add support for program_invocation[_short]_name

[musl] / src / env / __libc_start_main.c

diff --git a/src/env/__libc_start_main.c b/src/env/__libc_start_main.c
index e350fdb..aef9f9e 100644 (file)
--- a/src/env/__libc_start_main.c
+++ b/src/env/__libc_start_main.c
@@ -3,10 +3,14 @@
 
 void __init_tls(size_t *);
 void __init_security(size_t *);
+void __init_ldso_ctors(void);
 
 #define AUX_CNT 38
 
-void __init_libc(char **envp)
+extern size_t __hwcap, __sysinfo;
+extern char *__progname, *__progname_full;
+
+void __init_libc(char **envp, char *pn)
 {
        size_t i, *auxv, aux[AUX_CNT] = { 0 };
        __environ = envp;
@@ -14,6 +18,12 @@ void __init_libc(char **envp)
        libc.auxv = auxv = (void *)(envp+i+1);
        for (i=0; auxv[i]; i+=2) if (auxv[i]<AUX_CNT) aux[auxv[i]] = auxv[i+1];
        __hwcap = aux[AT_HWCAP];
+       __sysinfo = aux[AT_SYSINFO];
+
+       if (pn) {
+               __progname = __progname_full = pn;
+               for (i=0; pn[i]; i++) if (pn[i]=='/') __progname = pn+i+1;
+       }
 
        __init_tls(aux);
        __init_security(aux);
@@ -26,7 +36,7 @@ int __libc_start_main(
 {
        char **envp = argv+argc+1;
 
-       __init_libc(envp);
+       __init_libc(envp, argv[0]);
 
        libc.ldso_fini = ldso_fini;
        libc.fini = fini;
@@ -34,6 +44,10 @@ int __libc_start_main(
        /* Execute constructors (static) linked into the application */
        if (init) init(argc, argv, envp);
 
+#ifdef SHARED
+       __init_ldso_ctors();
+#endif
+
        /* Pass control to to application */
        exit(main(argc, argv, envp));
        return 0;