dynlink.c 38.2 KB
Newer Older
1
#define _GNU_SOURCE
R
Rich Felker 已提交
2 3 4 5 6 7 8 9 10 11 12
#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 <fcntl.h>
#include <sys/stat.h>
#include <errno.h>
R
Rich Felker 已提交
13
#include <link.h>
R
Rich Felker 已提交
14
#include <setjmp.h>
15
#include <pthread.h>
R
Rich Felker 已提交
16
#include <ctype.h>
17
#include <dlfcn.h>
18 19
#include "pthread_impl.h"
#include "libc.h"
R
Rich Felker 已提交
20

R
Rich Felker 已提交
21
static int errflag;
22
static char errbuf[128];
R
Rich Felker 已提交
23

24
#ifdef SHARED
R
Rich Felker 已提交
25

R
Rich Felker 已提交
26 27 28 29 30 31 32 33 34 35 36 37 38 39
#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 已提交
40 41 42
#define MAXP2(a,b) (-(-(a)&-(b)))
#define ALIGN(x,y) ((x)+(y)-1 & -(y))

43 44 45 46 47 48 49 50 51 52 53 54
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 已提交
55
	struct dso *next, *prev;
56

R
Rich Felker 已提交
57 58
	Phdr *phdr;
	int phnum;
R
Rich Felker 已提交
59 60
	int refcnt;
	Sym *syms;
61
	uint32_t *hashtab;
62
	uint32_t *ghashtab;
63
	int16_t *versym;
R
Rich Felker 已提交
64 65 66 67 68
	char *strings;
	unsigned char *map;
	size_t map_len;
	dev_t dev;
	ino_t ino;
69
	signed char global;
70 71
	char relocated;
	char constructed;
72
	char kernel_mapped;
73
	struct dso **deps, *needed_by;
74
	char *rpath_orig, *rpath;
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
void __init_libc(char **, char *);
95

96 97
const char *__libc_get_version(void);

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

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

114
#define AUX_CNT 38
R
Rich Felker 已提交
115 116 117 118 119 120 121 122 123 124 125
#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];
	}
}

126 127 128 129 130 131 132 133 134
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 已提交
135
{
136
	const unsigned char *s = (void *)s0;
R
Rich Felker 已提交
137 138 139 140 141 142 143 144
	uint_fast32_t h = 0;
	while (*s) {
		h = 16*h + *s++;
		h ^= h>>24 & 0xf0;
	}
	return h & 0xfffffff;
}

145 146 147 148 149 150 151 152 153 154
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 已提交
155 156
{
	size_t i;
R
Rich Felker 已提交
157 158 159
	Sym *syms = dso->syms;
	uint32_t *hashtab = dso->hashtab;
	char *strings = dso->strings;
R
Rich Felker 已提交
160
	for (i=hashtab[2+h%hashtab[0]]; i; i=hashtab[2+hashtab[0]+i]) {
161 162
		if ((!dso->versym || dso->versym[i] >= 0)
		    && (!strcmp(s, strings+syms[i].st_name)))
R
Rich Felker 已提交
163 164 165 166 167
			return syms+i;
	}
	return 0;
}

168 169
static Sym *gnu_lookup(const char *s, uint32_t h1, struct dso *dso)
{
170 171
	Sym *syms = dso->syms;
	char *strings = dso->strings;
172 173 174 175 176
	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;
177
	uint32_t i = buckets[h1 % nbuckets];
178

179
	if (!i) return 0;
180

181
	hashval = buckets + nbuckets + (i - hashtab[1]);
182

183
	for (h1 |= 1; ; i++) {
184
		h2 = *hashval++;
185 186 187
		if ((!dso->versym || dso->versym[i] >= 0)
		    && (h1 == (h2|1)) && !strcmp(s, strings + syms[i].st_name))
			return syms+i;
188 189 190 191 192 193
		if (h2 & 1) break;
	}

	return 0;
}

194
#define OK_TYPES (1<<STT_NOTYPE | 1<<STT_OBJECT | 1<<STT_FUNC | 1<<STT_COMMON | 1<<STT_TLS)
195
#define OK_BINDS (1<<STB_GLOBAL | 1<<STB_WEAK | 1<<STB_GNU_UNIQUE)
R
Rich Felker 已提交
196

197
static struct symdef find_sym(struct dso *dso, const char *s, int need_def)
R
Rich Felker 已提交
198
{
199
	uint32_t h = 0, gh = 0;
200
	struct symdef def = {0};
201 202 203 204 205 206 207
	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 已提交
208
	for (; dso; dso=dso->next) {
209 210
		Sym *sym;
		if (!dso->global) continue;
211 212 213 214 215 216 217
		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);
		}
218 219 220 221 222 223 224 225 226 227 228 229 230 231
		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 已提交
232
	}
233
	return def;
R
Rich Felker 已提交
234 235
}

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

	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;
254
			ctx = IS_COPY(type) ? head->next : head;
255 256
			def = find_sym(ctx, name, IS_PLT(type));
			if (!def.sym && sym->st_info>>4 != STB_WEAK) {
257 258
				snprintf(errbuf, sizeof errbuf,
					"Error relocating %s: %s: symbol not found",
259
					dso->name, name);
260
				if (runtime) longjmp(*rtld_fail, 1);
261
				dprintf(2, "%s\n", errbuf);
262 263
				ldso_fail = 1;
				continue;
R
Rich Felker 已提交
264
			}
265
		} else {
266 267 268
			sym = 0;
			def.sym = 0;
			def.dso = 0;
R
Rich Felker 已提交
269
		}
270 271 272
		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 已提交
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 303 304 305 306
/* 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);
	}
}

307
static void *map_library(int fd, struct dso *dso)
R
Rich Felker 已提交
308
{
309
	Ehdr buf[(896+sizeof(Ehdr))/sizeof(Ehdr)];
310
	void *allocated_buf=0;
R
Rich Felker 已提交
311 312 313 314 315
	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;
316
	Phdr *ph, *ph0;
R
Rich Felker 已提交
317
	unsigned prot;
318
	unsigned char *map=MAP_FAILED, *base;
319
	size_t dyn=0;
320
	size_t tls_image=0;
R
Rich Felker 已提交
321 322 323
	size_t i;

	ssize_t l = read(fd, buf, sizeof buf);
324
	eh = buf;
325 326 327
	if (l<0) return 0;
	if (l<sizeof *eh || (eh->e_type != ET_DYN && eh->e_type != ET_EXEC))
		goto noexec;
R
Rich Felker 已提交
328
	phsize = eh->e_phentsize * eh->e_phnum;
329 330 331 332 333 334 335 336
	if (phsize > sizeof buf - sizeof *eh) {
		allocated_buf = malloc(phsize);
		if (!allocated_buf) return 0;
		l = pread(fd, allocated_buf, phsize, eh->e_phoff);
		if (l < 0) goto error;
		if (l != phsize) goto noexec;
		ph = ph0 = allocated_buf;
	} else if (eh->e_phoff + phsize > l) {
337
		l = pread(fd, buf+1, phsize, eh->e_phoff);
338 339
		if (l < 0) goto error;
		if (l != phsize) goto noexec;
340 341 342
		ph = ph0 = (void *)(buf + 1);
	} else {
		ph = ph0 = (void *)((char *)buf + eh->e_phoff);
R
Rich Felker 已提交
343 344 345 346
	}
	for (i=eh->e_phnum; i; i--, ph=(void *)((char *)ph+eh->e_phentsize)) {
		if (ph->p_type == PT_DYNAMIC)
			dyn = ph->p_vaddr;
347 348 349 350 351 352
		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 已提交
353 354 355 356 357 358 359 360 361 362 363 364
		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;
		}
	}
365
	if (!dyn) goto noexec;
R
Rich Felker 已提交
366 367 368 369 370 371 372 373 374
	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. */
375
	map = mmap((void *)addr_min, map_len, prot, MAP_PRIVATE, fd, off_start);
376
	if (map==MAP_FAILED) goto error;
377 378 379 380 381 382
	/* If the loaded file is not relocatable and the requested address is
	 * not available, then the load operation must fail. */
	if (eh->e_type != ET_DYN && addr_min && map!=(void *)addr_min) {
		errno = EBUSY;
		goto error;
	}
R
Rich Felker 已提交
383
	base = map - addr_min;
384 385 386
	dso->phdr = 0;
	dso->phnum = 0;
	for (ph=ph0, i=eh->e_phnum; i; i--, ph=(void *)((char *)ph+eh->e_phentsize)) {
R
Rich Felker 已提交
387
		if (ph->p_type != PT_LOAD) continue;
388 389 390 391 392 393 394 395
		/* Check if the programs headers are in this load segment, and
		 * if so, record the address for use by dl_iterate_phdr. */
		if (!dso->phdr && eh->e_phoff >= ph->p_offset
		    && eh->e_phoff+phsize <= ph->p_offset+ph->p_filesz) {
			dso->phdr = (void *)(base + ph->p_vaddr
				+ (eh->e_phoff-ph->p_offset));
			dso->phnum = eh->e_phnum;
		}
R
Rich Felker 已提交
396 397 398 399 400 401 402 403
		/* 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));
404 405
		if (mmap(base+this_min, this_max-this_min, prot, MAP_PRIVATE|MAP_FIXED, fd, off_start) == MAP_FAILED)
			goto error;
R
Rich Felker 已提交
406 407 408 409
		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);
410 411
			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 已提交
412 413
		}
	}
R
Rich Felker 已提交
414 415
	for (i=0; ((size_t *)(base+dyn))[i]; i+=2)
		if (((size_t *)(base+dyn))[i]==DT_TEXTREL) {
416 417
			if (mprotect(map, map_len, PROT_READ|PROT_WRITE|PROT_EXEC) < 0)
				goto error;
R
Rich Felker 已提交
418 419
			break;
		}
420
	if (!runtime) reclaim_gaps(base, ph0, eh->e_phentsize, eh->e_phnum);
421 422 423 424 425
	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);
426
	free(allocated_buf);
R
Rich Felker 已提交
427
	return map;
428 429
noexec:
	errno = ENOEXEC;
430
error:
431 432
	if (map!=MAP_FAILED) munmap(map, map_len);
	free(allocated_buf);
433
	return 0;
R
Rich Felker 已提交
434 435
}

436
static int path_open(const char *name, const char *s, char *buf, size_t buf_size)
437
{
438 439
	size_t l;
	int fd;
440
	for (;;) {
441 442 443 444 445
		s += strspn(s, ":\n");
		l = strcspn(s, ":\n");
		if (l-1 >= INT_MAX) return -1;
		if (snprintf(buf, buf_size, "%.*s/%s", (int)l, s, name) >= buf_size)
			continue;
446
		if ((fd = open(buf, O_RDONLY|O_CLOEXEC))>=0) return fd;
447
		s += l;
448 449 450
	}
}

451 452 453 454 455 456 457 458 459 460 461 462 463
static int fixup_rpath(struct dso *p, char *buf, size_t buf_size)
{
	size_t n, l;
	const char *s, *t, *origin;
	char *d;
	if (p->rpath) return 0;
	if (!p->rpath_orig) return -1;
	if (!strchr(p->rpath_orig, '$')) {
		p->rpath = p->rpath_orig;
		return 0;
	}
	n = 0;
	s = p->rpath_orig;
R
Rich Felker 已提交
464 465 466
	while ((t=strchr(s, '$'))) {
		if (strncmp(t, "$ORIGIN", 7) && strncmp(t, "${ORIGIN}", 9))
			return -1;
467 468 469 470 471 472 473 474 475 476 477 478 479 480
		s = t+1;
		n++;
	}
	if (n > SSIZE_MAX/PATH_MAX) return -1;

	if (p->kernel_mapped) {
		/* $ORIGIN searches cannot be performed for the main program
		 * when it is suid/sgid/AT_SECURE. This is because the
		 * pathname is under the control of the caller of execve.
		 * For libraries, however, $ORIGIN can be processed safely
		 * since the library's pathname came from a trusted source
		 * (either system paths or a call to dlopen). */
		if (libc.secure)
			return -1;
R
Rich Felker 已提交
481 482
		l = readlink("/proc/self/exe", buf, buf_size);
		if (l >= buf_size)
483
			return -1;
R
Rich Felker 已提交
484
		buf[l] = 0;
485 486 487 488 489 490 491 492 493 494 495
		origin = buf;
	} else {
		origin = p->name;
	}
	t = strrchr(origin, '/');
	l = t ? t-origin : 0;
	p->rpath = malloc(strlen(p->rpath_orig) + n*l + 1);
	if (!p->rpath) return -1;

	d = p->rpath;
	s = p->rpath_orig;
R
Rich Felker 已提交
496
	while ((t=strchr(s, '$'))) {
497 498 499 500
		memcpy(d, s, t-s);
		d += t-s;
		memcpy(d, origin, l);
		d += l;
R
Rich Felker 已提交
501 502
		/* It was determined previously that the '$' is followed
		 * either by "ORIGIN" or "{ORIGIN}". */
503 504 505 506 507 508
		s = t + 7 + 2*(t[1]=='{');
	}
	strcpy(d, s);
	return 0;
}

509 510 511 512 513 514
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]);
515 516
	if (dyn[0]&(1<<DT_HASH))
		p->hashtab = (void *)(p->base + dyn[DT_HASH]);
517
	if (dyn[0]&(1<<DT_RPATH))
518
		p->rpath_orig = (void *)(p->strings + dyn[DT_RPATH]);
519 520
	if (search_vec(p->dynv, dyn, DT_GNU_HASH))
		p->ghashtab = (void *)(p->base + *dyn);
521 522
	if (search_vec(p->dynv, dyn, DT_VERSYM))
		p->versym = (void *)(p->base + *dyn);
523 524
}

525
static struct dso *load_library(const char *name, struct dso *needed_by)
R
Rich Felker 已提交
526
{
527
	char buf[2*NAME_MAX+2];
528
	const char *pathname;
529
	unsigned char *map;
530
	struct dso *p, temp_dso = {0};
R
Rich Felker 已提交
531 532
	int fd;
	struct stat st;
533 534
	size_t alloc_size;
	int n_th = 0;
535
	int is_self = 0;
R
Rich Felker 已提交
536 537 538 539 540 541 542 543

	/* 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;
544
			for (rp=reserved; *rp && strncmp(name+3, rp, l-3); rp+=strlen(rp)+1);
R
Rich Felker 已提交
545
			if (*rp) {
546 547 548 549 550 551 552 553 554 555 556 557
				if (ldd_mode) {
					/* Track which names have been resolved
					 * and only report each one once. */
					static unsigned reported;
					unsigned mask = 1U<<(rp-reserved);
					if (!(reported & mask)) {
						reported |= mask;
						dprintf(1, "\t%s => %s (%p)\n",
							name, ldso->name,
							ldso->base);
					}
				}
558
				is_self = 1;
R
Rich Felker 已提交
559 560 561
			}
		}
	}
562 563 564 565 566 567 568 569 570
	if (!strcmp(name, ldso->name)) is_self = 1;
	if (is_self) {
		if (!ldso->prev) {
			tail->next = ldso;
			ldso->prev = tail;
			tail = ldso->next ? ldso->next : ldso;
		}
		return ldso;
	}
571
	if (strchr(name, '/')) {
572
		pathname = name;
573
		fd = open(name, O_RDONLY|O_CLOEXEC);
R
Rich Felker 已提交
574
	} else {
575 576 577 578 579 580 581
		/* 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;
			}
		}
582
		if (strlen(name) > NAME_MAX) return 0;
583
		fd = -1;
584
		if (env_path) fd = path_open(name, env_path, buf, sizeof buf);
585
		for (p=needed_by; fd < 0 && p; p=p->needed_by)
586
			if (!fixup_rpath(p, buf, sizeof buf))
587
				fd = path_open(name, p->rpath, buf, sizeof buf);
588 589
		if (fd < 0) {
			if (!sys_path) {
590 591 592 593 594 595 596 597 598 599 600 601 602 603 604 605 606 607 608 609
				char *prefix = 0;
				size_t prefix_len;
				if (ldso->name[0]=='/') {
					char *s, *t, *z;
					for (s=t=z=ldso->name; *s; s++)
						if (*s=='/') z=t, t=s;
					prefix_len = z-ldso->name;
					if (prefix_len < PATH_MAX)
						prefix = ldso->name;
				}
				if (!prefix) {
					prefix = "";
					prefix_len = 0;
				}
				char etc_ldso_path[prefix_len + 1
					+ sizeof "/etc/ld-musl-" LDSO_ARCH ".path"];
				snprintf(etc_ldso_path, sizeof etc_ldso_path,
					"%.*s/etc/ld-musl-" LDSO_ARCH ".path",
					(int)prefix_len, prefix);
				FILE *f = fopen(etc_ldso_path, "rbe");
610
				if (f) {
611
					if (getdelim(&sys_path, (size_t[1]){0}, 0, f) <= 0) {
612
						free(sys_path);
613
						sys_path = "";
614
					}
615
					fclose(f);
616 617
				} else if (errno != ENOENT) {
					sys_path = "";
618 619
				}
			}
620 621
			if (!sys_path) sys_path = "/lib:/usr/local/lib:/usr/lib";
			fd = path_open(name, sys_path, buf, sizeof buf);
R
Rich Felker 已提交
622
		}
623
		pathname = buf;
R
Rich Felker 已提交
624 625 626 627 628 629 630 631
	}
	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) {
632 633 634
			/* 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. */
635 636
			if (!p->shortname && pathname != name)
				p->shortname = strrchr(p->name, '/')+1;
R
Rich Felker 已提交
637 638 639 640 641
			close(fd);
			p->refcnt++;
			return p;
		}
	}
642
	map = noload ? 0 : map_library(fd, &temp_dso);
R
Rich Felker 已提交
643 644
	close(fd);
	if (!map) return 0;
645 646 647 648 649 650 651 652 653 654

	/* 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);
655
		n_th = libc.threads_minus_1 + 1;
656 657 658 659
		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 已提交
660
	if (!p) {
661
		munmap(map, temp_dso.map_len);
R
Rich Felker 已提交
662 663
		return 0;
	}
664
	memcpy(p, &temp_dso, sizeof temp_dso);
665
	decode_dyn(p);
R
Rich Felker 已提交
666 667 668
	p->dev = st.st_dev;
	p->ino = st.st_ino;
	p->refcnt = 1;
669
	p->needed_by = needed_by;
R
Rich Felker 已提交
670
	p->name = p->buf;
671 672 673
	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;
674
	if (p->tls_image) {
675
		if (runtime && !__pthread_self_init()) {
676
			munmap(map, p->map_len);
677 678 679
			free(p);
			return 0;
		}
680
		p->tls_id = ++tls_cnt;
R
Rich Felker 已提交
681
		tls_align = MAXP2(tls_align, p->tls_align);
682 683 684 685 686
#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 已提交
687 688 689 690
		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;
691
#endif
692 693 694 695
		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 已提交
696 697 698 699 700

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

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

R
Rich Felker 已提交
703 704 705 706 707
	return p;
}

static void load_deps(struct dso *p)
{
708 709
	size_t i, ndeps=0;
	struct dso ***deps = &p->deps, **tmp, *dep;
R
Rich Felker 已提交
710 711 712
	for (; p; p=p->next) {
		for (i=0; p->dynv[i]; i+=2) {
			if (p->dynv[i] != DT_NEEDED) continue;
713
			dep = load_library(p->strings + p->dynv[i+1], p);
714
			if (!dep) {
715 716
				snprintf(errbuf, sizeof errbuf,
					"Error loading shared library %s: %m (needed by %s)",
R
Rich Felker 已提交
717
					p->strings + p->dynv[i+1], p->name);
718
				if (runtime) longjmp(*rtld_fail, 1);
719
				dprintf(2, "%s\n", errbuf);
720 721
				ldso_fail = 1;
				continue;
R
Rich Felker 已提交
722
			}
723 724
			if (runtime) {
				tmp = realloc(*deps, sizeof(*tmp)*(ndeps+2));
725
				if (!tmp) longjmp(*rtld_fail, 1);
726 727 728 729
				tmp[ndeps++] = dep;
				tmp[ndeps] = 0;
				*deps = tmp;
			}
R
Rich Felker 已提交
730 731 732 733
		}
	}
}

R
Rich Felker 已提交
734 735 736 737 738 739 740 741 742
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;
743
		load_library(s, 0);
R
Rich Felker 已提交
744 745 746 747
		*z = tmp;
	}
}

748 749 750 751 752
static void make_global(struct dso *p)
{
	for (; p; p=p->next) p->global = 1;
}

R
Rich Felker 已提交
753 754 755 756 757 758
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 已提交
759 760 761
#ifdef NEED_ARCH_RELOCS
		do_arch_relocs(p, head);
#endif
762 763 764 765
		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);
766
		p->relocated = 1;
R
Rich Felker 已提交
767 768 769
	}
}

770 771 772 773 774 775 776 777
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;
}

778 779 780 781 782 783 784 785 786 787 788 789 790 791 792 793
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;
}

794 795 796 797 798 799 800
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);
801 802
		if (dyn[0] & (1<<DT_FINI_ARRAY)) {
			size_t n = dyn[DT_FINI_ARRAYSZ]/sizeof(size_t);
803 804
			size_t *fn = (size_t *)(p->base + dyn[DT_FINI_ARRAY])+n;
			while (n--) ((void (*)(void))*--fn)();
805
		}
806
#ifndef NO_LEGACY_INITFINI
807
		if ((dyn[0] & (1<<DT_FINI)) && dyn[DT_FINI])
808
			((void (*)(void))(p->base + dyn[DT_FINI]))();
809
#endif
810 811 812
	}
}

813 814 815
static void do_init_fini(struct dso *p)
{
	size_t dyn[DYN_CNT] = {0};
816
	int need_locking = libc.threads_minus_1;
817 818 819 820
	/* 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);
821
	for (; p; p=p->prev) {
822 823
		if (p->constructed) continue;
		p->constructed = 1;
824
		decode_vec(p->dynv, dyn, DYN_CNT);
825
		if (dyn[0] & ((1<<DT_FINI) | (1<<DT_FINI_ARRAY))) {
826 827 828
			p->fini_next = fini_head;
			fini_head = p;
		}
829
#ifndef NO_LEGACY_INITFINI
830
		if ((dyn[0] & (1<<DT_INIT)) && dyn[DT_INIT])
831
			((void (*)(void))(p->base + dyn[DT_INIT]))();
832
#endif
833 834 835 836 837
		if (dyn[0] & (1<<DT_INIT_ARRAY)) {
			size_t n = dyn[DT_INIT_ARRAYSZ]/sizeof(size_t);
			size_t *fn = (void *)(p->base + dyn[DT_INIT_ARRAY]);
			while (n--) ((void (*)(void))*fn++)();
		}
838 839 840 841
		if (!need_locking && libc.threads_minus_1) {
			need_locking = 1;
			pthread_mutex_lock(&init_fini_lock);
		}
842
	}
843
	if (need_locking) pthread_mutex_unlock(&init_fini_lock);
844 845
}

846 847 848 849
void _dl_debug_state(void)
{
}

850 851 852 853 854 855 856 857 858 859 860 861 862
void __reset_tls()
{
	pthread_t self = __pthread_self();
	struct dso *p;
	for (p=head; p; p=p->next) {
		if (!p->tls_id || !self->dtv[p->tls_id]) continue;
		memcpy(self->dtv[p->tls_id], p->tls_image, p->tls_len);
		memset((char *)self->dtv[p->tls_id]+p->tls_len, 0,
			p->tls_size - p->tls_len);
		if (p->tls_id == (size_t)self->dtv[0]) break;
	}
}

863
void *__copy_tls(unsigned char *mem)
864
{
R
Rich Felker 已提交
865
	pthread_t td;
866
	struct dso *p;
R
Rich Felker 已提交
867 868 869

	if (!tls_cnt) return mem;

870
	void **dtv = (void *)mem;
871
	dtv[0] = (void *)tls_cnt;
R
Rich Felker 已提交
872

873 874 875 876 877 878 879 880 881 882 883 884
#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
885
	mem += libc.tls_size - sizeof(struct pthread);
R
Rich Felker 已提交
886 887 888 889
	mem -= (uintptr_t)mem & (tls_align-1);
	td = (pthread_t)mem;

	for (p=head; p; p=p->next) {
890
		if (!p->tls_id) continue;
R
Rich Felker 已提交
891 892
		dtv[p->tls_id] = mem - p->tls_offset;
		memcpy(dtv[p->tls_id], p->tls_image, p->tls_len);
893
	}
894
#endif
R
Rich Felker 已提交
895 896
	td->dtv = dtv;
	return td;
897 898
}

899
void *__tls_get_addr(size_t *v)
900 901
{
	pthread_t self = __pthread_self();
902
	if (v[0]<=(size_t)self->dtv[0] && self->dtv[v[0]])
903 904 905 906
		return (char *)self->dtv[v[0]]+v[1];

	/* Block signals to make accessing new TLS async-signal-safe */
	sigset_t set;
907
	pthread_sigmask(SIG_BLOCK, SIGALL_SET, &set);
908
	if (v[0]<=(size_t)self->dtv[0] && self->dtv[v[0]]) {
909 910 911 912 913 914 915 916 917 918 919 920
		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 */
921
	if (v[0] > (size_t)self->dtv[0]) {
922 923
		void **newdtv = p->new_dtv +
			(v[0]+1)*sizeof(void *)*a_fetch_add(&p->new_dtv_idx,1);
924
		memcpy(newdtv, self->dtv,
925 926 927
			((size_t)self->dtv[0]+1) * sizeof(void *));
		newdtv[0] = (void *)v[0];
		self->dtv = newdtv;
928
	}
929 930 931 932 933 934

	/* 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 已提交
935
	memcpy(mem, p->tls_image, p->tls_len);
936 937
	pthread_sigmask(SIG_SETMASK, &set, 0);
	return mem + v[1];
938 939
}

R
Rich Felker 已提交
940 941
static void update_tls_size()
{
942 943 944 945 946 947
	libc.tls_size = ALIGN(
		(1+tls_cnt) * sizeof(void *) +
		tls_offset +
		sizeof(struct pthread) +
		tls_align * 2,
	tls_align);
R
Rich Felker 已提交
948 949
}

950
void *__dynlink(int argc, char **argv)
R
Rich Felker 已提交
951
{
952
	size_t aux[AUX_CNT] = {0};
R
Rich Felker 已提交
953 954
	size_t i;
	Phdr *phdr;
955
	Ehdr *ehdr;
956
	static struct dso builtin_dsos[3];
R
Rich Felker 已提交
957 958
	struct dso *const app = builtin_dsos+0;
	struct dso *const lib = builtin_dsos+1;
959
	struct dso *const vdso = builtin_dsos+2;
R
Rich Felker 已提交
960
	char *env_preload=0;
R
Rich Felker 已提交
961
	size_t vdso_base;
962
	size_t *auxv;
963
	char **envp = argv+argc+1;
R
Rich Felker 已提交
964 965

	/* Find aux vector just past environ[] */
966 967 968
	for (i=argc+1; argv[i]; i++)
		if (!memcmp(argv[i], "LD_LIBRARY_PATH=", 16))
			env_path = argv[i]+16;
R
Rich Felker 已提交
969 970
		else if (!memcmp(argv[i], "LD_PRELOAD=", 11))
			env_preload = argv[i]+11;
R
Rich Felker 已提交
971 972 973 974
	auxv = (void *)(argv+i+1);

	decode_vec(auxv, aux, AUX_CNT);

975 976
	/* 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 已提交
977
	  || aux[AT_GID]!=aux[AT_EGID] || aux[AT_SECURE]) {
978
		env_path = 0;
R
Rich Felker 已提交
979
		env_preload = 0;
980
		libc.secure = 1;
981
	}
982
	libc.page_size = aux[AT_PAGESZ];
983

984 985 986 987 988 989 990 991 992
	/* 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;
	}

993 994 995
	/* The dynamic linker load address is passed by the kernel
	 * in the AUX vector, so this is easy. */
	lib->base = (void *)aux[AT_BASE];
996
	lib->name = lib->shortname = "libc.so";
997
	lib->global = 1;
998
	lib->kernel_mapped = 1;
999
	ehdr = (void *)lib->base;
R
Rich Felker 已提交
1000 1001 1002 1003
	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,
R
Rich Felker 已提交
1004
		ehdr->e_phnum, ehdr->e_phentsize));
1005 1006
	decode_dyn(lib);

1007
	if (aux[AT_PHDR]) {
1008
		size_t interp_off = 0;
1009
		size_t tls_image = 0;
1010
		/* Find load address of the main program, via AT_PHDR vs PT_PHDR. */
R
Rich Felker 已提交
1011 1012
		app->phdr = phdr = (void *)aux[AT_PHDR];
		app->phnum = aux[AT_PHNUM];
1013 1014 1015
		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);
1016 1017
			else if (phdr->p_type == PT_INTERP)
				interp_off = (size_t)phdr->p_vaddr;
1018 1019 1020 1021 1022 1023
			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;
			}
1024
		}
1025
		if (app->tls_size) app->tls_image = (char *)app->base + tls_image;
1026
		if (interp_off) lib->name = (char *)app->base + interp_off;
1027 1028 1029 1030 1031
		if ((aux[0] & (1UL<<AT_EXECFN))
		    && strncmp((char *)aux[AT_EXECFN], "/proc/", 6))
			app->name = (char *)aux[AT_EXECFN];
		else
			app->name = argv[0];
1032
		app->kernel_mapped = 1;
1033 1034
		app->dynv = (void *)(app->base + find_dyn(
			(void *)aux[AT_PHDR], aux[AT_PHNUM], aux[AT_PHENT]));
1035 1036
		find_map_range((void *)aux[AT_PHDR],
			aux[AT_PHNUM], aux[AT_PHENT], app);
1037 1038 1039
	} else {
		int fd;
		char *ldname = argv[0];
R
Rich Felker 已提交
1040
		size_t l = strlen(ldname);
1041 1042 1043 1044
		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]) {
1045 1046 1047 1048 1049
			dprintf(2, "musl libc\n"
				"Version %s\n"
				"Dynamic Program Loader\n"
				"Usage: %s [--] pathname%s\n",
				__libc_get_version(), ldname,
1050 1051 1052 1053 1054 1055 1056 1057 1058
				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;
1059
		ehdr = (void *)map_library(fd, app);
1060 1061 1062 1063 1064 1065
		if (!ehdr) {
			dprintf(2, "%s: %s: Not a valid dynamic program\n", ldname, argv[0]);
			_exit(1);
		}
		runtime = 0;
		close(fd);
1066
		lib->name = ldname;
1067
		app->name = argv[0];
1068
		aux[AT_ENTRY] = (size_t)app->base + ehdr->e_entry;
1069 1070 1071 1072 1073 1074 1075 1076 1077 1078
		/* Find the name that would have been used for the dynamic
		 * linker had ldd not taken its place. */
		if (ldd_mode) {
			for (i=0; i<app->phnum; i++) {
				if (app->phdr[i].p_type == PT_INTERP)
					lib->name = (void *)(app->base
						+ app->phdr[i].p_vaddr);
			}
			dprintf(1, "\t%s (%p)\n", lib->name, lib->base);
		}
1079
	}
1080
	if (app->tls_size) {
R
Rich Felker 已提交
1081
		app->tls_id = tls_cnt = 1;
1082 1083 1084 1085 1086 1087
#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
1088 1089 1090
		tls_offset = app->tls_offset = app->tls_size
			+ ( -((uintptr_t)app->tls_image + app->tls_size)
			& (app->tls_align-1) );
1091
#endif
R
Rich Felker 已提交
1092
		tls_align = MAXP2(tls_align, app->tls_align);
1093
	}
1094 1095 1096 1097
	app->global = 1;
	decode_dyn(app);

	/* Attach to vdso, if provided by the kernel */
R
Rich Felker 已提交
1098
	if (search_vec(auxv, &vdso_base, AT_SYSINFO_EHDR)) {
1099
		ehdr = (void *)vdso_base;
R
Rich Felker 已提交
1100 1101
		vdso->phdr = phdr = (void *)(vdso_base + ehdr->e_phoff);
		vdso->phnum = ehdr->e_phnum;
1102 1103 1104 1105 1106 1107
		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);
		}
1108 1109
		vdso->name = "";
		vdso->shortname = "linux-gate.so.1";
1110
		vdso->global = 1;
1111
		decode_dyn(vdso);
1112 1113 1114 1115
		vdso->prev = lib;
		lib->next = vdso;
	}

1116 1117 1118 1119 1120
	/* 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;
1121
	ldso = lib;
1122 1123 1124
	app->next = lib;
	reloc_all(lib);
	app->next = 0;
R
Rich Felker 已提交
1125

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

1128
	/* Donate unused parts of app and library mapping to malloc */
R
Rich Felker 已提交
1129 1130 1131
	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),
1132 1133
		ehdr->e_phentsize, ehdr->e_phnum);

1134
	/* Load preload/needed libraries, add their symbols to the global
1135 1136 1137
	 * 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 已提交
1138
	if (env_preload) load_preload(env_preload);
1139 1140
	load_deps(app);
	make_global(app);
1141

R
Rich Felker 已提交
1142 1143 1144 1145
	reloc_all(app->next);
	reloc_all(app);

	update_tls_size();
1146
	if (tls_cnt) {
1147
		void *mem = mmap(0, libc.tls_size, PROT_READ|PROT_WRITE,
1148 1149
			MAP_ANONYMOUS|MAP_PRIVATE, -1, 0);
		if (mem==MAP_FAILED ||
1150
		    !__install_initial_tls(__copy_tls(mem))) {
1151
			dprintf(2, "%s: Error getting %zu bytes thread-local storage: %m\n",
1152
				argv[0], libc.tls_size);
1153 1154 1155 1156
			_exit(127);
		}
	}

1157
	if (ldso_fail) _exit(127);
1158 1159
	if (ldd_mode) _exit(0);

1160 1161 1162 1163
	/* 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 已提交
1164
	runtime = 1;
1165

1166
#ifndef DYNAMIC_IS_RO
1167 1168 1169
	for (i=0; app->dynv[i]; i+=2)
		if (app->dynv[i]==DT_DEBUG)
			app->dynv[i+1] = (size_t)&debug;
1170
#endif
1171 1172 1173 1174 1175 1176 1177
	debug.ver = 1;
	debug.bp = _dl_debug_state;
	debug.head = head;
	debug.base = lib->base;
	debug.state = 0;
	_dl_debug_state();

1178
	if (ssp_used) __init_ssp((void *)aux[AT_RANDOM]);
1179
	__init_libc(envp, argv[0]);
1180
	atexit(do_fini);
1181
	errno = 0;
1182
	do_init_fini(tail);
1183 1184

	return (void *)aux[AT_ENTRY];
1185 1186
}

1187 1188
void *dlopen(const char *file, int mode)
{
R
Rich Felker 已提交
1189
	struct dso *volatile p, *orig_tail, *next;
R
Rich Felker 已提交
1190
	size_t orig_tls_cnt, orig_tls_offset, orig_tls_align;
1191
	size_t i;
1192
	int cs;
1193
	jmp_buf jb;
1194 1195 1196

	if (!file) return head;

1197
	pthread_setcancelstate(PTHREAD_CANCEL_DISABLE, &cs);
1198
	pthread_rwlock_wrlock(&lock);
1199
	__inhibit_ptc();
1200

1201 1202
	p = 0;
	orig_tls_cnt = tls_cnt;
R
Rich Felker 已提交
1203 1204
	orig_tls_offset = tls_offset;
	orig_tls_align = tls_align;
R
Rich Felker 已提交
1205
	orig_tail = tail;
1206
	noload = mode & RTLD_NOLOAD;
R
Rich Felker 已提交
1207

1208 1209
	rtld_fail = &jb;
	if (setjmp(*rtld_fail)) {
1210
		/* Clean up anything new that was (partially) loaded */
1211
		if (p && p->deps) for (i=0; p->deps[i]; i++)
1212 1213
			if (p->deps[i]->global < 0)
				p->deps[i]->global = 0;
1214 1215 1216 1217 1218 1219
		for (p=orig_tail->next; p; p=next) {
			next = p->next;
			munmap(p->map, p->map_len);
			free(p->deps);
			free(p);
		}
1220
		tls_cnt = orig_tls_cnt;
R
Rich Felker 已提交
1221 1222
		tls_offset = orig_tls_offset;
		tls_align = orig_tls_align;
1223 1224
		tail = orig_tail;
		tail->next = 0;
1225
		p = 0;
1226 1227
		errflag = 1;
		goto end;
1228
	} else p = load_library(file, head);
R
Rich Felker 已提交
1229 1230

	if (!p) {
1231 1232 1233 1234
		snprintf(errbuf, sizeof errbuf, noload ?
			"Library %s is not already loaded" :
			"Error loading shared library %s: %m",
			file);
R
Rich Felker 已提交
1235
		errflag = 1;
1236
		goto end;
1237 1238 1239 1240 1241
	}

	/* First load handling */
	if (!p->deps) {
		load_deps(p);
R
Rich Felker 已提交
1242
		if (p->deps) for (i=0; p->deps[i]; i++)
1243 1244 1245
			if (!p->deps[i]->global)
				p->deps[i]->global = -1;
		if (!p->global) p->global = -1;
1246
		reloc_all(p);
R
Rich Felker 已提交
1247
		if (p->deps) for (i=0; p->deps[i]; i++)
1248 1249 1250
			if (p->deps[i]->global < 0)
				p->deps[i]->global = 0;
		if (p->global < 0) p->global = 0;
1251 1252 1253
	}

	if (mode & RTLD_GLOBAL) {
R
Rich Felker 已提交
1254
		if (p->deps) for (i=0; p->deps[i]; i++)
1255 1256 1257 1258
			p->deps[i]->global = 1;
		p->global = 1;
	}

R
Rich Felker 已提交
1259
	update_tls_size();
1260

1261
	if (ssp_used) __init_ssp(libc.auxv);
1262

1263
	_dl_debug_state();
1264
	orig_tail = tail;
1265
end:
1266
	__release_ptc();
R
Rich Felker 已提交
1267
	if (p) gencnt++;
1268
	pthread_rwlock_unlock(&lock);
1269
	if (p) do_init_fini(orig_tail);
1270
	pthread_setcancelstate(cs, 0);
1271 1272 1273
	return p;
}

1274
static int invalid_dso_handle(void *h)
1275 1276 1277 1278 1279 1280 1281 1282
{
	struct dso *p;
	for (p=head; p; p=p->next) if (h==p) return 0;
	snprintf(errbuf, sizeof errbuf, "Invalid library handle %p", (void *)h);
	errflag = 1;
	return 1;
}

R
Rich Felker 已提交
1283
static void *do_dlsym(struct dso *p, const char *s, void *ra)
1284 1285
{
	size_t i;
1286
	uint32_t h = 0, gh = 0;
1287
	Sym *sym;
1288
	if (p == head || p == RTLD_DEFAULT || p == RTLD_NEXT) {
1289 1290 1291
		if (p == RTLD_DEFAULT) {
			p = head;
		} else if (p == RTLD_NEXT) {
1292 1293
			for (p=head; p && (unsigned char *)ra-p->map>p->map_len; p=p->next);
			if (!p) p=head;
1294
			p = p->next;
1295
		}
1296
		struct symdef def = find_sym(p, s, 0);
1297
		if (!def.sym) goto failed;
1298 1299
		if ((def.sym->st_info&0xf) == STT_TLS)
			return __tls_get_addr((size_t []){def.dso->tls_id, def.sym->st_value});
1300
		return def.dso->base + def.sym->st_value;
R
Rich Felker 已提交
1301
	}
1302 1303
	if (p != RTLD_DEFAULT && p != RTLD_NEXT && invalid_dso_handle(p))
		return 0;
1304 1305 1306 1307 1308 1309 1310
	if (p->ghashtab) {
		gh = gnu_hash(s);
		sym = gnu_lookup(s, gh, p);
	} else {
		h = sysv_hash(s);
		sym = sysv_lookup(s, h, p);
	}
1311 1312
	if (sym && (sym->st_info&0xf) == STT_TLS)
		return __tls_get_addr((size_t []){p->tls_id, sym->st_value});
1313 1314 1315
	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++) {
1316 1317
		if (p->deps[i]->ghashtab) {
			if (!gh) gh = gnu_hash(s);
1318
			sym = gnu_lookup(s, gh, p->deps[i]);
1319 1320 1321 1322
		} else {
			if (!h) h = sysv_hash(s);
			sym = sysv_lookup(s, h, p->deps[i]);
		}
1323 1324
		if (sym && (sym->st_info&0xf) == STT_TLS)
			return __tls_get_addr((size_t []){p->deps[i]->tls_id, sym->st_value});
1325 1326 1327
		if (sym && sym->st_value && (1<<(sym->st_info&0xf) & OK_TYPES))
			return p->deps[i]->base + sym->st_value;
	}
1328
failed:
R
Rich Felker 已提交
1329
	errflag = 1;
1330
	snprintf(errbuf, sizeof errbuf, "Symbol not found: %s", s);
1331 1332 1333
	return 0;
}

1334 1335 1336 1337 1338 1339 1340 1341 1342 1343 1344 1345 1346 1347 1348 1349 1350 1351 1352 1353 1354 1355 1356 1357 1358
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];
R
Rich Felker 已提交
1359
		for (i = nsym = 0; i < p->ghashtab[0]; i++) {
1360 1361 1362 1363 1364 1365 1366 1367 1368 1369 1370 1371
			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++) {
1372
		if (sym->st_value
1373 1374 1375 1376 1377 1378 1379 1380 1381 1382 1383 1384 1385 1386 1387 1388 1389 1390 1391 1392 1393 1394
		 && (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;
}

1395
void *__dlsym(void *restrict p, const char *restrict s, void *restrict ra)
1396 1397 1398
{
	void *res;
	pthread_rwlock_rdlock(&lock);
R
Rich Felker 已提交
1399
	res = do_dlsym(p, s, ra);
1400 1401 1402
	pthread_rwlock_unlock(&lock);
	return res;
}
R
Rich Felker 已提交
1403 1404 1405 1406 1407 1408 1409 1410 1411 1412 1413 1414 1415 1416 1417 1418 1419 1420 1421 1422 1423 1424 1425 1426 1427 1428

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;
}
1429
#else
1430
static int invalid_dso_handle(void *h)
1431 1432 1433 1434 1435
{
	snprintf(errbuf, sizeof errbuf, "Invalid library handle %p", (void *)h);
	errflag = 1;
	return 1;
}
1436 1437 1438 1439
void *dlopen(const char *file, int mode)
{
	return 0;
}
1440
void *__dlsym(void *restrict p, const char *restrict s, void *restrict ra)
1441 1442 1443
{
	return 0;
}
1444 1445 1446 1447
int __dladdr (void *addr, Dl_info *info)
{
	return 0;
}
1448
#endif
1449

R
Rich Felker 已提交
1450 1451 1452 1453 1454 1455 1456 1457 1458 1459 1460 1461
int __dlinfo(void *dso, int req, void *res)
{
	if (invalid_dso_handle(dso)) return -1;
	if (req != RTLD_DI_LINKMAP) {
		snprintf(errbuf, sizeof errbuf, "Unsupported request %d", req);
		errflag = 1;
		return -1;
	}
	*(struct link_map **)res = dso;
	return 0;
}

1462 1463
char *dlerror()
{
R
Rich Felker 已提交
1464 1465
	if (!errflag) return 0;
	errflag = 0;
1466
	return errbuf;
1467 1468 1469 1470
}

int dlclose(void *p)
{
1471
	return invalid_dso_handle(p);
1472
}