chpst: stop using data/bss
authorDenis Vlasenko <vda.linux@googlemail.com>
Fri, 5 Oct 2007 21:23:49 +0000 (21:23 -0000)
committerDenis Vlasenko <vda.linux@googlemail.com>
Fri, 5 Oct 2007 21:23:49 +0000 (21:23 -0000)
function                                             old     new   delta
chpst_main                                          1066    1089     +23
set_user                                               4       -      -4
root                                                   4       -      -4
nicelvl                                                4       -      -4
limitt                                                 4       -      -4
limits                                               196     192      -4
limitr                                                 4       -      -4
limitp                                                 4       -      -4
limito                                                 4       -      -4
limitl                                                 4       -      -4
limitf                                                 4       -      -4
limitd                                                 4       -      -4
limitc                                                 4       -      -4
limita                                                 4       -      -4
env_user                                               4       -      -4
env_dir                                                4       -      -4
------------------------------------------------------------------------------
(add/remove: 0/14 grow/shrink: 1/1 up/down: 23/-60)           Total: -37 bytes
   text    data     bss     dec     hex filename
 770892    1029    9520  781441   bec81 busybox_old
 770916     989    9496  781401   bec59 busybox_unstripped

runit/chpst.c

index 87cd8e04a281b8ab218f488f5f0eac72bcdb11fe..0c26d836e38fb002c7c68cfce0044e6d949b0129 100644 (file)
@@ -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,53 +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);
 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);