blob: f2bcd0f3d401e8dca13178d3b1a8dba64866dd0d [file] [log] [blame]
Rich Felkerf419bcb2012-08-26 21:09:26 -04001#define _GNU_SOURCE
Rich Felker51e2d832011-06-18 19:48:42 -04002#include <stdio.h>
3#include <stdlib.h>
4#include <string.h>
5#include <unistd.h>
6#include <stdint.h>
7#include <elf.h>
8#include <sys/mman.h>
9#include <limits.h>
10#include <stdint.h>
11#include <fcntl.h>
12#include <sys/stat.h>
13#include <errno.h>
14#include <limits.h>
15#include <elf.h>
Rich Felker6b3d5e52011-06-26 17:39:17 -040016#include <setjmp.h>
Rich Felker59ab43f2011-06-26 19:23:28 -040017#include <pthread.h>
Rich Felker2719cc82011-08-16 00:24:36 -040018#include <ctype.h>
Rich Felker59ab43f2011-06-26 19:23:28 -040019#include <dlfcn.h>
Rich Felker8431d792012-10-04 16:35:46 -040020#include "pthread_impl.h"
21#include "libc.h"
22#undef libc
Rich Felker51e2d832011-06-18 19:48:42 -040023
Rich Felkera9e85c02012-03-23 00:28:20 -040024static int errflag;
Rich Felkera5d10eb2012-04-23 12:03:31 -040025static char errbuf[128];
Rich Felkera9e85c02012-03-23 00:28:20 -040026
Rich Felkere864a292012-07-11 01:47:30 -040027#ifdef SHARED
Rich Felkera9e85c02012-03-23 00:28:20 -040028
Rich Felker51e2d832011-06-18 19:48:42 -040029#if ULONG_MAX == 0xffffffff
30typedef Elf32_Ehdr Ehdr;
31typedef Elf32_Phdr Phdr;
32typedef Elf32_Sym Sym;
33#define R_TYPE(x) ((x)&255)
34#define R_SYM(x) ((x)>>8)
35#else
36typedef Elf64_Ehdr Ehdr;
37typedef Elf64_Phdr Phdr;
38typedef Elf64_Sym Sym;
39#define R_TYPE(x) ((x)&0xffffffff)
40#define R_SYM(x) ((x)>>32)
41#endif
42
Rich Felker3ec8d292012-04-25 00:05:42 -040043struct debug {
44 int ver;
45 void *head;
46 void (*bp)(void);
47 int state;
48 void *base;
49};
50
51struct dso {
52 unsigned char *base;
53 char *name;
Rich Felker51e2d832011-06-18 19:48:42 -040054 size_t *dynv;
Rich Felker3ec8d292012-04-25 00:05:42 -040055 struct dso *next, *prev;
56
57 int refcnt;
Rich Felker51e2d832011-06-18 19:48:42 -040058 Sym *syms;
Rich Felker596d60c2011-06-18 22:52:01 -040059 uint32_t *hashtab;
Rich Felker2bd05a42012-08-25 17:13:28 -040060 uint32_t *ghashtab;
Rich Felker51e2d832011-06-18 19:48:42 -040061 char *strings;
Rich Felker51e2d832011-06-18 19:48:42 -040062 unsigned char *map;
63 size_t map_len;
64 dev_t dev;
65 ino_t ino;
Rich Felker6343ac82012-06-09 21:20:44 -040066 signed char global;
Rich Felker700a8152012-02-07 20:29:29 -050067 char relocated;
68 char constructed;
Rich Felker59ab43f2011-06-26 19:23:28 -040069 struct dso **deps;
Rich Felkerbc6a35f2012-10-04 20:04:13 -040070 void *tls_image;
Rich Felker9c748562012-10-04 22:48:33 -040071 size_t tls_len, tls_size, tls_align, tls_id, tls_offset;
Rich Felkerdcd60372012-10-05 11:51:50 -040072 void **new_dtv;
73 unsigned char *new_tls;
74 int new_dtv_idx, new_tls_idx;
Rich Felker5c1909a2012-05-27 16:01:44 -040075 char *shortname;
Rich Felker6b3d5e52011-06-26 17:39:17 -040076 char buf[];
Rich Felker51e2d832011-06-18 19:48:42 -040077};
78
Rich Felker9c748562012-10-04 22:48:33 -040079struct symdef {
80 Sym *sym;
81 struct dso *dso;
82};
83
Rich Felker59f40862012-08-05 13:46:39 -040084#include "reloc.h"
85
Rich Felker58aa5f42012-05-03 20:42:45 -040086void __init_ssp(size_t *);
Rich Felkerbc6a35f2012-10-04 20:04:13 -040087void *__install_initial_tls(void *);
Rich Felker60872cf2012-04-24 18:07:59 -040088
Rich Felker51e2d832011-06-18 19:48:42 -040089static struct dso *head, *tail, *libc;
Rich Felker191ebca2011-06-30 23:02:27 -040090static char *env_path, *sys_path, *r_path;
Rich Felker60872cf2012-04-24 18:07:59 -040091static int ssp_used;
Rich Felker6b3d5e52011-06-26 17:39:17 -040092static int runtime;
Rich Felker5c1909a2012-05-27 16:01:44 -040093static int ldd_mode;
Rich Felker04109502012-08-18 16:00:23 -040094static int ldso_fail;
Rich Felker6b3d5e52011-06-26 17:39:17 -040095static jmp_buf rtld_fail;
Rich Felker59ab43f2011-06-26 19:23:28 -040096static pthread_rwlock_t lock;
Rich Felker3ec8d292012-04-25 00:05:42 -040097static struct debug debug;
Rich Felker731e8ff2012-08-25 17:24:46 -040098static size_t *auxv;
Rich Felkerbc6a35f2012-10-04 20:04:13 -040099static size_t tls_cnt, tls_size;
Rich Felker3ec8d292012-04-25 00:05:42 -0400100
101struct debug *_dl_debug_addr = &debug;
Rich Felker51e2d832011-06-18 19:48:42 -0400102
Rich Felkera0458832011-08-16 07:46:42 -0400103#define AUX_CNT 24
Rich Felker51e2d832011-06-18 19:48:42 -0400104#define DYN_CNT 34
105
106static void decode_vec(size_t *v, size_t *a, size_t cnt)
107{
108 memset(a, 0, cnt*sizeof(size_t));
109 for (; v[0]; v+=2) if (v[0]<cnt) {
110 a[0] |= 1ULL<<v[0];
111 a[v[0]] = v[1];
112 }
113}
114
Rich Felker2bd05a42012-08-25 17:13:28 -0400115static int search_vec(size_t *v, size_t *r, size_t key)
116{
117 for (; v[0]!=key; v+=2)
118 if (!v[0]) return 0;
119 *r = v[1];
120 return 1;
121}
122
123static uint32_t sysv_hash(const char *s0)
Rich Felker51e2d832011-06-18 19:48:42 -0400124{
Rich Felker2adf2fb2012-01-17 00:34:58 -0500125 const unsigned char *s = (void *)s0;
Rich Felker51e2d832011-06-18 19:48:42 -0400126 uint_fast32_t h = 0;
127 while (*s) {
128 h = 16*h + *s++;
129 h ^= h>>24 & 0xf0;
130 }
131 return h & 0xfffffff;
132}
133
Rich Felker2bd05a42012-08-25 17:13:28 -0400134static uint32_t gnu_hash(const char *s0)
135{
136 const unsigned char *s = (void *)s0;
137 uint_fast32_t h = 5381;
138 for (; *s; s++)
139 h = h*33 + *s;
140 return h;
141}
142
143static Sym *sysv_lookup(const char *s, uint32_t h, struct dso *dso)
Rich Felker51e2d832011-06-18 19:48:42 -0400144{
145 size_t i;
Rich Felker05eff012012-08-05 02:38:35 -0400146 Sym *syms = dso->syms;
147 uint32_t *hashtab = dso->hashtab;
148 char *strings = dso->strings;
Rich Felker51e2d832011-06-18 19:48:42 -0400149 for (i=hashtab[2+h%hashtab[0]]; i; i=hashtab[2+hashtab[0]+i]) {
150 if (!strcmp(s, strings+syms[i].st_name))
151 return syms+i;
152 }
153 return 0;
154}
155
Rich Felker2bd05a42012-08-25 17:13:28 -0400156static Sym *gnu_lookup(const char *s, uint32_t h1, struct dso *dso)
157{
158 Sym *sym;
159 char *strings;
160 uint32_t *hashtab = dso->ghashtab;
161 uint32_t nbuckets = hashtab[0];
162 uint32_t *buckets = hashtab + 4 + hashtab[2]*(sizeof(size_t)/4);
163 uint32_t h2;
164 uint32_t *hashval;
165 uint32_t n = buckets[h1 % nbuckets];
166
167 if (!n) return 0;
168
169 strings = dso->strings;
170 sym = dso->syms + n;
171 hashval = buckets + nbuckets + (n - hashtab[1]);
172
173 for (h1 |= 1; ; sym++) {
174 h2 = *hashval++;
175 if ((h1 == (h2|1)) && !strcmp(s, strings + sym->st_name))
176 return sym;
177 if (h2 & 1) break;
178 }
179
180 return 0;
181}
182
Rich Felker9c748562012-10-04 22:48:33 -0400183#define OK_TYPES (1<<STT_NOTYPE | 1<<STT_OBJECT | 1<<STT_FUNC | 1<<STT_COMMON | 1<<STT_TLS)
Rich Felker427173b2011-07-24 02:19:47 -0400184#define OK_BINDS (1<<STB_GLOBAL | 1<<STB_WEAK)
Rich Felker51e2d832011-06-18 19:48:42 -0400185
Rich Felker9c748562012-10-04 22:48:33 -0400186static struct symdef find_sym(struct dso *dso, const char *s, int need_def)
Rich Felker51e2d832011-06-18 19:48:42 -0400187{
Rich Felker2bd05a42012-08-25 17:13:28 -0400188 uint32_t h = 0, gh = 0;
Rich Felker9c748562012-10-04 22:48:33 -0400189 struct symdef def = {0};
Rich Felker2bd05a42012-08-25 17:13:28 -0400190 if (dso->ghashtab) {
191 gh = gnu_hash(s);
Rich Felker2bd05a42012-08-25 17:13:28 -0400192 if (gh == 0x1f4039c9 && !strcmp(s, "__stack_chk_fail")) ssp_used = 1;
193 } else {
194 h = sysv_hash(s);
Rich Felker2bd05a42012-08-25 17:13:28 -0400195 if (h == 0x595a4cc && !strcmp(s, "__stack_chk_fail")) ssp_used = 1;
196 }
Rich Felker51e2d832011-06-18 19:48:42 -0400197 for (; dso; dso=dso->next) {
Rich Felker59ab43f2011-06-26 19:23:28 -0400198 Sym *sym;
199 if (!dso->global) continue;
Rich Felker2bd05a42012-08-25 17:13:28 -0400200 if (dso->ghashtab) {
201 if (!gh) gh = gnu_hash(s);
202 sym = gnu_lookup(s, gh, dso);
203 } else {
204 if (!h) h = sysv_hash(s);
205 sym = sysv_lookup(s, h, dso);
206 }
Rich Felker51e2d832011-06-18 19:48:42 -0400207 if (sym && (!need_def || sym->st_shndx) && sym->st_value
Rich Felker427173b2011-07-24 02:19:47 -0400208 && (1<<(sym->st_info&0xf) & OK_TYPES)
209 && (1<<(sym->st_info>>4) & OK_BINDS)) {
Rich Felker9c748562012-10-04 22:48:33 -0400210 if (def.sym && sym->st_info>>4 == STB_WEAK) continue;
211 def.sym = sym;
212 def.dso = dso;
Rich Felker427173b2011-07-24 02:19:47 -0400213 if (sym->st_info>>4 == STB_GLOBAL) break;
214 }
Rich Felker51e2d832011-06-18 19:48:42 -0400215 }
Rich Felker427173b2011-07-24 02:19:47 -0400216 return def;
Rich Felker51e2d832011-06-18 19:48:42 -0400217}
218
Rich Felker87d13a42012-08-05 02:49:02 -0400219static void do_relocs(struct dso *dso, size_t *rel, size_t rel_size, size_t stride)
Rich Felker51e2d832011-06-18 19:48:42 -0400220{
Rich Felker87d13a42012-08-05 02:49:02 -0400221 unsigned char *base = dso->base;
222 Sym *syms = dso->syms;
223 char *strings = dso->strings;
Rich Felker51e2d832011-06-18 19:48:42 -0400224 Sym *sym;
225 const char *name;
Rich Felker51e2d832011-06-18 19:48:42 -0400226 void *ctx;
227 int type;
228 int sym_index;
Rich Felker9c748562012-10-04 22:48:33 -0400229 struct symdef def;
Rich Felker51e2d832011-06-18 19:48:42 -0400230
231 for (; rel_size; rel+=stride, rel_size-=stride*sizeof(size_t)) {
Rich Felker51e2d832011-06-18 19:48:42 -0400232 type = R_TYPE(rel[1]);
233 sym_index = R_SYM(rel[1]);
234 if (sym_index) {
235 sym = syms + sym_index;
236 name = strings + sym->st_name;
Rich Felker7cb44cd2012-08-05 02:44:32 -0400237 ctx = IS_COPY(type) ? head->next : head;
Rich Felker9c748562012-10-04 22:48:33 -0400238 def = find_sym(ctx, name, IS_PLT(type));
239 if (!def.sym && sym->st_info>>4 != STB_WEAK) {
Rich Felkera5d10eb2012-04-23 12:03:31 -0400240 snprintf(errbuf, sizeof errbuf,
241 "Error relocating %s: %s: symbol not found",
Rich Felker87d13a42012-08-05 02:49:02 -0400242 dso->name, name);
Rich Felker6b3d5e52011-06-26 17:39:17 -0400243 if (runtime) longjmp(rtld_fail, 1);
Rich Felkera5d10eb2012-04-23 12:03:31 -0400244 dprintf(2, "%s\n", errbuf);
Rich Felker04109502012-08-18 16:00:23 -0400245 ldso_fail = 1;
246 continue;
Rich Felker6b3d5e52011-06-26 17:39:17 -0400247 }
Rich Felker7d9a5c62012-08-05 14:03:17 -0400248 } else {
Rich Felker9c748562012-10-04 22:48:33 -0400249 sym = 0;
250 def.sym = 0;
251 def.dso = 0;
Rich Felker51e2d832011-06-18 19:48:42 -0400252 }
Rich Felker9c748562012-10-04 22:48:33 -0400253 do_single_reloc(dso, base, (void *)(base + rel[0]), type,
254 stride>2 ? rel[2] : 0, sym, sym?sym->st_size:0, def,
255 def.sym?(size_t)(def.dso->base+def.sym->st_value):0);
Rich Felker51e2d832011-06-18 19:48:42 -0400256 }
257}
258
Rich Felker6717e622011-06-28 19:40:14 -0400259/* A huge hack: to make up for the wastefulness of shared libraries
260 * needing at least a page of dirty memory even if they have no global
261 * data, we reclaim the gaps at the beginning and end of writable maps
262 * and "donate" them to the heap by setting up minimal malloc
263 * structures and then freeing them. */
264
265static void reclaim(unsigned char *base, size_t start, size_t end)
266{
267 size_t *a, *z;
268 start = start + 6*sizeof(size_t)-1 & -4*sizeof(size_t);
269 end = (end & -4*sizeof(size_t)) - 2*sizeof(size_t);
270 if (start>end || end-start < 4*sizeof(size_t)) return;
271 a = (size_t *)(base + start);
272 z = (size_t *)(base + end);
273 a[-2] = 1;
274 a[-1] = z[0] = end-start + 2*sizeof(size_t) | 1;
275 z[1] = 1;
276 free(a);
277}
278
279static void reclaim_gaps(unsigned char *base, Phdr *ph, size_t phent, size_t phcnt)
280{
281 for (; phcnt--; ph=(void *)((char *)ph+phent)) {
282 if (ph->p_type!=PT_LOAD) continue;
283 if ((ph->p_flags&(PF_R|PF_W))!=(PF_R|PF_W)) continue;
284 reclaim(base, ph->p_vaddr & -PAGE_SIZE, ph->p_vaddr);
285 reclaim(base, ph->p_vaddr+ph->p_memsz,
286 ph->p_vaddr+ph->p_memsz+PAGE_SIZE-1 & -PAGE_SIZE);
287 }
288}
289
Rich Felkerbc6a35f2012-10-04 20:04:13 -0400290static void *map_library(int fd, struct dso *dso)
Rich Felker51e2d832011-06-18 19:48:42 -0400291{
Rich Felker59633c72011-06-25 12:26:08 -0400292 Ehdr buf[(896+sizeof(Ehdr))/sizeof(Ehdr)];
Rich Felker51e2d832011-06-18 19:48:42 -0400293 size_t phsize;
294 size_t addr_min=SIZE_MAX, addr_max=0, map_len;
295 size_t this_min, this_max;
296 off_t off_start;
297 Ehdr *eh;
298 Phdr *ph;
299 unsigned prot;
300 unsigned char *map, *base;
301 size_t dyn;
Rich Felkerbc6a35f2012-10-04 20:04:13 -0400302 size_t tls_image=0;
Rich Felker51e2d832011-06-18 19:48:42 -0400303 size_t i;
304
305 ssize_t l = read(fd, buf, sizeof buf);
306 if (l<sizeof *eh) return 0;
Rich Felker59633c72011-06-25 12:26:08 -0400307 eh = buf;
Rich Felker51e2d832011-06-18 19:48:42 -0400308 phsize = eh->e_phentsize * eh->e_phnum;
309 if (phsize + sizeof *eh > l) return 0;
310 if (eh->e_phoff + phsize > l) {
Rich Felker59633c72011-06-25 12:26:08 -0400311 l = pread(fd, buf+1, phsize, eh->e_phoff);
Rich Felker51e2d832011-06-18 19:48:42 -0400312 if (l != phsize) return 0;
313 eh->e_phoff = sizeof *eh;
314 }
315 ph = (void *)((char *)buf + eh->e_phoff);
316 for (i=eh->e_phnum; i; i--, ph=(void *)((char *)ph+eh->e_phentsize)) {
317 if (ph->p_type == PT_DYNAMIC)
318 dyn = ph->p_vaddr;
Rich Felkerbc6a35f2012-10-04 20:04:13 -0400319 if (ph->p_type == PT_TLS) {
320 tls_image = ph->p_vaddr;
321 dso->tls_align = ph->p_align;
322 dso->tls_len = ph->p_filesz;
323 dso->tls_size = ph->p_memsz;
324 }
Rich Felker51e2d832011-06-18 19:48:42 -0400325 if (ph->p_type != PT_LOAD) continue;
326 if (ph->p_vaddr < addr_min) {
327 addr_min = ph->p_vaddr;
328 off_start = ph->p_offset;
329 prot = (((ph->p_flags&PF_R) ? PROT_READ : 0) |
330 ((ph->p_flags&PF_W) ? PROT_WRITE: 0) |
331 ((ph->p_flags&PF_X) ? PROT_EXEC : 0));
332 }
333 if (ph->p_vaddr+ph->p_memsz > addr_max) {
334 addr_max = ph->p_vaddr+ph->p_memsz;
335 }
336 }
337 if (!dyn) return 0;
338 addr_max += PAGE_SIZE-1;
339 addr_max &= -PAGE_SIZE;
340 addr_min &= -PAGE_SIZE;
341 off_start &= -PAGE_SIZE;
342 map_len = addr_max - addr_min + off_start;
343 /* The first time, we map too much, possibly even more than
344 * the length of the file. This is okay because we will not
345 * use the invalid part; we just need to reserve the right
346 * amount of virtual address space to map over later. */
Rich Felkerbf301002011-06-28 14:20:41 -0400347 map = mmap((void *)addr_min, map_len, prot, MAP_PRIVATE, fd, off_start);
Rich Felker51e2d832011-06-18 19:48:42 -0400348 if (map==MAP_FAILED) return 0;
349 base = map - addr_min;
350 ph = (void *)((char *)buf + eh->e_phoff);
351 for (i=eh->e_phnum; i; i--, ph=(void *)((char *)ph+eh->e_phentsize)) {
352 if (ph->p_type != PT_LOAD) continue;
353 /* Reuse the existing mapping for the lowest-address LOAD */
354 if ((ph->p_vaddr & -PAGE_SIZE) == addr_min) continue;
355 this_min = ph->p_vaddr & -PAGE_SIZE;
356 this_max = ph->p_vaddr+ph->p_memsz+PAGE_SIZE-1 & -PAGE_SIZE;
357 off_start = ph->p_offset & -PAGE_SIZE;
358 prot = (((ph->p_flags&PF_R) ? PROT_READ : 0) |
359 ((ph->p_flags&PF_W) ? PROT_WRITE: 0) |
360 ((ph->p_flags&PF_X) ? PROT_EXEC : 0));
Rich Felkerf7d15dc2012-06-06 11:21:28 -0400361 if (mmap(base+this_min, this_max-this_min, prot, MAP_PRIVATE|MAP_FIXED, fd, off_start) == MAP_FAILED)
362 goto error;
Rich Felker51e2d832011-06-18 19:48:42 -0400363 if (ph->p_memsz > ph->p_filesz) {
364 size_t brk = (size_t)base+ph->p_vaddr+ph->p_filesz;
365 size_t pgbrk = brk+PAGE_SIZE-1 & -PAGE_SIZE;
366 memset((void *)brk, 0, pgbrk-brk & PAGE_SIZE-1);
Rich Felkerf7d15dc2012-06-06 11:21:28 -0400367 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)
368 goto error;
Rich Felker51e2d832011-06-18 19:48:42 -0400369 }
370 }
Rich Felker9f174132011-06-29 00:29:08 -0400371 for (i=0; ((size_t *)(base+dyn))[i]; i+=2)
372 if (((size_t *)(base+dyn))[i]==DT_TEXTREL) {
Rich Felkerf7d15dc2012-06-06 11:21:28 -0400373 if (mprotect(map, map_len, PROT_READ|PROT_WRITE|PROT_EXEC) < 0)
374 goto error;
Rich Felker9f174132011-06-29 00:29:08 -0400375 break;
376 }
Rich Felker6717e622011-06-28 19:40:14 -0400377 if (!runtime) reclaim_gaps(base, (void *)((char *)buf + eh->e_phoff),
378 eh->e_phentsize, eh->e_phnum);
Rich Felkerbc6a35f2012-10-04 20:04:13 -0400379 dso->map = map;
380 dso->map_len = map_len;
381 dso->base = base;
382 dso->dynv = (void *)(base+dyn);
383 if (dso->tls_size) dso->tls_image = (void *)(base+tls_image);
Rich Felker51e2d832011-06-18 19:48:42 -0400384 return map;
Rich Felkerf7d15dc2012-06-06 11:21:28 -0400385error:
386 munmap(map, map_len);
387 return 0;
Rich Felker51e2d832011-06-18 19:48:42 -0400388}
389
Rich Felker5c1909a2012-05-27 16:01:44 -0400390static int path_open(const char *name, const char *search, char *buf, size_t buf_size)
Rich Felker568b8072011-06-25 01:56:34 -0400391{
Rich Felker49388f32011-06-25 17:49:16 -0400392 const char *s=search, *z;
Rich Felker568b8072011-06-25 01:56:34 -0400393 int l, fd;
Rich Felker49388f32011-06-25 17:49:16 -0400394 for (;;) {
395 while (*s==':') s++;
396 if (!*s) return -1;
Rich Felker568b8072011-06-25 01:56:34 -0400397 z = strchr(s, ':');
398 l = z ? z-s : strlen(s);
Rich Felker5c1909a2012-05-27 16:01:44 -0400399 snprintf(buf, buf_size, "%.*s/%s", l, s, name);
Rich Felkerf2d08cf2012-09-29 17:59:50 -0400400 if ((fd = open(buf, O_RDONLY|O_CLOEXEC))>=0) return fd;
Rich Felker49388f32011-06-25 17:49:16 -0400401 s += l;
Rich Felker568b8072011-06-25 01:56:34 -0400402 }
Rich Felker568b8072011-06-25 01:56:34 -0400403}
404
Rich Felkerc82f4a32012-01-23 00:57:38 -0500405static void decode_dyn(struct dso *p)
406{
407 size_t dyn[DYN_CNT] = {0};
408 decode_vec(p->dynv, dyn, DYN_CNT);
409 p->syms = (void *)(p->base + dyn[DT_SYMTAB]);
Rich Felkerc82f4a32012-01-23 00:57:38 -0500410 p->strings = (void *)(p->base + dyn[DT_STRTAB]);
Rich Felker2bd05a42012-08-25 17:13:28 -0400411 if (dyn[0]&(1<<DT_HASH))
412 p->hashtab = (void *)(p->base + dyn[DT_HASH]);
413 if (search_vec(p->dynv, dyn, DT_GNU_HASH))
414 p->ghashtab = (void *)(p->base + *dyn);
Rich Felkerc82f4a32012-01-23 00:57:38 -0500415}
416
Rich Felker51e2d832011-06-18 19:48:42 -0400417static struct dso *load_library(const char *name)
418{
Rich Felker5c1909a2012-05-27 16:01:44 -0400419 char buf[2*NAME_MAX+2];
Rich Felker0420b872012-07-11 01:41:20 -0400420 const char *pathname;
Rich Felker51e2d832011-06-18 19:48:42 -0400421 unsigned char *base, *map;
422 size_t dyno, map_len;
Rich Felkerbc6a35f2012-10-04 20:04:13 -0400423 struct dso *p, temp_dso = {0};
Rich Felker51e2d832011-06-18 19:48:42 -0400424 int fd;
425 struct stat st;
Rich Felkerdcd60372012-10-05 11:51:50 -0400426 size_t alloc_size;
427 int n_th = 0;
Rich Felker51e2d832011-06-18 19:48:42 -0400428
429 /* Catch and block attempts to reload the implementation itself */
430 if (name[0]=='l' && name[1]=='i' && name[2]=='b') {
431 static const char *rp, reserved[] =
432 "c\0pthread\0rt\0m\0dl\0util\0xnet\0";
433 char *z = strchr(name, '.');
434 if (z) {
435 size_t l = z-name;
436 for (rp=reserved; *rp && memcmp(name+3, rp, l-3); rp+=strlen(rp)+1);
437 if (*rp) {
438 if (!libc->prev) {
439 tail->next = libc;
440 libc->prev = tail;
Rich Felker6ab444d2011-07-24 00:54:55 -0400441 tail = libc->next ? libc->next : libc;
Rich Felker51e2d832011-06-18 19:48:42 -0400442 }
443 return libc;
444 }
445 }
446 }
Rich Felker59ab43f2011-06-26 19:23:28 -0400447 if (strchr(name, '/')) {
Rich Felker0420b872012-07-11 01:41:20 -0400448 pathname = name;
Rich Felkerf2d08cf2012-09-29 17:59:50 -0400449 fd = open(name, O_RDONLY|O_CLOEXEC);
Rich Felker51e2d832011-06-18 19:48:42 -0400450 } else {
Rich Felker0420b872012-07-11 01:41:20 -0400451 /* Search for the name to see if it's already loaded */
452 for (p=head->next; p; p=p->next) {
453 if (p->shortname && !strcmp(p->shortname, name)) {
454 p->refcnt++;
455 return p;
456 }
457 }
Rich Felker59ab43f2011-06-26 19:23:28 -0400458 if (strlen(name) > NAME_MAX) return 0;
Rich Felker568b8072011-06-25 01:56:34 -0400459 fd = -1;
Rich Felker5c1909a2012-05-27 16:01:44 -0400460 if (r_path) fd = path_open(name, r_path, buf, sizeof buf);
461 if (fd < 0 && env_path) fd = path_open(name, env_path, buf, sizeof buf);
Rich Felker568b8072011-06-25 01:56:34 -0400462 if (fd < 0) {
463 if (!sys_path) {
Rich Felkerd712dd32012-09-29 18:14:46 -0400464 FILE *f = fopen(ETC_LDSO_PATH, "rbe");
Rich Felker568b8072011-06-25 01:56:34 -0400465 if (f) {
466 if (getline(&sys_path, (size_t[1]){0}, f) > 0)
467 sys_path[strlen(sys_path)-1]=0;
468 fclose(f);
469 }
470 }
Rich Felker5c1909a2012-05-27 16:01:44 -0400471 if (sys_path) fd = path_open(name, sys_path, buf, sizeof buf);
472 else fd = path_open(name, "/lib:/usr/local/lib:/usr/lib", buf, sizeof buf);
Rich Felker51e2d832011-06-18 19:48:42 -0400473 }
Rich Felker0420b872012-07-11 01:41:20 -0400474 pathname = buf;
Rich Felker51e2d832011-06-18 19:48:42 -0400475 }
476 if (fd < 0) return 0;
477 if (fstat(fd, &st) < 0) {
478 close(fd);
479 return 0;
480 }
481 for (p=head->next; p; p=p->next) {
482 if (p->dev == st.st_dev && p->ino == st.st_ino) {
Rich Felker0420b872012-07-11 01:41:20 -0400483 /* If this library was previously loaded with a
484 * pathname but a search found the same inode,
485 * setup its shortname so it can be found by name. */
486 if (!p->shortname) p->shortname = strrchr(p->name, '/')+1;
Rich Felker51e2d832011-06-18 19:48:42 -0400487 close(fd);
488 p->refcnt++;
489 return p;
490 }
491 }
Rich Felkerbc6a35f2012-10-04 20:04:13 -0400492 map = map_library(fd, &temp_dso);
Rich Felker51e2d832011-06-18 19:48:42 -0400493 close(fd);
494 if (!map) return 0;
Rich Felkerdcd60372012-10-05 11:51:50 -0400495
496 /* Allocate storage for the new DSO. When there is TLS, this
497 * storage must include a reservation for all pre-existing
498 * threads to obtain copies of both the new TLS, and an
499 * extended DTV capable of storing an additional slot for
500 * the newly-loaded DSO. */
501 alloc_size = sizeof *p + strlen(pathname) + 1;
502 if (runtime && temp_dso.tls_image) {
503 size_t per_th = temp_dso.tls_size + temp_dso.tls_align
504 + sizeof(void *) * (tls_cnt+3);
505 n_th = __libc.threads_minus_1 + 1;
506 if (n_th > SSIZE_MAX / per_th) alloc_size = SIZE_MAX;
507 else alloc_size += n_th * per_th;
508 }
509 p = calloc(1, alloc_size);
Rich Felker51e2d832011-06-18 19:48:42 -0400510 if (!p) {
511 munmap(map, map_len);
512 return 0;
513 }
Rich Felkerbc6a35f2012-10-04 20:04:13 -0400514 memcpy(p, &temp_dso, sizeof temp_dso);
Rich Felkerc82f4a32012-01-23 00:57:38 -0500515 decode_dyn(p);
Rich Felker51e2d832011-06-18 19:48:42 -0400516 p->dev = st.st_dev;
517 p->ino = st.st_ino;
Rich Felker51e2d832011-06-18 19:48:42 -0400518 p->refcnt = 1;
Rich Felker6b3d5e52011-06-26 17:39:17 -0400519 p->name = p->buf;
Rich Felker0420b872012-07-11 01:41:20 -0400520 strcpy(p->name, pathname);
521 /* Add a shortname only if name arg was not an explicit pathname. */
522 if (pathname != name) p->shortname = strrchr(p->name, '/')+1;
Rich Felkerdcd60372012-10-05 11:51:50 -0400523 if (p->tls_image) {
524 p->tls_id = ++tls_cnt;
525 tls_size += p->tls_size + p->tls_align + 8*sizeof(size_t) - 1
526 & -4*sizeof(size_t);
527 p->new_dtv = (void *)(-sizeof(size_t) &
528 (uintptr_t)(p->name+strlen(p->name)+sizeof(size_t)));
529 p->new_tls = (void *)(p->new_dtv + n_th*(tls_cnt+1));
530 }
Rich Felker51e2d832011-06-18 19:48:42 -0400531
532 tail->next = p;
533 p->prev = tail;
534 tail = p;
535
Rich Felker0420b872012-07-11 01:41:20 -0400536 if (ldd_mode) dprintf(1, "\t%s => %s (%p)\n", name, pathname, base);
Rich Felker5c1909a2012-05-27 16:01:44 -0400537
Rich Felker51e2d832011-06-18 19:48:42 -0400538 return p;
539}
540
541static void load_deps(struct dso *p)
542{
Rich Felker59ab43f2011-06-26 19:23:28 -0400543 size_t i, ndeps=0;
544 struct dso ***deps = &p->deps, **tmp, *dep;
Rich Felker51e2d832011-06-18 19:48:42 -0400545 for (; p; p=p->next) {
546 for (i=0; p->dynv[i]; i+=2) {
Rich Felker191ebca2011-06-30 23:02:27 -0400547 if (p->dynv[i] != DT_RPATH) continue;
548 r_path = (void *)(p->strings + p->dynv[i+1]);
549 }
550 for (i=0; p->dynv[i]; i+=2) {
Rich Felker51e2d832011-06-18 19:48:42 -0400551 if (p->dynv[i] != DT_NEEDED) continue;
Rich Felker59ab43f2011-06-26 19:23:28 -0400552 dep = load_library(p->strings + p->dynv[i+1]);
553 if (!dep) {
Rich Felkera5d10eb2012-04-23 12:03:31 -0400554 snprintf(errbuf, sizeof errbuf,
555 "Error loading shared library %s: %m (needed by %s)",
Rich Felker6b3d5e52011-06-26 17:39:17 -0400556 p->strings + p->dynv[i+1], p->name);
Rich Felkera5d10eb2012-04-23 12:03:31 -0400557 if (runtime) longjmp(rtld_fail, 1);
558 dprintf(2, "%s\n", errbuf);
Rich Felker04109502012-08-18 16:00:23 -0400559 ldso_fail = 1;
560 continue;
Rich Felker6b3d5e52011-06-26 17:39:17 -0400561 }
Rich Felker59ab43f2011-06-26 19:23:28 -0400562 if (runtime) {
563 tmp = realloc(*deps, sizeof(*tmp)*(ndeps+2));
564 if (!tmp) longjmp(rtld_fail, 1);
565 tmp[ndeps++] = dep;
566 tmp[ndeps] = 0;
567 *deps = tmp;
568 }
Rich Felker51e2d832011-06-18 19:48:42 -0400569 }
Rich Felker191ebca2011-06-30 23:02:27 -0400570 r_path = 0;
Rich Felker51e2d832011-06-18 19:48:42 -0400571 }
572}
573
Rich Felker2719cc82011-08-16 00:24:36 -0400574static void load_preload(char *s)
575{
576 int tmp;
577 char *z;
578 for (z=s; *z; s=z) {
579 for ( ; *s && isspace(*s); s++);
580 for (z=s; *z && !isspace(*z); z++);
581 tmp = *z;
582 *z = 0;
583 load_library(s);
584 *z = tmp;
585 }
586}
587
Rich Felker59ab43f2011-06-26 19:23:28 -0400588static void make_global(struct dso *p)
589{
590 for (; p; p=p->next) p->global = 1;
591}
592
Rich Felker51e2d832011-06-18 19:48:42 -0400593static void reloc_all(struct dso *p)
594{
595 size_t dyn[DYN_CNT] = {0};
596 for (; p; p=p->next) {
597 if (p->relocated) continue;
598 decode_vec(p->dynv, dyn, DYN_CNT);
Rich Felkerbabf8202012-08-05 12:50:26 -0400599#ifdef NEED_ARCH_RELOCS
600 do_arch_relocs(p, head);
601#endif
Rich Felker87d13a42012-08-05 02:49:02 -0400602 do_relocs(p, (void *)(p->base+dyn[DT_JMPREL]), dyn[DT_PLTRELSZ],
603 2+(dyn[DT_PLTREL]==DT_RELA));
604 do_relocs(p, (void *)(p->base+dyn[DT_REL]), dyn[DT_RELSZ], 2);
605 do_relocs(p, (void *)(p->base+dyn[DT_RELA]), dyn[DT_RELASZ], 3);
Rich Felker368ba4a2011-06-25 00:18:19 -0400606 p->relocated = 1;
Rich Felker51e2d832011-06-18 19:48:42 -0400607 }
608}
609
Rich Felkere8dbf002011-06-25 00:47:28 -0400610static void free_all(struct dso *p)
611{
612 struct dso *n;
613 while (p) {
614 n = p->next;
Rich Felker8b28aa92012-08-27 10:07:32 -0400615 if (p->map && p!=libc && p!=head) free(p);
Rich Felkere8dbf002011-06-25 00:47:28 -0400616 p = n;
617 }
618}
619
Rich Felkerc82f4a32012-01-23 00:57:38 -0500620static size_t find_dyn(Phdr *ph, size_t cnt, size_t stride)
621{
622 for (; cnt--; ph = (void *)((char *)ph + stride))
623 if (ph->p_type == PT_DYNAMIC)
624 return ph->p_vaddr;
625 return 0;
626}
627
Rich Felkerf419bcb2012-08-26 21:09:26 -0400628static void find_map_range(Phdr *ph, size_t cnt, size_t stride, struct dso *p)
629{
630 size_t min_addr = -1, max_addr = 0;
631 for (; cnt--; ph = (void *)((char *)ph + stride)) {
632 if (ph->p_type != PT_LOAD) continue;
633 if (ph->p_vaddr < min_addr)
634 min_addr = ph->p_vaddr;
635 if (ph->p_vaddr+ph->p_memsz > max_addr)
636 max_addr = ph->p_vaddr+ph->p_memsz;
637 }
638 min_addr &= -PAGE_SIZE;
639 max_addr = (max_addr + PAGE_SIZE-1) & -PAGE_SIZE;
640 p->map = p->base + min_addr;
641 p->map_len = max_addr - min_addr;
642}
643
Rich Felker4ce3cb52012-02-06 14:39:09 -0500644static void do_init_fini(struct dso *p)
645{
646 size_t dyn[DYN_CNT] = {0};
647 for (; p; p=p->prev) {
648 if (p->constructed) return;
649 decode_vec(p->dynv, dyn, DYN_CNT);
650 if (dyn[0] & (1<<DT_FINI))
651 atexit((void (*)(void))(p->base + dyn[DT_FINI]));
652 if (dyn[0] & (1<<DT_INIT))
653 ((void (*)(void))(p->base + dyn[DT_INIT]))();
654 p->constructed = 1;
655 }
656}
657
Rich Felker3ec8d292012-04-25 00:05:42 -0400658void _dl_debug_state(void)
659{
660}
661
Rich Felkerdcd60372012-10-05 11:51:50 -0400662void *__copy_tls(unsigned char *mem)
Rich Felkerbc6a35f2012-10-04 20:04:13 -0400663{
664 struct dso *p;
665 void **dtv = (void *)mem;
Rich Felkerdcd60372012-10-05 11:51:50 -0400666 dtv[0] = (void *)tls_cnt;
667 mem = (void *)(dtv + tls_cnt + 1);
Rich Felkerbc6a35f2012-10-04 20:04:13 -0400668 for (p=tail; p; p=p->prev) {
Rich Felkerdcd60372012-10-05 11:51:50 -0400669 if (!p->tls_id) continue;
Rich Felkerbc6a35f2012-10-04 20:04:13 -0400670 mem += -p->tls_len & (4*sizeof(size_t)-1);
671 mem += ((uintptr_t)p->tls_image - (uintptr_t)mem)
672 & (p->tls_align-1);
673 dtv[p->tls_id] = mem;
674 memcpy(mem, p->tls_image, p->tls_len);
675 mem += p->tls_size;
676 }
677 ((pthread_t)mem)->dtv = dtv;
678 return mem;
679}
680
Rich Felkerdcd60372012-10-05 11:51:50 -0400681void *__tls_get_addr(size_t *v)
Rich Felker9b153c02012-10-04 21:01:56 -0400682{
683 pthread_t self = __pthread_self();
Rich Felkerdcd60372012-10-05 11:51:50 -0400684 if (self->dtv && v[0]<=(size_t)self->dtv[0] && self->dtv[v[0]])
685 return (char *)self->dtv[v[0]]+v[1];
686
687 /* Block signals to make accessing new TLS async-signal-safe */
688 sigset_t set;
689 sigfillset(&set);
690 pthread_sigmask(SIG_BLOCK, &set, &set);
691 if (self->dtv && v[0]<=(size_t)self->dtv[0] && self->dtv[v[0]]) {
692 pthread_sigmask(SIG_SETMASK, &set, 0);
693 return (char *)self->dtv[v[0]]+v[1];
Rich Felker9b153c02012-10-04 21:01:56 -0400694 }
Rich Felkerdcd60372012-10-05 11:51:50 -0400695
696 /* This is safe without any locks held because, if the caller
697 * is able to request the Nth entry of the DTV, the DSO list
698 * must be valid at least that far out and it was synchronized
699 * at program startup or by an already-completed call to dlopen. */
700 struct dso *p;
701 for (p=head; p->tls_id != v[0]; p=p->next);
702
703 /* Get new DTV space from new DSO if needed */
704 if (!self->dtv || v[0] > (size_t)self->dtv[0]) {
705 void **newdtv = p->new_dtv +
706 (v[0]+1)*sizeof(void *)*a_fetch_add(&p->new_dtv_idx,1);
707 if (self->dtv) memcpy(newdtv, self->dtv,
708 ((size_t)self->dtv[0]+1) * sizeof(void *));
709 newdtv[0] = (void *)v[0];
710 self->dtv = newdtv;
711 }
712
713 /* Get new TLS memory from new DSO */
714 unsigned char *mem = p->new_tls +
715 (p->tls_size + p->tls_align) * a_fetch_add(&p->new_tls_idx,1);
716 mem += ((uintptr_t)p->tls_image - (uintptr_t)mem) & (p->tls_align-1);
717 self->dtv[v[0]] = mem;
718 memcpy(mem, p->tls_image, p->tls_size);
719 pthread_sigmask(SIG_SETMASK, &set, 0);
720 return mem + v[1];
Rich Felker9b153c02012-10-04 21:01:56 -0400721}
722
Rich Felkerc82f4a32012-01-23 00:57:38 -0500723void *__dynlink(int argc, char **argv)
Rich Felker51e2d832011-06-18 19:48:42 -0400724{
Rich Felker731e8ff2012-08-25 17:24:46 -0400725 size_t aux[AUX_CNT] = {0};
Rich Felker51e2d832011-06-18 19:48:42 -0400726 size_t i;
727 Phdr *phdr;
Rich Felker6717e622011-06-28 19:40:14 -0400728 Ehdr *ehdr;
Rich Felker6ab444d2011-07-24 00:54:55 -0400729 static struct dso builtin_dsos[3];
Rich Felkera53de812011-07-24 00:26:12 -0400730 struct dso *const app = builtin_dsos+0;
731 struct dso *const lib = builtin_dsos+1;
Rich Felker6ab444d2011-07-24 00:54:55 -0400732 struct dso *const vdso = builtin_dsos+2;
Rich Felker2719cc82011-08-16 00:24:36 -0400733 char *env_preload=0;
Rich Felkerdbcb3ad2012-08-25 17:31:59 -0400734 size_t vdso_base;
Rich Felker51e2d832011-06-18 19:48:42 -0400735
736 /* Find aux vector just past environ[] */
Rich Felker568b8072011-06-25 01:56:34 -0400737 for (i=argc+1; argv[i]; i++)
738 if (!memcmp(argv[i], "LD_LIBRARY_PATH=", 16))
739 env_path = argv[i]+16;
Rich Felker2719cc82011-08-16 00:24:36 -0400740 else if (!memcmp(argv[i], "LD_PRELOAD=", 11))
741 env_preload = argv[i]+11;
Rich Felker51e2d832011-06-18 19:48:42 -0400742 auxv = (void *)(argv+i+1);
743
744 decode_vec(auxv, aux, AUX_CNT);
745
Rich Felker568b8072011-06-25 01:56:34 -0400746 /* Only trust user/env if kernel says we're not suid/sgid */
747 if ((aux[0]&0x7800)!=0x7800 || aux[AT_UID]!=aux[AT_EUID]
Rich Felkera0458832011-08-16 07:46:42 -0400748 || aux[AT_GID]!=aux[AT_EGID] || aux[AT_SECURE]) {
Rich Felker568b8072011-06-25 01:56:34 -0400749 env_path = 0;
Rich Felker2719cc82011-08-16 00:24:36 -0400750 env_preload = 0;
Rich Felker568b8072011-06-25 01:56:34 -0400751 }
752
Rich Felker5c1909a2012-05-27 16:01:44 -0400753 /* If the dynamic linker was invoked as a program itself, AT_BASE
754 * will not be set. In that case, we assume the base address is
755 * the start of the page containing the PHDRs; I don't know any
756 * better approach... */
757 if (!aux[AT_BASE]) {
758 aux[AT_BASE] = aux[AT_PHDR] & -PAGE_SIZE;
759 aux[AT_PHDR] = aux[AT_PHENT] = aux[AT_PHNUM] = 0;
760 }
761
Rich Felkerc82f4a32012-01-23 00:57:38 -0500762 /* The dynamic linker load address is passed by the kernel
763 * in the AUX vector, so this is easy. */
764 lib->base = (void *)aux[AT_BASE];
Rich Felker5c1909a2012-05-27 16:01:44 -0400765 lib->name = lib->shortname = "libc.so";
Rich Felkerc82f4a32012-01-23 00:57:38 -0500766 lib->global = 1;
767 ehdr = (void *)lib->base;
Rich Felkerf419bcb2012-08-26 21:09:26 -0400768 find_map_range((void *)(aux[AT_BASE]+ehdr->e_phoff),
769 ehdr->e_phnum, ehdr->e_phentsize, lib);
Rich Felkerc82f4a32012-01-23 00:57:38 -0500770 lib->dynv = (void *)(lib->base + find_dyn(
771 (void *)(aux[AT_BASE]+ehdr->e_phoff),
772 ehdr->e_phnum, ehdr->e_phentsize));
773 decode_dyn(lib);
774
Rich Felker5c1909a2012-05-27 16:01:44 -0400775 if (aux[AT_PHDR]) {
Rich Felker649cec52012-07-13 01:31:02 -0400776 size_t interp_off = 0;
Rich Felkerbc6a35f2012-10-04 20:04:13 -0400777 size_t tls_image = 0;
Rich Felker5c1909a2012-05-27 16:01:44 -0400778 /* Find load address of the main program, via AT_PHDR vs PT_PHDR. */
779 phdr = (void *)aux[AT_PHDR];
780 for (i=aux[AT_PHNUM]; i; i--, phdr=(void *)((char *)phdr + aux[AT_PHENT])) {
781 if (phdr->p_type == PT_PHDR)
782 app->base = (void *)(aux[AT_PHDR] - phdr->p_vaddr);
Rich Felker649cec52012-07-13 01:31:02 -0400783 else if (phdr->p_type == PT_INTERP)
784 interp_off = (size_t)phdr->p_vaddr;
Rich Felkerbc6a35f2012-10-04 20:04:13 -0400785 else if (phdr->p_type == PT_TLS) {
786 tls_image = phdr->p_vaddr;
787 app->tls_len = phdr->p_filesz;
788 app->tls_size = phdr->p_memsz;
789 app->tls_align = phdr->p_align;
790 }
Rich Felker5c1909a2012-05-27 16:01:44 -0400791 }
Rich Felkerbc6a35f2012-10-04 20:04:13 -0400792 if (app->tls_size) app->tls_image = (char *)app->base + tls_image;
Rich Felker649cec52012-07-13 01:31:02 -0400793 if (interp_off) lib->name = (char *)app->base + interp_off;
Rich Felker0420b872012-07-11 01:41:20 -0400794 app->name = argv[0];
Rich Felker5c1909a2012-05-27 16:01:44 -0400795 app->dynv = (void *)(app->base + find_dyn(
796 (void *)aux[AT_PHDR], aux[AT_PHNUM], aux[AT_PHENT]));
Rich Felkerf419bcb2012-08-26 21:09:26 -0400797 find_map_range((void *)aux[AT_PHDR],
798 aux[AT_PHNUM], aux[AT_PHENT], app);
Rich Felker5c1909a2012-05-27 16:01:44 -0400799 } else {
800 int fd;
801 char *ldname = argv[0];
802 size_t dyno, l = strlen(ldname);
803 if (l >= 3 && !strcmp(ldname+l-3, "ldd")) ldd_mode = 1;
804 *argv++ = (void *)-1;
805 if (argv[0] && !strcmp(argv[0], "--")) *argv++ = (void *)-1;
806 if (!argv[0]) {
807 dprintf(2, "musl libc/dynamic program loader\n");
808 dprintf(2, "usage: %s pathname%s\n", ldname,
809 ldd_mode ? "" : " [args]");
810 _exit(1);
811 }
812 fd = open(argv[0], O_RDONLY);
813 if (fd < 0) {
814 dprintf(2, "%s: cannot load %s: %s\n", ldname, argv[0], strerror(errno));
815 _exit(1);
816 }
817 runtime = 1;
Rich Felkerbc6a35f2012-10-04 20:04:13 -0400818 ehdr = (void *)map_library(fd, app);
Rich Felker5c1909a2012-05-27 16:01:44 -0400819 if (!ehdr) {
820 dprintf(2, "%s: %s: Not a valid dynamic program\n", ldname, argv[0]);
821 _exit(1);
822 }
823 runtime = 0;
824 close(fd);
Rich Felker649cec52012-07-13 01:31:02 -0400825 lib->name = ldname;
Rich Felker0420b872012-07-11 01:41:20 -0400826 app->name = argv[0];
Rich Felker5c1909a2012-05-27 16:01:44 -0400827 aux[AT_ENTRY] = ehdr->e_entry;
Rich Felkere12fe652012-01-23 02:02:59 -0500828 }
Rich Felkerbc6a35f2012-10-04 20:04:13 -0400829 if (app->tls_size) {
830 app->tls_id = ++tls_cnt;
831 tls_size += app->tls_size+app->tls_align + 8*sizeof(size_t)-1
832 & -4*sizeof(size_t);
833 }
Rich Felkerc82f4a32012-01-23 00:57:38 -0500834 app->global = 1;
Rich Felker4ce3cb52012-02-06 14:39:09 -0500835 app->constructed = 1;
Rich Felkerc82f4a32012-01-23 00:57:38 -0500836 decode_dyn(app);
837
838 /* Attach to vdso, if provided by the kernel */
Rich Felkerdbcb3ad2012-08-25 17:31:59 -0400839 if (search_vec(auxv, &vdso_base, AT_SYSINFO_EHDR)) {
Rich Felker6ab444d2011-07-24 00:54:55 -0400840 ehdr = (void *)vdso_base;
841 phdr = (void *)(vdso_base + ehdr->e_phoff);
842 for (i=ehdr->e_phnum; i; i--, phdr=(void *)((char *)phdr + ehdr->e_phentsize)) {
843 if (phdr->p_type == PT_DYNAMIC)
844 vdso->dynv = (void *)(vdso_base + phdr->p_offset);
845 if (phdr->p_type == PT_LOAD)
846 vdso->base = (void *)(vdso_base - phdr->p_vaddr + phdr->p_offset);
847 }
Rich Felker5c1909a2012-05-27 16:01:44 -0400848 vdso->name = vdso->shortname = "linux-gate.so.1";
Rich Felker427173b2011-07-24 02:19:47 -0400849 vdso->global = 1;
Rich Felkerc82f4a32012-01-23 00:57:38 -0500850 decode_dyn(vdso);
Rich Felker6ab444d2011-07-24 00:54:55 -0400851 vdso->prev = lib;
852 lib->next = vdso;
853 }
854
Rich Felkerc82f4a32012-01-23 00:57:38 -0500855 /* Initial dso chain consists only of the app. We temporarily
856 * append the dynamic linker/libc so we can relocate it, then
857 * restore the initial chain in preparation for loading third
858 * party libraries (preload/needed). */
859 head = tail = app;
860 libc = lib;
861 app->next = lib;
862 reloc_all(lib);
863 app->next = 0;
Rich Felker51e2d832011-06-18 19:48:42 -0400864
Rich Felkerc82f4a32012-01-23 00:57:38 -0500865 /* PAST THIS POINT, ALL LIBC INTERFACES ARE FULLY USABLE. */
Rich Felker51e2d832011-06-18 19:48:42 -0400866
Rich Felkerc82f4a32012-01-23 00:57:38 -0500867 /* Donate unused parts of app and library mapping to malloc */
Rich Felkera53de812011-07-24 00:26:12 -0400868 reclaim_gaps(app->base, (void *)aux[AT_PHDR], aux[AT_PHENT], aux[AT_PHNUM]);
869 ehdr = (void *)lib->base;
870 reclaim_gaps(lib->base, (void *)(lib->base+ehdr->e_phoff),
Rich Felker6717e622011-06-28 19:40:14 -0400871 ehdr->e_phentsize, ehdr->e_phnum);
872
Rich Felkerc82f4a32012-01-23 00:57:38 -0500873 /* Load preload/needed libraries, add their symbols to the global
Rich Felkerfd7015d2012-01-23 18:32:40 -0500874 * namespace, and perform all remaining relocations. The main
875 * program must be relocated LAST since it may contain copy
876 * relocations which depend on libraries' relocations. */
Rich Felker2719cc82011-08-16 00:24:36 -0400877 if (env_preload) load_preload(env_preload);
Rich Felkerc82f4a32012-01-23 00:57:38 -0500878 load_deps(app);
879 make_global(app);
Rich Felker9c748562012-10-04 22:48:33 -0400880
881 /* Make an initial pass setting up TLS before performing relocs.
882 * This provides the TP-based offset of each DSO's TLS for
883 * use in TP-relative relocations. After relocations, we need
884 * to copy the TLS images again in case they had relocs. */
885 tls_size += sizeof(struct pthread) + 4*sizeof(size_t);
886 __libc.tls_size = tls_size;
Rich Felker9c748562012-10-04 22:48:33 -0400887 if (tls_cnt) {
888 struct dso *p;
889 void *mem = mmap(0, __libc.tls_size, PROT_READ|PROT_WRITE,
890 MAP_ANONYMOUS|MAP_PRIVATE, -1, 0);
891 if (mem==MAP_FAILED ||
Rich Felkerdcd60372012-10-05 11:51:50 -0400892 !__install_initial_tls(__copy_tls(mem))) {
Rich Felker9c748562012-10-04 22:48:33 -0400893 dprintf(2, "%s: Error getting %zu bytes thread-local storage: %m\n",
894 argv[0], tls_size);
895 _exit(127);
896 }
897 for (p=head; p; p=p->next) {
898 if (!p->tls_id) continue;
899 p->tls_offset = (char *)__pthread_self()
900 - (char *)__pthread_self()->dtv[p->tls_id];
901 }
902 }
903
Rich Felkerfd7015d2012-01-23 18:32:40 -0500904 reloc_all(app->next);
Rich Felkerc82f4a32012-01-23 00:57:38 -0500905 reloc_all(app);
Rich Felker51e2d832011-06-18 19:48:42 -0400906
Rich Felker9c748562012-10-04 22:48:33 -0400907 /* The initial DTV is located at the base of the memory
908 * allocated for TLS. Repeat copying TLS to pick up relocs. */
Rich Felkerdcd60372012-10-05 11:51:50 -0400909 if (tls_cnt) __copy_tls((void *)__pthread_self()->dtv);
Rich Felker9c748562012-10-04 22:48:33 -0400910
Rich Felker04109502012-08-18 16:00:23 -0400911 if (ldso_fail) _exit(127);
Rich Felker5c1909a2012-05-27 16:01:44 -0400912 if (ldd_mode) _exit(0);
913
Rich Felkerc82f4a32012-01-23 00:57:38 -0500914 /* Switch to runtime mode: any further failures in the dynamic
915 * linker are a reportable failure rather than a fatal startup
916 * error. If the dynamic loader (dlopen) will not be used, free
917 * all memory used by the dynamic linker. */
Rich Felkera53de812011-07-24 00:26:12 -0400918 runtime = 1;
Rich Felker4ce3cb52012-02-06 14:39:09 -0500919
Rich Felker7d9a5c62012-08-05 14:03:17 -0400920#ifndef DYNAMIC_IS_RO
Rich Felker3ec8d292012-04-25 00:05:42 -0400921 for (i=0; app->dynv[i]; i+=2)
922 if (app->dynv[i]==DT_DEBUG)
923 app->dynv[i+1] = (size_t)&debug;
Rich Felker7d9a5c62012-08-05 14:03:17 -0400924#endif
Rich Felker3ec8d292012-04-25 00:05:42 -0400925 debug.ver = 1;
926 debug.bp = _dl_debug_state;
927 debug.head = head;
928 debug.base = lib->base;
929 debug.state = 0;
930 _dl_debug_state();
931
Rich Felker58aa5f42012-05-03 20:42:45 -0400932 if (ssp_used) __init_ssp(auxv);
933
Rich Felker4ce3cb52012-02-06 14:39:09 -0500934 do_init_fini(tail);
935
Rich Felker51e2d832011-06-18 19:48:42 -0400936 errno = 0;
937 return (void *)aux[AT_ENTRY];
938}
Rich Felker59ab43f2011-06-26 19:23:28 -0400939
940void *dlopen(const char *file, int mode)
941{
Rich Felker642b7592012-10-05 01:15:25 -0400942 struct dso *volatile p, *orig_tail, *next;
Rich Felkerdcd60372012-10-05 11:51:50 -0400943 size_t orig_tls_cnt;
Rich Felker59ab43f2011-06-26 19:23:28 -0400944 size_t i;
Rich Felkerf2baf4d2012-02-07 20:31:27 -0500945 int cs;
Rich Felker59ab43f2011-06-26 19:23:28 -0400946
947 if (!file) return head;
948
Rich Felkerf2baf4d2012-02-07 20:31:27 -0500949 pthread_setcancelstate(PTHREAD_CANCEL_DISABLE, &cs);
Rich Felker59ab43f2011-06-26 19:23:28 -0400950 pthread_rwlock_wrlock(&lock);
Rich Felkerdcd60372012-10-05 11:51:50 -0400951 __inhibit_ptc();
Rich Felker59ab43f2011-06-26 19:23:28 -0400952
Rich Felkerdcd60372012-10-05 11:51:50 -0400953 p = 0;
954 orig_tls_cnt = tls_cnt;
Rich Felker642b7592012-10-05 01:15:25 -0400955 orig_tail = tail;
956
Rich Felker59ab43f2011-06-26 19:23:28 -0400957 if (setjmp(rtld_fail)) {
958 /* Clean up anything new that was (partially) loaded */
Rich Felkerdcd60372012-10-05 11:51:50 -0400959 if (p && p->deps) for (i=0; p->deps[i]; i++)
Rich Felker92ab5d82011-06-26 21:21:04 -0400960 if (p->deps[i]->global < 0)
961 p->deps[i]->global = 0;
Rich Felker59ab43f2011-06-26 19:23:28 -0400962 for (p=orig_tail->next; p; p=next) {
963 next = p->next;
964 munmap(p->map, p->map_len);
965 free(p->deps);
966 free(p);
967 }
Rich Felkerdcd60372012-10-05 11:51:50 -0400968 tls_cnt = orig_tls_cnt;
969 tls_size = __libc.tls_size;
Rich Felker59ab43f2011-06-26 19:23:28 -0400970 tail = orig_tail;
971 tail->next = 0;
Rich Felkerf2baf4d2012-02-07 20:31:27 -0500972 p = 0;
Rich Felkera5d10eb2012-04-23 12:03:31 -0400973 errflag = 1;
974 goto end;
Rich Felkera9e85c02012-03-23 00:28:20 -0400975 } else p = load_library(file);
976
977 if (!p) {
Rich Felkera5d10eb2012-04-23 12:03:31 -0400978 snprintf(errbuf, sizeof errbuf,
979 "Error loading shared library %s: %m", file);
Rich Felkera9e85c02012-03-23 00:28:20 -0400980 errflag = 1;
Rich Felkerf2baf4d2012-02-07 20:31:27 -0500981 goto end;
Rich Felker59ab43f2011-06-26 19:23:28 -0400982 }
983
Rich Felker59ab43f2011-06-26 19:23:28 -0400984 /* First load handling */
985 if (!p->deps) {
986 load_deps(p);
Rich Felker0e4dae32011-06-26 21:36:44 -0400987 if (p->deps) for (i=0; p->deps[i]; i++)
Rich Felker92ab5d82011-06-26 21:21:04 -0400988 if (!p->deps[i]->global)
989 p->deps[i]->global = -1;
990 if (!p->global) p->global = -1;
Rich Felker59ab43f2011-06-26 19:23:28 -0400991 reloc_all(p);
Rich Felker0e4dae32011-06-26 21:36:44 -0400992 if (p->deps) for (i=0; p->deps[i]; i++)
Rich Felker92ab5d82011-06-26 21:21:04 -0400993 if (p->deps[i]->global < 0)
994 p->deps[i]->global = 0;
995 if (p->global < 0) p->global = 0;
Rich Felker59ab43f2011-06-26 19:23:28 -0400996 }
997
998 if (mode & RTLD_GLOBAL) {
Rich Felker0e4dae32011-06-26 21:36:44 -0400999 if (p->deps) for (i=0; p->deps[i]; i++)
Rich Felker59ab43f2011-06-26 19:23:28 -04001000 p->deps[i]->global = 1;
1001 p->global = 1;
1002 }
1003
Rich Felkerdcd60372012-10-05 11:51:50 -04001004 __libc.tls_size = tls_size;
1005
Rich Felker731e8ff2012-08-25 17:24:46 -04001006 if (ssp_used) __init_ssp(auxv);
1007
Rich Felker3ec8d292012-04-25 00:05:42 -04001008 _dl_debug_state();
1009
Rich Felkerce4d97e2012-02-06 17:57:29 -05001010 do_init_fini(tail);
Rich Felker06933cc2011-06-26 22:09:32 -04001011end:
Rich Felkerdcd60372012-10-05 11:51:50 -04001012 __release_ptc();
Rich Felker59ab43f2011-06-26 19:23:28 -04001013 pthread_rwlock_unlock(&lock);
Rich Felkerf2baf4d2012-02-07 20:31:27 -05001014 pthread_setcancelstate(cs, 0);
Rich Felker59ab43f2011-06-26 19:23:28 -04001015 return p;
1016}
1017
Rich Felker623753a2011-08-16 00:42:13 -04001018static void *do_dlsym(struct dso *p, const char *s, void *ra)
Rich Felker59ab43f2011-06-26 19:23:28 -04001019{
1020 size_t i;
Rich Felker2bd05a42012-08-25 17:13:28 -04001021 uint32_t h = 0, gh = 0;
Rich Felker59ab43f2011-06-26 19:23:28 -04001022 Sym *sym;
Rich Felker9c748562012-10-04 22:48:33 -04001023 if (p == head || p == RTLD_DEFAULT || p == RTLD_NEXT) {
1024 if (p == RTLD_NEXT) {
1025 for (p=head; p && (unsigned char *)ra-p->map>p->map_len; p=p->next);
1026 if (!p) p=head;
1027 }
1028 struct symdef def = find_sym(p->next, s, 0);
1029 if (!def.sym) goto failed;
1030 return def.dso->base + def.sym->st_value;
Rich Felkera9e85c02012-03-23 00:28:20 -04001031 }
Rich Felker2bd05a42012-08-25 17:13:28 -04001032 if (p->ghashtab) {
1033 gh = gnu_hash(s);
1034 sym = gnu_lookup(s, gh, p);
1035 } else {
1036 h = sysv_hash(s);
1037 sym = sysv_lookup(s, h, p);
1038 }
Rich Felker59ab43f2011-06-26 19:23:28 -04001039 if (sym && sym->st_value && (1<<(sym->st_info&0xf) & OK_TYPES))
1040 return p->base + sym->st_value;
1041 if (p->deps) for (i=0; p->deps[i]; i++) {
Rich Felker2bd05a42012-08-25 17:13:28 -04001042 if (p->deps[i]->ghashtab) {
1043 if (!gh) gh = gnu_hash(s);
Rich Felkera5d61992012-08-25 17:40:27 -04001044 sym = gnu_lookup(s, gh, p->deps[i]);
Rich Felker2bd05a42012-08-25 17:13:28 -04001045 } else {
1046 if (!h) h = sysv_hash(s);
1047 sym = sysv_lookup(s, h, p->deps[i]);
1048 }
Rich Felker59ab43f2011-06-26 19:23:28 -04001049 if (sym && sym->st_value && (1<<(sym->st_info&0xf) & OK_TYPES))
1050 return p->deps[i]->base + sym->st_value;
1051 }
Rich Felker4027f4e2012-05-04 20:18:18 -04001052failed:
Rich Felkera9e85c02012-03-23 00:28:20 -04001053 errflag = 1;
Rich Felkera5d10eb2012-04-23 12:03:31 -04001054 snprintf(errbuf, sizeof errbuf, "Symbol not found: %s", s);
Rich Felker59ab43f2011-06-26 19:23:28 -04001055 return 0;
1056}
1057
Rich Felkerf419bcb2012-08-26 21:09:26 -04001058int __dladdr(void *addr, Dl_info *info)
1059{
1060 struct dso *p;
1061 Sym *sym;
1062 uint32_t nsym;
1063 char *strings;
1064 size_t i;
1065 void *best = 0;
1066 char *bestname;
1067
1068 pthread_rwlock_rdlock(&lock);
1069 for (p=head; p && (unsigned char *)addr-p->map>p->map_len; p=p->next);
1070 pthread_rwlock_unlock(&lock);
1071
1072 if (!p) return 0;
1073
1074 sym = p->syms;
1075 strings = p->strings;
1076 if (p->hashtab) {
1077 nsym = p->hashtab[1];
1078 } else {
1079 uint32_t *buckets;
1080 uint32_t *hashval;
1081 buckets = p->ghashtab + 4 + (p->ghashtab[2]*sizeof(size_t)/4);
1082 sym += p->ghashtab[1];
1083 for (i = 0; i < p->ghashtab[0]; i++) {
1084 if (buckets[i] > nsym)
1085 nsym = buckets[i];
1086 }
1087 if (nsym) {
1088 nsym -= p->ghashtab[1];
1089 hashval = buckets + p->ghashtab[0] + nsym;
1090 do nsym++;
1091 while (!(*hashval++ & 1));
1092 }
1093 }
1094
1095 for (; nsym; nsym--, sym++) {
1096 if (sym->st_shndx && sym->st_value
1097 && (1<<(sym->st_info&0xf) & OK_TYPES)
1098 && (1<<(sym->st_info>>4) & OK_BINDS)) {
1099 void *symaddr = p->base + sym->st_value;
1100 if (symaddr > addr || symaddr < best)
1101 continue;
1102 best = symaddr;
1103 bestname = strings + sym->st_name;
1104 if (addr == symaddr)
1105 break;
1106 }
1107 }
1108
1109 if (!best) return 0;
1110
1111 info->dli_fname = p->name;
1112 info->dli_fbase = p->base;
1113 info->dli_sname = bestname;
1114 info->dli_saddr = best;
1115
1116 return 1;
1117}
1118
Rich Felker400c5e52012-09-06 22:44:55 -04001119void *__dlsym(void *restrict p, const char *restrict s, void *restrict ra)
Rich Felker59ab43f2011-06-26 19:23:28 -04001120{
1121 void *res;
1122 pthread_rwlock_rdlock(&lock);
Rich Felker623753a2011-08-16 00:42:13 -04001123 res = do_dlsym(p, s, ra);
Rich Felker59ab43f2011-06-26 19:23:28 -04001124 pthread_rwlock_unlock(&lock);
1125 return res;
1126}
Rich Felker5a09a532012-02-03 03:16:07 -05001127#else
1128void *dlopen(const char *file, int mode)
1129{
1130 return 0;
1131}
Rich Felker400c5e52012-09-06 22:44:55 -04001132void *__dlsym(void *restrict p, const char *restrict s, void *restrict ra)
Rich Felker5a09a532012-02-03 03:16:07 -05001133{
1134 return 0;
1135}
Rich Felkerf419bcb2012-08-26 21:09:26 -04001136int __dladdr (void *addr, Dl_info *info)
1137{
1138 return 0;
1139}
Rich Felker5a09a532012-02-03 03:16:07 -05001140#endif
Rich Felker59ab43f2011-06-26 19:23:28 -04001141
1142char *dlerror()
1143{
Rich Felkera9e85c02012-03-23 00:28:20 -04001144 if (!errflag) return 0;
1145 errflag = 0;
Rich Felkera5d10eb2012-04-23 12:03:31 -04001146 return errbuf;
Rich Felker59ab43f2011-06-26 19:23:28 -04001147}
1148
1149int dlclose(void *p)
1150{
1151 return 0;
1152}