nsz Git - musl/blob - src/env/__libc_start_main.c

   1 #include <elf.h>
   2 #include <poll.h>
   3 #include <fcntl.h>
   4 #include "syscall.h"
   5 #include "atomic.h"
   6 #include "libc.h"
   7
   8 void __init_tls(size_t *);
   9
  10 #ifndef SHARED
  11 static void dummy() {}
  12 weak_alias(dummy, _init);
  13 extern void (*const __init_array_start)() __attribute__((weak));
  14 extern void (*const __init_array_end)() __attribute__((weak));
  15 #endif
  16
  17 static void dummy1(void *p) {}
  18 weak_alias(dummy1, __init_ssp);
  19
  20 #define AUX_CNT 38
  21
  22 extern size_t __hwcap, __sysinfo;
  23 extern char *__progname, *__progname_full;
  24
  25 #ifndef SHARED
  26 static
  27 #endif
  28 void __init_libc(char **envp, char *pn)
  29 {
  30         size_t i, *auxv, aux[AUX_CNT] = { 0 };
  31         __environ = envp;
  32         for (i=0; envp[i]; i++);
  33         libc.auxv = auxv = (void *)(envp+i+1);
  34         for (i=0; auxv[i]; i+=2) if (auxv[i]<AUX_CNT) aux[auxv[i]] = auxv[i+1];
  35         __hwcap = aux[AT_HWCAP];
  36         __sysinfo = aux[AT_SYSINFO];
  37         libc.page_size = aux[AT_PAGESZ];
  38
  39         if (pn) {
  40                 __progname = __progname_full = pn;
  41                 for (i=0; pn[i]; i++) if (pn[i]=='/') __progname = pn+i+1;
  42         }
  43
  44         __init_tls(aux);
  45         __init_ssp((void *)aux[AT_RANDOM]);
  46
  47         if (aux[AT_UID]==aux[AT_EUID] && aux[AT_GID]==aux[AT_EGID]
  48                 && !aux[AT_SECURE]) return;
  49
  50         struct pollfd pfd[3] = { {.fd=0}, {.fd=1}, {.fd=2} };
  51         __syscall(SYS_poll, pfd, 3, 0);
  52         for (i=0; i<3; i++) if (pfd[i].revents&POLLNVAL)
  53                 if (__syscall(SYS_open, "/dev/null", O_RDWR|O_LARGEFILE)<0)
  54                         a_crash();
  55         libc.secure = 1;
  56 }
  57
  58 int __libc_start_main(int (*main)(int,char **,char **), int argc, char **argv)
  59 {
  60         char **envp = argv+argc+1;
  61
  62 #ifndef SHARED
  63         __init_libc(envp, argv[0]);
  64         _init();
  65         uintptr_t a = (uintptr_t)&__init_array_start;
  66         for (; a<(uintptr_t)&__init_array_end; a+=sizeof(void(*)()))
  67                 (*(void (**)())a)();
  68 #endif
  69
  70         /* Pass control to to application */
  71         exit(main(argc, argv, envp));
  72         return 0;
  73 }