Magellan Linux

  • Back to homepage
    • /[magellan-source]/trunk/mkinitrd-magellan/busybox/runit/chpst.c

    Contents of /trunk/mkinitrd-magellan/busybox/runit/chpst.c

    Parent Directory Parent Directory | Revision Log Revision Log

    Revision 532 - (show annotations) (download)
    Sat Sep 1 22:45:15 2007 UTC (16 years, 9 months ago) by niro
    File MIME type: text/plain
    File size: 10033 byte(s) 
    -import if magellan mkinitrd; it is a fork of redhats mkinitrd-5.0.8 with all magellan patches and features; deprecates magellan-src/mkinitrd
    1 /*
    2 Copyright (c) 2001-2006, Gerrit Pape
    3 All rights reserved.
    4
    5 Redistribution and use in source and binary forms, with or without
    6 modification, are permitted provided that the following conditions are met:
    7
    8 1. Redistributions of source code must retain the above copyright notice,
    9 this list of conditions and the following disclaimer.
    10 2. Redistributions in binary form must reproduce the above copyright
    11 notice, this list of conditions and the following disclaimer in the
    12 documentation and/or other materials provided with the distribution.
    13 3. The name of the author may not be used to endorse or promote products
    14 derived from this software without specific prior written permission.
    15
    16 THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR IMPLIED
    17 WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
    18 MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO
    19 EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
    20 SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
    21 PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;
    22 OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
    23 WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
    24 OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
    25 ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
    26 */
    27
    28 /* Busyboxed by Denis Vlasenko <vda.linux@googlemail.com> */
    29 /* Dependencies on runit_lib.c removed */
    30
    31 #include "busybox.h"
    32
    33 #include <dirent.h>
    34
    35 // Must match constants in chpst_main!
    36 #define OPT_verbose (option_mask32 & 0x2000)
    37 #define OPT_pgrp (option_mask32 & 0x4000)
    38 #define OPT_nostdin (option_mask32 & 0x8000)
    39 #define OPT_nostdout (option_mask32 & 0x10000)
    40 #define OPT_nostderr (option_mask32 & 0x20000)
    41
    42 static char *set_user;
    43 static char *env_user;
    44 static const char *env_dir;
    45 static long limitd = -2;
    46 static long limits = -2;
    47 static long limitl = -2;
    48 static long limita = -2;
    49 static long limito = -2;
    50 static long limitp = -2;
    51 static long limitf = -2;
    52 static long limitc = -2;
    53 static long limitr = -2;
    54 static long limitt = -2;
    55 static int nicelvl;
    56 static const char *root;
    57
    58 static void suidgid(char *user)
    59 {
    60 struct bb_uidgid_t ugid;
    61
    62 if (!get_uidgid(&ugid, user, 1)) {
    63 bb_error_msg_and_die("unknown user/group: %s", user);
    64 }
    65 if (setgroups(1, &ugid.gid) == -1)
    66 bb_perror_msg_and_die("setgroups");
    67 xsetgid(ugid.gid);
    68 xsetuid(ugid.uid);
    69 }
    70
    71 static void euidgid(char *user)
    72 {
    73 struct bb_uidgid_t ugid;
    74
    75 if (!get_uidgid(&ugid, user, 1)) {
    76 bb_error_msg_and_die("unknown user/group: %s", user);
    77 }
    78 xsetenv("GID", utoa(ugid.gid));
    79 xsetenv("UID", utoa(ugid.uid));
    80 }
    81
    82 static void edir(const char *directory_name)
    83 {
    84 int wdir;
    85 DIR *dir;
    86 struct dirent *d;
    87 int fd;
    88
    89 wdir = xopen(".", O_RDONLY | O_NDELAY);
    90 xchdir(directory_name);
    91 dir = opendir(".");
    92 if (!dir)
    93 bb_perror_msg_and_die("opendir %s", directory_name);
    94 for (;;) {
    95 errno = 0;
    96 d = readdir(dir);
    97 if (!d) {
    98 if (errno)
    99 bb_perror_msg_and_die("readdir %s",
    100 directory_name);
    101 break;
    102 }
    103 if (d->d_name[0] == '.') continue;
    104 fd = open(d->d_name, O_RDONLY | O_NDELAY);
    105 if (fd < 0) {
    106 if ((errno == EISDIR) && env_dir) {
    107 if (OPT_verbose)
    108 bb_perror_msg("warning: %s/%s is a directory",
    109 directory_name, d->d_name);
    110 continue;
    111 } else
    112 bb_perror_msg_and_die("open %s/%s",
    113 directory_name, d->d_name);
    114 }
    115 if (fd >= 0) {
    116 char buf[256];
    117 char *tail;
    118 int size;
    119
    120 size = safe_read(fd, buf, sizeof(buf)-1);
    121 if (size < 0)
    122 bb_perror_msg_and_die("read %s/%s",
    123 directory_name, d->d_name);
    124 if (size == 0) {
    125 unsetenv(d->d_name);
    126 continue;
    127 }
    128 buf[size] = '\n';
    129 tail = memchr(buf, '\n', sizeof(buf));
    130 /* skip trailing whitespace */;
    131 while (1) {
    132 if (tail[0]==' ') tail[0] = '\0';
    133 if (tail[0]=='\t') tail[0] = '\0';
    134 if (tail[0]=='\n') tail[0] = '\0';
    135 if (tail == buf) break;
    136 tail--;
    137 }
    138 xsetenv(d->d_name, buf);
    139 }
    140 }
    141 closedir(dir);
    142 if (fchdir(wdir) == -1) bb_perror_msg_and_die("fchdir");
    143 close(wdir);
    144 }
    145
    146 static void limit(int what, long l)
    147 {
    148 struct rlimit r;
    149
    150 if (getrlimit(what, &r) == -1) bb_perror_msg_and_die("getrlimit");
    151 if ((l < 0) || (l > r.rlim_max))
    152 r.rlim_cur = r.rlim_max;
    153 else
    154 r.rlim_cur = l;
    155 if (setrlimit(what, &r) == -1) bb_perror_msg_and_die("setrlimit");
    156 }
    157
    158 static void slimit(void)
    159 {
    160 if (limitd >= -1) {
    161 #ifdef RLIMIT_DATA
    162 limit(RLIMIT_DATA, limitd);
    163 #else
    164 if (OPT_verbose) bb_error_msg("system does not support %s",
    165 "RLIMIT_DATA");
    166 #endif
    167 }
    168 if (limits >= -1) {
    169 #ifdef RLIMIT_STACK
    170 limit(RLIMIT_STACK, limits);
    171 #else
    172 if (OPT_verbose) bb_error_msg("system does not support %s",
    173 "RLIMIT_STACK");
    174 #endif
    175 }
    176 if (limitl >= -1) {
    177 #ifdef RLIMIT_MEMLOCK
    178 limit(RLIMIT_MEMLOCK, limitl);
    179 #else
    180 if (OPT_verbose) bb_error_msg("system does not support %s",
    181 "RLIMIT_MEMLOCK");
    182 #endif
    183 }
    184 if (limita >= -1) {
    185 #ifdef RLIMIT_VMEM
    186 limit(RLIMIT_VMEM, limita);
    187 #else
    188 #ifdef RLIMIT_AS
    189 limit(RLIMIT_AS, limita);
    190 #else
    191 if (OPT_verbose)
    192 bb_error_msg("system does not support %s",
    193 "RLIMIT_VMEM");
    194 #endif
    195 #endif
    196 }
    197 if (limito >= -1) {
    198 #ifdef RLIMIT_NOFILE
    199 limit(RLIMIT_NOFILE, limito);
    200 #else
    201 #ifdef RLIMIT_OFILE
    202 limit(RLIMIT_OFILE, limito);
    203 #else
    204 if (OPT_verbose)
    205 bb_error_msg("system does not support %s",
    206 "RLIMIT_NOFILE");
    207 #endif
    208 #endif
    209 }
    210 if (limitp >= -1) {
    211 #ifdef RLIMIT_NPROC
    212 limit(RLIMIT_NPROC, limitp);
    213 #else
    214 if (OPT_verbose) bb_error_msg("system does not support %s",
    215 "RLIMIT_NPROC");
    216 #endif
    217 }
    218 if (limitf >= -1) {
    219 #ifdef RLIMIT_FSIZE
    220 limit(RLIMIT_FSIZE, limitf);
    221 #else
    222 if (OPT_verbose) bb_error_msg("system does not support %s",
    223 "RLIMIT_FSIZE");
    224 #endif
    225 }
    226 if (limitc >= -1) {
    227 #ifdef RLIMIT_CORE
    228 limit(RLIMIT_CORE, limitc);
    229 #else
    230 if (OPT_verbose) bb_error_msg("system does not support %s",
    231 "RLIMIT_CORE");
    232 #endif
    233 }
    234 if (limitr >= -1) {
    235 #ifdef RLIMIT_RSS
    236 limit(RLIMIT_RSS, limitr);
    237 #else
    238 if (OPT_verbose) bb_error_msg("system does not support %s",
    239 "RLIMIT_RSS");
    240 #endif
    241 }
    242 if (limitt >= -1) {
    243 #ifdef RLIMIT_CPU
    244 limit(RLIMIT_CPU, limitt);
    245 #else
    246 if (OPT_verbose) bb_error_msg("system does not support %s",
    247 "RLIMIT_CPU");
    248 #endif
    249 }
    250 }
    251
    252 /* argv[0] */
    253 static void setuidgid(int, char **);
    254 static void envuidgid(int, char **);
    255 static void envdir(int, char **);
    256 static void softlimit(int, char **);
    257
    258 int chpst_main(int argc, char **argv)
    259 {
    260 if (applet_name[3] == 'd') envdir(argc, argv);
    261 if (applet_name[1] == 'o') softlimit(argc, argv);
    262 if (applet_name[0] == 's') setuidgid(argc, argv);
    263 if (applet_name[0] == 'e') envuidgid(argc, argv);
    264 // otherwise we are chpst
    265
    266 {
    267 char *m,*d,*o,*p,*f,*c,*r,*t,*n;
    268 getopt32(argc, argv, "+u:U:e:m:d:o:p:f:c:r:t:/:n:vP012",
    269 &set_user,&env_user,&env_dir,
    270 &m,&d,&o,&p,&f,&c,&r,&t,&root,&n);
    271 // if (option_mask32 & 0x1) // -u
    272 // if (option_mask32 & 0x2) // -U
    273 // if (option_mask32 & 0x4) // -e
    274 if (option_mask32 & 0x8) limits = limitl = limita = limitd = xatoul(m); // -m
    275 if (option_mask32 & 0x10) limitd = xatoul(d); // -d
    276 if (option_mask32 & 0x20) limito = xatoul(o); // -o
    277 if (option_mask32 & 0x40) limitp = xatoul(p); // -p
    278 if (option_mask32 & 0x80) limitf = xatoul(f); // -f
    279 if (option_mask32 & 0x100) limitc = xatoul(c); // -c
    280 if (option_mask32 & 0x200) limitr = xatoul(r); // -r
    281 if (option_mask32 & 0x400) limitt = xatoul(t); // -t
    282 // if (option_mask32 & 0x800) // -/
    283 if (option_mask32 & 0x1000) nicelvl = xatoi(n); // -n
    284 // The below consts should match #defines at top!
    285 //if (option_mask32 & 0x2000) OPT_verbose = 1; // -v
    286 //if (option_mask32 & 0x4000) OPT_pgrp = 1; // -P
    287 //if (option_mask32 & 0x8000) OPT_nostdin = 1; // -0
    288 //if (option_mask32 & 0x10000) OPT_nostdout = 1; // -1
    289 //if (option_mask32 & 0x20000) OPT_nostderr = 1; // -2
    290 }
    291 argv += optind;
    292 if (!argv || !*argv) bb_show_usage();
    293
    294 if (OPT_pgrp) setsid();
    295 if (env_dir) edir(env_dir);
    296 if (root) {
    297 xchdir(root);
    298 if (chroot(".") == -1)
    299 bb_perror_msg_and_die("chroot");
    300 }
    301 slimit();
    302 if (nicelvl) {
    303 errno = 0;
    304 if (nice(nicelvl) == -1)
    305 bb_perror_msg_and_die("nice");
    306 }
    307 if (env_user) euidgid(env_user);
    308 if (set_user) suidgid(set_user);
    309 if (OPT_nostdin) close(0);
    310 if (OPT_nostdout) close(1);
    311 if (OPT_nostderr) close(2);
    312 execvp(argv[0], argv);
    313 bb_perror_msg_and_die("exec %s", argv[0]);
    314 }
    315
    316 static void setuidgid(int argc, char **argv)
    317 {
    318 const char *account;
    319
    320 account = *++argv;
    321 if (!account) bb_show_usage();
    322 if (!*++argv) bb_show_usage();
    323 suidgid((char*)account);
    324 execvp(argv[0], argv);
    325 bb_perror_msg_and_die("exec %s", argv[0]);
    326 }
    327
    328 static void envuidgid(int argc, char **argv)
    329 {
    330 const char *account;
    331
    332 account = *++argv;
    333 if (!account) bb_show_usage();
    334 if (!*++argv) bb_show_usage();
    335 euidgid((char*)account);
    336 execvp(argv[0], argv);
    337 bb_perror_msg_and_die("exec %s", argv[0]);
    338 }
    339
    340 static void envdir(int argc, char **argv)
    341 {
    342 const char *dir;
    343
    344 dir = *++argv;
    345 if (!dir) bb_show_usage();
    346 if (!*++argv) bb_show_usage();
    347 edir(dir);
    348 execvp(argv[0], argv);
    349 bb_perror_msg_and_die("exec %s", argv[0]);
    350 }
    351
    352 static void softlimit(int argc, char **argv)
    353 {
    354 char *a,*c,*d,*f,*l,*m,*o,*p,*r,*s,*t;
    355 getopt32(argc, argv, "+a:c:d:f:l:m:o:p:r:s:t:",
    356 &a,&c,&d,&f,&l,&m,&o,&p,&r,&s,&t);
    357 if (option_mask32 & 0x001) limita = xatoul(a); // -a
    358 if (option_mask32 & 0x002) limitc = xatoul(c); // -c
    359 if (option_mask32 & 0x004) limitd = xatoul(d); // -d
    360 if (option_mask32 & 0x008) limitf = xatoul(f); // -f
    361 if (option_mask32 & 0x010) limitl = xatoul(l); // -l
    362 if (option_mask32 & 0x020) limits = limitl = limita = limitd = xatoul(m); // -m
    363 if (option_mask32 & 0x040) limito = xatoul(o); // -o
    364 if (option_mask32 & 0x080) limitp = xatoul(p); // -p
    365 if (option_mask32 & 0x100) limitr = xatoul(r); // -r
    366 if (option_mask32 & 0x200) limits = xatoul(s); // -s
    367 if (option_mask32 & 0x400) limitt = xatoul(t); // -t
    368 argv += optind;
    369 if (!argv[0]) bb_show_usage();
    370 slimit();
    371 execvp(argv[0], argv);
    372 bb_perror_msg_and_die("exec %s", argv[0]);
    373 }