[qemu.git] / hw / elf_ops.h

static void glue(bswap_ehdr, SZ)(struct elfhdr *ehdr)
{
    bswap16s(&ehdr->e_type);			/* Object file type */
    bswap16s(&ehdr->e_machine);		/* Architecture */
    bswap32s(&ehdr->e_version);		/* Object file version */
    bswapSZs(&ehdr->e_entry);		/* Entry point virtual address */
    bswapSZs(&ehdr->e_phoff);		/* Program header table file offset */
    bswapSZs(&ehdr->e_shoff);		/* Section header table file offset */
    bswap32s(&ehdr->e_flags);		/* Processor-specific flags */
    bswap16s(&ehdr->e_ehsize);		/* ELF header size in bytes */
    bswap16s(&ehdr->e_phentsize);		/* Program header table entry size */
    bswap16s(&ehdr->e_phnum);		/* Program header table entry count */
    bswap16s(&ehdr->e_shentsize);		/* Section header table entry size */
    bswap16s(&ehdr->e_shnum);		/* Section header table entry count */
    bswap16s(&ehdr->e_shstrndx);		/* Section header string table index */
}

static void glue(bswap_phdr, SZ)(struct elf_phdr *phdr)
{
    bswap32s(&phdr->p_type);			/* Segment type */
    bswapSZs(&phdr->p_offset);		/* Segment file offset */
    bswapSZs(&phdr->p_vaddr);		/* Segment virtual address */
    bswapSZs(&phdr->p_paddr);		/* Segment physical address */
    bswapSZs(&phdr->p_filesz);		/* Segment size in file */
    bswapSZs(&phdr->p_memsz);		/* Segment size in memory */
    bswap32s(&phdr->p_flags);		/* Segment flags */
    bswapSZs(&phdr->p_align);		/* Segment alignment */
}

static void glue(bswap_shdr, SZ)(struct elf_shdr *shdr)
{
    bswap32s(&shdr->sh_name);
    bswap32s(&shdr->sh_type);
    bswapSZs(&shdr->sh_flags);
    bswapSZs(&shdr->sh_addr);
    bswapSZs(&shdr->sh_offset);
    bswapSZs(&shdr->sh_size);
    bswap32s(&shdr->sh_link);
    bswap32s(&shdr->sh_info);
    bswapSZs(&shdr->sh_addralign);
    bswapSZs(&shdr->sh_entsize);
}

static void glue(bswap_sym, SZ)(struct elf_sym *sym)
{
    bswap32s(&sym->st_name);
    bswapSZs(&sym->st_value);
    bswapSZs(&sym->st_size);
    bswap16s(&sym->st_shndx);
}

static struct elf_shdr *glue(find_section, SZ)(struct elf_shdr *shdr_table,
                                               int n, int type)
{
    int i;
    for(i=0;i<n;i++) {
        if (shdr_table[i].sh_type == type)
            return shdr_table + i;
    }
    return NULL;
}

static int glue(symfind, SZ)(const void *s0, const void *s1)
{
    struct elf_sym *key = (struct elf_sym *)s0;
    struct elf_sym *sym = (struct elf_sym *)s1;
    int result = 0;
    if (key->st_value < sym->st_value) {
        result = -1;
    } else if (key->st_value >= sym->st_value + sym->st_size) {
        result = 1;
    }
    return result;
}

static const char *glue(lookup_symbol, SZ)(struct syminfo *s,
                                           target_phys_addr_t orig_addr)
{
    struct elf_sym *syms = glue(s->disas_symtab.elf, SZ);
    struct elf_sym key;
    struct elf_sym *sym;

    key.st_value = orig_addr;

    sym = bsearch(&key, syms, s->disas_num_syms, sizeof(*syms), glue(symfind, SZ));
    if (sym != NULL) {
        return s->disas_strtab + sym->st_name;
    }

    return "";
}

static int glue(symcmp, SZ)(const void *s0, const void *s1)
{
    struct elf_sym *sym0 = (struct elf_sym *)s0;
    struct elf_sym *sym1 = (struct elf_sym *)s1;
    return (sym0->st_value < sym1->st_value)
        ? -1
        : ((sym0->st_value > sym1->st_value) ? 1 : 0);
}

static int glue(load_symbols, SZ)(struct elfhdr *ehdr, int fd, int must_swab,
                                  int clear_lsb)
{
    struct elf_shdr *symtab, *strtab, *shdr_table = NULL;
    struct elf_sym *syms = NULL;
    struct syminfo *s;
    int nsyms, i;
    char *str = NULL;

    shdr_table = load_at(fd, ehdr->e_shoff,
                         sizeof(struct elf_shdr) * ehdr->e_shnum);
    if (!shdr_table)
        return -1;

    if (must_swab) {
        for (i = 0; i < ehdr->e_shnum; i++) {
            glue(bswap_shdr, SZ)(shdr_table + i);
        }
    }

    symtab = glue(find_section, SZ)(shdr_table, ehdr->e_shnum, SHT_SYMTAB);
    if (!symtab)
        goto fail;
    syms = load_at(fd, symtab->sh_offset, symtab->sh_size);
    if (!syms)
        goto fail;

    nsyms = symtab->sh_size / sizeof(struct elf_sym);

    i = 0;
    while (i < nsyms) {
        if (must_swab)
            glue(bswap_sym, SZ)(&syms[i]);
        /* We are only interested in function symbols.
           Throw everything else away.  */
        if (syms[i].st_shndx == SHN_UNDEF ||
                syms[i].st_shndx >= SHN_LORESERVE ||
                ELF_ST_TYPE(syms[i].st_info) != STT_FUNC) {
            nsyms--;
            if (i < nsyms) {
                syms[i] = syms[nsyms];
            }
            continue;
        }
        if (clear_lsb) {
            /* The bottom address bit marks a Thumb or MIPS16 symbol.  */
            syms[i].st_value &= ~(glue(glue(Elf, SZ), _Addr))1;
        }
        i++;
    }
    if (nsyms) {
        syms = g_realloc(syms, nsyms * sizeof(*syms));

        qsort(syms, nsyms, sizeof(*syms), glue(symcmp, SZ));
        for (i = 0; i < nsyms - 1; i++) {
            if (syms[i].st_size == 0) {
                syms[i].st_size = syms[i + 1].st_value - syms[i].st_value;
            }
        }
    } else {
        g_free(syms);
        syms = NULL;
    }

    /* String table */
    if (symtab->sh_link >= ehdr->e_shnum)
        goto fail;
    strtab = &shdr_table[symtab->sh_link];

    str = load_at(fd, strtab->sh_offset, strtab->sh_size);
    if (!str)
        goto fail;

    /* Commit */
    s = g_malloc0(sizeof(*s));
    s->lookup_symbol = glue(lookup_symbol, SZ);
    glue(s->disas_symtab.elf, SZ) = syms;
    s->disas_num_syms = nsyms;
    s->disas_strtab = str;
    s->next = syminfos;
    syminfos = s;
    g_free(shdr_table);
    return 0;
 fail:
    g_free(syms);
    g_free(str);
    g_free(shdr_table);
    return -1;
}

static int glue(load_elf, SZ)(const char *name, int fd,
                              uint64_t (*translate_fn)(void *, uint64_t),
                              void *translate_opaque,
                              int must_swab, uint64_t *pentry,
                              uint64_t *lowaddr, uint64_t *highaddr,
                              int elf_machine, int clear_lsb)
{
    struct elfhdr ehdr;
    struct elf_phdr *phdr = NULL, *ph;
    int size, i, total_size;
    elf_word mem_size;
    uint64_t addr, low = (uint64_t)-1, high = 0;
    uint8_t *data = NULL;
    char label[128];

    if (read(fd, &ehdr, sizeof(ehdr)) != sizeof(ehdr))
        goto fail;
    if (must_swab) {
        glue(bswap_ehdr, SZ)(&ehdr);
    }

    switch (elf_machine) {
        case EM_PPC64:
            if (EM_PPC64 != ehdr.e_machine)
                if (EM_PPC != ehdr.e_machine)
                    goto fail;
            break;
        case EM_X86_64:
            if (EM_X86_64 != ehdr.e_machine)
                if (EM_386 != ehdr.e_machine)
                    goto fail;
            break;
        case EM_MICROBLAZE:
            if (EM_MICROBLAZE != ehdr.e_machine)
                if (EM_MICROBLAZE_OLD != ehdr.e_machine)
                    goto fail;
            break;
        default:
            if (elf_machine != ehdr.e_machine)
                goto fail;
    }

    if (pentry)
   	*pentry = (uint64_t)(elf_sword)ehdr.e_entry;

    glue(load_symbols, SZ)(&ehdr, fd, must_swab, clear_lsb);

    size = ehdr.e_phnum * sizeof(phdr[0]);
    lseek(fd, ehdr.e_phoff, SEEK_SET);
    phdr = g_malloc0(size);
    if (!phdr)
        goto fail;
    if (read(fd, phdr, size) != size)
        goto fail;
    if (must_swab) {
        for(i = 0; i < ehdr.e_phnum; i++) {
            ph = &phdr[i];
            glue(bswap_phdr, SZ)(ph);
        }
    }

    total_size = 0;
    for(i = 0; i < ehdr.e_phnum; i++) {
        ph = &phdr[i];
        if (ph->p_type == PT_LOAD) {
            mem_size = ph->p_memsz;
            /* XXX: avoid allocating */
            data = g_malloc0(mem_size);
            if (ph->p_filesz > 0) {
                if (lseek(fd, ph->p_offset, SEEK_SET) < 0)
                    goto fail;
                if (read(fd, data, ph->p_filesz) != ph->p_filesz)
                    goto fail;
            }
            /* address_offset is hack for kernel images that are
               linked at the wrong physical address.  */
            if (translate_fn) {
                addr = translate_fn(translate_opaque, ph->p_paddr);
            } else {
                addr = ph->p_paddr;
            }

            snprintf(label, sizeof(label), "phdr #%d: %s", i, name);
            rom_add_blob_fixed(label, data, mem_size, addr);

            total_size += mem_size;
            if (addr < low)
                low = addr;
            if ((addr + mem_size) > high)
                high = addr + mem_size;

            g_free(data);
            data = NULL;
        }
    }
    g_free(phdr);
    if (lowaddr)
        *lowaddr = (uint64_t)(elf_sword)low;
    if (highaddr)
        *highaddr = (uint64_t)(elf_sword)high;
    return total_size;
 fail:
    g_free(data);
    g_free(phdr);
    return -1;
}
Commit	Line	Data
5fe141fd FB	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
5fafdf24	52	static struct elf_shdr glue(find_section, SZ)(struct elf_shdr shdr_table,
5fe141fd FB	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
49918a75 PB	63	static int glue(symfind, SZ)(const void s0, const void s1)
	64	{
	65	struct elf_sym key = (struct elf_sym )s0;
	66	struct elf_sym sym = (struct elf_sym )s1;
	67	int result = 0;
	68	if (key->st_value < sym->st_value) {
	69	result = -1;
ec822001	70	} else if (key->st_value >= sym->st_value + sym->st_size) {
49918a75 PB	71	result = 1;
	72	}
	73	return result;
	74	}
	75
ca20cf32	76	static const char glue(lookup_symbol, SZ)(struct syminfo s,
c227f099	77	target_phys_addr_t orig_addr)
49918a75 PB	78	{
	79	struct elf_sym *syms = glue(s->disas_symtab.elf, SZ);
	80	struct elf_sym key;
	81	struct elf_sym *sym;
	82
	83	key.st_value = orig_addr;
	84
	85	sym = bsearch(&key, syms, s->disas_num_syms, sizeof(*syms), glue(symfind, SZ));
660f11be	86	if (sym != NULL) {
49918a75 PB	87	return s->disas_strtab + sym->st_name;
	88	}
	89
	90	return "";
	91	}
	92
	93	static int glue(symcmp, SZ)(const void s0, const void s1)
	94	{
	95	struct elf_sym sym0 = (struct elf_sym )s0;
	96	struct elf_sym sym1 = (struct elf_sym )s1;
	97	return (sym0->st_value < sym1->st_value)
	98	? -1
	99	: ((sym0->st_value > sym1->st_value) ? 1 : 0);
	100	}
	101
ca20cf32 BS	102	static int glue(load_symbols, SZ)(struct elfhdr *ehdr, int fd, int must_swab,
ca20cf32 BS	103	int clear_lsb)
5fe141fd FB	104	{
	105	struct elf_shdr symtab, strtab, *shdr_table = NULL;
	106	struct elf_sym *syms = NULL;
5fe141fd FB	107	struct syminfo *s;
	108	int nsyms, i;
	109	char *str = NULL;
	110
5fafdf24	111	shdr_table = load_at(fd, ehdr->e_shoff,
5fe141fd FB	112	sizeof(struct elf_shdr) * ehdr->e_shnum);
	113	if (!shdr_table)
	114	return -1;
3b46e624	115
5fe141fd FB	116	if (must_swab) {
	117	for (i = 0; i < ehdr->e_shnum; i++) {
	118	glue(bswap_shdr, SZ)(shdr_table + i);
	119	}
	120	}
3b46e624	121
5fe141fd FB	122	symtab = glue(find_section, SZ)(shdr_table, ehdr->e_shnum, SHT_SYMTAB);
	123	if (!symtab)
	124	goto fail;
	125	syms = load_at(fd, symtab->sh_offset, symtab->sh_size);
	126	if (!syms)
	127	goto fail;
	128
	129	nsyms = symtab->sh_size / sizeof(struct elf_sym);
49918a75 PB	130
	131	i = 0;
	132	while (i < nsyms) {
5fe141fd FB	133	if (must_swab)
5fe141fd FB	134	glue(bswap_sym, SZ)(&syms[i]);
49918a75 PB	135	/* We are only interested in function symbols.
	136	Throw everything else away. */
	137	if (syms[i].st_shndx == SHN_UNDEF \|\|
	138	syms[i].st_shndx >= SHN_LORESERVE \|\|
	139	ELF_ST_TYPE(syms[i].st_info) != STT_FUNC) {
	140	nsyms--;
	141	if (i < nsyms) {
	142	syms[i] = syms[nsyms];
	143	}
	144	continue;
	145	}
ca20cf32 BS	146	if (clear_lsb) {
	147	/* The bottom address bit marks a Thumb or MIPS16 symbol. */
	148	syms[i].st_value &= ~(glue(glue(Elf, SZ), _Addr))1;
	149	}
49918a75	150	i++;
5fe141fd	151	}
3e372cf8	152	if (nsyms) {
7267c094	153	syms = g_realloc(syms, nsyms * sizeof(*syms));
49918a75	154
3e372cf8	155	qsort(syms, nsyms, sizeof(*syms), glue(symcmp, SZ));
e403e433 SW	156	for (i = 0; i < nsyms - 1; i++) {
	157	if (syms[i].st_size == 0) {
	158	syms[i].st_size = syms[i + 1].st_value - syms[i].st_value;
	159	}
	160	}
3e372cf8	161	} else {
7267c094	162	g_free(syms);
3e372cf8 AJ	163	syms = NULL;
3e372cf8 AJ	164	}
49918a75	165
5fe141fd FB	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)
49918a75	173	goto fail;
5fe141fd FB	174
5fe141fd FB	175	/* Commit */
7267c094	176	s = g_malloc0(sizeof(*s));
49918a75 PB	177	s->lookup_symbol = glue(lookup_symbol, SZ);
49918a75 PB	178	glue(s->disas_symtab.elf, SZ) = syms;
5fe141fd FB	179	s->disas_num_syms = nsyms;
	180	s->disas_strtab = str;
	181	s->next = syminfos;
	182	syminfos = s;
7267c094	183	g_free(shdr_table);
5fe141fd FB	184	return 0;
5fe141fd FB	185	fail:
7267c094 AL	186	g_free(syms);
	187	g_free(str);
	188	g_free(shdr_table);
5fe141fd FB	189	return -1;
	190	}
	191
409dbce5 AJ	192	static int glue(load_elf, SZ)(const char *name, int fd,
	193	uint64_t (translate_fn)(void , uint64_t),
	194	void *translate_opaque,
9596ebb7	195	int must_swab, uint64_t *pentry,
ca20cf32 BS	196	uint64_t lowaddr, uint64_t highaddr,
ca20cf32 BS	197	int elf_machine, int clear_lsb)
5fe141fd FB	198	{
	199	struct elfhdr ehdr;
	200	struct elf_phdr phdr = NULL, ph;
	201	int size, i, total_size;
9437454a	202	elf_word mem_size;
fd93a799	203	uint64_t addr, low = (uint64_t)-1, high = 0;
9ee3c029	204	uint8_t *data = NULL;
45a50b16	205	char label[128];
5fe141fd FB	206
	207	if (read(fd, &ehdr, sizeof(ehdr)) != sizeof(ehdr))
	208	goto fail;
	209	if (must_swab) {
	210	glue(bswap_ehdr, SZ)(&ehdr);
	211	}
	212
ca20cf32	213	switch (elf_machine) {
7f70c937 BS	214	case EM_PPC64:
	215	if (EM_PPC64 != ehdr.e_machine)
	216	if (EM_PPC != ehdr.e_machine)
	217	goto fail;
	218	break;
	219	case EM_X86_64:
	220	if (EM_X86_64 != ehdr.e_machine)
	221	if (EM_386 != ehdr.e_machine)
	222	goto fail;
	223	break;
16f04416 EI	224	case EM_MICROBLAZE:
	225	if (EM_MICROBLAZE != ehdr.e_machine)
	226	if (EM_MICROBLAZE_OLD != ehdr.e_machine)
	227	goto fail;
	228	break;
7f70c937	229	default:
ca20cf32	230	if (elf_machine != ehdr.e_machine)
7f70c937 BS	231	goto fail;
7f70c937 BS	232	}
9042c0e2	233
9ee3c029	234	if (pentry)
82790064	235	*pentry = (uint64_t)(elf_sword)ehdr.e_entry;
9ee3c029	236
ca20cf32	237	glue(load_symbols, SZ)(&ehdr, fd, must_swab, clear_lsb);
5fe141fd FB	238
	239	size = ehdr.e_phnum * sizeof(phdr[0]);
	240	lseek(fd, ehdr.e_phoff, SEEK_SET);
7267c094	241	phdr = g_malloc0(size);
5fe141fd FB	242	if (!phdr)
	243	goto fail;
	244	if (read(fd, phdr, size) != size)
04d4b0c3	245	goto fail;
5fe141fd FB	246	if (must_swab) {
	247	for(i = 0; i < ehdr.e_phnum; i++) {
	248	ph = &phdr[i];
	249	glue(bswap_phdr, SZ)(ph);
	250	}
	251	}
3b46e624	252
5fe141fd FB	253	total_size = 0;
	254	for(i = 0; i < ehdr.e_phnum; i++) {
	255	ph = &phdr[i];
	256	if (ph->p_type == PT_LOAD) {
	257	mem_size = ph->p_memsz;
	258	/* XXX: avoid allocating */
7267c094	259	data = g_malloc0(mem_size);
5fe141fd	260	if (ph->p_filesz > 0) {
9ee3c029	261	if (lseek(fd, ph->p_offset, SEEK_SET) < 0)
04d4b0c3	262	goto fail;
5fe141fd	263	if (read(fd, data, ph->p_filesz) != ph->p_filesz)
04d4b0c3	264	goto fail;
5fe141fd	265	}
83c1f87c PB	266	/* address_offset is hack for kernel images that are
83c1f87c PB	267	linked at the wrong physical address. */
409dbce5 AJ	268	if (translate_fn) {
	269	addr = translate_fn(translate_opaque, ph->p_paddr);
	270	} else {
	271	addr = ph->p_paddr;
	272	}
5fe141fd	273
45a50b16 GH	274	snprintf(label, sizeof(label), "phdr #%d: %s", i, name);
45a50b16 GH	275	rom_add_blob_fixed(label, data, mem_size, addr);
5fe141fd FB	276
5fe141fd FB	277	total_size += mem_size;
fd93a799	278	if (addr < low)
74287114	279	low = addr;
fd93a799	280	if ((addr + mem_size) > high)
74287114	281	high = addr + mem_size;
5fe141fd	282
7267c094	283	g_free(data);
9ee3c029	284	data = NULL;
5fe141fd FB	285	}
5fe141fd FB	286	}
7267c094	287	g_free(phdr);
74287114	288	if (lowaddr)
82790064	289	*lowaddr = (uint64_t)(elf_sword)low;
74287114	290	if (highaddr)
82790064	291	*highaddr = (uint64_t)(elf_sword)high;
5fe141fd	292	return total_size;
04d4b0c3	293	fail:
7267c094 AL	294	g_free(data);
7267c094 AL	295	g_free(phdr);
5fe141fd FB	296	return -1;
5fe141fd FB	297	}