Home | History | Annotate | Line # | Download | only in ld.elf_so
rtld.c revision 1.39
      1 /*	$NetBSD: rtld.c,v 1.39 2000/07/26 02:07:35 mycroft Exp $	 */
      2 
      3 /*
      4  * Copyright 1996 John D. Polstra.
      5  * Copyright 1996 Matt Thomas <matt (at) 3am-software.com>
      6  * All rights reserved.
      7  *
      8  * Redistribution and use in source and binary forms, with or without
      9  * modification, are permitted provided that the following conditions
     10  * are met:
     11  * 1. Redistributions of source code must retain the above copyright
     12  *    notice, this list of conditions and the following disclaimer.
     13  * 2. Redistributions in binary form must reproduce the above copyright
     14  *    notice, this list of conditions and the following disclaimer in the
     15  *    documentation and/or other materials provided with the distribution.
     16  * 3. All advertising materials mentioning features or use of this software
     17  *    must display the following acknowledgement:
     18  *      This product includes software developed by John Polstra.
     19  * 4. The name of the author may not be used to endorse or promote products
     20  *    derived from this software without specific prior written permission.
     21  *
     22  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
     23  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
     24  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
     25  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
     26  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
     27  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
     28  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
     29  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
     30  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
     31  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
     32  */
     33 
     34 /*
     35  * Dynamic linker for ELF.
     36  *
     37  * John Polstra <jdp (at) polstra.com>.
     38  */
     39 
     40 #include <err.h>
     41 #include <errno.h>
     42 #include <fcntl.h>
     43 #include <stdarg.h>
     44 #include <stdio.h>
     45 #include <stdlib.h>
     46 #include <string.h>
     47 #include <unistd.h>
     48 #include <sys/param.h>
     49 #include <sys/mman.h>
     50 #include <dirent.h>
     51 
     52 #include <ctype.h>
     53 
     54 #include <dlfcn.h>
     55 #include "debug.h"
     56 #include "rtld.h"
     57 
     58 #if !defined(lint)
     59 #include "sysident.h"
     60 #endif
     61 
     62 #define END_SYM		"_end"
     63 
     64 /*
     65  * Debugging support.
     66  */
     67 
     68 typedef void    (*funcptr) __P((void));
     69 
     70 /*
     71  * Function declarations.
     72  */
     73 static void     _rtld_init __P((caddr_t, int));
     74 static void     _rtld_exit __P((void));
     75 
     76 Elf_Addr        _rtld __P((Elf_Addr *));
     77 
     78 
     79 /*
     80  * Data declarations.
     81  */
     82 static char    *error_message;	/* Message for dlopen(), or NULL */
     83 
     84 struct r_debug  _rtld_debug;	/* for GDB; */
     85 bool            _rtld_trust;	/* False for setuid and setgid programs */
     86 Obj_Entry      *_rtld_objlist;	/* Head of linked list of shared objects */
     87 Obj_Entry     **_rtld_objtail;	/* Link field of last object in list */
     88 Obj_Entry      *_rtld_objmain;	/* The main program shared object */
     89 Obj_Entry       _rtld_objself;	/* The dynamic linker shared object */
     90 char            _rtld_path[] = _PATH_RTLD;
     91 unsigned long   _rtld_curmark;	/* Current mark value */
     92 Elf_Sym         _rtld_sym_zero;	/* For resolving undefined weak refs. */
     93 #ifdef	VARPSZ
     94 int		_rtld_pagesz;	/* Page size, as provided by kernel */
     95 #endif
     96 
     97 Objlist _rtld_list_global =	/* Objects dlopened with RTLD_GLOBAL */
     98   SIMPLEQ_HEAD_INITIALIZER(_rtld_list_global);
     99 Objlist _rtld_list_main =	/* Objects loaded at program startup */
    100   SIMPLEQ_HEAD_INITIALIZER(_rtld_list_main);
    101 
    102 Search_Path    *_rtld_default_paths;
    103 Search_Path    *_rtld_paths;
    104 
    105 Library_Xform  *_rtld_xforms;
    106 
    107 /*
    108  * Global declarations normally provided by crt0.
    109  */
    110 char           *__progname;
    111 char          **environ;
    112 
    113 #ifdef OLD_GOT
    114 extern Elf_Addr _GLOBAL_OFFSET_TABLE_[];
    115 #else
    116 extern Elf_Addr _GLOBAL_OFFSET_TABLE_[];
    117 extern Elf_Dyn  _DYNAMIC;
    118 #endif
    119 
    120 static void _rtld_call_fini_functions __P((Obj_Entry *));
    121 static void _rtld_call_init_functions __P((Obj_Entry *));
    122 static Obj_Entry *_rtld_dlcheck __P((void *));
    123 static void _rtld_init_dag __P((Obj_Entry *));
    124 static void _rtld_init_dag1 __P((Obj_Entry *, Obj_Entry *));
    125 static void _rtld_objlist_add __P((Objlist *, Obj_Entry *));
    126 static Objlist_Entry *_rtld_objlist_find __P((Objlist *, const Obj_Entry *));
    127 static void _rtld_objlist_remove __P((Objlist *, Obj_Entry *));
    128 static void _rtld_unload_object __P((Obj_Entry *, bool));
    129 static void _rtld_unref_dag __P((Obj_Entry *));
    130 static Obj_Entry *_rtld_obj_from_addr __P((const void *));
    131 
    132 static void
    133 _rtld_call_fini_functions(first)
    134 	Obj_Entry *first;
    135 {
    136 	Obj_Entry *obj;
    137 
    138 	for (obj = first; obj != NULL; obj = obj->next)
    139 		if (obj->fini != NULL)
    140 			(*obj->fini)();
    141 }
    142 
    143 static void
    144 _rtld_call_init_functions(first)
    145 	Obj_Entry *first;
    146 {
    147 	if (first != NULL) {
    148 		_rtld_call_init_functions(first->next);
    149 		if (first->init != NULL)
    150 			(*first->init)();
    151 	}
    152 }
    153 
    154 /*
    155  * Initialize the dynamic linker.  The argument is the address at which
    156  * the dynamic linker has been mapped into memory.  The primary task of
    157  * this function is to relocate the dynamic linker.
    158  */
    159 static void
    160 _rtld_init(mapbase, pagesz)
    161 	caddr_t mapbase;
    162 	int pagesz;
    163 {
    164 	Obj_Entry objself;/* The dynamic linker shared object */
    165 	const Elf_Ehdr *hdr = (Elf_Ehdr *) mapbase;
    166 #ifdef RTLD_RELOCATE_SELF
    167 	int dodebug = false;
    168 #else
    169 	int dodebug = true;
    170 #endif
    171 	int i;
    172 
    173 	memset(&objself, 0, sizeof objself);
    174 
    175 	/* Conjure up an Obj_Entry structure for the dynamic linker. */
    176 	objself.path = NULL;
    177 	objself.rtld = true;
    178 	objself.mapbase = mapbase;
    179 	objself.phdr = (Elf_Phdr *) (mapbase + hdr->e_phoff);
    180 	for (i = 0; i < hdr->e_phnum; i++) {
    181 		if (objself.phdr[i].p_type == PT_LOAD) {
    182 #ifdef	VARPSZ
    183 			/* We can't touch _rtld_pagesz yet so we can't use round_*() */
    184 #define	_rnd_down(x)	((x) & ~((long)pagesz-1))
    185 #define	_rnd_up(x)	_rnd_down((x) + pagesz - 1)
    186 			objself.textsize = _rnd_up(objself.phdr[i].p_vaddr + objself.phdr[i].p_memsz) - _rnd_down(objself.phdr[i].p_vaddr);
    187 #undef	_rnd_down(x)
    188 #undef	_rnd_up(x)
    189 #else
    190 			objself.textsize = round_up(objself.phdr[i].p_vaddr + objself.phdr[i].p_memsz) - round_down(objself.phdr[i].p_vaddr);
    191 #endif
    192 			break;
    193 		}
    194 	}
    195 
    196 #if defined(__mips__)
    197 	/*
    198 	* mips and ld.so currently linked at load address,
    199 	* so no relocation needed
    200 	*/
    201 	objself.relocbase = 0;
    202 #else
    203 	objself.relocbase = mapbase;
    204 #endif
    205 
    206 	objself.pltgot = NULL;
    207 
    208 #ifdef OLD_GOT
    209 	objself.dynamic = (Elf_Dyn *) _GLOBAL_OFFSET_TABLE_[0];
    210 #else
    211 	objself.dynamic = (Elf_Dyn *) & _DYNAMIC;
    212 #endif
    213 
    214 #ifdef RTLD_RELOCATE_SELF
    215 	/* We have not been relocated yet, so fix the dynamic address */
    216 	objself.dynamic = (Elf_Dyn *)
    217 		((u_long) mapbase + (char *) objself.dynamic);
    218 #endif				/* RTLD_RELOCATE_SELF */
    219 
    220 	_rtld_digest_dynamic(&objself);
    221 
    222 #ifdef __alpha__
    223 	/* XXX XXX XXX */
    224 	objself.pltgot = NULL;
    225 #endif
    226 	assert(objself.needed == NULL);
    227 
    228 #if !defined(__mips__) && !defined(__i386__) && !defined(__vax__)
    229 	/* no relocation for mips/i386 */
    230 	assert(!objself.textrel);
    231 #endif
    232 
    233 	_rtld_relocate_objects(&objself, true, dodebug);
    234 
    235 	/*
    236 	 * Now that we relocated ourselves, we can use globals.
    237 	 */
    238 	_rtld_objself = objself;
    239 
    240 	_rtld_objself.path = _rtld_path;
    241 	_rtld_add_paths(&_rtld_default_paths, RTLD_DEFAULT_LIBRARY_PATH, true);
    242 
    243 	/*
    244 	 * Set up the _rtld_objlist pointer, so that rtld symbols can be found.
    245 	 */
    246 	_rtld_objlist = &_rtld_objself;
    247 
    248 	/* Make the object list empty again. */
    249 	_rtld_objlist = NULL;
    250 	_rtld_objtail = &_rtld_objlist;
    251 
    252 	_rtld_debug.r_brk = _rtld_debug_state;
    253 	_rtld_debug.r_state = RT_CONSISTENT;
    254 }
    255 
    256 /*
    257  * Cleanup procedure.  It will be called (by the atexit() mechanism) just
    258  * before the process exits.
    259  */
    260 static void
    261 _rtld_exit()
    262 {
    263 	dbg(("rtld_exit()"));
    264 
    265 	_rtld_call_fini_functions(_rtld_objlist->next);
    266 }
    267 
    268 /*
    269  * Main entry point for dynamic linking.  The argument is the stack
    270  * pointer.  The stack is expected to be laid out as described in the
    271  * SVR4 ABI specification, Intel 386 Processor Supplement.  Specifically,
    272  * the stack pointer points to a word containing ARGC.  Following that
    273  * in the stack is a null-terminated sequence of pointers to argument
    274  * strings.  Then comes a null-terminated sequence of pointers to
    275  * environment strings.  Finally, there is a sequence of "auxiliary
    276  * vector" entries.
    277  *
    278  * This function returns the entry point for the main program, the dynamic
    279  * linker's exit procedure in sp[0], and a pointer to the main object in
    280  * sp[1].
    281  */
    282 Elf_Addr
    283 _rtld(sp)
    284 	Elf_Addr *sp;
    285 {
    286 	const AuxInfo  *pAUX_base, *pAUX_entry, *pAUX_execfd, *pAUX_phdr,
    287 	               *pAUX_phent, *pAUX_phnum;
    288 #ifdef	VARPSZ
    289 	const AuxInfo  *pAUX_pagesz;
    290 #endif
    291 	char          **env;
    292 	const AuxInfo  *aux;
    293 	const AuxInfo  *auxp;
    294 	Elf_Addr       *const osp = sp;
    295 	bool            bind_now = 0;
    296 	const char     *ld_bind_now;
    297 	const char    **argv;
    298 	int		argc;
    299 	Obj_Entry	*obj;
    300 	const char **real___progname;
    301 	const Obj_Entry **real___mainprog_obj;
    302 	char ***real_environ;
    303 #if defined(RTLD_DEBUG) && !defined(RTLD_RELOCATE_SELF)
    304 	int             i = 0;
    305 #endif
    306 
    307 	/*
    308          * On entry, the dynamic linker itself has not been relocated yet.
    309          * Be very careful not to reference any global data until after
    310          * _rtld_init has returned.  It is OK to reference file-scope statics
    311          * and string constants, and to call static and global functions.
    312          */
    313 	/* Find the auxiliary vector on the stack. */
    314 	/* first Elf_Word reserved to address of exit routine */
    315 #if defined(RTLD_DEBUG) && !defined(RTLD_RELOCATE_SELF)
    316 	dbg(("sp = %p, argc = %ld, argv = %p <%s>\n", sp, (long)sp[2],
    317 	     &sp[3], (char *) sp[3]));
    318 	dbg(("got is at %p, dynamic is at %p\n",
    319 	    _GLOBAL_OFFSET_TABLE_, &_DYNAMIC));
    320 	debug = 1;
    321 	dbg(("_ctype_ is %p\n", _ctype_));
    322 #endif
    323 
    324 	sp += 2;		/* skip over return argument space */
    325 	argv = (const char **) &sp[1];
    326 	argc = *(int *)sp;
    327 	sp += 2 + argc;		/* Skip over argc, arguments, and NULL
    328 				 * terminator */
    329 	env = (char **) sp;
    330 	while (*sp++ != 0) {	/* Skip over environment, and NULL terminator */
    331 #if defined(RTLD_DEBUG) && !defined(RTLD_RELOCATE_SELF)
    332 		dbg(("env[%d] = %p %s\n", i++, (void *)sp[-1], (char *)sp[-1]));
    333 #endif
    334 	}
    335 	aux = (const AuxInfo *) sp;
    336 
    337 	/* Digest the auxiliary vector. */
    338 	pAUX_base = pAUX_entry = pAUX_execfd = NULL;
    339 	pAUX_phdr = pAUX_phent = pAUX_phnum = NULL;
    340 #ifdef	VARPSZ
    341 	pAUX_pagesz = NULL;
    342 #endif
    343 	for (auxp = aux; auxp->a_type != AT_NULL; ++auxp) {
    344 		switch (auxp->a_type) {
    345 		case AT_BASE:
    346 			pAUX_base = auxp;
    347 			break;
    348 		case AT_ENTRY:
    349 			pAUX_entry = auxp;
    350 			break;
    351 		case AT_EXECFD:
    352 			pAUX_execfd = auxp;
    353 			break;
    354 		case AT_PHDR:
    355 			pAUX_phdr = auxp;
    356 			break;
    357 		case AT_PHENT:
    358 			pAUX_phent = auxp;
    359 			break;
    360 		case AT_PHNUM:
    361 			pAUX_phnum = auxp;
    362 			break;
    363 #ifdef	VARPSZ
    364 		case AT_PAGESZ:
    365 			pAUX_pagesz = auxp;
    366 			break;
    367 #endif
    368 		}
    369 	}
    370 
    371 	/* Initialize and relocate ourselves. */
    372 	assert(pAUX_base != NULL);
    373 #ifdef	VARPSZ
    374 	assert(pAUX_pagesz != NULL);
    375 	_rtld_init((caddr_t) pAUX_base->a_v, (int)pAUX_pagesz->a_v);
    376 #else
    377 	_rtld_init((caddr_t) pAUX_base->a_v, 0);
    378 #endif
    379 
    380 #ifdef	VARPSZ
    381 	_rtld_pagesz = (int)pAUX_pagesz->a_v;
    382 #endif
    383 
    384 #ifdef RTLD_DEBUG
    385 	dbg(("_ctype_ is %p\n", _ctype_));
    386 #endif
    387 
    388 	__progname = _rtld_objself.path;
    389 	environ = env;
    390 
    391 	_rtld_trust = geteuid() == getuid() && getegid() == getgid();
    392 
    393 	ld_bind_now = getenv("LD_BIND_NOW");
    394 	if (ld_bind_now != NULL && *ld_bind_now != '\0')
    395 		bind_now = true;
    396 	if (_rtld_trust) {
    397 #ifdef DEBUG
    398 		const char     *ld_debug = getenv("LD_DEBUG");
    399 		if (ld_debug != NULL && *ld_debug != '\0')
    400 			debug = 1;
    401 #endif
    402 		_rtld_add_paths(&_rtld_paths, getenv("LD_LIBRARY_PATH"), true);
    403 	}
    404 	_rtld_process_hints(&_rtld_paths, &_rtld_xforms, _PATH_LD_HINTS, true);
    405 	dbg(("%s is initialized, base address = %p", __progname,
    406 	     (void *) pAUX_base->a_v));
    407 
    408 	/*
    409          * Load the main program, or process its program header if it is
    410          * already loaded.
    411          */
    412 	if (pAUX_execfd != NULL) {	/* Load the main program. */
    413 		int             fd = pAUX_execfd->a_v;
    414 		dbg(("loading main program"));
    415 		_rtld_objmain = _rtld_map_object(argv[0], fd, NULL);
    416 		close(fd);
    417 		if (_rtld_objmain == NULL)
    418 			_rtld_die();
    419 	} else {		/* Main program already loaded. */
    420 		const Elf_Phdr *phdr;
    421 		int             phnum;
    422 		caddr_t         entry;
    423 
    424 		dbg(("processing main program's program header"));
    425 		assert(pAUX_phdr != NULL);
    426 		phdr = (const Elf_Phdr *) pAUX_phdr->a_v;
    427 		assert(pAUX_phnum != NULL);
    428 		phnum = pAUX_phnum->a_v;
    429 		assert(pAUX_phent != NULL);
    430 		assert(pAUX_phent->a_v == sizeof(Elf_Phdr));
    431 		assert(pAUX_entry != NULL);
    432 		entry = (caddr_t) pAUX_entry->a_v;
    433 		_rtld_objmain = _rtld_digest_phdr(phdr, phnum, entry);
    434 	}
    435 
    436 	_rtld_objmain->path = xstrdup("main program");
    437 	_rtld_objmain->mainprog = true;
    438 
    439 	/*
    440 	 * Get the actual dynamic linker pathname from the executable if
    441 	 * possible.  (It should always be possible.)  That ensures that
    442 	 * gdb will find the right dynamic linker even if a non-standard
    443 	 * one is being used.
    444 	 */
    445 	if (_rtld_objmain->interp != NULL &&
    446 	    strcmp(_rtld_objmain->interp, _rtld_objself.path) != 0) {
    447 		free(_rtld_objself.path);
    448 		_rtld_objself.path = xstrdup(_rtld_objmain->interp);
    449 	}
    450 
    451 	_rtld_digest_dynamic(_rtld_objmain);
    452 
    453 	_rtld_linkmap_add(_rtld_objmain);
    454 	_rtld_linkmap_add(&_rtld_objself);
    455 
    456 	/* Link the main program into the list of objects. */
    457 	*_rtld_objtail = _rtld_objmain;
    458 	_rtld_objtail = &_rtld_objmain->next;
    459 	++_rtld_objmain->refcount;
    460 
    461 	/* Initialize a fake symbol for resolving undefined weak references. */
    462 	_rtld_sym_zero.st_info = ELF_ST_INFO(STB_GLOBAL, STT_NOTYPE);
    463 	_rtld_sym_zero.st_shndx = SHN_ABS;
    464 
    465 	/*
    466 	 * Pre-load user-specified objects after the main program but before
    467 	 * any shared object dependencies.
    468 	 */
    469 	dbg(("preloading objects"));
    470 	if (_rtld_trust && _rtld_preload(getenv("LD_PRELOAD"), true) == -1)
    471 		_rtld_die();
    472 
    473 	dbg(("loading needed objects"));
    474 	if (_rtld_load_needed_objects(_rtld_objmain, true) == -1)
    475 		_rtld_die();
    476 
    477 	for (obj = _rtld_objlist;  obj != NULL;  obj = obj->next)
    478 		_rtld_objlist_add(&_rtld_list_main, obj);
    479 
    480 	dbg(("relocating objects"));
    481 	if (_rtld_relocate_objects(_rtld_objmain, bind_now, true) == -1)
    482 		_rtld_die();
    483 
    484 	dbg(("doing copy relocations"));
    485 	if (_rtld_do_copy_relocations(_rtld_objmain, true) == -1)
    486 		_rtld_die();
    487 
    488 	/*
    489 	 * Set the __progname,  environ and, __mainprog_obj before
    490 	 * calling anything that might use them.
    491 	 */
    492 	real___progname = _rtld_objmain_sym("__progname");
    493 	if (real___progname) {
    494 		if ((*real___progname = strrchr(argv[0], '/')) == NULL)
    495 			(*real___progname) = argv[0];
    496 		else
    497 			(*real___progname)++;
    498 	}
    499 	real_environ = _rtld_objmain_sym("environ");
    500 	if (real_environ)
    501 		*real_environ = environ;
    502 	real___mainprog_obj = _rtld_objmain_sym("__mainprog_obj");
    503 	if (real___mainprog_obj)
    504 		*real___mainprog_obj = _rtld_objmain;
    505 
    506 	dbg(("calling _init functions"));
    507 	_rtld_call_init_functions(_rtld_objmain->next);
    508 
    509 	dbg(("control at program entry point = %p, obj = %p, exit = %p",
    510 	     _rtld_objmain->entry, _rtld_objmain, _rtld_exit));
    511 
    512 	/*
    513 	 * Return with the entry point and the exit procedure in at the top
    514 	 * of stack.
    515 	 */
    516 
    517 	_rtld_debug_state();	/* say hello to gdb! */
    518 
    519 	((void **) osp)[0] = _rtld_exit;
    520 	((void **) osp)[1] = _rtld_objmain;
    521 	return (Elf_Addr) _rtld_objmain->entry;
    522 }
    523 
    524 void
    525 _rtld_die()
    526 {
    527 	const char *msg = _rtld_dlerror();
    528 
    529 	if (msg == NULL)
    530 		msg = "Fatal error";
    531 	xerrx(1, "%s\n", msg);
    532 }
    533 
    534 static Obj_Entry *
    535 _rtld_dlcheck(handle)
    536 	void *handle;
    537 {
    538 	Obj_Entry *obj;
    539 
    540 	for (obj = _rtld_objlist; obj != NULL; obj = obj->next)
    541 		if (obj == (Obj_Entry *) handle)
    542 			break;
    543 
    544 	if (obj == NULL || obj->dl_refcount == 0) {
    545 		xwarnx("Invalid shared object handle %p", handle);
    546 		return NULL;
    547 	}
    548 	return obj;
    549 }
    550 
    551 static void
    552 _rtld_init_dag(root)
    553 	Obj_Entry *root;
    554 {
    555 	_rtld_curmark++;
    556 	_rtld_init_dag1(root, root);
    557 }
    558 
    559 static void
    560 _rtld_init_dag1(root, obj)
    561 	Obj_Entry *root;
    562 	Obj_Entry *obj;
    563 {
    564 	const Needed_Entry *needed;
    565 
    566 	if (obj->mark == _rtld_curmark)
    567 		return;
    568 	obj->mark = _rtld_curmark;
    569 	_rtld_objlist_add(&obj->dldags, root);
    570 	_rtld_objlist_add(&root->dagmembers, obj);
    571 	for (needed = obj->needed; needed != NULL; needed = needed->next)
    572 		if (needed->obj != NULL)
    573 			_rtld_init_dag1(root, needed->obj);
    574 }
    575 
    576 /*
    577  * Note, this is called only for objects loaded by dlopen().
    578  */
    579 static void
    580 _rtld_unload_object(root, do_fini_funcs)
    581 	Obj_Entry *root;
    582 	bool do_fini_funcs;
    583 {
    584 	_rtld_unref_dag(root);
    585 	if (root->refcount == 0) { /* We are finished with some objects. */
    586 		Obj_Entry *obj;
    587 		Obj_Entry **linkp;
    588 		Objlist_Entry *elm;
    589 
    590 		/* Finalize objects that are about to be unmapped. */
    591 		if (do_fini_funcs)
    592 			for (obj = _rtld_objlist->next;  obj != NULL;  obj = obj->next)
    593 				if (obj->refcount == 0 && obj->fini != NULL)
    594 					(*obj->fini)();
    595 
    596 		/* Remove the DAG from all objects' DAG lists. */
    597 		for (elm = SIMPLEQ_FIRST(&root->dagmembers); elm; elm = SIMPLEQ_NEXT(elm, link))
    598 			_rtld_objlist_remove(&elm->obj->dldags, root);
    599 
    600 		/* Remove the DAG from the RTLD_GLOBAL list. */
    601 		_rtld_objlist_remove(&_rtld_list_global, root);
    602 
    603 		/* Unmap all objects that are no longer referenced. */
    604 		linkp = &_rtld_objlist->next;
    605 		while ((obj = *linkp) != NULL) {
    606 			if (obj->refcount == 0) {
    607 #ifdef RTLD_DEBUG
    608 				dbg(("unloading \"%s\"", obj->path));
    609 #endif
    610 				munmap(obj->mapbase, obj->mapsize);
    611 				_rtld_linkmap_delete(obj);
    612 				*linkp = obj->next;
    613 				_rtld_obj_free(obj);
    614 			} else
    615 				linkp = &obj->next;
    616 		}
    617 		_rtld_objtail = linkp;
    618 	}
    619 }
    620 
    621 static void
    622 _rtld_unref_dag(root)
    623 	Obj_Entry *root;
    624 {
    625 	assert(root);
    626 	assert(root->refcount != 0);
    627 	--root->refcount;
    628 	if (root->refcount == 0) {
    629 		const Needed_Entry *needed;
    630 
    631 		for (needed = root->needed; needed != NULL;
    632 		     needed = needed->next) {
    633 			if (needed->obj != NULL)
    634 				_rtld_unref_dag(needed->obj);
    635 		}
    636 	}
    637 }
    638 
    639 int
    640 _rtld_dlclose(handle)
    641 	void *handle;
    642 {
    643 	Obj_Entry *root = _rtld_dlcheck(handle);
    644 
    645 	if (root == NULL)
    646 		return -1;
    647 
    648 	_rtld_debug.r_state = RT_DELETE;
    649 	_rtld_debug_state();
    650 
    651 	--root->dl_refcount;
    652 	_rtld_unload_object(root, true);
    653 
    654 	_rtld_debug.r_state = RT_CONSISTENT;
    655 	_rtld_debug_state();
    656 
    657 	return 0;
    658 }
    659 
    660 char *
    661 _rtld_dlerror()
    662 {
    663 	char *msg = error_message;
    664 	error_message = NULL;
    665 	return msg;
    666 }
    667 
    668 void *
    669 _rtld_dlopen(name, mode)
    670 	const char *name;
    671 	int mode;
    672 {
    673 	Obj_Entry **old_obj_tail = _rtld_objtail;
    674 	Obj_Entry *obj = NULL;
    675 
    676 	_rtld_debug.r_state = RT_ADD;
    677 	_rtld_debug_state();
    678 
    679 	if (name == NULL) {
    680 		obj = _rtld_objmain;
    681 		obj->refcount++;
    682 	} else {
    683 		char *path = _rtld_find_library(name, _rtld_objmain);
    684 		if (path != NULL)
    685 			obj = _rtld_load_object(path, true);
    686 	}
    687 
    688 	if (obj != NULL) {
    689 		++obj->dl_refcount;
    690 		if ((mode & RTLD_GLOBAL) &&
    691 		    _rtld_objlist_find(&_rtld_list_global, obj) == NULL)
    692 			_rtld_objlist_add(&_rtld_list_global, obj);
    693 		if (*old_obj_tail != NULL) {	/* We loaded something new. */
    694 			assert(*old_obj_tail == obj);
    695 
    696 			if (_rtld_load_needed_objects(obj, true) == -1 ||
    697 			    (_rtld_init_dag(obj),
    698 			    _rtld_relocate_objects(obj,
    699 			    ((mode & 3) == RTLD_NOW), true)) == -1) {
    700 				_rtld_unload_object(obj, false);
    701 				obj->dl_refcount--;
    702 				obj = NULL;
    703 			} else
    704 				_rtld_call_init_functions(obj);
    705 		}
    706 	}
    707 	_rtld_debug.r_state = RT_CONSISTENT;
    708 	_rtld_debug_state();
    709 
    710 	return obj;
    711 }
    712 
    713 /*
    714  * Find a symbol in the main program.
    715  */
    716 void *
    717 _rtld_objmain_sym(name)
    718 	const char *name;
    719 {
    720 	unsigned long hash;
    721 	const Elf_Sym *def;
    722 	const Obj_Entry *obj;
    723 
    724 	hash = _rtld_elf_hash(name);
    725 	obj = _rtld_objmain;
    726 
    727 	_rtld_curmark++;
    728 	def = _rtld_symlook_list(name, hash, &_rtld_list_main, &obj, true);
    729 
    730 	if (def != NULL)
    731 		return obj->relocbase + def->st_value;
    732 	return(NULL);
    733 }
    734 
    735 void *
    736 _rtld_dlsym(handle, name)
    737 	void *handle;
    738 	const char *name;
    739 {
    740 	const Obj_Entry *obj;
    741 	unsigned long hash;
    742 	const Elf_Sym *def;
    743 	const Obj_Entry *defobj;
    744 
    745 	hash = _rtld_elf_hash(name);
    746 	def = NULL;
    747 	defobj = NULL;
    748 
    749 	if (handle == NULL
    750 #if 0
    751 	    || handle == RTLD_NEXT
    752 #endif
    753 	) {
    754 		void *retaddr;
    755 
    756 		retaddr = __builtin_return_address(0); /* __GNUC__ only */
    757 		if ((obj = _rtld_obj_from_addr(retaddr)) == NULL) {
    758 			_rtld_error("Cannot determine caller's shared object");
    759 			return NULL;
    760 		}
    761 		if (handle == NULL) { /* Just the caller's shared object. */
    762 			def = _rtld_symlook_obj(name, hash, obj, true);
    763 			defobj = obj;
    764 		} else { /* All the shared objects after the caller's */
    765 			while ((obj = obj->next) != NULL) {
    766 				if ((def = _rtld_symlook_obj(name, hash, obj, true)) != NULL) {
    767 					defobj = obj;
    768 					break;
    769 				}
    770 			}
    771 		}
    772 	} else {
    773 		if ((obj = _rtld_dlcheck(handle)) == NULL)
    774 			return NULL;
    775 
    776 		if (obj->mainprog) {
    777 			/* Search main program and all libraries loaded by it. */
    778 			_rtld_curmark++;
    779 			def = _rtld_symlook_list(name, hash, &_rtld_list_main, &defobj, true);
    780 		} else {
    781 			/*
    782 			 * XXX - This isn't correct.  The search should include the whole
    783 			 * DAG rooted at the given object.
    784 			 */
    785 			def = _rtld_symlook_obj(name, hash, obj, true);
    786 			defobj = obj;
    787 		}
    788 	}
    789 
    790 	if (def != NULL)
    791 		return defobj->relocbase + def->st_value;
    792 
    793 	_rtld_error("Undefined symbol \"%s\"", name);
    794 	return NULL;
    795 }
    796 
    797 int
    798 _rtld_dladdr(addr, info)
    799 	const void *addr;
    800 	Dl_info *info;
    801 {
    802 	const Obj_Entry *obj;
    803 	const Elf_Sym *def;
    804 	void *symbol_addr;
    805 	unsigned long symoffset;
    806 
    807 	obj = _rtld_obj_from_addr(addr);
    808 	if (obj == NULL) {
    809 		_rtld_error("No shared object contains address");
    810 		return 0;
    811 	}
    812 	info->dli_fname = obj->path;
    813 	info->dli_fbase = obj->mapbase;
    814 	info->dli_saddr = (void *)0;
    815 	info->dli_sname = NULL;
    816 
    817 	/*
    818 	 * Walk the symbol list looking for the symbol whose address is
    819 	 * closest to the address sent in.
    820 	 */
    821 	for (symoffset = 0; symoffset < obj->nchains; symoffset++) {
    822 		def = obj->symtab + symoffset;
    823 
    824 		/*
    825 		 * For skip the symbol if st_shndx is either SHN_UNDEF or
    826 		 * SHN_COMMON.
    827 		 */
    828 		if (def->st_shndx == SHN_UNDEF || def->st_shndx == SHN_COMMON)
    829 			continue;
    830 
    831 		/*
    832 		 * If the symbol is greater than the specified address, or if it
    833 		 * is further away from addr than the current nearest symbol,
    834 		 * then reject it.
    835 		 */
    836 		symbol_addr = obj->relocbase + def->st_value;
    837 		if (symbol_addr > addr || symbol_addr < info->dli_saddr)
    838 			continue;
    839 
    840 		/* Update our idea of the nearest symbol. */
    841 		info->dli_sname = obj->strtab + def->st_name;
    842 		info->dli_saddr = symbol_addr;
    843 
    844 		/* Exact match? */
    845 		if (info->dli_saddr == addr)
    846 			break;
    847 	}
    848 	return 1;
    849 }
    850 
    851 /*
    852  * Error reporting function.  Use it like printf.  If formats the message
    853  * into a buffer, and sets things up so that the next call to dlerror()
    854  * will return the message.
    855  */
    856 void
    857 #ifdef __STDC__
    858 _rtld_error(const char *fmt,...)
    859 #else
    860 _rtld_error(va_alist)
    861 	va_dcl
    862 #endif
    863 {
    864 	static char     buf[512];
    865 	va_list         ap;
    866 #ifdef __STDC__
    867 	va_start(ap, fmt);
    868 #else
    869 	const char *fmt;
    870 
    871 	va_start(ap);
    872 	fmt = va_arg(ap, const char *);
    873 #endif
    874 	xvsnprintf(buf, sizeof buf, fmt, ap);
    875 	error_message = buf;
    876 	va_end(ap);
    877 }
    878 
    879 void
    880 _rtld_debug_state()
    881 {
    882 	/* do nothing */
    883 }
    884 
    885 void
    886 _rtld_linkmap_add(obj)
    887 	Obj_Entry *obj;
    888 {
    889 	struct link_map *l = &obj->linkmap;
    890 	struct link_map *prev;
    891 
    892 	obj->linkmap.l_name = obj->path;
    893 	obj->linkmap.l_addr = obj->mapbase;
    894 	obj->linkmap.l_ld = obj->dynamic;
    895 #ifdef __mips__
    896 	/* GDB needs load offset on MIPS to use the symbols */
    897 	obj->linkmap.l_offs = obj->relocbase;
    898 #endif
    899 #ifdef __vax__
    900 	/* VAX shared libaries don't start at a vaddr of 0 */
    901 	obj->linkmap.l_addr -= obj->vaddrbase;
    902 #endif
    903 
    904 	if (_rtld_debug.r_map == NULL) {
    905 		_rtld_debug.r_map = l;
    906 		return;
    907 	}
    908 	for (prev = _rtld_debug.r_map; prev->l_next != NULL; prev = prev->l_next);
    909 	l->l_prev = prev;
    910 	prev->l_next = l;
    911 	l->l_next = NULL;
    912 }
    913 
    914 void
    915 _rtld_linkmap_delete(obj)
    916 	Obj_Entry *obj;
    917 {
    918 	struct link_map *l = &obj->linkmap;
    919 
    920 	if (l->l_prev == NULL) {
    921 		if ((_rtld_debug.r_map = l->l_next) != NULL)
    922 			l->l_next->l_prev = NULL;
    923 		return;
    924 	}
    925 	if ((l->l_prev->l_next = l->l_next) != NULL)
    926 		l->l_next->l_prev = l->l_prev;
    927 }
    928 
    929 static Obj_Entry *
    930 _rtld_obj_from_addr(const void *addr)
    931 {
    932 	unsigned long endhash;
    933 	Obj_Entry *obj;
    934 
    935 	endhash = _rtld_elf_hash(END_SYM);
    936 	for (obj = _rtld_objlist;  obj != NULL;  obj = obj->next) {
    937 		const Elf_Sym *endsym;
    938 
    939 		if (addr < (void *) obj->mapbase)
    940 			continue;
    941 		if ((endsym = _rtld_symlook_obj(END_SYM, endhash, obj, true)) == NULL)
    942 			continue; /* No "end" symbol?! */
    943 		if (addr < (void *) (obj->relocbase + endsym->st_value))
    944 			return obj;
    945 	}
    946 	return NULL;
    947 }
    948 
    949 static void
    950 _rtld_objlist_add(list, obj)
    951 	Objlist *list;
    952 	Obj_Entry *obj;
    953 {
    954 	Objlist_Entry *elm;
    955 
    956 	elm = NEW(Objlist_Entry);
    957 	elm->obj = obj;
    958 	SIMPLEQ_INSERT_TAIL(list, elm, link);
    959 }
    960 
    961 static Objlist_Entry *
    962 _rtld_objlist_find(Objlist *list, const Obj_Entry *obj)
    963 {
    964 	Objlist_Entry *elm;
    965 
    966 	for (elm = SIMPLEQ_FIRST(list); elm; elm = SIMPLEQ_NEXT(elm, link)) {
    967 		if (elm->obj == obj)
    968 			return elm;
    969 	}
    970 	return NULL;
    971 }
    972 
    973 static void
    974 _rtld_objlist_remove(list, obj)
    975 	Objlist *list;
    976 	Obj_Entry *obj;
    977 {
    978 	Objlist_Entry *elm;
    979 
    980 	if ((elm = _rtld_objlist_find(list, obj)) != NULL) {
    981 		if ((list)->sqh_first == (elm)) {
    982 			SIMPLEQ_REMOVE_HEAD(list, elm, link);
    983 		}
    984 		else {
    985 			struct Struct_Objlist_Entry *curelm = (list)->sqh_first;
    986 			while (curelm->link.sqe_next != (elm))
    987 				curelm = curelm->link.sqe_next;
    988 			if((curelm->link.sqe_next =
    989 			    curelm->link.sqe_next->link.sqe_next) == NULL)
    990 				(list)->sqh_last = &(curelm)->link.sqe_next;
    991 		}
    992 		free(elm);
    993 	}
    994 }
    995