dynlink.c 45.5 KB
Newer Older
1
#define _GNU_SOURCE
R
Rich Felker 已提交
2 3
#include <stdio.h>
#include <stdlib.h>
4
#include <stdarg.h>
5
#include <stddef.h>
R
Rich Felker 已提交
6 7 8 9 10 11 12 13 14
#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 已提交
15
#include <link.h>
R
Rich Felker 已提交
16
#include <setjmp.h>
17
#include <pthread.h>
R
Rich Felker 已提交
18
#include <ctype.h>
19
#include <dlfcn.h>
20 21
#include "pthread_impl.h"
#include "libc.h"
R
Rich Felker 已提交
22
#include "dynlink.h"
R
Rich Felker 已提交
23

24
static void error(const char *, ...);
R
Rich Felker 已提交
25

26
#ifdef SHARED
R
Rich Felker 已提交
27

R
Rich Felker 已提交
28 29 30
#define MAXP2(a,b) (-(-(a)&-(b)))
#define ALIGN(x,y) ((x)+(y)-1 & -(y))

31 32 33 34 35 36 37 38
struct debug {
	int ver;
	void *head;
	void (*bp)(void);
	int state;
	void *base;
};

39 40 41 42 43
struct td_index {
	size_t args[2];
	struct td_index *next;
};

44 45 46 47
struct dso {
	unsigned char *base;
	char *name;
	size_t *dynv;
R
Rich Felker 已提交
48
	struct dso *next, *prev;
49

R
Rich Felker 已提交
50 51
	Phdr *phdr;
	int phnum;
52
	size_t phentsize;
R
Rich Felker 已提交
53 54
	int refcnt;
	Sym *syms;
55
	uint32_t *hashtab;
56
	uint32_t *ghashtab;
57
	int16_t *versym;
R
Rich Felker 已提交
58 59 60 61 62
	char *strings;
	unsigned char *map;
	size_t map_len;
	dev_t dev;
	ino_t ino;
63
	signed char global;
64 65
	char relocated;
	char constructed;
66
	char kernel_mapped;
67
	struct dso **deps, *needed_by;
68
	char *rpath_orig, *rpath;
69
	void *tls_image;
70
	size_t tls_len, tls_size, tls_align, tls_id, tls_offset;
T
Timo Teräs 已提交
71
	size_t relro_start, relro_end;
72 73
	void **new_dtv;
	unsigned char *new_tls;
74
	volatile int new_dtv_idx, new_tls_idx;
75
	struct td_index *td_index;
76
	struct dso *fini_next;
77
	char *shortname;
R
Rich Felker 已提交
78
	char buf[];
R
Rich Felker 已提交
79 80
};

81 82 83 84 85
struct symdef {
	Sym *sym;
	struct dso *dso;
};

86
int __init_tp(void *);
87
void __init_libc(char **, char *);
88

89 90
const char *__libc_get_version(void);

91 92 93 94 95 96 97
static struct builtin_tls {
	char c;
	struct pthread pt;
	void *space[16];
} builtin_tls[1];
#define MIN_TLS_ALIGN offsetof(struct builtin_tls, pt)

98 99 100
#define ADDEND_LIMIT 4096
static size_t *saved_addends, *apply_addends_to;

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

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

R
Rich Felker 已提交
118 119 120 121 122 123
static int dl_strcmp(const char *l, const char *r)
{
	for (; *l==*r && *l; l++, r++);
	return *(unsigned char *)l - *(unsigned char *)r;
}
#define strcmp(l,r) dl_strcmp(l,r)
R
Rich Felker 已提交
124 125 126

static void decode_vec(size_t *v, size_t *a, size_t cnt)
{
R
Rich Felker 已提交
127 128 129 130
	size_t i;
	for (i=0; i<cnt; i++) a[i] = 0;
	for (; v[0]; v+=2) if (v[0]-1<cnt-1) {
		a[0] |= 1UL<<v[0];
R
Rich Felker 已提交
131 132 133 134
		a[v[0]] = v[1];
	}
}

135 136 137 138 139 140 141 142 143
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 已提交
144
{
145
	const unsigned char *s = (void *)s0;
R
Rich Felker 已提交
146 147 148 149 150 151 152 153
	uint_fast32_t h = 0;
	while (*s) {
		h = 16*h + *s++;
		h ^= h>>24 & 0xf0;
	}
	return h & 0xfffffff;
}

154 155 156 157 158 159 160 161 162 163
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 已提交
164 165
{
	size_t i;
R
Rich Felker 已提交
166 167 168
	Sym *syms = dso->syms;
	uint32_t *hashtab = dso->hashtab;
	char *strings = dso->strings;
R
Rich Felker 已提交
169
	for (i=hashtab[2+h%hashtab[0]]; i; i=hashtab[2+hashtab[0]+i]) {
170 171
		if ((!dso->versym || dso->versym[i] >= 0)
		    && (!strcmp(s, strings+syms[i].st_name)))
R
Rich Felker 已提交
172 173 174 175 176
			return syms+i;
	}
	return 0;
}

177 178
static Sym *gnu_lookup(const char *s, uint32_t h1, struct dso *dso)
{
179 180
	Sym *syms = dso->syms;
	char *strings = dso->strings;
181 182 183 184 185
	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;
186
	uint32_t i = buckets[h1 % nbuckets];
187

188
	if (!i) return 0;
189

190
	hashval = buckets + nbuckets + (i - hashtab[1]);
191

192
	for (h1 |= 1; ; i++) {
193
		h2 = *hashval++;
194 195 196
		if ((!dso->versym || dso->versym[i] >= 0)
		    && (h1 == (h2|1)) && !strcmp(s, strings + syms[i].st_name))
			return syms+i;
197 198 199 200 201 202
		if (h2 & 1) break;
	}

	return 0;
}

203
#define OK_TYPES (1<<STT_NOTYPE | 1<<STT_OBJECT | 1<<STT_FUNC | 1<<STT_COMMON | 1<<STT_TLS)
204
#define OK_BINDS (1<<STB_GLOBAL | 1<<STB_WEAK | 1<<STB_GNU_UNIQUE)
R
Rich Felker 已提交
205

206 207 208 209
#ifndef ARCH_SYM_REJECT_UND
#define ARCH_SYM_REJECT_UND(s) 0
#endif

210
static struct symdef find_sym(struct dso *dso, const char *s, int need_def)
R
Rich Felker 已提交
211
{
212
	uint32_t h = 0, gh = 0;
213
	struct symdef def = {0};
R
Rich Felker 已提交
214
	for (; dso; dso=dso->next) {
215 216
		Sym *sym;
		if (!dso->global) continue;
217 218 219 220 221 222 223
		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);
		}
224 225
		if (!sym) continue;
		if (!sym->st_shndx)
226 227
			if (need_def || (sym->st_info&0xf) == STT_TLS
			    || ARCH_SYM_REJECT_UND(sym))
228 229 230 231 232 233 234 235 236 237 238
				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 已提交
239
	}
240
	return def;
R
Rich Felker 已提交
241 242
}

243
__attribute__((__visibility__("hidden")))
244 245
ptrdiff_t __tlsdesc_static(), __tlsdesc_dynamic();

246
static void do_relocs(struct dso *dso, size_t *rel, size_t rel_size, size_t stride)
R
Rich Felker 已提交
247
{
248 249 250
	unsigned char *base = dso->base;
	Sym *syms = dso->syms;
	char *strings = dso->strings;
R
Rich Felker 已提交
251 252 253
	Sym *sym;
	const char *name;
	void *ctx;
R
Rich Felker 已提交
254
	int type;
R
Rich Felker 已提交
255
	int sym_index;
256
	struct symdef def;
257 258 259 260
	size_t *reloc_addr;
	size_t sym_val;
	size_t tls_val;
	size_t addend;
261 262 263 264 265 266 267 268
	int skip_relative = 0, reuse_addends = 0, save_slot = 0;

	if (dso == &ldso) {
		/* Only ldso's REL table needs addend saving/reuse. */
		if (rel == apply_addends_to)
			reuse_addends = 1;
		skip_relative = 1;
	}
R
Rich Felker 已提交
269 270

	for (; rel_size; rel+=stride, rel_size-=stride*sizeof(size_t)) {
271
		if (skip_relative && IS_RELATIVE(rel[1])) continue;
R
Rich Felker 已提交
272
		type = R_TYPE(rel[1]);
R
Rich Felker 已提交
273
		sym_index = R_SYM(rel[1]);
274
		reloc_addr = (void *)(base + rel[0]);
R
Rich Felker 已提交
275 276 277
		if (sym_index) {
			sym = syms + sym_index;
			name = strings + sym->st_name;
278 279
			ctx = type==REL_COPY ? head->next : head;
			def = find_sym(ctx, name, type==REL_PLT);
280 281
			if (!def.sym && (sym->st_shndx != SHN_UNDEF
			    || sym->st_info>>4 != STB_WEAK)) {
282
				error("Error relocating %s: %s: symbol not found",
283
					dso->name, name);
284
				if (runtime) longjmp(*rtld_fail, 1);
285
				continue;
R
Rich Felker 已提交
286
			}
287
		} else {
288 289
			sym = 0;
			def.sym = 0;
290 291 292
			def.dso = dso;
		}

293 294 295 296 297 298 299 300 301 302 303 304 305 306
		if (stride > 2) {
			addend = rel[2];
		} else if (type==REL_GOT || type==REL_PLT|| type==REL_COPY) {
			addend = 0;
		} else if (reuse_addends) {
			/* Save original addend in stage 2 where the dso
			 * chain consists of just ldso; otherwise read back
			 * saved addend since the inline one was clobbered. */
			if (head==&ldso)
				saved_addends[save_slot] = *reloc_addr;
			addend = saved_addends[save_slot++];
		} else {
			addend = *reloc_addr;
		}
307 308 309 310 311

		sym_val = def.sym ? (size_t)def.dso->base+def.sym->st_value : 0;
		tls_val = def.sym ? def.sym->st_value : 0;

		switch(type) {
R
Rich Felker 已提交
312 313
		case REL_NONE:
			break;
314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344 345 346 347 348 349 350 351 352
		case REL_OFFSET:
			addend -= (size_t)reloc_addr;
		case REL_SYMBOLIC:
		case REL_GOT:
		case REL_PLT:
			*reloc_addr = sym_val + addend;
			break;
		case REL_RELATIVE:
			*reloc_addr = (size_t)base + addend;
			break;
		case REL_SYM_OR_REL:
			if (sym) *reloc_addr = sym_val + addend;
			else *reloc_addr = (size_t)base + addend;
			break;
		case REL_COPY:
			memcpy(reloc_addr, (void *)sym_val, sym->st_size);
			break;
		case REL_OFFSET32:
			*(uint32_t *)reloc_addr = sym_val + addend
				- (size_t)reloc_addr;
			break;
		case REL_DTPMOD:
			*reloc_addr = def.dso->tls_id;
			break;
		case REL_DTPOFF:
			*reloc_addr = tls_val + addend;
			break;
#ifdef TLS_ABOVE_TP
		case REL_TPOFF:
			*reloc_addr = tls_val + def.dso->tls_offset + TPOFF_K + addend;
			break;
#else
		case REL_TPOFF:
			*reloc_addr = tls_val - def.dso->tls_offset + addend;
			break;
		case REL_TPOFF_NEG:
			*reloc_addr = def.dso->tls_offset - tls_val + addend;
			break;
#endif
353 354 355 356
		case REL_TLSDESC:
			if (stride<3) addend = reloc_addr[1];
			if (runtime && def.dso->tls_id >= static_tls_cnt) {
				struct td_index *new = malloc(sizeof *new);
357 358
				if (!new) {
					error(
359 360
					"Error relocating %s: cannot allocate TLSDESC for %s",
					dso->name, sym ? name : "(local)" );
361
					longjmp(*rtld_fail, 1);
362
				}
363 364 365 366 367 368 369 370 371 372 373 374 375 376 377 378 379
				new->next = dso->td_index;
				dso->td_index = new;
				new->args[0] = def.dso->tls_id;
				new->args[1] = tls_val + addend;
				reloc_addr[0] = (size_t)__tlsdesc_dynamic;
				reloc_addr[1] = (size_t)new;
			} else {
				reloc_addr[0] = (size_t)__tlsdesc_static;
#ifdef TLS_ABOVE_TP
				reloc_addr[1] = tls_val + def.dso->tls_offset
					+ TPOFF_K + addend;
#else
				reloc_addr[1] = tls_val - def.dso->tls_offset
					+ addend;
#endif
			}
			break;
R
Rich Felker 已提交
380 381 382
		default:
			error("Error relocating %s: unsupported relocation type %d",
				dso->name, type);
383
			if (runtime) longjmp(*rtld_fail, 1);
R
Rich Felker 已提交
384
			continue;
R
Rich Felker 已提交
385 386 387 388
		}
	}
}

389 390 391 392 393 394
/* 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. */

T
Timo Teräs 已提交
395
static void reclaim(struct dso *dso, size_t start, size_t end)
396 397
{
	size_t *a, *z;
T
Timo Teräs 已提交
398 399
	if (start >= dso->relro_start && start < dso->relro_end) start = dso->relro_end;
	if (end   >= dso->relro_start && end   < dso->relro_end) end = dso->relro_start;
400 401 402
	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;
T
Timo Teräs 已提交
403 404
	a = (size_t *)(dso->base + start);
	z = (size_t *)(dso->base + end);
405 406 407 408 409 410
	a[-2] = 1;
	a[-1] = z[0] = end-start + 2*sizeof(size_t) | 1;
	z[1] = 1;
	free(a);
}

411
static void reclaim_gaps(struct dso *dso)
412
{
413 414
	Phdr *ph = dso->phdr;
	size_t phcnt = dso->phnum;
415

416
	for (; phcnt--; ph=(void *)((char *)ph+dso->phentsize)) {
417 418
		if (ph->p_type!=PT_LOAD) continue;
		if ((ph->p_flags&(PF_R|PF_W))!=(PF_R|PF_W)) continue;
T
Timo Teräs 已提交
419 420
		reclaim(dso, ph->p_vaddr & -PAGE_SIZE, ph->p_vaddr);
		reclaim(dso, ph->p_vaddr+ph->p_memsz,
421 422 423 424
			ph->p_vaddr+ph->p_memsz+PAGE_SIZE-1 & -PAGE_SIZE);
	}
}

425
static void *map_library(int fd, struct dso *dso)
R
Rich Felker 已提交
426
{
427
	Ehdr buf[(896+sizeof(Ehdr))/sizeof(Ehdr)];
428
	void *allocated_buf=0;
R
Rich Felker 已提交
429 430 431 432 433
	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;
434
	Phdr *ph, *ph0;
R
Rich Felker 已提交
435
	unsigned prot;
436
	unsigned char *map=MAP_FAILED, *base;
437
	size_t dyn=0;
438
	size_t tls_image=0;
R
Rich Felker 已提交
439 440 441
	size_t i;

	ssize_t l = read(fd, buf, sizeof buf);
442
	eh = buf;
443 444 445
	if (l<0) return 0;
	if (l<sizeof *eh || (eh->e_type != ET_DYN && eh->e_type != ET_EXEC))
		goto noexec;
R
Rich Felker 已提交
446
	phsize = eh->e_phentsize * eh->e_phnum;
447 448 449 450 451 452 453 454
	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) {
455
		l = pread(fd, buf+1, phsize, eh->e_phoff);
456 457
		if (l < 0) goto error;
		if (l != phsize) goto noexec;
458 459 460
		ph = ph0 = (void *)(buf + 1);
	} else {
		ph = ph0 = (void *)((char *)buf + eh->e_phoff);
R
Rich Felker 已提交
461 462
	}
	for (i=eh->e_phnum; i; i--, ph=(void *)((char *)ph+eh->e_phentsize)) {
463
		if (ph->p_type == PT_DYNAMIC) {
R
Rich Felker 已提交
464
			dyn = ph->p_vaddr;
465
		} else if (ph->p_type == PT_TLS) {
466 467 468 469
			tls_image = ph->p_vaddr;
			dso->tls_align = ph->p_align;
			dso->tls_len = ph->p_filesz;
			dso->tls_size = ph->p_memsz;
T
Timo Teräs 已提交
470 471 472
		} else if (ph->p_type == PT_GNU_RELRO) {
			dso->relro_start = ph->p_vaddr & -PAGE_SIZE;
			dso->relro_end = (ph->p_vaddr + ph->p_memsz) & -PAGE_SIZE;
473
		}
R
Rich Felker 已提交
474 475 476 477 478 479 480 481 482 483 484 485
		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;
		}
	}
486
	if (!dyn) goto noexec;
R
Rich Felker 已提交
487 488 489 490 491 492 493 494 495
	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. */
496
	map = mmap((void *)addr_min, map_len, prot, MAP_PRIVATE, fd, off_start);
497
	if (map==MAP_FAILED) goto error;
498 499 500 501 502 503
	/* 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 已提交
504
	base = map - addr_min;
505 506 507
	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 已提交
508
		if (ph->p_type != PT_LOAD) continue;
509 510 511 512 513 514 515
		/* 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;
516
			dso->phentsize = eh->e_phentsize;
517
		}
R
Rich Felker 已提交
518 519 520 521 522 523 524 525
		/* 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));
526 527
		if (mmap(base+this_min, this_max-this_min, prot, MAP_PRIVATE|MAP_FIXED, fd, off_start) == MAP_FAILED)
			goto error;
R
Rich Felker 已提交
528 529 530 531
		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);
532 533
			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 已提交
534 535
		}
	}
R
Rich Felker 已提交
536 537
	for (i=0; ((size_t *)(base+dyn))[i]; i+=2)
		if (((size_t *)(base+dyn))[i]==DT_TEXTREL) {
538 539
			if (mprotect(map, map_len, PROT_READ|PROT_WRITE|PROT_EXEC) < 0)
				goto error;
R
Rich Felker 已提交
540 541
			break;
		}
542 543 544 545 546
	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);
547
	if (!runtime) reclaim_gaps(dso);
548
	free(allocated_buf);
R
Rich Felker 已提交
549
	return map;
550 551
noexec:
	errno = ENOEXEC;
552
error:
553 554
	if (map!=MAP_FAILED) munmap(map, map_len);
	free(allocated_buf);
555
	return 0;
R
Rich Felker 已提交
556 557
}

558
static int path_open(const char *name, const char *s, char *buf, size_t buf_size)
559
{
560 561
	size_t l;
	int fd;
562
	for (;;) {
563 564 565
		s += strspn(s, ":\n");
		l = strcspn(s, ":\n");
		if (l-1 >= INT_MAX) return -1;
566 567 568 569 570 571 572 573 574 575 576 577 578 579
		if (snprintf(buf, buf_size, "%.*s/%s", (int)l, s, name) < buf_size) {
			if ((fd = open(buf, O_RDONLY|O_CLOEXEC))>=0) return fd;
			switch (errno) {
			case ENOENT:
			case ENOTDIR:
			case EACCES:
			case ENAMETOOLONG:
				break;
			default:
				/* Any negative value but -1 will inhibit
				 * futher path search. */
				return -2;
			}
		}
580
		s += l;
581 582 583
	}
}

584 585 586 587 588
static int fixup_rpath(struct dso *p, char *buf, size_t buf_size)
{
	size_t n, l;
	const char *s, *t, *origin;
	char *d;
589
	if (p->rpath || !p->rpath_orig) return 0;
590 591 592 593 594 595
	if (!strchr(p->rpath_orig, '$')) {
		p->rpath = p->rpath_orig;
		return 0;
	}
	n = 0;
	s = p->rpath_orig;
R
Rich Felker 已提交
596 597
	while ((t=strchr(s, '$'))) {
		if (strncmp(t, "$ORIGIN", 7) && strncmp(t, "${ORIGIN}", 9))
598
			return 0;
599 600 601
		s = t+1;
		n++;
	}
602
	if (n > SSIZE_MAX/PATH_MAX) return 0;
603 604 605 606 607 608 609 610 611

	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)
612
			return 0;
R
Rich Felker 已提交
613
		l = readlink("/proc/self/exe", buf, buf_size);
614 615 616 617 618 619
		if (l == -1) switch (errno) {
		case ENOENT:
		case ENOTDIR:
		case EACCES:
			break;
		default:
620
			return -1;
621 622 623
		}
		if (l >= buf_size)
			return 0;
R
Rich Felker 已提交
624
		buf[l] = 0;
625 626 627 628 629 630 631 632 633 634 635
		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 已提交
636
	while ((t=strchr(s, '$'))) {
637 638 639 640
		memcpy(d, s, t-s);
		d += t-s;
		memcpy(d, origin, l);
		d += l;
R
Rich Felker 已提交
641 642
		/* It was determined previously that the '$' is followed
		 * either by "ORIGIN" or "{ORIGIN}". */
643 644 645 646 647 648
		s = t + 7 + 2*(t[1]=='{');
	}
	strcpy(d, s);
	return 0;
}

649 650
static void decode_dyn(struct dso *p)
{
651
	size_t dyn[DYN_CNT];
652 653 654
	decode_vec(p->dynv, dyn, DYN_CNT);
	p->syms = (void *)(p->base + dyn[DT_SYMTAB]);
	p->strings = (void *)(p->base + dyn[DT_STRTAB]);
655 656
	if (dyn[0]&(1<<DT_HASH))
		p->hashtab = (void *)(p->base + dyn[DT_HASH]);
657
	if (dyn[0]&(1<<DT_RPATH))
658
		p->rpath_orig = (void *)(p->strings + dyn[DT_RPATH]);
659 660
	if (dyn[0]&(1<<DT_RUNPATH))
		p->rpath_orig = (void *)(p->strings + dyn[DT_RUNPATH]);
661 662
	if (search_vec(p->dynv, dyn, DT_GNU_HASH))
		p->ghashtab = (void *)(p->base + *dyn);
663 664
	if (search_vec(p->dynv, dyn, DT_VERSYM))
		p->versym = (void *)(p->base + *dyn);
665 666
}

667
static struct dso *load_library(const char *name, struct dso *needed_by)
R
Rich Felker 已提交
668
{
669
	char buf[2*NAME_MAX+2];
670
	const char *pathname;
671
	unsigned char *map;
672
	struct dso *p, temp_dso = {0};
R
Rich Felker 已提交
673 674
	int fd;
	struct stat st;
675 676
	size_t alloc_size;
	int n_th = 0;
677
	int is_self = 0;
R
Rich Felker 已提交
678

679 680 681 682 683
	if (!*name) {
		errno = EINVAL;
		return 0;
	}

R
Rich Felker 已提交
684 685 686 687 688 689 690
	/* 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;
691
			for (rp=reserved; *rp && strncmp(name+3, rp, l-3); rp+=strlen(rp)+1);
R
Rich Felker 已提交
692
			if (*rp) {
693 694 695 696 697 698 699 700
				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",
R
Rich Felker 已提交
701 702
							name, ldso.name,
							ldso.base);
703 704
					}
				}
705
				is_self = 1;
R
Rich Felker 已提交
706 707 708
			}
		}
	}
R
Rich Felker 已提交
709
	if (!strcmp(name, ldso.name)) is_self = 1;
710
	if (is_self) {
R
Rich Felker 已提交
711 712 713 714
		if (!ldso.prev) {
			tail->next = &ldso;
			ldso.prev = tail;
			tail = ldso.next ? ldso.next : &ldso;
715
		}
R
Rich Felker 已提交
716
		return &ldso;
717
	}
718
	if (strchr(name, '/')) {
719
		pathname = name;
720
		fd = open(name, O_RDONLY|O_CLOEXEC);
R
Rich Felker 已提交
721
	} else {
722 723 724 725 726 727 728
		/* 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;
			}
		}
729
		if (strlen(name) > NAME_MAX) return 0;
730
		fd = -1;
731
		if (env_path) fd = path_open(name, env_path, buf, sizeof buf);
732 733 734 735
		for (p=needed_by; fd == -1 && p; p=p->needed_by) {
			if (fixup_rpath(p, buf, sizeof buf) < 0)
				fd = -2; /* Inhibit further search. */
			if (p->rpath)
736
				fd = path_open(name, p->rpath, buf, sizeof buf);
737
		}
738
		if (fd == -1) {
739
			if (!sys_path) {
740 741
				char *prefix = 0;
				size_t prefix_len;
R
Rich Felker 已提交
742
				if (ldso.name[0]=='/') {
743
					char *s, *t, *z;
R
Rich Felker 已提交
744
					for (s=t=z=ldso.name; *s; s++)
745
						if (*s=='/') z=t, t=s;
R
Rich Felker 已提交
746
					prefix_len = z-ldso.name;
747
					if (prefix_len < PATH_MAX)
R
Rich Felker 已提交
748
						prefix = ldso.name;
749 750 751 752 753 754 755 756 757 758 759
				}
				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");
760
				if (f) {
761
					if (getdelim(&sys_path, (size_t[1]){0}, 0, f) <= 0) {
762
						free(sys_path);
763
						sys_path = "";
764
					}
765
					fclose(f);
766 767
				} else if (errno != ENOENT) {
					sys_path = "";
768 769
				}
			}
770 771
			if (!sys_path) sys_path = "/lib:/usr/local/lib:/usr/lib";
			fd = path_open(name, sys_path, buf, sizeof buf);
R
Rich Felker 已提交
772
		}
773
		pathname = buf;
R
Rich Felker 已提交
774 775 776 777 778 779 780 781
	}
	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) {
782 783 784
			/* 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. */
785 786
			if (!p->shortname && pathname != name)
				p->shortname = strrchr(p->name, '/')+1;
R
Rich Felker 已提交
787 788 789 790 791
			close(fd);
			p->refcnt++;
			return p;
		}
	}
792
	map = noload ? 0 : map_library(fd, &temp_dso);
R
Rich Felker 已提交
793 794
	close(fd);
	if (!map) return 0;
795 796 797 798 799 800 801 802 803 804

	/* 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);
805
		n_th = libc.threads_minus_1 + 1;
806 807 808 809
		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 已提交
810
	if (!p) {
811
		munmap(map, temp_dso.map_len);
R
Rich Felker 已提交
812 813
		return 0;
	}
814
	memcpy(p, &temp_dso, sizeof temp_dso);
815
	decode_dyn(p);
R
Rich Felker 已提交
816 817 818
	p->dev = st.st_dev;
	p->ino = st.st_ino;
	p->refcnt = 1;
819
	p->needed_by = needed_by;
R
Rich Felker 已提交
820
	p->name = p->buf;
821 822 823
	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;
824 825
	if (p->tls_image) {
		p->tls_id = ++tls_cnt;
R
Rich Felker 已提交
826
		tls_align = MAXP2(tls_align, p->tls_align);
827 828 829 830 831
#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 已提交
832 833 834 835
		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;
836
#endif
837 838 839 840
		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 已提交
841 842 843 844 845

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

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

R
Rich Felker 已提交
848 849 850 851 852
	return p;
}

static void load_deps(struct dso *p)
{
853 854
	size_t i, ndeps=0;
	struct dso ***deps = &p->deps, **tmp, *dep;
R
Rich Felker 已提交
855 856 857
	for (; p; p=p->next) {
		for (i=0; p->dynv[i]; i+=2) {
			if (p->dynv[i] != DT_NEEDED) continue;
858
			dep = load_library(p->strings + p->dynv[i+1], p);
859
			if (!dep) {
860
				error("Error loading shared library %s: %m (needed by %s)",
R
Rich Felker 已提交
861
					p->strings + p->dynv[i+1], p->name);
862
				if (runtime) longjmp(*rtld_fail, 1);
863
				continue;
R
Rich Felker 已提交
864
			}
865 866
			if (runtime) {
				tmp = realloc(*deps, sizeof(*tmp)*(ndeps+2));
867
				if (!tmp) longjmp(*rtld_fail, 1);
868 869 870 871
				tmp[ndeps++] = dep;
				tmp[ndeps] = 0;
				*deps = tmp;
			}
R
Rich Felker 已提交
872 873 874 875
		}
	}
}

R
Rich Felker 已提交
876 877 878 879 880
static void load_preload(char *s)
{
	int tmp;
	char *z;
	for (z=s; *z; s=z) {
881 882
		for (   ; *s && (isspace(*s) || *s==':'); s++);
		for (z=s; *z && !isspace(*z) && *z!=':'; z++);
R
Rich Felker 已提交
883 884
		tmp = *z;
		*z = 0;
885
		load_library(s, 0);
R
Rich Felker 已提交
886 887 888 889
		*z = tmp;
	}
}

890 891 892 893 894
static void make_global(struct dso *p)
{
	for (; p; p=p->next) p->global = 1;
}

R
Rich Felker 已提交
895 896 897 898 899
static void do_mips_relocs(struct dso *p, size_t *got)
{
	size_t i, j, rel[2];
	unsigned char *base = p->base;
	i=0; search_vec(p->dynv, &i, DT_MIPS_LOCAL_GOTNO);
900
	if (p==&ldso) {
R
Rich Felker 已提交
901 902 903 904 905 906 907 908 909 910 911 912 913 914
		got += i;
	} else {
		while (i--) *got++ += (size_t)base;
	}
	j=0; search_vec(p->dynv, &j, DT_MIPS_GOTSYM);
	i=0; search_vec(p->dynv, &i, DT_MIPS_SYMTABNO);
	Sym *sym = p->syms + j;
	rel[0] = (unsigned char *)got - base;
	for (i-=j; i; i--, sym++, rel[0]+=sizeof(size_t)) {
		rel[1] = sym-p->syms << 8 | R_MIPS_JUMP_SLOT;
		do_relocs(p, rel, sizeof rel, 2);
	}
}

R
Rich Felker 已提交
915 916
static void reloc_all(struct dso *p)
{
917
	size_t dyn[DYN_CNT];
R
Rich Felker 已提交
918 919 920
	for (; p; p=p->next) {
		if (p->relocated) continue;
		decode_vec(p->dynv, dyn, DYN_CNT);
R
Rich Felker 已提交
921 922
		if (NEED_MIPS_GOT_RELOCS)
			do_mips_relocs(p, (void *)(p->base+dyn[DT_PLTGOT]));
923 924 925 926
		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);
T
Timo Teräs 已提交
927

R
Rich Felker 已提交
928
		if (head != &ldso && p->relro_start != p->relro_end &&
T
Timo Teräs 已提交
929
		    mprotect(p->base+p->relro_start, p->relro_end-p->relro_start, PROT_READ) < 0) {
930
			error("Error relocating %s: RELRO protection failed: %m",
T
Timo Teräs 已提交
931
				p->name);
932
			if (runtime) longjmp(*rtld_fail, 1);
T
Timo Teräs 已提交
933 934
		}

935
		p->relocated = 1;
R
Rich Felker 已提交
936 937 938
	}
}

939
static void kernel_mapped_dso(struct dso *p)
940
{
941 942 943 944 945 946
	size_t min_addr = -1, max_addr = 0, cnt;
	Phdr *ph = p->phdr;
	for (cnt = p->phnum; cnt--; ph = (void *)((char *)ph + p->phentsize)) {
		if (ph->p_type == PT_DYNAMIC) {
			p->dynv = (void *)(p->base + ph->p_vaddr);
		} else if (ph->p_type == PT_GNU_RELRO) {
T
Timo Teräs 已提交
947 948 949
			p->relro_start = ph->p_vaddr & -PAGE_SIZE;
			p->relro_end = (ph->p_vaddr + ph->p_memsz) & -PAGE_SIZE;
		}
950 951 952 953 954 955 956 957 958 959
		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;
960
	p->kernel_mapped = 1;
961 962
}

963 964 965
static void do_fini()
{
	struct dso *p;
966
	size_t dyn[DYN_CNT];
967 968 969
	for (p=fini_head; p; p=p->fini_next) {
		if (!p->constructed) continue;
		decode_vec(p->dynv, dyn, DYN_CNT);
970 971
		if (dyn[0] & (1<<DT_FINI_ARRAY)) {
			size_t n = dyn[DT_FINI_ARRAYSZ]/sizeof(size_t);
972 973
			size_t *fn = (size_t *)(p->base + dyn[DT_FINI_ARRAY])+n;
			while (n--) ((void (*)(void))*--fn)();
974
		}
975
#ifndef NO_LEGACY_INITFINI
976
		if ((dyn[0] & (1<<DT_FINI)) && dyn[DT_FINI])
977
			((void (*)(void))(p->base + dyn[DT_FINI]))();
978
#endif
979 980 981
	}
}

982 983
static void do_init_fini(struct dso *p)
{
984
	size_t dyn[DYN_CNT];
985
	int need_locking = libc.threads_minus_1;
986 987 988 989
	/* 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);
990
	for (; p; p=p->prev) {
991 992
		if (p->constructed) continue;
		p->constructed = 1;
993
		decode_vec(p->dynv, dyn, DYN_CNT);
994
		if (dyn[0] & ((1<<DT_FINI) | (1<<DT_FINI_ARRAY))) {
995 996 997
			p->fini_next = fini_head;
			fini_head = p;
		}
998
#ifndef NO_LEGACY_INITFINI
999
		if ((dyn[0] & (1<<DT_INIT)) && dyn[DT_INIT])
1000
			((void (*)(void))(p->base + dyn[DT_INIT]))();
1001
#endif
1002 1003 1004 1005 1006
		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++)();
		}
1007 1008 1009 1010
		if (!need_locking && libc.threads_minus_1) {
			need_locking = 1;
			pthread_mutex_lock(&init_fini_lock);
		}
1011
	}
1012
	if (need_locking) pthread_mutex_unlock(&init_fini_lock);
1013 1014
}

1015
static void dl_debug_state(void)
1016 1017 1018
{
}

1019 1020
weak_alias(dl_debug_state, _dl_debug_state);

1021 1022 1023 1024 1025 1026 1027 1028 1029 1030 1031 1032 1033
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;
	}
}

1034
void *__copy_tls(unsigned char *mem)
1035
{
R
Rich Felker 已提交
1036
	pthread_t td;
1037
	struct dso *p;
1038
	void **dtv;
R
Rich Felker 已提交
1039

1040
#ifdef TLS_ABOVE_TP
1041 1042
	dtv = (void **)(mem + libc.tls_size) - (tls_cnt + 1);

1043 1044 1045 1046 1047 1048 1049 1050 1051 1052
	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
1053 1054
	dtv = (void **)mem;

1055
	mem += libc.tls_size - sizeof(struct pthread);
R
Rich Felker 已提交
1056 1057 1058 1059
	mem -= (uintptr_t)mem & (tls_align-1);
	td = (pthread_t)mem;

	for (p=head; p; p=p->next) {
1060
		if (!p->tls_id) continue;
R
Rich Felker 已提交
1061 1062
		dtv[p->tls_id] = mem - p->tls_offset;
		memcpy(dtv[p->tls_id], p->tls_image, p->tls_len);
1063
	}
1064
#endif
1065
	dtv[0] = (void *)tls_cnt;
1066
	td->dtv = td->dtv_copy = dtv;
R
Rich Felker 已提交
1067
	return td;
1068 1069
}

1070
__attribute__((__visibility__("hidden")))
1071
void *__tls_get_new(size_t *v)
1072 1073
{
	pthread_t self = __pthread_self();
1074 1075 1076

	/* Block signals to make accessing new TLS async-signal-safe */
	sigset_t set;
1077
	__block_all_sigs(&set);
1078
	if (v[0]<=(size_t)self->dtv[0]) {
1079
		__restore_sigs(&set);
1080 1081 1082 1083 1084 1085 1086 1087 1088 1089 1090
		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 */
1091
	if (v[0] > (size_t)self->dtv[0]) {
1092 1093
		void **newdtv = p->new_dtv +
			(v[0]+1)*sizeof(void *)*a_fetch_add(&p->new_dtv_idx,1);
1094
		memcpy(newdtv, self->dtv,
1095 1096
			((size_t)self->dtv[0]+1) * sizeof(void *));
		newdtv[0] = (void *)v[0];
1097
		self->dtv = self->dtv_copy = newdtv;
1098
	}
1099

1100 1101 1102 1103 1104 1105 1106 1107 1108 1109 1110 1111
	/* Get new TLS memory from all new DSOs up to the requested one */
	unsigned char *mem;
	for (p=head; ; p=p->next) {
		if (!p->tls_id || self->dtv[p->tls_id]) continue;
		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[p->tls_id] = mem;
		memcpy(mem, p->tls_image, p->tls_len);
		if (p->tls_id == v[0]) break;
	}
1112
	__restore_sigs(&set);
1113
	return mem + v[1];
1114 1115
}

R
Rich Felker 已提交
1116 1117
static void update_tls_size()
{
1118 1119 1120 1121 1122 1123
	libc.tls_size = ALIGN(
		(1+tls_cnt) * sizeof(void *) +
		tls_offset +
		sizeof(struct pthread) +
		tls_align * 2,
	tls_align);
R
Rich Felker 已提交
1124 1125
}

R
Rich Felker 已提交
1126 1127 1128 1129 1130 1131 1132 1133 1134 1135 1136
/* Stage 1 of the dynamic linker is defined in dlstart.c. It calls the
 * following stage 2 and stage 3 functions via primitive symbolic lookup
 * since it does not have access to their addresses to begin with. */

/* Stage 2 of the dynamic linker is called after relative relocations 
 * have been processed. It can make function calls to static functions
 * and access string literals and static data, but cannot use extern
 * symbols. Its job is to perform symbolic relocations on the dynamic
 * linker itself, but some of the relocations performed may need to be
 * replaced later due to copy relocations in the main program. */

1137
void __dls2(unsigned char *base, size_t *sp)
R
Rich Felker 已提交
1138
{
R
Rich Felker 已提交
1139 1140 1141 1142 1143 1144 1145 1146 1147 1148
	Ehdr *ehdr = (void *)base;
	ldso.base = base;
	ldso.name = ldso.shortname = "libc.so";
	ldso.global = 1;
	ldso.phnum = ehdr->e_phnum;
	ldso.phdr = (void *)(base + ehdr->e_phoff);
	ldso.phentsize = ehdr->e_phentsize;
	kernel_mapped_dso(&ldso);
	decode_dyn(&ldso);

1149 1150 1151 1152 1153 1154 1155 1156 1157 1158 1159 1160 1161 1162 1163 1164
	/* Prepare storage for to save clobbered REL addends so they
	 * can be reused in stage 3. There should be very few. If
	 * something goes wrong and there are a huge number, abort
	 * instead of risking stack overflow. */
	size_t dyn[DYN_CNT];
	decode_vec(ldso.dynv, dyn, DYN_CNT);
	size_t *rel = (void *)(base+dyn[DT_REL]);
	size_t rel_size = dyn[DT_RELSZ];
	size_t symbolic_rel_cnt = 0;
	apply_addends_to = rel;
	for (; rel_size; rel+=2, rel_size-=2*sizeof(size_t))
		if (!IS_RELATIVE(rel[1])) symbolic_rel_cnt++;
	if (symbolic_rel_cnt >= ADDEND_LIMIT) a_crash();
	size_t addends[symbolic_rel_cnt+1];
	saved_addends = addends;

R
Rich Felker 已提交
1165 1166 1167 1168
	head = &ldso;
	reloc_all(&ldso);

	ldso.relocated = 0;
1169 1170 1171 1172 1173 1174

	/* Call dynamic linker stage-3, __dls3, looking it up
	 * symbolically as a barrier against moving the address
	 * load across the above relocation processing. */
	struct symdef dls3_def = find_sym(&ldso, "__dls3", 0);
	((stage3_func)(ldso.base+dls3_def.sym->st_value))(sp);
R
Rich Felker 已提交
1175 1176 1177 1178 1179 1180 1181 1182 1183 1184
}

/* Stage 3 of the dynamic linker is called with the dynamic linker/libc
 * fully functional. Its job is to load (if not already loaded) and
 * process dependencies and relocations for the main application and
 * transfer control to its entry point. */

_Noreturn void __dls3(size_t *sp)
{
	static struct dso app, vdso;
1185
	size_t aux[AUX_CNT], *auxv;
R
Rich Felker 已提交
1186
	size_t i;
R
Rich Felker 已提交
1187
	char *env_preload=0;
R
Rich Felker 已提交
1188
	size_t vdso_base;
R
Rich Felker 已提交
1189 1190 1191
	int argc = *sp;
	char **argv = (void *)(sp+1);
	char **argv_orig = argv;
1192
	char **envp = argv+argc+1;
1193 1194 1195 1196 1197 1198

	/* Setup early thread pointer in builtin_tls for ldso/libc itself to
	 * use during dynamic linking. If possible it will also serve as the
	 * thread pointer at runtime. */
	libc.tls_size = sizeof builtin_tls;
	if (__init_tp(__copy_tls((void *)builtin_tls)) < 0) {
1199
		a_crash();
1200
	}
R
Rich Felker 已提交
1201 1202

	/* Find aux vector just past environ[] */
1203 1204 1205
	for (i=argc+1; argv[i]; i++)
		if (!memcmp(argv[i], "LD_LIBRARY_PATH=", 16))
			env_path = argv[i]+16;
R
Rich Felker 已提交
1206 1207
		else if (!memcmp(argv[i], "LD_PRELOAD=", 11))
			env_preload = argv[i]+11;
R
Rich Felker 已提交
1208 1209 1210 1211
	auxv = (void *)(argv+i+1);

	decode_vec(auxv, aux, AUX_CNT);

1212 1213
	/* 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 已提交
1214
	  || aux[AT_GID]!=aux[AT_EGID] || aux[AT_SECURE]) {
1215
		env_path = 0;
R
Rich Felker 已提交
1216
		env_preload = 0;
1217
		libc.secure = 1;
1218
	}
1219
	libc.page_size = aux[AT_PAGESZ];
1220
	libc.auxv = auxv;
1221

R
Rich Felker 已提交
1222 1223 1224 1225
	/* If the main program was already loaded by the kernel,
	 * AT_PHDR will point to some location other than the dynamic
	 * linker's program headers. */
	if (aux[AT_PHDR] != (size_t)ldso.phdr) {
1226
		size_t interp_off = 0;
1227
		size_t tls_image = 0;
1228
		/* Find load address of the main program, via AT_PHDR vs PT_PHDR. */
R
Rich Felker 已提交
1229 1230 1231
		Phdr *phdr = app.phdr = (void *)aux[AT_PHDR];
		app.phnum = aux[AT_PHNUM];
		app.phentsize = aux[AT_PHENT];
1232 1233
		for (i=aux[AT_PHNUM]; i; i--, phdr=(void *)((char *)phdr + aux[AT_PHENT])) {
			if (phdr->p_type == PT_PHDR)
R
Rich Felker 已提交
1234
				app.base = (void *)(aux[AT_PHDR] - phdr->p_vaddr);
1235 1236
			else if (phdr->p_type == PT_INTERP)
				interp_off = (size_t)phdr->p_vaddr;
1237 1238
			else if (phdr->p_type == PT_TLS) {
				tls_image = phdr->p_vaddr;
R
Rich Felker 已提交
1239 1240 1241
				app.tls_len = phdr->p_filesz;
				app.tls_size = phdr->p_memsz;
				app.tls_align = phdr->p_align;
1242
			}
1243
		}
R
Rich Felker 已提交
1244 1245
		if (app.tls_size) app.tls_image = (char *)app.base + tls_image;
		if (interp_off) ldso.name = (char *)app.base + interp_off;
1246 1247
		if ((aux[0] & (1UL<<AT_EXECFN))
		    && strncmp((char *)aux[AT_EXECFN], "/proc/", 6))
R
Rich Felker 已提交
1248
			app.name = (char *)aux[AT_EXECFN];
1249
		else
R
Rich Felker 已提交
1250 1251
			app.name = argv[0];
		kernel_mapped_dso(&app);
1252 1253 1254
	} else {
		int fd;
		char *ldname = argv[0];
R
Rich Felker 已提交
1255
		size_t l = strlen(ldname);
1256
		if (l >= 3 && !strcmp(ldname+l-3, "ldd")) ldd_mode = 1;
R
Rich Felker 已提交
1257
		argv++;
1258 1259 1260 1261 1262 1263 1264 1265 1266 1267 1268 1269 1270 1271 1272 1273 1274 1275 1276
		while (argv[0] && argv[0][0]=='-' && argv[0][1]=='-') {
			char *opt = argv[0]+2;
			*argv++ = (void *)-1;
			if (!*opt) {
				break;
			} else if (!memcmp(opt, "list", 5)) {
				ldd_mode = 1;
			} else if (!memcmp(opt, "library-path", 12)) {
				if (opt[12]=='=') env_path = opt+13;
				else if (opt[12]) *argv = 0;
				else if (*argv) env_path = *argv++;
			} else if (!memcmp(opt, "preload", 7)) {
				if (opt[7]=='=') env_preload = opt+8;
				else if (opt[7]) *argv = 0;
				else if (*argv) env_preload = *argv++;
			} else {
				argv[0] = 0;
			}
		}
R
Rich Felker 已提交
1277
		argv[-1] = (void *)(argc - (argv-argv_orig));
1278
		if (!argv[0]) {
1279 1280 1281
			dprintf(2, "musl libc\n"
				"Version %s\n"
				"Dynamic Program Loader\n"
1282
				"Usage: %s [options] [--] pathname%s\n",
1283
				__libc_get_version(), ldname,
1284 1285 1286 1287 1288 1289 1290 1291 1292
				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;
R
Rich Felker 已提交
1293
		Ehdr *ehdr = (void *)map_library(fd, &app);
1294 1295 1296 1297 1298 1299
		if (!ehdr) {
			dprintf(2, "%s: %s: Not a valid dynamic program\n", ldname, argv[0]);
			_exit(1);
		}
		runtime = 0;
		close(fd);
R
Rich Felker 已提交
1300 1301 1302
		ldso.name = ldname;
		app.name = argv[0];
		aux[AT_ENTRY] = (size_t)app.base + ehdr->e_entry;
1303 1304 1305
		/* Find the name that would have been used for the dynamic
		 * linker had ldd not taken its place. */
		if (ldd_mode) {
R
Rich Felker 已提交
1306 1307 1308 1309
			for (i=0; i<app.phnum; i++) {
				if (app.phdr[i].p_type == PT_INTERP)
					ldso.name = (void *)(app.base
						+ app.phdr[i].p_vaddr);
1310
			}
R
Rich Felker 已提交
1311
			dprintf(1, "\t%s (%p)\n", ldso.name, ldso.base);
1312
		}
1313
	}
R
Rich Felker 已提交
1314 1315
	if (app.tls_size) {
		app.tls_id = tls_cnt = 1;
1316
#ifdef TLS_ABOVE_TP
R
Rich Felker 已提交
1317 1318 1319 1320
		app.tls_offset = 0;
		tls_offset = app.tls_size
			+ ( -((uintptr_t)app.tls_image + app.tls_size)
			& (app.tls_align-1) );
1321
#else
R
Rich Felker 已提交
1322 1323 1324
		tls_offset = app.tls_offset = app.tls_size
			+ ( -((uintptr_t)app.tls_image + app.tls_size)
			& (app.tls_align-1) );
1325
#endif
R
Rich Felker 已提交
1326
		tls_align = MAXP2(tls_align, app.tls_align);
1327
	}
R
Rich Felker 已提交
1328 1329
	app.global = 1;
	decode_dyn(&app);
1330 1331

	/* Attach to vdso, if provided by the kernel */
R
Rich Felker 已提交
1332
	if (search_vec(auxv, &vdso_base, AT_SYSINFO_EHDR)) {
R
Rich Felker 已提交
1333 1334 1335 1336
		Ehdr *ehdr = (void *)vdso_base;
		Phdr *phdr = vdso.phdr = (void *)(vdso_base + ehdr->e_phoff);
		vdso.phnum = ehdr->e_phnum;
		vdso.phentsize = ehdr->e_phentsize;
1337 1338
		for (i=ehdr->e_phnum; i; i--, phdr=(void *)((char *)phdr + ehdr->e_phentsize)) {
			if (phdr->p_type == PT_DYNAMIC)
R
Rich Felker 已提交
1339
				vdso.dynv = (void *)(vdso_base + phdr->p_offset);
1340
			if (phdr->p_type == PT_LOAD)
R
Rich Felker 已提交
1341
				vdso.base = (void *)(vdso_base - phdr->p_vaddr + phdr->p_offset);
1342
		}
R
Rich Felker 已提交
1343 1344 1345 1346 1347 1348 1349
		vdso.name = "";
		vdso.shortname = "linux-gate.so.1";
		vdso.global = 1;
		vdso.relocated = 1;
		decode_dyn(&vdso);
		vdso.prev = &ldso;
		ldso.next = &vdso;
1350 1351
	}

R
Rich Felker 已提交
1352 1353
	/* Initial dso chain consists only of the app. */
	head = tail = &app;
R
Rich Felker 已提交
1354

1355
	/* Donate unused parts of app and library mapping to malloc */
R
Rich Felker 已提交
1356 1357
	reclaim_gaps(&app);
	reclaim_gaps(&ldso);
1358

1359
	/* Load preload/needed libraries, add their symbols to the global
R
Rich Felker 已提交
1360
	 * namespace, and perform all remaining relocations. */
R
Rich Felker 已提交
1361
	if (env_preload) load_preload(env_preload);
R
Rich Felker 已提交
1362 1363
	load_deps(&app);
	make_global(&app);
1364

T
Timo Teräs 已提交
1365
#ifndef DYNAMIC_IS_RO
R
Rich Felker 已提交
1366 1367 1368
	for (i=0; app.dynv[i]; i+=2)
		if (app.dynv[i]==DT_DEBUG)
			app.dynv[i+1] = (size_t)&debug;
T
Timo Teräs 已提交
1369 1370
#endif

R
Rich Felker 已提交
1371 1372 1373 1374
	/* The main program must be relocated LAST since it may contin
	 * copy relocations which depend on libraries' relocations. */
	reloc_all(app.next);
	reloc_all(&app);
R
Rich Felker 已提交
1375 1376

	update_tls_size();
1377 1378
	if (libc.tls_size > sizeof builtin_tls || tls_align > MIN_TLS_ALIGN) {
		void *initial_tls = calloc(libc.tls_size, 1);
1379
		if (!initial_tls) {
1380
			dprintf(2, "%s: Error getting %zu bytes thread-local storage: %m\n",
1381
				argv[0], libc.tls_size);
1382 1383
			_exit(127);
		}
1384
		if (__init_tp(__copy_tls(initial_tls)) < 0) {
1385
			a_crash();
1386
		}
1387
	} else {
1388 1389 1390 1391 1392 1393 1394 1395
		size_t tmp_tls_size = libc.tls_size;
		pthread_t self = __pthread_self();
		/* Temporarily set the tls size to the full size of
		 * builtin_tls so that __copy_tls will use the same layout
		 * as it did for before. Then check, just to be safe. */
		libc.tls_size = sizeof builtin_tls;
		if (__copy_tls((void*)builtin_tls) != self) a_crash();
		libc.tls_size = tmp_tls_size;
1396
	}
1397
	static_tls_cnt = tls_cnt;
1398

1399
	if (ldso_fail) _exit(127);
1400 1401
	if (ldd_mode) _exit(0);

1402 1403
	/* Switch to runtime mode: any further failures in the dynamic
	 * linker are a reportable failure rather than a fatal startup
R
Rich Felker 已提交
1404
	 * error. */
R
Rich Felker 已提交
1405
	runtime = 1;
1406

1407
	debug.ver = 1;
1408
	debug.bp = dl_debug_state;
1409
	debug.head = head;
R
Rich Felker 已提交
1410
	debug.base = ldso.base;
1411 1412 1413
	debug.state = 0;
	_dl_debug_state();

1414
	__init_libc(envp, argv[0]);
1415
	atexit(do_fini);
1416
	errno = 0;
1417
	do_init_fini(tail);
1418

R
Rich Felker 已提交
1419 1420
	CRTJMP((void *)aux[AT_ENTRY], argv-1);
	for(;;);
1421 1422
}

1423 1424
void *dlopen(const char *file, int mode)
{
R
Rich Felker 已提交
1425
	struct dso *volatile p, *orig_tail, *next;
R
Rich Felker 已提交
1426
	size_t orig_tls_cnt, orig_tls_offset, orig_tls_align;
1427
	size_t i;
1428
	int cs;
1429
	jmp_buf jb;
1430 1431 1432

	if (!file) return head;

1433
	pthread_setcancelstate(PTHREAD_CANCEL_DISABLE, &cs);
1434
	pthread_rwlock_wrlock(&lock);
1435
	__inhibit_ptc();
1436

1437 1438
	p = 0;
	orig_tls_cnt = tls_cnt;
R
Rich Felker 已提交
1439 1440
	orig_tls_offset = tls_offset;
	orig_tls_align = tls_align;
R
Rich Felker 已提交
1441
	orig_tail = tail;
1442
	noload = mode & RTLD_NOLOAD;
R
Rich Felker 已提交
1443

1444 1445
	rtld_fail = &jb;
	if (setjmp(*rtld_fail)) {
1446
		/* Clean up anything new that was (partially) loaded */
1447
		if (p && p->deps) for (i=0; p->deps[i]; i++)
1448 1449
			if (p->deps[i]->global < 0)
				p->deps[i]->global = 0;
1450 1451 1452
		for (p=orig_tail->next; p; p=next) {
			next = p->next;
			munmap(p->map, p->map_len);
1453 1454 1455 1456 1457
			while (p->td_index) {
				void *tmp = p->td_index->next;
				free(p->td_index);
				p->td_index = tmp;
			}
1458 1459
			if (p->rpath != p->rpath_orig)
				free(p->rpath);
1460 1461 1462
			free(p->deps);
			free(p);
		}
1463
		tls_cnt = orig_tls_cnt;
R
Rich Felker 已提交
1464 1465
		tls_offset = orig_tls_offset;
		tls_align = orig_tls_align;
1466 1467
		tail = orig_tail;
		tail->next = 0;
1468
		p = 0;
1469
		goto end;
1470
	} else p = load_library(file, head);
R
Rich Felker 已提交
1471 1472

	if (!p) {
1473
		error(noload ?
1474 1475 1476
			"Library %s is not already loaded" :
			"Error loading shared library %s: %m",
			file);
1477
		goto end;
1478 1479 1480 1481 1482
	}

	/* First load handling */
	if (!p->deps) {
		load_deps(p);
R
Rich Felker 已提交
1483
		if (p->deps) for (i=0; p->deps[i]; i++)
1484 1485 1486
			if (!p->deps[i]->global)
				p->deps[i]->global = -1;
		if (!p->global) p->global = -1;
1487
		reloc_all(p);
R
Rich Felker 已提交
1488
		if (p->deps) for (i=0; p->deps[i]; i++)
1489 1490 1491
			if (p->deps[i]->global < 0)
				p->deps[i]->global = 0;
		if (p->global < 0) p->global = 0;
1492 1493 1494
	}

	if (mode & RTLD_GLOBAL) {
R
Rich Felker 已提交
1495
		if (p->deps) for (i=0; p->deps[i]; i++)
1496 1497 1498 1499
			p->deps[i]->global = 1;
		p->global = 1;
	}

R
Rich Felker 已提交
1500
	update_tls_size();
1501
	_dl_debug_state();
1502
	orig_tail = tail;
1503
end:
1504
	__release_ptc();
R
Rich Felker 已提交
1505
	if (p) gencnt++;
1506
	pthread_rwlock_unlock(&lock);
1507
	if (p) do_init_fini(orig_tail);
1508
	pthread_setcancelstate(cs, 0);
1509 1510 1511
	return p;
}

1512
static int invalid_dso_handle(void *h)
1513 1514 1515
{
	struct dso *p;
	for (p=head; p; p=p->next) if (h==p) return 0;
1516
	error("Invalid library handle %p", (void *)h);
1517 1518 1519
	return 1;
}

1520 1521
void *__tls_get_addr(size_t *);

R
Rich Felker 已提交
1522
static void *do_dlsym(struct dso *p, const char *s, void *ra)
1523 1524
{
	size_t i;
1525
	uint32_t h = 0, gh = 0;
1526
	Sym *sym;
1527
	if (p == head || p == RTLD_DEFAULT || p == RTLD_NEXT) {
1528 1529 1530
		if (p == RTLD_DEFAULT) {
			p = head;
		} else if (p == RTLD_NEXT) {
1531 1532
			for (p=head; p && (unsigned char *)ra-p->map>p->map_len; p=p->next);
			if (!p) p=head;
1533
			p = p->next;
1534
		}
1535
		struct symdef def = find_sym(p, s, 0);
1536
		if (!def.sym) goto failed;
1537 1538
		if ((def.sym->st_info&0xf) == STT_TLS)
			return __tls_get_addr((size_t []){def.dso->tls_id, def.sym->st_value});
1539
		return def.dso->base + def.sym->st_value;
R
Rich Felker 已提交
1540
	}
1541
	if (invalid_dso_handle(p))
1542
		return 0;
1543 1544 1545 1546 1547 1548 1549
	if (p->ghashtab) {
		gh = gnu_hash(s);
		sym = gnu_lookup(s, gh, p);
	} else {
		h = sysv_hash(s);
		sym = sysv_lookup(s, h, p);
	}
1550 1551
	if (sym && (sym->st_info&0xf) == STT_TLS)
		return __tls_get_addr((size_t []){p->tls_id, sym->st_value});
1552 1553 1554
	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++) {
1555 1556
		if (p->deps[i]->ghashtab) {
			if (!gh) gh = gnu_hash(s);
1557
			sym = gnu_lookup(s, gh, p->deps[i]);
1558 1559 1560 1561
		} else {
			if (!h) h = sysv_hash(s);
			sym = sysv_lookup(s, h, p->deps[i]);
		}
1562 1563
		if (sym && (sym->st_info&0xf) == STT_TLS)
			return __tls_get_addr((size_t []){p->deps[i]->tls_id, sym->st_value});
1564 1565 1566
		if (sym && sym->st_value && (1<<(sym->st_info&0xf) & OK_TYPES))
			return p->deps[i]->base + sym->st_value;
	}
1567
failed:
1568
	error("Symbol not found: %s", s);
1569 1570 1571
	return 0;
}

1572
int __dladdr(const void *addr, Dl_info *info)
1573 1574 1575 1576 1577 1578 1579 1580 1581 1582 1583 1584 1585 1586 1587 1588 1589 1590 1591 1592 1593 1594 1595 1596
{
	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 已提交
1597
		for (i = nsym = 0; i < p->ghashtab[0]; i++) {
1598 1599 1600 1601 1602 1603 1604 1605 1606 1607 1608 1609
			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++) {
1610
		if (sym->st_value
1611 1612 1613 1614 1615 1616 1617 1618 1619 1620 1621 1622 1623 1624 1625 1626 1627 1628 1629 1630 1631 1632
		 && (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;
}

1633
__attribute__((__visibility__("hidden")))
1634
void *__dlsym(void *restrict p, const char *restrict s, void *restrict ra)
1635 1636 1637
{
	void *res;
	pthread_rwlock_rdlock(&lock);
R
Rich Felker 已提交
1638
	res = do_dlsym(p, s, ra);
1639 1640 1641
	pthread_rwlock_unlock(&lock);
	return res;
}
R
Rich Felker 已提交
1642 1643 1644 1645 1646 1647 1648 1649 1650 1651 1652 1653 1654 1655 1656 1657 1658 1659 1660 1661 1662 1663 1664 1665 1666 1667

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;
}
1668
#else
1669
static int invalid_dso_handle(void *h)
1670
{
1671
	error("Invalid library handle %p", (void *)h);
1672 1673
	return 1;
}
1674 1675
void *dlopen(const char *file, int mode)
{
1676
	error("Dynamic loading not supported");
1677 1678
	return 0;
}
1679
void *__dlsym(void *restrict p, const char *restrict s, void *restrict ra)
1680
{
1681
	error("Symbol not found: %s", s);
1682 1683
	return 0;
}
1684
int __dladdr (const void *addr, Dl_info *info)
1685 1686 1687
{
	return 0;
}
1688
#endif
1689

R
Rich Felker 已提交
1690 1691 1692 1693
int __dlinfo(void *dso, int req, void *res)
{
	if (invalid_dso_handle(dso)) return -1;
	if (req != RTLD_DI_LINKMAP) {
1694
		error("Unsupported request %d", req);
R
Rich Felker 已提交
1695 1696 1697 1698 1699 1700
		return -1;
	}
	*(struct link_map **)res = dso;
	return 0;
}

1701 1702
char *dlerror()
{
1703 1704 1705 1706 1707 1708 1709 1710
	pthread_t self = __pthread_self();
	if (!self->dlerror_flag) return 0;
	self->dlerror_flag = 0;
	char *s = self->dlerror_buf;
	if (s == (void *)-1)
		return "Dynamic linker failed to allocate memory for error message";
	else
		return s;
1711 1712 1713 1714
}

int dlclose(void *p)
{
1715
	return invalid_dso_handle(p);
1716
}
1717 1718 1719 1720 1721 1722 1723 1724 1725 1726 1727 1728 1729 1730 1731 1732 1733 1734 1735 1736 1737 1738 1739 1740 1741 1742 1743 1744 1745 1746 1747 1748 1749 1750 1751 1752 1753

void __dl_thread_cleanup(void)
{
	pthread_t self = __pthread_self();
	if (self->dlerror_buf != (void *)-1)
		free(self->dlerror_buf);
}

static void error(const char *fmt, ...)
{
	va_list ap;
	va_start(ap, fmt);
#ifdef SHARED
	if (!runtime) {
		vdprintf(2, fmt, ap);
		dprintf(2, "\n");
		ldso_fail = 1;
		va_end(ap);
		return;
	}
#endif
	pthread_t self = __pthread_self();
	if (self->dlerror_buf != (void *)-1)
		free(self->dlerror_buf);
	size_t len = vsnprintf(0, 0, fmt, ap);
	va_end(ap);
	char *buf = malloc(len+1);
	if (buf) {
		va_start(ap, fmt);
		vsnprintf(buf, len+1, fmt, ap);
		va_end(ap);
	} else {
		buf = (void *)-1;	
	}
	self->dlerror_buf = buf;
	self->dlerror_flag = 1;
}