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 struct elf_shdr *glue(find_section, SZ)(struct elf_shdr *shdr_table,
  53                                                int n, int type)
  54 {
  55     int i;
  56     for(i=0;i<n;i++) {
  57         if (shdr_table[i].sh_type == type)
  58             return shdr_table + i;
  59     }
  60     return NULL;
  61 }
  62
  63 static int glue(symfind, SZ)(const void *s0, const void *s1)
  64 {
  65     hwaddr addr = *(hwaddr *)s0;
  66     struct elf_sym *sym = (struct elf_sym *)s1;
  67     int result = 0;
  68     if (addr < sym->st_value) {
  69         result = -1;
  70     } else if (addr >= sym->st_value + sym->st_size) {
  71         result = 1;
  72     }
  73     return result;
  74 }
  75
  76 static const char *glue(lookup_symbol, SZ)(struct syminfo *s,
  77                                            hwaddr orig_addr)
  78 {
  79     struct elf_sym *syms = glue(s->disas_symtab.elf, SZ);
  80     struct elf_sym *sym;
  81
  82     sym = bsearch(&orig_addr, syms, s->disas_num_syms, sizeof(*syms),
  83                   glue(symfind, SZ));
  84     if (sym != NULL) {
  85         return s->disas_strtab + sym->st_name;
  86     }
  87
  88     return "";
  89 }
  90
  91 static int glue(symcmp, SZ)(const void *s0, const void *s1)
  92 {
  93     struct elf_sym *sym0 = (struct elf_sym *)s0;
  94     struct elf_sym *sym1 = (struct elf_sym *)s1;
  95     return (sym0->st_value < sym1->st_value)
  96         ? -1
  97         : ((sym0->st_value > sym1->st_value) ? 1 : 0);
  98 }
  99
 100 static int glue(load_symbols, SZ)(struct elfhdr *ehdr, int fd, int must_swab,
 101                                   int clear_lsb)
 102 {
 103     struct elf_shdr *symtab, *strtab, *shdr_table = NULL;
 104     struct elf_sym *syms = NULL;
 105     struct syminfo *s;
 106     int nsyms, i;
 107     char *str = NULL;
 108
 109     shdr_table = load_at(fd, ehdr->e_shoff,
 110                          sizeof(struct elf_shdr) * ehdr->e_shnum);
 111     if (!shdr_table)
 112         return -1;
 113
 114     if (must_swab) {
 115         for (i = 0; i < ehdr->e_shnum; i++) {
 116             glue(bswap_shdr, SZ)(shdr_table + i);
 117         }
 118     }
 119
 120     symtab = glue(find_section, SZ)(shdr_table, ehdr->e_shnum, SHT_SYMTAB);
 121     if (!symtab)
 122         goto fail;
 123     syms = load_at(fd, symtab->sh_offset, symtab->sh_size);
 124     if (!syms)
 125         goto fail;
 126
 127     nsyms = symtab->sh_size / sizeof(struct elf_sym);
 128
 129     i = 0;
 130     while (i < nsyms) {
 131         if (must_swab)
 132             glue(bswap_sym, SZ)(&syms[i]);
 133         /* We are only interested in function symbols.
 134            Throw everything else away.  */
 135         if (syms[i].st_shndx == SHN_UNDEF ||
 136                 syms[i].st_shndx >= SHN_LORESERVE ||
 137                 ELF_ST_TYPE(syms[i].st_info) != STT_FUNC) {
 138             nsyms--;
 139             if (i < nsyms) {
 140                 syms[i] = syms[nsyms];
 141             }
 142             continue;
 143         }
 144         if (clear_lsb) {
 145             /* The bottom address bit marks a Thumb or MIPS16 symbol.  */
 146             syms[i].st_value &= ~(glue(glue(Elf, SZ), _Addr))1;
 147         }
 148         i++;
 149     }
 150     if (nsyms) {
 151         syms = g_realloc(syms, nsyms * sizeof(*syms));
 152
 153         qsort(syms, nsyms, sizeof(*syms), glue(symcmp, SZ));
 154         for (i = 0; i < nsyms - 1; i++) {
 155             if (syms[i].st_size == 0) {
 156                 syms[i].st_size = syms[i + 1].st_value - syms[i].st_value;
 157             }
 158         }
 159     } else {
 160         g_free(syms);
 161         syms = NULL;
 162     }
 163
 164     /* String table */
 165     if (symtab->sh_link >= ehdr->e_shnum)
 166         goto fail;
 167     strtab = &shdr_table[symtab->sh_link];
 168
 169     str = load_at(fd, strtab->sh_offset, strtab->sh_size);
 170     if (!str)
 171         goto fail;
 172
 173     /* Commit */
 174     s = g_malloc0(sizeof(*s));
 175     s->lookup_symbol = glue(lookup_symbol, SZ);
 176     glue(s->disas_symtab.elf, SZ) = syms;
 177     s->disas_num_syms = nsyms;
 178     s->disas_strtab = str;
 179     s->next = syminfos;
 180     syminfos = s;
 181     g_free(shdr_table);
 182     return 0;
 183  fail:
 184     g_free(syms);
 185     g_free(str);
 186     g_free(shdr_table);
 187     return -1;
 188 }
 189
 190 static int glue(load_elf, SZ)(const char *name, int fd,
 191                               uint64_t (*translate_fn)(void *, uint64_t),
 192                               void *translate_opaque,
 193                               int must_swab, uint64_t *pentry,
 194                               uint64_t *lowaddr, uint64_t *highaddr,
 195                               int elf_machine, int clear_lsb)
 196 {
 197     struct elfhdr ehdr;
 198     struct elf_phdr *phdr = NULL, *ph;
 199     int size, i, total_size;
 200     elf_word mem_size, file_size;
 201     uint64_t addr, low = (uint64_t)-1, high = 0;
 202     uint8_t *data = NULL;
 203     char label[128];
 204     int ret = ELF_LOAD_FAILED;
 205
 206     if (read(fd, &ehdr, sizeof(ehdr)) != sizeof(ehdr))
 207         goto fail;
 208     if (must_swab) {
 209         glue(bswap_ehdr, SZ)(&ehdr);
 210     }
 211
 212     switch (elf_machine) {
 213         case EM_PPC64:
 214             if (EM_PPC64 != ehdr.e_machine)
 215                 if (EM_PPC != ehdr.e_machine) {
 216                     ret = ELF_LOAD_WRONG_ARCH;
 217                     goto fail;
 218                 }
 219             break;
 220         case EM_X86_64:
 221             if (EM_X86_64 != ehdr.e_machine)
 222                 if (EM_386 != ehdr.e_machine) {
 223                     ret = ELF_LOAD_WRONG_ARCH;
 224                     goto fail;
 225                 }
 226             break;
 227         case EM_MICROBLAZE:
 228             if (EM_MICROBLAZE != ehdr.e_machine)
 229                 if (EM_MICROBLAZE_OLD != ehdr.e_machine) {
 230                     ret = ELF_LOAD_WRONG_ARCH;
 231                     goto fail;
 232                 }
 233             break;
 234         default:
 235             if (elf_machine != ehdr.e_machine) {
 236                 ret = ELF_LOAD_WRONG_ARCH;
 237                 goto fail;
 238             }
 239     }
 240
 241     if (pentry)
 242         *pentry = (uint64_t)(elf_sword)ehdr.e_entry;
 243
 244     glue(load_symbols, SZ)(&ehdr, fd, must_swab, clear_lsb);
 245
 246     size = ehdr.e_phnum * sizeof(phdr[0]);
 247     lseek(fd, ehdr.e_phoff, SEEK_SET);
 248     phdr = g_malloc0(size);
 249     if (!phdr)
 250         goto fail;
 251     if (read(fd, phdr, size) != size)
 252         goto fail;
 253     if (must_swab) {
 254         for(i = 0; i < ehdr.e_phnum; i++) {
 255             ph = &phdr[i];
 256             glue(bswap_phdr, SZ)(ph);
 257         }
 258     }
 259
 260     total_size = 0;
 261     for(i = 0; i < ehdr.e_phnum; i++) {
 262         ph = &phdr[i];
 263         if (ph->p_type == PT_LOAD) {
 264             mem_size = ph->p_memsz; /* Size of the ROM */
 265             file_size = ph->p_filesz; /* Size of the allocated data */
 266             data = g_malloc0(file_size);
 267             if (ph->p_filesz > 0) {
 268                 if (lseek(fd, ph->p_offset, SEEK_SET) < 0) {
 269                     goto fail;
 270                 }
 271                 if (read(fd, data, file_size) != file_size) {
 272                     goto fail;
 273                 }
 274             }
 275             /* address_offset is hack for kernel images that are
 276                linked at the wrong physical address.  */
 277             if (translate_fn) {
 278                 addr = translate_fn(translate_opaque, ph->p_paddr);
 279             } else {
 280                 addr = ph->p_paddr;
 281             }
 282
 283             /* the entry pointer in the ELF header is a virtual
 284              * address, if the text segments paddr and vaddr differ
 285              * we need to adjust the entry */
 286             if (pentry && !translate_fn &&
 287                     ph->p_vaddr != ph->p_paddr &&
 288                     ehdr.e_entry >= ph->p_vaddr &&
 289                     ehdr.e_entry < ph->p_vaddr + ph->p_filesz &&
 290                     ph->p_flags & PF_X) {
 291                 *pentry = ehdr.e_entry - ph->p_vaddr + ph->p_paddr;
 292             }
 293
 294             snprintf(label, sizeof(label), "phdr #%d: %s", i, name);
 295
 296             /* rom_add_elf_program() seize the ownership of 'data' */
 297             rom_add_elf_program(label, data, file_size, mem_size, addr);
 298
 299             total_size += mem_size;
 300             if (addr < low)
 301                 low = addr;
 302             if ((addr + mem_size) > high)
 303                 high = addr + mem_size;
 304
 305             data = NULL;
 306         }
 307     }
 308     g_free(phdr);
 309     if (lowaddr)
 310         *lowaddr = (uint64_t)(elf_sword)low;
 311     if (highaddr)
 312         *highaddr = (uint64_t)(elf_sword)high;
 313     return total_size;
 314  fail:
 315     g_free(data);
 316     g_free(phdr);
 317     return ret;
 318 }