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