| 0,0 → 1,163 |
|---|
| /* |
| * Copyright (c) 2008 Jiri Svoboda |
| * All rights reserved. |
| * |
| * Redistribution and use in source and binary forms, with or without |
| * modification, are permitted provided that the following conditions |
| * are met: |
| * |
| * - Redistributions of source code must retain the above copyright |
| * notice, this list of conditions and the following disclaimer. |
| * - Redistributions in binary form must reproduce the above copyright |
| * notice, this list of conditions and the following disclaimer in the |
| * documentation and/or other materials provided with the distribution. |
| * - The name of the author may not be used to endorse or promote products |
| * derived from this software without specific prior written permission. |
| * |
| * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR |
| * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES |
| * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. |
| * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, |
| * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT |
| * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
| * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
| * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
| * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF |
| * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
| */ |
| |
| /** @addtogroup rtld rtld |
| * @brief |
| * @{ |
| */ |
| /** |
| * @file |
| */ |
| |
| #include <stdio.h> |
| #include <stdlib.h> |
| #include <string.h> |
| |
| #include <rtld.h> |
| #include <symbol.h> |
| #include <elf.h> |
| |
| /* |
| * Hash tables are 32-bit (elf_word) even for 64-bit ELF files. |
| */ |
| static elf_word elf_hash(const unsigned char *name) |
| { |
| elf_word h = 0, g; |
| |
| while (*name) { |
| h = (h << 4) + *name++; |
| g = h & 0xf0000000; |
| if (g != 0) h ^= g >> 24; |
| h &= ~g; |
| } |
| |
| return h; |
| } |
| |
| static elf_symbol_t *def_find_in_module(char *name, module_t *m) |
| { |
| elf_symbol_t *sym_table; |
| elf_symbol_t *s, *sym; |
| elf_word nbucket; |
| elf_word nchain; |
| elf_word i; |
| char *s_name; |
| elf_word bucket; |
| |
| // module_name = m->dyn.soname; |
| // printf("def_find_in_module('%s', %s)\n", name, module_name); |
| |
| sym_table = m->dyn.sym_tab; |
| nbucket = m->dyn.hash[0]; |
| nchain = m->dyn.hash[1]; |
| |
| bucket = elf_hash((unsigned char *)name) % nbucket; |
| i = m->dyn.hash[2 + bucket]; |
| |
| sym = NULL; |
| while (i != STN_UNDEF) { |
| s = &sym_table[i]; |
| s_name = m->dyn.str_tab + s->st_name; |
| |
| if (strcmp(name, s_name) == 0) { |
| sym = s; |
| break; |
| } |
| |
| i = m->dyn.hash[2 + nbucket + i]; |
| } |
| |
| if (!sym) |
| return NULL; /* Not found */ |
| |
| if (sym->st_shndx == SHN_UNDEF) { |
| /* Not a definition */ |
| return NULL; |
| } |
| |
| return sym; /* Found */ |
| } |
| |
| /** Find the definition of a symbol. |
| * |
| * By definition in System V ABI, if module origin has the flag DT_SYMBOLIC, |
| * origin is searched first. Otherwise, or if the symbol hasn't been found, |
| * the module dependency graph is searched breadth-first, beginning |
| * from the executable program. |
| * |
| * @param name Name of the symbol to search for. |
| * @param origin Module in which the dependency originates. |
| * @param mod (output) Will be filled with a pointer to the module |
| * that contains the symbol. |
| */ |
| elf_symbol_t *symbol_def_find(char *name, module_t *origin, module_t **mod) |
| { |
| module_t *m; |
| elf_symbol_t *sym; |
| |
| /* FIXME: support DT_SYMBOLIC */ |
| //m = origin; |
| |
| /* Start in the executable program */ |
| m = runtime_env.program; |
| |
| while (true) { |
| sym = def_find_in_module(name, m); |
| if (sym != NULL) { |
| *mod = m; |
| return sym; |
| } |
| |
| if (m->n_deps < 1) break; |
| |
| /* FIXME: support branching */ |
| if (m->n_deps > 1) { |
| printf("error: BFS unimplemented\n"); |
| exit(1); |
| } |
| |
| m = m->deps[0]; |
| } |
| |
| printf("Error, symbol '%s' not found anywhere\n", name); |
| exit(1); |
| return NULL; /* Not found */ |
| } |
| |
| uintptr_t symbol_get_addr(elf_symbol_t *sym, module_t *m) |
| { |
| if (sym->st_shndx == SHN_ABS) { |
| /* Do not add bias to absolute symbols */ |
| return sym->st_value; |
| } else { |
| return sym->st_value + m->bias; |
| } |
| } |
| |
| /** @} |
| */ |