include/hw/elf_ops.h

   1 static void glue(bswap_ehdr, SZ)(struct elfhdr *ehdr)
   2 {
   3     bswap16s(&ehdr->e_type);                    /* Object file type */
   4     bswap16s(&ehdr->e_machine);         /* Architecture */
   5     bswap32s(&ehdr->e_version);         /* Object file version */
   6     bswapSZs(&ehdr->e_entry);           /* Entry point virtual address */
   7     bswapSZs(&ehdr->e_phoff);           /* Program header table file offset */
   8     bswapSZs(&ehdr->e_shoff);           /* Section header table file offset */
   9     bswap32s(&ehdr->e_flags);           /* Processor-specific flags */
  10     bswap16s(&ehdr->e_ehsize);          /* ELF header size in bytes */
  11     bswap16s(&ehdr->e_phentsize);               /* Program header table entry size */
  12     bswap16s(&ehdr->e_phnum);           /* Program header table entry count */
  13     bswap16s(&ehdr->e_shentsize);               /* Section header table entry size */
  14     bswap16s(&ehdr->e_shnum);           /* Section header table entry count */
  15     bswap16s(&ehdr->e_shstrndx);                /* Section header string table index */
  16 }
  17
  18 static void glue(bswap_phdr, SZ)(struct elf_phdr *phdr)
  19 {
  20     bswap32s(&phdr->p_type);                    /* Segment type */
  21     bswapSZs(&phdr->p_offset);          /* Segment file offset */
  22     bswapSZs(&phdr->p_vaddr);           /* Segment virtual address */
  23     bswapSZs(&phdr->p_paddr);           /* Segment physical address */
  24     bswapSZs(&phdr->p_filesz);          /* Segment size in file */
  25     bswapSZs(&phdr->p_memsz);           /* Segment size in memory */
  26     bswap32s(&phdr->p_flags);           /* Segment flags */
  27     bswapSZs(&phdr->p_align);           /* Segment alignment */
  28 }
  29
  30 static void glue(bswap_shdr, SZ)(struct elf_shdr *shdr)
  31 {
  32     bswap32s(&shdr->sh_name);
  33     bswap32s(&shdr->sh_type);
  34     bswapSZs(&shdr->sh_flags);
  35     bswapSZs(&shdr->sh_addr);
  36     bswapSZs(&shdr->sh_offset);
  37     bswapSZs(&shdr->sh_size);
  38     bswap32s(&shdr->sh_link);
  39     bswap32s(&shdr->sh_info);
  40     bswapSZs(&shdr->sh_addralign);
  41     bswapSZs(&shdr->sh_entsize);
  42 }
  43
  44 static void glue(bswap_sym, SZ)(struct elf_sym *sym)
  45 {
  46     bswap32s(&sym->st_name);
  47     bswapSZs(&sym->st_value);
  48     bswapSZs(&sym->st_size);
  49     bswap16s(&sym->st_shndx);
  50 }
  51
  52 static void glue(bswap_rela, SZ)(struct elf_rela *rela)
  53 {
  54     bswapSZs(&rela->r_offset);
  55     bswapSZs(&rela->r_info);
  56     bswapSZs((elf_word *)&rela->r_addend);
  57 }
  58
  59 static struct elf_shdr *glue(find_section, SZ)(struct elf_shdr *shdr_table,
  60                                                int n, int type)
  61 {
  62     int i;
  63     for(i=0;i<n;i++) {
  64         if (shdr_table[i].sh_type == type)
  65             return shdr_table + i;
  66     }
  67     return NULL;
  68 }
  69
  70 static int glue(symfind, SZ)(const void *s0, const void *s1)
  71 {
  72     hwaddr addr = *(hwaddr *)s0;
  73     struct elf_sym *sym = (struct elf_sym *)s1;
  74     int result = 0;
  75     if (addr < sym->st_value) {
  76         result = -1;
  77     } else if (addr >= sym->st_value + sym->st_size) {
  78         result = 1;
  79     }
  80     return result;
  81 }
  82
  83 static const char *glue(lookup_symbol, SZ)(struct syminfo *s,
  84                                            hwaddr orig_addr)
  85 {
  86     struct elf_sym *syms = glue(s->disas_symtab.elf, SZ);
  87     struct elf_sym *sym;
  88
  89     sym = bsearch(&orig_addr, syms, s->disas_num_syms, sizeof(*syms),
  90                   glue(symfind, SZ));
  91     if (sym != NULL) {
  92         return s->disas_strtab + sym->st_name;
  93     }
  94
  95     return "";
  96 }
  97
  98 static int glue(symcmp, SZ)(const void *s0, const void *s1)
  99 {
 100     struct elf_sym *sym0 = (struct elf_sym *)s0;
 101     struct elf_sym *sym1 = (struct elf_sym *)s1;
 102     return (sym0->st_value < sym1->st_value)
 103         ? -1
 104         : ((sym0->st_value > sym1->st_value) ? 1 : 0);
 105 }
 106
 107 static int glue(load_symbols, SZ)(struct elfhdr *ehdr, int fd, int must_swab,
 108                                   int clear_lsb)
 109 {
 110     struct elf_shdr *symtab, *strtab, *shdr_table = NULL;
 111     struct elf_sym *syms = NULL;
 112     struct syminfo *s;
 113     int nsyms, i;
 114     char *str = NULL;
 115
 116     shdr_table = load_at(fd, ehdr->e_shoff,
 117                          sizeof(struct elf_shdr) * ehdr->e_shnum);
 118     if (!shdr_table)
 119         return -1;
 120
 121     if (must_swab) {
 122         for (i = 0; i < ehdr->e_shnum; i++) {
 123             glue(bswap_shdr, SZ)(shdr_table + i);
 124         }
 125     }
 126
 127     symtab = glue(find_section, SZ)(shdr_table, ehdr->e_shnum, SHT_SYMTAB);
 128     if (!symtab)
 129         goto fail;
 130     syms = load_at(fd, symtab->sh_offset, symtab->sh_size);
 131     if (!syms)
 132         goto fail;
 133
 134     nsyms = symtab->sh_size / sizeof(struct elf_sym);
 135
 136     i = 0;
 137     while (i < nsyms) {
 138         if (must_swab)
 139             glue(bswap_sym, SZ)(&syms[i]);
 140         /* We are only interested in function symbols.
 141            Throw everything else away.  */
 142         if (syms[i].st_shndx == SHN_UNDEF ||
 143                 syms[i].st_shndx >= SHN_LORESERVE ||
 144                 ELF_ST_TYPE(syms[i].st_info) != STT_FUNC) {
 145             nsyms--;
 146             if (i < nsyms) {
 147                 syms[i] = syms[nsyms];
 148             }
 149             continue;
 150         }
 151         if (clear_lsb) {
 152             /* The bottom address bit marks a Thumb or MIPS16 symbol.  */
 153             syms[i].st_value &= ~(glue(glue(Elf, SZ), _Addr))1;
 154         }
 155         i++;
 156     }
 157     syms = g_realloc(syms, nsyms * sizeof(*syms));
 158
 159     qsort(syms, nsyms, sizeof(*syms), glue(symcmp, SZ));
 160     for (i = 0; i < nsyms - 1; i++) {
 161         if (syms[i].st_size == 0) {
 162             syms[i].st_size = syms[i + 1].st_value - syms[i].st_value;
 163         }
 164     }
 165
 166     /* String table */
 167     if (symtab->sh_link >= ehdr->e_shnum)
 168         goto fail;
 169     strtab = &shdr_table[symtab->sh_link];
 170
 171     str = load_at(fd, strtab->sh_offset, strtab->sh_size);
 172     if (!str)
 173         goto fail;
 174
 175     /* Commit */
 176     s = g_malloc0(sizeof(*s));
 177     s->lookup_symbol = glue(lookup_symbol, SZ);
 178     glue(s->disas_symtab.elf, SZ) = syms;
 179     s->disas_num_syms = nsyms;
 180     s->disas_strtab = str;
 181     s->next = syminfos;
 182     syminfos = s;
 183     g_free(shdr_table);
 184     return 0;
 185  fail:
 186     g_free(syms);
 187     g_free(str);
 188     g_free(shdr_table);
 189     return -1;
 190 }
 191
 192 static int glue(elf_reloc, SZ)(struct elfhdr *ehdr, int fd, int must_swab,
 193                                uint64_t (*translate_fn)(void *, uint64_t),
 194                                void *translate_opaque, uint8_t *data,
 195                                struct elf_phdr *ph, int elf_machine)
 196 {
 197     struct elf_shdr *reltab, *shdr_table = NULL;
 198     struct elf_rela *rels = NULL;
 199     int nrels, i, ret = -1;
 200     elf_word wordval;
 201     void *addr;
 202
 203     shdr_table = load_at(fd, ehdr->e_shoff,
 204                          sizeof(struct elf_shdr) * ehdr->e_shnum);
 205     if (!shdr_table) {
 206         return -1;
 207     }
 208     if (must_swab) {
 209         for (i = 0; i < ehdr->e_shnum; i++) {
 210             glue(bswap_shdr, SZ)(&shdr_table[i]);
 211         }
 212     }
 213
 214     reltab = glue(find_section, SZ)(shdr_table, ehdr->e_shnum, SHT_RELA);
 215     if (!reltab) {
 216         goto fail;
 217     }
 218     rels = load_at(fd, reltab->sh_offset, reltab->sh_size);
 219     if (!rels) {
 220         goto fail;
 221     }
 222     nrels = reltab->sh_size / sizeof(struct elf_rela);
 223
 224     for (i = 0; i < nrels; i++) {
 225         if (must_swab) {
 226             glue(bswap_rela, SZ)(&rels[i]);
 227         }
 228         if (rels[i].r_offset < ph->p_vaddr ||
 229             rels[i].r_offset >= ph->p_vaddr + ph->p_filesz) {
 230             continue;
 231         }
 232         addr = &data[rels[i].r_offset - ph->p_vaddr];
 233         switch (elf_machine) {
 234         case EM_S390:
 235             switch (rels[i].r_info) {
 236             case R_390_RELATIVE:
 237                 wordval = *(elf_word *)addr;
 238                 if (must_swab) {
 239                     bswapSZs(&wordval);
 240                 }
 241                 wordval = translate_fn(translate_opaque, wordval);
 242                 if (must_swab) {
 243                     bswapSZs(&wordval);
 244                 }
 245                 *(elf_word *)addr = wordval;
 246                 break;
 247             default:
 248                 fprintf(stderr, "Unsupported relocation type %i!\n",
 249                         (int)rels[i].r_info);
 250             }
 251         }
 252     }
 253
 254     ret = 0;
 255 fail:
 256     g_free(rels);
 257     g_free(shdr_table);
 258     return ret;
 259 }
 260
 261 static int glue(load_elf, SZ)(const char *name, int fd,
 262                               uint64_t (*translate_fn)(void *, uint64_t),
 263                               void *translate_opaque,
 264                               int must_swab, uint64_t *pentry,
 265                               uint64_t *lowaddr, uint64_t *highaddr,
 266                               int elf_machine, int clear_lsb, int data_swab)
 267 {
 268     struct elfhdr ehdr;
 269     struct elf_phdr *phdr = NULL, *ph;
 270     int size, i, total_size;
 271     elf_word mem_size, file_size;
 272     uint64_t addr, low = (uint64_t)-1, high = 0;
 273     uint8_t *data = NULL;
 274     char label[128];
 275     int ret = ELF_LOAD_FAILED;
 276
 277     if (read(fd, &ehdr, sizeof(ehdr)) != sizeof(ehdr))
 278         goto fail;
 279     if (must_swab) {
 280         glue(bswap_ehdr, SZ)(&ehdr);
 281     }
 282
 283     switch (elf_machine) {
 284         case EM_PPC64:
 285             if (ehdr.e_machine != EM_PPC64) {
 286                 if (ehdr.e_machine != EM_PPC) {
 287                     ret = ELF_LOAD_WRONG_ARCH;
 288                     goto fail;
 289                 }
 290             }
 291             break;
 292         case EM_X86_64:
 293             if (ehdr.e_machine != EM_X86_64) {
 294                 if (ehdr.e_machine != EM_386) {
 295                     ret = ELF_LOAD_WRONG_ARCH;
 296                     goto fail;
 297                 }
 298             }
 299             break;
 300         case EM_MICROBLAZE:
 301             if (ehdr.e_machine != EM_MICROBLAZE) {
 302                 if (ehdr.e_machine != EM_MICROBLAZE_OLD) {
 303                     ret = ELF_LOAD_WRONG_ARCH;
 304                     goto fail;
 305                 }
 306             }
 307             break;
 308         case EM_MOXIE:
 309             if (ehdr.e_machine != EM_MOXIE) {
 310                 if (ehdr.e_machine != EM_MOXIE_OLD) {
 311                     ret = ELF_LOAD_WRONG_ARCH;
 312                     goto fail;
 313                 }
 314             }
 315             break;
 316         default:
 317             if (elf_machine != ehdr.e_machine) {
 318                 ret = ELF_LOAD_WRONG_ARCH;
 319                 goto fail;
 320             }
 321     }
 322
 323     if (pentry)
 324         *pentry = (uint64_t)(elf_sword)ehdr.e_entry;
 325
 326     glue(load_symbols, SZ)(&ehdr, fd, must_swab, clear_lsb);
 327
 328     size = ehdr.e_phnum * sizeof(phdr[0]);
 329     if (lseek(fd, ehdr.e_phoff, SEEK_SET) != ehdr.e_phoff) {
 330         goto fail;
 331     }
 332     phdr = g_malloc0(size);
 333     if (!phdr)
 334         goto fail;
 335     if (read(fd, phdr, size) != size)
 336         goto fail;
 337     if (must_swab) {
 338         for(i = 0; i < ehdr.e_phnum; i++) {
 339             ph = &phdr[i];
 340             glue(bswap_phdr, SZ)(ph);
 341         }
 342     }
 343
 344     total_size = 0;
 345     for(i = 0; i < ehdr.e_phnum; i++) {
 346         ph = &phdr[i];
 347         if (ph->p_type == PT_LOAD) {
 348             mem_size = ph->p_memsz; /* Size of the ROM */
 349             file_size = ph->p_filesz; /* Size of the allocated data */
 350             data = g_malloc0(file_size);
 351             if (ph->p_filesz > 0) {
 352                 if (lseek(fd, ph->p_offset, SEEK_SET) < 0) {
 353                     goto fail;
 354                 }
 355                 if (read(fd, data, file_size) != file_size) {
 356                     goto fail;
 357                 }
 358             }
 359             /* address_offset is hack for kernel images that are
 360                linked at the wrong physical address.  */
 361             if (translate_fn) {
 362                 addr = translate_fn(translate_opaque, ph->p_paddr);
 363                 glue(elf_reloc, SZ)(&ehdr, fd, must_swab,  translate_fn,
 364                                     translate_opaque, data, ph, elf_machine);
 365             } else {
 366                 addr = ph->p_paddr;
 367             }
 368
 369             if (data_swab) {
 370                 int j;
 371                 for (j = 0; j < file_size; j += (1 << data_swab)) {
 372                     uint8_t *dp = data + j;
 373                     switch (data_swab) {
 374                     case (1):
 375                         *(uint16_t *)dp = bswap16(*(uint16_t *)dp);
 376                         break;
 377                     case (2):
 378                         *(uint32_t *)dp = bswap32(*(uint32_t *)dp);
 379                         break;
 380                     case (3):
 381                         *(uint64_t *)dp = bswap64(*(uint64_t *)dp);
 382                         break;
 383                     default:
 384                         g_assert_not_reached();
 385                     }
 386                 }
 387             }
 388
 389             /* the entry pointer in the ELF header is a virtual
 390              * address, if the text segments paddr and vaddr differ
 391              * we need to adjust the entry */
 392             if (pentry && !translate_fn &&
 393                     ph->p_vaddr != ph->p_paddr &&
 394                     ehdr.e_entry >= ph->p_vaddr &&
 395                     ehdr.e_entry < ph->p_vaddr + ph->p_filesz &&
 396                     ph->p_flags & PF_X) {
 397                 *pentry = ehdr.e_entry - ph->p_vaddr + ph->p_paddr;
 398             }
 399
 400             snprintf(label, sizeof(label), "phdr #%d: %s", i, name);
 401
 402             /* rom_add_elf_program() seize the ownership of 'data' */
 403             rom_add_elf_program(label, data, file_size, mem_size, addr);
 404
 405             total_size += mem_size;
 406             if (addr < low)
 407                 low = addr;
 408             if ((addr + mem_size) > high)
 409                 high = addr + mem_size;
 410
 411             data = NULL;
 412         }
 413     }
 414     g_free(phdr);
 415     if (lowaddr)
 416         *lowaddr = (uint64_t)(elf_sword)low;
 417     if (highaddr)
 418         *highaddr = (uint64_t)(elf_sword)high;
 419     return total_size;
 420  fail:
 421     g_free(data);
 422     g_free(phdr);
 423     return ret;
 424 }