/*
* 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
* @brief
* @{
*/
/**
* @file
*/
#include <stdio.h>
#include <stdlib.h>
#include <str.h>
#include <elf/elf.h>
#include <rtld/module.h>
#include <rtld/rtld.h>
#include <rtld/rtld_debug.h>
#include <rtld/symbol.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(const char *name, module_t *m)
{
if (m->dyn.hash == NULL) {
/* No hash table */
return NULL;
}
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;
DPRINTF("def_find_in_module('%s', %s)\n", name, m->dyn.soname);
sym_table = m->dyn.sym_tab;
nbucket = m->dyn.hash[0];
/* nchain = m->dyn.hash[1]; XXX Use to check HT range */
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 (str_cmp(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 in a module and its deps.
*
* Search the module dependency graph is breadth-first, beginning
* from the module @a start. Thus, @start and all its dependencies
* get searched.
*
* @param name Name of the symbol to search for.
* @param start Module in which to start the search..
* @param mod (output) Will be filled with a pointer to the module
* that contains the symbol.
*/
elf_symbol_t *symbol_bfs_find(const char *name, module_t *start,
module_t **mod)
{
module_t *m, *dm;
elf_symbol_t *sym, *s;
list_t queue;
size_t i;
/*
* Do a BFS using the queue_link and bfs_tag fields.
* Vertices (modules) are tagged the moment they are inserted
* into the queue. This prevents from visiting the same vertex
* more times in case of circular dependencies.
*/
/* Mark all vertices (modules) as unvisited */
modules_untag(start->rtld);
/*
* Insert root (the program) into the queue and tag it.
*
* We disable the dangling-pointer warning because the compiler incorrectly
* assumes that we leak local address (queue) to a parent scope (to start
* argument). However, we always empty the list so the pointer cannot
* actually escape. Probably the compiler can never statically analyze that
* correctly.
*/
list_initialize(&queue);
start->bfs_tag = true;
#pragma GCC diagnostic push
#if defined(__GNUC__) && (__GNUC__ >= 12)
#pragma GCC diagnostic ignored "-Wdangling-pointer"
#endif
list_append(&start->queue_link, &queue);
#pragma GCC diagnostic pop
/* If the symbol is found, it will be stored in 'sym' */
sym = NULL;
/* While queue is not empty */
while (!list_empty(&queue)) {
/* Pop first element from the queue */
m = list_get_instance(list_first(&queue), module_t, queue_link);
list_remove(&m->queue_link);
/* If ssf_noroot is specified, do not look in start module */
s = def_find_in_module(name, m);
if (s != NULL) {
/* Symbol found */
sym = s;
*mod = m;
break;
}
/*
* Insert m's untagged dependencies into the queue
* and tag them.
*/
for (i = 0; i < m->n_deps; ++i) {
dm = m->deps[i];
if (dm->bfs_tag == false) {
dm->bfs_tag = true;
list_append(&dm->queue_link, &queue);
}
}
}
/* Empty the queue so that we leave it in a clean state */
while (!list_empty(&queue))
list_remove(list_first(&queue));
if (!sym) {
return NULL; /* Not found */
}
return sym; /* Symbol 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, search global modules in the default
* order.
*
* @param name Name of the symbol to search for.
* @param origin Module in which the dependency originates.
* @param flags @c ssf_none or @c ssf_noexec to not look for the symbol
* in the executable program.
* @param mod (output) Will be filled with a pointer to the module
* that contains the symbol.
*/
elf_symbol_t *symbol_def_find(const char *name, module_t *origin,
symbol_search_flags_t flags, module_t **mod)
{
elf_symbol_t *s;
DPRINTF("symbol_def_find('%s', origin='%s'\n",
name, origin->dyn.soname);
if (origin->dyn.symbolic && (!origin->exec || (flags & ssf_noexec) == 0)) {
DPRINTF("symbolic->find '%s' in module '%s'\n", name, origin->dyn.soname);
/*
* Origin module has a DT_SYMBOLIC flag.
* Try this module first
*/
s = def_find_in_module(name, origin);
if (s != NULL) {
/* Found */
*mod = origin;
return s;
}
}
/* Not DT_SYMBOLIC or no match. Now try other locations. */
list_foreach(origin->rtld->modules, modules_link, module_t, m) {
DPRINTF("module '%s' local?\n", m->dyn.soname);
if (!m->local && (!m->exec || (flags & ssf_noexec) == 0)) {
DPRINTF("!local->find '%s' in module '%s'\n", name, m->dyn.soname);
s = def_find_in_module(name, m);
if (s != NULL) {
/* Found */
*mod = m;
return s;
}
}
}
/* Finally, try origin. */
DPRINTF("try finding '%s' in origin '%s'\n", name,
origin->dyn.soname);
if (!origin->exec || (flags & ssf_noexec) == 0) {
s = def_find_in_module(name, origin);
if (s != NULL) {
/* Found */
*mod = origin;
return s;
}
}
DPRINTF("'%s' not found\n", name);
return NULL;
}
/** Get symbol address.
*
* @param sym Symbol
* @param m Module contaning the symbol
* @param tcb TCB of the thread whose thread-local variable instance should
* be returned. If @a tcb is @c NULL then @c NULL is returned for
* thread-local variables.
*
* @return Symbol address
*/
void *symbol_get_addr(elf_symbol_t *sym, module_t *m, tcb_t *tcb)
{
if (elf_st_type(sym->st_info) == STT_TLS) {
if (tcb == NULL)
return NULL;
return rtld_tls_get_addr(m->rtld, tcb, m->id, sym->st_value);
} else if (sym->st_shndx == SHN_ABS) {
/* Do not add bias to absolute symbols */
return (void *) sym->st_value;
} else {
return (void *) (sym->st_value + m->bias);
}
}
/** @}
*/