dynlink.c 32.3 KB
Newer Older
1
#define _GNU_SOURCE
R
Rich Felker 已提交
2 3 4 5 6 7 8 9 10 11 12 13 14 15
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <stdint.h>
#include <elf.h>
#include <sys/mman.h>
#include <limits.h>
#include <stdint.h>
#include <fcntl.h>
#include <sys/stat.h>
#include <errno.h>
#include <limits.h>
#include <elf.h>
R
Rich Felker 已提交
16
#include <link.h>
R
Rich Felker 已提交
17
#include <setjmp.h>
18
#include <pthread.h>
R
Rich Felker 已提交
19
#include <ctype.h>
20
#include <dlfcn.h>
21 22
#include "pthread_impl.h"
#include "libc.h"
R
Rich Felker 已提交
23

R
Rich Felker 已提交
24
static int errflag;
25
static char errbuf[128];
R
Rich Felker 已提交
26

27
#ifdef SHARED
R
Rich Felker 已提交
28

R
Rich Felker 已提交
29 30 31 32 33 34 35 36 37 38 39 40 41 42
#if ULONG_MAX == 0xffffffff
typedef Elf32_Ehdr Ehdr;
typedef Elf32_Phdr Phdr;
typedef Elf32_Sym Sym;
#define R_TYPE(x) ((x)&255)
#define R_SYM(x) ((x)>>8)
#else
typedef Elf64_Ehdr Ehdr;
typedef Elf64_Phdr Phdr;
typedef Elf64_Sym Sym;
#define R_TYPE(x) ((x)&0xffffffff)
#define R_SYM(x) ((x)>>32)
#endif

R
Rich Felker 已提交
43 44 45
#define MAXP2(a,b) (-(-(a)&-(b)))
#define ALIGN(x,y) ((x)+(y)-1 & -(y))

46 47 48 49 50 51 52 53 54 55 56 57
struct debug {
	int ver;
	void *head;
	void (*bp)(void);
	int state;
	void *base;
};

struct dso {
	unsigned char *base;
	char *name;
	size_t *dynv;
R
Rich Felker 已提交
58
	struct dso *next, *prev;
59

R
Rich Felker 已提交
60 61
	Phdr *phdr;
	int phnum;
R
Rich Felker 已提交
62 63
	int refcnt;
	Sym *syms;
64
	uint32_t *hashtab;
65
	uint32_t *ghashtab;
R
Rich Felker 已提交
66 67 68 69 70
	char *strings;
	unsigned char *map;
	size_t map_len;
	dev_t dev;
	ino_t ino;
71
	signed char global;
72 73
	char relocated;
	char constructed;
74
	struct dso **deps;
75
	void *tls_image;
76
	size_t tls_len, tls_size, tls_align, tls_id, tls_offset;
77 78 79
	void **new_dtv;
	unsigned char *new_tls;
	int new_dtv_idx, new_tls_idx;
80
	struct dso *fini_next;
81
	char *shortname;
R
Rich Felker 已提交
82
	char buf[];
R
Rich Felker 已提交
83 84
};

85 86 87 88 89
struct symdef {
	Sym *sym;
	struct dso *dso;
};

90 91
#include "reloc.h"

92
void __init_ssp(size_t *);
93
void *__install_initial_tls(void *);
94

95
static struct dso *head, *tail, *ldso, *fini_head;
96
static char *env_path, *sys_path, *r_path;
R
Rich Felker 已提交
97
static unsigned long long gencnt;
98
static int ssp_used;
R
Rich Felker 已提交
99
static int runtime;
100
static int ldd_mode;
101
static int ldso_fail;
R
Rich Felker 已提交
102
static jmp_buf rtld_fail;
103
static pthread_rwlock_t lock;
104
static struct debug debug;
105
static size_t tls_cnt, tls_offset, tls_align = 4*sizeof(size_t);
106
static pthread_mutex_t init_fini_lock = { ._m_type = PTHREAD_MUTEX_RECURSIVE };
107 108

struct debug *_dl_debug_addr = &debug;
R
Rich Felker 已提交
109

110
#define AUX_CNT 38
R
Rich Felker 已提交
111 112 113 114 115 116 117 118 119 120 121
#define DYN_CNT 34

static void decode_vec(size_t *v, size_t *a, size_t cnt)
{
	memset(a, 0, cnt*sizeof(size_t));
	for (; v[0]; v+=2) if (v[0]<cnt) {
		a[0] |= 1ULL<<v[0];
		a[v[0]] = v[1];
	}
}

122 123 124 125 126 127 128 129 130
static int search_vec(size_t *v, size_t *r, size_t key)
{
	for (; v[0]!=key; v+=2)
		if (!v[0]) return 0;
	*r = v[1];
	return 1;
}

static uint32_t sysv_hash(const char *s0)
R
Rich Felker 已提交
131
{
132
	const unsigned char *s = (void *)s0;
R
Rich Felker 已提交
133 134 135 136 137 138 139 140
	uint_fast32_t h = 0;
	while (*s) {
		h = 16*h + *s++;
		h ^= h>>24 & 0xf0;
	}
	return h & 0xfffffff;
}

141 142 143 144 145 146 147 148 149 150
static uint32_t gnu_hash(const char *s0)
{
	const unsigned char *s = (void *)s0;
	uint_fast32_t h = 5381;
	for (; *s; s++)
		h = h*33 + *s;
	return h;
}

static Sym *sysv_lookup(const char *s, uint32_t h, struct dso *dso)
R
Rich Felker 已提交
151 152
{
	size_t i;
R
Rich Felker 已提交
153 154 155
	Sym *syms = dso->syms;
	uint32_t *hashtab = dso->hashtab;
	char *strings = dso->strings;
R
Rich Felker 已提交
156 157 158 159 160 161 162
	for (i=hashtab[2+h%hashtab[0]]; i; i=hashtab[2+hashtab[0]+i]) {
		if (!strcmp(s, strings+syms[i].st_name))
			return syms+i;
	}
	return 0;
}

163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189
static Sym *gnu_lookup(const char *s, uint32_t h1, struct dso *dso)
{
	Sym *sym;
	char *strings;
	uint32_t *hashtab = dso->ghashtab;
	uint32_t nbuckets = hashtab[0];
	uint32_t *buckets = hashtab + 4 + hashtab[2]*(sizeof(size_t)/4);
	uint32_t h2;
	uint32_t *hashval;
	uint32_t n = buckets[h1 % nbuckets];

	if (!n) return 0;

	strings = dso->strings;
	sym = dso->syms + n;
	hashval = buckets + nbuckets + (n - hashtab[1]);

	for (h1 |= 1; ; sym++) {
		h2 = *hashval++;
		if ((h1 == (h2|1)) && !strcmp(s, strings + sym->st_name))
			return sym;
		if (h2 & 1) break;
	}

	return 0;
}

190
#define OK_TYPES (1<<STT_NOTYPE | 1<<STT_OBJECT | 1<<STT_FUNC | 1<<STT_COMMON | 1<<STT_TLS)
191
#define OK_BINDS (1<<STB_GLOBAL | 1<<STB_WEAK)
R
Rich Felker 已提交
192

193
static struct symdef find_sym(struct dso *dso, const char *s, int need_def)
R
Rich Felker 已提交
194
{
195
	uint32_t h = 0, gh = 0;
196
	struct symdef def = {0};
197 198 199 200 201 202 203
	if (dso->ghashtab) {
		gh = gnu_hash(s);
		if (gh == 0x1f4039c9 && !strcmp(s, "__stack_chk_fail")) ssp_used = 1;
	} else {
		h = sysv_hash(s);
		if (h == 0x595a4cc && !strcmp(s, "__stack_chk_fail")) ssp_used = 1;
	}
R
Rich Felker 已提交
204
	for (; dso; dso=dso->next) {
205 206
		Sym *sym;
		if (!dso->global) continue;
207 208 209 210 211 212 213
		if (dso->ghashtab) {
			if (!gh) gh = gnu_hash(s);
			sym = gnu_lookup(s, gh, dso);
		} else {
			if (!h) h = sysv_hash(s);
			sym = sysv_lookup(s, h, dso);
		}
214 215 216 217 218 219 220 221 222 223 224 225 226 227
		if (!sym) continue;
		if (!sym->st_shndx)
			if (need_def || (sym->st_info&0xf) == STT_TLS)
				continue;
		if (!sym->st_value)
			if ((sym->st_info&0xf) != STT_TLS)
				continue;
		if (!(1<<(sym->st_info&0xf) & OK_TYPES)) continue;
		if (!(1<<(sym->st_info>>4) & OK_BINDS)) continue;

		if (def.sym && sym->st_info>>4 == STB_WEAK) continue;
		def.sym = sym;
		def.dso = dso;
		if (sym->st_info>>4 == STB_GLOBAL) break;
R
Rich Felker 已提交
228
	}
229
	return def;
R
Rich Felker 已提交
230 231
}

232
static void do_relocs(struct dso *dso, size_t *rel, size_t rel_size, size_t stride)
R
Rich Felker 已提交
233
{
234 235 236
	unsigned char *base = dso->base;
	Sym *syms = dso->syms;
	char *strings = dso->strings;
R
Rich Felker 已提交
237 238 239 240 241
	Sym *sym;
	const char *name;
	void *ctx;
	int type;
	int sym_index;
242
	struct symdef def;
R
Rich Felker 已提交
243 244 245 246 247 248 249

	for (; rel_size; rel+=stride, rel_size-=stride*sizeof(size_t)) {
		type = R_TYPE(rel[1]);
		sym_index = R_SYM(rel[1]);
		if (sym_index) {
			sym = syms + sym_index;
			name = strings + sym->st_name;
250
			ctx = IS_COPY(type) ? head->next : head;
251 252
			def = find_sym(ctx, name, IS_PLT(type));
			if (!def.sym && sym->st_info>>4 != STB_WEAK) {
253 254
				snprintf(errbuf, sizeof errbuf,
					"Error relocating %s: %s: symbol not found",
255
					dso->name, name);
R
Rich Felker 已提交
256
				if (runtime) longjmp(rtld_fail, 1);
257
				dprintf(2, "%s\n", errbuf);
258 259
				ldso_fail = 1;
				continue;
R
Rich Felker 已提交
260
			}
261
		} else {
262 263 264
			sym = 0;
			def.sym = 0;
			def.dso = 0;
R
Rich Felker 已提交
265
		}
266 267 268
		do_single_reloc(dso, base, (void *)(base + rel[0]), type,
			stride>2 ? rel[2] : 0, sym, sym?sym->st_size:0, def,
			def.sym?(size_t)(def.dso->base+def.sym->st_value):0);
R
Rich Felker 已提交
269 270 271
	}
}

272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302
/* A huge hack: to make up for the wastefulness of shared libraries
 * needing at least a page of dirty memory even if they have no global
 * data, we reclaim the gaps at the beginning and end of writable maps
 * and "donate" them to the heap by setting up minimal malloc
 * structures and then freeing them. */

static void reclaim(unsigned char *base, size_t start, size_t end)
{
	size_t *a, *z;
	start = start + 6*sizeof(size_t)-1 & -4*sizeof(size_t);
	end = (end & -4*sizeof(size_t)) - 2*sizeof(size_t);
	if (start>end || end-start < 4*sizeof(size_t)) return;
	a = (size_t *)(base + start);
	z = (size_t *)(base + end);
	a[-2] = 1;
	a[-1] = z[0] = end-start + 2*sizeof(size_t) | 1;
	z[1] = 1;
	free(a);
}

static void reclaim_gaps(unsigned char *base, Phdr *ph, size_t phent, size_t phcnt)
{
	for (; phcnt--; ph=(void *)((char *)ph+phent)) {
		if (ph->p_type!=PT_LOAD) continue;
		if ((ph->p_flags&(PF_R|PF_W))!=(PF_R|PF_W)) continue;
		reclaim(base, ph->p_vaddr & -PAGE_SIZE, ph->p_vaddr);
		reclaim(base, ph->p_vaddr+ph->p_memsz,
			ph->p_vaddr+ph->p_memsz+PAGE_SIZE-1 & -PAGE_SIZE);
	}
}

303
static void *map_library(int fd, struct dso *dso)
R
Rich Felker 已提交
304
{
305
	Ehdr buf[(896+sizeof(Ehdr))/sizeof(Ehdr)];
R
Rich Felker 已提交
306 307 308 309 310 311 312 313 314
	size_t phsize;
	size_t addr_min=SIZE_MAX, addr_max=0, map_len;
	size_t this_min, this_max;
	off_t off_start;
	Ehdr *eh;
	Phdr *ph;
	unsigned prot;
	unsigned char *map, *base;
	size_t dyn;
315
	size_t tls_image=0;
R
Rich Felker 已提交
316 317 318 319
	size_t i;

	ssize_t l = read(fd, buf, sizeof buf);
	if (l<sizeof *eh) return 0;
320
	eh = buf;
R
Rich Felker 已提交
321 322 323
	phsize = eh->e_phentsize * eh->e_phnum;
	if (phsize + sizeof *eh > l) return 0;
	if (eh->e_phoff + phsize > l) {
324
		l = pread(fd, buf+1, phsize, eh->e_phoff);
R
Rich Felker 已提交
325 326 327 328
		if (l != phsize) return 0;
		eh->e_phoff = sizeof *eh;
	}
	ph = (void *)((char *)buf + eh->e_phoff);
R
Rich Felker 已提交
329 330
	dso->phdr = ph;
	dso->phnum = eh->e_phnum;
R
Rich Felker 已提交
331 332 333
	for (i=eh->e_phnum; i; i--, ph=(void *)((char *)ph+eh->e_phentsize)) {
		if (ph->p_type == PT_DYNAMIC)
			dyn = ph->p_vaddr;
334 335 336 337 338 339
		if (ph->p_type == PT_TLS) {
			tls_image = ph->p_vaddr;
			dso->tls_align = ph->p_align;
			dso->tls_len = ph->p_filesz;
			dso->tls_size = ph->p_memsz;
		}
R
Rich Felker 已提交
340 341 342 343 344 345 346 347 348 349 350 351 352 353 354 355 356 357 358 359 360 361
		if (ph->p_type != PT_LOAD) continue;
		if (ph->p_vaddr < addr_min) {
			addr_min = ph->p_vaddr;
			off_start = ph->p_offset;
			prot = (((ph->p_flags&PF_R) ? PROT_READ : 0) |
				((ph->p_flags&PF_W) ? PROT_WRITE: 0) |
				((ph->p_flags&PF_X) ? PROT_EXEC : 0));
		}
		if (ph->p_vaddr+ph->p_memsz > addr_max) {
			addr_max = ph->p_vaddr+ph->p_memsz;
		}
	}
	if (!dyn) return 0;
	addr_max += PAGE_SIZE-1;
	addr_max &= -PAGE_SIZE;
	addr_min &= -PAGE_SIZE;
	off_start &= -PAGE_SIZE;
	map_len = addr_max - addr_min + off_start;
	/* The first time, we map too much, possibly even more than
	 * the length of the file. This is okay because we will not
	 * use the invalid part; we just need to reserve the right
	 * amount of virtual address space to map over later. */
362
	map = mmap((void *)addr_min, map_len, prot, MAP_PRIVATE, fd, off_start);
R
Rich Felker 已提交
363 364 365 366 367 368 369 370 371 372 373 374 375
	if (map==MAP_FAILED) return 0;
	base = map - addr_min;
	ph = (void *)((char *)buf + eh->e_phoff);
	for (i=eh->e_phnum; i; i--, ph=(void *)((char *)ph+eh->e_phentsize)) {
		if (ph->p_type != PT_LOAD) continue;
		/* Reuse the existing mapping for the lowest-address LOAD */
		if ((ph->p_vaddr & -PAGE_SIZE) == addr_min) continue;
		this_min = ph->p_vaddr & -PAGE_SIZE;
		this_max = ph->p_vaddr+ph->p_memsz+PAGE_SIZE-1 & -PAGE_SIZE;
		off_start = ph->p_offset & -PAGE_SIZE;
		prot = (((ph->p_flags&PF_R) ? PROT_READ : 0) |
			((ph->p_flags&PF_W) ? PROT_WRITE: 0) |
			((ph->p_flags&PF_X) ? PROT_EXEC : 0));
376 377
		if (mmap(base+this_min, this_max-this_min, prot, MAP_PRIVATE|MAP_FIXED, fd, off_start) == MAP_FAILED)
			goto error;
R
Rich Felker 已提交
378 379 380 381
		if (ph->p_memsz > ph->p_filesz) {
			size_t brk = (size_t)base+ph->p_vaddr+ph->p_filesz;
			size_t pgbrk = brk+PAGE_SIZE-1 & -PAGE_SIZE;
			memset((void *)brk, 0, pgbrk-brk & PAGE_SIZE-1);
382 383
			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)
				goto error;
R
Rich Felker 已提交
384 385
		}
	}
R
Rich Felker 已提交
386 387
	for (i=0; ((size_t *)(base+dyn))[i]; i+=2)
		if (((size_t *)(base+dyn))[i]==DT_TEXTREL) {
388 389
			if (mprotect(map, map_len, PROT_READ|PROT_WRITE|PROT_EXEC) < 0)
				goto error;
R
Rich Felker 已提交
390 391
			break;
		}
392 393
	if (!runtime) reclaim_gaps(base, (void *)((char *)buf + eh->e_phoff),
		eh->e_phentsize, eh->e_phnum);
394 395 396 397 398
	dso->map = map;
	dso->map_len = map_len;
	dso->base = base;
	dso->dynv = (void *)(base+dyn);
	if (dso->tls_size) dso->tls_image = (void *)(base+tls_image);
R
Rich Felker 已提交
399
	return map;
400 401 402
error:
	munmap(map, map_len);
	return 0;
R
Rich Felker 已提交
403 404
}

405
static int path_open(const char *name, const char *search, char *buf, size_t buf_size)
406
{
407
	const char *s=search, *z;
408
	int l, fd;
409 410 411
	for (;;) {
		while (*s==':') s++;
		if (!*s) return -1;
412 413
		z = strchr(s, ':');
		l = z ? z-s : strlen(s);
414
		snprintf(buf, buf_size, "%.*s/%s", l, s, name);
415
		if ((fd = open(buf, O_RDONLY|O_CLOEXEC))>=0) return fd;
416
		s += l;
417 418 419
	}
}

420 421 422 423 424 425
static void decode_dyn(struct dso *p)
{
	size_t dyn[DYN_CNT] = {0};
	decode_vec(p->dynv, dyn, DYN_CNT);
	p->syms = (void *)(p->base + dyn[DT_SYMTAB]);
	p->strings = (void *)(p->base + dyn[DT_STRTAB]);
426 427 428 429
	if (dyn[0]&(1<<DT_HASH))
		p->hashtab = (void *)(p->base + dyn[DT_HASH]);
	if (search_vec(p->dynv, dyn, DT_GNU_HASH))
		p->ghashtab = (void *)(p->base + *dyn);
430 431
}

R
Rich Felker 已提交
432 433
static struct dso *load_library(const char *name)
{
434
	char buf[2*NAME_MAX+2];
435
	const char *pathname;
R
Rich Felker 已提交
436
	unsigned char *base, *map;
R
Rich Felker 已提交
437
	size_t map_len;
438
	struct dso *p, temp_dso = {0};
R
Rich Felker 已提交
439 440
	int fd;
	struct stat st;
441 442
	size_t alloc_size;
	int n_th = 0;
R
Rich Felker 已提交
443 444 445 446 447 448 449 450 451 452

	/* Catch and block attempts to reload the implementation itself */
	if (name[0]=='l' && name[1]=='i' && name[2]=='b') {
		static const char *rp, reserved[] =
			"c\0pthread\0rt\0m\0dl\0util\0xnet\0";
		char *z = strchr(name, '.');
		if (z) {
			size_t l = z-name;
			for (rp=reserved; *rp && memcmp(name+3, rp, l-3); rp+=strlen(rp)+1);
			if (*rp) {
453 454 455 456
				if (!ldso->prev) {
					tail->next = ldso;
					ldso->prev = tail;
					tail = ldso->next ? ldso->next : ldso;
R
Rich Felker 已提交
457
				}
458
				return ldso;
R
Rich Felker 已提交
459 460 461
			}
		}
	}
462
	if (strchr(name, '/')) {
463
		pathname = name;
464
		fd = open(name, O_RDONLY|O_CLOEXEC);
R
Rich Felker 已提交
465
	} else {
466 467 468 469 470 471 472
		/* Search for the name to see if it's already loaded */
		for (p=head->next; p; p=p->next) {
			if (p->shortname && !strcmp(p->shortname, name)) {
				p->refcnt++;
				return p;
			}
		}
473
		if (strlen(name) > NAME_MAX) return 0;
474
		fd = -1;
475 476
		if (r_path) fd = path_open(name, r_path, buf, sizeof buf);
		if (fd < 0 && env_path) fd = path_open(name, env_path, buf, sizeof buf);
477 478
		if (fd < 0) {
			if (!sys_path) {
479
				FILE *f = fopen(ETC_LDSO_PATH, "rbe");
480
				if (f) {
481 482 483
					if (getline(&sys_path, (size_t[1]){0}, f) > 0) {
						size_t l = strlen(sys_path);
						if (l && sys_path[l-1]=='\n')
484
							sys_path[l-1] = 0;
485
					}
486 487 488
					fclose(f);
				}
			}
489 490
			if (!sys_path) sys_path = "/lib:/usr/local/lib:/usr/lib";
			fd = path_open(name, sys_path, buf, sizeof buf);
R
Rich Felker 已提交
491
		}
492
		pathname = buf;
R
Rich Felker 已提交
493 494 495 496 497 498 499 500
	}
	if (fd < 0) return 0;
	if (fstat(fd, &st) < 0) {
		close(fd);
		return 0;
	}
	for (p=head->next; p; p=p->next) {
		if (p->dev == st.st_dev && p->ino == st.st_ino) {
501 502 503
			/* If this library was previously loaded with a
			 * pathname but a search found the same inode,
			 * setup its shortname so it can be found by name. */
504 505
			if (!p->shortname && pathname != name)
				p->shortname = strrchr(p->name, '/')+1;
R
Rich Felker 已提交
506 507 508 509 510
			close(fd);
			p->refcnt++;
			return p;
		}
	}
511
	map = map_library(fd, &temp_dso);
R
Rich Felker 已提交
512 513
	close(fd);
	if (!map) return 0;
514 515 516 517 518 519 520 521 522 523

	/* Allocate storage for the new DSO. When there is TLS, this
	 * storage must include a reservation for all pre-existing
	 * threads to obtain copies of both the new TLS, and an
	 * extended DTV capable of storing an additional slot for
	 * the newly-loaded DSO. */
	alloc_size = sizeof *p + strlen(pathname) + 1;
	if (runtime && temp_dso.tls_image) {
		size_t per_th = temp_dso.tls_size + temp_dso.tls_align
			+ sizeof(void *) * (tls_cnt+3);
524
		n_th = libc.threads_minus_1 + 1;
525 526 527 528
		if (n_th > SSIZE_MAX / per_th) alloc_size = SIZE_MAX;
		else alloc_size += n_th * per_th;
	}
	p = calloc(1, alloc_size);
R
Rich Felker 已提交
529 530 531 532
	if (!p) {
		munmap(map, map_len);
		return 0;
	}
533
	memcpy(p, &temp_dso, sizeof temp_dso);
534
	decode_dyn(p);
R
Rich Felker 已提交
535 536 537
	p->dev = st.st_dev;
	p->ino = st.st_ino;
	p->refcnt = 1;
R
Rich Felker 已提交
538
	p->name = p->buf;
539 540 541
	strcpy(p->name, pathname);
	/* Add a shortname only if name arg was not an explicit pathname. */
	if (pathname != name) p->shortname = strrchr(p->name, '/')+1;
542
	if (p->tls_image) {
543
		if (runtime && !__pthread_self_init()) {
544 545 546 547
			free(p);
			munmap(map, map_len);
			return 0;
		}
548
		p->tls_id = ++tls_cnt;
R
Rich Felker 已提交
549
		tls_align = MAXP2(tls_align, p->tls_align);
550 551 552 553 554
#ifdef TLS_ABOVE_TP
		p->tls_offset = tls_offset + ( (tls_align-1) &
			-(tls_offset + (uintptr_t)p->tls_image) );
		tls_offset += p->tls_size;
#else
R
Rich Felker 已提交
555 556 557 558
		tls_offset += p->tls_size + p->tls_align - 1;
		tls_offset -= (tls_offset + (uintptr_t)p->tls_image)
			& (p->tls_align-1);
		p->tls_offset = tls_offset;
559
#endif
560 561 562 563
		p->new_dtv = (void *)(-sizeof(size_t) &
			(uintptr_t)(p->name+strlen(p->name)+sizeof(size_t)));
		p->new_tls = (void *)(p->new_dtv + n_th*(tls_cnt+1));
	}
R
Rich Felker 已提交
564 565 566 567 568

	tail->next = p;
	p->prev = tail;
	tail = p;

569
	if (ldd_mode) dprintf(1, "\t%s => %s (%p)\n", name, pathname, base);
570

R
Rich Felker 已提交
571 572 573 574 575
	return p;
}

static void load_deps(struct dso *p)
{
576 577
	size_t i, ndeps=0;
	struct dso ***deps = &p->deps, **tmp, *dep;
R
Rich Felker 已提交
578
	for (; p; p=p->next) {
579 580 581 582
		for (i=0; p->dynv[i]; i+=2) {
			if (p->dynv[i] != DT_RPATH) continue;
			r_path = (void *)(p->strings + p->dynv[i+1]);
		}
R
Rich Felker 已提交
583 584
		for (i=0; p->dynv[i]; i+=2) {
			if (p->dynv[i] != DT_NEEDED) continue;
585 586
			dep = load_library(p->strings + p->dynv[i+1]);
			if (!dep) {
587 588
				snprintf(errbuf, sizeof errbuf,
					"Error loading shared library %s: %m (needed by %s)",
R
Rich Felker 已提交
589
					p->strings + p->dynv[i+1], p->name);
590 591
				if (runtime) longjmp(rtld_fail, 1);
				dprintf(2, "%s\n", errbuf);
592 593
				ldso_fail = 1;
				continue;
R
Rich Felker 已提交
594
			}
595 596 597 598 599 600 601
			if (runtime) {
				tmp = realloc(*deps, sizeof(*tmp)*(ndeps+2));
				if (!tmp) longjmp(rtld_fail, 1);
				tmp[ndeps++] = dep;
				tmp[ndeps] = 0;
				*deps = tmp;
			}
R
Rich Felker 已提交
602
		}
603
		r_path = 0;
R
Rich Felker 已提交
604 605 606
	}
}

R
Rich Felker 已提交
607 608 609 610 611 612 613 614 615 616 617 618 619 620
static void load_preload(char *s)
{
	int tmp;
	char *z;
	for (z=s; *z; s=z) {
		for (   ; *s && isspace(*s); s++);
		for (z=s; *z && !isspace(*z); z++);
		tmp = *z;
		*z = 0;
		load_library(s);
		*z = tmp;
	}
}

621 622 623 624 625
static void make_global(struct dso *p)
{
	for (; p; p=p->next) p->global = 1;
}

R
Rich Felker 已提交
626 627 628 629 630 631
static void reloc_all(struct dso *p)
{
	size_t dyn[DYN_CNT] = {0};
	for (; p; p=p->next) {
		if (p->relocated) continue;
		decode_vec(p->dynv, dyn, DYN_CNT);
R
Rich Felker 已提交
632 633 634
#ifdef NEED_ARCH_RELOCS
		do_arch_relocs(p, head);
#endif
635 636 637 638
		do_relocs(p, (void *)(p->base+dyn[DT_JMPREL]), dyn[DT_PLTRELSZ],
			2+(dyn[DT_PLTREL]==DT_RELA));
		do_relocs(p, (void *)(p->base+dyn[DT_REL]), dyn[DT_RELSZ], 2);
		do_relocs(p, (void *)(p->base+dyn[DT_RELA]), dyn[DT_RELASZ], 3);
639
		p->relocated = 1;
R
Rich Felker 已提交
640 641 642
	}
}

643 644 645 646 647 648 649 650
static size_t find_dyn(Phdr *ph, size_t cnt, size_t stride)
{
	for (; cnt--; ph = (void *)((char *)ph + stride))
		if (ph->p_type == PT_DYNAMIC)
			return ph->p_vaddr;
	return 0;
}

651 652 653 654 655 656 657 658 659 660 661 662 663 664 665 666
static void find_map_range(Phdr *ph, size_t cnt, size_t stride, struct dso *p)
{
	size_t min_addr = -1, max_addr = 0;
	for (; cnt--; ph = (void *)((char *)ph + stride)) {
		if (ph->p_type != PT_LOAD) continue;
		if (ph->p_vaddr < min_addr)
			min_addr = ph->p_vaddr;
		if (ph->p_vaddr+ph->p_memsz > max_addr)
			max_addr = ph->p_vaddr+ph->p_memsz;
	}
	min_addr &= -PAGE_SIZE;
	max_addr = (max_addr + PAGE_SIZE-1) & -PAGE_SIZE;
	p->map = p->base + min_addr;
	p->map_len = max_addr - min_addr;
}

667 668 669 670 671 672 673 674 675 676 677
static void do_fini()
{
	struct dso *p;
	size_t dyn[DYN_CNT] = {0};
	for (p=fini_head; p; p=p->fini_next) {
		if (!p->constructed) continue;
		decode_vec(p->dynv, dyn, DYN_CNT);
		((void (*)(void))(p->base + dyn[DT_FINI]))();
	}
}

678 679 680
static void do_init_fini(struct dso *p)
{
	size_t dyn[DYN_CNT] = {0};
681
	int need_locking = libc.threads_minus_1;
682 683 684 685
	/* Allow recursive calls that arise when a library calls
	 * dlopen from one of its constructors, but block any
	 * other threads until all ctors have finished. */
	if (need_locking) pthread_mutex_lock(&init_fini_lock);
686
	for (; p; p=p->prev) {
687 688
		if (p->constructed) continue;
		p->constructed = 1;
689
		decode_vec(p->dynv, dyn, DYN_CNT);
690 691 692 693
		if (dyn[0] & (1<<DT_FINI)) {
			p->fini_next = fini_head;
			fini_head = p;
		}
694 695 696
		if (dyn[0] & (1<<DT_INIT))
			((void (*)(void))(p->base + dyn[DT_INIT]))();
	}
697
	if (need_locking) pthread_mutex_unlock(&init_fini_lock);
698 699
}

700 701 702 703
void _dl_debug_state(void)
{
}

704
void *__copy_tls(unsigned char *mem)
705
{
R
Rich Felker 已提交
706
	pthread_t td;
707
	struct dso *p;
R
Rich Felker 已提交
708 709 710

	if (!tls_cnt) return mem;

711
	void **dtv = (void *)mem;
712
	dtv[0] = (void *)tls_cnt;
R
Rich Felker 已提交
713

714 715 716 717 718 719 720 721 722 723 724 725
#ifdef TLS_ABOVE_TP
	mem += sizeof(void *) * (tls_cnt+1);
	mem += -((uintptr_t)mem + sizeof(struct pthread)) & (tls_align-1);
	td = (pthread_t)mem;
	mem += sizeof(struct pthread);

	for (p=head; p; p=p->next) {
		if (!p->tls_id) continue;
		dtv[p->tls_id] = mem + p->tls_offset;
		memcpy(dtv[p->tls_id], p->tls_image, p->tls_len);
	}
#else
726
	mem += libc.tls_size - sizeof(struct pthread);
R
Rich Felker 已提交
727 728 729 730
	mem -= (uintptr_t)mem & (tls_align-1);
	td = (pthread_t)mem;

	for (p=head; p; p=p->next) {
731
		if (!p->tls_id) continue;
R
Rich Felker 已提交
732 733
		dtv[p->tls_id] = mem - p->tls_offset;
		memcpy(dtv[p->tls_id], p->tls_image, p->tls_len);
734
	}
735
#endif
R
Rich Felker 已提交
736 737
	td->dtv = dtv;
	return td;
738 739
}

740
void *__tls_get_addr(size_t *v)
741 742
{
	pthread_t self = __pthread_self();
743 744 745 746 747
	if (self->dtv && v[0]<=(size_t)self->dtv[0] && self->dtv[v[0]])
		return (char *)self->dtv[v[0]]+v[1];

	/* Block signals to make accessing new TLS async-signal-safe */
	sigset_t set;
748
	pthread_sigmask(SIG_BLOCK, SIGALL_SET, &set);
749 750 751 752 753 754 755 756 757 758 759 760 761 762 763 764 765 766 767 768
	if (self->dtv && v[0]<=(size_t)self->dtv[0] && self->dtv[v[0]]) {
		pthread_sigmask(SIG_SETMASK, &set, 0);
		return (char *)self->dtv[v[0]]+v[1];
	}

	/* This is safe without any locks held because, if the caller
	 * is able to request the Nth entry of the DTV, the DSO list
	 * must be valid at least that far out and it was synchronized
	 * at program startup or by an already-completed call to dlopen. */
	struct dso *p;
	for (p=head; p->tls_id != v[0]; p=p->next);

	/* Get new DTV space from new DSO if needed */
	if (!self->dtv || v[0] > (size_t)self->dtv[0]) {
		void **newdtv = p->new_dtv +
			(v[0]+1)*sizeof(void *)*a_fetch_add(&p->new_dtv_idx,1);
		if (self->dtv) memcpy(newdtv, self->dtv,
			((size_t)self->dtv[0]+1) * sizeof(void *));
		newdtv[0] = (void *)v[0];
		self->dtv = newdtv;
769
	}
770 771 772 773 774 775

	/* Get new TLS memory from new DSO */
	unsigned char *mem = p->new_tls +
		(p->tls_size + p->tls_align) * a_fetch_add(&p->new_tls_idx,1);
	mem += ((uintptr_t)p->tls_image - (uintptr_t)mem) & (p->tls_align-1);
	self->dtv[v[0]] = mem;
R
Rich Felker 已提交
776
	memcpy(mem, p->tls_image, p->tls_len);
777 778
	pthread_sigmask(SIG_SETMASK, &set, 0);
	return mem + v[1];
779 780
}

R
Rich Felker 已提交
781 782
static void update_tls_size()
{
783 784 785 786 787 788
	libc.tls_size = ALIGN(
		(1+tls_cnt) * sizeof(void *) +
		tls_offset +
		sizeof(struct pthread) +
		tls_align * 2,
	tls_align);
R
Rich Felker 已提交
789 790
}

791
void *__dynlink(int argc, char **argv)
R
Rich Felker 已提交
792
{
793
	size_t aux[AUX_CNT] = {0};
R
Rich Felker 已提交
794 795
	size_t i;
	Phdr *phdr;
796
	Ehdr *ehdr;
797
	static struct dso builtin_dsos[3];
R
Rich Felker 已提交
798 799
	struct dso *const app = builtin_dsos+0;
	struct dso *const lib = builtin_dsos+1;
800
	struct dso *const vdso = builtin_dsos+2;
R
Rich Felker 已提交
801
	char *env_preload=0;
R
Rich Felker 已提交
802
	size_t vdso_base;
803
	size_t *auxv;
R
Rich Felker 已提交
804 805

	/* Find aux vector just past environ[] */
806 807 808
	for (i=argc+1; argv[i]; i++)
		if (!memcmp(argv[i], "LD_LIBRARY_PATH=", 16))
			env_path = argv[i]+16;
R
Rich Felker 已提交
809 810
		else if (!memcmp(argv[i], "LD_PRELOAD=", 11))
			env_preload = argv[i]+11;
R
Rich Felker 已提交
811 812 813 814
	auxv = (void *)(argv+i+1);

	decode_vec(auxv, aux, AUX_CNT);

815 816
	/* Only trust user/env if kernel says we're not suid/sgid */
	if ((aux[0]&0x7800)!=0x7800 || aux[AT_UID]!=aux[AT_EUID]
R
Rich Felker 已提交
817
	  || aux[AT_GID]!=aux[AT_EGID] || aux[AT_SECURE]) {
818
		env_path = 0;
R
Rich Felker 已提交
819
		env_preload = 0;
820 821
	}

822 823 824 825 826 827 828 829 830
	/* If the dynamic linker was invoked as a program itself, AT_BASE
	 * will not be set. In that case, we assume the base address is
	 * the start of the page containing the PHDRs; I don't know any
	 * better approach... */
	if (!aux[AT_BASE]) {
		aux[AT_BASE] = aux[AT_PHDR] & -PAGE_SIZE;
		aux[AT_PHDR] = aux[AT_PHENT] = aux[AT_PHNUM] = 0;
	}

831 832 833
	/* The dynamic linker load address is passed by the kernel
	 * in the AUX vector, so this is easy. */
	lib->base = (void *)aux[AT_BASE];
834
	lib->name = lib->shortname = "libc.so";
835 836
	lib->global = 1;
	ehdr = (void *)lib->base;
R
Rich Felker 已提交
837 838 839 840 841
	lib->phnum = ehdr->e_phnum;
	lib->phdr = (void *)(aux[AT_BASE]+ehdr->e_phoff);
	find_map_range(lib->phdr, ehdr->e_phnum, ehdr->e_phentsize, lib);
	lib->dynv = (void *)(lib->base + find_dyn(lib->phdr,
                    ehdr->e_phnum, ehdr->e_phentsize));
842 843
	decode_dyn(lib);

844
	if (aux[AT_PHDR]) {
845
		size_t interp_off = 0;
846
		size_t tls_image = 0;
847
		/* Find load address of the main program, via AT_PHDR vs PT_PHDR. */
R
Rich Felker 已提交
848 849
		app->phdr = phdr = (void *)aux[AT_PHDR];
		app->phnum = aux[AT_PHNUM];
850 851 852
		for (i=aux[AT_PHNUM]; i; i--, phdr=(void *)((char *)phdr + aux[AT_PHENT])) {
			if (phdr->p_type == PT_PHDR)
				app->base = (void *)(aux[AT_PHDR] - phdr->p_vaddr);
853 854
			else if (phdr->p_type == PT_INTERP)
				interp_off = (size_t)phdr->p_vaddr;
855 856 857 858 859 860
			else if (phdr->p_type == PT_TLS) {
				tls_image = phdr->p_vaddr;
				app->tls_len = phdr->p_filesz;
				app->tls_size = phdr->p_memsz;
				app->tls_align = phdr->p_align;
			}
861
		}
862
		if (app->tls_size) app->tls_image = (char *)app->base + tls_image;
863
		if (interp_off) lib->name = (char *)app->base + interp_off;
864
		app->name = argv[0];
865 866
		app->dynv = (void *)(app->base + find_dyn(
			(void *)aux[AT_PHDR], aux[AT_PHNUM], aux[AT_PHENT]));
867 868
		find_map_range((void *)aux[AT_PHDR],
			aux[AT_PHNUM], aux[AT_PHENT], app);
869 870 871
	} else {
		int fd;
		char *ldname = argv[0];
R
Rich Felker 已提交
872
		size_t l = strlen(ldname);
873 874 875 876 877 878 879 880 881 882 883 884 885 886 887
		if (l >= 3 && !strcmp(ldname+l-3, "ldd")) ldd_mode = 1;
		*argv++ = (void *)-1;
		if (argv[0] && !strcmp(argv[0], "--")) *argv++ = (void *)-1;
		if (!argv[0]) {
			dprintf(2, "musl libc/dynamic program loader\n");
			dprintf(2, "usage: %s pathname%s\n", ldname,
				ldd_mode ? "" : " [args]");
			_exit(1);
		}
		fd = open(argv[0], O_RDONLY);
		if (fd < 0) {
			dprintf(2, "%s: cannot load %s: %s\n", ldname, argv[0], strerror(errno));
			_exit(1);
		}
		runtime = 1;
888
		ehdr = (void *)map_library(fd, app);
889 890 891 892 893 894
		if (!ehdr) {
			dprintf(2, "%s: %s: Not a valid dynamic program\n", ldname, argv[0]);
			_exit(1);
		}
		runtime = 0;
		close(fd);
895
		lib->name = ldname;
896
		app->name = argv[0];
R
Rich Felker 已提交
897 898
		app->phnum = ehdr->e_phnum;
		app->phdr = (void *)(app->base + ehdr->e_phoff);
899
		aux[AT_ENTRY] = ehdr->e_entry;
900
	}
901
	if (app->tls_size) {
R
Rich Felker 已提交
902
		app->tls_id = tls_cnt = 1;
903 904 905 906 907 908
#ifdef TLS_ABOVE_TP
		app->tls_offset = 0;
		tls_offset = app->tls_size
			+ ( -((uintptr_t)app->tls_image + app->tls_size)
			& (app->tls_align-1) );
#else
909 910 911
		tls_offset = app->tls_offset = app->tls_size
			+ ( -((uintptr_t)app->tls_image + app->tls_size)
			& (app->tls_align-1) );
912
#endif
R
Rich Felker 已提交
913
		tls_align = MAXP2(tls_align, app->tls_align);
914
	}
915
	app->global = 1;
916
	app->constructed = 1;
917 918 919
	decode_dyn(app);

	/* Attach to vdso, if provided by the kernel */
R
Rich Felker 已提交
920
	if (search_vec(auxv, &vdso_base, AT_SYSINFO_EHDR)) {
921
		ehdr = (void *)vdso_base;
R
Rich Felker 已提交
922 923
		vdso->phdr = phdr = (void *)(vdso_base + ehdr->e_phoff);
		vdso->phnum = ehdr->e_phnum;
924 925 926 927 928 929
		for (i=ehdr->e_phnum; i; i--, phdr=(void *)((char *)phdr + ehdr->e_phentsize)) {
			if (phdr->p_type == PT_DYNAMIC)
				vdso->dynv = (void *)(vdso_base + phdr->p_offset);
			if (phdr->p_type == PT_LOAD)
				vdso->base = (void *)(vdso_base - phdr->p_vaddr + phdr->p_offset);
		}
930 931
		vdso->name = "";
		vdso->shortname = "linux-gate.so.1";
932
		vdso->global = 1;
933
		decode_dyn(vdso);
934 935 936 937
		vdso->prev = lib;
		lib->next = vdso;
	}

938 939 940 941 942
	/* Initial dso chain consists only of the app. We temporarily
	 * append the dynamic linker/libc so we can relocate it, then
	 * restore the initial chain in preparation for loading third
	 * party libraries (preload/needed). */
	head = tail = app;
943
	ldso = lib;
944 945 946
	app->next = lib;
	reloc_all(lib);
	app->next = 0;
R
Rich Felker 已提交
947

948
	/* PAST THIS POINT, ALL LIBC INTERFACES ARE FULLY USABLE. */
R
Rich Felker 已提交
949

950
	/* Donate unused parts of app and library mapping to malloc */
R
Rich Felker 已提交
951 952 953
	reclaim_gaps(app->base, (void *)aux[AT_PHDR], aux[AT_PHENT], aux[AT_PHNUM]);
	ehdr = (void *)lib->base;
	reclaim_gaps(lib->base, (void *)(lib->base+ehdr->e_phoff),
954 955
		ehdr->e_phentsize, ehdr->e_phnum);

956
	/* Load preload/needed libraries, add their symbols to the global
957 958 959
	 * namespace, and perform all remaining relocations. The main
	 * program must be relocated LAST since it may contain copy
	 * relocations which depend on libraries' relocations. */
R
Rich Felker 已提交
960
	if (env_preload) load_preload(env_preload);
961 962
	load_deps(app);
	make_global(app);
963

R
Rich Felker 已提交
964 965 966 967
	reloc_all(app->next);
	reloc_all(app);

	update_tls_size();
968
	if (tls_cnt) {
969
		void *mem = mmap(0, libc.tls_size, PROT_READ|PROT_WRITE,
970 971
			MAP_ANONYMOUS|MAP_PRIVATE, -1, 0);
		if (mem==MAP_FAILED ||
972
		    !__install_initial_tls(__copy_tls(mem))) {
973
			dprintf(2, "%s: Error getting %zu bytes thread-local storage: %m\n",
974
				argv[0], libc.tls_size);
975 976 977 978
			_exit(127);
		}
	}

979
	if (ldso_fail) _exit(127);
980 981
	if (ldd_mode) _exit(0);

982 983 984 985
	/* Switch to runtime mode: any further failures in the dynamic
	 * linker are a reportable failure rather than a fatal startup
	 * error. If the dynamic loader (dlopen) will not be used, free
	 * all memory used by the dynamic linker. */
R
Rich Felker 已提交
986
	runtime = 1;
987

988
#ifndef DYNAMIC_IS_RO
989 990 991
	for (i=0; app->dynv[i]; i+=2)
		if (app->dynv[i]==DT_DEBUG)
			app->dynv[i+1] = (size_t)&debug;
992
#endif
993 994 995 996 997 998 999
	debug.ver = 1;
	debug.bp = _dl_debug_state;
	debug.head = head;
	debug.base = lib->base;
	debug.state = 0;
	_dl_debug_state();

1000
	if (ssp_used) __init_ssp((void *)aux[AT_RANDOM]);
1001

R
Rich Felker 已提交
1002 1003 1004
	errno = 0;
	return (void *)aux[AT_ENTRY];
}
1005

1006 1007 1008 1009 1010 1011
void __init_ldso_ctors(void)
{
	atexit(do_fini);
	do_init_fini(tail);
}

1012 1013
void *dlopen(const char *file, int mode)
{
R
Rich Felker 已提交
1014
	struct dso *volatile p, *orig_tail, *next;
R
Rich Felker 已提交
1015
	size_t orig_tls_cnt, orig_tls_offset, orig_tls_align;
1016
	size_t i;
1017
	int cs;
1018 1019 1020

	if (!file) return head;

1021
	pthread_setcancelstate(PTHREAD_CANCEL_DISABLE, &cs);
1022
	pthread_rwlock_wrlock(&lock);
1023
	__inhibit_ptc();
1024

1025 1026
	p = 0;
	orig_tls_cnt = tls_cnt;
R
Rich Felker 已提交
1027 1028
	orig_tls_offset = tls_offset;
	orig_tls_align = tls_align;
R
Rich Felker 已提交
1029 1030
	orig_tail = tail;

1031 1032
	if (setjmp(rtld_fail)) {
		/* Clean up anything new that was (partially) loaded */
1033
		if (p && p->deps) for (i=0; p->deps[i]; i++)
1034 1035
			if (p->deps[i]->global < 0)
				p->deps[i]->global = 0;
1036 1037 1038 1039 1040 1041
		for (p=orig_tail->next; p; p=next) {
			next = p->next;
			munmap(p->map, p->map_len);
			free(p->deps);
			free(p);
		}
1042
		tls_cnt = orig_tls_cnt;
R
Rich Felker 已提交
1043 1044
		tls_offset = orig_tls_offset;
		tls_align = orig_tls_align;
1045 1046
		tail = orig_tail;
		tail->next = 0;
1047
		p = 0;
1048 1049
		errflag = 1;
		goto end;
R
Rich Felker 已提交
1050 1051 1052
	} else p = load_library(file);

	if (!p) {
1053 1054
		snprintf(errbuf, sizeof errbuf,
			"Error loading shared library %s: %m", file);
R
Rich Felker 已提交
1055
		errflag = 1;
1056
		goto end;
1057 1058 1059 1060 1061
	}

	/* First load handling */
	if (!p->deps) {
		load_deps(p);
R
Rich Felker 已提交
1062
		if (p->deps) for (i=0; p->deps[i]; i++)
1063 1064 1065
			if (!p->deps[i]->global)
				p->deps[i]->global = -1;
		if (!p->global) p->global = -1;
1066
		reloc_all(p);
R
Rich Felker 已提交
1067
		if (p->deps) for (i=0; p->deps[i]; i++)
1068 1069 1070
			if (p->deps[i]->global < 0)
				p->deps[i]->global = 0;
		if (p->global < 0) p->global = 0;
1071 1072 1073
	}

	if (mode & RTLD_GLOBAL) {
R
Rich Felker 已提交
1074
		if (p->deps) for (i=0; p->deps[i]; i++)
1075 1076 1077 1078
			p->deps[i]->global = 1;
		p->global = 1;
	}

R
Rich Felker 已提交
1079
	update_tls_size();
1080

1081
	if (ssp_used) __init_ssp(libc.auxv);
1082

1083
	_dl_debug_state();
1084
	orig_tail = tail;
1085
end:
1086
	__release_ptc();
R
Rich Felker 已提交
1087
	if (p) gencnt++;
1088
	pthread_rwlock_unlock(&lock);
1089
	if (p) do_init_fini(orig_tail);
1090
	pthread_setcancelstate(cs, 0);
1091 1092 1093
	return p;
}

R
Rich Felker 已提交
1094
static void *do_dlsym(struct dso *p, const char *s, void *ra)
1095 1096
{
	size_t i;
1097
	uint32_t h = 0, gh = 0;
1098
	Sym *sym;
1099
	if (p == head || p == RTLD_DEFAULT || p == RTLD_NEXT) {
1100 1101 1102
		if (p == RTLD_DEFAULT) {
			p = head;
		} else if (p == RTLD_NEXT) {
1103 1104
			for (p=head; p && (unsigned char *)ra-p->map>p->map_len; p=p->next);
			if (!p) p=head;
1105
			p = p->next;
1106
		}
1107
		struct symdef def = find_sym(p, s, 0);
1108
		if (!def.sym) goto failed;
1109 1110
		if ((def.sym->st_info&0xf) == STT_TLS)
			return __tls_get_addr((size_t []){def.dso->tls_id, def.sym->st_value});
1111
		return def.dso->base + def.sym->st_value;
R
Rich Felker 已提交
1112
	}
1113 1114 1115 1116 1117 1118 1119
	if (p->ghashtab) {
		gh = gnu_hash(s);
		sym = gnu_lookup(s, gh, p);
	} else {
		h = sysv_hash(s);
		sym = sysv_lookup(s, h, p);
	}
1120 1121
	if (sym && (sym->st_info&0xf) == STT_TLS)
		return __tls_get_addr((size_t []){p->tls_id, sym->st_value});
1122 1123 1124
	if (sym && sym->st_value && (1<<(sym->st_info&0xf) & OK_TYPES))
		return p->base + sym->st_value;
	if (p->deps) for (i=0; p->deps[i]; i++) {
1125 1126
		if (p->deps[i]->ghashtab) {
			if (!gh) gh = gnu_hash(s);
1127
			sym = gnu_lookup(s, gh, p->deps[i]);
1128 1129 1130 1131
		} else {
			if (!h) h = sysv_hash(s);
			sym = sysv_lookup(s, h, p->deps[i]);
		}
1132 1133
		if (sym && (sym->st_info&0xf) == STT_TLS)
			return __tls_get_addr((size_t []){p->deps[i]->tls_id, sym->st_value});
1134 1135 1136
		if (sym && sym->st_value && (1<<(sym->st_info&0xf) & OK_TYPES))
			return p->deps[i]->base + sym->st_value;
	}
1137
failed:
R
Rich Felker 已提交
1138
	errflag = 1;
1139
	snprintf(errbuf, sizeof errbuf, "Symbol not found: %s", s);
1140 1141 1142
	return 0;
}

1143 1144 1145 1146 1147 1148 1149 1150 1151 1152 1153 1154 1155 1156 1157 1158 1159 1160 1161 1162 1163 1164 1165 1166 1167 1168 1169 1170 1171 1172 1173 1174 1175 1176 1177 1178 1179 1180 1181 1182 1183 1184 1185 1186 1187 1188 1189 1190 1191 1192 1193 1194 1195 1196 1197 1198 1199 1200 1201 1202 1203
int __dladdr(void *addr, Dl_info *info)
{
	struct dso *p;
	Sym *sym;
	uint32_t nsym;
	char *strings;
	size_t i;
	void *best = 0;
	char *bestname;

	pthread_rwlock_rdlock(&lock);
	for (p=head; p && (unsigned char *)addr-p->map>p->map_len; p=p->next);
	pthread_rwlock_unlock(&lock);

	if (!p) return 0;

	sym = p->syms;
	strings = p->strings;
	if (p->hashtab) {
		nsym = p->hashtab[1];
	} else {
		uint32_t *buckets;
		uint32_t *hashval;
		buckets = p->ghashtab + 4 + (p->ghashtab[2]*sizeof(size_t)/4);
		sym += p->ghashtab[1];
		for (i = 0; i < p->ghashtab[0]; i++) {
			if (buckets[i] > nsym)
				nsym = buckets[i];
		}
		if (nsym) {
			nsym -= p->ghashtab[1];
			hashval = buckets + p->ghashtab[0] + nsym;
			do nsym++;
			while (!(*hashval++ & 1));
		}
	}

	for (; nsym; nsym--, sym++) {
		if (sym->st_shndx && sym->st_value
		 && (1<<(sym->st_info&0xf) & OK_TYPES)
		 && (1<<(sym->st_info>>4) & OK_BINDS)) {
			void *symaddr = p->base + sym->st_value;
			if (symaddr > addr || symaddr < best)
				continue;
			best = symaddr;
			bestname = strings + sym->st_name;
			if (addr == symaddr)
				break;
		}
	}

	if (!best) return 0;

	info->dli_fname = p->name;
	info->dli_fbase = p->base;
	info->dli_sname = bestname;
	info->dli_saddr = best;

	return 1;
}

1204
void *__dlsym(void *restrict p, const char *restrict s, void *restrict ra)
1205 1206 1207
{
	void *res;
	pthread_rwlock_rdlock(&lock);
R
Rich Felker 已提交
1208
	res = do_dlsym(p, s, ra);
1209 1210 1211
	pthread_rwlock_unlock(&lock);
	return res;
}
R
Rich Felker 已提交
1212 1213 1214 1215 1216 1217 1218 1219 1220 1221 1222 1223 1224 1225 1226 1227 1228 1229 1230 1231 1232 1233 1234 1235 1236 1237

int dl_iterate_phdr(int(*callback)(struct dl_phdr_info *info, size_t size, void *data), void *data)
{
	struct dso *current;
	struct dl_phdr_info info;
	int ret = 0;
	for(current = head; current;) {
		info.dlpi_addr      = (uintptr_t)current->base;
		info.dlpi_name      = current->name;
		info.dlpi_phdr      = current->phdr;
		info.dlpi_phnum     = current->phnum;
		info.dlpi_adds      = gencnt;
		info.dlpi_subs      = 0;
		info.dlpi_tls_modid = current->tls_id;
		info.dlpi_tls_data  = current->tls_image;

		ret = (callback)(&info, sizeof (info), data);

		if (ret != 0) break;

		pthread_rwlock_rdlock(&lock);
		current = current->next;
		pthread_rwlock_unlock(&lock);
	}
	return ret;
}
1238 1239 1240 1241 1242
#else
void *dlopen(const char *file, int mode)
{
	return 0;
}
1243
void *__dlsym(void *restrict p, const char *restrict s, void *restrict ra)
1244 1245 1246
{
	return 0;
}
1247 1248 1249 1250
int __dladdr (void *addr, Dl_info *info)
{
	return 0;
}
1251
#endif
1252 1253 1254

char *dlerror()
{
R
Rich Felker 已提交
1255 1256
	if (!errflag) return 0;
	errflag = 0;
1257
	return errbuf;
1258 1259 1260 1261 1262 1263
}

int dlclose(void *p)
{
	return 0;
}