nsz Git - musl/blob - 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 void _dlstart_c(size_t *sp, size_t *dynv)
  13 {
  14         size_t i, aux[AUX_CNT], dyn[DYN_CNT];
  15
  16         int argc = *sp;
  17         char **argv = (void *)(sp+1);
  18
  19         for (i=argc+1; argv[i]; i++);
  20         size_t *auxv = (void *)(argv+i+1);
  21
  22         for (i=0; i<AUX_CNT; i++) aux[i] = 0;
  23         for (i=0; auxv[i]; i+=2) if (auxv[i]<AUX_CNT)
  24                 aux[auxv[i]] = auxv[i+1];
  25
  26         for (i=0; i<DYN_CNT; i++) dyn[i] = 0;
  27         for (i=0; dynv[i]; i+=2) if (dynv[i]<DYN_CNT)
  28                 dyn[dynv[i]] = dynv[i+1];
  29
  30         /* If the dynamic linker is invoked as a command, its load
  31          * address is not available in the aux vector. Instead, compute
  32          * the load address as the difference between &_DYNAMIC and the
  33          * virtual address in the PT_DYNAMIC program header. */
  34         unsigned char *base = (void *)aux[AT_BASE];
  35         if (!base) {
  36                 size_t phnum = aux[AT_PHNUM];
  37                 size_t phentsize = aux[AT_PHENT];
  38                 Phdr *ph = (void *)aux[AT_PHDR];
  39                 for (i=phnum; i--; ph = (void *)((char *)ph + phentsize)) {
  40                         if (ph->p_type == PT_DYNAMIC) {
  41                                 base = (void *)((size_t)dynv - ph->p_vaddr);
  42                                 break;
  43                         }
  44                 }
  45         }
  46
  47         /* MIPS uses an ugly packed form for GOT relocations. Since we
  48          * can't make function calls yet and the code is tiny anyway,
  49          * it's simply inlined here. */
  50         if (NEED_MIPS_GOT_RELOCS) {
  51                 size_t local_cnt = 0;
  52                 size_t *got = (void *)(base + dyn[DT_PLTGOT]);
  53                 for (i=0; dynv[i]; i+=2) if (dynv[i]==DT_MIPS_LOCAL_GOTNO)
  54                         local_cnt = dynv[i+1];
  55                 for (i=0; i<local_cnt; i++) got[i] += (size_t)base;
  56         }
  57
  58         /* The use of the reloc_info structure and nested loops is a trick
  59          * to work around the fact that we can't necessarily make function
  60          * calls yet. Each struct in the array serves like the arguments
  61          * to a function call. */
  62         struct {
  63                 void *rel;
  64                 size_t size;
  65                 size_t stride;
  66         } reloc_info[] = {
  67                 { base+dyn[DT_JMPREL], dyn[DT_PLTRELSZ], 2+(dyn[DT_PLTREL]==DT_RELA) },
  68                 { base+dyn[DT_REL], dyn[DT_RELSZ], 2 },
  69                 { base+dyn[DT_RELA], dyn[DT_RELASZ], 3 },
  70                 { 0, 0, 0 }
  71         };
  72
  73         for (i=0; reloc_info[i].stride; i++) {
  74                 size_t *rel = reloc_info[i].rel;
  75                 size_t rel_size = reloc_info[i].size;
  76                 size_t stride = reloc_info[i].stride;
  77                 for (; rel_size; rel+=stride, rel_size-=stride*sizeof(size_t)) {
  78                         if (!IS_RELATIVE(rel[1])) continue;
  79                         size_t *rel_addr = (void *)(base + rel[0]);
  80                         size_t addend = stride==3 ? rel[2] : *rel_addr;
  81                         *rel_addr = (size_t)base + addend;
  82                 }
  83         }
  84
  85         const char *strings = (void *)(base + dyn[DT_STRTAB]);
  86         const Sym *syms = (void *)(base + dyn[DT_SYMTAB]);
  87
  88         /* Call dynamic linker stage-2, __dls2 */
  89         for (i=0; ;i++) {
  90                 const char *s = strings + syms[i].st_name;
  91                 if (s[0]=='_' && s[1]=='_' && s[2]=='d'
  92                  && s[3]=='l' && s[4]=='s' && s[5]=='2' && !s[6])
  93                         break;
  94         }
  95         ((stage2_func)(base + syms[i].st_value))(base);
  96
  97         /* Call dynamic linker stage-3, __dls3 */
  98         for (i=0; ;i++) {
  99                 const char *s = strings + syms[i].st_name;
 100                 if (s[0]=='_' && s[1]=='_' && s[2]=='d'
 101                  && s[3]=='l' && s[4]=='s' && s[5]=='3' && !s[6])
 102                         break;
 103         }
 104         ((stage3_func)(base + syms[i].st_value))(sp);
 105 }
 106
 107 #endif