#pragma ident "@(#)getdata.c 1.23 08/05/31 SMI"
#include <stdlib.h>
#include <assert.h>
#include <errno.h>
#include <libelf.h>
#include "decl.h"
#include "msg.h"
static const size_t align32[ELF_T_NUM] =
{
1,
sizeof (Elf32),
sizeof (Elf32),
sizeof (Elf32),
sizeof (Elf32_Half),
sizeof (Elf32),
sizeof (Elf32),
sizeof (Elf32),
sizeof (Elf32),
sizeof (Elf32),
sizeof (Elf32),
sizeof (Elf32),
sizeof (Elf32),
sizeof (Elf32),
sizeof (Elf32),
sizeof (Elf64_Sxword),
sizeof (Elf64),
sizeof (Elf32_Half),
sizeof (Elf32),
sizeof (Elf32_Lword),
sizeof (Elf32_Lword),
sizeof (Elf32_Word)
};
#define Nalign32 (sizeof (align32)/sizeof (align32[0]))
static const size_t align64[ELF_T_NUM] =
{
1,
sizeof (Elf64),
sizeof (Elf64),
sizeof (Elf64),
sizeof (Elf64_Half),
sizeof (Elf64),
sizeof (Elf64),
sizeof (Elf64),
sizeof (Elf64),
sizeof (Elf64),
sizeof (Elf64_Word),
sizeof (Elf64),
sizeof (Elf64_Word),
sizeof (Elf64),
sizeof (Elf64),
sizeof (Elf64),
sizeof (Elf64),
sizeof (Elf32_Half),
sizeof (Elf32),
sizeof (Elf64),
sizeof (Elf64),
sizeof (Elf64_Word)
};
#define Nalign64 (sizeof (align64)/sizeof (align64[0]))
#define NALIGN(elf) ((elf->ed_class == ELFCLASS32) ? Nalign32 : Nalign64)
#define ALIGN(elf) ((elf->ed_class == ELFCLASS32) ? align32 : align64)
Elf_Data *
_elf_locked_getdata(Elf_Scn * scn, Elf_Data * data)
{
Dnode * d = (Dnode *)data;
Elf * elf;
Elf_Data src;
unsigned work;
assert(!elf_threaded || RW_LOCK_HELD(&(scn->s_elf->ed_rwlock)));
assert(!elf_threaded || MUTEX_HELD(&(scn->s_mutex)));
elf = scn->s_elf;
if ((scn->s_myflags & SF_READY) == 0) {
UPGRADELOCKS(elf, scn)
if ((scn->s_myflags & SF_READY) == 0)
(void) _elf_cookscn(scn);
DOWNGRADELOCKS(elf, scn)
}
if (d == 0)
d = scn->s_hdnode;
else
d = d->db_next;
if (scn->s_err != 0) {
_elf_seterr((Msg)scn->s_err, 0);
return (0);
}
if (d == 0) {
return (0);
}
if (d->db_scn != scn) {
_elf_seterr(EREQ_DATA, 0);
return (0);
}
if (d->db_myflags & DBF_READY) {
return (&d->db_data);
}
elf = scn->s_elf;
ELFACCESSDATA(work, _elf_work)
d->db_data.d_version = work;
if ((d->db_off == 0) || (d->db_fsz == 0)) {
d->db_myflags |= DBF_READY;
return (&d->db_data);
}
if (elf->ed_class == ELFCLASS32) {
Elf32_Shdr *sh = scn->s_shdr;
size_t sz = sh->sh_entsize;
Elf_Type t = d->db_data.d_type;
if ((t != ELF_T_BYTE) &&
(sz > 1) && (sz != elf32_fsize(t, 1, elf->ed_version))) {
_elf_seterr(EFMT_ENTSZ, 0);
return (0);
}
} else if (elf->ed_class == ELFCLASS64) {
Elf64_Shdr *sh = scn->s_shdr;
Elf64_Xword sz = sh->sh_entsize;
Elf_Type t = d->db_data.d_type;
if (t != ELF_T_BYTE && sz > 1 &&
sz != elf64_fsize(t, 1, elf->ed_version)) {
_elf_seterr(EFMT_ENTSZ, 0);
return (0);
}
} else {
_elf_seterr(EREQ_CLASS, 0);
return (0);
}
if ((d->db_off < 0) || (d->db_off >= elf->ed_fsz) ||
(elf->ed_fsz - d->db_off < d->db_fsz)) {
_elf_seterr(EFMT_DATA, 0);
return (0);
}
src.d_buf = (Elf_Void *)(elf->ed_ident + d->db_off);
src.d_size = d->db_fsz;
src.d_type = d->db_data.d_type;
src.d_version = elf->ed_version;
if (elf->ed_vm) {
UPGRADELOCKS(elf, scn)
if (_elf_vm(elf, (size_t)d->db_off, d->db_fsz) != OK_YES) {
DOWNGRADELOCKS(elf, scn)
return (0);
}
DOWNGRADELOCKS(elf, scn)
}
if (elf->ed_kind == ELF_K_MACHO && NULL == data) {
if (elf->ed_class == ELFCLASS32) {
Elf32_Shdr *sh = scn->s_shdr;
d->db_data.d_buf = elf->ed_image + sh->sh_offset;
d->db_data.d_size = sh->sh_size;
d->db_myflags |= DBF_READY;
return &d->db_data;
} else if (elf->ed_class == ELFCLASS64) {
Elf64_Shdr *sh = scn->s_shdr;
d->db_data.d_buf = elf->ed_image + sh->sh_offset;
d->db_data.d_size = sh->sh_size;
d->db_myflags |= DBF_READY;
return &d->db_data;
}
}
switch (elf->ed_status) {
case ES_COOKED:
if ((size_t)d->db_data.d_type >= NALIGN(elf)) {
_elf_seterr(EBUG_COOKTYPE, 0);
return (0);
}
if (d->db_data.d_size <= src.d_size) {
d->db_data.d_buf = (Elf_Void *)(elf->ed_ident +
d->db_off);
if (((uintptr_t)d->db_data.d_buf
% ALIGN(elf)[d->db_data.d_type]) == 0) {
break;
} else {
d->db_data.d_buf = 0;
}
}
case ES_FROZEN:
if ((d->db_buf = malloc(d->db_data.d_size)) == 0) {
_elf_seterr(EMEM_DATA, errno);
return (0);
}
d->db_data.d_buf = d->db_buf;
break;
default:
_elf_seterr(EBUG_COOKSTAT, 0);
return (0);
}
if (elf->ed_class == ELFCLASS32) {
if (elf32_xlatetom(&d->db_data, &src, elf->ed_encode) == 0)
return (0);
} else {
if (elf64_xlatetom(&d->db_data, &src, elf->ed_encode) == 0)
return (0);
}
d->db_myflags |= DBF_READY;
return (&d->db_data);
}
Elf_Data *
elf_getdata(Elf_Scn * scn, Elf_Data * data)
{
Elf_Data * rc;
Elf * elf;
if (scn == 0)
return (0);
elf = scn->s_elf;
READLOCKS(elf, scn);
rc = _elf_locked_getdata(scn, data);
READUNLOCKS(elf, scn);
return (rc);
}