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