dynlink.c 41.7 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

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

26
#ifdef SHARED
R
Rich Felker 已提交
27

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

45 46 47 48 49 50 51 52
struct debug {
	int ver;
	void *head;
	void (*bp)(void);
	int state;
	void *base;
};

53 54 55 56 57
struct td_index {
	size_t args[2];
	struct td_index *next;
};

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

R
Rich Felker 已提交
64 65
	Phdr *phdr;
	int phnum;
66
	size_t phentsize;
R
Rich Felker 已提交
67 68
	int refcnt;
	Sym *syms;
69
	uint32_t *hashtab;
70
	uint32_t *ghashtab;
71
	int16_t *versym;
R
Rich Felker 已提交
72 73 74 75 76
	char *strings;
	unsigned char *map;
	size_t map_len;
	dev_t dev;
	ino_t ino;
77
	signed char global;
78 79
	char relocated;
	char constructed;
80
	char kernel_mapped;
81
	struct dso **deps, *needed_by;
82
	char *rpath_orig, *rpath;
83
	void *tls_image;
84
	size_t tls_len, tls_size, tls_align, tls_id, tls_offset;
T
Timo Teräs 已提交
85
	size_t relro_start, relro_end;
86 87 88
	void **new_dtv;
	unsigned char *new_tls;
	int new_dtv_idx, new_tls_idx;
89
	struct td_index *td_index;
90
	struct dso *fini_next;
91
	char *shortname;
R
Rich Felker 已提交
92
	char buf[];
R
Rich Felker 已提交
93 94
};

95 96 97 98 99
struct symdef {
	Sym *sym;
	struct dso *dso;
};

100 101 102 103 104 105 106 107 108 109 110 111 112 113 114
enum {
	REL_ERR,
	REL_SYMBOLIC,
	REL_GOT,
	REL_PLT,
	REL_RELATIVE,
	REL_OFFSET,
	REL_OFFSET32,
	REL_COPY,
	REL_SYM_OR_REL,
	REL_TLS, /* everything past here is TLS */
	REL_DTPMOD,
	REL_DTPOFF,
	REL_TPOFF,
	REL_TPOFF_NEG,
115
	REL_TLSDESC,
116 117
};

118 119
#include "reloc.h"

120
int __init_tp(void *);
121
void __init_libc(char **, char *);
122

123 124
const char *__libc_get_version(void);

125
static struct dso *head, *tail, *ldso, *fini_head;
126
static char *env_path, *sys_path;
R
Rich Felker 已提交
127
static unsigned long long gencnt;
R
Rich Felker 已提交
128
static int runtime;
129
static int ldd_mode;
130
static int ldso_fail;
131
static int noload;
132
static jmp_buf *rtld_fail;
133
static pthread_rwlock_t lock;
134
static struct debug debug;
135
static size_t tls_cnt, tls_offset, tls_align = 4*sizeof(size_t);
136
static size_t static_tls_cnt;
137
static pthread_mutex_t init_fini_lock = { ._m_type = PTHREAD_MUTEX_RECURSIVE };
138
static long long builtin_tls[(sizeof(struct pthread) + 64)/sizeof(long long)];
139 140

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

142
#define AUX_CNT 38
R
Rich Felker 已提交
143 144 145 146 147 148 149 150 151 152 153
#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];
	}
}

154 155 156 157 158 159 160 161
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;
}

162 163 164 165 166 167 168 169 170 171 172
static void error(const char *fmt, ...)
{
	va_list ap;
	va_start(ap, fmt);
	vsnprintf(errbuf, sizeof errbuf, fmt, ap);
	va_end(ap);
	if (runtime) longjmp(*rtld_fail, 1);
	dprintf(2, "%s\n", errbuf);
	ldso_fail = 1;
}

173
static uint32_t sysv_hash(const char *s0)
R
Rich Felker 已提交
174
{
175
	const unsigned char *s = (void *)s0;
R
Rich Felker 已提交
176 177 178 179 180 181 182 183
	uint_fast32_t h = 0;
	while (*s) {
		h = 16*h + *s++;
		h ^= h>>24 & 0xf0;
	}
	return h & 0xfffffff;
}

184 185 186 187 188 189 190 191 192 193
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 已提交
194 195
{
	size_t i;
R
Rich Felker 已提交
196 197 198
	Sym *syms = dso->syms;
	uint32_t *hashtab = dso->hashtab;
	char *strings = dso->strings;
R
Rich Felker 已提交
199
	for (i=hashtab[2+h%hashtab[0]]; i; i=hashtab[2+hashtab[0]+i]) {
200 201
		if ((!dso->versym || dso->versym[i] >= 0)
		    && (!strcmp(s, strings+syms[i].st_name)))
R
Rich Felker 已提交
202 203 204 205 206
			return syms+i;
	}
	return 0;
}

207 208
static Sym *gnu_lookup(const char *s, uint32_t h1, struct dso *dso)
{
209 210
	Sym *syms = dso->syms;
	char *strings = dso->strings;
211 212 213 214 215
	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;
216
	uint32_t i = buckets[h1 % nbuckets];
217

218
	if (!i) return 0;
219

220
	hashval = buckets + nbuckets + (i - hashtab[1]);
221

222
	for (h1 |= 1; ; i++) {
223
		h2 = *hashval++;
224 225 226
		if ((!dso->versym || dso->versym[i] >= 0)
		    && (h1 == (h2|1)) && !strcmp(s, strings + syms[i].st_name))
			return syms+i;
227 228 229 230 231 232
		if (h2 & 1) break;
	}

	return 0;
}

233
#define OK_TYPES (1<<STT_NOTYPE | 1<<STT_OBJECT | 1<<STT_FUNC | 1<<STT_COMMON | 1<<STT_TLS)
234
#define OK_BINDS (1<<STB_GLOBAL | 1<<STB_WEAK | 1<<STB_GNU_UNIQUE)
R
Rich Felker 已提交
235

236 237 238 239
#ifndef ARCH_SYM_REJECT_UND
#define ARCH_SYM_REJECT_UND(s) 0
#endif

240
static struct symdef find_sym(struct dso *dso, const char *s, int need_def)
R
Rich Felker 已提交
241
{
242
	uint32_t h = 0, gh = 0;
243
	struct symdef def = {0};
R
Rich Felker 已提交
244
	for (; dso; dso=dso->next) {
245 246
		Sym *sym;
		if (!dso->global) continue;
247 248 249 250 251 252 253
		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);
		}
254 255
		if (!sym) continue;
		if (!sym->st_shndx)
256 257
			if (need_def || (sym->st_info&0xf) == STT_TLS
			    || ARCH_SYM_REJECT_UND(sym))
258 259 260 261 262 263 264 265 266 267 268
				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 已提交
269
	}
270
	return def;
R
Rich Felker 已提交
271 272
}

273 274
#define NO_INLINE_ADDEND (1<<REL_COPY | 1<<REL_GOT | 1<<REL_PLT)

275 276
ptrdiff_t __tlsdesc_static(), __tlsdesc_dynamic();

277
static void do_relocs(struct dso *dso, size_t *rel, size_t rel_size, size_t stride)
R
Rich Felker 已提交
278
{
279 280 281
	unsigned char *base = dso->base;
	Sym *syms = dso->syms;
	char *strings = dso->strings;
R
Rich Felker 已提交
282 283 284
	Sym *sym;
	const char *name;
	void *ctx;
285
	int astype, type;
R
Rich Felker 已提交
286
	int sym_index;
287
	struct symdef def;
288 289 290 291
	size_t *reloc_addr;
	size_t sym_val;
	size_t tls_val;
	size_t addend;
R
Rich Felker 已提交
292 293

	for (; rel_size; rel+=stride, rel_size-=stride*sizeof(size_t)) {
294 295 296 297
		astype = R_TYPE(rel[1]);
		if (!astype) continue;
		type = remap_rel(astype);
		if (!type) {
298
			error("Error relocating %s: unsupported relocation type %d",
299 300 301
				dso->name, astype);
			continue;
		}
R
Rich Felker 已提交
302
		sym_index = R_SYM(rel[1]);
303
		reloc_addr = (void *)(base + rel[0]);
R
Rich Felker 已提交
304 305 306
		if (sym_index) {
			sym = syms + sym_index;
			name = strings + sym->st_name;
307 308
			ctx = type==REL_COPY ? head->next : head;
			def = find_sym(ctx, name, type==REL_PLT);
309 310
			if (!def.sym && (sym->st_shndx != SHN_UNDEF
			    || sym->st_info>>4 != STB_WEAK)) {
311
				error("Error relocating %s: %s: symbol not found",
312
					dso->name, name);
313
				continue;
R
Rich Felker 已提交
314
			}
315
		} else {
316 317
			sym = 0;
			def.sym = 0;
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 353 354 355 356 357 358 359 360 361 362 363 364 365 366 367
			def.dso = dso;
		}

		addend = stride>2 ? rel[2]
			: (1<<type & NO_INLINE_ADDEND) ? 0
			: *reloc_addr;

		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) {
		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
368 369 370 371
		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);
372
				if (!new) error(
373 374 375 376 377 378 379 380 381 382 383 384 385 386 387 388 389 390 391
					"Error relocating %s: cannot allocate TLSDESC for %s",
					dso->name, sym ? name : "(local)" );
				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 已提交
392 393 394 395
		}
	}
}

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

418
static void reclaim_gaps(struct dso *dso)
419
{
420 421
	Phdr *ph = dso->phdr;
	size_t phcnt = dso->phnum;
422

423
	for (; phcnt--; ph=(void *)((char *)ph+dso->phentsize)) {
424 425
		if (ph->p_type!=PT_LOAD) continue;
		if ((ph->p_flags&(PF_R|PF_W))!=(PF_R|PF_W)) continue;
T
Timo Teräs 已提交
426 427
		reclaim(dso, ph->p_vaddr & -PAGE_SIZE, ph->p_vaddr);
		reclaim(dso, ph->p_vaddr+ph->p_memsz,
428 429 430 431
			ph->p_vaddr+ph->p_memsz+PAGE_SIZE-1 & -PAGE_SIZE);
	}
}

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

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

565
static int path_open(const char *name, const char *s, char *buf, size_t buf_size)
566
{
567 568
	size_t l;
	int fd;
569
	for (;;) {
570 571 572 573 574
		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;
575
		if ((fd = open(buf, O_RDONLY|O_CLOEXEC))>=0) return fd;
576
		s += l;
577 578 579
	}
}

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

	if (p->kernel_mapped) {
		/* $ORIGIN searches cannot be performed for the main program
		 * when it is suid/sgid/AT_SECURE. This is because the
		 * pathname is under the control of the caller of execve.
		 * For libraries, however, $ORIGIN can be processed safely
		 * since the library's pathname came from a trusted source
		 * (either system paths or a call to dlopen). */
		if (libc.secure)
			return -1;
R
Rich Felker 已提交
610 611
		l = readlink("/proc/self/exe", buf, buf_size);
		if (l >= buf_size)
612
			return -1;
R
Rich Felker 已提交
613
		buf[l] = 0;
614 615 616 617 618 619 620 621 622 623 624
		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 已提交
625
	while ((t=strchr(s, '$'))) {
626 627 628 629
		memcpy(d, s, t-s);
		d += t-s;
		memcpy(d, origin, l);
		d += l;
R
Rich Felker 已提交
630 631
		/* It was determined previously that the '$' is followed
		 * either by "ORIGIN" or "{ORIGIN}". */
632 633 634 635 636 637
		s = t + 7 + 2*(t[1]=='{');
	}
	strcpy(d, s);
	return 0;
}

638 639 640 641 642 643
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]);
644 645
	if (dyn[0]&(1<<DT_HASH))
		p->hashtab = (void *)(p->base + dyn[DT_HASH]);
646
	if (dyn[0]&(1<<DT_RPATH))
647
		p->rpath_orig = (void *)(p->strings + dyn[DT_RPATH]);
648 649
	if (search_vec(p->dynv, dyn, DT_GNU_HASH))
		p->ghashtab = (void *)(p->base + *dyn);
650 651
	if (search_vec(p->dynv, dyn, DT_VERSYM))
		p->versym = (void *)(p->base + *dyn);
652 653
}

654
static struct dso *load_library(const char *name, struct dso *needed_by)
R
Rich Felker 已提交
655
{
656
	char buf[2*NAME_MAX+2];
657
	const char *pathname;
658
	unsigned char *map;
659
	struct dso *p, temp_dso = {0};
R
Rich Felker 已提交
660 661
	int fd;
	struct stat st;
662 663
	size_t alloc_size;
	int n_th = 0;
664
	int is_self = 0;
R
Rich Felker 已提交
665 666 667 668 669 670 671 672

	/* 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;
673
			for (rp=reserved; *rp && strncmp(name+3, rp, l-3); rp+=strlen(rp)+1);
R
Rich Felker 已提交
674
			if (*rp) {
675 676 677 678 679 680 681 682 683 684 685 686
				if (ldd_mode) {
					/* Track which names have been resolved
					 * and only report each one once. */
					static unsigned reported;
					unsigned mask = 1U<<(rp-reserved);
					if (!(reported & mask)) {
						reported |= mask;
						dprintf(1, "\t%s => %s (%p)\n",
							name, ldso->name,
							ldso->base);
					}
				}
687
				is_self = 1;
R
Rich Felker 已提交
688 689 690
			}
		}
	}
691 692 693 694 695 696 697 698 699
	if (!strcmp(name, ldso->name)) is_self = 1;
	if (is_self) {
		if (!ldso->prev) {
			tail->next = ldso;
			ldso->prev = tail;
			tail = ldso->next ? ldso->next : ldso;
		}
		return ldso;
	}
700
	if (strchr(name, '/')) {
701
		pathname = name;
702
		fd = open(name, O_RDONLY|O_CLOEXEC);
R
Rich Felker 已提交
703
	} else {
704 705 706 707 708 709 710
		/* 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;
			}
		}
711
		if (strlen(name) > NAME_MAX) return 0;
712
		fd = -1;
713
		if (env_path) fd = path_open(name, env_path, buf, sizeof buf);
714
		for (p=needed_by; fd < 0 && p; p=p->needed_by)
715
			if (!fixup_rpath(p, buf, sizeof buf))
716
				fd = path_open(name, p->rpath, buf, sizeof buf);
717 718
		if (fd < 0) {
			if (!sys_path) {
719 720 721 722 723 724 725 726 727 728 729 730 731 732 733 734 735 736 737 738
				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");
739
				if (f) {
740
					if (getdelim(&sys_path, (size_t[1]){0}, 0, f) <= 0) {
741
						free(sys_path);
742
						sys_path = "";
743
					}
744
					fclose(f);
745 746
				} else if (errno != ENOENT) {
					sys_path = "";
747 748
				}
			}
749 750
			if (!sys_path) sys_path = "/lib:/usr/local/lib:/usr/lib";
			fd = path_open(name, sys_path, buf, sizeof buf);
R
Rich Felker 已提交
751
		}
752
		pathname = buf;
R
Rich Felker 已提交
753 754 755 756 757 758 759 760
	}
	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) {
761 762 763
			/* 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. */
764 765
			if (!p->shortname && pathname != name)
				p->shortname = strrchr(p->name, '/')+1;
R
Rich Felker 已提交
766 767 768 769 770
			close(fd);
			p->refcnt++;
			return p;
		}
	}
771
	map = noload ? 0 : map_library(fd, &temp_dso);
R
Rich Felker 已提交
772 773
	close(fd);
	if (!map) return 0;
774 775 776 777 778 779 780 781 782 783

	/* 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);
784
		n_th = libc.threads_minus_1 + 1;
785 786 787 788
		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 已提交
789
	if (!p) {
790
		munmap(map, temp_dso.map_len);
R
Rich Felker 已提交
791 792
		return 0;
	}
793
	memcpy(p, &temp_dso, sizeof temp_dso);
794
	decode_dyn(p);
R
Rich Felker 已提交
795 796 797
	p->dev = st.st_dev;
	p->ino = st.st_ino;
	p->refcnt = 1;
798
	p->needed_by = needed_by;
R
Rich Felker 已提交
799
	p->name = p->buf;
800 801 802
	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;
803
	if (p->tls_image) {
804
		if (runtime && !libc.has_thread_pointer) {
805
			munmap(map, p->map_len);
806
			free(p);
807
			errno = ENOSYS;
808 809
			return 0;
		}
810
		p->tls_id = ++tls_cnt;
R
Rich Felker 已提交
811
		tls_align = MAXP2(tls_align, p->tls_align);
812 813 814 815 816
#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 已提交
817 818 819 820
		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;
821
#endif
822 823 824 825
		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 已提交
826 827 828 829 830

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

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

R
Rich Felker 已提交
833 834 835 836 837
	return p;
}

static void load_deps(struct dso *p)
{
838 839
	size_t i, ndeps=0;
	struct dso ***deps = &p->deps, **tmp, *dep;
R
Rich Felker 已提交
840 841 842
	for (; p; p=p->next) {
		for (i=0; p->dynv[i]; i+=2) {
			if (p->dynv[i] != DT_NEEDED) continue;
843
			dep = load_library(p->strings + p->dynv[i+1], p);
844
			if (!dep) {
845
				error("Error loading shared library %s: %m (needed by %s)",
R
Rich Felker 已提交
846
					p->strings + p->dynv[i+1], p->name);
847
				continue;
R
Rich Felker 已提交
848
			}
849 850
			if (runtime) {
				tmp = realloc(*deps, sizeof(*tmp)*(ndeps+2));
851
				if (!tmp) longjmp(*rtld_fail, 1);
852 853 854 855
				tmp[ndeps++] = dep;
				tmp[ndeps] = 0;
				*deps = tmp;
			}
R
Rich Felker 已提交
856 857 858 859
		}
	}
}

R
Rich Felker 已提交
860 861 862 863 864
static void load_preload(char *s)
{
	int tmp;
	char *z;
	for (z=s; *z; s=z) {
865 866
		for (   ; *s && (isspace(*s) || *s==':'); s++);
		for (z=s; *z && !isspace(*z) && *z!=':'; z++);
R
Rich Felker 已提交
867 868
		tmp = *z;
		*z = 0;
869
		load_library(s, 0);
R
Rich Felker 已提交
870 871 872 873
		*z = tmp;
	}
}

874 875 876 877 878
static void make_global(struct dso *p)
{
	for (; p; p=p->next) p->global = 1;
}

R
Rich Felker 已提交
879 880 881 882 883 884
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 已提交
885 886 887
#ifdef NEED_ARCH_RELOCS
		do_arch_relocs(p, head);
#endif
888 889 890 891
		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 已提交
892 893 894

		if (p->relro_start != p->relro_end &&
		    mprotect(p->base+p->relro_start, p->relro_end-p->relro_start, PROT_READ) < 0) {
895
			error("Error relocating %s: RELRO protection failed: %m",
T
Timo Teräs 已提交
896 897 898
				p->name);
		}

899
		p->relocated = 1;
R
Rich Felker 已提交
900 901 902
	}
}

903
static void kernel_mapped_dso(struct dso *p)
904
{
905 906 907 908 909 910
	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 已提交
911 912 913
			p->relro_start = ph->p_vaddr & -PAGE_SIZE;
			p->relro_end = (ph->p_vaddr + ph->p_memsz) & -PAGE_SIZE;
		}
914 915 916 917 918 919 920 921 922 923
		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;
924
	p->kernel_mapped = 1;
925 926
}

927 928 929 930 931 932 933
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);
934 935
		if (dyn[0] & (1<<DT_FINI_ARRAY)) {
			size_t n = dyn[DT_FINI_ARRAYSZ]/sizeof(size_t);
936 937
			size_t *fn = (size_t *)(p->base + dyn[DT_FINI_ARRAY])+n;
			while (n--) ((void (*)(void))*--fn)();
938
		}
939
#ifndef NO_LEGACY_INITFINI
940
		if ((dyn[0] & (1<<DT_FINI)) && dyn[DT_FINI])
941
			((void (*)(void))(p->base + dyn[DT_FINI]))();
942
#endif
943 944 945
	}
}

946 947 948
static void do_init_fini(struct dso *p)
{
	size_t dyn[DYN_CNT] = {0};
949
	int need_locking = libc.threads_minus_1;
950 951 952 953
	/* 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);
954
	for (; p; p=p->prev) {
955 956
		if (p->constructed) continue;
		p->constructed = 1;
957
		decode_vec(p->dynv, dyn, DYN_CNT);
958
		if (dyn[0] & ((1<<DT_FINI) | (1<<DT_FINI_ARRAY))) {
959 960 961
			p->fini_next = fini_head;
			fini_head = p;
		}
962
#ifndef NO_LEGACY_INITFINI
963
		if ((dyn[0] & (1<<DT_INIT)) && dyn[DT_INIT])
964
			((void (*)(void))(p->base + dyn[DT_INIT]))();
965
#endif
966 967 968 969 970
		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++)();
		}
971 972 973 974
		if (!need_locking && libc.threads_minus_1) {
			need_locking = 1;
			pthread_mutex_lock(&init_fini_lock);
		}
975
	}
976
	if (need_locking) pthread_mutex_unlock(&init_fini_lock);
977 978
}

979 980 981 982
void _dl_debug_state(void)
{
}

983 984 985 986 987 988 989 990 991 992 993 994 995
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;
	}
}

996
void *__copy_tls(unsigned char *mem)
997
{
R
Rich Felker 已提交
998
	pthread_t td;
999
	struct dso *p;
R
Rich Felker 已提交
1000

1001
	void **dtv = (void *)mem;
1002
	dtv[0] = (void *)tls_cnt;
1003 1004 1005 1006 1007
	if (!tls_cnt) {
		td = (void *)(dtv+1);
		td->dtv = dtv;
		return td;
	}
R
Rich Felker 已提交
1008

1009 1010 1011 1012 1013 1014 1015 1016 1017 1018 1019 1020
#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
1021
	mem += libc.tls_size - sizeof(struct pthread);
R
Rich Felker 已提交
1022 1023 1024 1025
	mem -= (uintptr_t)mem & (tls_align-1);
	td = (pthread_t)mem;

	for (p=head; p; p=p->next) {
1026
		if (!p->tls_id) continue;
R
Rich Felker 已提交
1027 1028
		dtv[p->tls_id] = mem - p->tls_offset;
		memcpy(dtv[p->tls_id], p->tls_image, p->tls_len);
1029
	}
1030
#endif
R
Rich Felker 已提交
1031 1032
	td->dtv = dtv;
	return td;
1033 1034
}

1035
void *__tls_get_new(size_t *v)
1036 1037
{
	pthread_t self = __pthread_self();
1038 1039 1040

	/* Block signals to make accessing new TLS async-signal-safe */
	sigset_t set;
1041
	__block_all_sigs(&set);
1042
	if (v[0]<=(size_t)self->dtv[0]) {
1043
		__restore_sigs(&set);
1044 1045 1046 1047 1048 1049 1050 1051 1052 1053 1054
		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 */
1055
	if (v[0] > (size_t)self->dtv[0]) {
1056 1057
		void **newdtv = p->new_dtv +
			(v[0]+1)*sizeof(void *)*a_fetch_add(&p->new_dtv_idx,1);
1058
		memcpy(newdtv, self->dtv,
1059 1060 1061
			((size_t)self->dtv[0]+1) * sizeof(void *));
		newdtv[0] = (void *)v[0];
		self->dtv = newdtv;
1062
	}
1063

1064 1065 1066 1067 1068 1069 1070 1071 1072 1073 1074 1075
	/* 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;
	}
1076
	__restore_sigs(&set);
1077
	return mem + v[1];
1078 1079
}

R
Rich Felker 已提交
1080 1081
static void update_tls_size()
{
1082 1083 1084 1085 1086 1087
	libc.tls_size = ALIGN(
		(1+tls_cnt) * sizeof(void *) +
		tls_offset +
		sizeof(struct pthread) +
		tls_align * 2,
	tls_align);
R
Rich Felker 已提交
1088 1089
}

1090
void *__dynlink(int argc, char **argv)
R
Rich Felker 已提交
1091
{
1092
	size_t aux[AUX_CNT] = {0};
R
Rich Felker 已提交
1093 1094
	size_t i;
	Phdr *phdr;
1095
	Ehdr *ehdr;
1096
	static struct dso builtin_dsos[3];
R
Rich Felker 已提交
1097 1098
	struct dso *const app = builtin_dsos+0;
	struct dso *const lib = builtin_dsos+1;
1099
	struct dso *const vdso = builtin_dsos+2;
R
Rich Felker 已提交
1100
	char *env_preload=0;
R
Rich Felker 已提交
1101
	size_t vdso_base;
1102
	size_t *auxv;
1103
	char **envp = argv+argc+1;
1104
	void *initial_tls;
R
Rich Felker 已提交
1105 1106

	/* Find aux vector just past environ[] */
1107 1108 1109
	for (i=argc+1; argv[i]; i++)
		if (!memcmp(argv[i], "LD_LIBRARY_PATH=", 16))
			env_path = argv[i]+16;
R
Rich Felker 已提交
1110 1111
		else if (!memcmp(argv[i], "LD_PRELOAD=", 11))
			env_preload = argv[i]+11;
R
Rich Felker 已提交
1112 1113 1114 1115
	auxv = (void *)(argv+i+1);

	decode_vec(auxv, aux, AUX_CNT);

1116 1117
	/* 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 已提交
1118
	  || aux[AT_GID]!=aux[AT_EGID] || aux[AT_SECURE]) {
1119
		env_path = 0;
R
Rich Felker 已提交
1120
		env_preload = 0;
1121
		libc.secure = 1;
1122
	}
1123
	libc.page_size = aux[AT_PAGESZ];
1124

1125 1126 1127 1128 1129 1130 1131 1132 1133
	/* 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;
	}

1134 1135 1136
	/* The dynamic linker load address is passed by the kernel
	 * in the AUX vector, so this is easy. */
	lib->base = (void *)aux[AT_BASE];
1137
	lib->name = lib->shortname = "libc.so";
1138 1139
	lib->global = 1;
	ehdr = (void *)lib->base;
R
Rich Felker 已提交
1140 1141
	lib->phnum = ehdr->e_phnum;
	lib->phdr = (void *)(aux[AT_BASE]+ehdr->e_phoff);
1142 1143
	lib->phentsize = ehdr->e_phentsize;
	kernel_mapped_dso(lib);
1144 1145
	decode_dyn(lib);

1146
	if (aux[AT_PHDR]) {
1147
		size_t interp_off = 0;
1148
		size_t tls_image = 0;
1149
		/* Find load address of the main program, via AT_PHDR vs PT_PHDR. */
R
Rich Felker 已提交
1150 1151
		app->phdr = phdr = (void *)aux[AT_PHDR];
		app->phnum = aux[AT_PHNUM];
1152
		app->phentsize = aux[AT_PHENT];
1153 1154 1155
		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);
1156 1157
			else if (phdr->p_type == PT_INTERP)
				interp_off = (size_t)phdr->p_vaddr;
1158 1159 1160 1161 1162 1163
			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;
			}
1164
		}
1165
		if (app->tls_size) app->tls_image = (char *)app->base + tls_image;
1166
		if (interp_off) lib->name = (char *)app->base + interp_off;
1167 1168 1169 1170 1171
		if ((aux[0] & (1UL<<AT_EXECFN))
		    && strncmp((char *)aux[AT_EXECFN], "/proc/", 6))
			app->name = (char *)aux[AT_EXECFN];
		else
			app->name = argv[0];
1172
		kernel_mapped_dso(app);
1173 1174 1175
	} else {
		int fd;
		char *ldname = argv[0];
R
Rich Felker 已提交
1176
		size_t l = strlen(ldname);
1177 1178
		if (l >= 3 && !strcmp(ldname+l-3, "ldd")) ldd_mode = 1;
		*argv++ = (void *)-1;
1179 1180 1181 1182 1183 1184 1185 1186 1187 1188 1189 1190 1191 1192 1193 1194 1195 1196 1197 1198
		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;
			}
			argv[-1] = (void *)-1;
		}
1199
		if (!argv[0]) {
1200 1201 1202
			dprintf(2, "musl libc\n"
				"Version %s\n"
				"Dynamic Program Loader\n"
1203
				"Usage: %s [options] [--] pathname%s\n",
1204
				__libc_get_version(), ldname,
1205 1206 1207 1208 1209 1210 1211 1212 1213
				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;
1214
		ehdr = (void *)map_library(fd, app);
1215 1216 1217 1218 1219 1220
		if (!ehdr) {
			dprintf(2, "%s: %s: Not a valid dynamic program\n", ldname, argv[0]);
			_exit(1);
		}
		runtime = 0;
		close(fd);
1221
		lib->name = ldname;
1222
		app->name = argv[0];
1223
		aux[AT_ENTRY] = (size_t)app->base + ehdr->e_entry;
1224 1225 1226 1227 1228 1229 1230 1231 1232 1233
		/* Find the name that would have been used for the dynamic
		 * linker had ldd not taken its place. */
		if (ldd_mode) {
			for (i=0; i<app->phnum; i++) {
				if (app->phdr[i].p_type == PT_INTERP)
					lib->name = (void *)(app->base
						+ app->phdr[i].p_vaddr);
			}
			dprintf(1, "\t%s (%p)\n", lib->name, lib->base);
		}
1234
	}
1235
	if (app->tls_size) {
R
Rich Felker 已提交
1236
		app->tls_id = tls_cnt = 1;
1237 1238 1239 1240 1241 1242
#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
1243 1244 1245
		tls_offset = app->tls_offset = app->tls_size
			+ ( -((uintptr_t)app->tls_image + app->tls_size)
			& (app->tls_align-1) );
1246
#endif
R
Rich Felker 已提交
1247
		tls_align = MAXP2(tls_align, app->tls_align);
1248
	}
1249 1250 1251 1252
	app->global = 1;
	decode_dyn(app);

	/* Attach to vdso, if provided by the kernel */
R
Rich Felker 已提交
1253
	if (search_vec(auxv, &vdso_base, AT_SYSINFO_EHDR)) {
1254
		ehdr = (void *)vdso_base;
R
Rich Felker 已提交
1255 1256
		vdso->phdr = phdr = (void *)(vdso_base + ehdr->e_phoff);
		vdso->phnum = ehdr->e_phnum;
1257
		vdso->phentsize = ehdr->e_phentsize;
1258 1259 1260 1261 1262 1263
		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);
		}
1264 1265
		vdso->name = "";
		vdso->shortname = "linux-gate.so.1";
1266
		vdso->global = 1;
1267
		decode_dyn(vdso);
1268 1269 1270 1271
		vdso->prev = lib;
		lib->next = vdso;
	}

1272 1273 1274 1275 1276
	/* 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;
1277
	ldso = lib;
1278 1279 1280
	app->next = lib;
	reloc_all(lib);
	app->next = 0;
R
Rich Felker 已提交
1281

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

1284
	/* Donate unused parts of app and library mapping to malloc */
1285 1286
	reclaim_gaps(app);
	reclaim_gaps(lib);
1287

1288
	/* Load preload/needed libraries, add their symbols to the global
1289 1290 1291
	 * 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 已提交
1292
	if (env_preload) load_preload(env_preload);
1293 1294
	load_deps(app);
	make_global(app);
1295

T
Timo Teräs 已提交
1296 1297 1298 1299 1300 1301
#ifndef DYNAMIC_IS_RO
	for (i=0; app->dynv[i]; i+=2)
		if (app->dynv[i]==DT_DEBUG)
			app->dynv[i+1] = (size_t)&debug;
#endif

R
Rich Felker 已提交
1302 1303 1304 1305
	reloc_all(app->next);
	reloc_all(app);

	update_tls_size();
1306 1307 1308
	if (libc.tls_size > sizeof builtin_tls) {
		initial_tls = calloc(libc.tls_size, 1);
		if (!initial_tls) {
1309
			dprintf(2, "%s: Error getting %zu bytes thread-local storage: %m\n",
1310
				argv[0], libc.tls_size);
1311 1312
			_exit(127);
		}
1313 1314 1315 1316 1317 1318
	} else {
		initial_tls = builtin_tls;
	}
	if (__init_tp(__copy_tls(initial_tls)) < 0 && tls_cnt) {
		dprintf(2, "%s: Thread-local storage not supported by kernel.\n", argv[0]);
		_exit(127);
1319
	}
1320
	static_tls_cnt = tls_cnt;
1321

1322
	if (ldso_fail) _exit(127);
1323 1324
	if (ldd_mode) _exit(0);

1325 1326 1327 1328
	/* 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 已提交
1329
	runtime = 1;
1330

1331 1332 1333 1334 1335 1336 1337
	debug.ver = 1;
	debug.bp = _dl_debug_state;
	debug.head = head;
	debug.base = lib->base;
	debug.state = 0;
	_dl_debug_state();

1338
	__init_libc(envp, argv[0]);
1339
	atexit(do_fini);
1340
	errno = 0;
1341
	do_init_fini(tail);
1342 1343

	return (void *)aux[AT_ENTRY];
1344 1345
}

1346 1347
void *dlopen(const char *file, int mode)
{
R
Rich Felker 已提交
1348
	struct dso *volatile p, *orig_tail, *next;
R
Rich Felker 已提交
1349
	size_t orig_tls_cnt, orig_tls_offset, orig_tls_align;
1350
	size_t i;
1351
	int cs;
1352
	jmp_buf jb;
1353 1354 1355

	if (!file) return head;

1356
	pthread_setcancelstate(PTHREAD_CANCEL_DISABLE, &cs);
1357
	pthread_rwlock_wrlock(&lock);
1358
	__inhibit_ptc();
1359

1360 1361
	p = 0;
	orig_tls_cnt = tls_cnt;
R
Rich Felker 已提交
1362 1363
	orig_tls_offset = tls_offset;
	orig_tls_align = tls_align;
R
Rich Felker 已提交
1364
	orig_tail = tail;
1365
	noload = mode & RTLD_NOLOAD;
R
Rich Felker 已提交
1366

1367 1368
	rtld_fail = &jb;
	if (setjmp(*rtld_fail)) {
1369
		/* Clean up anything new that was (partially) loaded */
1370
		if (p && p->deps) for (i=0; p->deps[i]; i++)
1371 1372
			if (p->deps[i]->global < 0)
				p->deps[i]->global = 0;
1373 1374 1375
		for (p=orig_tail->next; p; p=next) {
			next = p->next;
			munmap(p->map, p->map_len);
1376 1377 1378 1379 1380
			while (p->td_index) {
				void *tmp = p->td_index->next;
				free(p->td_index);
				p->td_index = tmp;
			}
1381 1382 1383
			free(p->deps);
			free(p);
		}
1384
		tls_cnt = orig_tls_cnt;
R
Rich Felker 已提交
1385 1386
		tls_offset = orig_tls_offset;
		tls_align = orig_tls_align;
1387 1388
		tail = orig_tail;
		tail->next = 0;
1389
		p = 0;
1390 1391
		errflag = 1;
		goto end;
1392
	} else p = load_library(file, head);
R
Rich Felker 已提交
1393 1394

	if (!p) {
1395 1396 1397 1398
		snprintf(errbuf, sizeof errbuf, noload ?
			"Library %s is not already loaded" :
			"Error loading shared library %s: %m",
			file);
R
Rich Felker 已提交
1399
		errflag = 1;
1400
		goto end;
1401 1402 1403 1404 1405
	}

	/* First load handling */
	if (!p->deps) {
		load_deps(p);
R
Rich Felker 已提交
1406
		if (p->deps) for (i=0; p->deps[i]; i++)
1407 1408 1409
			if (!p->deps[i]->global)
				p->deps[i]->global = -1;
		if (!p->global) p->global = -1;
1410
		reloc_all(p);
R
Rich Felker 已提交
1411
		if (p->deps) for (i=0; p->deps[i]; i++)
1412 1413 1414
			if (p->deps[i]->global < 0)
				p->deps[i]->global = 0;
		if (p->global < 0) p->global = 0;
1415 1416 1417
	}

	if (mode & RTLD_GLOBAL) {
R
Rich Felker 已提交
1418
		if (p->deps) for (i=0; p->deps[i]; i++)
1419 1420 1421 1422
			p->deps[i]->global = 1;
		p->global = 1;
	}

R
Rich Felker 已提交
1423
	update_tls_size();
1424
	_dl_debug_state();
1425
	orig_tail = tail;
1426
end:
1427
	__release_ptc();
R
Rich Felker 已提交
1428
	if (p) gencnt++;
1429
	pthread_rwlock_unlock(&lock);
1430
	if (p) do_init_fini(orig_tail);
1431
	pthread_setcancelstate(cs, 0);
1432 1433 1434
	return p;
}

1435
static int invalid_dso_handle(void *h)
1436 1437 1438 1439 1440 1441 1442 1443
{
	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;
}

1444 1445
void *__tls_get_addr(size_t *);

R
Rich Felker 已提交
1446
static void *do_dlsym(struct dso *p, const char *s, void *ra)
1447 1448
{
	size_t i;
1449
	uint32_t h = 0, gh = 0;
1450
	Sym *sym;
1451
	if (p == head || p == RTLD_DEFAULT || p == RTLD_NEXT) {
1452 1453 1454
		if (p == RTLD_DEFAULT) {
			p = head;
		} else if (p == RTLD_NEXT) {
1455 1456
			for (p=head; p && (unsigned char *)ra-p->map>p->map_len; p=p->next);
			if (!p) p=head;
1457
			p = p->next;
1458
		}
1459
		struct symdef def = find_sym(p, s, 0);
1460
		if (!def.sym) goto failed;
1461 1462
		if ((def.sym->st_info&0xf) == STT_TLS)
			return __tls_get_addr((size_t []){def.dso->tls_id, def.sym->st_value});
1463
		return def.dso->base + def.sym->st_value;
R
Rich Felker 已提交
1464
	}
1465 1466
	if (p != RTLD_DEFAULT && p != RTLD_NEXT && invalid_dso_handle(p))
		return 0;
1467 1468 1469 1470 1471 1472 1473
	if (p->ghashtab) {
		gh = gnu_hash(s);
		sym = gnu_lookup(s, gh, p);
	} else {
		h = sysv_hash(s);
		sym = sysv_lookup(s, h, p);
	}
1474 1475
	if (sym && (sym->st_info&0xf) == STT_TLS)
		return __tls_get_addr((size_t []){p->tls_id, sym->st_value});
1476 1477 1478
	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++) {
1479 1480
		if (p->deps[i]->ghashtab) {
			if (!gh) gh = gnu_hash(s);
1481
			sym = gnu_lookup(s, gh, p->deps[i]);
1482 1483 1484 1485
		} else {
			if (!h) h = sysv_hash(s);
			sym = sysv_lookup(s, h, p->deps[i]);
		}
1486 1487
		if (sym && (sym->st_info&0xf) == STT_TLS)
			return __tls_get_addr((size_t []){p->deps[i]->tls_id, sym->st_value});
1488 1489 1490
		if (sym && sym->st_value && (1<<(sym->st_info&0xf) & OK_TYPES))
			return p->deps[i]->base + sym->st_value;
	}
1491
failed:
R
Rich Felker 已提交
1492
	errflag = 1;
1493
	snprintf(errbuf, sizeof errbuf, "Symbol not found: %s", s);
1494 1495 1496
	return 0;
}

1497
int __dladdr(const void *addr, Dl_info *info)
1498 1499 1500 1501 1502 1503 1504 1505 1506 1507 1508 1509 1510 1511 1512 1513 1514 1515 1516 1517 1518 1519 1520 1521
{
	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 已提交
1522
		for (i = nsym = 0; i < p->ghashtab[0]; i++) {
1523 1524 1525 1526 1527 1528 1529 1530 1531 1532 1533 1534
			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++) {
1535
		if (sym->st_value
1536 1537 1538 1539 1540 1541 1542 1543 1544 1545 1546 1547 1548 1549 1550 1551 1552 1553 1554 1555 1556 1557
		 && (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;
}

1558
void *__dlsym(void *restrict p, const char *restrict s, void *restrict ra)
1559 1560 1561
{
	void *res;
	pthread_rwlock_rdlock(&lock);
R
Rich Felker 已提交
1562
	res = do_dlsym(p, s, ra);
1563 1564 1565
	pthread_rwlock_unlock(&lock);
	return res;
}
R
Rich Felker 已提交
1566 1567 1568 1569 1570 1571 1572 1573 1574 1575 1576 1577 1578 1579 1580 1581 1582 1583 1584 1585 1586 1587 1588 1589 1590 1591

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;
}
1592
#else
1593
static int invalid_dso_handle(void *h)
1594 1595 1596 1597 1598
{
	snprintf(errbuf, sizeof errbuf, "Invalid library handle %p", (void *)h);
	errflag = 1;
	return 1;
}
1599 1600 1601 1602
void *dlopen(const char *file, int mode)
{
	return 0;
}
1603
void *__dlsym(void *restrict p, const char *restrict s, void *restrict ra)
1604 1605 1606
{
	return 0;
}
1607
int __dladdr (const void *addr, Dl_info *info)
1608 1609 1610
{
	return 0;
}
1611
#endif
1612

R
Rich Felker 已提交
1613 1614 1615 1616 1617 1618 1619 1620 1621 1622 1623 1624
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;
}

1625 1626
char *dlerror()
{
R
Rich Felker 已提交
1627 1628
	if (!errflag) return 0;
	errflag = 0;
1629
	return errbuf;
1630 1631 1632 1633
}

int dlclose(void *p)
{
1634
	return invalid_dso_handle(p);
1635
}