rtld.c revision 1.152 1 1.152 christos /* $NetBSD: rtld.c,v 1.152 2011/08/13 22:24:24 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.87 mycroft * Copyright 2002 Charles M. Hannum <root (at) ihack.net>
7 1.1 cgd * All rights reserved.
8 1.1 cgd *
9 1.1 cgd * Redistribution and use in source and binary forms, with or without
10 1.1 cgd * modification, are permitted provided that the following conditions
11 1.1 cgd * are met:
12 1.1 cgd * 1. Redistributions of source code must retain the above copyright
13 1.1 cgd * notice, this list of conditions and the following disclaimer.
14 1.1 cgd * 2. Redistributions in binary form must reproduce the above copyright
15 1.1 cgd * notice, this list of conditions and the following disclaimer in the
16 1.1 cgd * documentation and/or other materials provided with the distribution.
17 1.1 cgd * 3. All advertising materials mentioning features or use of this software
18 1.1 cgd * must display the following acknowledgement:
19 1.1 cgd * This product includes software developed by John Polstra.
20 1.1 cgd * 4. The name of the author may not be used to endorse or promote products
21 1.1 cgd * derived from this software without specific prior written permission.
22 1.1 cgd *
23 1.1 cgd * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
24 1.1 cgd * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
25 1.1 cgd * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
26 1.1 cgd * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
27 1.1 cgd * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
28 1.1 cgd * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
29 1.1 cgd * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
30 1.1 cgd * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
31 1.1 cgd * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
32 1.1 cgd * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
33 1.1 cgd */
34 1.1 cgd
35 1.1 cgd /*
36 1.1 cgd * Dynamic linker for ELF.
37 1.1 cgd *
38 1.1 cgd * John Polstra <jdp (at) polstra.com>.
39 1.1 cgd */
40 1.1 cgd
41 1.107 skrll #include <sys/cdefs.h>
42 1.107 skrll #ifndef lint
43 1.152 christos __RCSID("$NetBSD: rtld.c,v 1.152 2011/08/13 22:24:24 christos Exp $");
44 1.107 skrll #endif /* not lint */
45 1.107 skrll
46 1.141 joerg #include <sys/param.h>
47 1.141 joerg #include <sys/atomic.h>
48 1.141 joerg #include <sys/mman.h>
49 1.1 cgd #include <err.h>
50 1.1 cgd #include <errno.h>
51 1.1 cgd #include <fcntl.h>
52 1.141 joerg #include <lwp.h>
53 1.1 cgd #include <stdarg.h>
54 1.1 cgd #include <stdio.h>
55 1.1 cgd #include <stdlib.h>
56 1.1 cgd #include <string.h>
57 1.1 cgd #include <unistd.h>
58 1.1 cgd #include <dirent.h>
59 1.1 cgd
60 1.1 cgd #include <ctype.h>
61 1.1 cgd
62 1.2 cgd #include <dlfcn.h>
63 1.1 cgd #include "debug.h"
64 1.1 cgd #include "rtld.h"
65 1.3 cgd
66 1.20 kleink #if !defined(lint)
67 1.3 cgd #include "sysident.h"
68 1.20 kleink #endif
69 1.1 cgd
70 1.1 cgd /*
71 1.1 cgd * Function declarations.
72 1.1 cgd */
73 1.113 christos static void _rtld_init(caddr_t, caddr_t, const char *);
74 1.98 skrll static void _rtld_exit(void);
75 1.15 christos
76 1.98 skrll Elf_Addr _rtld(Elf_Addr *, Elf_Addr);
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.131 skrll u_int _rtld_objcount; /* Number of objects in _rtld_objlist */
91 1.131 skrll u_int _rtld_objloads; /* Number of objects loaded in _rtld_objlist */
92 1.147 joerg u_int _rtld_objgen; /* Generation count for _rtld_objlist */
93 1.120 matt const char _rtld_path[] = _PATH_RTLD;
94 1.120 matt
95 1.120 matt /* Initialize a fake symbol for resolving undefined weak references. */
96 1.120 matt Elf_Sym _rtld_sym_zero = {
97 1.120 matt .st_info = ELF_ST_INFO(STB_GLOBAL, STT_NOTYPE),
98 1.120 matt .st_shndx = SHN_ABS,
99 1.120 matt };
100 1.126 skrll size_t _rtld_pagesz; /* Page size, as provided by kernel */
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.141 joerg static volatile bool _rtld_mutex_may_recurse;
114 1.141 joerg
115 1.108 uwe #if defined(RTLD_DEBUG)
116 1.112 scw #ifndef __sh__
117 1.9 tv extern Elf_Addr _GLOBAL_OFFSET_TABLE_[];
118 1.108 uwe #else /* 32-bit SuperH */
119 1.108 uwe register Elf_Addr *_GLOBAL_OFFSET_TABLE_ asm("r12");
120 1.108 uwe #endif
121 1.108 uwe #endif /* RTLD_DEBUG */
122 1.15 christos extern Elf_Dyn _DYNAMIC;
123 1.8 tv
124 1.148 joerg static void _rtld_call_fini_functions(sigset_t *, int);
125 1.148 joerg static void _rtld_call_init_functions(sigset_t *);
126 1.117 ad static void _rtld_initlist_visit(Objlist *, Obj_Entry *, int);
127 1.117 ad static void _rtld_initlist_tsort(Objlist *, int);
128 1.98 skrll static Obj_Entry *_rtld_dlcheck(void *);
129 1.98 skrll static void _rtld_init_dag(Obj_Entry *);
130 1.98 skrll static void _rtld_init_dag1(Obj_Entry *, Obj_Entry *);
131 1.98 skrll static void _rtld_objlist_remove(Objlist *, Obj_Entry *);
132 1.117 ad static void _rtld_objlist_clear(Objlist *);
133 1.148 joerg static void _rtld_unload_object(sigset_t *, Obj_Entry *, bool);
134 1.98 skrll static void _rtld_unref_dag(Obj_Entry *);
135 1.98 skrll static Obj_Entry *_rtld_obj_from_addr(const void *);
136 1.15 christos
137 1.1 cgd static void
138 1.148 joerg _rtld_call_fini_functions(sigset_t *mask, int force)
139 1.1 cgd {
140 1.117 ad Objlist_Entry *elm;
141 1.117 ad Objlist finilist;
142 1.14 pk Obj_Entry *obj;
143 1.147 joerg void (*fini)(void);
144 1.147 joerg u_int cur_objgen;
145 1.1 cgd
146 1.117 ad dbg(("_rtld_call_fini_functions(%d)", force));
147 1.117 ad
148 1.147 joerg restart:
149 1.147 joerg cur_objgen = ++_rtld_objgen;
150 1.117 ad SIMPLEQ_INIT(&finilist);
151 1.117 ad _rtld_initlist_tsort(&finilist, 1);
152 1.117 ad
153 1.117 ad /* First pass: objects _not_ marked with DF_1_INITFIRST. */
154 1.117 ad SIMPLEQ_FOREACH(elm, &finilist, link) {
155 1.117 ad obj = elm->obj;
156 1.117 ad if (obj->refcount > 0 && !force) {
157 1.117 ad continue;
158 1.117 ad }
159 1.137 skrll if (obj->fini == NULL || obj->fini_called || obj->z_initfirst) {
160 1.117 ad continue;
161 1.117 ad }
162 1.117 ad dbg (("calling fini function %s at %p", obj->path,
163 1.117 ad (void *)obj->fini));
164 1.117 ad obj->fini_called = 1;
165 1.147 joerg /*
166 1.147 joerg * XXX This can race against a concurrent dlclose().
167 1.147 joerg * XXX In that case, the object could be unmapped before
168 1.147 joerg * XXX the fini() call is done.
169 1.147 joerg */
170 1.147 joerg fini = obj->fini;
171 1.148 joerg _rtld_exclusive_exit(mask);
172 1.147 joerg (*fini)();
173 1.148 joerg _rtld_exclusive_enter(mask);
174 1.147 joerg if (_rtld_objgen != cur_objgen) {
175 1.147 joerg dbg(("restarting fini iteration"));
176 1.147 joerg _rtld_objlist_clear(&finilist);
177 1.147 joerg goto restart;
178 1.147 joerg }
179 1.117 ad }
180 1.117 ad
181 1.117 ad /* Second pass: objects marked with DF_1_INITFIRST. */
182 1.117 ad SIMPLEQ_FOREACH(elm, &finilist, link) {
183 1.117 ad obj = elm->obj;
184 1.117 ad if (obj->refcount > 0 && !force) {
185 1.117 ad continue;
186 1.117 ad }
187 1.117 ad if (obj->fini == NULL || obj->fini_called) {
188 1.117 ad continue;
189 1.117 ad }
190 1.117 ad dbg (("calling fini function %s at %p (DF_1_INITFIRST)",
191 1.117 ad obj->path, (void *)obj->fini));
192 1.117 ad obj->fini_called = 1;
193 1.147 joerg /* XXX See above for the race condition here */
194 1.147 joerg fini = obj->fini;
195 1.148 joerg _rtld_exclusive_exit(mask);
196 1.147 joerg (*fini)();
197 1.148 joerg _rtld_exclusive_enter(mask);
198 1.147 joerg if (_rtld_objgen != cur_objgen) {
199 1.147 joerg dbg(("restarting fini iteration"));
200 1.147 joerg _rtld_objlist_clear(&finilist);
201 1.147 joerg goto restart;
202 1.147 joerg }
203 1.117 ad }
204 1.117 ad
205 1.117 ad _rtld_objlist_clear(&finilist);
206 1.1 cgd }
207 1.1 cgd
208 1.1 cgd static void
209 1.148 joerg _rtld_call_init_functions(sigset_t *mask)
210 1.1 cgd {
211 1.117 ad Objlist_Entry *elm;
212 1.117 ad Objlist initlist;
213 1.117 ad Obj_Entry *obj;
214 1.147 joerg void (*init)(void);
215 1.147 joerg u_int cur_objgen;
216 1.94 simonb
217 1.117 ad dbg(("_rtld_call_init_functions()"));
218 1.147 joerg
219 1.147 joerg restart:
220 1.147 joerg cur_objgen = ++_rtld_objgen;
221 1.117 ad SIMPLEQ_INIT(&initlist);
222 1.117 ad _rtld_initlist_tsort(&initlist, 0);
223 1.117 ad
224 1.117 ad /* First pass: objects marked with DF_1_INITFIRST. */
225 1.117 ad SIMPLEQ_FOREACH(elm, &initlist, link) {
226 1.117 ad obj = elm->obj;
227 1.137 skrll if (obj->init == NULL || obj->init_called || !obj->z_initfirst) {
228 1.117 ad continue;
229 1.117 ad }
230 1.117 ad dbg (("calling init function %s at %p (DF_1_INITFIRST)",
231 1.117 ad obj->path, (void *)obj->init));
232 1.117 ad obj->init_called = 1;
233 1.147 joerg init = obj->init;
234 1.148 joerg _rtld_exclusive_exit(mask);
235 1.147 joerg (*init)();
236 1.148 joerg _rtld_exclusive_enter(mask);
237 1.147 joerg if (_rtld_objgen != cur_objgen) {
238 1.147 joerg dbg(("restarting init iteration"));
239 1.147 joerg _rtld_objlist_clear(&initlist);
240 1.147 joerg goto restart;
241 1.147 joerg }
242 1.117 ad }
243 1.117 ad
244 1.117 ad /* Second pass: all other objects. */
245 1.117 ad SIMPLEQ_FOREACH(elm, &initlist, link) {
246 1.117 ad obj = elm->obj;
247 1.117 ad if (obj->init == NULL || obj->init_called) {
248 1.117 ad continue;
249 1.117 ad }
250 1.117 ad dbg (("calling init function %s at %p", obj->path,
251 1.117 ad (void *)obj->init));
252 1.117 ad obj->init_called = 1;
253 1.147 joerg init = obj->init;
254 1.148 joerg _rtld_exclusive_exit(mask);
255 1.147 joerg (*init)();
256 1.148 joerg _rtld_exclusive_enter(mask);
257 1.147 joerg if (_rtld_objgen != cur_objgen) {
258 1.147 joerg dbg(("restarting init iteration"));
259 1.147 joerg _rtld_objlist_clear(&initlist);
260 1.147 joerg goto restart;
261 1.147 joerg }
262 1.14 pk }
263 1.117 ad
264 1.117 ad _rtld_objlist_clear(&initlist);
265 1.1 cgd }
266 1.14 pk
267 1.1 cgd /*
268 1.1 cgd * Initialize the dynamic linker. The argument is the address at which
269 1.1 cgd * the dynamic linker has been mapped into memory. The primary task of
270 1.105 skrll * this function is to create an Obj_Entry for the dynamic linker and
271 1.105 skrll * to resolve the PLT relocation for platforms that need it (those that
272 1.105 skrll * define __HAVE_FUNCTION_DESCRIPTORS
273 1.1 cgd */
274 1.1 cgd static void
275 1.114 christos _rtld_init(caddr_t mapbase, caddr_t relocbase, const char *execname)
276 1.1 cgd {
277 1.94 simonb
278 1.14 pk /* Conjure up an Obj_Entry structure for the dynamic linker. */
279 1.120 matt _rtld_objself.path = __UNCONST(_rtld_path);
280 1.120 matt _rtld_objself.pathlen = sizeof(_rtld_path)-1;
281 1.80 mycroft _rtld_objself.rtld = true;
282 1.80 mycroft _rtld_objself.mapbase = mapbase;
283 1.80 mycroft _rtld_objself.relocbase = relocbase;
284 1.80 mycroft _rtld_objself.dynamic = (Elf_Dyn *) &_DYNAMIC;
285 1.119 christos _rtld_objself.strtab = "_rtld_sym_zero";
286 1.1 cgd
287 1.120 matt /*
288 1.128 skrll * Set value to -relocbase so that
289 1.128 skrll *
290 1.128 skrll * _rtld_objself.relocbase + _rtld_sym_zero.st_value == 0
291 1.128 skrll *
292 1.120 matt * This allows unresolved references to weak symbols to be computed
293 1.128 skrll * to a value of 0.
294 1.120 matt */
295 1.120 matt _rtld_sym_zero.st_value = -(uintptr_t)relocbase;
296 1.120 matt
297 1.113 christos _rtld_digest_dynamic(_rtld_path, &_rtld_objself);
298 1.104 skrll assert(!_rtld_objself.needed);
299 1.105 skrll #if !defined(__hppa__)
300 1.103 skrll assert(!_rtld_objself.pltrel && !_rtld_objself.pltrela);
301 1.105 skrll #else
302 1.105 skrll _rtld_relocate_plt_objects(&_rtld_objself);
303 1.105 skrll #endif
304 1.105 skrll #if !defined(__mips__) && !defined(__hppa__)
305 1.104 skrll assert(!_rtld_objself.pltgot);
306 1.103 skrll #endif
307 1.97 skrll #if !defined(__arm__) && !defined(__mips__) && !defined(__sh__)
308 1.97 skrll /* ARM, MIPS and SH{3,5} have a bogus DT_TEXTREL. */
309 1.103 skrll assert(!_rtld_objself.textrel);
310 1.80 mycroft #endif
311 1.13 christos
312 1.114 christos _rtld_add_paths(execname, &_rtld_default_paths,
313 1.114 christos RTLD_DEFAULT_LIBRARY_PATH);
314 1.13 christos
315 1.123 mrg #ifdef RTLD_ARCH_SUBDIR
316 1.123 mrg _rtld_add_paths(execname, &_rtld_default_paths,
317 1.123 mrg RTLD_DEFAULT_LIBRARY_PATH "/" RTLD_ARCH_SUBDIR);
318 1.123 mrg #endif
319 1.123 mrg
320 1.14 pk /*
321 1.14 pk * Set up the _rtld_objlist pointer, so that rtld symbols can be found.
322 1.14 pk */
323 1.14 pk _rtld_objlist = &_rtld_objself;
324 1.1 cgd
325 1.14 pk /* Make the object list empty again. */
326 1.14 pk _rtld_objlist = NULL;
327 1.14 pk _rtld_objtail = &_rtld_objlist;
328 1.129 roy _rtld_objcount = 0;
329 1.1 cgd
330 1.14 pk _rtld_debug.r_brk = _rtld_debug_state;
331 1.14 pk _rtld_debug.r_state = RT_CONSISTENT;
332 1.1 cgd }
333 1.14 pk
334 1.1 cgd /*
335 1.1 cgd * Cleanup procedure. It will be called (by the atexit() mechanism) just
336 1.1 cgd * before the process exits.
337 1.1 cgd */
338 1.1 cgd static void
339 1.98 skrll _rtld_exit(void)
340 1.1 cgd {
341 1.148 joerg sigset_t mask;
342 1.148 joerg
343 1.15 christos dbg(("rtld_exit()"));
344 1.1 cgd
345 1.148 joerg _rtld_exclusive_enter(&mask);
346 1.145 joerg
347 1.148 joerg _rtld_call_fini_functions(&mask, 1);
348 1.145 joerg
349 1.148 joerg _rtld_exclusive_exit(&mask);
350 1.1 cgd }
351 1.14 pk
352 1.1 cgd /*
353 1.1 cgd * Main entry point for dynamic linking. The argument is the stack
354 1.1 cgd * pointer. The stack is expected to be laid out as described in the
355 1.1 cgd * SVR4 ABI specification, Intel 386 Processor Supplement. Specifically,
356 1.1 cgd * the stack pointer points to a word containing ARGC. Following that
357 1.1 cgd * in the stack is a null-terminated sequence of pointers to argument
358 1.1 cgd * strings. Then comes a null-terminated sequence of pointers to
359 1.1 cgd * environment strings. Finally, there is a sequence of "auxiliary
360 1.1 cgd * vector" entries.
361 1.1 cgd *
362 1.17 kleink * This function returns the entry point for the main program, the dynamic
363 1.17 kleink * linker's exit procedure in sp[0], and a pointer to the main object in
364 1.17 kleink * sp[1].
365 1.1 cgd */
366 1.1 cgd Elf_Addr
367 1.98 skrll _rtld(Elf_Addr *sp, Elf_Addr relocbase)
368 1.1 cgd {
369 1.15 christos const AuxInfo *pAUX_base, *pAUX_entry, *pAUX_execfd, *pAUX_phdr,
370 1.53 christos *pAUX_phent, *pAUX_phnum, *pAUX_euid, *pAUX_egid,
371 1.53 christos *pAUX_ruid, *pAUX_rgid;
372 1.18 ws const AuxInfo *pAUX_pagesz;
373 1.134 joerg char **env, **oenvp;
374 1.15 christos const AuxInfo *aux;
375 1.15 christos const AuxInfo *auxp;
376 1.139 joerg Obj_Entry *obj;
377 1.39 mycroft Elf_Addr *const osp = sp;
378 1.15 christos bool bind_now = 0;
379 1.134 joerg const char *ld_bind_now, *ld_preload, *ld_library_path;
380 1.15 christos const char **argv;
381 1.114 christos const char *execname;
382 1.40 mycroft long argc;
383 1.31 erh const char **real___progname;
384 1.34 christos const Obj_Entry **real___mainprog_obj;
385 1.31 erh char ***real_environ;
386 1.148 joerg sigset_t mask;
387 1.134 joerg #ifdef DEBUG
388 1.138 pooka const char *ld_debug;
389 1.138 pooka #endif
390 1.138 pooka #ifdef RTLD_DEBUG
391 1.117 ad int i = 0;
392 1.12 christos #endif
393 1.1 cgd
394 1.15 christos /*
395 1.15 christos * On entry, the dynamic linker itself has not been relocated yet.
396 1.15 christos * Be very careful not to reference any global data until after
397 1.15 christos * _rtld_init has returned. It is OK to reference file-scope statics
398 1.15 christos * and string constants, and to call static and global functions.
399 1.15 christos */
400 1.15 christos /* Find the auxiliary vector on the stack. */
401 1.15 christos /* first Elf_Word reserved to address of exit routine */
402 1.60 mycroft #if defined(RTLD_DEBUG)
403 1.75 mycroft debug = 1;
404 1.88 fvdl dbg(("sp = %p, argc = %ld, argv = %p <%s> relocbase %p", sp,
405 1.88 fvdl (long)sp[2], &sp[3], (char *) sp[3], (void *)relocbase));
406 1.152 christos #if 0
407 1.75 mycroft dbg(("got is at %p, dynamic is at %p", _GLOBAL_OFFSET_TABLE_,
408 1.75 mycroft &_DYNAMIC));
409 1.152 christos #endif
410 1.71 junyoung dbg(("_ctype_ is %p", _ctype_));
411 1.1 cgd #endif
412 1.1 cgd
413 1.15 christos sp += 2; /* skip over return argument space */
414 1.15 christos argv = (const char **) &sp[1];
415 1.40 mycroft argc = *(long *)sp;
416 1.39 mycroft sp += 2 + argc; /* Skip over argc, arguments, and NULL
417 1.15 christos * terminator */
418 1.15 christos env = (char **) sp;
419 1.15 christos while (*sp++ != 0) { /* Skip over environment, and NULL terminator */
420 1.60 mycroft #if defined(RTLD_DEBUG)
421 1.71 junyoung dbg(("env[%d] = %p %s", i++, (void *)sp[-1], (char *)sp[-1]));
422 1.1 cgd #endif
423 1.15 christos }
424 1.15 christos aux = (const AuxInfo *) sp;
425 1.1 cgd
426 1.15 christos pAUX_base = pAUX_entry = pAUX_execfd = NULL;
427 1.15 christos pAUX_phdr = pAUX_phent = pAUX_phnum = NULL;
428 1.53 christos pAUX_euid = pAUX_ruid = pAUX_egid = pAUX_rgid = NULL;
429 1.18 ws pAUX_pagesz = NULL;
430 1.44 pk
431 1.114 christos execname = NULL;
432 1.114 christos
433 1.69 mycroft /* Digest the auxiliary vector. */
434 1.24 kleink for (auxp = aux; auxp->a_type != AT_NULL; ++auxp) {
435 1.24 kleink switch (auxp->a_type) {
436 1.24 kleink case AT_BASE:
437 1.15 christos pAUX_base = auxp;
438 1.15 christos break;
439 1.24 kleink case AT_ENTRY:
440 1.15 christos pAUX_entry = auxp;
441 1.15 christos break;
442 1.24 kleink case AT_EXECFD:
443 1.15 christos pAUX_execfd = auxp;
444 1.15 christos break;
445 1.24 kleink case AT_PHDR:
446 1.15 christos pAUX_phdr = auxp;
447 1.15 christos break;
448 1.24 kleink case AT_PHENT:
449 1.15 christos pAUX_phent = auxp;
450 1.15 christos break;
451 1.24 kleink case AT_PHNUM:
452 1.15 christos pAUX_phnum = auxp;
453 1.15 christos break;
454 1.54 mycroft #ifdef AT_EUID
455 1.53 christos case AT_EUID:
456 1.53 christos pAUX_euid = auxp;
457 1.53 christos break;
458 1.53 christos case AT_RUID:
459 1.53 christos pAUX_ruid = auxp;
460 1.53 christos break;
461 1.53 christos case AT_EGID:
462 1.53 christos pAUX_egid = auxp;
463 1.53 christos break;
464 1.53 christos case AT_RGID:
465 1.53 christos pAUX_rgid = auxp;
466 1.53 christos break;
467 1.54 mycroft #endif
468 1.114 christos #ifdef AT_SUN_EXECNAME
469 1.114 christos case AT_SUN_EXECNAME:
470 1.114 christos execname = (const char *)(const void *)auxp->a_v;
471 1.114 christos break;
472 1.114 christos #endif
473 1.24 kleink case AT_PAGESZ:
474 1.18 ws pAUX_pagesz = auxp;
475 1.18 ws break;
476 1.15 christos }
477 1.15 christos }
478 1.18 ws
479 1.69 mycroft /* Initialize and relocate ourselves. */
480 1.90 junyoung if (pAUX_base == NULL) {
481 1.90 junyoung _rtld_error("Bad pAUX_base");
482 1.90 junyoung _rtld_die();
483 1.90 junyoung }
484 1.69 mycroft assert(pAUX_pagesz != NULL);
485 1.24 kleink _rtld_pagesz = (int)pAUX_pagesz->a_v;
486 1.114 christos _rtld_init((caddr_t)pAUX_base->a_v, (caddr_t)relocbase, execname);
487 1.1 cgd
488 1.15 christos __progname = _rtld_objself.path;
489 1.15 christos environ = env;
490 1.1 cgd
491 1.53 christos _rtld_trust = ((pAUX_euid ? (uid_t)pAUX_euid->a_v : geteuid()) ==
492 1.53 christos (pAUX_ruid ? (uid_t)pAUX_ruid->a_v : getuid())) &&
493 1.53 christos ((pAUX_egid ? (gid_t)pAUX_egid->a_v : getegid()) ==
494 1.53 christos (pAUX_rgid ? (gid_t)pAUX_rgid->a_v : getgid()));
495 1.1 cgd
496 1.134 joerg #ifdef DEBUG
497 1.134 joerg ld_debug = NULL;
498 1.134 joerg #endif
499 1.134 joerg ld_bind_now = NULL;
500 1.134 joerg ld_library_path = NULL;
501 1.134 joerg ld_preload = NULL;
502 1.134 joerg /*
503 1.134 joerg * Inline avoid using normal getenv/unsetenv here as the libc
504 1.134 joerg * code is quite a bit more complicated.
505 1.134 joerg */
506 1.134 joerg for (oenvp = env; *env != NULL; ++env) {
507 1.134 joerg static const char bind_var[] = "LD_BIND_NOW=";
508 1.134 joerg static const char debug_var[] = "LD_DEBUG=";
509 1.134 joerg static const char path_var[] = "LD_LIBRARY_PATH=";
510 1.134 joerg static const char preload_var[] = "LD_PRELOAD=";
511 1.134 joerg #define LEN(x) (sizeof(x) - 1)
512 1.134 joerg
513 1.134 joerg if ((*env)[0] != 'L' || (*env)[1] != 'D') {
514 1.134 joerg /*
515 1.134 joerg * Special case to skip most entries without
516 1.134 joerg * the more expensive calls to strncmp.
517 1.134 joerg */
518 1.134 joerg *oenvp++ = *env;
519 1.134 joerg } else if (strncmp(*env, debug_var, LEN(debug_var)) == 0) {
520 1.134 joerg if (_rtld_trust) {
521 1.134 joerg #ifdef DEBUG
522 1.134 joerg ld_debug = *env + LEN(debug_var);
523 1.134 joerg #endif
524 1.134 joerg *oenvp++ = *env;
525 1.134 joerg }
526 1.134 joerg } else if (strncmp(*env, bind_var, LEN(bind_var)) == 0) {
527 1.134 joerg ld_bind_now = *env + LEN(bind_var);
528 1.134 joerg } else if (strncmp(*env, path_var, LEN(path_var)) == 0) {
529 1.134 joerg if (_rtld_trust) {
530 1.134 joerg ld_library_path = *env + LEN(path_var);
531 1.134 joerg *oenvp++ = *env;
532 1.134 joerg }
533 1.134 joerg } else if (strncmp(*env, preload_var, LEN(preload_var)) == 0) {
534 1.134 joerg if (_rtld_trust) {
535 1.134 joerg ld_preload = *env + LEN(preload_var);
536 1.134 joerg *oenvp++ = *env;
537 1.134 joerg }
538 1.134 joerg } else {
539 1.134 joerg *oenvp++ = *env;
540 1.134 joerg }
541 1.134 joerg #undef LEN
542 1.134 joerg }
543 1.134 joerg *oenvp++ = NULL;
544 1.134 joerg
545 1.15 christos if (ld_bind_now != NULL && *ld_bind_now != '\0')
546 1.15 christos bind_now = true;
547 1.15 christos if (_rtld_trust) {
548 1.6 mhitch #ifdef DEBUG
549 1.75 mycroft #ifdef RTLD_DEBUG
550 1.75 mycroft debug = 0;
551 1.75 mycroft #endif
552 1.15 christos if (ld_debug != NULL && *ld_debug != '\0')
553 1.15 christos debug = 1;
554 1.15 christos #endif
555 1.134 joerg _rtld_add_paths(execname, &_rtld_paths, ld_library_path);
556 1.111 christos } else {
557 1.115 christos execname = NULL;
558 1.15 christos }
559 1.114 christos _rtld_process_hints(execname, &_rtld_paths, &_rtld_xforms,
560 1.113 christos _PATH_LD_HINTS);
561 1.66 junyoung dbg(("dynamic linker is initialized, mapbase=%p, relocbase=%p",
562 1.58 mycroft _rtld_objself.mapbase, _rtld_objself.relocbase));
563 1.15 christos
564 1.15 christos /*
565 1.15 christos * Load the main program, or process its program header if it is
566 1.15 christos * already loaded.
567 1.15 christos */
568 1.15 christos if (pAUX_execfd != NULL) { /* Load the main program. */
569 1.24 kleink int fd = pAUX_execfd->a_v;
570 1.110 christos const char *obj_name = argv[0] ? argv[0] : "main program";
571 1.15 christos dbg(("loading main program"));
572 1.110 christos _rtld_objmain = _rtld_map_object(obj_name, fd, NULL);
573 1.15 christos close(fd);
574 1.15 christos if (_rtld_objmain == NULL)
575 1.15 christos _rtld_die();
576 1.15 christos } else { /* Main program already loaded. */
577 1.15 christos const Elf_Phdr *phdr;
578 1.15 christos int phnum;
579 1.15 christos caddr_t entry;
580 1.15 christos
581 1.15 christos dbg(("processing main program's program header"));
582 1.15 christos assert(pAUX_phdr != NULL);
583 1.24 kleink phdr = (const Elf_Phdr *) pAUX_phdr->a_v;
584 1.15 christos assert(pAUX_phnum != NULL);
585 1.24 kleink phnum = pAUX_phnum->a_v;
586 1.15 christos assert(pAUX_phent != NULL);
587 1.24 kleink assert(pAUX_phent->a_v == sizeof(Elf_Phdr));
588 1.15 christos assert(pAUX_entry != NULL);
589 1.24 kleink entry = (caddr_t) pAUX_entry->a_v;
590 1.15 christos _rtld_objmain = _rtld_digest_phdr(phdr, phnum, entry);
591 1.84 mycroft _rtld_objmain->path = xstrdup(argv[0] ? argv[0] :
592 1.84 mycroft "main program");
593 1.89 junyoung _rtld_objmain->pathlen = strlen(_rtld_objmain->path);
594 1.15 christos }
595 1.15 christos
596 1.92 mycroft _rtld_objmain->mainprog = true;
597 1.92 mycroft
598 1.25 mycroft /*
599 1.25 mycroft * Get the actual dynamic linker pathname from the executable if
600 1.25 mycroft * possible. (It should always be possible.) That ensures that
601 1.25 mycroft * gdb will find the right dynamic linker even if a non-standard
602 1.25 mycroft * one is being used.
603 1.25 mycroft */
604 1.25 mycroft if (_rtld_objmain->interp != NULL &&
605 1.78 mycroft strcmp(_rtld_objmain->interp, _rtld_objself.path) != 0)
606 1.25 mycroft _rtld_objself.path = xstrdup(_rtld_objmain->interp);
607 1.66 junyoung dbg(("actual dynamic linker is %s", _rtld_objself.path));
608 1.25 mycroft
609 1.114 christos _rtld_digest_dynamic(execname, _rtld_objmain);
610 1.15 christos
611 1.84 mycroft /* Link the main program into the list of objects. */
612 1.84 mycroft *_rtld_objtail = _rtld_objmain;
613 1.84 mycroft _rtld_objtail = &_rtld_objmain->next;
614 1.131 skrll _rtld_objcount++;
615 1.131 skrll _rtld_objloads++;
616 1.84 mycroft
617 1.15 christos _rtld_linkmap_add(_rtld_objmain);
618 1.15 christos _rtld_linkmap_add(&_rtld_objself);
619 1.15 christos
620 1.15 christos ++_rtld_objmain->refcount;
621 1.84 mycroft _rtld_objmain->mainref = 1;
622 1.117 ad _rtld_objlist_push_tail(&_rtld_list_main, _rtld_objmain);
623 1.19 kleink
624 1.134 joerg if (ld_preload) {
625 1.111 christos /*
626 1.111 christos * Pre-load user-specified objects after the main program
627 1.111 christos * but before any shared object dependencies.
628 1.111 christos */
629 1.111 christos dbg(("preloading objects"));
630 1.134 joerg if (_rtld_preload(ld_preload) == -1)
631 1.127 christos _rtld_die();
632 1.134 joerg }
633 1.15 christos
634 1.15 christos dbg(("loading needed objects"));
635 1.137 skrll if (_rtld_load_needed_objects(_rtld_objmain, _RTLD_MAIN) == -1)
636 1.15 christos _rtld_die();
637 1.15 christos
638 1.151 nonaka dbg(("checking for required versions"));
639 1.151 nonaka for (obj = _rtld_objlist; obj != NULL; obj = obj->next) {
640 1.151 nonaka if (_rtld_verify_object_versions(obj) == -1)
641 1.151 nonaka _rtld_die();
642 1.151 nonaka }
643 1.151 nonaka
644 1.139 joerg #if defined(__HAVE_TLS_VARIANT_I) || defined(__HAVE_TLS_VARIANT_II)
645 1.139 joerg dbg(("initializing initial Thread Local Storage"));
646 1.139 joerg /*
647 1.139 joerg * All initial objects get the TLS space from the static block.
648 1.139 joerg */
649 1.139 joerg for (obj = _rtld_objlist; obj != NULL; obj = obj->next)
650 1.139 joerg _rtld_tls_offset_allocate(obj);
651 1.139 joerg _rtld_tls_initial_allocation();
652 1.139 joerg #endif
653 1.139 joerg
654 1.15 christos dbg(("relocating objects"));
655 1.81 mycroft if (_rtld_relocate_objects(_rtld_objmain, bind_now) == -1)
656 1.15 christos _rtld_die();
657 1.15 christos
658 1.15 christos dbg(("doing copy relocations"));
659 1.60 mycroft if (_rtld_do_copy_relocations(_rtld_objmain) == -1)
660 1.15 christos _rtld_die();
661 1.15 christos
662 1.31 erh /*
663 1.34 christos * Set the __progname, environ and, __mainprog_obj before
664 1.34 christos * calling anything that might use them.
665 1.31 erh */
666 1.31 erh real___progname = _rtld_objmain_sym("__progname");
667 1.31 erh if (real___progname) {
668 1.64 christos if (argv[0] != NULL) {
669 1.64 christos if ((*real___progname = strrchr(argv[0], '/')) == NULL)
670 1.64 christos (*real___progname) = argv[0];
671 1.64 christos else
672 1.64 christos (*real___progname)++;
673 1.64 christos } else {
674 1.64 christos (*real___progname) = NULL;
675 1.64 christos }
676 1.31 erh }
677 1.31 erh real_environ = _rtld_objmain_sym("environ");
678 1.31 erh if (real_environ)
679 1.31 erh *real_environ = environ;
680 1.99 skrll /*
681 1.99 skrll * Set __mainprog_obj for old binaries.
682 1.99 skrll */
683 1.34 christos real___mainprog_obj = _rtld_objmain_sym("__mainprog_obj");
684 1.34 christos if (real___mainprog_obj)
685 1.34 christos *real___mainprog_obj = _rtld_objmain;
686 1.31 erh
687 1.148 joerg _rtld_exclusive_enter(&mask);
688 1.141 joerg
689 1.15 christos dbg(("calling _init functions"));
690 1.148 joerg _rtld_call_init_functions(&mask);
691 1.15 christos
692 1.15 christos dbg(("control at program entry point = %p, obj = %p, exit = %p",
693 1.15 christos _rtld_objmain->entry, _rtld_objmain, _rtld_exit));
694 1.15 christos
695 1.148 joerg _rtld_exclusive_exit(&mask);
696 1.141 joerg
697 1.15 christos /*
698 1.15 christos * Return with the entry point and the exit procedure in at the top
699 1.15 christos * of stack.
700 1.15 christos */
701 1.15 christos
702 1.15 christos _rtld_debug_state(); /* say hello to gdb! */
703 1.15 christos
704 1.15 christos ((void **) osp)[0] = _rtld_exit;
705 1.15 christos ((void **) osp)[1] = _rtld_objmain;
706 1.15 christos return (Elf_Addr) _rtld_objmain->entry;
707 1.1 cgd }
708 1.1 cgd
709 1.1 cgd void
710 1.98 skrll _rtld_die(void)
711 1.1 cgd {
712 1.99 skrll const char *msg = dlerror();
713 1.1 cgd
714 1.15 christos if (msg == NULL)
715 1.15 christos msg = "Fatal error";
716 1.52 soren xerrx(1, "%s", msg);
717 1.1 cgd }
718 1.1 cgd
719 1.1 cgd static Obj_Entry *
720 1.98 skrll _rtld_dlcheck(void *handle)
721 1.1 cgd {
722 1.15 christos Obj_Entry *obj;
723 1.1 cgd
724 1.15 christos for (obj = _rtld_objlist; obj != NULL; obj = obj->next)
725 1.15 christos if (obj == (Obj_Entry *) handle)
726 1.15 christos break;
727 1.15 christos
728 1.15 christos if (obj == NULL || obj->dl_refcount == 0) {
729 1.15 christos xwarnx("Invalid shared object handle %p", handle);
730 1.15 christos return NULL;
731 1.15 christos }
732 1.15 christos return obj;
733 1.1 cgd }
734 1.1 cgd
735 1.1 cgd static void
736 1.117 ad _rtld_initlist_visit(Objlist* list, Obj_Entry *obj, int rev)
737 1.117 ad {
738 1.117 ad Needed_Entry* elm;
739 1.117 ad
740 1.117 ad /* dbg(("_rtld_initlist_visit(%s)", obj->path)); */
741 1.117 ad
742 1.117 ad if (obj->init_done)
743 1.117 ad return;
744 1.117 ad obj->init_done = 1;
745 1.117 ad
746 1.117 ad for (elm = obj->needed; elm != NULL; elm = elm->next) {
747 1.117 ad if (elm->obj != NULL) {
748 1.117 ad _rtld_initlist_visit(list, elm->obj, rev);
749 1.117 ad }
750 1.117 ad }
751 1.117 ad
752 1.117 ad if (rev) {
753 1.117 ad _rtld_objlist_push_head(list, obj);
754 1.117 ad } else {
755 1.117 ad _rtld_objlist_push_tail(list, obj);
756 1.117 ad }
757 1.117 ad }
758 1.117 ad
759 1.117 ad static void
760 1.117 ad _rtld_initlist_tsort(Objlist* list, int rev)
761 1.117 ad {
762 1.117 ad dbg(("_rtld_initlist_tsort"));
763 1.117 ad
764 1.117 ad Obj_Entry* obj;
765 1.117 ad
766 1.117 ad for (obj = _rtld_objlist->next; obj; obj = obj->next) {
767 1.117 ad obj->init_done = 0;
768 1.117 ad }
769 1.117 ad
770 1.117 ad for (obj = _rtld_objlist->next; obj; obj = obj->next) {
771 1.117 ad _rtld_initlist_visit(list, obj, rev);
772 1.117 ad }
773 1.117 ad }
774 1.117 ad
775 1.117 ad static void
776 1.98 skrll _rtld_init_dag(Obj_Entry *root)
777 1.25 mycroft {
778 1.94 simonb
779 1.25 mycroft _rtld_init_dag1(root, root);
780 1.25 mycroft }
781 1.25 mycroft
782 1.25 mycroft static void
783 1.98 skrll _rtld_init_dag1(Obj_Entry *root, Obj_Entry *obj)
784 1.25 mycroft {
785 1.25 mycroft const Needed_Entry *needed;
786 1.25 mycroft
787 1.84 mycroft if (!obj->mainref) {
788 1.84 mycroft if (_rtld_objlist_find(&obj->dldags, root))
789 1.84 mycroft return;
790 1.133 skrll dbg(("add %p (%s) to %p (%s) DAG", obj, obj->path, root,
791 1.84 mycroft root->path));
792 1.117 ad _rtld_objlist_push_tail(&obj->dldags, root);
793 1.117 ad _rtld_objlist_push_tail(&root->dagmembers, obj);
794 1.84 mycroft }
795 1.25 mycroft for (needed = obj->needed; needed != NULL; needed = needed->next)
796 1.25 mycroft if (needed->obj != NULL)
797 1.25 mycroft _rtld_init_dag1(root, needed->obj);
798 1.25 mycroft }
799 1.25 mycroft
800 1.25 mycroft /*
801 1.25 mycroft * Note, this is called only for objects loaded by dlopen().
802 1.25 mycroft */
803 1.25 mycroft static void
804 1.148 joerg _rtld_unload_object(sigset_t *mask, Obj_Entry *root, bool do_fini_funcs)
805 1.25 mycroft {
806 1.94 simonb
807 1.25 mycroft _rtld_unref_dag(root);
808 1.25 mycroft if (root->refcount == 0) { /* We are finished with some objects. */
809 1.25 mycroft Obj_Entry *obj;
810 1.25 mycroft Obj_Entry **linkp;
811 1.25 mycroft Objlist_Entry *elm;
812 1.25 mycroft
813 1.25 mycroft /* Finalize objects that are about to be unmapped. */
814 1.25 mycroft if (do_fini_funcs)
815 1.148 joerg _rtld_call_fini_functions(mask, 0);
816 1.25 mycroft
817 1.25 mycroft /* Remove the DAG from all objects' DAG lists. */
818 1.50 lukem SIMPLEQ_FOREACH(elm, &root->dagmembers, link)
819 1.25 mycroft _rtld_objlist_remove(&elm->obj->dldags, root);
820 1.25 mycroft
821 1.25 mycroft /* Remove the DAG from the RTLD_GLOBAL list. */
822 1.84 mycroft if (root->globalref) {
823 1.84 mycroft root->globalref = 0;
824 1.84 mycroft _rtld_objlist_remove(&_rtld_list_global, root);
825 1.84 mycroft }
826 1.25 mycroft
827 1.25 mycroft /* Unmap all objects that are no longer referenced. */
828 1.25 mycroft linkp = &_rtld_objlist->next;
829 1.25 mycroft while ((obj = *linkp) != NULL) {
830 1.25 mycroft if (obj->refcount == 0) {
831 1.25 mycroft dbg(("unloading \"%s\"", obj->path));
832 1.118 ad if (obj->ehdr != MAP_FAILED)
833 1.118 ad munmap(obj->ehdr, _rtld_pagesz);
834 1.25 mycroft munmap(obj->mapbase, obj->mapsize);
835 1.45 christos _rtld_objlist_remove(&_rtld_list_global, obj);
836 1.25 mycroft _rtld_linkmap_delete(obj);
837 1.25 mycroft *linkp = obj->next;
838 1.129 roy _rtld_objcount--;
839 1.25 mycroft _rtld_obj_free(obj);
840 1.25 mycroft } else
841 1.25 mycroft linkp = &obj->next;
842 1.25 mycroft }
843 1.25 mycroft _rtld_objtail = linkp;
844 1.25 mycroft }
845 1.25 mycroft }
846 1.25 mycroft
847 1.137 skrll void
848 1.137 skrll _rtld_ref_dag(Obj_Entry *root)
849 1.137 skrll {
850 1.137 skrll const Needed_Entry *needed;
851 1.137 skrll
852 1.137 skrll assert(root);
853 1.137 skrll
854 1.137 skrll ++root->refcount;
855 1.137 skrll
856 1.137 skrll dbg(("incremented reference on \"%s\" (%d)", root->path,
857 1.137 skrll root->refcount));
858 1.137 skrll for (needed = root->needed; needed != NULL;
859 1.137 skrll needed = needed->next) {
860 1.137 skrll if (needed->obj != NULL)
861 1.137 skrll _rtld_ref_dag(needed->obj);
862 1.137 skrll }
863 1.137 skrll }
864 1.137 skrll
865 1.25 mycroft static void
866 1.98 skrll _rtld_unref_dag(Obj_Entry *root)
867 1.1 cgd {
868 1.94 simonb
869 1.33 jdolecek assert(root);
870 1.15 christos assert(root->refcount != 0);
871 1.137 skrll
872 1.15 christos --root->refcount;
873 1.137 skrll dbg(("decremented reference on \"%s\" (%d)", root->path,
874 1.137 skrll root->refcount));
875 1.137 skrll
876 1.15 christos if (root->refcount == 0) {
877 1.15 christos const Needed_Entry *needed;
878 1.15 christos
879 1.15 christos for (needed = root->needed; needed != NULL;
880 1.32 jdolecek needed = needed->next) {
881 1.32 jdolecek if (needed->obj != NULL)
882 1.32 jdolecek _rtld_unref_dag(needed->obj);
883 1.32 jdolecek }
884 1.15 christos }
885 1.1 cgd }
886 1.1 cgd
887 1.106 thorpej __strong_alias(__dlclose,dlclose)
888 1.1 cgd int
889 1.99 skrll dlclose(void *handle)
890 1.1 cgd {
891 1.143 joerg Obj_Entry *root;
892 1.148 joerg sigset_t mask;
893 1.1 cgd
894 1.144 joerg dbg(("dlclose of %p", handle));
895 1.143 joerg
896 1.148 joerg _rtld_exclusive_enter(&mask);
897 1.143 joerg
898 1.143 joerg root = _rtld_dlcheck(handle);
899 1.143 joerg
900 1.143 joerg if (root == NULL) {
901 1.148 joerg _rtld_exclusive_exit(&mask);
902 1.15 christos return -1;
903 1.143 joerg }
904 1.1 cgd
905 1.15 christos _rtld_debug.r_state = RT_DELETE;
906 1.15 christos _rtld_debug_state();
907 1.15 christos
908 1.15 christos --root->dl_refcount;
909 1.148 joerg _rtld_unload_object(&mask, root, true);
910 1.15 christos
911 1.15 christos _rtld_debug.r_state = RT_CONSISTENT;
912 1.15 christos _rtld_debug_state();
913 1.1 cgd
914 1.148 joerg _rtld_exclusive_exit(&mask);
915 1.143 joerg
916 1.15 christos return 0;
917 1.1 cgd }
918 1.1 cgd
919 1.106 thorpej __strong_alias(__dlerror,dlerror)
920 1.1 cgd char *
921 1.99 skrll dlerror(void)
922 1.1 cgd {
923 1.15 christos char *msg = error_message;
924 1.94 simonb
925 1.15 christos error_message = NULL;
926 1.15 christos return msg;
927 1.1 cgd }
928 1.1 cgd
929 1.106 thorpej __strong_alias(__dlopen,dlopen)
930 1.1 cgd void *
931 1.99 skrll dlopen(const char *name, int mode)
932 1.15 christos {
933 1.15 christos Obj_Entry **old_obj_tail = _rtld_objtail;
934 1.15 christos Obj_Entry *obj = NULL;
935 1.137 skrll int flags = _RTLD_DLOPEN;
936 1.137 skrll bool nodelete;
937 1.137 skrll bool now;
938 1.148 joerg sigset_t mask;
939 1.151 nonaka int result;
940 1.137 skrll
941 1.144 joerg dbg(("dlopen of %s %d", name, mode));
942 1.144 joerg
943 1.148 joerg _rtld_exclusive_enter(&mask);
944 1.141 joerg
945 1.137 skrll flags |= (mode & RTLD_GLOBAL) ? _RTLD_GLOBAL : 0;
946 1.137 skrll flags |= (mode & RTLD_NOLOAD) ? _RTLD_NOLOAD : 0;
947 1.137 skrll
948 1.137 skrll nodelete = (mode & RTLD_NODELETE) ? true : false;
949 1.137 skrll now = ((mode & RTLD_MODEMASK) == RTLD_NOW) ? true : false;
950 1.15 christos
951 1.15 christos _rtld_debug.r_state = RT_ADD;
952 1.15 christos _rtld_debug_state();
953 1.15 christos
954 1.15 christos if (name == NULL) {
955 1.15 christos obj = _rtld_objmain;
956 1.25 mycroft obj->refcount++;
957 1.67 mycroft } else
958 1.137 skrll obj = _rtld_load_library(name, _rtld_objmain, flags);
959 1.137 skrll
960 1.1 cgd
961 1.15 christos if (obj != NULL) {
962 1.15 christos ++obj->dl_refcount;
963 1.15 christos if (*old_obj_tail != NULL) { /* We loaded something new. */
964 1.15 christos assert(*old_obj_tail == obj);
965 1.15 christos
966 1.151 nonaka result = _rtld_load_needed_objects(obj, flags);
967 1.151 nonaka if (result != -1) {
968 1.151 nonaka Objlist_Entry *entry;
969 1.151 nonaka _rtld_init_dag(obj);
970 1.151 nonaka SIMPLEQ_FOREACH(entry, &obj->dagmembers, link) {
971 1.151 nonaka result = _rtld_verify_object_versions(entry->obj);
972 1.151 nonaka if (result == -1)
973 1.151 nonaka break;
974 1.151 nonaka }
975 1.151 nonaka }
976 1.151 nonaka if (result == -1 || _rtld_relocate_objects(obj,
977 1.151 nonaka (now || obj->z_now)) == -1) {
978 1.148 joerg _rtld_unload_object(&mask, obj, false);
979 1.25 mycroft obj->dl_refcount--;
980 1.15 christos obj = NULL;
981 1.117 ad } else {
982 1.148 joerg _rtld_call_init_functions(&mask);
983 1.117 ad }
984 1.15 christos }
985 1.137 skrll if (obj != NULL) {
986 1.137 skrll if ((nodelete || obj->z_nodelete) && !obj->ref_nodel) {
987 1.137 skrll dbg(("dlopen obj %s nodelete", obj->path));
988 1.137 skrll _rtld_ref_dag(obj);
989 1.137 skrll obj->z_nodelete = obj->ref_nodel = true;
990 1.137 skrll }
991 1.137 skrll }
992 1.15 christos }
993 1.15 christos _rtld_debug.r_state = RT_CONSISTENT;
994 1.15 christos _rtld_debug_state();
995 1.1 cgd
996 1.148 joerg _rtld_exclusive_exit(&mask);
997 1.141 joerg
998 1.15 christos return obj;
999 1.31 erh }
1000 1.31 erh
1001 1.31 erh /*
1002 1.31 erh * Find a symbol in the main program.
1003 1.31 erh */
1004 1.31 erh void *
1005 1.98 skrll _rtld_objmain_sym(const char *name)
1006 1.31 erh {
1007 1.31 erh unsigned long hash;
1008 1.31 erh const Elf_Sym *def;
1009 1.31 erh const Obj_Entry *obj;
1010 1.129 roy DoneList donelist;
1011 1.31 erh
1012 1.31 erh hash = _rtld_elf_hash(name);
1013 1.31 erh obj = _rtld_objmain;
1014 1.129 roy _rtld_donelist_init(&donelist);
1015 1.31 erh
1016 1.151 nonaka def = _rtld_symlook_list(name, hash, &_rtld_list_main, &obj, 0,
1017 1.151 nonaka NULL, &donelist);
1018 1.129 roy
1019 1.31 erh if (def != NULL)
1020 1.31 erh return obj->relocbase + def->st_value;
1021 1.136 skrll return NULL;
1022 1.1 cgd }
1023 1.1 cgd
1024 1.121 macallan #ifdef __powerpc__
1025 1.121 macallan static void *
1026 1.121 macallan hackish_return_address(void)
1027 1.121 macallan {
1028 1.121 macallan return __builtin_return_address(1);
1029 1.121 macallan }
1030 1.121 macallan #endif
1031 1.121 macallan
1032 1.141 joerg #ifdef __HAVE_FUNCTION_DESCRIPTORS
1033 1.148 joerg #define lookup_mutex_enter() _rtld_exclusive_enter(&mask)
1034 1.148 joerg #define lookup_mutex_exit() _rtld_exclusive_exit(&mask)
1035 1.141 joerg #else
1036 1.141 joerg #define lookup_mutex_enter() _rtld_shared_enter()
1037 1.141 joerg #define lookup_mutex_exit() _rtld_shared_exit()
1038 1.141 joerg #endif
1039 1.141 joerg
1040 1.151 nonaka static void *
1041 1.151 nonaka do_dlsym(void *handle, const char *name, const Ver_Entry *ventry)
1042 1.15 christos {
1043 1.25 mycroft const Obj_Entry *obj;
1044 1.92 mycroft unsigned long hash;
1045 1.25 mycroft const Elf_Sym *def;
1046 1.15 christos const Obj_Entry *defobj;
1047 1.93 christos void *retaddr;
1048 1.148 joerg DoneList donelist;
1049 1.151 nonaka const u_int flags = SYMLOOK_DLSYM | SYMLOOK_IN_PLT;
1050 1.148 joerg #ifdef __HAVE_FUNCTION_DESCRIPTORS
1051 1.148 joerg sigset_t mask;
1052 1.148 joerg #endif
1053 1.141 joerg
1054 1.141 joerg lookup_mutex_enter();
1055 1.141 joerg
1056 1.92 mycroft hash = _rtld_elf_hash(name);
1057 1.92 mycroft def = NULL;
1058 1.92 mycroft defobj = NULL;
1059 1.92 mycroft
1060 1.93 christos switch ((intptr_t)handle) {
1061 1.93 christos case (intptr_t)NULL:
1062 1.93 christos case (intptr_t)RTLD_NEXT:
1063 1.93 christos case (intptr_t)RTLD_DEFAULT:
1064 1.93 christos case (intptr_t)RTLD_SELF:
1065 1.121 macallan #ifdef __powerpc__
1066 1.121 macallan retaddr = hackish_return_address();
1067 1.121 macallan #else
1068 1.121 macallan retaddr = __builtin_return_address(0);
1069 1.121 macallan #endif
1070 1.30 christos if ((obj = _rtld_obj_from_addr(retaddr)) == NULL) {
1071 1.30 christos _rtld_error("Cannot determine caller's shared object");
1072 1.141 joerg lookup_mutex_exit();
1073 1.30 christos return NULL;
1074 1.30 christos }
1075 1.93 christos
1076 1.93 christos switch ((intptr_t)handle) {
1077 1.93 christos case (intptr_t)NULL: /* Just the caller's shared object. */
1078 1.151 nonaka def = _rtld_symlook_obj(name, hash, obj, flags, ventry);
1079 1.92 mycroft defobj = obj;
1080 1.93 christos break;
1081 1.93 christos
1082 1.93 christos case (intptr_t)RTLD_NEXT: /* Objects after callers */
1083 1.93 christos obj = obj->next;
1084 1.93 christos /*FALLTHROUGH*/
1085 1.93 christos
1086 1.93 christos case (intptr_t)RTLD_SELF: /* Caller included */
1087 1.93 christos for (; obj; obj = obj->next) {
1088 1.93 christos if ((def = _rtld_symlook_obj(name, hash, obj,
1089 1.151 nonaka flags, ventry)) != NULL) {
1090 1.92 mycroft defobj = obj;
1091 1.92 mycroft break;
1092 1.92 mycroft }
1093 1.92 mycroft }
1094 1.93 christos break;
1095 1.93 christos
1096 1.93 christos case (intptr_t)RTLD_DEFAULT:
1097 1.93 christos def = _rtld_symlook_default(name, hash, obj, &defobj,
1098 1.151 nonaka flags, ventry);
1099 1.93 christos break;
1100 1.93 christos
1101 1.93 christos default:
1102 1.93 christos abort();
1103 1.92 mycroft }
1104 1.93 christos break;
1105 1.93 christos
1106 1.93 christos default:
1107 1.141 joerg if ((obj = _rtld_dlcheck(handle)) == NULL) {
1108 1.141 joerg lookup_mutex_exit();
1109 1.30 christos return NULL;
1110 1.141 joerg }
1111 1.141 joerg
1112 1.129 roy _rtld_donelist_init(&donelist);
1113 1.129 roy
1114 1.92 mycroft if (obj->mainprog) {
1115 1.93 christos /* Search main program and all libraries loaded by it */
1116 1.93 christos def = _rtld_symlook_list(name, hash, &_rtld_list_main,
1117 1.151 nonaka &defobj, flags, ventry, &donelist);
1118 1.92 mycroft } else {
1119 1.122 skrll Needed_Entry fake;
1120 1.129 roy DoneList depth;
1121 1.122 skrll
1122 1.122 skrll /* Search the object and all the libraries loaded by it. */
1123 1.122 skrll fake.next = NULL;
1124 1.124 christos fake.obj = __UNCONST(obj);
1125 1.122 skrll fake.name = 0;
1126 1.129 roy
1127 1.129 roy _rtld_donelist_init(&depth);
1128 1.122 skrll def = _rtld_symlook_needed(name, hash, &fake, &defobj,
1129 1.151 nonaka flags, ventry, &donelist, &depth);
1130 1.25 mycroft }
1131 1.129 roy
1132 1.93 christos break;
1133 1.30 christos }
1134 1.30 christos
1135 1.51 fredette if (def != NULL) {
1136 1.141 joerg void *p;
1137 1.51 fredette #ifdef __HAVE_FUNCTION_DESCRIPTORS
1138 1.141 joerg if (ELF_ST_TYPE(def->st_info) == STT_FUNC) {
1139 1.142 joerg p = (void *)_rtld_function_descriptor_alloc(defobj,
1140 1.142 joerg def, 0);
1141 1.141 joerg lookup_mutex_exit();
1142 1.141 joerg return p;
1143 1.141 joerg }
1144 1.51 fredette #endif /* __HAVE_FUNCTION_DESCRIPTORS */
1145 1.141 joerg p = defobj->relocbase + def->st_value;
1146 1.141 joerg lookup_mutex_exit();
1147 1.141 joerg return p;
1148 1.51 fredette }
1149 1.92 mycroft
1150 1.101 mycroft _rtld_error("Undefined symbol \"%s\"", name);
1151 1.141 joerg lookup_mutex_exit();
1152 1.92 mycroft return NULL;
1153 1.25 mycroft }
1154 1.1 cgd
1155 1.151 nonaka __strong_alias(__dlsym,dlsym)
1156 1.151 nonaka void *
1157 1.151 nonaka dlsym(void *handle, const char *name)
1158 1.151 nonaka {
1159 1.151 nonaka
1160 1.151 nonaka dbg(("dlsym of %s in %p", name, handle));
1161 1.151 nonaka
1162 1.151 nonaka return do_dlsym(handle, name, NULL);
1163 1.151 nonaka }
1164 1.151 nonaka
1165 1.151 nonaka __strong_alias(__dlvsym,dlvsym)
1166 1.151 nonaka void *
1167 1.151 nonaka dlvsym(void *handle, const char *name, const char *version)
1168 1.151 nonaka {
1169 1.151 nonaka Ver_Entry *ventry = NULL;
1170 1.151 nonaka Ver_Entry ver_entry;
1171 1.151 nonaka
1172 1.151 nonaka dbg(("dlvsym of %s@%s in %p", name, version ? version : NULL, handle));
1173 1.151 nonaka
1174 1.151 nonaka if (version != NULL) {
1175 1.151 nonaka ver_entry.name = version;
1176 1.151 nonaka ver_entry.file = NULL;
1177 1.151 nonaka ver_entry.hash = _rtld_elf_hash(version);
1178 1.151 nonaka ver_entry.flags = 0;
1179 1.151 nonaka ventry = &ver_entry;
1180 1.151 nonaka }
1181 1.151 nonaka return do_dlsym(handle, name, ventry);
1182 1.151 nonaka }
1183 1.151 nonaka
1184 1.106 thorpej __strong_alias(__dladdr,dladdr)
1185 1.25 mycroft int
1186 1.99 skrll dladdr(const void *addr, Dl_info *info)
1187 1.25 mycroft {
1188 1.25 mycroft const Obj_Entry *obj;
1189 1.51 fredette const Elf_Sym *def, *best_def;
1190 1.25 mycroft void *symbol_addr;
1191 1.25 mycroft unsigned long symoffset;
1192 1.149 joerg #ifdef __HAVE_FUNCTION_DESCRIPTORS
1193 1.149 joerg sigset_t mask;
1194 1.149 joerg #endif
1195 1.141 joerg
1196 1.144 joerg dbg(("dladdr of %p", addr));
1197 1.144 joerg
1198 1.141 joerg lookup_mutex_enter();
1199 1.141 joerg
1200 1.51 fredette #ifdef __HAVE_FUNCTION_DESCRIPTORS
1201 1.51 fredette addr = _rtld_function_descriptor_function(addr);
1202 1.51 fredette #endif /* __HAVE_FUNCTION_DESCRIPTORS */
1203 1.51 fredette
1204 1.25 mycroft obj = _rtld_obj_from_addr(addr);
1205 1.25 mycroft if (obj == NULL) {
1206 1.25 mycroft _rtld_error("No shared object contains address");
1207 1.141 joerg lookup_mutex_enter();
1208 1.25 mycroft return 0;
1209 1.25 mycroft }
1210 1.25 mycroft info->dli_fname = obj->path;
1211 1.25 mycroft info->dli_fbase = obj->mapbase;
1212 1.25 mycroft info->dli_saddr = (void *)0;
1213 1.25 mycroft info->dli_sname = NULL;
1214 1.25 mycroft
1215 1.15 christos /*
1216 1.25 mycroft * Walk the symbol list looking for the symbol whose address is
1217 1.25 mycroft * closest to the address sent in.
1218 1.25 mycroft */
1219 1.51 fredette best_def = NULL;
1220 1.25 mycroft for (symoffset = 0; symoffset < obj->nchains; symoffset++) {
1221 1.25 mycroft def = obj->symtab + symoffset;
1222 1.25 mycroft
1223 1.25 mycroft /*
1224 1.25 mycroft * For skip the symbol if st_shndx is either SHN_UNDEF or
1225 1.25 mycroft * SHN_COMMON.
1226 1.25 mycroft */
1227 1.25 mycroft if (def->st_shndx == SHN_UNDEF || def->st_shndx == SHN_COMMON)
1228 1.25 mycroft continue;
1229 1.25 mycroft
1230 1.25 mycroft /*
1231 1.25 mycroft * If the symbol is greater than the specified address, or if it
1232 1.25 mycroft * is further away from addr than the current nearest symbol,
1233 1.25 mycroft * then reject it.
1234 1.25 mycroft */
1235 1.25 mycroft symbol_addr = obj->relocbase + def->st_value;
1236 1.25 mycroft if (symbol_addr > addr || symbol_addr < info->dli_saddr)
1237 1.25 mycroft continue;
1238 1.25 mycroft
1239 1.25 mycroft /* Update our idea of the nearest symbol. */
1240 1.25 mycroft info->dli_sname = obj->strtab + def->st_name;
1241 1.25 mycroft info->dli_saddr = symbol_addr;
1242 1.51 fredette best_def = def;
1243 1.1 cgd
1244 1.25 mycroft /* Exact match? */
1245 1.25 mycroft if (info->dli_saddr == addr)
1246 1.25 mycroft break;
1247 1.25 mycroft }
1248 1.51 fredette
1249 1.51 fredette #ifdef __HAVE_FUNCTION_DESCRIPTORS
1250 1.51 fredette if (best_def != NULL && ELF_ST_TYPE(best_def->st_info) == STT_FUNC)
1251 1.51 fredette info->dli_saddr = (void *)_rtld_function_descriptor_alloc(obj,
1252 1.51 fredette best_def, 0);
1253 1.51 fredette #endif /* __HAVE_FUNCTION_DESCRIPTORS */
1254 1.51 fredette
1255 1.141 joerg lookup_mutex_exit();
1256 1.25 mycroft return 1;
1257 1.1 cgd }
1258 1.1 cgd
1259 1.125 pooka __strong_alias(__dlinfo,dlinfo)
1260 1.125 pooka int
1261 1.125 pooka dlinfo(void *handle, int req, void *v)
1262 1.125 pooka {
1263 1.125 pooka const Obj_Entry *obj;
1264 1.125 pooka void *retaddr;
1265 1.125 pooka
1266 1.144 joerg dbg(("dlinfo for %p %d", handle, req));
1267 1.144 joerg
1268 1.141 joerg _rtld_shared_enter();
1269 1.141 joerg
1270 1.125 pooka if (handle == RTLD_SELF) {
1271 1.125 pooka #ifdef __powerpc__
1272 1.125 pooka retaddr = hackish_return_address();
1273 1.125 pooka #else
1274 1.125 pooka retaddr = __builtin_return_address(0);
1275 1.125 pooka #endif
1276 1.125 pooka if ((obj = _rtld_obj_from_addr(retaddr)) == NULL) {
1277 1.125 pooka _rtld_error("Cannot determine caller's shared object");
1278 1.141 joerg _rtld_shared_exit();
1279 1.125 pooka return -1;
1280 1.125 pooka }
1281 1.125 pooka } else {
1282 1.125 pooka if ((obj = _rtld_dlcheck(handle)) == NULL) {
1283 1.125 pooka _rtld_error("Invalid handle");
1284 1.141 joerg _rtld_shared_exit();
1285 1.125 pooka return -1;
1286 1.125 pooka }
1287 1.125 pooka }
1288 1.125 pooka
1289 1.125 pooka switch (req) {
1290 1.125 pooka case RTLD_DI_LINKMAP:
1291 1.125 pooka {
1292 1.125 pooka const struct link_map **map = v;
1293 1.125 pooka
1294 1.125 pooka *map = &obj->linkmap;
1295 1.125 pooka break;
1296 1.125 pooka }
1297 1.125 pooka
1298 1.125 pooka default:
1299 1.125 pooka _rtld_error("Invalid request");
1300 1.141 joerg _rtld_shared_exit();
1301 1.125 pooka return -1;
1302 1.125 pooka }
1303 1.125 pooka
1304 1.141 joerg _rtld_shared_exit();
1305 1.125 pooka return 0;
1306 1.125 pooka }
1307 1.125 pooka
1308 1.131 skrll __strong_alias(__dl_iterate_phdr,dl_iterate_phdr);
1309 1.131 skrll int
1310 1.131 skrll dl_iterate_phdr(int (*callback)(struct dl_phdr_info *, size_t, void *), void *param)
1311 1.131 skrll {
1312 1.131 skrll struct dl_phdr_info phdr_info;
1313 1.131 skrll const Obj_Entry *obj;
1314 1.131 skrll int error = 0;
1315 1.131 skrll
1316 1.144 joerg dbg(("dl_iterate_phdr"));
1317 1.144 joerg
1318 1.141 joerg _rtld_shared_enter();
1319 1.141 joerg
1320 1.131 skrll for (obj = _rtld_objlist; obj != NULL; obj = obj->next) {
1321 1.131 skrll phdr_info.dlpi_addr = (Elf_Addr)obj->relocbase;
1322 1.131 skrll phdr_info.dlpi_name = STAILQ_FIRST(&obj->names) ?
1323 1.131 skrll STAILQ_FIRST(&obj->names)->name : obj->path;
1324 1.131 skrll phdr_info.dlpi_phdr = obj->phdr;
1325 1.131 skrll phdr_info.dlpi_phnum = obj->phsize / sizeof(obj->phdr[0]);
1326 1.140 joerg #if defined(__HAVE_TLS_VARIANT_I) || defined(__HAVE_TLS_VARIANT_II)
1327 1.140 joerg phdr_info.dlpi_tls_modid = obj->tlsindex;
1328 1.140 joerg phdr_info.dlpi_tls_data = obj->tlsinit;
1329 1.140 joerg #else
1330 1.131 skrll phdr_info.dlpi_tls_modid = 0;
1331 1.131 skrll phdr_info.dlpi_tls_data = 0;
1332 1.131 skrll #endif
1333 1.131 skrll phdr_info.dlpi_adds = _rtld_objloads;
1334 1.131 skrll phdr_info.dlpi_subs = _rtld_objloads - _rtld_objcount;
1335 1.131 skrll
1336 1.141 joerg /* XXXlocking: exit point */
1337 1.131 skrll error = callback(&phdr_info, sizeof(phdr_info), param);
1338 1.131 skrll if (error)
1339 1.131 skrll break;
1340 1.131 skrll }
1341 1.131 skrll
1342 1.141 joerg _rtld_shared_exit();
1343 1.131 skrll return error;
1344 1.131 skrll }
1345 1.131 skrll
1346 1.1 cgd /*
1347 1.1 cgd * Error reporting function. Use it like printf. If formats the message
1348 1.1 cgd * into a buffer, and sets things up so that the next call to dlerror()
1349 1.1 cgd * will return the message.
1350 1.1 cgd */
1351 1.1 cgd void
1352 1.15 christos _rtld_error(const char *fmt,...)
1353 1.1 cgd {
1354 1.15 christos static char buf[512];
1355 1.15 christos va_list ap;
1356 1.49 wiz
1357 1.15 christos va_start(ap, fmt);
1358 1.15 christos xvsnprintf(buf, sizeof buf, fmt, ap);
1359 1.15 christos error_message = buf;
1360 1.15 christos va_end(ap);
1361 1.1 cgd }
1362 1.14 pk
1363 1.1 cgd void
1364 1.98 skrll _rtld_debug_state(void)
1365 1.1 cgd {
1366 1.94 simonb
1367 1.15 christos /* do nothing */
1368 1.1 cgd }
1369 1.1 cgd
1370 1.1 cgd void
1371 1.98 skrll _rtld_linkmap_add(Obj_Entry *obj)
1372 1.1 cgd {
1373 1.15 christos struct link_map *l = &obj->linkmap;
1374 1.15 christos struct link_map *prev;
1375 1.1 cgd
1376 1.15 christos obj->linkmap.l_name = obj->path;
1377 1.68 mycroft obj->linkmap.l_addr = obj->relocbase;
1378 1.15 christos obj->linkmap.l_ld = obj->dynamic;
1379 1.6 mhitch #ifdef __mips__
1380 1.72 mycroft /* XXX This field is not standard and will be removed eventually. */
1381 1.15 christos obj->linkmap.l_offs = obj->relocbase;
1382 1.6 mhitch #endif
1383 1.1 cgd
1384 1.15 christos if (_rtld_debug.r_map == NULL) {
1385 1.15 christos _rtld_debug.r_map = l;
1386 1.15 christos return;
1387 1.15 christos }
1388 1.109 skrll
1389 1.109 skrll /*
1390 1.109 skrll * Scan to the end of the list, but not past the entry for the
1391 1.109 skrll * dynamic linker, which we want to keep at the very end.
1392 1.109 skrll */
1393 1.109 skrll for (prev = _rtld_debug.r_map;
1394 1.109 skrll prev->l_next != NULL && prev->l_next != &_rtld_objself.linkmap;
1395 1.109 skrll prev = prev->l_next);
1396 1.109 skrll
1397 1.15 christos l->l_prev = prev;
1398 1.109 skrll l->l_next = prev->l_next;
1399 1.109 skrll if (l->l_next != NULL)
1400 1.109 skrll l->l_next->l_prev = l;
1401 1.15 christos prev->l_next = l;
1402 1.1 cgd }
1403 1.1 cgd
1404 1.1 cgd void
1405 1.98 skrll _rtld_linkmap_delete(Obj_Entry *obj)
1406 1.1 cgd {
1407 1.15 christos struct link_map *l = &obj->linkmap;
1408 1.1 cgd
1409 1.15 christos if (l->l_prev == NULL) {
1410 1.15 christos if ((_rtld_debug.r_map = l->l_next) != NULL)
1411 1.15 christos l->l_next->l_prev = NULL;
1412 1.15 christos return;
1413 1.15 christos }
1414 1.15 christos if ((l->l_prev->l_next = l->l_next) != NULL)
1415 1.15 christos l->l_next->l_prev = l->l_prev;
1416 1.25 mycroft }
1417 1.25 mycroft
1418 1.25 mycroft static Obj_Entry *
1419 1.25 mycroft _rtld_obj_from_addr(const void *addr)
1420 1.25 mycroft {
1421 1.25 mycroft Obj_Entry *obj;
1422 1.25 mycroft
1423 1.25 mycroft for (obj = _rtld_objlist; obj != NULL; obj = obj->next) {
1424 1.25 mycroft if (addr < (void *) obj->mapbase)
1425 1.25 mycroft continue;
1426 1.67 mycroft if (addr < (void *) (obj->mapbase + obj->mapsize))
1427 1.25 mycroft return obj;
1428 1.25 mycroft }
1429 1.25 mycroft return NULL;
1430 1.25 mycroft }
1431 1.25 mycroft
1432 1.25 mycroft static void
1433 1.117 ad _rtld_objlist_clear(Objlist *list)
1434 1.117 ad {
1435 1.117 ad while (!SIMPLEQ_EMPTY(list)) {
1436 1.117 ad Objlist_Entry* elm = SIMPLEQ_FIRST(list);
1437 1.117 ad SIMPLEQ_REMOVE_HEAD(list, link);
1438 1.117 ad xfree(elm);
1439 1.117 ad }
1440 1.117 ad }
1441 1.117 ad
1442 1.117 ad static void
1443 1.98 skrll _rtld_objlist_remove(Objlist *list, Obj_Entry *obj)
1444 1.25 mycroft {
1445 1.25 mycroft Objlist_Entry *elm;
1446 1.25 mycroft
1447 1.25 mycroft if ((elm = _rtld_objlist_find(list, obj)) != NULL) {
1448 1.50 lukem SIMPLEQ_REMOVE(list, elm, Struct_Objlist_Entry, link);
1449 1.116 ad xfree(elm);
1450 1.25 mycroft }
1451 1.1 cgd }
1452 1.141 joerg
1453 1.141 joerg #define RTLD_EXCLUSIVE_MASK 0x80000000U
1454 1.141 joerg static volatile unsigned int _rtld_mutex;
1455 1.141 joerg static volatile unsigned int _rtld_waiter_exclusive;
1456 1.141 joerg static volatile unsigned int _rtld_waiter_shared;
1457 1.141 joerg
1458 1.141 joerg void
1459 1.141 joerg _rtld_shared_enter(void)
1460 1.141 joerg {
1461 1.141 joerg unsigned int cur;
1462 1.141 joerg lwpid_t waiter, self = 0;
1463 1.141 joerg
1464 1.141 joerg membar_enter();
1465 1.141 joerg
1466 1.141 joerg for (;;) {
1467 1.141 joerg cur = _rtld_mutex;
1468 1.141 joerg /*
1469 1.141 joerg * First check if we are currently not exclusively locked.
1470 1.141 joerg */
1471 1.141 joerg if ((cur & RTLD_EXCLUSIVE_MASK) == 0) {
1472 1.141 joerg /* Yes, so increment use counter */
1473 1.141 joerg if (atomic_cas_uint(&_rtld_mutex, cur, cur + 1) != cur)
1474 1.141 joerg continue;
1475 1.141 joerg return;
1476 1.141 joerg }
1477 1.141 joerg /*
1478 1.141 joerg * Someone has an exclusive lock. Puts us on the waiter list.
1479 1.141 joerg */
1480 1.141 joerg if (!self)
1481 1.141 joerg self = _lwp_self();
1482 1.141 joerg if (cur == (self | RTLD_EXCLUSIVE_MASK)) {
1483 1.141 joerg if (_rtld_mutex_may_recurse)
1484 1.141 joerg return;
1485 1.141 joerg _rtld_error("dead lock detected");
1486 1.141 joerg _rtld_die();
1487 1.141 joerg }
1488 1.141 joerg waiter = atomic_swap_uint(&_rtld_waiter_shared, self);
1489 1.141 joerg /*
1490 1.141 joerg * Check for race against _rtld_exclusive_exit before sleeping.
1491 1.141 joerg */
1492 1.141 joerg if ((_rtld_mutex & RTLD_EXCLUSIVE_MASK) ||
1493 1.141 joerg _rtld_waiter_exclusive)
1494 1.141 joerg _lwp_park(NULL, -1, __UNVOLATILE(&_rtld_mutex), NULL);
1495 1.141 joerg /* Try to remove us from the waiter list. */
1496 1.141 joerg atomic_cas_uint(&_rtld_waiter_shared, self, 0);
1497 1.141 joerg if (waiter)
1498 1.141 joerg _lwp_unpark(waiter, __UNVOLATILE(&_rtld_mutex));
1499 1.141 joerg }
1500 1.141 joerg }
1501 1.141 joerg
1502 1.141 joerg void
1503 1.141 joerg _rtld_shared_exit(void)
1504 1.141 joerg {
1505 1.141 joerg lwpid_t waiter;
1506 1.141 joerg
1507 1.141 joerg /*
1508 1.141 joerg * Shared lock taken after an exclusive lock.
1509 1.141 joerg * Just assume this is a partial recursion.
1510 1.141 joerg */
1511 1.141 joerg if (_rtld_mutex & RTLD_EXCLUSIVE_MASK)
1512 1.141 joerg return;
1513 1.141 joerg
1514 1.141 joerg /*
1515 1.141 joerg * Wakeup LWPs waiting for an exclusive lock if this is the last
1516 1.141 joerg * LWP on the shared lock.
1517 1.141 joerg */
1518 1.141 joerg if (atomic_dec_uint_nv(&_rtld_mutex))
1519 1.141 joerg return;
1520 1.141 joerg if ((waiter = _rtld_waiter_exclusive) != 0)
1521 1.141 joerg _lwp_unpark(waiter, __UNVOLATILE(&_rtld_mutex));
1522 1.141 joerg
1523 1.141 joerg membar_exit();
1524 1.141 joerg }
1525 1.141 joerg
1526 1.141 joerg void
1527 1.148 joerg _rtld_exclusive_enter(sigset_t *mask)
1528 1.141 joerg {
1529 1.141 joerg lwpid_t waiter, self = _lwp_self();
1530 1.141 joerg unsigned int locked_value = (unsigned int)self | RTLD_EXCLUSIVE_MASK;
1531 1.141 joerg unsigned int cur;
1532 1.148 joerg sigset_t blockmask;
1533 1.148 joerg
1534 1.148 joerg sigfillset(&blockmask);
1535 1.148 joerg sigprocmask(SIG_BLOCK, &blockmask, mask);
1536 1.141 joerg
1537 1.141 joerg membar_enter();
1538 1.141 joerg
1539 1.141 joerg for (;;) {
1540 1.141 joerg if (atomic_cas_uint(&_rtld_mutex, 0, locked_value) == 0)
1541 1.141 joerg break;
1542 1.141 joerg waiter = atomic_swap_uint(&_rtld_waiter_exclusive, self);
1543 1.141 joerg cur = _rtld_mutex;
1544 1.141 joerg if (cur == locked_value) {
1545 1.141 joerg _rtld_error("dead lock detected");
1546 1.141 joerg _rtld_die();
1547 1.141 joerg }
1548 1.141 joerg if (cur)
1549 1.141 joerg _lwp_park(NULL, -1, __UNVOLATILE(&_rtld_mutex), NULL);
1550 1.141 joerg atomic_cas_uint(&_rtld_waiter_exclusive, self, 0);
1551 1.141 joerg if (waiter)
1552 1.141 joerg _lwp_unpark(waiter, __UNVOLATILE(&_rtld_mutex));
1553 1.141 joerg }
1554 1.141 joerg }
1555 1.141 joerg
1556 1.141 joerg void
1557 1.148 joerg _rtld_exclusive_exit(sigset_t *mask)
1558 1.141 joerg {
1559 1.141 joerg lwpid_t waiter;
1560 1.141 joerg
1561 1.141 joerg _rtld_mutex = 0;
1562 1.141 joerg if ((waiter = _rtld_waiter_exclusive) != 0)
1563 1.141 joerg _lwp_unpark(waiter, __UNVOLATILE(&_rtld_mutex));
1564 1.141 joerg
1565 1.141 joerg if ((waiter = _rtld_waiter_shared) != 0)
1566 1.141 joerg _lwp_unpark(waiter, __UNVOLATILE(&_rtld_mutex));
1567 1.141 joerg
1568 1.141 joerg membar_exit();
1569 1.148 joerg sigprocmask(SIG_SETMASK, mask, NULL);
1570 1.141 joerg }
1571