[1765] | 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 "libbb.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 | int chpst_main(int argc, char **argv)
|
---|
| 260 | {
|
---|
| 261 | if (applet_name[3] == 'd') envdir(argc, argv);
|
---|
| 262 | if (applet_name[1] == 'o') softlimit(argc, argv);
|
---|
| 263 | if (applet_name[0] == 's') setuidgid(argc, argv);
|
---|
| 264 | if (applet_name[0] == 'e') envuidgid(argc, argv);
|
---|
| 265 | // otherwise we are chpst
|
---|
| 266 |
|
---|
| 267 | {
|
---|
| 268 | char *m,*d,*o,*p,*f,*c,*r,*t,*n;
|
---|
| 269 | getopt32(argv, "+u:U:e:m:d:o:p:f:c:r:t:/:n:vP012",
|
---|
| 270 | &set_user,&env_user,&env_dir,
|
---|
| 271 | &m,&d,&o,&p,&f,&c,&r,&t,&root,&n);
|
---|
| 272 | // if (option_mask32 & 0x1) // -u
|
---|
| 273 | // if (option_mask32 & 0x2) // -U
|
---|
| 274 | // if (option_mask32 & 0x4) // -e
|
---|
| 275 | if (option_mask32 & 0x8) limits = limitl = limita = limitd = xatoul(m); // -m
|
---|
| 276 | if (option_mask32 & 0x10) limitd = xatoul(d); // -d
|
---|
| 277 | if (option_mask32 & 0x20) limito = xatoul(o); // -o
|
---|
| 278 | if (option_mask32 & 0x40) limitp = xatoul(p); // -p
|
---|
| 279 | if (option_mask32 & 0x80) limitf = xatoul(f); // -f
|
---|
| 280 | if (option_mask32 & 0x100) limitc = xatoul(c); // -c
|
---|
| 281 | if (option_mask32 & 0x200) limitr = xatoul(r); // -r
|
---|
| 282 | if (option_mask32 & 0x400) limitt = xatoul(t); // -t
|
---|
| 283 | // if (option_mask32 & 0x800) // -/
|
---|
| 284 | if (option_mask32 & 0x1000) nicelvl = xatoi(n); // -n
|
---|
| 285 | // The below consts should match #defines at top!
|
---|
| 286 | //if (option_mask32 & 0x2000) OPT_verbose = 1; // -v
|
---|
| 287 | //if (option_mask32 & 0x4000) OPT_pgrp = 1; // -P
|
---|
| 288 | //if (option_mask32 & 0x8000) OPT_nostdin = 1; // -0
|
---|
| 289 | //if (option_mask32 & 0x10000) OPT_nostdout = 1; // -1
|
---|
| 290 | //if (option_mask32 & 0x20000) OPT_nostderr = 1; // -2
|
---|
| 291 | }
|
---|
| 292 | argv += optind;
|
---|
| 293 | if (!argv || !*argv) bb_show_usage();
|
---|
| 294 |
|
---|
| 295 | if (OPT_pgrp) setsid();
|
---|
| 296 | if (env_dir) edir(env_dir);
|
---|
| 297 | if (root) {
|
---|
| 298 | xchdir(root);
|
---|
| 299 | if (chroot(".") == -1)
|
---|
| 300 | bb_perror_msg_and_die("chroot");
|
---|
| 301 | }
|
---|
| 302 | slimit();
|
---|
| 303 | if (nicelvl) {
|
---|
| 304 | errno = 0;
|
---|
| 305 | if (nice(nicelvl) == -1)
|
---|
| 306 | bb_perror_msg_and_die("nice");
|
---|
| 307 | }
|
---|
| 308 | if (env_user) euidgid(env_user);
|
---|
| 309 | if (set_user) suidgid(set_user);
|
---|
| 310 | if (OPT_nostdin) close(0);
|
---|
| 311 | if (OPT_nostdout) close(1);
|
---|
| 312 | if (OPT_nostderr) close(2);
|
---|
| 313 | BB_EXECVP(argv[0], argv);
|
---|
| 314 | bb_perror_msg_and_die("exec %s", argv[0]);
|
---|
| 315 | }
|
---|
| 316 |
|
---|
| 317 | static void setuidgid(int argc, char **argv)
|
---|
| 318 | {
|
---|
| 319 | const char *account;
|
---|
| 320 |
|
---|
| 321 | account = *++argv;
|
---|
| 322 | if (!account) bb_show_usage();
|
---|
| 323 | if (!*++argv) bb_show_usage();
|
---|
| 324 | suidgid((char*)account);
|
---|
| 325 | BB_EXECVP(argv[0], argv);
|
---|
| 326 | bb_perror_msg_and_die("exec %s", argv[0]);
|
---|
| 327 | }
|
---|
| 328 |
|
---|
| 329 | static void envuidgid(int argc, char **argv)
|
---|
| 330 | {
|
---|
| 331 | const char *account;
|
---|
| 332 |
|
---|
| 333 | account = *++argv;
|
---|
| 334 | if (!account) bb_show_usage();
|
---|
| 335 | if (!*++argv) bb_show_usage();
|
---|
| 336 | euidgid((char*)account);
|
---|
| 337 | BB_EXECVP(argv[0], argv);
|
---|
| 338 | bb_perror_msg_and_die("exec %s", argv[0]);
|
---|
| 339 | }
|
---|
| 340 |
|
---|
| 341 | static void envdir(int argc, char **argv)
|
---|
| 342 | {
|
---|
| 343 | const char *dir;
|
---|
| 344 |
|
---|
| 345 | dir = *++argv;
|
---|
| 346 | if (!dir) bb_show_usage();
|
---|
| 347 | if (!*++argv) bb_show_usage();
|
---|
| 348 | edir(dir);
|
---|
| 349 | BB_EXECVP(argv[0], argv);
|
---|
| 350 | bb_perror_msg_and_die("exec %s", argv[0]);
|
---|
| 351 | }
|
---|
| 352 |
|
---|
| 353 | static void softlimit(int argc, char **argv)
|
---|
| 354 | {
|
---|
| 355 | char *a,*c,*d,*f,*l,*m,*o,*p,*r,*s,*t;
|
---|
| 356 | getopt32(argv, "+a:c:d:f:l:m:o:p:r:s:t:",
|
---|
| 357 | &a,&c,&d,&f,&l,&m,&o,&p,&r,&s,&t);
|
---|
| 358 | if (option_mask32 & 0x001) limita = xatoul(a); // -a
|
---|
| 359 | if (option_mask32 & 0x002) limitc = xatoul(c); // -c
|
---|
| 360 | if (option_mask32 & 0x004) limitd = xatoul(d); // -d
|
---|
| 361 | if (option_mask32 & 0x008) limitf = xatoul(f); // -f
|
---|
| 362 | if (option_mask32 & 0x010) limitl = xatoul(l); // -l
|
---|
| 363 | if (option_mask32 & 0x020) limits = limitl = limita = limitd = xatoul(m); // -m
|
---|
| 364 | if (option_mask32 & 0x040) limito = xatoul(o); // -o
|
---|
| 365 | if (option_mask32 & 0x080) limitp = xatoul(p); // -p
|
---|
| 366 | if (option_mask32 & 0x100) limitr = xatoul(r); // -r
|
---|
| 367 | if (option_mask32 & 0x200) limits = xatoul(s); // -s
|
---|
| 368 | if (option_mask32 & 0x400) limitt = xatoul(t); // -t
|
---|
| 369 | argv += optind;
|
---|
| 370 | if (!argv[0]) bb_show_usage();
|
---|
| 371 | slimit();
|
---|
| 372 | BB_EXECVP(argv[0], argv);
|
---|
| 373 | bb_perror_msg_and_die("exec %s", argv[0]);
|
---|
| 374 | }
|
---|