tac: *really* add tac.c now
[oweals/busybox.git] / runit / chpst.c
index 14b8b5a157ae59e680a606196bbda3ab5dd084df..0f605cc455ffbae8af072ba838a17ad1275f22b8 100644 (file)
@@ -28,7 +28,7 @@ ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 /* Busyboxed by Denis Vlasenko <vda.linux@googlemail.com> */
 /* Dependencies on runit_lib.c removed */
 
-#include "busybox.h"
+#include "libbb.h"
 
 #include <dirent.h>
 
@@ -39,21 +39,43 @@ ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 #define OPT_nostdout (option_mask32 & 0x10000)
 #define OPT_nostderr (option_mask32 & 0x20000)
 
-static char *set_user;
-static char *env_user;
-static const char *env_dir;
-static long limitd = -2;
-static long limits = -2;
-static long limitl = -2;
-static long limita = -2;
-static long limito = -2;
-static long limitp = -2;
-static long limitf = -2;
-static long limitc = -2;
-static long limitr = -2;
-static long limitt = -2;
-static int nicelvl;
-static const char *root;
+struct globals {
+       char *set_user;
+       char *env_user;
+       const char *env_dir;
+       const char *root;
+       long limitd; /* limitX are initialized to -2 */
+       long limits;
+       long limitl;
+       long limita;
+       long limito;
+       long limitp;
+       long limitf;
+       long limitc;
+       long limitr;
+       long limitt;
+       int nicelvl;
+};
+#define G (*(struct globals*)&bb_common_bufsiz1)
+#define set_user (G.set_user)
+#define env_user (G.env_user)
+#define env_dir  (G.env_dir )
+#define root     (G.root    )
+#define limitd   (G.limitd  )
+#define limits   (G.limits  )
+#define limitl   (G.limitl  )
+#define limita   (G.limita  )
+#define limito   (G.limito  )
+#define limitp   (G.limitp  )
+#define limitf   (G.limitf  )
+#define limitc   (G.limitc  )
+#define limitr   (G.limitr  )
+#define limitt   (G.limitt  )
+#define nicelvl  (G.nicelvl )
+#define INIT_G() do { \
+       long *p = &limitd; \
+       do *p++ = -2; while (p <= &limitt); \
+} while (0)
 
 static void suidgid(char *user)
 {
@@ -100,7 +122,8 @@ static void edir(const char *directory_name)
                                                directory_name);
                        break;
                }
-               if (d->d_name[0] == '.') continue;
+               if (d->d_name[0] == '.')
+                       continue;
                fd = open(d->d_name, O_RDONLY | O_NDELAY);
                if (fd < 0) {
                        if ((errno == EISDIR) && env_dir) {
@@ -129,9 +152,9 @@ static void edir(const char *directory_name)
                        tail = memchr(buf, '\n', sizeof(buf));
                        /* skip trailing whitespace */;
                        while (1) {
-                               if (tail[0]==' ') tail[0] = '\0';
-                               if (tail[0]=='\t') tail[0] = '\0';
-                               if (tail[0]=='\n') tail[0] = '\0';
+                               if (tail[0] == ' ') tail[0] = '\0';
+                               if (tail[0] == '\t') tail[0] = '\0';
+                               if (tail[0] == '\n') tail[0] = '\0';
                                if (tail == buf) break;
                                tail--;
                        }
@@ -139,7 +162,8 @@ static void edir(const char *directory_name)
                }
        }
        closedir(dir);
-       if (fchdir(wdir) == -1) bb_perror_msg_and_die("fchdir");
+       if (fchdir(wdir) == -1)
+               bb_perror_msg_and_die("fchdir");
        close(wdir);
 }
 
@@ -147,12 +171,14 @@ static void limit(int what, long l)
 {
        struct rlimit r;
 
-       if (getrlimit(what, &r) == -1) bb_perror_msg_and_die("getrlimit");
+       if (getrlimit(what, &r) == -1)
+               bb_perror_msg_and_die("getrlimit");
        if ((l < 0) || (l > r.rlim_max))
                r.rlim_cur = r.rlim_max;
        else
                r.rlim_cur = l;
-       if (setrlimit(what, &r) == -1) bb_perror_msg_and_die("setrlimit");
+       if (setrlimit(what, &r) == -1)
+               bb_perror_msg_and_die("setrlimit");
 }
 
 static void slimit(void)
@@ -161,24 +187,27 @@ static void slimit(void)
 #ifdef RLIMIT_DATA
                limit(RLIMIT_DATA, limitd);
 #else
-               if (OPT_verbose) bb_error_msg("system does not support %s",
-                               "RLIMIT_DATA");
+               if (OPT_verbose)
+                       bb_error_msg("system does not support RLIMIT_%s",
+                               "DATA");
 #endif
        }
        if (limits >= -1) {
 #ifdef RLIMIT_STACK
                limit(RLIMIT_STACK, limits);
 #else
-               if (OPT_verbose) bb_error_msg("system does not support %s",
-                               "RLIMIT_STACK");
+               if (OPT_verbose)
+                       bb_error_msg("system does not support RLIMIT_%s",
+                               "STACK");
 #endif
        }
        if (limitl >= -1) {
 #ifdef RLIMIT_MEMLOCK
                limit(RLIMIT_MEMLOCK, limitl);
 #else
-               if (OPT_verbose) bb_error_msg("system does not support %s",
-                               "RLIMIT_MEMLOCK");
+               if (OPT_verbose)
+                       bb_error_msg("system does not support RLIMIT_%s",
+                               "MEMLOCK");
 #endif
        }
        if (limita >= -1) {
@@ -189,8 +218,8 @@ static void slimit(void)
                limit(RLIMIT_AS, limita);
 #else
                if (OPT_verbose)
-                       bb_error_msg("system does not support %s",
-                               "RLIMIT_VMEM");
+                       bb_error_msg("system does not support RLIMIT_%s",
+                               "VMEM");
 #endif
 #endif
        }
@@ -202,8 +231,8 @@ static void slimit(void)
                limit(RLIMIT_OFILE, limito);
 #else
                if (OPT_verbose)
-                       bb_error_msg("system does not support %s",
-                               "RLIMIT_NOFILE");
+                       bb_error_msg("system does not support RLIMIT_%s",
+                               "NOFILE");
 #endif
 #endif
        }
@@ -211,52 +240,60 @@ static void slimit(void)
 #ifdef RLIMIT_NPROC
                limit(RLIMIT_NPROC, limitp);
 #else
-               if (OPT_verbose) bb_error_msg("system does not support %s",
-                               "RLIMIT_NPROC");
+               if (OPT_verbose)
+                       bb_error_msg("system does not support RLIMIT_%s",
+                               "NPROC");
 #endif
        }
        if (limitf >= -1) {
 #ifdef RLIMIT_FSIZE
                limit(RLIMIT_FSIZE, limitf);
 #else
-               if (OPT_verbose) bb_error_msg("system does not support %s",
-                               "RLIMIT_FSIZE");
+               if (OPT_verbose)
+                       bb_error_msg("system does not support RLIMIT_%s",
+                               "FSIZE");
 #endif
        }
        if (limitc >= -1) {
 #ifdef RLIMIT_CORE
                limit(RLIMIT_CORE, limitc);
 #else
-               if (OPT_verbose) bb_error_msg("system does not support %s",
-                               "RLIMIT_CORE");
+               if (OPT_verbose)
+                       bb_error_msg("system does not support RLIMIT_%s",
+                               "CORE");
 #endif
        }
        if (limitr >= -1) {
 #ifdef RLIMIT_RSS
                limit(RLIMIT_RSS, limitr);
 #else
-               if (OPT_verbose) bb_error_msg("system does not support %s",
-                               "RLIMIT_RSS");
+               if (OPT_verbose)
+                       bb_error_msg("system does not support RLIMIT_%s",
+                               "RSS");
 #endif
        }
        if (limitt >= -1) {
 #ifdef RLIMIT_CPU
                limit(RLIMIT_CPU, limitt);
 #else
-               if (OPT_verbose) bb_error_msg("system does not support %s",
-                               "RLIMIT_CPU");
+               if (OPT_verbose)
+                       bb_error_msg("system does not support RLIMIT_%s",
+                               "CPU");
 #endif
        }
 }
 
 /* argv[0] */
-static void setuidgid(int, char **);
-static void envuidgid(int, char **);
-static void envdir(int, char **);
-static void softlimit(int, char **);
+static void setuidgid(int, char **) ATTRIBUTE_NORETURN;
+static void envuidgid(int, char **) ATTRIBUTE_NORETURN;
+static void envdir(int, char **) ATTRIBUTE_NORETURN;
+static void softlimit(int, char **) ATTRIBUTE_NORETURN;
 
+int chpst_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
 int chpst_main(int argc, char **argv)
 {
+       INIT_G();
+
        if (applet_name[3] == 'd') envdir(argc, argv);
        if (applet_name[1] == 'o') softlimit(argc, argv);
        if (applet_name[0] == 's') setuidgid(argc, argv);
@@ -265,7 +302,7 @@ int chpst_main(int argc, char **argv)
 
        {
                char *m,*d,*o,*p,*f,*c,*r,*t,*n;
-               getopt32(argc, argv, "+u:U:e:m:d:o:p:f:c:r:t:/:n:vP012",
+               getopt32(argv, "+u:U:e:m:d:o:p:f:c:r:t:/:n:vP012",
                                &set_user,&env_user,&env_dir,
                                &m,&d,&o,&p,&f,&c,&r,&t,&root,&n);
                // if (option_mask32 & 0x1) // -u
@@ -290,7 +327,7 @@ int chpst_main(int argc, char **argv)
        }
        argv += optind;
        if (!argv || !*argv) bb_show_usage();
-       
+
        if (OPT_pgrp) setsid();
        if (env_dir) edir(env_dir);
        if (root) {
@@ -309,7 +346,7 @@ int chpst_main(int argc, char **argv)
        if (OPT_nostdin) close(0);
        if (OPT_nostdout) close(1);
        if (OPT_nostderr) close(2);
-       execvp(argv[0], argv);
+       BB_EXECVP(argv[0], argv);
        bb_perror_msg_and_die("exec %s", argv[0]);
 }
 
@@ -321,7 +358,7 @@ static void setuidgid(int argc, char **argv)
        if (!account) bb_show_usage();
        if (!*++argv) bb_show_usage();
        suidgid((char*)account);
-       execvp(argv[0], argv);
+       BB_EXECVP(argv[0], argv);
        bb_perror_msg_and_die("exec %s", argv[0]);
 }
 
@@ -333,7 +370,7 @@ static void envuidgid(int argc, char **argv)
        if (!account) bb_show_usage();
        if (!*++argv) bb_show_usage();
        euidgid((char*)account);
-       execvp(argv[0], argv);
+       BB_EXECVP(argv[0], argv);
        bb_perror_msg_and_die("exec %s", argv[0]);
 }
 
@@ -345,14 +382,14 @@ static void envdir(int argc, char **argv)
        if (!dir) bb_show_usage();
        if (!*++argv) bb_show_usage();
        edir(dir);
-       execvp(argv[0], argv);
+       BB_EXECVP(argv[0], argv);
        bb_perror_msg_and_die("exec %s", argv[0]);
 }
 
 static void softlimit(int argc, char **argv)
 {
        char *a,*c,*d,*f,*l,*m,*o,*p,*r,*s,*t;
-       getopt32(argc, argv, "+a:c:d:f:l:m:o:p:r:s:t:",
+       getopt32(argv, "+a:c:d:f:l:m:o:p:r:s:t:",
                        &a,&c,&d,&f,&l,&m,&o,&p,&r,&s,&t);
        if (option_mask32 & 0x001) limita = xatoul(a); // -a
        if (option_mask32 & 0x002) limitc = xatoul(c); // -c
@@ -368,6 +405,6 @@ static void softlimit(int argc, char **argv)
        argv += optind;
        if (!argv[0]) bb_show_usage();
        slimit();
-       execvp(argv[0], argv);
+       BB_EXECVP(argv[0], argv);
        bb_perror_msg_and_die("exec %s", argv[0]);
 }