|
@@ -53,6 +53,7 @@ struct dso {
|
|
|
int refcnt;
|
|
|
Sym *syms;
|
|
|
uint32_t *hashtab;
|
|
|
+ uint32_t *ghashtab;
|
|
|
char *strings;
|
|
|
unsigned char *map;
|
|
|
size_t map_len;
|
|
@@ -95,7 +96,15 @@ static void decode_vec(size_t *v, size_t *a, size_t cnt)
|
|
|
}
|
|
|
}
|
|
|
|
|
|
-static uint32_t hash(const char *s0)
|
|
|
+static int search_vec(size_t *v, size_t *r, size_t key)
|
|
|
+{
|
|
|
+ for (; v[0]!=key; v+=2)
|
|
|
+ if (!v[0]) return 0;
|
|
|
+ *r = v[1];
|
|
|
+ return 1;
|
|
|
+}
|
|
|
+
|
|
|
+static uint32_t sysv_hash(const char *s0)
|
|
|
{
|
|
|
const unsigned char *s = (void *)s0;
|
|
|
uint_fast32_t h = 0;
|
|
@@ -106,7 +115,16 @@ static uint32_t hash(const char *s0)
|
|
|
return h & 0xfffffff;
|
|
|
}
|
|
|
|
|
|
-static Sym *lookup(const char *s, uint32_t h, struct dso *dso)
|
|
|
+static uint32_t gnu_hash(const char *s0)
|
|
|
+{
|
|
|
+ const unsigned char *s = (void *)s0;
|
|
|
+ uint_fast32_t h = 5381;
|
|
|
+ for (; *s; s++)
|
|
|
+ h = h*33 + *s;
|
|
|
+ return h;
|
|
|
+}
|
|
|
+
|
|
|
+static Sym *sysv_lookup(const char *s, uint32_t h, struct dso *dso)
|
|
|
{
|
|
|
size_t i;
|
|
|
Sym *syms = dso->syms;
|
|
@@ -119,20 +137,61 @@ static Sym *lookup(const char *s, uint32_t h, struct dso *dso)
|
|
|
return 0;
|
|
|
}
|
|
|
|
|
|
+static Sym *gnu_lookup(const char *s, uint32_t h1, struct dso *dso)
|
|
|
+{
|
|
|
+ Sym *sym;
|
|
|
+ char *strings;
|
|
|
+ uint32_t *hashtab = dso->ghashtab;
|
|
|
+ uint32_t nbuckets = hashtab[0];
|
|
|
+ uint32_t *buckets = hashtab + 4 + hashtab[2]*(sizeof(size_t)/4);
|
|
|
+ uint32_t h2;
|
|
|
+ uint32_t *hashval;
|
|
|
+ uint32_t n = buckets[h1 % nbuckets];
|
|
|
+
|
|
|
+ if (!n) return 0;
|
|
|
+
|
|
|
+ strings = dso->strings;
|
|
|
+ sym = dso->syms + n;
|
|
|
+ hashval = buckets + nbuckets + (n - hashtab[1]);
|
|
|
+
|
|
|
+ for (h1 |= 1; ; sym++) {
|
|
|
+ h2 = *hashval++;
|
|
|
+ if ((h1 == (h2|1)) && !strcmp(s, strings + sym->st_name))
|
|
|
+ return sym;
|
|
|
+ if (h2 & 1) break;
|
|
|
+ }
|
|
|
+
|
|
|
+ return 0;
|
|
|
+}
|
|
|
+
|
|
|
#define OK_TYPES (1<<STT_NOTYPE | 1<<STT_OBJECT | 1<<STT_FUNC | 1<<STT_COMMON)
|
|
|
#define OK_BINDS (1<<STB_GLOBAL | 1<<STB_WEAK)
|
|
|
|
|
|
static void *find_sym(struct dso *dso, const char *s, int need_def)
|
|
|
{
|
|
|
- uint32_t h = hash(s);
|
|
|
+ uint32_t h = 0, gh = 0;
|
|
|
void *def = 0;
|
|
|
- if (h==0x6b366be && !strcmp(s, "dlopen")) rtld_used = 1;
|
|
|
- if (h==0x6b3afd && !strcmp(s, "dlsym")) rtld_used = 1;
|
|
|
- if (h==0x595a4cc && !strcmp(s, "__stack_chk_fail")) ssp_used = 1;
|
|
|
+ if (dso->ghashtab) {
|
|
|
+ gh = gnu_hash(s);
|
|
|
+ if (gh == 0xf9040207 && !strcmp(s, "dlopen")) rtld_used = 1;
|
|
|
+ if (gh == 0xf4dc4ae && !strcmp(s, "dlsym")) rtld_used = 1;
|
|
|
+ if (gh == 0x1f4039c9 && !strcmp(s, "__stack_chk_fail")) ssp_used = 1;
|
|
|
+ } else {
|
|
|
+ h = sysv_hash(s);
|
|
|
+ if (h == 0x6b366be && !strcmp(s, "dlopen")) rtld_used = 1;
|
|
|
+ if (h == 0x6b3afd && !strcmp(s, "dlsym")) rtld_used = 1;
|
|
|
+ if (h == 0x595a4cc && !strcmp(s, "__stack_chk_fail")) ssp_used = 1;
|
|
|
+ }
|
|
|
for (; dso; dso=dso->next) {
|
|
|
Sym *sym;
|
|
|
if (!dso->global) continue;
|
|
|
- sym = lookup(s, h, dso);
|
|
|
+ if (dso->ghashtab) {
|
|
|
+ if (!gh) gh = gnu_hash(s);
|
|
|
+ sym = gnu_lookup(s, gh, dso);
|
|
|
+ } else {
|
|
|
+ if (!h) h = sysv_hash(s);
|
|
|
+ sym = sysv_lookup(s, h, dso);
|
|
|
+ }
|
|
|
if (sym && (!need_def || sym->st_shndx) && sym->st_value
|
|
|
&& (1<<(sym->st_info&0xf) & OK_TYPES)
|
|
|
&& (1<<(sym->st_info>>4) & OK_BINDS)) {
|
|
@@ -325,8 +384,11 @@ static void decode_dyn(struct dso *p)
|
|
|
size_t dyn[DYN_CNT] = {0};
|
|
|
decode_vec(p->dynv, dyn, DYN_CNT);
|
|
|
p->syms = (void *)(p->base + dyn[DT_SYMTAB]);
|
|
|
- p->hashtab = (void *)(p->base + dyn[DT_HASH]);
|
|
|
p->strings = (void *)(p->base + dyn[DT_STRTAB]);
|
|
|
+ if (dyn[0]&(1<<DT_HASH))
|
|
|
+ p->hashtab = (void *)(p->base + dyn[DT_HASH]);
|
|
|
+ if (search_vec(p->dynv, dyn, DT_GNU_HASH))
|
|
|
+ p->ghashtab = (void *)(p->base + *dyn);
|
|
|
}
|
|
|
|
|
|
static struct dso *load_library(const char *name)
|
|
@@ -788,7 +850,7 @@ end:
|
|
|
static void *do_dlsym(struct dso *p, const char *s, void *ra)
|
|
|
{
|
|
|
size_t i;
|
|
|
- uint32_t h;
|
|
|
+ uint32_t h = 0, gh = 0;
|
|
|
Sym *sym;
|
|
|
if (p == RTLD_NEXT) {
|
|
|
for (p=head; p && (unsigned char *)ra-p->map>p->map_len; p=p->next);
|
|
@@ -802,12 +864,23 @@ static void *do_dlsym(struct dso *p, const char *s, void *ra)
|
|
|
if (!res) goto failed;
|
|
|
return res;
|
|
|
}
|
|
|
- h = hash(s);
|
|
|
- sym = lookup(s, h, p);
|
|
|
+ if (p->ghashtab) {
|
|
|
+ gh = gnu_hash(s);
|
|
|
+ sym = gnu_lookup(s, gh, p);
|
|
|
+ } else {
|
|
|
+ h = sysv_hash(s);
|
|
|
+ sym = sysv_lookup(s, h, p);
|
|
|
+ }
|
|
|
if (sym && sym->st_value && (1<<(sym->st_info&0xf) & OK_TYPES))
|
|
|
return p->base + sym->st_value;
|
|
|
if (p->deps) for (i=0; p->deps[i]; i++) {
|
|
|
- sym = lookup(s, h, p->deps[i]);
|
|
|
+ if (p->deps[i]->ghashtab) {
|
|
|
+ if (!gh) gh = gnu_hash(s);
|
|
|
+ sym = gnu_lookup(s, h, p->deps[i]);
|
|
|
+ } else {
|
|
|
+ if (!h) h = sysv_hash(s);
|
|
|
+ sym = sysv_lookup(s, h, p->deps[i]);
|
|
|
+ }
|
|
|
if (sym && sym->st_value && (1<<(sym->st_info&0xf) & OK_TYPES))
|
|
|
return p->deps[i]->base + sym->st_value;
|
|
|
}
|