simplify/shrink relocation processing in dynamic linker stage 1
[musl] / src / ldso / dlstart.c
1 #include <stddef.h>
2 #include "dynlink.h"
3
4 #ifdef SHARED
5
6 #ifndef START
7 #define START "_dlstart"
8 #endif
9
10 #include "crt_arch.h"
11
12 __attribute__((__visibility__("hidden")))
13 void _dlstart_c(size_t *sp, size_t *dynv)
14 {
15         size_t i, aux[AUX_CNT], dyn[DYN_CNT];
16
17         int argc = *sp;
18         char **argv = (void *)(sp+1);
19
20         for (i=argc+1; argv[i]; i++);
21         size_t *auxv = (void *)(argv+i+1);
22
23         for (i=0; i<AUX_CNT; i++) aux[i] = 0;
24         for (i=0; auxv[i]; i+=2) if (auxv[i]<AUX_CNT)
25                 aux[auxv[i]] = auxv[i+1];
26
27         for (i=0; i<DYN_CNT; i++) dyn[i] = 0;
28         for (i=0; dynv[i]; i+=2) if (dynv[i]<DYN_CNT)
29                 dyn[dynv[i]] = dynv[i+1];
30
31         /* If the dynamic linker is invoked as a command, its load
32          * address is not available in the aux vector. Instead, compute
33          * the load address as the difference between &_DYNAMIC and the
34          * virtual address in the PT_DYNAMIC program header. */
35         unsigned char *base = (void *)aux[AT_BASE];
36         if (!base) {
37                 size_t phnum = aux[AT_PHNUM];
38                 size_t phentsize = aux[AT_PHENT];
39                 Phdr *ph = (void *)aux[AT_PHDR];
40                 for (i=phnum; i--; ph = (void *)((char *)ph + phentsize)) {
41                         if (ph->p_type == PT_DYNAMIC) {
42                                 base = (void *)((size_t)dynv - ph->p_vaddr);
43                                 break;
44                         }
45                 }
46         }
47
48         /* MIPS uses an ugly packed form for GOT relocations. Since we
49          * can't make function calls yet and the code is tiny anyway,
50          * it's simply inlined here. */
51         if (NEED_MIPS_GOT_RELOCS) {
52                 size_t local_cnt = 0;
53                 size_t *got = (void *)(base + dyn[DT_PLTGOT]);
54                 for (i=0; dynv[i]; i+=2) if (dynv[i]==DT_MIPS_LOCAL_GOTNO)
55                         local_cnt = dynv[i+1];
56                 for (i=0; i<local_cnt; i++) got[i] += (size_t)base;
57         }
58
59         size_t *rel, rel_size;
60
61         rel = (void *)(base+dyn[DT_REL]);
62         rel_size = dyn[DT_RELSZ];
63         for (; rel_size; rel+=2, rel_size-=2*sizeof(size_t)) {
64                 if (!IS_RELATIVE(rel[1])) continue;
65                 size_t *rel_addr = (void *)(base + rel[0]);
66                 *rel_addr += (size_t)base;
67         }
68
69         rel = (void *)(base+dyn[DT_RELA]);
70         rel_size = dyn[DT_RELASZ];
71         for (; rel_size; rel+=3, rel_size-=3*sizeof(size_t)) {
72                 if (!IS_RELATIVE(rel[1])) continue;
73                 size_t *rel_addr = (void *)(base + rel[0]);
74                 *rel_addr = (size_t)base + rel[2];
75         }
76
77         const char *strings = (void *)(base + dyn[DT_STRTAB]);
78         const Sym *syms = (void *)(base + dyn[DT_SYMTAB]);
79
80         /* Call dynamic linker stage-2, __dls2 */
81         for (i=0; ;i++) {
82                 const char *s = strings + syms[i].st_name;
83                 if (s[0]=='_' && s[1]=='_' && s[2]=='d'
84                  && s[3]=='l' && s[4]=='s' && s[5]=='2' && !s[6])
85                         break;
86         }
87         ((stage2_func)(base + syms[i].st_value))(base);
88
89         /* Call dynamic linker stage-3, __dls3 */
90         for (i=0; ;i++) {
91                 const char *s = strings + syms[i].st_name;
92                 if (s[0]=='_' && s[1]=='_' && s[2]=='d'
93                  && s[3]=='l' && s[4]=='s' && s[5]=='3' && !s[6])
94                         break;
95         }
96         ((stage3_func)(base + syms[i].st_value))(sp);
97 }
98
99 #endif