Denis Vlasenko | f0a97fb | 2006-10-03 17:52:24 +0000 | [diff] [blame] | 1 | #include "busybox.h" |
| 2 | |
| 3 | #include <sys/types.h> |
| 4 | #include <sys/resource.h> |
| 5 | #include <grp.h> |
| 6 | |
| 7 | #include "uidgid.h" |
| 8 | |
| 9 | #include <sys/types.h> |
| 10 | #include <dirent.h> |
| 11 | |
| 12 | static unsigned option_mask; |
| 13 | // Must meatch constants in chpst_main! |
| 14 | #define OPT_verbose (option_mask & 0x2000) |
| 15 | #define OPT_pgrp (option_mask & 0x4000) |
| 16 | #define OPT_nostdin (option_mask & 0x8000) |
| 17 | #define OPT_nostdout (option_mask & 0x10000) |
| 18 | #define OPT_nostderr (option_mask & 0x20000) |
| 19 | |
| 20 | static char *set_user; |
| 21 | static char *env_user; |
| 22 | static const char *env_dir; |
| 23 | static long limitd = -2; |
| 24 | static long limits = -2; |
| 25 | static long limitl = -2; |
| 26 | static long limita = -2; |
| 27 | static long limito = -2; |
| 28 | static long limitp = -2; |
| 29 | static long limitf = -2; |
| 30 | static long limitc = -2; |
| 31 | static long limitr = -2; |
| 32 | static long limitt = -2; |
| 33 | static long nicelvl; |
| 34 | static const char *root; |
| 35 | |
| 36 | static void suidgid(char *user, unsigned dogrp) |
| 37 | { |
| 38 | struct uidgid ugid; |
| 39 | |
| 40 | if (!uidgid_get(&ugid, user, dogrp)) { |
| 41 | if (dogrp) |
| 42 | bb_error_msg_and_die("unknown user/group: %s", user); |
| 43 | else |
| 44 | bb_error_msg_and_die("unknown account: %s", user); |
| 45 | } |
| 46 | if (setgroups(ugid.gids, ugid.gid) == -1) |
| 47 | bb_perror_msg_and_die("setgroups"); |
| 48 | xsetgid(*ugid.gid); |
| 49 | xsetuid(ugid.uid); |
| 50 | } |
| 51 | |
| 52 | static void euidgid(char *user, unsigned dogrp) |
| 53 | { |
| 54 | struct uidgid ugid; |
| 55 | |
| 56 | if (!uidgid_get(&ugid, user, dogrp)) { |
| 57 | if (dogrp) |
| 58 | bb_error_msg_and_die("unknown user/group: %s", user); |
| 59 | else |
| 60 | bb_error_msg_and_die("unknown account: %s", user); |
| 61 | } |
| 62 | //FIXME: ultoa needed here! |
| 63 | xsetenv("GID", utoa(*ugid.gid)); |
| 64 | xsetenv("UID", utoa(ugid.uid)); |
| 65 | } |
| 66 | |
| 67 | static void edir(const char *directory_name) |
| 68 | { |
| 69 | int wdir; |
| 70 | DIR *dir; |
| 71 | struct dirent *d; |
| 72 | int fd; |
| 73 | |
| 74 | wdir = xopen(".", O_RDONLY | O_NDELAY); |
| 75 | xchdir(directory_name); |
| 76 | dir = opendir("."); |
| 77 | if (!dir) |
| 78 | bb_perror_msg_and_die("opendir %s", directory_name); |
| 79 | for (;;) { |
| 80 | errno = 0; |
| 81 | d = readdir(dir); |
| 82 | if (!d) { |
| 83 | if (errno) bb_perror_msg_and_die("readdir %s", directory_name); |
| 84 | break; |
| 85 | } |
| 86 | if (d->d_name[0] == '.') continue; |
| 87 | fd = open(d->d_name, O_RDONLY | O_NDELAY); |
| 88 | if (fd < 0) { |
| 89 | if ((errno == EISDIR) && env_dir) { |
| 90 | if (OPT_verbose) |
| 91 | bb_perror_msg("warning: %s/%s is a directory", directory_name, |
| 92 | d->d_name); |
| 93 | continue; |
| 94 | } else |
| 95 | bb_perror_msg_and_die("open %s/%s", directory_name, /* was exiting 111 */ |
| 96 | d->d_name); |
| 97 | } |
| 98 | if (fd >= 0) { |
| 99 | char buf[256]; |
| 100 | char *tail; |
| 101 | int size; |
| 102 | |
| 103 | size = safe_read(fd, buf, sizeof(buf)-1); |
| 104 | if (size < 0) |
| 105 | bb_perror_msg_and_die("read %s/%s", directory_name, /* was exiting 111 */ |
| 106 | d->d_name); |
| 107 | if (size == 0) { |
| 108 | xsetenv(d->d_name, ""); |
| 109 | continue; |
| 110 | } |
| 111 | buf[size] = '\n'; |
| 112 | tail = memchr(buf, '\n', sizeof(buf)); |
| 113 | /* skip trailing whitespace */; |
| 114 | while (1) { |
| 115 | if (tail[0]==' ') tail[0] = '\0'; |
| 116 | if (tail[0]=='\t') tail[0] = '\0'; |
| 117 | if (tail[0]=='\n') tail[0] = '\0'; |
| 118 | if (tail == buf) break; |
| 119 | tail--; |
| 120 | } |
| 121 | xsetenv(d->d_name, buf); |
| 122 | } |
| 123 | } |
| 124 | closedir(dir); |
| 125 | if (fchdir(wdir) == -1) bb_perror_msg_and_die("fchdir"); |
| 126 | close(wdir); |
| 127 | } |
| 128 | |
| 129 | static void limit(int what, long l) |
| 130 | { |
| 131 | struct rlimit r; |
| 132 | |
| 133 | if (getrlimit(what, &r) == -1) bb_perror_msg_and_die("getrlimit"); |
| 134 | if ((l < 0) || (l > r.rlim_max)) |
| 135 | r.rlim_cur = r.rlim_max; |
| 136 | else |
| 137 | r.rlim_cur = l; |
| 138 | if (setrlimit(what, &r) == -1) bb_perror_msg_and_die("setrlimit"); |
| 139 | } |
| 140 | |
| 141 | static void slimit(void) |
| 142 | { |
| 143 | if (limitd >= -1) { |
| 144 | #ifdef RLIMIT_DATA |
| 145 | limit(RLIMIT_DATA, limitd); |
| 146 | #else |
| 147 | if (OPT_verbose) bb_error_msg("system does not support %s", "RLIMIT_DATA"); |
| 148 | #endif |
| 149 | } |
| 150 | if (limits >= -1) { |
| 151 | #ifdef RLIMIT_STACK |
| 152 | limit(RLIMIT_STACK, limits); |
| 153 | #else |
| 154 | if (OPT_verbose) bb_error_msg("system does not support %s", "RLIMIT_STACK"); |
| 155 | #endif |
| 156 | } |
| 157 | if (limitl >= -1) { |
| 158 | #ifdef RLIMIT_MEMLOCK |
| 159 | limit(RLIMIT_MEMLOCK, limitl); |
| 160 | #else |
| 161 | if (OPT_verbose) bb_error_msg("system does not support %s", "RLIMIT_MEMLOCK"); |
| 162 | #endif |
| 163 | } |
| 164 | if (limita >= -1) { |
| 165 | #ifdef RLIMIT_VMEM |
| 166 | limit(RLIMIT_VMEM, limita); |
| 167 | #else |
| 168 | #ifdef RLIMIT_AS |
| 169 | limit(RLIMIT_AS, limita); |
| 170 | #else |
| 171 | if (OPT_verbose) |
| 172 | bb_error_msg("system does not support %s", "RLIMIT_VMEM"); |
| 173 | #endif |
| 174 | #endif |
| 175 | } |
| 176 | if (limito >= -1) { |
| 177 | #ifdef RLIMIT_NOFILE |
| 178 | limit(RLIMIT_NOFILE, limito); |
| 179 | #else |
| 180 | #ifdef RLIMIT_OFILE |
| 181 | limit(RLIMIT_OFILE, limito); |
| 182 | #else |
| 183 | if (OPT_verbose) |
| 184 | bb_error_msg("system does not support %s", "RLIMIT_NOFILE"); |
| 185 | #endif |
| 186 | #endif |
| 187 | } |
| 188 | if (limitp >= -1) { |
| 189 | #ifdef RLIMIT_NPROC |
| 190 | limit(RLIMIT_NPROC, limitp); |
| 191 | #else |
| 192 | if (OPT_verbose) bb_error_msg("system does not support %s", "RLIMIT_NPROC"); |
| 193 | #endif |
| 194 | } |
| 195 | if (limitf >= -1) { |
| 196 | #ifdef RLIMIT_FSIZE |
| 197 | limit(RLIMIT_FSIZE, limitf); |
| 198 | #else |
| 199 | if (OPT_verbose) bb_error_msg("system does not support %s", "RLIMIT_FSIZE"); |
| 200 | #endif |
| 201 | } |
| 202 | if (limitc >= -1) { |
| 203 | #ifdef RLIMIT_CORE |
| 204 | limit(RLIMIT_CORE, limitc); |
| 205 | #else |
| 206 | if (OPT_verbose) bb_error_msg("system does not support %s", "RLIMIT_CORE"); |
| 207 | #endif |
| 208 | } |
| 209 | if (limitr >= -1) { |
| 210 | #ifdef RLIMIT_RSS |
| 211 | limit(RLIMIT_RSS, limitr); |
| 212 | #else |
| 213 | if (OPT_verbose) bb_error_msg("system does not support %s", "RLIMIT_RSS"); |
| 214 | #endif |
| 215 | } |
| 216 | if (limitt >= -1) { |
| 217 | #ifdef RLIMIT_CPU |
| 218 | limit(RLIMIT_CPU, limitt); |
| 219 | #else |
| 220 | if (OPT_verbose) bb_error_msg("system does not support %s", "RLIMIT_CPU"); |
| 221 | #endif |
| 222 | } |
| 223 | } |
| 224 | |
| 225 | /* argv[0] */ |
| 226 | static void setuidgid(int, char **); |
| 227 | static void envuidgid(int, char **); |
| 228 | static void envdir(int, char **); |
| 229 | static void softlimit(int, char **); |
| 230 | |
| 231 | int chpst_main(int argc, char **argv) |
| 232 | { |
| 233 | if (bb_applet_name[3] == 'd') envdir(argc, argv); |
| 234 | if (bb_applet_name[1] == 'o') softlimit(argc, argv); |
| 235 | if (bb_applet_name[0] == 's') setuidgid(argc, argv); |
| 236 | if (bb_applet_name[0] == 'e') envuidgid(argc, argv); |
| 237 | // otherwise we are.......... chpst |
| 238 | |
| 239 | { |
| 240 | char *m,*d,*o,*p,*f,*c,*r,*t,*n; |
Denis Vlasenko | 67b23e6 | 2006-10-03 21:00:06 +0000 | [diff] [blame^] | 241 | option_mask = getopt32(argc, argv, "u:U:e:m:d:o:p:f:c:r:t:/:n:vP012", |
Denis Vlasenko | f0a97fb | 2006-10-03 17:52:24 +0000 | [diff] [blame] | 242 | &set_user,&env_user,&env_dir, |
| 243 | &m,&d,&o,&p,&f,&c,&r,&t,&root,&n); |
| 244 | // if (option_mask & 0x1) // -u |
| 245 | // if (option_mask & 0x2) // -U |
| 246 | // if (option_mask & 0x4) // -e |
| 247 | if (option_mask & 0x8) limits = limitl = limita = limitd = bb_xgetularg10(m); // -m |
| 248 | if (option_mask & 0x10) limitd = bb_xgetularg10(d); // -d |
| 249 | if (option_mask & 0x20) limito = bb_xgetularg10(o); // -o |
| 250 | if (option_mask & 0x40) limitp = bb_xgetularg10(p); // -p |
| 251 | if (option_mask & 0x80) limitf = bb_xgetularg10(f); // -f |
| 252 | if (option_mask & 0x100) limitc = bb_xgetularg10(c); // -c |
| 253 | if (option_mask & 0x200) limitr = bb_xgetularg10(r); // -r |
| 254 | if (option_mask & 0x400) limitt = bb_xgetularg10(t); // -t |
| 255 | // if (option_mask & 0x800) // -/ |
| 256 | if (option_mask & 0x1000) nicelvl = bb_xgetlarg_bnd_sfx(n, 10, -20, 20, NULL); // -n |
| 257 | // The below consts should match #defines at top! |
| 258 | //if (option_mask & 0x2000) OPT_verbose = 1; // -v |
| 259 | //if (option_mask & 0x4000) OPT_pgrp = 1; // -P |
| 260 | //if (option_mask & 0x8000) OPT_nostdin = 1; // -0 |
| 261 | //if (option_mask & 0x10000) OPT_nostdout = 1; // -1 |
| 262 | //if (option_mask & 0x20000) OPT_nostderr = 1; // -2 |
| 263 | } |
| 264 | if (!argv || !*argv) bb_show_usage(); |
| 265 | |
| 266 | if (OPT_pgrp) setsid(); |
| 267 | if (env_dir) edir(env_dir); |
| 268 | if (root) { |
| 269 | xchdir(root); |
| 270 | if (chroot(".") == -1) |
| 271 | bb_perror_msg_and_die("chroot"); |
| 272 | } |
| 273 | slimit(); |
| 274 | if (nicelvl) { |
| 275 | errno = 0; |
| 276 | if (nice(nicelvl) == -1) |
| 277 | bb_perror_msg_and_die("nice"); |
| 278 | } |
| 279 | if (env_user) euidgid(env_user, 1); |
| 280 | if (set_user) suidgid(set_user, 1); |
| 281 | if (OPT_nostdin) close(0); |
| 282 | if (OPT_nostdout) close(1); |
| 283 | if (OPT_nostderr) close(2); |
| 284 | execvp(argv[0], argv); |
| 285 | bb_perror_msg_and_die("exec %s", argv[0]); |
| 286 | } |
| 287 | |
| 288 | static void setuidgid(int argc, char **argv) |
| 289 | { |
| 290 | const char *account; |
| 291 | |
| 292 | account = *++argv; |
| 293 | if (!account) bb_show_usage(); |
| 294 | if (!*++argv) bb_show_usage(); |
| 295 | suidgid((char*)account, 0); |
| 296 | execvp(argv[0], argv); |
| 297 | bb_perror_msg_and_die("exec %s", argv[0]); |
| 298 | } |
| 299 | |
| 300 | static void envuidgid(int argc, char **argv) |
| 301 | { |
| 302 | const char *account; |
| 303 | |
| 304 | account = *++argv; |
| 305 | if (!account) bb_show_usage(); |
| 306 | if (!*++argv) bb_show_usage(); |
| 307 | euidgid((char*)account, 0); |
| 308 | execvp(argv[0], argv); |
| 309 | bb_perror_msg_and_die("exec %s", argv[0]); |
| 310 | } |
| 311 | |
| 312 | static void envdir(int argc, char **argv) |
| 313 | { |
| 314 | const char *dir; |
| 315 | |
| 316 | dir = *++argv; |
| 317 | if (!dir) bb_show_usage(); |
| 318 | if (!*++argv) bb_show_usage(); |
| 319 | edir(dir); |
| 320 | execvp(argv[0], argv); |
| 321 | bb_perror_msg_and_die("exec %s", argv[0]); |
| 322 | } |
| 323 | |
| 324 | static void softlimit(int argc, char **argv) |
| 325 | { |
| 326 | char *a,*c,*d,*f,*l,*m,*o,*p,*r,*s,*t; |
Denis Vlasenko | 67b23e6 | 2006-10-03 21:00:06 +0000 | [diff] [blame^] | 327 | option_mask = getopt32(argc, argv, "a:c:d:f:l:m:o:p:r:s:t:", |
Denis Vlasenko | f0a97fb | 2006-10-03 17:52:24 +0000 | [diff] [blame] | 328 | &a,&c,&d,&f,&l,&m,&o,&p,&r,&s,&t); |
| 329 | if (option_mask & 0x001) limita = bb_xgetularg10(a); // -a |
| 330 | if (option_mask & 0x002) limitc = bb_xgetularg10(c); // -c |
| 331 | if (option_mask & 0x004) limitd = bb_xgetularg10(d); // -d |
| 332 | if (option_mask & 0x008) limitf = bb_xgetularg10(f); // -f |
| 333 | if (option_mask & 0x010) limitl = bb_xgetularg10(l); // -l |
| 334 | if (option_mask & 0x020) limits = limitl = limita = limitd = bb_xgetularg10(m); // -m |
| 335 | if (option_mask & 0x040) limito = bb_xgetularg10(o); // -o |
| 336 | if (option_mask & 0x080) limitp = bb_xgetularg10(p); // -p |
| 337 | if (option_mask & 0x100) limitr = bb_xgetularg10(r); // -r |
| 338 | if (option_mask & 0x200) limits = bb_xgetularg10(s); // -s |
| 339 | if (option_mask & 0x400) limitt = bb_xgetularg10(t); // -t |
| 340 | argv += optind; |
| 341 | if (!argv[0]) bb_show_usage(); |
| 342 | slimit(); |
| 343 | execvp(argv[0], argv); |
| 344 | bb_perror_msg_and_die("exec %s", argv[0]); |
| 345 | } |