init/init.c

   1 /* vi: set sw=4 ts=4: */
   2 /*
   3  * Mini init implementation for busybox
   4  *
   5  *
   6  * Copyright (C) 1995, 1996 by Bruce Perens <bruce@pixar.com>.
   7  * Adjusted by so many folks, it's impossible to keep track.
   8  *
   9  * This program is free software; you can redistribute it and/or modify
  10  * it under the terms of the GNU General Public License as published by
  11  * the Free Software Foundation; either version 2 of the License, or
  12  * (at your option) any later version.
  13  *
  14  * This program is distributed in the hope that it will be useful,
  15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  17  * General Public License for more details.
  18  *
  19  * You should have received a copy of the GNU General Public License
  20  * along with this program; if not, write to the Free Software
  21  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
  22  *
  23  */
  24
  25 /* Turn this on to disable all the dangerous
  26    rebooting stuff when debugging.
  27 #define DEBUG_INIT
  28 */
  29
  30 #include "busybox.h"
  31 #include <stdio.h>
  32 #include <stdlib.h>
  33 #include <errno.h>
  34 #include <paths.h>
  35 #include <signal.h>
  36 #include <stdarg.h>
  37 #include <string.h>
  38 #include <termios.h>
  39 #include <unistd.h>
  40 #include <sys/fcntl.h>
  41 #include <sys/ioctl.h>
  42 #include <sys/mount.h>
  43 #include <sys/types.h>
  44 #include <sys/wait.h>
  45 #ifdef BB_SYSLOGD
  46 # include <sys/syslog.h>
  47 #endif
  48
  49 #define bb_need_full_version
  50 #define BB_DECLARE_EXTERN
  51 #include "messages.c"
  52
  53 /* From <linux/vt.h> */
  54 struct vt_stat {
  55         unsigned short v_active;        /* active vt */
  56         unsigned short v_signal;        /* signal to send */
  57         unsigned short v_state;         /* vt bitmask */
  58 };
  59 static const int VT_GETSTATE = 0x5603;  /* get global vt state info */
  60
  61 /* From <linux/serial.h> */
  62 struct serial_struct {
  63         int     type;
  64         int     line;
  65         int     port;
  66         int     irq;
  67         int     flags;
  68         int     xmit_fifo_size;
  69         int     custom_divisor;
  70         int     baud_base;
  71         unsigned short  close_delay;
  72         char    reserved_char[2];
  73         int     hub6;
  74         unsigned short  closing_wait; /* time to wait before closing */
  75         unsigned short  closing_wait2; /* no longer used... */
  76         int     reserved[4];
  77 };
  78
  79
  80
  81 #ifndef RB_HALT_SYSTEM
  82 static const int RB_HALT_SYSTEM = 0xcdef0123;
  83 static const int RB_ENABLE_CAD = 0x89abcdef;
  84 static const int RB_DISABLE_CAD = 0;
  85 #define RB_POWER_OFF    0x4321fedc
  86 static const int RB_AUTOBOOT = 0x01234567;
  87 #if defined(__GLIBC__) || defined (__UCLIBC__)
  88 #include <sys/reboot.h>
  89   #define init_reboot(magic) reboot(magic)
  90 #else
  91   #define init_reboot(magic) reboot(0xfee1dead, 672274793, magic)
  92 #endif
  93 #endif
  94
  95 #ifndef _PATH_STDPATH
  96 #define _PATH_STDPATH   "/usr/bin:/bin:/usr/sbin:/sbin"
  97 #endif
  98
  99
 100 #if defined BB_FEATURE_INIT_COREDUMPS
 101 /*
 102  * When a file named CORE_ENABLE_FLAG_FILE exists, setrlimit is called
 103  * before processes are spawned to set core file size as unlimited.
 104  * This is for debugging only.  Don't use this is production, unless
 105  * you want core dumps lying about....
 106  */
 107 #define CORE_ENABLE_FLAG_FILE "/.init_enable_core"
 108 #include <sys/resource.h>
 109 #include <sys/time.h>
 110 #endif
 111
 112 #define KERNEL_VERSION(a,b,c) (((a) << 16) + ((b) << 8) + (c))
 113
 114 #if defined(__GLIBC__)
 115 #include <sys/kdaemon.h>
 116 #else
 117 static _syscall2(int, bdflush, int, func, int, data);
 118 #endif                                                  /* __GLIBC__ */
 119
 120
 121 #define VT_PRIMARY   "/dev/tty1"     /* Primary virtual console */
 122 #define VT_SECONDARY "/dev/tty2"     /* Virtual console */
 123 #define VT_THIRD     "/dev/tty3"     /* Virtual console */
 124 #define VT_FOURTH    "/dev/tty4"     /* Virtual console */
 125 #define VT_LOG       "/dev/tty5"     /* Virtual console */
 126 #define SERIAL_CON0  "/dev/ttyS0"    /* Primary serial console */
 127 #define SERIAL_CON1  "/dev/ttyS1"    /* Serial console */
 128 #define SHELL        "-/bin/sh"      /* Default shell */
 129 #define INITTAB      "/etc/inittab"  /* inittab file location */
 130 #ifndef INIT_SCRIPT
 131 #define INIT_SCRIPT  "/etc/init.d/rcS"   /* Default sysinit script. */
 132 #endif
 133
 134 static const int LOG = 0x1;
 135 static const int CONSOLE = 0x2;
 136
 137 /* Allowed init action types */
 138 typedef enum {
 139         SYSINIT = 1,
 140         RESPAWN,
 141         ASKFIRST,
 142         WAIT,
 143         ONCE,
 144         CTRLALTDEL
 145 } initActionEnum;
 146
 147 /* A mapping between "inittab" action name strings and action type codes. */
 148 typedef struct initActionType {
 149         const char *name;
 150         initActionEnum action;
 151 } initActionType;
 152
 153 static const struct initActionType actions[] = {
 154         {"sysinit", SYSINIT},
 155         {"respawn", RESPAWN},
 156         {"askfirst", ASKFIRST},
 157         {"wait", WAIT},
 158         {"once", ONCE},
 159         {"ctrlaltdel", CTRLALTDEL},
 160         {0, 0}
 161 };
 162
 163 /* Set up a linked list of initActions, to be read from inittab */
 164 typedef struct initActionTag initAction;
 165 struct initActionTag {
 166         pid_t pid;
 167         char process[256];
 168         char console[256];
 169         initAction *nextPtr;
 170         initActionEnum action;
 171 };
 172 initAction *initActionList = NULL;
 173
 174
 175 static char *secondConsole = VT_SECONDARY;
 176 static char *thirdConsole  = VT_THIRD;
 177 static char *fourthConsole = VT_FOURTH;
 178 static char *log           = VT_LOG;
 179 static int  kernelVersion  = 0;
 180 static char termType[32]   = "TERM=linux";
 181 static char console[32]    = _PATH_CONSOLE;
 182
 183 static void delete_initAction(initAction * action);
 184
 185
 186 /* Print a message to the specified device.
 187  * Device may be bitwise-or'd from LOG | CONSOLE */
 188 static void message(int device, char *fmt, ...)
 189                    __attribute__ ((format (printf, 2, 3)));
 190 static void message(int device, char *fmt, ...)
 191 {
 192         va_list arguments;
 193         int fd;
 194
 195 #ifdef BB_SYSLOGD
 196
 197         /* Log the message to syslogd */
 198         if (device & LOG) {
 199                 char msg[1024];
 200
 201                 va_start(arguments, fmt);
 202                 vsnprintf(msg, sizeof(msg), fmt, arguments);
 203                 va_end(arguments);
 204                 openlog("init", 0, LOG_USER);
 205                 syslog(LOG_USER|LOG_INFO, msg);
 206                 closelog();
 207         }
 208 #else
 209         static int log_fd = -1;
 210
 211         /* Take full control of the log tty, and never close it.
 212          * It's mine, all mine!  Muhahahaha! */
 213         if (log_fd < 0) {
 214                 if (log == NULL) {
 215                         /* don't even try to log, because there is no such console */
 216                         log_fd = -2;
 217                         /* log to main console instead */
 218                         device = CONSOLE;
 219                 } else if ((log_fd = device_open(log, O_RDWR|O_NDELAY)) < 0) {
 220                         log_fd = -2;
 221                         fprintf(stderr, "Bummer, can't write to log on %s!\r\n", log);
 222                         fflush(stderr);
 223                         log = NULL;
 224                         device = CONSOLE;
 225                 }
 226         }
 227         if ((device & LOG) && (log_fd >= 0)) {
 228                 va_start(arguments, fmt);
 229                 vdprintf(log_fd, fmt, arguments);
 230                 va_end(arguments);
 231         }
 232 #endif
 233
 234         if (device & CONSOLE) {
 235                 /* Always send console messages to /dev/console so people will see them. */
 236                 if (
 237                         (fd =
 238                          device_open(_PATH_CONSOLE,
 239                                                  O_WRONLY | O_NOCTTY | O_NDELAY)) >= 0) {
 240                         va_start(arguments, fmt);
 241                         vdprintf(fd, fmt, arguments);
 242                         va_end(arguments);
 243                         close(fd);
 244                 } else {
 245                         fprintf(stderr, "Bummer, can't print: ");
 246                         va_start(arguments, fmt);
 247                         vfprintf(stderr, fmt, arguments);
 248                         fflush(stderr);
 249                         va_end(arguments);
 250                 }
 251         }
 252 }
 253
 254 /* Set terminal settings to reasonable defaults */
 255 void set_term(int fd)
 256 {
 257         struct termios tty;
 258
 259         tcgetattr(fd, &tty);
 260
 261         /* set control chars */
 262         tty.c_cc[VINTR]  = 3;   /* C-c */
 263         tty.c_cc[VQUIT]  = 28;  /* C-\ */
 264         tty.c_cc[VERASE] = 127; /* C-? */
 265         tty.c_cc[VKILL]  = 21;  /* C-u */
 266         tty.c_cc[VEOF]   = 4;   /* C-d */
 267         tty.c_cc[VSTART] = 17;  /* C-q */
 268         tty.c_cc[VSTOP]  = 19;  /* C-s */
 269         tty.c_cc[VSUSP]  = 26;  /* C-z */
 270
 271         /* use line dicipline 0 */
 272         tty.c_line = 0;
 273
 274         /* Make it be sane */
 275         tty.c_cflag &= CBAUD|CBAUDEX|CSIZE|CSTOPB|PARENB|PARODD;
 276         tty.c_cflag |= HUPCL|CLOCAL;
 277
 278         /* input modes */
 279         tty.c_iflag = ICRNL | IXON | IXOFF;
 280
 281         /* output modes */
 282         tty.c_oflag = OPOST | ONLCR;
 283
 284         /* local modes */
 285         tty.c_lflag =
 286                 ISIG | ICANON | ECHO | ECHOE | ECHOK | ECHOCTL | ECHOKE | IEXTEN;
 287
 288         tcsetattr(fd, TCSANOW, &tty);
 289 }
 290
 291 /* How much memory does this machine have?
 292    Units are kBytes to avoid overflow on 4GB machines */
 293 static int check_free_memory()
 294 {
 295         struct sysinfo info;
 296         unsigned int result, u, s=10;
 297
 298         if (sysinfo(&info) != 0) {
 299                 perror_msg("Error checking free memory: ");
 300                 return -1;
 301         }
 302
 303         /* Kernels 2.0.x and 2.2.x return info.mem_unit==0 with values in bytes.
 304          * Kernels 2.4.0 return info.mem_unit in bytes. */
 305         u = info.mem_unit;
 306         if (u==0) u=1;
 307         while ( (u&1) == 0 && s > 0 ) { u>>=1; s--; }
 308         result = (info.totalram>>s) + (info.totalswap>>s);
 309         result = result*u;
 310         if (result < 0) result = INT_MAX;
 311         return result;
 312 }
 313
 314 static void console_init()
 315 {
 316         int fd;
 317         int tried_devcons = 0;
 318         int tried_vtprimary = 0;
 319         struct vt_stat vt;
 320         struct serial_struct sr;
 321         char *s;
 322
 323         if ((s = getenv("TERM")) != NULL) {
 324                 snprintf(termType, sizeof(termType) - 1, "TERM=%s", s);
 325         }
 326
 327         if ((s = getenv("CONSOLE")) != NULL) {
 328                 snprintf(console, sizeof(console) - 1, "%s", s);
 329         }
 330 #if #cpu(sparc)
 331         /* sparc kernel supports console=tty[ab] parameter which is also
 332          * passed to init, so catch it here */
 333         else if ((s = getenv("console")) != NULL) {
 334                 /* remap tty[ab] to /dev/ttyS[01] */
 335                 if (strcmp(s, "ttya") == 0)
 336                         snprintf(console, sizeof(console) - 1, "%s", SERIAL_CON0);
 337                 else if (strcmp(s, "ttyb") == 0)
 338                         snprintf(console, sizeof(console) - 1, "%s", SERIAL_CON1);
 339         }
 340 #endif
 341         else {
 342                 /* 2.2 kernels: identify the real console backend and try to use it */
 343                 if (ioctl(0, TIOCGSERIAL, &sr) == 0) {
 344                         /* this is a serial console */
 345                         snprintf(console, sizeof(console) - 1, "/dev/ttyS%d", sr.line);
 346                 } else if (ioctl(0, VT_GETSTATE, &vt) == 0) {
 347                         /* this is linux virtual tty */
 348                         snprintf(console, sizeof(console) - 1, "/dev/tty%d",
 349                                          vt.v_active);
 350                 } else {
 351                         snprintf(console, sizeof(console) - 1, "%s", _PATH_CONSOLE);
 352                         tried_devcons++;
 353                 }
 354         }
 355
 356         while ((fd = open(console, O_RDONLY | O_NONBLOCK)) < 0) {
 357                 /* Can't open selected console -- try /dev/console */
 358                 if (!tried_devcons) {
 359                         tried_devcons++;
 360                         snprintf(console, sizeof(console) - 1, "%s", _PATH_CONSOLE);
 361                         continue;
 362                 }
 363                 /* Can't open selected console -- try vt1 */
 364                 if (!tried_vtprimary) {
 365                         tried_vtprimary++;
 366                         snprintf(console, sizeof(console) - 1, "%s", VT_PRIMARY);
 367                         continue;
 368                 }
 369                 break;
 370         }
 371         if (fd < 0) {
 372                 /* Perhaps we should panic here? */
 373                 snprintf(console, sizeof(console) - 1, "/dev/null");
 374         } else {
 375                 /* check for serial console and disable logging to tty5 & running a
 376                    * shell to tty2-4 */
 377                 if (ioctl(0, TIOCGSERIAL, &sr) == 0) {
 378                         log = NULL;
 379                         secondConsole = NULL;
 380                         thirdConsole = NULL;
 381                         fourthConsole = NULL;
 382                         /* Force the TERM setting to vt102 for serial console --
 383                          * iff TERM is set to linux (the default) */
 384                         if (strcmp( termType, "TERM=linux" ) == 0)
 385                                 snprintf(termType, sizeof(termType) - 1, "TERM=vt102");
 386                         message(LOG | CONSOLE,
 387                                         "serial console detected.  Disabling virtual terminals.\r\n");
 388                 }
 389                 close(fd);
 390         }
 391         message(LOG, "console=%s\n", console);
 392 }
 393
 394 static pid_t run(char *command, char *terminal, int get_enter)
 395 {
 396         int i, fd;
 397         pid_t pid;
 398         char *tmpCmd;
 399         char *cmd[255], *cmdpath;
 400         char buf[255];
 401         static const char press_enter[] =
 402
 403 #ifdef CUSTOMIZED_BANNER
 404 #include CUSTOMIZED_BANNER
 405 #endif
 406
 407                 "\nPlease press Enter to activate this console. ";
 408         char *environment[] = {
 409                 "HOME=/",
 410                 "PATH=/usr/bin:/bin:/usr/sbin:/sbin",
 411                 "SHELL=/bin/sh",
 412                 termType,
 413                 "USER=root",
 414                 0
 415         };
 416
 417         if ((pid = fork()) == 0) {
 418                 /* Clean up */
 419                 ioctl(0, TIOCNOTTY, 0);
 420                 close(0);
 421                 close(1);
 422                 close(2);
 423                 setsid();
 424
 425                 /* Reset signal handlers set for parent process */
 426                 signal(SIGUSR1, SIG_DFL);
 427                 signal(SIGUSR2, SIG_DFL);
 428                 signal(SIGINT, SIG_DFL);
 429                 signal(SIGTERM, SIG_DFL);
 430                 signal(SIGHUP, SIG_DFL);
 431
 432                 if ((fd = device_open(terminal, O_RDWR)) < 0) {
 433                         message(LOG | CONSOLE, "Bummer, can't open %s\r\n", terminal);
 434                         exit(1);
 435                 }
 436                 dup2(fd, 0);
 437                 dup2(fd, 1);
 438                 dup2(fd, 2);
 439                 ioctl(0, TIOCSCTTY, 1);
 440                 tcsetpgrp(0, getpgrp());
 441                 set_term(0);
 442
 443                 if (get_enter == TRUE) {
 444                         /*
 445                          * Save memory by not exec-ing anything large (like a shell)
 446                          * before the user wants it. This is critical if swap is not
 447                          * enabled and the system has low memory. Generally this will
 448                          * be run on the second virtual console, and the first will
 449                          * be allowed to start a shell or whatever an init script
 450                          * specifies.
 451                          */
 452 #ifdef DEBUG_INIT
 453                         pid_t shell_pgid = getpid();
 454                         message(LOG, "Waiting for enter to start '%s' (pid %d, console %s)\r\n",
 455                                         command, shell_pgid, terminal);
 456 #endif
 457                         write(fileno(stdout), press_enter, sizeof(press_enter) - 1);
 458                         getc(stdin);
 459                 }
 460
 461 #ifdef DEBUG_INIT
 462                 /* Log the process name and args */
 463                 message(LOG, "Starting pid %d, console %s: '%s'\r\n",
 464                                 shell_pgid, terminal, command);
 465 #endif
 466
 467                 /* See if any special /bin/sh requiring characters are present */
 468                 if (strpbrk(command, "~`!$^&*()=|\\{}[];\"'<>?") != NULL) {
 469                         cmd[0] = SHELL;
 470                         cmd[1] = "-c";
 471                         strcpy(buf, "exec ");
 472                         strncat(buf, command, sizeof(buf) - strlen(buf) - 1);
 473                         cmd[2] = buf;
 474                         cmd[3] = NULL;
 475                 } else {
 476                         /* Convert command (char*) into cmd (char**, one word per string) */
 477                         for (tmpCmd = command, i = 0;
 478                                  (tmpCmd = strsep(&command, " \t")) != NULL;) {
 479                                 if (*tmpCmd != '\0') {
 480                                         cmd[i] = tmpCmd;
 481                                         tmpCmd++;
 482                                         i++;
 483                                 }
 484                         }
 485                         cmd[i] = NULL;
 486                 }
 487
 488                cmdpath = cmd[0];
 489
 490                /*
 491                    Interactive shells want to see a dash in argv[0].  This
 492                    typically is handled by login, argv will be setup this
 493                    way if a dash appears at the front of the command path
 494                    (like "-/bin/sh").
 495                */
 496
 497                if (*cmdpath == '-') {
 498                        char *s;
 499
 500                        /* skip over the dash */
 501                          ++cmdpath;
 502
 503                        /* find the last component in the command pathname */
 504                                                  s = get_last_path_component(cmdpath);
 505
 506                        /* make a new argv[0] */
 507                        if ((cmd[0] = malloc(strlen(s)+2)) == NULL) {
 508                                message(LOG | CONSOLE, "malloc failed");
 509                                cmd[0] = cmdpath;
 510                        } else {
 511                                cmd[0][0] = '-';
 512                                strcpy(cmd[0]+1, s);
 513                        }
 514                }
 515
 516 #if defined BB_FEATURE_INIT_COREDUMPS
 517                 {
 518                         struct stat sb;
 519                         if (stat (CORE_ENABLE_FLAG_FILE, &sb) == 0) {
 520                                 struct rlimit limit;
 521                                 limit.rlim_cur = RLIM_INFINITY;
 522                                 limit.rlim_max = RLIM_INFINITY;
 523                                 setrlimit(RLIMIT_CORE, &limit);
 524                         }
 525                 }
 526 #endif
 527
 528                 /* Now run it.  The new program will take over this PID,
 529                  * so nothing further in init.c should be run. */
 530                 execve(cmdpath, cmd, environment);
 531
 532                 /* We're still here?  Some error happened. */
 533                 message(LOG | CONSOLE, "Bummer, could not run '%s': %s\n", cmdpath,
 534                                 strerror(errno));
 535                 exit(-1);
 536         }
 537         return pid;
 538 }
 539
 540 static int waitfor(char *command, char *terminal, int get_enter)
 541 {
 542         int status, wpid;
 543         int pid = run(command, terminal, get_enter);
 544
 545         while (1) {
 546                 wpid = wait(&status);
 547                 if (wpid > 0 && wpid != pid) {
 548                         continue;
 549                 }
 550                 if (wpid == pid)
 551                         break;
 552         }
 553         return wpid;
 554 }
 555
 556 /* Make sure there is enough memory to do something useful. *
 557  * Calls "swapon -a" if needed so be sure /etc/fstab is present... */
 558 static void check_memory()
 559 {
 560         struct stat statBuf;
 561
 562         if (check_free_memory() > 1000)
 563                 return;
 564
 565         if (stat("/etc/fstab", &statBuf) == 0) {
 566                 /* swapon -a requires /proc typically */
 567                 waitfor("mount proc /proc -t proc", console, FALSE);
 568                 /* Try to turn on swap */
 569                 waitfor("swapon -a", console, FALSE);
 570                 if (check_free_memory() < 1000)
 571                         goto goodnight;
 572         } else
 573                 goto goodnight;
 574         return;
 575
 576   goodnight:
 577         message(CONSOLE,
 578                         "Sorry, your computer does not have enough memory.\r\n");
 579         while (1)
 580                 sleep(1);
 581 }
 582
 583 /* Run all commands to be run right before halt/reboot */
 584 static void run_lastAction(void)
 585 {
 586         initAction *a;
 587         for (a = initActionList; a; a = a->nextPtr) {
 588                 if (a->action == CTRLALTDEL) {
 589                         waitfor(a->process, a->console, FALSE);
 590                         delete_initAction(a);
 591                 }
 592         }
 593 }
 594
 595
 596 #ifndef DEBUG_INIT
 597 static void shutdown_system(void)
 598 {
 599
 600         /* first disable our SIGHUP signal */
 601         signal(SIGHUP, SIG_DFL);
 602
 603         /* Allow Ctrl-Alt-Del to reboot system. */
 604         init_reboot(RB_ENABLE_CAD);
 605
 606         message(CONSOLE|LOG, "\r\nThe system is going down NOW !!\r\n");
 607         sync();
 608
 609         /* Send signals to every process _except_ pid 1 */
 610         message(CONSOLE|LOG, "Sending SIGTERM to all processes.\r\n");
 611         kill(-1, SIGTERM);
 612         sleep(1);
 613         sync();
 614
 615         message(CONSOLE|LOG, "Sending SIGKILL to all processes.\r\n");
 616         kill(-1, SIGKILL);
 617         sleep(1);
 618
 619         /* run everything to be run at "ctrlaltdel" */
 620         run_lastAction();
 621
 622         sync();
 623         if (kernelVersion > 0 && kernelVersion <= KERNEL_VERSION(2,2,11)) {
 624                 /* bdflush, kupdate not needed for kernels >2.2.11 */
 625                 bdflush(1, 0);
 626                 sync();
 627         }
 628 }
 629
 630 static void halt_signal(int sig)
 631 {
 632         shutdown_system();
 633         message(CONSOLE|LOG,
 634                         "The system is halted. Press %s or turn off power\r\n",
 635                         (secondConsole == NULL) /* serial console */
 636                         ? "Reset" : "CTRL-ALT-DEL");
 637         sync();
 638
 639         /* allow time for last message to reach serial console */
 640         sleep(2);
 641
 642         if (sig == SIGUSR2 && kernelVersion >= KERNEL_VERSION(2,2,0))
 643                 init_reboot(RB_POWER_OFF);
 644         else
 645                 init_reboot(RB_HALT_SYSTEM);
 646         exit(0);
 647 }
 648
 649 static void reboot_signal(int sig)
 650 {
 651         shutdown_system();
 652         message(CONSOLE|LOG, "Please stand by while rebooting the system.\r\n");
 653         sync();
 654
 655         /* allow time for last message to reach serial console */
 656         sleep(2);
 657
 658         init_reboot(RB_AUTOBOOT);
 659         exit(0);
 660 }
 661
 662 #if defined BB_FEATURE_INIT_CHROOT
 663
 664 #warning BB_FEATURE_INIT_CHROOT is out of date and should be rewritten to us
 665 #warning pivot root instead.  Do not even bother till this work is done...
 666 #warning You have been warned.
 667
 668 #if ! defined BB_FEATURE_USE_PROCFS
 669 #error Sorry, I depend on the /proc filesystem right now.
 670 #endif
 671
 672 static void check_chroot(int sig)
 673 {
 674         char *argv_init[2] = { "init", NULL, };
 675         char *envp_init[3] = { "HOME=/", "TERM=linux", NULL, };
 676         char rootpath[256], *tc;
 677         int fd;
 678
 679         if ((fd = open("/proc/sys/kernel/init-chroot", O_RDONLY)) == -1) {
 680                 message(CONSOLE,
 681                                 "SIGHUP recived, but could not open proc file\r\n");
 682                 sleep(2);
 683                 return;
 684         }
 685         if (read(fd, rootpath, sizeof(rootpath)) == -1) {
 686                 message(CONSOLE,
 687                                 "SIGHUP recived, but could not read proc file\r\n");
 688                 sleep(2);
 689                 return;
 690         }
 691         close(fd);
 692
 693         if (rootpath[0] == '\0') {
 694                 message(CONSOLE,
 695                                 "SIGHUP recived, but new root is not valid: %s\r\n",
 696                                 rootpath);
 697                 sleep(2);
 698                 return;
 699         }
 700
 701         tc = strrchr(rootpath, '\n');
 702         *tc = '\0';
 703
 704         /* Ok, making it this far means we commit */
 705         message(CONSOLE, "Please stand by, changing root to `%s'.\r\n",
 706                         rootpath);
 707
 708         /* kill all other programs first */
 709         message(CONSOLE, "Sending SIGTERM to all processes.\r\n");
 710         kill(-1, SIGTERM);
 711         sleep(2);
 712         sync();
 713
 714         message(CONSOLE, "Sending SIGKILL to all processes.\r\n");
 715         kill(-1, SIGKILL);
 716         sleep(2);
 717         sync();
 718
 719         /* ok, we don't need /proc anymore. we also assume that the signaling
 720          * process left the rest of the filesystems alone for us */
 721         umount("/proc");
 722
 723         /* Ok, now we chroot. Hopefully we only have two things mounted, the
 724          * new chroot'd mount point, and the old "/" mount. s,
 725          * we go ahead and unmount the old "/". This should trigger the kernel
 726          * to set things up the Right Way(tm). */
 727
 728         if (!chroot(rootpath))
 729                 umount("/dev/root");
 730
 731         /* If the chroot fails, we are already too far to turn back, so we
 732          * continue and hope that executing init below will revive the system */
 733
 734         /* close all of our descriptors and open new ones */
 735         close(0);
 736         close(1);
 737         close(2);
 738         open("/dev/console", O_RDWR, 0);
 739         dup(0);
 740         dup(0);
 741
 742         message(CONSOLE, "Executing real init...\r\n");
 743         /* execute init in the (hopefully) new root */
 744         execve("/sbin/init", argv_init, envp_init);
 745
 746         message(CONSOLE,
 747                         "ERROR: Could not exec new init. Press %s to reboot.\r\n",
 748                         (secondConsole == NULL) /* serial console */
 749                         ? "Reset" : "CTRL-ALT-DEL");
 750         return;
 751 }
 752 #endif                                                  /* BB_FEATURE_INIT_CHROOT */
 753
 754 #endif                                                  /* ! DEBUG_INIT */
 755
 756 void new_initAction(initActionEnum action, char *process, char *cons)
 757 {
 758         initAction *newAction;
 759
 760         if (*cons == '\0')
 761                 cons = console;
 762
 763         /* If BusyBox detects that a serial console is in use,
 764          * then entries not refering to the console or null devices will _not_ be run.
 765          * The exception to this rule is the null device.
 766          */
 767         if (secondConsole == NULL && strcmp(cons, console)
 768                 && strcmp(cons, "/dev/null"))
 769                 return;
 770
 771         newAction = calloc((size_t) (1), sizeof(initAction));
 772         if (!newAction) {
 773                 message(LOG | CONSOLE, "Memory allocation failure\n");
 774                 while (1)
 775                         sleep(1);
 776         }
 777         newAction->nextPtr = initActionList;
 778         initActionList = newAction;
 779         strncpy(newAction->process, process, 255);
 780         newAction->action = action;
 781         strncpy(newAction->console, cons, 255);
 782         newAction->pid = 0;
 783 //    message(LOG|CONSOLE, "process='%s' action='%d' console='%s'\n",
 784 //      newAction->process, newAction->action, newAction->console);
 785 }
 786
 787 static void delete_initAction(initAction * action)
 788 {
 789         initAction *a, *b = NULL;
 790
 791         for (a = initActionList; a; b = a, a = a->nextPtr) {
 792                 if (a == action) {
 793                         if (b == NULL) {
 794                                 initActionList = a->nextPtr;
 795                         } else {
 796                                 b->nextPtr = a->nextPtr;
 797                         }
 798                         free(a);
 799                         break;
 800                 }
 801         }
 802 }
 803
 804 /* NOTE that if BB_FEATURE_USE_INITTAB is NOT defined,
 805  * then parse_inittab() simply adds in some default
 806  * actions(i.e runs INIT_SCRIPT and then starts a pair
 807  * of "askfirst" shells).  If BB_FEATURE_USE_INITTAB
 808  * _is_ defined, but /etc/inittab is missing, this
 809  * results in the same set of default behaviors.
 810  * */
 811 void parse_inittab(void)
 812 {
 813 #ifdef BB_FEATURE_USE_INITTAB
 814         FILE *file;
 815         char buf[256], lineAsRead[256], tmpConsole[256];
 816         char *id, *runlev, *action, *process, *eol;
 817         const struct initActionType *a = actions;
 818         int foundIt;
 819
 820
 821         file = fopen(INITTAB, "r");
 822         if (file == NULL) {
 823                 /* No inittab file -- set up some default behavior */
 824 #endif
 825                 /* Swapoff on halt/reboot */
 826                 new_initAction(CTRLALTDEL, "/sbin/swapoff -a", console);
 827                 /* Umount all filesystems on halt/reboot */
 828                 new_initAction(CTRLALTDEL, "/bin/umount -a -r", console);
 829                 /* Askfirst shell on tty1 */
 830                 new_initAction(ASKFIRST, SHELL, console);
 831                 /* Askfirst shell on tty2 */
 832                 if (secondConsole != NULL)
 833                         new_initAction(ASKFIRST, SHELL, secondConsole);
 834                 /* Askfirst shell on tty3 */
 835                 if (thirdConsole != NULL)
 836                         new_initAction(ASKFIRST, SHELL, thirdConsole);
 837                 /* Askfirst shell on tty4 */
 838                 if (fourthConsole != NULL)
 839                         new_initAction(ASKFIRST, SHELL, fourthConsole);
 840                 /* sysinit */
 841                 new_initAction(SYSINIT, INIT_SCRIPT, console);
 842
 843                 return;
 844 #ifdef BB_FEATURE_USE_INITTAB
 845         }
 846
 847         while (fgets(buf, 255, file) != NULL) {
 848                 foundIt = FALSE;
 849                 /* Skip leading spaces */
 850                 for (id = buf; *id == ' ' || *id == '\t'; id++);
 851
 852                 /* Skip the line if it's a comment */
 853                 if (*id == '#' || *id == '\n')
 854                         continue;
 855
 856                 /* Trim the trailing \n */
 857                 eol = strrchr(id, '\n');
 858                 if (eol != NULL)
 859                         *eol = '\0';
 860
 861                 /* Keep a copy around for posterity's sake (and error msgs) */
 862                 strcpy(lineAsRead, buf);
 863
 864                 /* Separate the ID field from the runlevels */
 865                 runlev = strchr(id, ':');
 866                 if (runlev == NULL || *(runlev + 1) == '\0') {
 867                         message(LOG | CONSOLE, "Bad inittab entry: %s\n", lineAsRead);
 868                         continue;
 869                 } else {
 870                         *runlev = '\0';
 871                         ++runlev;
 872                 }
 873
 874                 /* Separate the runlevels from the action */
 875                 action = strchr(runlev, ':');
 876                 if (action == NULL || *(action + 1) == '\0') {
 877                         message(LOG | CONSOLE, "Bad inittab entry: %s\n", lineAsRead);
 878                         continue;
 879                 } else {
 880                         *action = '\0';
 881                         ++action;
 882                 }
 883
 884                 /* Separate the action from the process */
 885                 process = strchr(action, ':');
 886                 if (process == NULL || *(process + 1) == '\0') {
 887                         message(LOG | CONSOLE, "Bad inittab entry: %s\n", lineAsRead);
 888                         continue;
 889                 } else {
 890                         *process = '\0';
 891                         ++process;
 892                 }
 893
 894                 /* Ok, now process it */
 895                 a = actions;
 896                 while (a->name != 0) {
 897                         if (strcmp(a->name, action) == 0) {
 898                                 if (*id != '\0') {
 899                                         struct stat statBuf;
 900
 901                                         strcpy(tmpConsole, "/dev/");
 902                                         strncat(tmpConsole, id, 200);
 903                                         if (stat(tmpConsole, &statBuf) != 0) {
 904                                                 message(LOG | CONSOLE,
 905                                                                 "device '%s' does not exist.  Did you read the directions?\n",
 906                                                                 tmpConsole);
 907                                                 break;
 908                                         }
 909                                         id = tmpConsole;
 910                                 }
 911                                 new_initAction(a->action, process, id);
 912                                 foundIt = TRUE;
 913                         }
 914                         a++;
 915                 }
 916                 if (foundIt == TRUE)
 917                         continue;
 918                 else {
 919                         /* Choke on an unknown action */
 920                         message(LOG | CONSOLE, "Bad inittab entry: %s\n", lineAsRead);
 921                 }
 922         }
 923         return;
 924 #endif /* BB_FEATURE_USE_INITTAB */
 925 }
 926
 927
 928
 929 extern int init_main(int argc, char **argv)
 930 {
 931         initAction *a;
 932         pid_t wpid;
 933         int status;
 934
 935 #ifndef DEBUG_INIT
 936         /* Expect to be invoked as init with PID=1 or be invoked as linuxrc */
 937         if (getpid() != 1
 938 #ifdef BB_FEATURE_LINUXRC
 939                         && strstr(applet_name, "linuxrc") == NULL
 940 #endif
 941                           )
 942         {
 943                         usage("init\n\nInit is the parent of all processes.\n\n"
 944                                   "This version of init is designed to be run only "
 945                                   "by the kernel.\n");
 946         }
 947         /* Set up sig handlers  -- be sure to
 948          * clear all of these in run() */
 949         signal(SIGUSR1, halt_signal);
 950         signal(SIGUSR2, halt_signal);
 951         signal(SIGINT, reboot_signal);
 952         signal(SIGTERM, reboot_signal);
 953 #if defined BB_FEATURE_INIT_CHROOT
 954         signal(SIGHUP, check_chroot);
 955 #endif
 956
 957         /* Turn off rebooting via CTL-ALT-DEL -- we get a
 958          * SIGINT on CAD so we can shut things down gracefully... */
 959         init_reboot(RB_DISABLE_CAD);
 960 #endif
 961
 962         /* Figure out what kernel this is running */
 963         kernelVersion = get_kernel_revision();
 964
 965         /* Figure out where the default console should be */
 966         console_init();
 967
 968         /* Close whatever files are open, and reset the console. */
 969         close(0);
 970         close(1);
 971         close(2);
 972         set_term(0);
 973         chdir("/");
 974         setsid();
 975
 976         /* Make sure PATH is set to something sane */
 977         putenv(_PATH_STDPATH);
 978
 979         /* Hello world */
 980 #ifndef DEBUG_INIT
 981         message(
 982 #if ! defined BB_FEATURE_EXTRA_QUIET
 983                         CONSOLE|
 984 #endif
 985                         LOG,
 986                         "init started:  %s\r\n", full_version);
 987 #else
 988         message(
 989 #if ! defined BB_FEATURE_EXTRA_QUIET
 990                         CONSOLE|
 991 #endif
 992                         LOG,
 993                         "init(%d) started:  %s\r\n", getpid(), full_version);
 994 #endif
 995
 996
 997         /* Make sure there is enough memory to do something useful. */
 998         check_memory();
 999
1000         /* Check if we are supposed to be in single user mode */
1001         if (argc > 1 && (!strcmp(argv[1], "single") ||
1002                                          !strcmp(argv[1], "-s") || !strcmp(argv[1], "1"))) {
1003                 /* Ask first then start a shell on tty2-4 */
1004                 if (secondConsole != NULL)
1005                         new_initAction(ASKFIRST, SHELL, secondConsole);
1006                 if (thirdConsole != NULL)
1007                         new_initAction(ASKFIRST, SHELL, thirdConsole);
1008                 if (fourthConsole != NULL)
1009                         new_initAction(ASKFIRST, SHELL, fourthConsole);
1010                 /* Start a shell on tty1 */
1011                 new_initAction(RESPAWN, SHELL, console);
1012         } else {
1013                 /* Not in single user mode -- see what inittab says */
1014
1015                 /* NOTE that if BB_FEATURE_USE_INITTAB is NOT defined,
1016                  * then parse_inittab() simply adds in some default
1017                  * actions(i.e runs INIT_SCRIPT and then starts a pair
1018                  * of "askfirst" shells */
1019                 parse_inittab();
1020         }
1021
1022         /* Fix up argv[0] to be certain we claim to be init */
1023         argv[0]="init";
1024
1025         if (argc > 1)
1026                 strncpy(argv[1], "\0", strlen(argv[1])+1);
1027
1028         /* Now run everything that needs to be run */
1029
1030         /* First run the sysinit command */
1031         for (a = initActionList; a; a = a->nextPtr) {
1032                 if (a->action == SYSINIT) {
1033                         waitfor(a->process, a->console, FALSE);
1034                         /* Now remove the "sysinit" entry from the list */
1035                         delete_initAction(a);
1036                 }
1037         }
1038         /* Next run anything that wants to block */
1039         for (a = initActionList; a; a = a->nextPtr) {
1040                 if (a->action == WAIT) {
1041                         waitfor(a->process, a->console, FALSE);
1042                         /* Now remove the "wait" entry from the list */
1043                         delete_initAction(a);
1044                 }
1045         }
1046         /* Next run anything to be run only once */
1047         for (a = initActionList; a; a = a->nextPtr) {
1048                 if (a->action == ONCE) {
1049                         run(a->process, a->console, FALSE);
1050                         /* Now remove the "once" entry from the list */
1051                         delete_initAction(a);
1052                 }
1053         }
1054         /* If there is nothing else to do, stop */
1055         if (initActionList == NULL) {
1056                 message(LOG | CONSOLE,
1057                                 "No more tasks for init -- sleeping forever.\n");
1058                 while (1)
1059                         sleep(1);
1060         }
1061
1062         /* Now run the looping stuff for the rest of forever */
1063         while (1) {
1064                 for (a = initActionList; a; a = a->nextPtr) {
1065                         /* Only run stuff with pid==0.  If they have
1066                          * a pid, that means they are still running */
1067                         if (a->pid == 0) {
1068                                 switch (a->action) {
1069                                 case RESPAWN:
1070                                         /* run the respawn stuff */
1071                                         a->pid = run(a->process, a->console, FALSE);
1072                                         break;
1073                                 case ASKFIRST:
1074                                         /* run the askfirst stuff */
1075                                         a->pid = run(a->process, a->console, TRUE);
1076                                         break;
1077                                         /* silence the compiler's incessant whining */
1078                                 default:
1079                                         break;
1080                                 }
1081                         }
1082                 }
1083                 /* Wait for a child process to exit */
1084                 wpid = wait(&status);
1085                 if (wpid > 0) {
1086                         /* Find out who died and clean up their corpse */
1087                         for (a = initActionList; a; a = a->nextPtr) {
1088                                 if (a->pid == wpid) {
1089                                         a->pid = 0;
1090                                         message(LOG,
1091                                                         "Process '%s' (pid %d) exited.  Scheduling it for restart.\n",
1092                                                         a->process, wpid);
1093                                 }
1094                         }
1095                 }
1096                 sleep(1);
1097         }
1098 }
1099
1100 /*
1101 Local Variables:
1102 c-file-style: "linux"
1103 c-basic-offset: 4
1104 tab-width: 4
1105 End:
1106 */