dynlink.c 34.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;
102
static int noload;
R
Rich Felker 已提交
103
static jmp_buf rtld_fail;
104
static pthread_rwlock_t lock;
105
static struct debug debug;
106
static size_t tls_cnt, tls_offset, tls_align = 4*sizeof(size_t);
107
static pthread_mutex_t init_fini_lock = { ._m_type = PTHREAD_MUTEX_RECURSIVE };
108 109

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

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

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

142 143 144 145 146 147 148 149 150 151
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 已提交
152 153
{
	size_t i;
R
Rich Felker 已提交
154 155 156
	Sym *syms = dso->syms;
	uint32_t *hashtab = dso->hashtab;
	char *strings = dso->strings;
R
Rich Felker 已提交
157 158 159 160 161 162 163
	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;
}

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 190
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;
}

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

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

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

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

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

	ssize_t l = read(fd, buf, sizeof buf);
	if (l<sizeof *eh) return 0;
321
	eh = buf;
R
Rich Felker 已提交
322 323 324
	phsize = eh->e_phentsize * eh->e_phnum;
	if (phsize + sizeof *eh > l) return 0;
	if (eh->e_phoff + phsize > l) {
325
		l = pread(fd, buf+1, phsize, eh->e_phoff);
R
Rich Felker 已提交
326
		if (l != phsize) return 0;
327 328 329
		ph = ph0 = (void *)(buf + 1);
	} else {
		ph = ph0 = (void *)((char *)buf + eh->e_phoff);
R
Rich Felker 已提交
330 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
	if (map==MAP_FAILED) return 0;
	base = map - addr_min;
365 366 367
	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 已提交
368
		if (ph->p_type != PT_LOAD) continue;
369 370 371 372 373 374 375 376
		/* 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 已提交
377 378 379 380 381 382 383 384
		/* 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));
385 386
		if (mmap(base+this_min, this_max-this_min, prot, MAP_PRIVATE|MAP_FIXED, fd, off_start) == MAP_FAILED)
			goto error;
R
Rich Felker 已提交
387 388 389 390
		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);
391 392
			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 已提交
393 394
		}
	}
R
Rich Felker 已提交
395 396
	for (i=0; ((size_t *)(base+dyn))[i]; i+=2)
		if (((size_t *)(base+dyn))[i]==DT_TEXTREL) {
397 398
			if (mprotect(map, map_len, PROT_READ|PROT_WRITE|PROT_EXEC) < 0)
				goto error;
R
Rich Felker 已提交
399 400
			break;
		}
401
	if (!runtime) reclaim_gaps(base, ph0, eh->e_phentsize, eh->e_phnum);
402 403 404 405 406
	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 已提交
407
	return map;
408 409 410
error:
	munmap(map, map_len);
	return 0;
R
Rich Felker 已提交
411 412
}

413
static int path_open(const char *name, const char *s, char *buf, size_t buf_size)
414
{
415 416
	size_t l;
	int fd;
417
	for (;;) {
418 419 420 421 422
		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;
423
		if ((fd = open(buf, O_RDONLY|O_CLOEXEC))>=0) return fd;
424
		s += l;
425 426 427
	}
}

428 429 430 431 432 433
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]);
434 435 436 437
	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);
438 439
}

R
Rich Felker 已提交
440 441
static struct dso *load_library(const char *name)
{
442
	char buf[2*NAME_MAX+2];
443
	const char *pathname;
444
	unsigned char *map;
445
	struct dso *p, temp_dso = {0};
R
Rich Felker 已提交
446 447
	int fd;
	struct stat st;
448 449
	size_t alloc_size;
	int n_th = 0;
R
Rich Felker 已提交
450 451 452 453 454 455 456 457 458 459

	/* 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) {
460 461 462 463
				if (!ldso->prev) {
					tail->next = ldso;
					ldso->prev = tail;
					tail = ldso->next ? ldso->next : ldso;
R
Rich Felker 已提交
464
				}
465
				return ldso;
R
Rich Felker 已提交
466 467 468
			}
		}
	}
469
	if (strchr(name, '/')) {
470
		pathname = name;
471
		fd = open(name, O_RDONLY|O_CLOEXEC);
R
Rich Felker 已提交
472
	} else {
473 474 475 476 477 478 479
		/* 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;
			}
		}
480
		if (strlen(name) > NAME_MAX) return 0;
481
		fd = -1;
482 483
		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);
484 485
		if (fd < 0) {
			if (!sys_path) {
486 487 488 489 490 491 492 493 494 495 496 497 498 499 500 501 502 503 504 505
				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");
506
				if (f) {
507
					if (getdelim(&sys_path, (size_t[1]){0}, 0, f) <= 0) {
508
						free(sys_path);
509
						sys_path = "";
510
					}
511 512 513
					fclose(f);
				}
			}
514 515
			if (!sys_path) sys_path = "/lib:/usr/local/lib:/usr/lib";
			fd = path_open(name, sys_path, buf, sizeof buf);
R
Rich Felker 已提交
516
		}
517
		pathname = buf;
R
Rich Felker 已提交
518 519 520 521 522 523 524 525
	}
	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) {
526 527 528
			/* 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. */
529 530
			if (!p->shortname && pathname != name)
				p->shortname = strrchr(p->name, '/')+1;
R
Rich Felker 已提交
531 532 533 534 535
			close(fd);
			p->refcnt++;
			return p;
		}
	}
536
	map = noload ? 0 : map_library(fd, &temp_dso);
R
Rich Felker 已提交
537 538
	close(fd);
	if (!map) return 0;
539 540 541 542 543 544 545 546 547 548

	/* 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);
549
		n_th = libc.threads_minus_1 + 1;
550 551 552 553
		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 已提交
554
	if (!p) {
555
		munmap(map, temp_dso.map_len);
R
Rich Felker 已提交
556 557
		return 0;
	}
558
	memcpy(p, &temp_dso, sizeof temp_dso);
559
	decode_dyn(p);
R
Rich Felker 已提交
560 561 562
	p->dev = st.st_dev;
	p->ino = st.st_ino;
	p->refcnt = 1;
R
Rich Felker 已提交
563
	p->name = p->buf;
564 565 566
	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;
567
	if (p->tls_image) {
568
		if (runtime && !__pthread_self_init()) {
569
			munmap(map, p->map_len);
570 571 572
			free(p);
			return 0;
		}
573
		p->tls_id = ++tls_cnt;
R
Rich Felker 已提交
574
		tls_align = MAXP2(tls_align, p->tls_align);
575 576 577 578 579
#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 已提交
580 581 582 583
		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;
584
#endif
585 586 587 588
		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 已提交
589 590 591 592 593

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

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

R
Rich Felker 已提交
596 597 598 599 600
	return p;
}

static void load_deps(struct dso *p)
{
601 602
	size_t i, ndeps=0;
	struct dso ***deps = &p->deps, **tmp, *dep;
R
Rich Felker 已提交
603
	for (; p; p=p->next) {
604 605 606 607
		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 已提交
608 609
		for (i=0; p->dynv[i]; i+=2) {
			if (p->dynv[i] != DT_NEEDED) continue;
610 611
			dep = load_library(p->strings + p->dynv[i+1]);
			if (!dep) {
612 613
				snprintf(errbuf, sizeof errbuf,
					"Error loading shared library %s: %m (needed by %s)",
R
Rich Felker 已提交
614
					p->strings + p->dynv[i+1], p->name);
615 616
				if (runtime) longjmp(rtld_fail, 1);
				dprintf(2, "%s\n", errbuf);
617 618
				ldso_fail = 1;
				continue;
R
Rich Felker 已提交
619
			}
620 621 622 623 624 625 626
			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 已提交
627
		}
628
		r_path = 0;
R
Rich Felker 已提交
629 630 631
	}
}

R
Rich Felker 已提交
632 633 634 635 636 637 638 639 640 641 642 643 644 645
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;
	}
}

646 647 648 649 650
static void make_global(struct dso *p)
{
	for (; p; p=p->next) p->global = 1;
}

R
Rich Felker 已提交
651 652 653 654 655 656
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 已提交
657 658 659
#ifdef NEED_ARCH_RELOCS
		do_arch_relocs(p, head);
#endif
660 661 662 663
		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);
664
		p->relocated = 1;
R
Rich Felker 已提交
665 666 667
	}
}

668 669 670 671 672 673 674 675
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;
}

676 677 678 679 680 681 682 683 684 685 686 687 688 689 690 691
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;
}

692 693 694 695 696 697 698
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);
699 700
		if (dyn[0] & (1<<DT_FINI_ARRAY)) {
			size_t n = dyn[DT_FINI_ARRAYSZ]/sizeof(size_t);
701 702
			size_t *fn = (size_t *)(p->base + dyn[DT_FINI_ARRAY])+n;
			while (n--) ((void (*)(void))*--fn)();
703 704 705
		}
		if (dyn[0] & (1<<DT_FINI))
			((void (*)(void))(p->base + dyn[DT_FINI]))();
706 707 708
	}
}

709 710 711
static void do_init_fini(struct dso *p)
{
	size_t dyn[DYN_CNT] = {0};
712
	int need_locking = libc.threads_minus_1;
713 714 715 716
	/* 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);
717
	for (; p; p=p->prev) {
718 719
		if (p->constructed) continue;
		p->constructed = 1;
720
		decode_vec(p->dynv, dyn, DYN_CNT);
721
		if (dyn[0] & ((1<<DT_FINI) | (1<<DT_FINI_ARRAY))) {
722 723 724
			p->fini_next = fini_head;
			fini_head = p;
		}
725 726
		if (dyn[0] & (1<<DT_INIT))
			((void (*)(void))(p->base + dyn[DT_INIT]))();
727 728 729 730 731
		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++)();
		}
732 733 734 735
		if (!need_locking && libc.threads_minus_1) {
			need_locking = 1;
			pthread_mutex_lock(&init_fini_lock);
		}
736
	}
737
	if (need_locking) pthread_mutex_unlock(&init_fini_lock);
738 739
}

740 741 742 743
void _dl_debug_state(void)
{
}

744
void *__copy_tls(unsigned char *mem)
745
{
R
Rich Felker 已提交
746
	pthread_t td;
747
	struct dso *p;
R
Rich Felker 已提交
748 749 750

	if (!tls_cnt) return mem;

751
	void **dtv = (void *)mem;
752
	dtv[0] = (void *)tls_cnt;
R
Rich Felker 已提交
753

754 755 756 757 758 759 760 761 762 763 764 765
#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
766
	mem += libc.tls_size - sizeof(struct pthread);
R
Rich Felker 已提交
767 768 769 770
	mem -= (uintptr_t)mem & (tls_align-1);
	td = (pthread_t)mem;

	for (p=head; p; p=p->next) {
771
		if (!p->tls_id) continue;
R
Rich Felker 已提交
772 773
		dtv[p->tls_id] = mem - p->tls_offset;
		memcpy(dtv[p->tls_id], p->tls_image, p->tls_len);
774
	}
775
#endif
R
Rich Felker 已提交
776 777
	td->dtv = dtv;
	return td;
778 779
}

780
void *__tls_get_addr(size_t *v)
781 782
{
	pthread_t self = __pthread_self();
783
	if (v[0]<=(size_t)self->dtv[0] && self->dtv[v[0]])
784 785 786 787
		return (char *)self->dtv[v[0]]+v[1];

	/* Block signals to make accessing new TLS async-signal-safe */
	sigset_t set;
788
	pthread_sigmask(SIG_BLOCK, SIGALL_SET, &set);
789
	if (v[0]<=(size_t)self->dtv[0] && self->dtv[v[0]]) {
790 791 792 793 794 795 796 797 798 799 800 801
		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 */
802
	if (v[0] > (size_t)self->dtv[0]) {
803 804
		void **newdtv = p->new_dtv +
			(v[0]+1)*sizeof(void *)*a_fetch_add(&p->new_dtv_idx,1);
805
		memcpy(newdtv, self->dtv,
806 807 808
			((size_t)self->dtv[0]+1) * sizeof(void *));
		newdtv[0] = (void *)v[0];
		self->dtv = newdtv;
809
	}
810 811 812 813 814 815

	/* 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 已提交
816
	memcpy(mem, p->tls_image, p->tls_len);
817 818
	pthread_sigmask(SIG_SETMASK, &set, 0);
	return mem + v[1];
819 820
}

R
Rich Felker 已提交
821 822
static void update_tls_size()
{
823 824 825 826 827 828
	libc.tls_size = ALIGN(
		(1+tls_cnt) * sizeof(void *) +
		tls_offset +
		sizeof(struct pthread) +
		tls_align * 2,
	tls_align);
R
Rich Felker 已提交
829 830
}

831
void *__dynlink(int argc, char **argv)
R
Rich Felker 已提交
832
{
833
	size_t aux[AUX_CNT] = {0};
R
Rich Felker 已提交
834 835
	size_t i;
	Phdr *phdr;
836
	Ehdr *ehdr;
837
	static struct dso builtin_dsos[3];
R
Rich Felker 已提交
838 839
	struct dso *const app = builtin_dsos+0;
	struct dso *const lib = builtin_dsos+1;
840
	struct dso *const vdso = builtin_dsos+2;
R
Rich Felker 已提交
841
	char *env_preload=0;
R
Rich Felker 已提交
842
	size_t vdso_base;
843
	size_t *auxv;
R
Rich Felker 已提交
844 845

	/* Find aux vector just past environ[] */
846 847 848
	for (i=argc+1; argv[i]; i++)
		if (!memcmp(argv[i], "LD_LIBRARY_PATH=", 16))
			env_path = argv[i]+16;
R
Rich Felker 已提交
849 850
		else if (!memcmp(argv[i], "LD_PRELOAD=", 11))
			env_preload = argv[i]+11;
R
Rich Felker 已提交
851 852 853 854
	auxv = (void *)(argv+i+1);

	decode_vec(auxv, aux, AUX_CNT);

855 856
	/* 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 已提交
857
	  || aux[AT_GID]!=aux[AT_EGID] || aux[AT_SECURE]) {
858
		env_path = 0;
R
Rich Felker 已提交
859
		env_preload = 0;
860 861
	}

862 863 864 865 866 867 868 869 870
	/* 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;
	}

871 872 873
	/* The dynamic linker load address is passed by the kernel
	 * in the AUX vector, so this is easy. */
	lib->base = (void *)aux[AT_BASE];
874
	lib->name = lib->shortname = "libc.so";
875 876
	lib->global = 1;
	ehdr = (void *)lib->base;
R
Rich Felker 已提交
877 878 879 880 881
	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));
882 883
	decode_dyn(lib);

884
	if (aux[AT_PHDR]) {
885
		size_t interp_off = 0;
886
		size_t tls_image = 0;
887
		/* Find load address of the main program, via AT_PHDR vs PT_PHDR. */
R
Rich Felker 已提交
888 889
		app->phdr = phdr = (void *)aux[AT_PHDR];
		app->phnum = aux[AT_PHNUM];
890 891 892
		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);
893 894
			else if (phdr->p_type == PT_INTERP)
				interp_off = (size_t)phdr->p_vaddr;
895 896 897 898 899 900
			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;
			}
901
		}
902
		if (app->tls_size) app->tls_image = (char *)app->base + tls_image;
903
		if (interp_off) lib->name = (char *)app->base + interp_off;
904
		app->name = argv[0];
905 906
		app->dynv = (void *)(app->base + find_dyn(
			(void *)aux[AT_PHDR], aux[AT_PHNUM], aux[AT_PHENT]));
907 908
		find_map_range((void *)aux[AT_PHDR],
			aux[AT_PHNUM], aux[AT_PHENT], app);
909 910 911
	} else {
		int fd;
		char *ldname = argv[0];
R
Rich Felker 已提交
912
		size_t l = strlen(ldname);
913 914 915 916 917 918 919 920 921 922 923 924 925 926 927
		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;
928
		ehdr = (void *)map_library(fd, app);
929 930 931 932 933 934
		if (!ehdr) {
			dprintf(2, "%s: %s: Not a valid dynamic program\n", ldname, argv[0]);
			_exit(1);
		}
		runtime = 0;
		close(fd);
935
		lib->name = ldname;
936
		app->name = argv[0];
R
Rich Felker 已提交
937 938
		app->phnum = ehdr->e_phnum;
		app->phdr = (void *)(app->base + ehdr->e_phoff);
939
		aux[AT_ENTRY] = ehdr->e_entry;
940
	}
941
	if (app->tls_size) {
R
Rich Felker 已提交
942
		app->tls_id = tls_cnt = 1;
943 944 945 946 947 948
#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
949 950 951
		tls_offset = app->tls_offset = app->tls_size
			+ ( -((uintptr_t)app->tls_image + app->tls_size)
			& (app->tls_align-1) );
952
#endif
R
Rich Felker 已提交
953
		tls_align = MAXP2(tls_align, app->tls_align);
954
	}
955
	app->global = 1;
956
	app->constructed = 1;
957 958 959
	decode_dyn(app);

	/* Attach to vdso, if provided by the kernel */
R
Rich Felker 已提交
960
	if (search_vec(auxv, &vdso_base, AT_SYSINFO_EHDR)) {
961
		ehdr = (void *)vdso_base;
R
Rich Felker 已提交
962 963
		vdso->phdr = phdr = (void *)(vdso_base + ehdr->e_phoff);
		vdso->phnum = ehdr->e_phnum;
964 965 966 967 968 969
		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);
		}
970 971
		vdso->name = "";
		vdso->shortname = "linux-gate.so.1";
972
		vdso->global = 1;
973
		decode_dyn(vdso);
974 975 976 977
		vdso->prev = lib;
		lib->next = vdso;
	}

978 979 980 981 982
	/* 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;
983
	ldso = lib;
984 985 986
	app->next = lib;
	reloc_all(lib);
	app->next = 0;
R
Rich Felker 已提交
987

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

990
	/* Donate unused parts of app and library mapping to malloc */
R
Rich Felker 已提交
991 992 993
	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),
994 995
		ehdr->e_phentsize, ehdr->e_phnum);

996
	/* Load preload/needed libraries, add their symbols to the global
997 998 999
	 * 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 已提交
1000
	if (env_preload) load_preload(env_preload);
1001 1002
	load_deps(app);
	make_global(app);
1003

R
Rich Felker 已提交
1004 1005 1006 1007
	reloc_all(app->next);
	reloc_all(app);

	update_tls_size();
1008
	if (tls_cnt) {
1009
		void *mem = mmap(0, libc.tls_size, PROT_READ|PROT_WRITE,
1010 1011
			MAP_ANONYMOUS|MAP_PRIVATE, -1, 0);
		if (mem==MAP_FAILED ||
1012
		    !__install_initial_tls(__copy_tls(mem))) {
1013
			dprintf(2, "%s: Error getting %zu bytes thread-local storage: %m\n",
1014
				argv[0], libc.tls_size);
1015 1016 1017 1018
			_exit(127);
		}
	}

1019
	if (ldso_fail) _exit(127);
1020 1021
	if (ldd_mode) _exit(0);

1022 1023 1024 1025
	/* 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 已提交
1026
	runtime = 1;
1027

1028
#ifndef DYNAMIC_IS_RO
1029 1030 1031
	for (i=0; app->dynv[i]; i+=2)
		if (app->dynv[i]==DT_DEBUG)
			app->dynv[i+1] = (size_t)&debug;
1032
#endif
1033 1034 1035 1036 1037 1038 1039
	debug.ver = 1;
	debug.bp = _dl_debug_state;
	debug.head = head;
	debug.base = lib->base;
	debug.state = 0;
	_dl_debug_state();

1040
	if (ssp_used) __init_ssp((void *)aux[AT_RANDOM]);
1041

R
Rich Felker 已提交
1042 1043 1044
	errno = 0;
	return (void *)aux[AT_ENTRY];
}
1045

1046 1047 1048 1049 1050 1051
void __init_ldso_ctors(void)
{
	atexit(do_fini);
	do_init_fini(tail);
}

1052 1053
void *dlopen(const char *file, int mode)
{
R
Rich Felker 已提交
1054
	struct dso *volatile p, *orig_tail, *next;
R
Rich Felker 已提交
1055
	size_t orig_tls_cnt, orig_tls_offset, orig_tls_align;
1056
	size_t i;
1057
	int cs;
1058 1059 1060

	if (!file) return head;

1061
	pthread_setcancelstate(PTHREAD_CANCEL_DISABLE, &cs);
1062
	pthread_rwlock_wrlock(&lock);
1063
	__inhibit_ptc();
1064

1065 1066
	p = 0;
	orig_tls_cnt = tls_cnt;
R
Rich Felker 已提交
1067 1068
	orig_tls_offset = tls_offset;
	orig_tls_align = tls_align;
R
Rich Felker 已提交
1069
	orig_tail = tail;
1070
	noload = mode & RTLD_NOLOAD;
R
Rich Felker 已提交
1071

1072 1073
	if (setjmp(rtld_fail)) {
		/* Clean up anything new that was (partially) loaded */
1074
		if (p && p->deps) for (i=0; p->deps[i]; i++)
1075 1076
			if (p->deps[i]->global < 0)
				p->deps[i]->global = 0;
1077 1078 1079 1080 1081 1082
		for (p=orig_tail->next; p; p=next) {
			next = p->next;
			munmap(p->map, p->map_len);
			free(p->deps);
			free(p);
		}
1083
		tls_cnt = orig_tls_cnt;
R
Rich Felker 已提交
1084 1085
		tls_offset = orig_tls_offset;
		tls_align = orig_tls_align;
1086 1087
		tail = orig_tail;
		tail->next = 0;
1088
		p = 0;
1089 1090
		errflag = 1;
		goto end;
R
Rich Felker 已提交
1091 1092 1093
	} else p = load_library(file);

	if (!p) {
1094 1095 1096 1097
		snprintf(errbuf, sizeof errbuf, noload ?
			"Library %s is not already loaded" :
			"Error loading shared library %s: %m",
			file);
R
Rich Felker 已提交
1098
		errflag = 1;
1099
		goto end;
1100 1101 1102 1103 1104
	}

	/* First load handling */
	if (!p->deps) {
		load_deps(p);
R
Rich Felker 已提交
1105
		if (p->deps) for (i=0; p->deps[i]; i++)
1106 1107 1108
			if (!p->deps[i]->global)
				p->deps[i]->global = -1;
		if (!p->global) p->global = -1;
1109
		reloc_all(p);
R
Rich Felker 已提交
1110
		if (p->deps) for (i=0; p->deps[i]; i++)
1111 1112 1113
			if (p->deps[i]->global < 0)
				p->deps[i]->global = 0;
		if (p->global < 0) p->global = 0;
1114 1115 1116
	}

	if (mode & RTLD_GLOBAL) {
R
Rich Felker 已提交
1117
		if (p->deps) for (i=0; p->deps[i]; i++)
1118 1119 1120 1121
			p->deps[i]->global = 1;
		p->global = 1;
	}

R
Rich Felker 已提交
1122
	update_tls_size();
1123

1124
	if (ssp_used) __init_ssp(libc.auxv);
1125

1126
	_dl_debug_state();
1127
	orig_tail = tail;
1128
end:
1129
	__release_ptc();
R
Rich Felker 已提交
1130
	if (p) gencnt++;
1131
	pthread_rwlock_unlock(&lock);
1132
	if (p) do_init_fini(orig_tail);
1133
	pthread_setcancelstate(cs, 0);
1134 1135 1136
	return p;
}

1137
static int invalid_dso_handle(void *h)
1138 1139 1140 1141 1142 1143 1144 1145
{
	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 已提交
1146
static void *do_dlsym(struct dso *p, const char *s, void *ra)
1147 1148
{
	size_t i;
1149
	uint32_t h = 0, gh = 0;
1150
	Sym *sym;
1151
	if (p == head || p == RTLD_DEFAULT || p == RTLD_NEXT) {
1152 1153 1154
		if (p == RTLD_DEFAULT) {
			p = head;
		} else if (p == RTLD_NEXT) {
1155 1156
			for (p=head; p && (unsigned char *)ra-p->map>p->map_len; p=p->next);
			if (!p) p=head;
1157
			p = p->next;
1158
		}
1159
		struct symdef def = find_sym(p, s, 0);
1160
		if (!def.sym) goto failed;
1161 1162
		if ((def.sym->st_info&0xf) == STT_TLS)
			return __tls_get_addr((size_t []){def.dso->tls_id, def.sym->st_value});
1163
		return def.dso->base + def.sym->st_value;
R
Rich Felker 已提交
1164
	}
1165 1166
	if (p != RTLD_DEFAULT && p != RTLD_NEXT && invalid_dso_handle(p))
		return 0;
1167 1168 1169 1170 1171 1172 1173
	if (p->ghashtab) {
		gh = gnu_hash(s);
		sym = gnu_lookup(s, gh, p);
	} else {
		h = sysv_hash(s);
		sym = sysv_lookup(s, h, p);
	}
1174 1175
	if (sym && (sym->st_info&0xf) == STT_TLS)
		return __tls_get_addr((size_t []){p->tls_id, sym->st_value});
1176 1177 1178
	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++) {
1179 1180
		if (p->deps[i]->ghashtab) {
			if (!gh) gh = gnu_hash(s);
1181
			sym = gnu_lookup(s, gh, p->deps[i]);
1182 1183 1184 1185
		} else {
			if (!h) h = sysv_hash(s);
			sym = sysv_lookup(s, h, p->deps[i]);
		}
1186 1187
		if (sym && (sym->st_info&0xf) == STT_TLS)
			return __tls_get_addr((size_t []){p->deps[i]->tls_id, sym->st_value});
1188 1189 1190
		if (sym && sym->st_value && (1<<(sym->st_info&0xf) & OK_TYPES))
			return p->deps[i]->base + sym->st_value;
	}
1191
failed:
R
Rich Felker 已提交
1192
	errflag = 1;
1193
	snprintf(errbuf, sizeof errbuf, "Symbol not found: %s", s);
1194 1195 1196
	return 0;
}

1197 1198 1199 1200 1201 1202 1203 1204 1205 1206 1207 1208 1209 1210 1211 1212 1213 1214 1215 1216 1217 1218 1219 1220 1221 1222 1223 1224 1225 1226 1227 1228 1229 1230 1231 1232 1233 1234
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++) {
1235
		if (sym->st_value
1236 1237 1238 1239 1240 1241 1242 1243 1244 1245 1246 1247 1248 1249 1250 1251 1252 1253 1254 1255 1256 1257
		 && (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;
}

1258
void *__dlsym(void *restrict p, const char *restrict s, void *restrict ra)
1259 1260 1261
{
	void *res;
	pthread_rwlock_rdlock(&lock);
R
Rich Felker 已提交
1262
	res = do_dlsym(p, s, ra);
1263 1264 1265
	pthread_rwlock_unlock(&lock);
	return res;
}
R
Rich Felker 已提交
1266 1267 1268 1269 1270 1271 1272 1273 1274 1275 1276 1277 1278 1279 1280 1281 1282 1283 1284 1285 1286 1287 1288 1289 1290 1291

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;
}
1292
#else
1293
static int invalid_dso_handle(void *h)
1294 1295 1296 1297 1298
{
	snprintf(errbuf, sizeof errbuf, "Invalid library handle %p", (void *)h);
	errflag = 1;
	return 1;
}
1299 1300 1301 1302
void *dlopen(const char *file, int mode)
{
	return 0;
}
1303
void *__dlsym(void *restrict p, const char *restrict s, void *restrict ra)
1304 1305 1306
{
	return 0;
}
1307 1308 1309 1310
int __dladdr (void *addr, Dl_info *info)
{
	return 0;
}
1311
#endif
1312

R
Rich Felker 已提交
1313 1314 1315 1316 1317 1318 1319 1320 1321 1322 1323 1324
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;
}

1325 1326
char *dlerror()
{
R
Rich Felker 已提交
1327 1328
	if (!errflag) return 0;
	errflag = 0;
1329
	return errbuf;
1330 1331 1332 1333
}

int dlclose(void *p)
{
1334
	return invalid_dso_handle(p);
1335
}