blob: 1d5e2977825e612540886a636e3343b10031a8a9 [file] [log] [blame]
Rich Felkerf419bcb2012-08-26 21:09:26 -04001#define _GNU_SOURCE
Rich Felker51e2d832011-06-18 19:48:42 -04002#include <stdio.h>
3#include <stdlib.h>
4#include <string.h>
5#include <unistd.h>
6#include <stdint.h>
7#include <elf.h>
8#include <sys/mman.h>
9#include <limits.h>
10#include <stdint.h>
11#include <fcntl.h>
12#include <sys/stat.h>
13#include <errno.h>
14#include <limits.h>
15#include <elf.h>
Rich Felker18c0e022012-10-31 21:27:48 -040016#include <link.h>
Rich Felker6b3d5e52011-06-26 17:39:17 -040017#include <setjmp.h>
Rich Felker59ab43f2011-06-26 19:23:28 -040018#include <pthread.h>
Rich Felker2719cc82011-08-16 00:24:36 -040019#include <ctype.h>
Rich Felker59ab43f2011-06-26 19:23:28 -040020#include <dlfcn.h>
Rich Felker8431d792012-10-04 16:35:46 -040021#include "pthread_impl.h"
22#include "libc.h"
Rich Felker51e2d832011-06-18 19:48:42 -040023
Rich Felkera9e85c02012-03-23 00:28:20 -040024static int errflag;
Rich Felkera5d10eb2012-04-23 12:03:31 -040025static char errbuf[128];
Rich Felkera9e85c02012-03-23 00:28:20 -040026
Rich Felkere864a292012-07-11 01:47:30 -040027#ifdef SHARED
Rich Felkera9e85c02012-03-23 00:28:20 -040028
Rich Felker51e2d832011-06-18 19:48:42 -040029#if ULONG_MAX == 0xffffffff
30typedef Elf32_Ehdr Ehdr;
31typedef Elf32_Phdr Phdr;
32typedef Elf32_Sym Sym;
33#define R_TYPE(x) ((x)&255)
34#define R_SYM(x) ((x)>>8)
35#else
36typedef Elf64_Ehdr Ehdr;
37typedef Elf64_Phdr Phdr;
38typedef Elf64_Sym Sym;
39#define R_TYPE(x) ((x)&0xffffffff)
40#define R_SYM(x) ((x)>>32)
41#endif
42
Rich Felkercf3fd3d2012-10-06 01:22:51 -040043#define MAXP2(a,b) (-(-(a)&-(b)))
44#define ALIGN(x,y) ((x)+(y)-1 & -(y))
45
Rich Felker3ec8d292012-04-25 00:05:42 -040046struct debug {
47 int ver;
48 void *head;
49 void (*bp)(void);
50 int state;
51 void *base;
52};
53
54struct dso {
55 unsigned char *base;
56 char *name;
Rich Felker51e2d832011-06-18 19:48:42 -040057 size_t *dynv;
Rich Felker3ec8d292012-04-25 00:05:42 -040058 struct dso *next, *prev;
59
Rich Felker18c0e022012-10-31 21:27:48 -040060 Phdr *phdr;
61 int phnum;
Rich Felker3ec8d292012-04-25 00:05:42 -040062 int refcnt;
Rich Felker51e2d832011-06-18 19:48:42 -040063 Sym *syms;
Rich Felker596d60c2011-06-18 22:52:01 -040064 uint32_t *hashtab;
Rich Felker2bd05a42012-08-25 17:13:28 -040065 uint32_t *ghashtab;
Rich Felker51e2d832011-06-18 19:48:42 -040066 char *strings;
Rich Felker51e2d832011-06-18 19:48:42 -040067 unsigned char *map;
68 size_t map_len;
69 dev_t dev;
70 ino_t ino;
Rich Felker6343ac82012-06-09 21:20:44 -040071 signed char global;
Rich Felker700a8152012-02-07 20:29:29 -050072 char relocated;
73 char constructed;
Rich Felker59ab43f2011-06-26 19:23:28 -040074 struct dso **deps;
Rich Felkerbc6a35f2012-10-04 20:04:13 -040075 void *tls_image;
Rich Felker9c748562012-10-04 22:48:33 -040076 size_t tls_len, tls_size, tls_align, tls_id, tls_offset;
Rich Felkerdcd60372012-10-05 11:51:50 -040077 void **new_dtv;
78 unsigned char *new_tls;
79 int new_dtv_idx, new_tls_idx;
Rich Felkerf4f77c02012-10-05 13:09:09 -040080 struct dso *fini_next;
Rich Felker5c1909a2012-05-27 16:01:44 -040081 char *shortname;
Rich Felker6b3d5e52011-06-26 17:39:17 -040082 char buf[];
Rich Felker51e2d832011-06-18 19:48:42 -040083};
84
Rich Felker9c748562012-10-04 22:48:33 -040085struct symdef {
86 Sym *sym;
87 struct dso *dso;
88};
89
Rich Felker59f40862012-08-05 13:46:39 -040090#include "reloc.h"
91
Rich Felker58aa5f42012-05-03 20:42:45 -040092void __init_ssp(size_t *);
Rich Felkerbc6a35f2012-10-04 20:04:13 -040093void *__install_initial_tls(void *);
Rich Felker60872cf2012-04-24 18:07:59 -040094
Rich Felkere23d3582012-10-13 23:25:20 -040095static struct dso *head, *tail, *ldso, *fini_head;
Rich Felker191ebca2011-06-30 23:02:27 -040096static char *env_path, *sys_path, *r_path;
Rich Felker18c0e022012-10-31 21:27:48 -040097static unsigned long long gencnt;
Rich Felker60872cf2012-04-24 18:07:59 -040098static int ssp_used;
Rich Felker6b3d5e52011-06-26 17:39:17 -040099static int runtime;
Rich Felker5c1909a2012-05-27 16:01:44 -0400100static int ldd_mode;
Rich Felker04109502012-08-18 16:00:23 -0400101static int ldso_fail;
Rich Felker4d07e552013-01-23 22:07:45 -0500102static int noload;
Rich Felker6b3d5e52011-06-26 17:39:17 -0400103static jmp_buf rtld_fail;
Rich Felker59ab43f2011-06-26 19:23:28 -0400104static pthread_rwlock_t lock;
Rich Felker3ec8d292012-04-25 00:05:42 -0400105static struct debug debug;
Rich Felkerc62b9f32012-10-14 19:56:50 -0400106static size_t tls_cnt, tls_offset, tls_align = 4*sizeof(size_t);
Rich Felkerf4f77c02012-10-05 13:09:09 -0400107static pthread_mutex_t init_fini_lock = { ._m_type = PTHREAD_MUTEX_RECURSIVE };
Rich Felker3ec8d292012-04-25 00:05:42 -0400108
109struct debug *_dl_debug_addr = &debug;
Rich Felker51e2d832011-06-18 19:48:42 -0400110
Rich Felker0a96a372012-10-07 21:43:46 -0400111#define AUX_CNT 38
Rich Felker51e2d832011-06-18 19:48:42 -0400112#define DYN_CNT 34
113
114static void decode_vec(size_t *v, size_t *a, size_t cnt)
115{
116 memset(a, 0, cnt*sizeof(size_t));
117 for (; v[0]; v+=2) if (v[0]<cnt) {
118 a[0] |= 1ULL<<v[0];
119 a[v[0]] = v[1];
120 }
121}
122
Rich Felker2bd05a42012-08-25 17:13:28 -0400123static int search_vec(size_t *v, size_t *r, size_t key)
124{
125 for (; v[0]!=key; v+=2)
126 if (!v[0]) return 0;
127 *r = v[1];
128 return 1;
129}
130
131static uint32_t sysv_hash(const char *s0)
Rich Felker51e2d832011-06-18 19:48:42 -0400132{
Rich Felker2adf2fb2012-01-17 00:34:58 -0500133 const unsigned char *s = (void *)s0;
Rich Felker51e2d832011-06-18 19:48:42 -0400134 uint_fast32_t h = 0;
135 while (*s) {
136 h = 16*h + *s++;
137 h ^= h>>24 & 0xf0;
138 }
139 return h & 0xfffffff;
140}
141
Rich Felker2bd05a42012-08-25 17:13:28 -0400142static uint32_t gnu_hash(const char *s0)
143{
144 const unsigned char *s = (void *)s0;
145 uint_fast32_t h = 5381;
146 for (; *s; s++)
147 h = h*33 + *s;
148 return h;
149}
150
151static Sym *sysv_lookup(const char *s, uint32_t h, struct dso *dso)
Rich Felker51e2d832011-06-18 19:48:42 -0400152{
153 size_t i;
Rich Felker05eff012012-08-05 02:38:35 -0400154 Sym *syms = dso->syms;
155 uint32_t *hashtab = dso->hashtab;
156 char *strings = dso->strings;
Rich Felker51e2d832011-06-18 19:48:42 -0400157 for (i=hashtab[2+h%hashtab[0]]; i; i=hashtab[2+hashtab[0]+i]) {
158 if (!strcmp(s, strings+syms[i].st_name))
159 return syms+i;
160 }
161 return 0;
162}
163
Rich Felker2bd05a42012-08-25 17:13:28 -0400164static Sym *gnu_lookup(const char *s, uint32_t h1, struct dso *dso)
165{
166 Sym *sym;
167 char *strings;
168 uint32_t *hashtab = dso->ghashtab;
169 uint32_t nbuckets = hashtab[0];
170 uint32_t *buckets = hashtab + 4 + hashtab[2]*(sizeof(size_t)/4);
171 uint32_t h2;
172 uint32_t *hashval;
173 uint32_t n = buckets[h1 % nbuckets];
174
175 if (!n) return 0;
176
177 strings = dso->strings;
178 sym = dso->syms + n;
179 hashval = buckets + nbuckets + (n - hashtab[1]);
180
181 for (h1 |= 1; ; sym++) {
182 h2 = *hashval++;
183 if ((h1 == (h2|1)) && !strcmp(s, strings + sym->st_name))
184 return sym;
185 if (h2 & 1) break;
186 }
187
188 return 0;
189}
190
Rich Felker9c748562012-10-04 22:48:33 -0400191#define OK_TYPES (1<<STT_NOTYPE | 1<<STT_OBJECT | 1<<STT_FUNC | 1<<STT_COMMON | 1<<STT_TLS)
Rich Felker427173b2011-07-24 02:19:47 -0400192#define OK_BINDS (1<<STB_GLOBAL | 1<<STB_WEAK)
Rich Felker51e2d832011-06-18 19:48:42 -0400193
Rich Felker9c748562012-10-04 22:48:33 -0400194static struct symdef find_sym(struct dso *dso, const char *s, int need_def)
Rich Felker51e2d832011-06-18 19:48:42 -0400195{
Rich Felker2bd05a42012-08-25 17:13:28 -0400196 uint32_t h = 0, gh = 0;
Rich Felker9c748562012-10-04 22:48:33 -0400197 struct symdef def = {0};
Rich Felker2bd05a42012-08-25 17:13:28 -0400198 if (dso->ghashtab) {
199 gh = gnu_hash(s);
Rich Felker2bd05a42012-08-25 17:13:28 -0400200 if (gh == 0x1f4039c9 && !strcmp(s, "__stack_chk_fail")) ssp_used = 1;
201 } else {
202 h = sysv_hash(s);
Rich Felker2bd05a42012-08-25 17:13:28 -0400203 if (h == 0x595a4cc && !strcmp(s, "__stack_chk_fail")) ssp_used = 1;
204 }
Rich Felker51e2d832011-06-18 19:48:42 -0400205 for (; dso; dso=dso->next) {
Rich Felker59ab43f2011-06-26 19:23:28 -0400206 Sym *sym;
207 if (!dso->global) continue;
Rich Felker2bd05a42012-08-25 17:13:28 -0400208 if (dso->ghashtab) {
209 if (!gh) gh = gnu_hash(s);
210 sym = gnu_lookup(s, gh, dso);
211 } else {
212 if (!h) h = sysv_hash(s);
213 sym = sysv_lookup(s, h, dso);
214 }
Rich Felkerbd174312012-10-06 01:36:11 -0400215 if (!sym) continue;
216 if (!sym->st_shndx)
217 if (need_def || (sym->st_info&0xf) == STT_TLS)
218 continue;
219 if (!sym->st_value)
220 if ((sym->st_info&0xf) != STT_TLS)
221 continue;
222 if (!(1<<(sym->st_info&0xf) & OK_TYPES)) continue;
223 if (!(1<<(sym->st_info>>4) & OK_BINDS)) continue;
224
225 if (def.sym && sym->st_info>>4 == STB_WEAK) continue;
226 def.sym = sym;
227 def.dso = dso;
228 if (sym->st_info>>4 == STB_GLOBAL) break;
Rich Felker51e2d832011-06-18 19:48:42 -0400229 }
Rich Felker427173b2011-07-24 02:19:47 -0400230 return def;
Rich Felker51e2d832011-06-18 19:48:42 -0400231}
232
Rich Felker87d13a42012-08-05 02:49:02 -0400233static void do_relocs(struct dso *dso, size_t *rel, size_t rel_size, size_t stride)
Rich Felker51e2d832011-06-18 19:48:42 -0400234{
Rich Felker87d13a42012-08-05 02:49:02 -0400235 unsigned char *base = dso->base;
236 Sym *syms = dso->syms;
237 char *strings = dso->strings;
Rich Felker51e2d832011-06-18 19:48:42 -0400238 Sym *sym;
239 const char *name;
Rich Felker51e2d832011-06-18 19:48:42 -0400240 void *ctx;
241 int type;
242 int sym_index;
Rich Felker9c748562012-10-04 22:48:33 -0400243 struct symdef def;
Rich Felker51e2d832011-06-18 19:48:42 -0400244
245 for (; rel_size; rel+=stride, rel_size-=stride*sizeof(size_t)) {
Rich Felker51e2d832011-06-18 19:48:42 -0400246 type = R_TYPE(rel[1]);
247 sym_index = R_SYM(rel[1]);
248 if (sym_index) {
249 sym = syms + sym_index;
250 name = strings + sym->st_name;
Rich Felker7cb44cd2012-08-05 02:44:32 -0400251 ctx = IS_COPY(type) ? head->next : head;
Rich Felker9c748562012-10-04 22:48:33 -0400252 def = find_sym(ctx, name, IS_PLT(type));
253 if (!def.sym && sym->st_info>>4 != STB_WEAK) {
Rich Felkera5d10eb2012-04-23 12:03:31 -0400254 snprintf(errbuf, sizeof errbuf,
255 "Error relocating %s: %s: symbol not found",
Rich Felker87d13a42012-08-05 02:49:02 -0400256 dso->name, name);
Rich Felker6b3d5e52011-06-26 17:39:17 -0400257 if (runtime) longjmp(rtld_fail, 1);
Rich Felkera5d10eb2012-04-23 12:03:31 -0400258 dprintf(2, "%s\n", errbuf);
Rich Felker04109502012-08-18 16:00:23 -0400259 ldso_fail = 1;
260 continue;
Rich Felker6b3d5e52011-06-26 17:39:17 -0400261 }
Rich Felker7d9a5c62012-08-05 14:03:17 -0400262 } else {
Rich Felker9c748562012-10-04 22:48:33 -0400263 sym = 0;
264 def.sym = 0;
265 def.dso = 0;
Rich Felker51e2d832011-06-18 19:48:42 -0400266 }
Rich Felker9c748562012-10-04 22:48:33 -0400267 do_single_reloc(dso, base, (void *)(base + rel[0]), type,
268 stride>2 ? rel[2] : 0, sym, sym?sym->st_size:0, def,
269 def.sym?(size_t)(def.dso->base+def.sym->st_value):0);
Rich Felker51e2d832011-06-18 19:48:42 -0400270 }
271}
272
Rich Felker6717e622011-06-28 19:40:14 -0400273/* A huge hack: to make up for the wastefulness of shared libraries
274 * needing at least a page of dirty memory even if they have no global
275 * data, we reclaim the gaps at the beginning and end of writable maps
276 * and "donate" them to the heap by setting up minimal malloc
277 * structures and then freeing them. */
278
279static void reclaim(unsigned char *base, size_t start, size_t end)
280{
281 size_t *a, *z;
282 start = start + 6*sizeof(size_t)-1 & -4*sizeof(size_t);
283 end = (end & -4*sizeof(size_t)) - 2*sizeof(size_t);
284 if (start>end || end-start < 4*sizeof(size_t)) return;
285 a = (size_t *)(base + start);
286 z = (size_t *)(base + end);
287 a[-2] = 1;
288 a[-1] = z[0] = end-start + 2*sizeof(size_t) | 1;
289 z[1] = 1;
290 free(a);
291}
292
293static void reclaim_gaps(unsigned char *base, Phdr *ph, size_t phent, size_t phcnt)
294{
295 for (; phcnt--; ph=(void *)((char *)ph+phent)) {
296 if (ph->p_type!=PT_LOAD) continue;
297 if ((ph->p_flags&(PF_R|PF_W))!=(PF_R|PF_W)) continue;
298 reclaim(base, ph->p_vaddr & -PAGE_SIZE, ph->p_vaddr);
299 reclaim(base, ph->p_vaddr+ph->p_memsz,
300 ph->p_vaddr+ph->p_memsz+PAGE_SIZE-1 & -PAGE_SIZE);
301 }
302}
303
Rich Felkerbc6a35f2012-10-04 20:04:13 -0400304static void *map_library(int fd, struct dso *dso)
Rich Felker51e2d832011-06-18 19:48:42 -0400305{
Rich Felker59633c72011-06-25 12:26:08 -0400306 Ehdr buf[(896+sizeof(Ehdr))/sizeof(Ehdr)];
Rich Felker51e2d832011-06-18 19:48:42 -0400307 size_t phsize;
308 size_t addr_min=SIZE_MAX, addr_max=0, map_len;
309 size_t this_min, this_max;
310 off_t off_start;
311 Ehdr *eh;
Rich Felker30763fd2013-07-10 14:38:20 -0400312 Phdr *ph, *ph0;
Rich Felker51e2d832011-06-18 19:48:42 -0400313 unsigned prot;
314 unsigned char *map, *base;
315 size_t dyn;
Rich Felkerbc6a35f2012-10-04 20:04:13 -0400316 size_t tls_image=0;
Rich Felker51e2d832011-06-18 19:48:42 -0400317 size_t i;
318
319 ssize_t l = read(fd, buf, sizeof buf);
320 if (l<sizeof *eh) return 0;
Rich Felker59633c72011-06-25 12:26:08 -0400321 eh = buf;
Rich Felker51e2d832011-06-18 19:48:42 -0400322 phsize = eh->e_phentsize * eh->e_phnum;
323 if (phsize + sizeof *eh > l) return 0;
324 if (eh->e_phoff + phsize > l) {
Rich Felker59633c72011-06-25 12:26:08 -0400325 l = pread(fd, buf+1, phsize, eh->e_phoff);
Rich Felker51e2d832011-06-18 19:48:42 -0400326 if (l != phsize) return 0;
Rich Felker30763fd2013-07-10 14:38:20 -0400327 ph = ph0 = (void *)(buf + 1);
328 } else {
329 ph = ph0 = (void *)((char *)buf + eh->e_phoff);
Rich Felker51e2d832011-06-18 19:48:42 -0400330 }
Rich Felker51e2d832011-06-18 19:48:42 -0400331 for (i=eh->e_phnum; i; i--, ph=(void *)((char *)ph+eh->e_phentsize)) {
332 if (ph->p_type == PT_DYNAMIC)
333 dyn = ph->p_vaddr;
Rich Felkerbc6a35f2012-10-04 20:04:13 -0400334 if (ph->p_type == PT_TLS) {
335 tls_image = ph->p_vaddr;
336 dso->tls_align = ph->p_align;
337 dso->tls_len = ph->p_filesz;
338 dso->tls_size = ph->p_memsz;
339 }
Rich Felker51e2d832011-06-18 19:48:42 -0400340 if (ph->p_type != PT_LOAD) continue;
341 if (ph->p_vaddr < addr_min) {
342 addr_min = ph->p_vaddr;
343 off_start = ph->p_offset;
344 prot = (((ph->p_flags&PF_R) ? PROT_READ : 0) |
345 ((ph->p_flags&PF_W) ? PROT_WRITE: 0) |
346 ((ph->p_flags&PF_X) ? PROT_EXEC : 0));
347 }
348 if (ph->p_vaddr+ph->p_memsz > addr_max) {
349 addr_max = ph->p_vaddr+ph->p_memsz;
350 }
351 }
352 if (!dyn) return 0;
353 addr_max += PAGE_SIZE-1;
354 addr_max &= -PAGE_SIZE;
355 addr_min &= -PAGE_SIZE;
356 off_start &= -PAGE_SIZE;
357 map_len = addr_max - addr_min + off_start;
358 /* The first time, we map too much, possibly even more than
359 * the length of the file. This is okay because we will not
360 * use the invalid part; we just need to reserve the right
361 * amount of virtual address space to map over later. */
Rich Felkerbf301002011-06-28 14:20:41 -0400362 map = mmap((void *)addr_min, map_len, prot, MAP_PRIVATE, fd, off_start);
Rich Felker51e2d832011-06-18 19:48:42 -0400363 if (map==MAP_FAILED) return 0;
364 base = map - addr_min;
Rich Felker30763fd2013-07-10 14:38:20 -0400365 dso->phdr = 0;
366 dso->phnum = 0;
367 for (ph=ph0, i=eh->e_phnum; i; i--, ph=(void *)((char *)ph+eh->e_phentsize)) {
Rich Felker51e2d832011-06-18 19:48:42 -0400368 if (ph->p_type != PT_LOAD) continue;
Rich Felker30763fd2013-07-10 14:38:20 -0400369 /* Check if the programs headers are in this load segment, and
370 * if so, record the address for use by dl_iterate_phdr. */
371 if (!dso->phdr && eh->e_phoff >= ph->p_offset
372 && eh->e_phoff+phsize <= ph->p_offset+ph->p_filesz) {
373 dso->phdr = (void *)(base + ph->p_vaddr
374 + (eh->e_phoff-ph->p_offset));
375 dso->phnum = eh->e_phnum;
376 }
Rich Felker51e2d832011-06-18 19:48:42 -0400377 /* Reuse the existing mapping for the lowest-address LOAD */
378 if ((ph->p_vaddr & -PAGE_SIZE) == addr_min) continue;
379 this_min = ph->p_vaddr & -PAGE_SIZE;
380 this_max = ph->p_vaddr+ph->p_memsz+PAGE_SIZE-1 & -PAGE_SIZE;
381 off_start = ph->p_offset & -PAGE_SIZE;
382 prot = (((ph->p_flags&PF_R) ? PROT_READ : 0) |
383 ((ph->p_flags&PF_W) ? PROT_WRITE: 0) |
384 ((ph->p_flags&PF_X) ? PROT_EXEC : 0));
Rich Felkerf7d15dc2012-06-06 11:21:28 -0400385 if (mmap(base+this_min, this_max-this_min, prot, MAP_PRIVATE|MAP_FIXED, fd, off_start) == MAP_FAILED)
386 goto error;
Rich Felker51e2d832011-06-18 19:48:42 -0400387 if (ph->p_memsz > ph->p_filesz) {
388 size_t brk = (size_t)base+ph->p_vaddr+ph->p_filesz;
389 size_t pgbrk = brk+PAGE_SIZE-1 & -PAGE_SIZE;
390 memset((void *)brk, 0, pgbrk-brk & PAGE_SIZE-1);
Rich Felkerf7d15dc2012-06-06 11:21:28 -0400391 if (pgbrk-(size_t)base < this_max && mmap((void *)pgbrk, (size_t)base+this_max-pgbrk, prot, MAP_PRIVATE|MAP_FIXED|MAP_ANONYMOUS, -1, 0) == MAP_FAILED)
392 goto error;
Rich Felker51e2d832011-06-18 19:48:42 -0400393 }
394 }
Rich Felker9f174132011-06-29 00:29:08 -0400395 for (i=0; ((size_t *)(base+dyn))[i]; i+=2)
396 if (((size_t *)(base+dyn))[i]==DT_TEXTREL) {
Rich Felkerf7d15dc2012-06-06 11:21:28 -0400397 if (mprotect(map, map_len, PROT_READ|PROT_WRITE|PROT_EXEC) < 0)
398 goto error;
Rich Felker9f174132011-06-29 00:29:08 -0400399 break;
400 }
Rich Felker30763fd2013-07-10 14:38:20 -0400401 if (!runtime) reclaim_gaps(base, ph0, eh->e_phentsize, eh->e_phnum);
Rich Felkerbc6a35f2012-10-04 20:04:13 -0400402 dso->map = map;
403 dso->map_len = map_len;
404 dso->base = base;
405 dso->dynv = (void *)(base+dyn);
406 if (dso->tls_size) dso->tls_image = (void *)(base+tls_image);
Rich Felker51e2d832011-06-18 19:48:42 -0400407 return map;
Rich Felkerf7d15dc2012-06-06 11:21:28 -0400408error:
409 munmap(map, map_len);
410 return 0;
Rich Felker51e2d832011-06-18 19:48:42 -0400411}
412
Rich Felker8c203ea2013-04-20 11:51:58 -0400413static int path_open(const char *name, const char *s, char *buf, size_t buf_size)
Rich Felker568b8072011-06-25 01:56:34 -0400414{
Rich Felker8c203ea2013-04-20 11:51:58 -0400415 size_t l;
416 int fd;
Rich Felker49388f32011-06-25 17:49:16 -0400417 for (;;) {
Rich Felker8c203ea2013-04-20 11:51:58 -0400418 s += strspn(s, ":\n");
419 l = strcspn(s, ":\n");
420 if (l-1 >= INT_MAX) return -1;
421 if (snprintf(buf, buf_size, "%.*s/%s", (int)l, s, name) >= buf_size)
422 continue;
Rich Felkerf2d08cf2012-09-29 17:59:50 -0400423 if ((fd = open(buf, O_RDONLY|O_CLOEXEC))>=0) return fd;
Rich Felker49388f32011-06-25 17:49:16 -0400424 s += l;
Rich Felker568b8072011-06-25 01:56:34 -0400425 }
Rich Felker568b8072011-06-25 01:56:34 -0400426}
427
Rich Felkerc82f4a32012-01-23 00:57:38 -0500428static void decode_dyn(struct dso *p)
429{
430 size_t dyn[DYN_CNT] = {0};
431 decode_vec(p->dynv, dyn, DYN_CNT);
432 p->syms = (void *)(p->base + dyn[DT_SYMTAB]);
Rich Felkerc82f4a32012-01-23 00:57:38 -0500433 p->strings = (void *)(p->base + dyn[DT_STRTAB]);
Rich Felker2bd05a42012-08-25 17:13:28 -0400434 if (dyn[0]&(1<<DT_HASH))
435 p->hashtab = (void *)(p->base + dyn[DT_HASH]);
436 if (search_vec(p->dynv, dyn, DT_GNU_HASH))
437 p->ghashtab = (void *)(p->base + *dyn);
Rich Felkerc82f4a32012-01-23 00:57:38 -0500438}
439
Rich Felker51e2d832011-06-18 19:48:42 -0400440static struct dso *load_library(const char *name)
441{
Rich Felker5c1909a2012-05-27 16:01:44 -0400442 char buf[2*NAME_MAX+2];
Rich Felker0420b872012-07-11 01:41:20 -0400443 const char *pathname;
Rich Felker1d7c4f82012-12-15 23:34:08 -0500444 unsigned char *map;
Rich Felkerbc6a35f2012-10-04 20:04:13 -0400445 struct dso *p, temp_dso = {0};
Rich Felker51e2d832011-06-18 19:48:42 -0400446 int fd;
447 struct stat st;
Rich Felkerdcd60372012-10-05 11:51:50 -0400448 size_t alloc_size;
449 int n_th = 0;
Rich Felker51e2d832011-06-18 19:48:42 -0400450
451 /* Catch and block attempts to reload the implementation itself */
452 if (name[0]=='l' && name[1]=='i' && name[2]=='b') {
453 static const char *rp, reserved[] =
454 "c\0pthread\0rt\0m\0dl\0util\0xnet\0";
455 char *z = strchr(name, '.');
456 if (z) {
457 size_t l = z-name;
458 for (rp=reserved; *rp && memcmp(name+3, rp, l-3); rp+=strlen(rp)+1);
459 if (*rp) {
Rich Felkere23d3582012-10-13 23:25:20 -0400460 if (!ldso->prev) {
461 tail->next = ldso;
462 ldso->prev = tail;
463 tail = ldso->next ? ldso->next : ldso;
Rich Felker51e2d832011-06-18 19:48:42 -0400464 }
Rich Felkere23d3582012-10-13 23:25:20 -0400465 return ldso;
Rich Felker51e2d832011-06-18 19:48:42 -0400466 }
467 }
468 }
Rich Felker59ab43f2011-06-26 19:23:28 -0400469 if (strchr(name, '/')) {
Rich Felker0420b872012-07-11 01:41:20 -0400470 pathname = name;
Rich Felkerf2d08cf2012-09-29 17:59:50 -0400471 fd = open(name, O_RDONLY|O_CLOEXEC);
Rich Felker51e2d832011-06-18 19:48:42 -0400472 } else {
Rich Felker0420b872012-07-11 01:41:20 -0400473 /* Search for the name to see if it's already loaded */
474 for (p=head->next; p; p=p->next) {
475 if (p->shortname && !strcmp(p->shortname, name)) {
476 p->refcnt++;
477 return p;
478 }
479 }
Rich Felker59ab43f2011-06-26 19:23:28 -0400480 if (strlen(name) > NAME_MAX) return 0;
Rich Felker568b8072011-06-25 01:56:34 -0400481 fd = -1;
Rich Felker5c1909a2012-05-27 16:01:44 -0400482 if (r_path) fd = path_open(name, r_path, buf, sizeof buf);
483 if (fd < 0 && env_path) fd = path_open(name, env_path, buf, sizeof buf);
Rich Felker568b8072011-06-25 01:56:34 -0400484 if (fd < 0) {
485 if (!sys_path) {
Rich Felkerf389c492013-07-18 19:29:44 -0400486 char *prefix = 0;
487 size_t prefix_len;
488 if (ldso->name[0]=='/') {
489 char *s, *t, *z;
490 for (s=t=z=ldso->name; *s; s++)
491 if (*s=='/') z=t, t=s;
492 prefix_len = z-ldso->name;
493 if (prefix_len < PATH_MAX)
494 prefix = ldso->name;
495 }
496 if (!prefix) {
497 prefix = "";
498 prefix_len = 0;
499 }
500 char etc_ldso_path[prefix_len + 1
501 + sizeof "/etc/ld-musl-" LDSO_ARCH ".path"];
502 snprintf(etc_ldso_path, sizeof etc_ldso_path,
503 "%.*s/etc/ld-musl-" LDSO_ARCH ".path",
504 (int)prefix_len, prefix);
505 FILE *f = fopen(etc_ldso_path, "rbe");
Rich Felker568b8072011-06-25 01:56:34 -0400506 if (f) {
Rich Felker11bc1732013-06-26 10:17:29 -0400507 if (getdelim(&sys_path, (size_t[1]){0}, 0, f) <= 0) {
Rich Felker59b481d2013-06-26 10:51:36 -0400508 free(sys_path);
Rich Felker11bc1732013-06-26 10:17:29 -0400509 sys_path = "";
Rich Felker65465102012-11-09 13:49:40 -0500510 }
Rich Felker568b8072011-06-25 01:56:34 -0400511 fclose(f);
512 }
513 }
Rich Felker40d5f7e2012-11-08 22:41:16 -0500514 if (!sys_path) sys_path = "/lib:/usr/local/lib:/usr/lib";
515 fd = path_open(name, sys_path, buf, sizeof buf);
Rich Felker51e2d832011-06-18 19:48:42 -0400516 }
Rich Felker0420b872012-07-11 01:41:20 -0400517 pathname = buf;
Rich Felker51e2d832011-06-18 19:48:42 -0400518 }
519 if (fd < 0) return 0;
520 if (fstat(fd, &st) < 0) {
521 close(fd);
522 return 0;
523 }
524 for (p=head->next; p; p=p->next) {
525 if (p->dev == st.st_dev && p->ino == st.st_ino) {
Rich Felker0420b872012-07-11 01:41:20 -0400526 /* If this library was previously loaded with a
527 * pathname but a search found the same inode,
528 * setup its shortname so it can be found by name. */
Rich Felker5f88c0e2012-10-05 12:09:54 -0400529 if (!p->shortname && pathname != name)
530 p->shortname = strrchr(p->name, '/')+1;
Rich Felker51e2d832011-06-18 19:48:42 -0400531 close(fd);
532 p->refcnt++;
533 return p;
534 }
535 }
Rich Felker4d07e552013-01-23 22:07:45 -0500536 map = noload ? 0 : map_library(fd, &temp_dso);
Rich Felker51e2d832011-06-18 19:48:42 -0400537 close(fd);
538 if (!map) return 0;
Rich Felkerdcd60372012-10-05 11:51:50 -0400539
540 /* Allocate storage for the new DSO. When there is TLS, this
541 * storage must include a reservation for all pre-existing
542 * threads to obtain copies of both the new TLS, and an
543 * extended DTV capable of storing an additional slot for
544 * the newly-loaded DSO. */
545 alloc_size = sizeof *p + strlen(pathname) + 1;
546 if (runtime && temp_dso.tls_image) {
547 size_t per_th = temp_dso.tls_size + temp_dso.tls_align
548 + sizeof(void *) * (tls_cnt+3);
Rich Felkere23d3582012-10-13 23:25:20 -0400549 n_th = libc.threads_minus_1 + 1;
Rich Felkerdcd60372012-10-05 11:51:50 -0400550 if (n_th > SSIZE_MAX / per_th) alloc_size = SIZE_MAX;
551 else alloc_size += n_th * per_th;
552 }
553 p = calloc(1, alloc_size);
Rich Felker51e2d832011-06-18 19:48:42 -0400554 if (!p) {
Rich Felker74025c82013-02-02 00:59:25 -0500555 munmap(map, temp_dso.map_len);
Rich Felker51e2d832011-06-18 19:48:42 -0400556 return 0;
557 }
Rich Felkerbc6a35f2012-10-04 20:04:13 -0400558 memcpy(p, &temp_dso, sizeof temp_dso);
Rich Felkerc82f4a32012-01-23 00:57:38 -0500559 decode_dyn(p);
Rich Felker51e2d832011-06-18 19:48:42 -0400560 p->dev = st.st_dev;
561 p->ino = st.st_ino;
Rich Felker51e2d832011-06-18 19:48:42 -0400562 p->refcnt = 1;
Rich Felker6b3d5e52011-06-26 17:39:17 -0400563 p->name = p->buf;
Rich Felker0420b872012-07-11 01:41:20 -0400564 strcpy(p->name, pathname);
565 /* Add a shortname only if name arg was not an explicit pathname. */
566 if (pathname != name) p->shortname = strrchr(p->name, '/')+1;
Rich Felkerdcd60372012-10-05 11:51:50 -0400567 if (p->tls_image) {
Rich Felker017bf142012-10-07 20:19:11 -0400568 if (runtime && !__pthread_self_init()) {
Rich Felker74025c82013-02-02 00:59:25 -0500569 munmap(map, p->map_len);
Rich Felker92e1cd92012-10-06 16:56:35 -0400570 free(p);
Rich Felker92e1cd92012-10-06 16:56:35 -0400571 return 0;
572 }
Rich Felkerdcd60372012-10-05 11:51:50 -0400573 p->tls_id = ++tls_cnt;
Rich Felkercf3fd3d2012-10-06 01:22:51 -0400574 tls_align = MAXP2(tls_align, p->tls_align);
Rich Felker9ec42832012-10-15 18:51:53 -0400575#ifdef TLS_ABOVE_TP
576 p->tls_offset = tls_offset + ( (tls_align-1) &
577 -(tls_offset + (uintptr_t)p->tls_image) );
578 tls_offset += p->tls_size;
579#else
Rich Felkercf3fd3d2012-10-06 01:22:51 -0400580 tls_offset += p->tls_size + p->tls_align - 1;
581 tls_offset -= (tls_offset + (uintptr_t)p->tls_image)
582 & (p->tls_align-1);
583 p->tls_offset = tls_offset;
Rich Felker9ec42832012-10-15 18:51:53 -0400584#endif
Rich Felkerdcd60372012-10-05 11:51:50 -0400585 p->new_dtv = (void *)(-sizeof(size_t) &
586 (uintptr_t)(p->name+strlen(p->name)+sizeof(size_t)));
587 p->new_tls = (void *)(p->new_dtv + n_th*(tls_cnt+1));
588 }
Rich Felker51e2d832011-06-18 19:48:42 -0400589
590 tail->next = p;
591 p->prev = tail;
592 tail = p;
593
Rich Felker1d7c4f82012-12-15 23:34:08 -0500594 if (ldd_mode) dprintf(1, "\t%s => %s (%p)\n", name, pathname, p->base);
Rich Felker5c1909a2012-05-27 16:01:44 -0400595
Rich Felker51e2d832011-06-18 19:48:42 -0400596 return p;
597}
598
599static void load_deps(struct dso *p)
600{
Rich Felker59ab43f2011-06-26 19:23:28 -0400601 size_t i, ndeps=0;
602 struct dso ***deps = &p->deps, **tmp, *dep;
Rich Felker51e2d832011-06-18 19:48:42 -0400603 for (; p; p=p->next) {
604 for (i=0; p->dynv[i]; i+=2) {
Rich Felker191ebca2011-06-30 23:02:27 -0400605 if (p->dynv[i] != DT_RPATH) continue;
606 r_path = (void *)(p->strings + p->dynv[i+1]);
607 }
608 for (i=0; p->dynv[i]; i+=2) {
Rich Felker51e2d832011-06-18 19:48:42 -0400609 if (p->dynv[i] != DT_NEEDED) continue;
Rich Felker59ab43f2011-06-26 19:23:28 -0400610 dep = load_library(p->strings + p->dynv[i+1]);
611 if (!dep) {
Rich Felkera5d10eb2012-04-23 12:03:31 -0400612 snprintf(errbuf, sizeof errbuf,
613 "Error loading shared library %s: %m (needed by %s)",
Rich Felker6b3d5e52011-06-26 17:39:17 -0400614 p->strings + p->dynv[i+1], p->name);
Rich Felkera5d10eb2012-04-23 12:03:31 -0400615 if (runtime) longjmp(rtld_fail, 1);
616 dprintf(2, "%s\n", errbuf);
Rich Felker04109502012-08-18 16:00:23 -0400617 ldso_fail = 1;
618 continue;
Rich Felker6b3d5e52011-06-26 17:39:17 -0400619 }
Rich Felker59ab43f2011-06-26 19:23:28 -0400620 if (runtime) {
621 tmp = realloc(*deps, sizeof(*tmp)*(ndeps+2));
622 if (!tmp) longjmp(rtld_fail, 1);
623 tmp[ndeps++] = dep;
624 tmp[ndeps] = 0;
625 *deps = tmp;
626 }
Rich Felker51e2d832011-06-18 19:48:42 -0400627 }
Rich Felker191ebca2011-06-30 23:02:27 -0400628 r_path = 0;
Rich Felker51e2d832011-06-18 19:48:42 -0400629 }
630}
631
Rich Felker2719cc82011-08-16 00:24:36 -0400632static void load_preload(char *s)
633{
634 int tmp;
635 char *z;
636 for (z=s; *z; s=z) {
637 for ( ; *s && isspace(*s); s++);
638 for (z=s; *z && !isspace(*z); z++);
639 tmp = *z;
640 *z = 0;
641 load_library(s);
642 *z = tmp;
643 }
644}
645
Rich Felker59ab43f2011-06-26 19:23:28 -0400646static void make_global(struct dso *p)
647{
648 for (; p; p=p->next) p->global = 1;
649}
650
Rich Felker51e2d832011-06-18 19:48:42 -0400651static void reloc_all(struct dso *p)
652{
653 size_t dyn[DYN_CNT] = {0};
654 for (; p; p=p->next) {
655 if (p->relocated) continue;
656 decode_vec(p->dynv, dyn, DYN_CNT);
Rich Felkerbabf8202012-08-05 12:50:26 -0400657#ifdef NEED_ARCH_RELOCS
658 do_arch_relocs(p, head);
659#endif
Rich Felker87d13a42012-08-05 02:49:02 -0400660 do_relocs(p, (void *)(p->base+dyn[DT_JMPREL]), dyn[DT_PLTRELSZ],
661 2+(dyn[DT_PLTREL]==DT_RELA));
662 do_relocs(p, (void *)(p->base+dyn[DT_REL]), dyn[DT_RELSZ], 2);
663 do_relocs(p, (void *)(p->base+dyn[DT_RELA]), dyn[DT_RELASZ], 3);
Rich Felker368ba4a2011-06-25 00:18:19 -0400664 p->relocated = 1;
Rich Felker51e2d832011-06-18 19:48:42 -0400665 }
666}
667
Rich Felkerc82f4a32012-01-23 00:57:38 -0500668static size_t find_dyn(Phdr *ph, size_t cnt, size_t stride)
669{
670 for (; cnt--; ph = (void *)((char *)ph + stride))
671 if (ph->p_type == PT_DYNAMIC)
672 return ph->p_vaddr;
673 return 0;
674}
675
Rich Felkerf419bcb2012-08-26 21:09:26 -0400676static void find_map_range(Phdr *ph, size_t cnt, size_t stride, struct dso *p)
677{
678 size_t min_addr = -1, max_addr = 0;
679 for (; cnt--; ph = (void *)((char *)ph + stride)) {
680 if (ph->p_type != PT_LOAD) continue;
681 if (ph->p_vaddr < min_addr)
682 min_addr = ph->p_vaddr;
683 if (ph->p_vaddr+ph->p_memsz > max_addr)
684 max_addr = ph->p_vaddr+ph->p_memsz;
685 }
686 min_addr &= -PAGE_SIZE;
687 max_addr = (max_addr + PAGE_SIZE-1) & -PAGE_SIZE;
688 p->map = p->base + min_addr;
689 p->map_len = max_addr - min_addr;
690}
691
Rich Felkerf4f77c02012-10-05 13:09:09 -0400692static void do_fini()
693{
694 struct dso *p;
695 size_t dyn[DYN_CNT] = {0};
696 for (p=fini_head; p; p=p->fini_next) {
697 if (!p->constructed) continue;
698 decode_vec(p->dynv, dyn, DYN_CNT);
699 ((void (*)(void))(p->base + dyn[DT_FINI]))();
700 }
701}
702
Rich Felker4ce3cb52012-02-06 14:39:09 -0500703static void do_init_fini(struct dso *p)
704{
705 size_t dyn[DYN_CNT] = {0};
Rich Felkere23d3582012-10-13 23:25:20 -0400706 int need_locking = libc.threads_minus_1;
Rich Felkerf4f77c02012-10-05 13:09:09 -0400707 /* Allow recursive calls that arise when a library calls
708 * dlopen from one of its constructors, but block any
709 * other threads until all ctors have finished. */
710 if (need_locking) pthread_mutex_lock(&init_fini_lock);
Rich Felker4ce3cb52012-02-06 14:39:09 -0500711 for (; p; p=p->prev) {
Rich Felkerf4f77c02012-10-05 13:09:09 -0400712 if (p->constructed) continue;
713 p->constructed = 1;
Rich Felker4ce3cb52012-02-06 14:39:09 -0500714 decode_vec(p->dynv, dyn, DYN_CNT);
Rich Felkerf4f77c02012-10-05 13:09:09 -0400715 if (dyn[0] & (1<<DT_FINI)) {
716 p->fini_next = fini_head;
717 fini_head = p;
718 }
Rich Felker4ce3cb52012-02-06 14:39:09 -0500719 if (dyn[0] & (1<<DT_INIT))
720 ((void (*)(void))(p->base + dyn[DT_INIT]))();
Rich Felker509b50e2013-06-29 02:24:02 -0400721 if (!need_locking && libc.threads_minus_1) {
722 need_locking = 1;
723 pthread_mutex_lock(&init_fini_lock);
724 }
Rich Felker4ce3cb52012-02-06 14:39:09 -0500725 }
Rich Felkerf4f77c02012-10-05 13:09:09 -0400726 if (need_locking) pthread_mutex_unlock(&init_fini_lock);
Rich Felker4ce3cb52012-02-06 14:39:09 -0500727}
728
Rich Felker3ec8d292012-04-25 00:05:42 -0400729void _dl_debug_state(void)
730{
731}
732
Rich Felkerdcd60372012-10-05 11:51:50 -0400733void *__copy_tls(unsigned char *mem)
Rich Felkerbc6a35f2012-10-04 20:04:13 -0400734{
Rich Felkercf3fd3d2012-10-06 01:22:51 -0400735 pthread_t td;
Rich Felkerbc6a35f2012-10-04 20:04:13 -0400736 struct dso *p;
Rich Felkercf3fd3d2012-10-06 01:22:51 -0400737
738 if (!tls_cnt) return mem;
739
Rich Felkerbc6a35f2012-10-04 20:04:13 -0400740 void **dtv = (void *)mem;
Rich Felkerdcd60372012-10-05 11:51:50 -0400741 dtv[0] = (void *)tls_cnt;
Rich Felkercf3fd3d2012-10-06 01:22:51 -0400742
Rich Felker9ec42832012-10-15 18:51:53 -0400743#ifdef TLS_ABOVE_TP
744 mem += sizeof(void *) * (tls_cnt+1);
745 mem += -((uintptr_t)mem + sizeof(struct pthread)) & (tls_align-1);
746 td = (pthread_t)mem;
747 mem += sizeof(struct pthread);
748
749 for (p=head; p; p=p->next) {
750 if (!p->tls_id) continue;
751 dtv[p->tls_id] = mem + p->tls_offset;
752 memcpy(dtv[p->tls_id], p->tls_image, p->tls_len);
753 }
754#else
Rich Felkere23d3582012-10-13 23:25:20 -0400755 mem += libc.tls_size - sizeof(struct pthread);
Rich Felkercf3fd3d2012-10-06 01:22:51 -0400756 mem -= (uintptr_t)mem & (tls_align-1);
Rich Felkercf3fd3d2012-10-06 01:22:51 -0400757 td = (pthread_t)mem;
758
759 for (p=head; p; p=p->next) {
Rich Felkerdcd60372012-10-05 11:51:50 -0400760 if (!p->tls_id) continue;
Rich Felkercf3fd3d2012-10-06 01:22:51 -0400761 dtv[p->tls_id] = mem - p->tls_offset;
762 memcpy(dtv[p->tls_id], p->tls_image, p->tls_len);
Rich Felkerbc6a35f2012-10-04 20:04:13 -0400763 }
Rich Felker9ec42832012-10-15 18:51:53 -0400764#endif
Rich Felkercf3fd3d2012-10-06 01:22:51 -0400765 td->dtv = dtv;
766 return td;
Rich Felkerbc6a35f2012-10-04 20:04:13 -0400767}
768
Rich Felkerdcd60372012-10-05 11:51:50 -0400769void *__tls_get_addr(size_t *v)
Rich Felker9b153c02012-10-04 21:01:56 -0400770{
771 pthread_t self = __pthread_self();
Rich Felker44b4d092013-06-03 16:35:59 -0400772 if (v[0]<=(size_t)self->dtv[0] && self->dtv[v[0]])
Rich Felkerdcd60372012-10-05 11:51:50 -0400773 return (char *)self->dtv[v[0]]+v[1];
774
775 /* Block signals to make accessing new TLS async-signal-safe */
776 sigset_t set;
Rich Felker00902c72012-10-06 23:57:51 -0400777 pthread_sigmask(SIG_BLOCK, SIGALL_SET, &set);
Rich Felker44b4d092013-06-03 16:35:59 -0400778 if (v[0]<=(size_t)self->dtv[0] && self->dtv[v[0]]) {
Rich Felkerdcd60372012-10-05 11:51:50 -0400779 pthread_sigmask(SIG_SETMASK, &set, 0);
780 return (char *)self->dtv[v[0]]+v[1];
Rich Felker9b153c02012-10-04 21:01:56 -0400781 }
Rich Felkerdcd60372012-10-05 11:51:50 -0400782
783 /* This is safe without any locks held because, if the caller
784 * is able to request the Nth entry of the DTV, the DSO list
785 * must be valid at least that far out and it was synchronized
786 * at program startup or by an already-completed call to dlopen. */
787 struct dso *p;
788 for (p=head; p->tls_id != v[0]; p=p->next);
789
790 /* Get new DTV space from new DSO if needed */
Rich Felker44b4d092013-06-03 16:35:59 -0400791 if (v[0] > (size_t)self->dtv[0]) {
Rich Felkerdcd60372012-10-05 11:51:50 -0400792 void **newdtv = p->new_dtv +
793 (v[0]+1)*sizeof(void *)*a_fetch_add(&p->new_dtv_idx,1);
Rich Felker44b4d092013-06-03 16:35:59 -0400794 memcpy(newdtv, self->dtv,
Rich Felkerdcd60372012-10-05 11:51:50 -0400795 ((size_t)self->dtv[0]+1) * sizeof(void *));
796 newdtv[0] = (void *)v[0];
797 self->dtv = newdtv;
798 }
799
800 /* Get new TLS memory from new DSO */
801 unsigned char *mem = p->new_tls +
802 (p->tls_size + p->tls_align) * a_fetch_add(&p->new_tls_idx,1);
803 mem += ((uintptr_t)p->tls_image - (uintptr_t)mem) & (p->tls_align-1);
804 self->dtv[v[0]] = mem;
Rich Felkercf3fd3d2012-10-06 01:22:51 -0400805 memcpy(mem, p->tls_image, p->tls_len);
Rich Felkerdcd60372012-10-05 11:51:50 -0400806 pthread_sigmask(SIG_SETMASK, &set, 0);
807 return mem + v[1];
Rich Felker9b153c02012-10-04 21:01:56 -0400808}
809
Rich Felkercf3fd3d2012-10-06 01:22:51 -0400810static void update_tls_size()
811{
Rich Felker9ec42832012-10-15 18:51:53 -0400812 libc.tls_size = ALIGN(
813 (1+tls_cnt) * sizeof(void *) +
814 tls_offset +
815 sizeof(struct pthread) +
816 tls_align * 2,
817 tls_align);
Rich Felkercf3fd3d2012-10-06 01:22:51 -0400818}
819
Rich Felkerc82f4a32012-01-23 00:57:38 -0500820void *__dynlink(int argc, char **argv)
Rich Felker51e2d832011-06-18 19:48:42 -0400821{
Rich Felker731e8ff2012-08-25 17:24:46 -0400822 size_t aux[AUX_CNT] = {0};
Rich Felker51e2d832011-06-18 19:48:42 -0400823 size_t i;
824 Phdr *phdr;
Rich Felker6717e622011-06-28 19:40:14 -0400825 Ehdr *ehdr;
Rich Felker6ab444d2011-07-24 00:54:55 -0400826 static struct dso builtin_dsos[3];
Rich Felkera53de812011-07-24 00:26:12 -0400827 struct dso *const app = builtin_dsos+0;
828 struct dso *const lib = builtin_dsos+1;
Rich Felker6ab444d2011-07-24 00:54:55 -0400829 struct dso *const vdso = builtin_dsos+2;
Rich Felker2719cc82011-08-16 00:24:36 -0400830 char *env_preload=0;
Rich Felkerdbcb3ad2012-08-25 17:31:59 -0400831 size_t vdso_base;
Rich Felker2f2f1152012-11-01 23:49:57 -0400832 size_t *auxv;
Rich Felker51e2d832011-06-18 19:48:42 -0400833
834 /* Find aux vector just past environ[] */
Rich Felker568b8072011-06-25 01:56:34 -0400835 for (i=argc+1; argv[i]; i++)
836 if (!memcmp(argv[i], "LD_LIBRARY_PATH=", 16))
837 env_path = argv[i]+16;
Rich Felker2719cc82011-08-16 00:24:36 -0400838 else if (!memcmp(argv[i], "LD_PRELOAD=", 11))
839 env_preload = argv[i]+11;
Rich Felker51e2d832011-06-18 19:48:42 -0400840 auxv = (void *)(argv+i+1);
841
842 decode_vec(auxv, aux, AUX_CNT);
843
Rich Felker568b8072011-06-25 01:56:34 -0400844 /* Only trust user/env if kernel says we're not suid/sgid */
845 if ((aux[0]&0x7800)!=0x7800 || aux[AT_UID]!=aux[AT_EUID]
Rich Felkera0458832011-08-16 07:46:42 -0400846 || aux[AT_GID]!=aux[AT_EGID] || aux[AT_SECURE]) {
Rich Felker568b8072011-06-25 01:56:34 -0400847 env_path = 0;
Rich Felker2719cc82011-08-16 00:24:36 -0400848 env_preload = 0;
Rich Felker568b8072011-06-25 01:56:34 -0400849 }
850
Rich Felker5c1909a2012-05-27 16:01:44 -0400851 /* If the dynamic linker was invoked as a program itself, AT_BASE
852 * will not be set. In that case, we assume the base address is
853 * the start of the page containing the PHDRs; I don't know any
854 * better approach... */
855 if (!aux[AT_BASE]) {
856 aux[AT_BASE] = aux[AT_PHDR] & -PAGE_SIZE;
857 aux[AT_PHDR] = aux[AT_PHENT] = aux[AT_PHNUM] = 0;
858 }
859
Rich Felkerc82f4a32012-01-23 00:57:38 -0500860 /* The dynamic linker load address is passed by the kernel
861 * in the AUX vector, so this is easy. */
862 lib->base = (void *)aux[AT_BASE];
Rich Felker5c1909a2012-05-27 16:01:44 -0400863 lib->name = lib->shortname = "libc.so";
Rich Felkerc82f4a32012-01-23 00:57:38 -0500864 lib->global = 1;
865 ehdr = (void *)lib->base;
Rich Felker18c0e022012-10-31 21:27:48 -0400866 lib->phnum = ehdr->e_phnum;
867 lib->phdr = (void *)(aux[AT_BASE]+ehdr->e_phoff);
868 find_map_range(lib->phdr, ehdr->e_phnum, ehdr->e_phentsize, lib);
869 lib->dynv = (void *)(lib->base + find_dyn(lib->phdr,
870 ehdr->e_phnum, ehdr->e_phentsize));
Rich Felkerc82f4a32012-01-23 00:57:38 -0500871 decode_dyn(lib);
872
Rich Felker5c1909a2012-05-27 16:01:44 -0400873 if (aux[AT_PHDR]) {
Rich Felker649cec52012-07-13 01:31:02 -0400874 size_t interp_off = 0;
Rich Felkerbc6a35f2012-10-04 20:04:13 -0400875 size_t tls_image = 0;
Rich Felker5c1909a2012-05-27 16:01:44 -0400876 /* Find load address of the main program, via AT_PHDR vs PT_PHDR. */
Rich Felker18c0e022012-10-31 21:27:48 -0400877 app->phdr = phdr = (void *)aux[AT_PHDR];
878 app->phnum = aux[AT_PHNUM];
Rich Felker5c1909a2012-05-27 16:01:44 -0400879 for (i=aux[AT_PHNUM]; i; i--, phdr=(void *)((char *)phdr + aux[AT_PHENT])) {
880 if (phdr->p_type == PT_PHDR)
881 app->base = (void *)(aux[AT_PHDR] - phdr->p_vaddr);
Rich Felker649cec52012-07-13 01:31:02 -0400882 else if (phdr->p_type == PT_INTERP)
883 interp_off = (size_t)phdr->p_vaddr;
Rich Felkerbc6a35f2012-10-04 20:04:13 -0400884 else if (phdr->p_type == PT_TLS) {
885 tls_image = phdr->p_vaddr;
886 app->tls_len = phdr->p_filesz;
887 app->tls_size = phdr->p_memsz;
888 app->tls_align = phdr->p_align;
889 }
Rich Felker5c1909a2012-05-27 16:01:44 -0400890 }
Rich Felkerbc6a35f2012-10-04 20:04:13 -0400891 if (app->tls_size) app->tls_image = (char *)app->base + tls_image;
Rich Felker649cec52012-07-13 01:31:02 -0400892 if (interp_off) lib->name = (char *)app->base + interp_off;
Rich Felker0420b872012-07-11 01:41:20 -0400893 app->name = argv[0];
Rich Felker5c1909a2012-05-27 16:01:44 -0400894 app->dynv = (void *)(app->base + find_dyn(
895 (void *)aux[AT_PHDR], aux[AT_PHNUM], aux[AT_PHENT]));
Rich Felkerf419bcb2012-08-26 21:09:26 -0400896 find_map_range((void *)aux[AT_PHDR],
897 aux[AT_PHNUM], aux[AT_PHENT], app);
Rich Felker5c1909a2012-05-27 16:01:44 -0400898 } else {
899 int fd;
900 char *ldname = argv[0];
Rich Felkera617a8e2012-11-01 23:46:39 -0400901 size_t l = strlen(ldname);
Rich Felker5c1909a2012-05-27 16:01:44 -0400902 if (l >= 3 && !strcmp(ldname+l-3, "ldd")) ldd_mode = 1;
903 *argv++ = (void *)-1;
904 if (argv[0] && !strcmp(argv[0], "--")) *argv++ = (void *)-1;
905 if (!argv[0]) {
906 dprintf(2, "musl libc/dynamic program loader\n");
907 dprintf(2, "usage: %s pathname%s\n", ldname,
908 ldd_mode ? "" : " [args]");
909 _exit(1);
910 }
911 fd = open(argv[0], O_RDONLY);
912 if (fd < 0) {
913 dprintf(2, "%s: cannot load %s: %s\n", ldname, argv[0], strerror(errno));
914 _exit(1);
915 }
916 runtime = 1;
Rich Felkerbc6a35f2012-10-04 20:04:13 -0400917 ehdr = (void *)map_library(fd, app);
Rich Felker5c1909a2012-05-27 16:01:44 -0400918 if (!ehdr) {
919 dprintf(2, "%s: %s: Not a valid dynamic program\n", ldname, argv[0]);
920 _exit(1);
921 }
922 runtime = 0;
923 close(fd);
Rich Felker649cec52012-07-13 01:31:02 -0400924 lib->name = ldname;
Rich Felker0420b872012-07-11 01:41:20 -0400925 app->name = argv[0];
Rich Felker18c0e022012-10-31 21:27:48 -0400926 app->phnum = ehdr->e_phnum;
927 app->phdr = (void *)(app->base + ehdr->e_phoff);
Rich Felker5c1909a2012-05-27 16:01:44 -0400928 aux[AT_ENTRY] = ehdr->e_entry;
Rich Felkere12fe652012-01-23 02:02:59 -0500929 }
Rich Felkerbc6a35f2012-10-04 20:04:13 -0400930 if (app->tls_size) {
Rich Felkercf3fd3d2012-10-06 01:22:51 -0400931 app->tls_id = tls_cnt = 1;
Rich Felker9ec42832012-10-15 18:51:53 -0400932#ifdef TLS_ABOVE_TP
933 app->tls_offset = 0;
934 tls_offset = app->tls_size
935 + ( -((uintptr_t)app->tls_image + app->tls_size)
936 & (app->tls_align-1) );
937#else
Rich Felkerc62b9f32012-10-14 19:56:50 -0400938 tls_offset = app->tls_offset = app->tls_size
939 + ( -((uintptr_t)app->tls_image + app->tls_size)
940 & (app->tls_align-1) );
Rich Felker9ec42832012-10-15 18:51:53 -0400941#endif
Rich Felkercf3fd3d2012-10-06 01:22:51 -0400942 tls_align = MAXP2(tls_align, app->tls_align);
Rich Felkerbc6a35f2012-10-04 20:04:13 -0400943 }
Rich Felkerc82f4a32012-01-23 00:57:38 -0500944 app->global = 1;
Rich Felker4ce3cb52012-02-06 14:39:09 -0500945 app->constructed = 1;
Rich Felkerc82f4a32012-01-23 00:57:38 -0500946 decode_dyn(app);
947
948 /* Attach to vdso, if provided by the kernel */
Rich Felkerdbcb3ad2012-08-25 17:31:59 -0400949 if (search_vec(auxv, &vdso_base, AT_SYSINFO_EHDR)) {
Rich Felker6ab444d2011-07-24 00:54:55 -0400950 ehdr = (void *)vdso_base;
Rich Felker18c0e022012-10-31 21:27:48 -0400951 vdso->phdr = phdr = (void *)(vdso_base + ehdr->e_phoff);
952 vdso->phnum = ehdr->e_phnum;
Rich Felker6ab444d2011-07-24 00:54:55 -0400953 for (i=ehdr->e_phnum; i; i--, phdr=(void *)((char *)phdr + ehdr->e_phentsize)) {
954 if (phdr->p_type == PT_DYNAMIC)
955 vdso->dynv = (void *)(vdso_base + phdr->p_offset);
956 if (phdr->p_type == PT_LOAD)
957 vdso->base = (void *)(vdso_base - phdr->p_vaddr + phdr->p_offset);
958 }
Rich Felker75a31fa2012-11-25 20:56:31 -0500959 vdso->name = "";
960 vdso->shortname = "linux-gate.so.1";
Rich Felker427173b2011-07-24 02:19:47 -0400961 vdso->global = 1;
Rich Felkerc82f4a32012-01-23 00:57:38 -0500962 decode_dyn(vdso);
Rich Felker6ab444d2011-07-24 00:54:55 -0400963 vdso->prev = lib;
964 lib->next = vdso;
965 }
966
Rich Felkerc82f4a32012-01-23 00:57:38 -0500967 /* Initial dso chain consists only of the app. We temporarily
968 * append the dynamic linker/libc so we can relocate it, then
969 * restore the initial chain in preparation for loading third
970 * party libraries (preload/needed). */
971 head = tail = app;
Rich Felkere23d3582012-10-13 23:25:20 -0400972 ldso = lib;
Rich Felkerc82f4a32012-01-23 00:57:38 -0500973 app->next = lib;
974 reloc_all(lib);
975 app->next = 0;
Rich Felker51e2d832011-06-18 19:48:42 -0400976
Rich Felkerc82f4a32012-01-23 00:57:38 -0500977 /* PAST THIS POINT, ALL LIBC INTERFACES ARE FULLY USABLE. */
Rich Felker51e2d832011-06-18 19:48:42 -0400978
Rich Felkerc82f4a32012-01-23 00:57:38 -0500979 /* Donate unused parts of app and library mapping to malloc */
Rich Felkera53de812011-07-24 00:26:12 -0400980 reclaim_gaps(app->base, (void *)aux[AT_PHDR], aux[AT_PHENT], aux[AT_PHNUM]);
981 ehdr = (void *)lib->base;
982 reclaim_gaps(lib->base, (void *)(lib->base+ehdr->e_phoff),
Rich Felker6717e622011-06-28 19:40:14 -0400983 ehdr->e_phentsize, ehdr->e_phnum);
984
Rich Felkerc82f4a32012-01-23 00:57:38 -0500985 /* Load preload/needed libraries, add their symbols to the global
Rich Felkerfd7015d2012-01-23 18:32:40 -0500986 * namespace, and perform all remaining relocations. The main
987 * program must be relocated LAST since it may contain copy
988 * relocations which depend on libraries' relocations. */
Rich Felker2719cc82011-08-16 00:24:36 -0400989 if (env_preload) load_preload(env_preload);
Rich Felkerc82f4a32012-01-23 00:57:38 -0500990 load_deps(app);
991 make_global(app);
Rich Felker9c748562012-10-04 22:48:33 -0400992
Rich Felkercf3fd3d2012-10-06 01:22:51 -0400993 reloc_all(app->next);
994 reloc_all(app);
995
996 update_tls_size();
Rich Felker9c748562012-10-04 22:48:33 -0400997 if (tls_cnt) {
Rich Felkere23d3582012-10-13 23:25:20 -0400998 void *mem = mmap(0, libc.tls_size, PROT_READ|PROT_WRITE,
Rich Felker9c748562012-10-04 22:48:33 -0400999 MAP_ANONYMOUS|MAP_PRIVATE, -1, 0);
1000 if (mem==MAP_FAILED ||
Rich Felkerdcd60372012-10-05 11:51:50 -04001001 !__install_initial_tls(__copy_tls(mem))) {
Rich Felker9c748562012-10-04 22:48:33 -04001002 dprintf(2, "%s: Error getting %zu bytes thread-local storage: %m\n",
Rich Felkere23d3582012-10-13 23:25:20 -04001003 argv[0], libc.tls_size);
Rich Felker9c748562012-10-04 22:48:33 -04001004 _exit(127);
1005 }
Rich Felker9c748562012-10-04 22:48:33 -04001006 }
1007
Rich Felker04109502012-08-18 16:00:23 -04001008 if (ldso_fail) _exit(127);
Rich Felker5c1909a2012-05-27 16:01:44 -04001009 if (ldd_mode) _exit(0);
1010
Rich Felkerc82f4a32012-01-23 00:57:38 -05001011 /* Switch to runtime mode: any further failures in the dynamic
1012 * linker are a reportable failure rather than a fatal startup
1013 * error. If the dynamic loader (dlopen) will not be used, free
1014 * all memory used by the dynamic linker. */
Rich Felkera53de812011-07-24 00:26:12 -04001015 runtime = 1;
Rich Felker4ce3cb52012-02-06 14:39:09 -05001016
Rich Felker7d9a5c62012-08-05 14:03:17 -04001017#ifndef DYNAMIC_IS_RO
Rich Felker3ec8d292012-04-25 00:05:42 -04001018 for (i=0; app->dynv[i]; i+=2)
1019 if (app->dynv[i]==DT_DEBUG)
1020 app->dynv[i+1] = (size_t)&debug;
Rich Felker7d9a5c62012-08-05 14:03:17 -04001021#endif
Rich Felker3ec8d292012-04-25 00:05:42 -04001022 debug.ver = 1;
1023 debug.bp = _dl_debug_state;
1024 debug.head = head;
1025 debug.base = lib->base;
1026 debug.state = 0;
1027 _dl_debug_state();
1028
Rich Felker0a96a372012-10-07 21:43:46 -04001029 if (ssp_used) __init_ssp((void *)aux[AT_RANDOM]);
Rich Felker58aa5f42012-05-03 20:42:45 -04001030
Rich Felker51e2d832011-06-18 19:48:42 -04001031 errno = 0;
1032 return (void *)aux[AT_ENTRY];
1033}
Rich Felker59ab43f2011-06-26 19:23:28 -04001034
Rich Felkera7936f62012-11-30 17:56:23 -05001035void __init_ldso_ctors(void)
1036{
1037 atexit(do_fini);
1038 do_init_fini(tail);
1039}
1040
Rich Felker59ab43f2011-06-26 19:23:28 -04001041void *dlopen(const char *file, int mode)
1042{
Rich Felker642b7592012-10-05 01:15:25 -04001043 struct dso *volatile p, *orig_tail, *next;
Rich Felkercf3fd3d2012-10-06 01:22:51 -04001044 size_t orig_tls_cnt, orig_tls_offset, orig_tls_align;
Rich Felker59ab43f2011-06-26 19:23:28 -04001045 size_t i;
Rich Felkerf2baf4d2012-02-07 20:31:27 -05001046 int cs;
Rich Felker59ab43f2011-06-26 19:23:28 -04001047
1048 if (!file) return head;
1049
Rich Felkerf2baf4d2012-02-07 20:31:27 -05001050 pthread_setcancelstate(PTHREAD_CANCEL_DISABLE, &cs);
Rich Felker59ab43f2011-06-26 19:23:28 -04001051 pthread_rwlock_wrlock(&lock);
Rich Felkerdcd60372012-10-05 11:51:50 -04001052 __inhibit_ptc();
Rich Felker59ab43f2011-06-26 19:23:28 -04001053
Rich Felkerdcd60372012-10-05 11:51:50 -04001054 p = 0;
1055 orig_tls_cnt = tls_cnt;
Rich Felkercf3fd3d2012-10-06 01:22:51 -04001056 orig_tls_offset = tls_offset;
1057 orig_tls_align = tls_align;
Rich Felker642b7592012-10-05 01:15:25 -04001058 orig_tail = tail;
Rich Felker4d07e552013-01-23 22:07:45 -05001059 noload = mode & RTLD_NOLOAD;
Rich Felker642b7592012-10-05 01:15:25 -04001060
Rich Felker59ab43f2011-06-26 19:23:28 -04001061 if (setjmp(rtld_fail)) {
1062 /* Clean up anything new that was (partially) loaded */
Rich Felkerdcd60372012-10-05 11:51:50 -04001063 if (p && p->deps) for (i=0; p->deps[i]; i++)
Rich Felker92ab5d82011-06-26 21:21:04 -04001064 if (p->deps[i]->global < 0)
1065 p->deps[i]->global = 0;
Rich Felker59ab43f2011-06-26 19:23:28 -04001066 for (p=orig_tail->next; p; p=next) {
1067 next = p->next;
1068 munmap(p->map, p->map_len);
1069 free(p->deps);
1070 free(p);
1071 }
Rich Felkerdcd60372012-10-05 11:51:50 -04001072 tls_cnt = orig_tls_cnt;
Rich Felkercf3fd3d2012-10-06 01:22:51 -04001073 tls_offset = orig_tls_offset;
1074 tls_align = orig_tls_align;
Rich Felker59ab43f2011-06-26 19:23:28 -04001075 tail = orig_tail;
1076 tail->next = 0;
Rich Felkerf2baf4d2012-02-07 20:31:27 -05001077 p = 0;
Rich Felkera5d10eb2012-04-23 12:03:31 -04001078 errflag = 1;
1079 goto end;
Rich Felkera9e85c02012-03-23 00:28:20 -04001080 } else p = load_library(file);
1081
1082 if (!p) {
Rich Felker4d07e552013-01-23 22:07:45 -05001083 snprintf(errbuf, sizeof errbuf, noload ?
1084 "Library %s is not already loaded" :
1085 "Error loading shared library %s: %m",
1086 file);
Rich Felkera9e85c02012-03-23 00:28:20 -04001087 errflag = 1;
Rich Felkerf2baf4d2012-02-07 20:31:27 -05001088 goto end;
Rich Felker59ab43f2011-06-26 19:23:28 -04001089 }
1090
Rich Felker59ab43f2011-06-26 19:23:28 -04001091 /* First load handling */
1092 if (!p->deps) {
1093 load_deps(p);
Rich Felker0e4dae32011-06-26 21:36:44 -04001094 if (p->deps) for (i=0; p->deps[i]; i++)
Rich Felker92ab5d82011-06-26 21:21:04 -04001095 if (!p->deps[i]->global)
1096 p->deps[i]->global = -1;
1097 if (!p->global) p->global = -1;
Rich Felker59ab43f2011-06-26 19:23:28 -04001098 reloc_all(p);
Rich Felker0e4dae32011-06-26 21:36:44 -04001099 if (p->deps) for (i=0; p->deps[i]; i++)
Rich Felker92ab5d82011-06-26 21:21:04 -04001100 if (p->deps[i]->global < 0)
1101 p->deps[i]->global = 0;
1102 if (p->global < 0) p->global = 0;
Rich Felker59ab43f2011-06-26 19:23:28 -04001103 }
1104
1105 if (mode & RTLD_GLOBAL) {
Rich Felker0e4dae32011-06-26 21:36:44 -04001106 if (p->deps) for (i=0; p->deps[i]; i++)
Rich Felker59ab43f2011-06-26 19:23:28 -04001107 p->deps[i]->global = 1;
1108 p->global = 1;
1109 }
1110
Rich Felkercf3fd3d2012-10-06 01:22:51 -04001111 update_tls_size();
Rich Felkerdcd60372012-10-05 11:51:50 -04001112
Rich Felker2f2f1152012-11-01 23:49:57 -04001113 if (ssp_used) __init_ssp(libc.auxv);
Rich Felker731e8ff2012-08-25 17:24:46 -04001114
Rich Felker3ec8d292012-04-25 00:05:42 -04001115 _dl_debug_state();
Rich Felkerf4f77c02012-10-05 13:09:09 -04001116 orig_tail = tail;
Rich Felker06933cc2011-06-26 22:09:32 -04001117end:
Rich Felkerdcd60372012-10-05 11:51:50 -04001118 __release_ptc();
Rich Felker18c0e022012-10-31 21:27:48 -04001119 if (p) gencnt++;
Rich Felker59ab43f2011-06-26 19:23:28 -04001120 pthread_rwlock_unlock(&lock);
Rich Felkerf4f77c02012-10-05 13:09:09 -04001121 if (p) do_init_fini(orig_tail);
Rich Felkerf2baf4d2012-02-07 20:31:27 -05001122 pthread_setcancelstate(cs, 0);
Rich Felker59ab43f2011-06-26 19:23:28 -04001123 return p;
1124}
1125
Rich Felker4d982802013-01-16 11:49:00 -05001126static int invalid_dso_handle(void *h)
Rich Felker6468fc92013-01-10 14:05:40 -05001127{
1128 struct dso *p;
1129 for (p=head; p; p=p->next) if (h==p) return 0;
1130 snprintf(errbuf, sizeof errbuf, "Invalid library handle %p", (void *)h);
1131 errflag = 1;
1132 return 1;
1133}
1134
Rich Felker623753a2011-08-16 00:42:13 -04001135static void *do_dlsym(struct dso *p, const char *s, void *ra)
Rich Felker59ab43f2011-06-26 19:23:28 -04001136{
1137 size_t i;
Rich Felker2bd05a42012-08-25 17:13:28 -04001138 uint32_t h = 0, gh = 0;
Rich Felker59ab43f2011-06-26 19:23:28 -04001139 Sym *sym;
Rich Felker9c748562012-10-04 22:48:33 -04001140 if (p == head || p == RTLD_DEFAULT || p == RTLD_NEXT) {
Rich Felkerdeb15b32012-10-19 21:41:30 -04001141 if (p == RTLD_DEFAULT) {
1142 p = head;
1143 } else if (p == RTLD_NEXT) {
Rich Felker9c748562012-10-04 22:48:33 -04001144 for (p=head; p && (unsigned char *)ra-p->map>p->map_len; p=p->next);
1145 if (!p) p=head;
Rich Felkerdeb15b32012-10-19 21:41:30 -04001146 p = p->next;
Rich Felker9c748562012-10-04 22:48:33 -04001147 }
Rich Felkerdeb15b32012-10-19 21:41:30 -04001148 struct symdef def = find_sym(p, s, 0);
Rich Felker9c748562012-10-04 22:48:33 -04001149 if (!def.sym) goto failed;
Rich Felker0a1c2c12012-10-19 21:57:56 -04001150 if ((def.sym->st_info&0xf) == STT_TLS)
1151 return __tls_get_addr((size_t []){def.dso->tls_id, def.sym->st_value});
Rich Felker9c748562012-10-04 22:48:33 -04001152 return def.dso->base + def.sym->st_value;
Rich Felkera9e85c02012-03-23 00:28:20 -04001153 }
Rich Felker637dd2d2013-01-23 20:21:36 -05001154 if (p != RTLD_DEFAULT && p != RTLD_NEXT && invalid_dso_handle(p))
1155 return 0;
Rich Felker2bd05a42012-08-25 17:13:28 -04001156 if (p->ghashtab) {
1157 gh = gnu_hash(s);
1158 sym = gnu_lookup(s, gh, p);
1159 } else {
1160 h = sysv_hash(s);
1161 sym = sysv_lookup(s, h, p);
1162 }
Rich Felker0a1c2c12012-10-19 21:57:56 -04001163 if (sym && (sym->st_info&0xf) == STT_TLS)
1164 return __tls_get_addr((size_t []){p->tls_id, sym->st_value});
Rich Felker59ab43f2011-06-26 19:23:28 -04001165 if (sym && sym->st_value && (1<<(sym->st_info&0xf) & OK_TYPES))
1166 return p->base + sym->st_value;
1167 if (p->deps) for (i=0; p->deps[i]; i++) {
Rich Felker2bd05a42012-08-25 17:13:28 -04001168 if (p->deps[i]->ghashtab) {
1169 if (!gh) gh = gnu_hash(s);
Rich Felkera5d61992012-08-25 17:40:27 -04001170 sym = gnu_lookup(s, gh, p->deps[i]);
Rich Felker2bd05a42012-08-25 17:13:28 -04001171 } else {
1172 if (!h) h = sysv_hash(s);
1173 sym = sysv_lookup(s, h, p->deps[i]);
1174 }
Rich Felker0a1c2c12012-10-19 21:57:56 -04001175 if (sym && (sym->st_info&0xf) == STT_TLS)
1176 return __tls_get_addr((size_t []){p->deps[i]->tls_id, sym->st_value});
Rich Felker59ab43f2011-06-26 19:23:28 -04001177 if (sym && sym->st_value && (1<<(sym->st_info&0xf) & OK_TYPES))
1178 return p->deps[i]->base + sym->st_value;
1179 }
Rich Felker4027f4e2012-05-04 20:18:18 -04001180failed:
Rich Felkera9e85c02012-03-23 00:28:20 -04001181 errflag = 1;
Rich Felkera5d10eb2012-04-23 12:03:31 -04001182 snprintf(errbuf, sizeof errbuf, "Symbol not found: %s", s);
Rich Felker59ab43f2011-06-26 19:23:28 -04001183 return 0;
1184}
1185
Rich Felkerf419bcb2012-08-26 21:09:26 -04001186int __dladdr(void *addr, Dl_info *info)
1187{
1188 struct dso *p;
1189 Sym *sym;
1190 uint32_t nsym;
1191 char *strings;
1192 size_t i;
1193 void *best = 0;
1194 char *bestname;
1195
1196 pthread_rwlock_rdlock(&lock);
1197 for (p=head; p && (unsigned char *)addr-p->map>p->map_len; p=p->next);
1198 pthread_rwlock_unlock(&lock);
1199
1200 if (!p) return 0;
1201
1202 sym = p->syms;
1203 strings = p->strings;
1204 if (p->hashtab) {
1205 nsym = p->hashtab[1];
1206 } else {
1207 uint32_t *buckets;
1208 uint32_t *hashval;
1209 buckets = p->ghashtab + 4 + (p->ghashtab[2]*sizeof(size_t)/4);
1210 sym += p->ghashtab[1];
1211 for (i = 0; i < p->ghashtab[0]; i++) {
1212 if (buckets[i] > nsym)
1213 nsym = buckets[i];
1214 }
1215 if (nsym) {
1216 nsym -= p->ghashtab[1];
1217 hashval = buckets + p->ghashtab[0] + nsym;
1218 do nsym++;
1219 while (!(*hashval++ & 1));
1220 }
1221 }
1222
1223 for (; nsym; nsym--, sym++) {
Rich Felkercdc5c742013-01-16 11:47:35 -05001224 if (sym->st_value
Rich Felkerf419bcb2012-08-26 21:09:26 -04001225 && (1<<(sym->st_info&0xf) & OK_TYPES)
1226 && (1<<(sym->st_info>>4) & OK_BINDS)) {
1227 void *symaddr = p->base + sym->st_value;
1228 if (symaddr > addr || symaddr < best)
1229 continue;
1230 best = symaddr;
1231 bestname = strings + sym->st_name;
1232 if (addr == symaddr)
1233 break;
1234 }
1235 }
1236
1237 if (!best) return 0;
1238
1239 info->dli_fname = p->name;
1240 info->dli_fbase = p->base;
1241 info->dli_sname = bestname;
1242 info->dli_saddr = best;
1243
1244 return 1;
1245}
1246
Rich Felker400c5e52012-09-06 22:44:55 -04001247void *__dlsym(void *restrict p, const char *restrict s, void *restrict ra)
Rich Felker59ab43f2011-06-26 19:23:28 -04001248{
1249 void *res;
1250 pthread_rwlock_rdlock(&lock);
Rich Felker623753a2011-08-16 00:42:13 -04001251 res = do_dlsym(p, s, ra);
Rich Felker59ab43f2011-06-26 19:23:28 -04001252 pthread_rwlock_unlock(&lock);
1253 return res;
1254}
Rich Felker18c0e022012-10-31 21:27:48 -04001255
1256int dl_iterate_phdr(int(*callback)(struct dl_phdr_info *info, size_t size, void *data), void *data)
1257{
1258 struct dso *current;
1259 struct dl_phdr_info info;
1260 int ret = 0;
1261 for(current = head; current;) {
1262 info.dlpi_addr = (uintptr_t)current->base;
1263 info.dlpi_name = current->name;
1264 info.dlpi_phdr = current->phdr;
1265 info.dlpi_phnum = current->phnum;
1266 info.dlpi_adds = gencnt;
1267 info.dlpi_subs = 0;
1268 info.dlpi_tls_modid = current->tls_id;
1269 info.dlpi_tls_data = current->tls_image;
1270
1271 ret = (callback)(&info, sizeof (info), data);
1272
1273 if (ret != 0) break;
1274
1275 pthread_rwlock_rdlock(&lock);
1276 current = current->next;
1277 pthread_rwlock_unlock(&lock);
1278 }
1279 return ret;
1280}
Rich Felker5a09a532012-02-03 03:16:07 -05001281#else
Rich Felker4d982802013-01-16 11:49:00 -05001282static int invalid_dso_handle(void *h)
Rich Felker6468fc92013-01-10 14:05:40 -05001283{
1284 snprintf(errbuf, sizeof errbuf, "Invalid library handle %p", (void *)h);
1285 errflag = 1;
1286 return 1;
1287}
Rich Felker5a09a532012-02-03 03:16:07 -05001288void *dlopen(const char *file, int mode)
1289{
1290 return 0;
1291}
Rich Felker400c5e52012-09-06 22:44:55 -04001292void *__dlsym(void *restrict p, const char *restrict s, void *restrict ra)
Rich Felker5a09a532012-02-03 03:16:07 -05001293{
1294 return 0;
1295}
Rich Felkerf419bcb2012-08-26 21:09:26 -04001296int __dladdr (void *addr, Dl_info *info)
1297{
1298 return 0;
1299}
Rich Felker5a09a532012-02-03 03:16:07 -05001300#endif
Rich Felker59ab43f2011-06-26 19:23:28 -04001301
Rich Felker780cbbe2013-06-29 12:46:46 -04001302int __dlinfo(void *dso, int req, void *res)
1303{
1304 if (invalid_dso_handle(dso)) return -1;
1305 if (req != RTLD_DI_LINKMAP) {
1306 snprintf(errbuf, sizeof errbuf, "Unsupported request %d", req);
1307 errflag = 1;
1308 return -1;
1309 }
1310 *(struct link_map **)res = dso;
1311 return 0;
1312}
1313
Rich Felker59ab43f2011-06-26 19:23:28 -04001314char *dlerror()
1315{
Rich Felkera9e85c02012-03-23 00:28:20 -04001316 if (!errflag) return 0;
1317 errflag = 0;
Rich Felkera5d10eb2012-04-23 12:03:31 -04001318 return errbuf;
Rich Felker59ab43f2011-06-26 19:23:28 -04001319}
1320
1321int dlclose(void *p)
1322{
Rich Felker6468fc92013-01-10 14:05:40 -05001323 return invalid_dso_handle(p);
Rich Felker59ab43f2011-06-26 19:23:28 -04001324}