libbb/appletlib.c

   1 /* vi: set sw=4 ts=4: */
   2 /*
   3  * Utility routines.
   4  *
   5  * Copyright (C) tons of folks.  Tracking down who wrote what
   6  * isn't something I'm going to worry about...  If you wrote something
   7  * here, please feel free to acknowledge your work.
   8  *
   9  * Based in part on code from sash, Copyright (c) 1999 by David I. Bell
  10  * Permission has been granted to redistribute this code under GPL.
  11  *
  12  * Licensed under GPLv2 or later, see file LICENSE in this source tree.
  13  */
  14
  15 /* We are trying to not use printf, this benefits the case when selected
  16  * applets are really simple. Example:
  17  *
  18  * $ ./busybox
  19  * ...
  20  * Currently defined functions:
  21  *         basename, false, true
  22  *
  23  * $ size busybox
  24  *    text    data     bss     dec     hex filename
  25  *    4473      52      72    4597    11f5 busybox
  26  *
  27  * FEATURE_INSTALLER or FEATURE_SUID will still link printf routines in. :(
  28  */
  29 #include "busybox.h"
  30
  31 #if !(defined(__FreeBSD__) || defined(__OpenBSD__) || defined(__NetBSD__) \
  32     || defined(__APPLE__) \
  33     )
  34 # include <malloc.h> /* for mallopt */
  35 #endif
  36
  37 #include <sys/prctl.h>
  38 #ifndef PR_SET_NAME
  39 #define PR_SET_NAME 15
  40 #endif
  41 #ifndef PR_GET_NAME
  42 #define PR_GET_NAME 16
  43 #endif
  44
  45 /* Declare <applet>_main() */
  46 #define PROTOTYPES
  47 #include "applets.h"
  48 #undef PROTOTYPES
  49
  50 /* Include generated applet names, pointers to <applet>_main, etc */
  51 #include "applet_tables.h"
  52 /* ...and if applet_tables generator says we have only one applet... */
  53 #ifdef SINGLE_APPLET_MAIN
  54 # undef ENABLE_FEATURE_INDIVIDUAL
  55 # define ENABLE_FEATURE_INDIVIDUAL 1
  56 # undef IF_FEATURE_INDIVIDUAL
  57 # define IF_FEATURE_INDIVIDUAL(...) __VA_ARGS__
  58 #endif
  59
  60 #include "usage_compressed.h"
  61
  62
  63 /* "Do not compress usage text if uncompressed text is small
  64  *  and we don't include bunzip2 code for other reasons"
  65  *
  66  * Useful for mass one-applet rebuild (bunzip2 code is ~2.7k).
  67  *
  68  * Unlike BUNZIP2, if FEATURE_SEAMLESS_BZ2 is on, bunzip2 code is built but
  69  * still may be unused if none of the selected applets calls open_zipped()
  70  * or its friends; we test for (FEATURE_SEAMLESS_BZ2 && <APPLET>) instead.
  71  * For example, only if TAR and FEATURE_SEAMLESS_BZ2 are both selected,
  72  * then bunzip2 code will be linked in anyway, and disabling help compression
  73  * would be not optimal:
  74  */
  75 #if UNPACKED_USAGE_LENGTH < 4*1024 \
  76  && !(ENABLE_FEATURE_SEAMLESS_BZ2 && ENABLE_TAR) \
  77  && !(ENABLE_FEATURE_SEAMLESS_BZ2 && ENABLE_MODPROBE) \
  78  && !(ENABLE_FEATURE_SEAMLESS_BZ2 && ENABLE_INSMOD) \
  79  && !(ENABLE_FEATURE_SEAMLESS_BZ2 && ENABLE_DEPMOD) \
  80  && !(ENABLE_FEATURE_SEAMLESS_BZ2 && ENABLE_MAN) \
  81  && !ENABLE_BUNZIP2 \
  82  && !ENABLE_BZCAT
  83 # undef  ENABLE_FEATURE_COMPRESS_USAGE
  84 # define ENABLE_FEATURE_COMPRESS_USAGE 0
  85 #endif
  86
  87
  88 unsigned FAST_FUNC string_array_len(char **argv)
  89 {
  90         char **start = argv;
  91
  92         while (*argv)
  93                 argv++;
  94
  95         return argv - start;
  96 }
  97
  98
  99 #if ENABLE_SHOW_USAGE && !ENABLE_FEATURE_COMPRESS_USAGE
 100 static const char usage_messages[] ALIGN1 = UNPACKED_USAGE;
 101 #else
 102 # define usage_messages 0
 103 #endif
 104
 105 #if ENABLE_FEATURE_COMPRESS_USAGE
 106
 107 static const char packed_usage[] ALIGN1 = { PACKED_USAGE };
 108 # include "bb_archive.h"
 109 static const char *unpack_usage_messages(void)
 110 {
 111         char *outbuf = NULL;
 112         bunzip_data *bd;
 113         int i;
 114
 115         i = start_bunzip(&bd,
 116                         /* src_fd: */ -1,
 117                         /* inbuf:  */ packed_usage,
 118                         /* len:    */ sizeof(packed_usage));
 119         /* read_bunzip can longjmp to start_bunzip, and ultimately
 120          * end up here with i != 0 on read data errors! Not trivial */
 121         if (!i) {
 122                 /* Cannot use xmalloc: will leak bd in NOFORK case! */
 123                 outbuf = malloc_or_warn(sizeof(UNPACKED_USAGE));
 124                 if (outbuf)
 125                         read_bunzip(bd, outbuf, sizeof(UNPACKED_USAGE));
 126         }
 127         dealloc_bunzip(bd);
 128         return outbuf;
 129 }
 130 # define dealloc_usage_messages(s) free(s)
 131
 132 #else
 133
 134 # define unpack_usage_messages() usage_messages
 135 # define dealloc_usage_messages(s) ((void)(s))
 136
 137 #endif /* FEATURE_COMPRESS_USAGE */
 138
 139
 140 void FAST_FUNC bb_show_usage(void)
 141 {
 142         if (ENABLE_SHOW_USAGE) {
 143 #ifdef SINGLE_APPLET_STR
 144                 /* Imagine that this applet is "true". Dont suck in printf! */
 145                 const char *usage_string = unpack_usage_messages();
 146
 147                 if (*usage_string == '\b') {
 148                         full_write2_str("No help available.\n\n");
 149                 } else {
 150                         full_write2_str("Usage: "SINGLE_APPLET_STR" ");
 151                         full_write2_str(usage_string);
 152                         full_write2_str("\n\n");
 153                 }
 154                 if (ENABLE_FEATURE_CLEAN_UP)
 155                         dealloc_usage_messages((char*)usage_string);
 156 #else
 157                 const char *p;
 158                 const char *usage_string = p = unpack_usage_messages();
 159                 int ap = find_applet_by_name(applet_name);
 160
 161                 if (ap < 0) /* never happens, paranoia */
 162                         xfunc_die();
 163                 while (ap) {
 164                         while (*p++) continue;
 165                         ap--;
 166                 }
 167                 full_write2_str(bb_banner);
 168                 full_write2_str(" multi-call binary.\n");
 169                 if (*p == '\b')
 170                         full_write2_str("\nNo help available.\n\n");
 171                 else {
 172                         full_write2_str("\nUsage: ");
 173                         full_write2_str(applet_name);
 174                         full_write2_str(" ");
 175                         full_write2_str(p);
 176                         full_write2_str("\n");
 177                 }
 178                 if (ENABLE_FEATURE_CLEAN_UP)
 179                         dealloc_usage_messages((char*)usage_string);
 180 #endif
 181         }
 182         xfunc_die();
 183 }
 184
 185 int FAST_FUNC find_applet_by_name(const char *name)
 186 {
 187         unsigned i, max;
 188         int j;
 189         const char *p;
 190
 191 /* The commented-out word-at-a-time code is ~40% faster, but +160 bytes.
 192  * "Faster" here saves ~0.5 microsecond of real time - not worth it.
 193  */
 194 #if 0 /*BB_UNALIGNED_MEMACCESS_OK && BB_LITTLE_ENDIAN*/
 195         uint32_t n32;
 196
 197         /* Handle all names < 2 chars long early */
 198         if (name[0] == '\0')
 199                 return -1; /* "" is not a valid applet name */
 200         if (name[1] == '\0') {
 201                 if (!ENABLE_TEST)
 202                         return -1; /* 1-char name is not valid */
 203                 if (name[0] != ']')
 204                         return -1; /* 1-char name which isn't "[" is not valid */
 205                 /* applet "[" is always applet #0: */
 206                 return 0;
 207         }
 208 #endif
 209
 210         p = applet_names;
 211         i = 0;
 212 #if KNOWN_APPNAME_OFFSETS <= 0
 213         max = NUM_APPLETS;
 214 #else
 215         max = NUM_APPLETS * KNOWN_APPNAME_OFFSETS;
 216         for (j = ARRAY_SIZE(applet_nameofs)-1; j >= 0; j--) {
 217                 const char *pp = applet_names + applet_nameofs[j];
 218                 if (strcmp(name, pp) >= 0) {
 219                         //bb_error_msg("name:'%s' >= pp:'%s'", name, pp);
 220                         p = pp;
 221                         i = max - NUM_APPLETS;
 222                         break;
 223                 }
 224                 max -= NUM_APPLETS;
 225         }
 226         max /= (unsigned)KNOWN_APPNAME_OFFSETS;
 227         i /= (unsigned)KNOWN_APPNAME_OFFSETS;
 228         //bb_error_msg("name:'%s' starting from:'%s' i:%u max:%u", name, p, i, max);
 229 #endif
 230
 231         /* Open-coded linear search without strcmp/strlen calls for speed */
 232
 233 #if 0 /*BB_UNALIGNED_MEMACCESS_OK && BB_LITTLE_ENDIAN*/
 234         /* skip "[\0" name, it's surely not it */
 235         if (ENABLE_TEST && LONE_CHAR(p, '['))
 236                 i++, p += 2;
 237         /* All remaining applet names in p[] are at least 2 chars long */
 238         /* name[] is also at least 2 chars long */
 239
 240         n32 = (name[0] << 0) | (name[1] << 8) | (name[2] << 16);
 241         while (i < max) {
 242                 uint32_t p32;
 243                 char ch;
 244
 245                 /* Quickly check match of the first 3 bytes */
 246                 move_from_unaligned32(p32, p);
 247                 p += 3;
 248                 if ((p32 & 0x00ffffff) != n32) {
 249                         /* Most likely case: 3 first bytes do not match */
 250                         i++;
 251                         if ((p32 & 0x00ff0000) == '\0')
 252                                 continue; // p[2] was NUL
 253                         p++;
 254                         if ((p32 & 0xff000000) == '\0')
 255                                 continue; // p[3] was NUL
 256                         /* p[0..3] aren't matching and none is NUL, check the rest */
 257                         while (*p++ != '\0')
 258                                 continue;
 259                         continue;
 260                 }
 261
 262                 /* Unlikely branch: first 3 bytes ([0..2]) match */
 263                 if ((p32 & 0x00ff0000) == '\0') {
 264                         /* name is 2-byte long, it is full match */
 265                         //bb_error_msg("found:'%s' i:%u", name, i);
 266                         return i;
 267                 }
 268                 /* Check remaining bytes [3..NUL] */
 269                 ch = (p32 >> 24);
 270                 j = 3;
 271                 while (ch == name[j]) {
 272                         if (ch == '\0') {
 273                                 //bb_error_msg("found:'%s' i:%u", name, i);
 274                                 return i;
 275                         }
 276                         ch = *++p;
 277                         j++;
 278                 }
 279                 /* Not a match. Skip it, including NUL */
 280                 while (ch != '\0')
 281                         ch = *++p;
 282                 p++;
 283                 i++;
 284         }
 285         return -1;
 286 #else
 287         while (i < max) {
 288                 char ch;
 289                 j = 0;
 290                 /* Do we see "name\0" in applet_names[p] position? */
 291                 while ((ch = *p) == name[j]) {
 292                         if (ch == '\0') {
 293                                 //bb_error_msg("found:'%s' i:%u", name, i);
 294                                 return i; /* yes */
 295                         }
 296                         p++;
 297                         j++;
 298                 }
 299                 /* No.
 300                  * p => 1st non-matching char in applet_names[],
 301                  * skip to and including NUL.
 302                  */
 303                 while (ch != '\0')
 304                         ch = *++p;
 305                 p++;
 306                 i++;
 307         }
 308         return -1;
 309 #endif
 310 }
 311
 312
 313 void lbb_prepare(const char *applet
 314                 IF_FEATURE_INDIVIDUAL(, char **argv))
 315                                 MAIN_EXTERNALLY_VISIBLE;
 316 void lbb_prepare(const char *applet
 317                 IF_FEATURE_INDIVIDUAL(, char **argv))
 318 {
 319 #ifdef __GLIBC__
 320         (*(int **)&bb_errno) = __errno_location();
 321         barrier();
 322 #endif
 323         applet_name = applet;
 324
 325         if (ENABLE_LOCALE_SUPPORT)
 326                 setlocale(LC_ALL, "");
 327
 328 #if ENABLE_FEATURE_INDIVIDUAL
 329         /* Redundant for busybox (run_applet_and_exit covers that case)
 330          * but needed for "individual applet" mode */
 331         if (argv[1]
 332          && !argv[2]
 333          && strcmp(argv[1], "--help") == 0
 334          && !is_prefixed_with(applet, "busybox")
 335         ) {
 336                 /* Special case. POSIX says "test --help"
 337                  * should be no different from e.g. "test --foo".  */
 338                 if (!ENABLE_TEST || strcmp(applet_name, "test") != 0)
 339                         bb_show_usage();
 340         }
 341 #endif
 342 }
 343
 344 /* The code below can well be in applets/applets.c, as it is used only
 345  * for busybox binary, not "individual" binaries.
 346  * However, keeping it here and linking it into libbusybox.so
 347  * (together with remaining tiny applets/applets.o)
 348  * makes it possible to avoid --whole-archive at link time.
 349  * This makes (shared busybox) + libbusybox smaller.
 350  * (--gc-sections would be even better....)
 351  */
 352
 353 const char *applet_name;
 354 #if !BB_MMU
 355 bool re_execed;
 356 #endif
 357
 358
 359 /* If not built as a single-applet executable... */
 360 #if !defined(SINGLE_APPLET_MAIN)
 361
 362 IF_FEATURE_SUID(static uid_t ruid;)  /* real uid */
 363
 364 # if ENABLE_FEATURE_SUID_CONFIG
 365
 366 static struct suid_config_t {
 367         /* next ptr must be first: this struct needs to be llist-compatible */
 368         struct suid_config_t *m_next;
 369         struct bb_uidgid_t m_ugid;
 370         int m_applet;
 371         mode_t m_mode;
 372 } *suid_config;
 373
 374 static bool suid_cfg_readable;
 375
 376 /* libbb candidate */
 377 static char *get_trimmed_slice(char *s, char *e)
 378 {
 379         /* First, consider the value at e to be nul and back up until we
 380          * reach a non-space char.  Set the char after that (possibly at
 381          * the original e) to nul. */
 382         while (e-- > s) {
 383                 if (!isspace(*e)) {
 384                         break;
 385                 }
 386         }
 387         e[1] = '\0';
 388
 389         /* Next, advance past all leading space and return a ptr to the
 390          * first non-space char; possibly the terminating nul. */
 391         return skip_whitespace(s);
 392 }
 393
 394 static void parse_config_file(void)
 395 {
 396         /* Don't depend on the tools to combine strings. */
 397         static const char config_file[] ALIGN1 = "/etc/busybox.conf";
 398
 399         struct suid_config_t *sct_head;
 400         int applet_no;
 401         FILE *f;
 402         const char *errmsg;
 403         unsigned lc;
 404         smallint section;
 405         struct stat st;
 406
 407         ruid = getuid();
 408         if (ruid == 0) /* run by root - don't need to even read config file */
 409                 return;
 410
 411         if ((stat(config_file, &st) != 0)       /* No config file? */
 412          || !S_ISREG(st.st_mode)                /* Not a regular file? */
 413          || (st.st_uid != 0)                    /* Not owned by root? */
 414          || (st.st_mode & (S_IWGRP | S_IWOTH))  /* Writable by non-root? */
 415          || !(f = fopen_for_read(config_file))  /* Cannot open? */
 416         ) {
 417                 return;
 418         }
 419
 420         suid_cfg_readable = 1;
 421         sct_head = NULL;
 422         section = lc = 0;
 423
 424         while (1) {
 425                 char buffer[256];
 426                 char *s;
 427
 428                 if (!fgets(buffer, sizeof(buffer), f)) { /* Are we done? */
 429                         // Looks like bloat
 430                         //if (ferror(f)) {   /* Make sure it wasn't a read error. */
 431                         //      errmsg = "reading";
 432                         //      goto pe_label;
 433                         //}
 434                         fclose(f);
 435                         suid_config = sct_head; /* Success, so set the pointer. */
 436                         return;
 437                 }
 438
 439                 s = buffer;
 440                 lc++;                                   /* Got a (partial) line. */
 441
 442                 /* If a line is too long for our buffer, we consider it an error.
 443                  * The following test does mistreat one corner case though.
 444                  * If the final line of the file does not end with a newline and
 445                  * yet exactly fills the buffer, it will be treated as too long
 446                  * even though there isn't really a problem.  But it isn't really
 447                  * worth adding code to deal with such an unlikely situation, and
 448                  * we do err on the side of caution.  Besides, the line would be
 449                  * too long if it did end with a newline. */
 450                 if (!strchr(s, '\n') && !feof(f)) {
 451                         errmsg = "line too long";
 452                         goto pe_label;
 453                 }
 454
 455                 /* Trim leading and trailing whitespace, ignoring comments, and
 456                  * check if the resulting string is empty. */
 457                 s = get_trimmed_slice(s, strchrnul(s, '#'));
 458                 if (!*s) {
 459                         continue;
 460                 }
 461
 462                 /* Check for a section header. */
 463
 464                 if (*s == '[') {
 465                         /* Unlike the old code, we ignore leading and trailing
 466                          * whitespace for the section name.  We also require that
 467                          * there are no stray characters after the closing bracket. */
 468                         char *e = strchr(s, ']');
 469                         if (!e   /* Missing right bracket? */
 470                          || e[1] /* Trailing characters? */
 471                          || !*(s = get_trimmed_slice(s+1, e)) /* Missing name? */
 472                         ) {
 473                                 errmsg = "section header";
 474                                 goto pe_label;
 475                         }
 476                         /* Right now we only have one section so just check it.
 477                          * If more sections are added in the future, please don't
 478                          * resort to cascading ifs with multiple strcasecmp calls.
 479                          * That kind of bloated code is all too common.  A loop
 480                          * and a string table would be a better choice unless the
 481                          * number of sections is very small. */
 482                         if (strcasecmp(s, "SUID") == 0) {
 483                                 section = 1;
 484                                 continue;
 485                         }
 486                         section = -1;   /* Unknown section so set to skip. */
 487                         continue;
 488                 }
 489
 490                 /* Process sections. */
 491
 492                 if (section == 1) {             /* SUID */
 493                         /* Since we trimmed leading and trailing space above, we're
 494                          * now looking for strings of the form
 495                          *    <key>[::space::]*=[::space::]*<value>
 496                          * where both key and value could contain inner whitespace. */
 497
 498                         /* First get the key (an applet name in our case). */
 499                         char *e = strchr(s, '=');
 500                         if (e) {
 501                                 s = get_trimmed_slice(s, e);
 502                         }
 503                         if (!e || !*s) {        /* Missing '=' or empty key. */
 504                                 errmsg = "keyword";
 505                                 goto pe_label;
 506                         }
 507
 508                         /* Ok, we have an applet name.  Process the rhs if this
 509                          * applet is currently built in and ignore it otherwise.
 510                          * Note: this can hide config file bugs which only pop
 511                          * up when the busybox configuration is changed. */
 512                         applet_no = find_applet_by_name(s);
 513                         if (applet_no >= 0) {
 514                                 unsigned i;
 515                                 struct suid_config_t *sct;
 516
 517                                 /* Note: We currently don't check for duplicates!
 518                                  * The last config line for each applet will be the
 519                                  * one used since we insert at the head of the list.
 520                                  * I suppose this could be considered a feature. */
 521                                 sct = xzalloc(sizeof(*sct));
 522                                 sct->m_applet = applet_no;
 523                                 /*sct->m_mode = 0;*/
 524                                 sct->m_next = sct_head;
 525                                 sct_head = sct;
 526
 527                                 /* Get the specified mode. */
 528
 529                                 e = skip_whitespace(e+1);
 530
 531                                 for (i = 0; i < 3; i++) {
 532                                         /* There are 4 chars for each of user/group/other.
 533                                          * "x-xx" instead of "x-" are to make
 534                                          * "idx > 3" check catch invalid chars.
 535                                          */
 536                                         static const char mode_chars[] ALIGN1 = "Ssx-" "Ssx-" "x-xx";
 537                                         static const unsigned short mode_mask[] ALIGN2 = {
 538                                                 S_ISUID, S_ISUID|S_IXUSR, S_IXUSR, 0, /* Ssx- */
 539                                                 S_ISGID, S_ISGID|S_IXGRP, S_IXGRP, 0, /* Ssx- */
 540                                                                           S_IXOTH, 0  /*   x- */
 541                                         };
 542                                         const char *q = strchrnul(mode_chars + 4*i, *e);
 543                                         unsigned idx = q - (mode_chars + 4*i);
 544                                         if (idx > 3) {
 545                                                 errmsg = "mode";
 546                                                 goto pe_label;
 547                                         }
 548                                         sct->m_mode |= mode_mask[q - mode_chars];
 549                                         e++;
 550                                 }
 551
 552                                 /* Now get the user/group info. */
 553
 554                                 s = skip_whitespace(e);
 555                                 /* Default is 0.0, else parse USER.GROUP: */
 556                                 if (*s) {
 557                                         /* We require whitespace between mode and USER.GROUP */
 558                                         if ((s == e) || !(e = strchr(s, '.'))) {
 559                                                 errmsg = "uid.gid";
 560                                                 goto pe_label;
 561                                         }
 562                                         *e = ':'; /* get_uidgid needs USER:GROUP syntax */
 563                                         if (get_uidgid(&sct->m_ugid, s) == 0) {
 564                                                 errmsg = "unknown user/group";
 565                                                 goto pe_label;
 566                                         }
 567                                 }
 568                         }
 569                         continue;
 570                 }
 571
 572                 /* Unknown sections are ignored. */
 573
 574                 /* Encountering configuration lines prior to seeing a
 575                  * section header is treated as an error.  This is how
 576                  * the old code worked, but it may not be desirable.
 577                  * We may want to simply ignore such lines in case they
 578                  * are used in some future version of busybox. */
 579                 if (!section) {
 580                         errmsg = "keyword outside section";
 581                         goto pe_label;
 582                 }
 583         } /* while (1) */
 584
 585  pe_label:
 586         fclose(f);
 587         bb_error_msg("parse error in %s, line %u: %s", config_file, lc, errmsg);
 588
 589         /* Release any allocated memory before returning. */
 590         llist_free((llist_t*)sct_head, NULL);
 591 }
 592 # else
 593 static inline void parse_config_file(void)
 594 {
 595         IF_FEATURE_SUID(ruid = getuid();)
 596 }
 597 # endif /* FEATURE_SUID_CONFIG */
 598
 599
 600 # if ENABLE_FEATURE_SUID && NUM_APPLETS > 0
 601 #  if ENABLE_FEATURE_SUID_CONFIG
 602 /* check if u is member of group g */
 603 static int ingroup(uid_t u, gid_t g)
 604 {
 605         struct group *grp = getgrgid(g);
 606         if (grp) {
 607                 char **mem;
 608                 for (mem = grp->gr_mem; *mem; mem++) {
 609                         struct passwd *pwd = getpwnam(*mem);
 610                         if (pwd && (pwd->pw_uid == u))
 611                                 return 1;
 612                 }
 613         }
 614         return 0;
 615 }
 616 #  endif
 617
 618 static void check_suid(int applet_no)
 619 {
 620         gid_t rgid;  /* real gid */
 621
 622         if (ruid == 0) /* set by parse_config_file() */
 623                 return; /* run by root - no need to check more */
 624         rgid = getgid();
 625
 626 #  if ENABLE_FEATURE_SUID_CONFIG
 627         if (suid_cfg_readable) {
 628                 uid_t uid;
 629                 struct suid_config_t *sct;
 630                 mode_t m;
 631
 632                 for (sct = suid_config; sct; sct = sct->m_next) {
 633                         if (sct->m_applet == applet_no)
 634                                 goto found;
 635                 }
 636                 goto check_need_suid;
 637  found:
 638                 /* Is this user allowed to run this applet? */
 639                 m = sct->m_mode;
 640                 if (sct->m_ugid.uid == ruid)
 641                         /* same uid */
 642                         m >>= 6;
 643                 else if ((sct->m_ugid.gid == rgid) || ingroup(ruid, sct->m_ugid.gid))
 644                         /* same group / in group */
 645                         m >>= 3;
 646                 if (!(m & S_IXOTH)) /* is x bit not set? */
 647                         bb_error_msg_and_die("you have no permission to run this applet");
 648
 649                 /* We set effective AND saved ids. If saved-id is not set
 650                  * like we do below, seteuid(0) can still later succeed! */
 651
 652                 /* Are we directed to change gid
 653                  * (APPLET = *s* USER.GROUP or APPLET = *S* USER.GROUP)?
 654                  */
 655                 if (sct->m_mode & S_ISGID)
 656                         rgid = sct->m_ugid.gid;
 657                 /* else: we will set egid = rgid, thus dropping sgid effect */
 658                 if (setresgid(-1, rgid, rgid))
 659                         bb_perror_msg_and_die("setresgid");
 660
 661                 /* Are we directed to change uid
 662                  * (APPLET = s** USER.GROUP or APPLET = S** USER.GROUP)?
 663                  */
 664                 uid = ruid;
 665                 if (sct->m_mode & S_ISUID)
 666                         uid = sct->m_ugid.uid;
 667                 /* else: we will set euid = ruid, thus dropping suid effect */
 668                 if (setresuid(-1, uid, uid))
 669                         bb_perror_msg_and_die("setresuid");
 670
 671                 goto ret;
 672         }
 673 #   if !ENABLE_FEATURE_SUID_CONFIG_QUIET
 674         {
 675                 static bool onetime = 0;
 676
 677                 if (!onetime) {
 678                         onetime = 1;
 679                         bb_error_msg("using fallback suid method");
 680                 }
 681         }
 682 #   endif
 683  check_need_suid:
 684 #  endif
 685         if (APPLET_SUID(applet_no) == BB_SUID_REQUIRE) {
 686                 /* Real uid is not 0. If euid isn't 0 too, suid bit
 687                  * is most probably not set on our executable */
 688                 if (geteuid())
 689                         bb_error_msg_and_die("must be suid to work properly");
 690         } else if (APPLET_SUID(applet_no) == BB_SUID_DROP) {
 691                 /*
 692                  * Drop all privileges.
 693                  *
 694                  * Don't check for errors: in normal use, they are impossible,
 695                  * and in special cases, exiting is harmful. Example:
 696                  * 'unshare --user' when user's shell is also from busybox.
 697                  *
 698                  * 'unshare --user' creates a new user namespace without any
 699                  * uid mappings. Thus, busybox binary is setuid nobody:nogroup
 700                  * within the namespace, as that is the only user. However,
 701                  * since no uids are mapped, calls to setgid/setuid
 702                  * fail (even though they would do nothing).
 703                  */
 704                 setgid(rgid);
 705                 setuid(ruid);
 706         }
 707 #  if ENABLE_FEATURE_SUID_CONFIG
 708  ret: ;
 709         llist_free((llist_t*)suid_config, NULL);
 710 #  endif
 711 }
 712 # else
 713 #  define check_suid(x) ((void)0)
 714 # endif /* FEATURE_SUID */
 715
 716
 717 # if ENABLE_FEATURE_INSTALLER
 718 static const char usr_bin [] ALIGN1 = "/usr/bin/";
 719 static const char usr_sbin[] ALIGN1 = "/usr/sbin/";
 720 static const char *const install_dir[] = {
 721         &usr_bin [8], /* "/" */
 722         &usr_bin [4], /* "/bin/" */
 723         &usr_sbin[4]  /* "/sbin/" */
 724 #  if !ENABLE_INSTALL_NO_USR
 725         ,usr_bin
 726         ,usr_sbin
 727 #  endif
 728 };
 729
 730 /* create (sym)links for each applet */
 731 static void install_links(const char *busybox, int use_symbolic_links,
 732                 char *custom_install_dir)
 733 {
 734         /* directory table
 735          * this should be consistent w/ the enum,
 736          * busybox.h::bb_install_loc_t, or else... */
 737         int (*lf)(const char *, const char *);
 738         char *fpc;
 739         const char *appname = applet_names;
 740         unsigned i;
 741         int rc;
 742
 743         lf = link;
 744         if (use_symbolic_links)
 745                 lf = symlink;
 746
 747         for (i = 0; i < ARRAY_SIZE(applet_main); i++) {
 748                 fpc = concat_path_file(
 749                                 custom_install_dir ? custom_install_dir : install_dir[APPLET_INSTALL_LOC(i)],
 750                                 appname);
 751                 // debug: bb_error_msg("%slinking %s to busybox",
 752                 //              use_symbolic_links ? "sym" : "", fpc);
 753                 rc = lf(busybox, fpc);
 754                 if (rc != 0 && errno != EEXIST) {
 755                         bb_simple_perror_msg(fpc);
 756                 }
 757                 free(fpc);
 758                 while (*appname++ != '\0')
 759                         continue;
 760         }
 761 }
 762 # elif ENABLE_BUSYBOX
 763 static void install_links(const char *busybox UNUSED_PARAM,
 764                 int use_symbolic_links UNUSED_PARAM,
 765                 char *custom_install_dir UNUSED_PARAM)
 766 {
 767 }
 768 # endif
 769
 770 static void run_applet_and_exit(const char *name, char **argv) NORETURN;
 771
 772 # if ENABLE_BUSYBOX
 773 #  if ENABLE_FEATURE_SH_STANDALONE && ENABLE_FEATURE_TAB_COMPLETION
 774     /*
 775      * Insert "busybox" into applet table as well.
 776      * This makes standalone shell tab-complete this name too.
 777      * (Otherwise having "busybox" in applet table is not necessary,
 778      * there is other code which routes "busyboxANY_SUFFIX" name
 779      * to busybox_main()).
 780      */
 781 //usage:#define busybox_trivial_usage NOUSAGE_STR
 782 //usage:#define busybox_full_usage ""
 783 //applet:IF_BUSYBOX(IF_FEATURE_SH_STANDALONE(IF_FEATURE_TAB_COMPLETION(APPLET(busybox, BB_DIR_BIN, BB_SUID_MAYBE))))
 784 int busybox_main(int argc, char *argv[]) MAIN_EXTERNALLY_VISIBLE;
 785 #  else
 786 #   define busybox_main(argc,argv) busybox_main(argv)
 787 static
 788 #  endif
 789 int busybox_main(int argc UNUSED_PARAM, char **argv)
 790 {
 791         if (!argv[1]) {
 792                 /* Called without arguments */
 793                 const char *a;
 794                 int col;
 795                 unsigned output_width;
 796  help:
 797                 output_width = get_terminal_width(2);
 798
 799                 dup2(1, 2);
 800                 full_write2_str(bb_banner); /* reuse const string */
 801                 full_write2_str(" multi-call binary.\n"); /* reuse */
 802                 full_write2_str(
 803                         "BusyBox is copyrighted by many authors between 1998-2015.\n"
 804                         "Licensed under GPLv2. See source distribution for detailed\n"
 805                         "copyright notices.\n"
 806                         "\n"
 807                         "Usage: busybox [function [arguments]...]\n"
 808                         "   or: busybox --list"IF_FEATURE_INSTALLER("[-full]")"\n"
 809                         IF_FEATURE_INSTALLER(
 810                         "   or: busybox --install [-s] [DIR]\n"
 811                         )
 812                         "   or: function [arguments]...\n"
 813                         "\n"
 814                         IF_NOT_FEATURE_SH_STANDALONE(
 815                         "\tBusyBox is a multi-call binary that combines many common Unix\n"
 816                         "\tutilities into a single executable.  Most people will create a\n"
 817                         "\tlink to busybox for each function they wish to use and BusyBox\n"
 818                         "\twill act like whatever it was invoked as.\n"
 819                         )
 820                         IF_FEATURE_SH_STANDALONE(
 821                         "\tBusyBox is a multi-call binary that combines many common Unix\n"
 822                         "\tutilities into a single executable.  The shell in this build\n"
 823                         "\tis configured to run built-in utilities without $PATH search.\n"
 824                         "\tYou don't need to install a link to busybox for each utility.\n"
 825                         "\tTo run external program, use full path (/sbin/ip instead of ip).\n"
 826                         )
 827                         "\n"
 828                         "Currently defined functions:\n"
 829                 );
 830                 col = 0;
 831                 a = applet_names;
 832                 /* prevent last comma to be in the very last pos */
 833                 output_width--;
 834                 while (*a) {
 835                         int len2 = strlen(a) + 2;
 836                         if (col >= (int)output_width - len2) {
 837                                 full_write2_str(",\n");
 838                                 col = 0;
 839                         }
 840                         if (col == 0) {
 841                                 col = 6;
 842                                 full_write2_str("\t");
 843                         } else {
 844                                 full_write2_str(", ");
 845                         }
 846                         full_write2_str(a);
 847                         col += len2;
 848                         a += len2 - 1;
 849                 }
 850                 full_write2_str("\n");
 851                 return 0;
 852         }
 853
 854         if (is_prefixed_with(argv[1], "--list")) {
 855                 unsigned i = 0;
 856                 const char *a = applet_names;
 857                 dup2(1, 2);
 858                 while (*a) {
 859 #  if ENABLE_FEATURE_INSTALLER
 860                         if (argv[1][6]) /* --list-full? */
 861                                 full_write2_str(install_dir[APPLET_INSTALL_LOC(i)] + 1);
 862 #  endif
 863                         full_write2_str(a);
 864                         full_write2_str("\n");
 865                         i++;
 866                         while (*a++ != '\0')
 867                                 continue;
 868                 }
 869                 return 0;
 870         }
 871
 872         if (ENABLE_FEATURE_INSTALLER && strcmp(argv[1], "--install") == 0) {
 873                 int use_symbolic_links;
 874                 const char *busybox;
 875
 876                 busybox = xmalloc_readlink(bb_busybox_exec_path);
 877                 if (!busybox) {
 878                         /* bb_busybox_exec_path is usually "/proc/self/exe".
 879                          * In chroot, readlink("/proc/self/exe") usually fails.
 880                          * In such case, better use argv[0] as symlink target
 881                          * if it is a full path name.
 882                          */
 883                         if (argv[0][0] != '/')
 884                                 bb_error_msg_and_die("'%s' is not an absolute path", argv[0]);
 885                         busybox = argv[0];
 886                 }
 887                 /* busybox --install [-s] [DIR]:
 888                  * -s: make symlinks
 889                  * DIR: directory to install links to
 890                  */
 891                 use_symbolic_links = (argv[2] && strcmp(argv[2], "-s") == 0 && ++argv);
 892                 install_links(busybox, use_symbolic_links, argv[2]);
 893                 return 0;
 894         }
 895
 896         if (strcmp(argv[1], "--help") == 0) {
 897                 /* "busybox --help [<applet>]" */
 898                 if (!argv[2])
 899                         goto help;
 900                 /* convert to "<applet> --help" */
 901                 argv[0] = argv[2];
 902                 argv[2] = NULL;
 903         } else {
 904                 /* "busybox <applet> arg1 arg2 ..." */
 905                 argv++;
 906         }
 907         /* We support "busybox /a/path/to/applet args..." too. Allows for
 908          * "#!/bin/busybox"-style wrappers */
 909         applet_name = bb_get_last_path_component_nostrip(argv[0]);
 910         run_applet_and_exit(applet_name, argv);
 911 }
 912 # endif
 913
 914 #if defined(__linux__) && (NUM_APPLETS > 1)
 915 void FAST_FUNC set_task_comm(const char *comm)
 916 {
 917         /* okay if too long (truncates) */
 918         prctl(PR_SET_NAME, (long)comm, 0, 0, 0);
 919 }
 920 #endif
 921
 922 # if NUM_APPLETS > 0
 923 void FAST_FUNC run_applet_no_and_exit(int applet_no, const char *name, char **argv)
 924 {
 925         int argc = string_array_len(argv);
 926
 927         /* Reinit some shared global data */
 928         xfunc_error_retval = EXIT_FAILURE;
 929         /*
 930          * We do not use argv[0]: do not want to repeat massaging of
 931          * "-/sbin/halt" -> "halt", for example.
 932          */
 933         applet_name = name;
 934
 935         /* Special case. POSIX says "test --help"
 936          * should be no different from e.g. "test --foo".
 937          * Thus for "test", we skip --help check.
 938          * "true" and "false" are also special.
 939          */
 940         if (1
 941 #  if defined APPLET_NO_test
 942          && applet_no != APPLET_NO_test
 943 #  endif
 944 #  if defined APPLET_NO_true
 945          && applet_no != APPLET_NO_true
 946 #  endif
 947 #  if defined APPLET_NO_false
 948          && applet_no != APPLET_NO_false
 949 #  endif
 950         ) {
 951                 if (argc == 2 && strcmp(argv[1], "--help") == 0) {
 952                         /* Make "foo --help" exit with 0: */
 953                         xfunc_error_retval = 0;
 954                         bb_show_usage();
 955                 }
 956         }
 957         if (ENABLE_FEATURE_SUID)
 958                 check_suid(applet_no);
 959         xfunc_error_retval = applet_main[applet_no](argc, argv);
 960         /* Note: applet_main() may also not return (die on a xfunc or such) */
 961         xfunc_die();
 962 }
 963 # endif /* NUM_APPLETS > 0 */
 964
 965 # if ENABLE_BUSYBOX || NUM_APPLETS > 0
 966 static NORETURN void run_applet_and_exit(const char *name, char **argv)
 967 {
 968 #  if ENABLE_BUSYBOX
 969         if (is_prefixed_with(name, "busybox"))
 970                 exit(busybox_main(/*unused:*/ 0, argv));
 971 #  endif
 972 #  if NUM_APPLETS > 0
 973         /* find_applet_by_name() search is more expensive, so goes second */
 974         {
 975                 int applet = find_applet_by_name(name);
 976                 if (applet >= 0)
 977                         run_applet_no_and_exit(applet, name, argv);
 978         }
 979 #  endif
 980
 981         /*bb_error_msg_and_die("applet not found"); - links in printf */
 982         full_write2_str(applet_name);
 983         full_write2_str(": applet not found\n");
 984         /* POSIX: "If a command is not found, the exit status shall be 127" */
 985         exit(127);
 986 }
 987 # endif
 988
 989 #endif /* !defined(SINGLE_APPLET_MAIN) */
 990
 991
 992 #if ENABLE_BUILD_LIBBUSYBOX
 993 int lbb_main(char **argv)
 994 #else
 995 int main(int argc UNUSED_PARAM, char **argv)
 996 #endif
 997 {
 998 #if 0
 999         /* TODO: find a use for a block of memory between end of .bss
1000          * and end of page. For example, I'm getting "_end:0x812e698 2408 bytes"
1001          * - more than 2k of wasted memory (in this particular build)
1002          * *per each running process*!
1003          * (If your linker does not generate "_end" name, weak attribute
1004          * makes &_end == NULL, end_len == 0 here.)
1005          */
1006         extern char _end[] __attribute__((weak));
1007         unsigned end_len = (-(int)_end) & 0xfff;
1008         printf("_end:%p %u bytes\n", &_end, end_len);
1009 #endif
1010
1011         /* Tweak malloc for reduced memory consumption */
1012 #ifdef M_TRIM_THRESHOLD
1013         /* M_TRIM_THRESHOLD is the maximum amount of freed top-most memory
1014          * to keep before releasing to the OS
1015          * Default is way too big: 256k
1016          */
1017         mallopt(M_TRIM_THRESHOLD, 8 * 1024);
1018 #endif
1019 #ifdef M_MMAP_THRESHOLD
1020         /* M_MMAP_THRESHOLD is the request size threshold for using mmap()
1021          * Default is too big: 256k
1022          */
1023         mallopt(M_MMAP_THRESHOLD, 32 * 1024 - 256);
1024 #endif
1025 #if 0 /*def M_TOP_PAD*/
1026         /* When the program break is increased, then M_TOP_PAD bytes are added
1027          * to the sbrk(2) request. When the heap is trimmed because of free(3),
1028          * this much free space is preserved at the top of the heap.
1029          * glibc default seems to be way too big: 128k, but need to verify.
1030          */
1031         mallopt(M_TOP_PAD, 8 * 1024);
1032 #endif
1033
1034 #if !BB_MMU
1035         /* NOMMU re-exec trick sets high-order bit in first byte of name */
1036         if (argv[0][0] & 0x80) {
1037                 re_execed = 1;
1038                 argv[0][0] &= 0x7f;
1039         }
1040 #endif
1041
1042 #if defined(SINGLE_APPLET_MAIN)
1043
1044         /* Only one applet is selected in .config */
1045         if (argv[1] && is_prefixed_with(argv[0], "busybox")) {
1046                 /* "busybox <applet> <params>" should still work as expected */
1047                 argv++;
1048         }
1049         /* applet_names in this case is just "applet\0\0" */
1050         lbb_prepare(applet_names IF_FEATURE_INDIVIDUAL(, argv));
1051 # if ENABLE_BUILD_LIBBUSYBOX
1052         return SINGLE_APPLET_MAIN(string_array_len(argv), argv);
1053 # else
1054         return SINGLE_APPLET_MAIN(argc, argv);
1055 # endif
1056
1057 #elif !ENABLE_BUSYBOX && NUM_APPLETS == 0
1058
1059         full_write2_str(bb_basename(argv[0]));
1060         full_write2_str(": no applets enabled\n");
1061         exit(127);
1062
1063 #else
1064
1065         lbb_prepare("busybox" IF_FEATURE_INDIVIDUAL(, argv));
1066 # if !ENABLE_BUSYBOX
1067         if (argv[1] && is_prefixed_with(bb_basename(argv[0]), "busybox"))
1068                 argv++;
1069 # endif
1070         applet_name = argv[0];
1071         if (applet_name[0] == '-')
1072                 applet_name++;
1073         applet_name = bb_basename(applet_name);
1074
1075         /* If we are a result of execv("/proc/self/exe"), fix ugly comm of "exe" */
1076         if (ENABLE_FEATURE_SH_STANDALONE
1077          || ENABLE_FEATURE_PREFER_APPLETS
1078          || !BB_MMU
1079         ) {
1080                 if (NUM_APPLETS > 1)
1081                         set_task_comm(applet_name);
1082         }
1083
1084         parse_config_file(); /* ...maybe, if FEATURE_SUID_CONFIG */
1085         run_applet_and_exit(applet_name, argv);
1086
1087 #endif
1088 }