init/init.c

   1 /*
   2  * Mini init implementation for busybox
   3  *
   4  *
   5  * Copyright (C) 1995, 1996 by Bruce Perens <bruce@pixar.com>.
   6  * Adjusted by so many folks, it's impossible to keep track.
   7  *
   8  * This program is free software; you can redistribute it and/or modify
   9  * it under the terms of the GNU General Public License as published by
  10  * the Free Software Foundation; either version 2 of the License, or
  11  * (at your option) any later version.
  12  *
  13  * This program is distributed in the hope that it will be useful,
  14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  16  * General Public License for more details.
  17  *
  18  * You should have received a copy of the GNU General Public License
  19  * along with this program; if not, write to the Free Software
  20  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
  21  *
  22  */
  23
  24 #include "internal.h"
  25 #include <stdio.h>
  26 #include <stdlib.h>
  27 #include <stdarg.h>
  28 #include <unistd.h>
  29 #include <errno.h>
  30 #include <signal.h>
  31 #include <termios.h>
  32 #include <paths.h>
  33 #include <sys/types.h>
  34 #include <sys/fcntl.h>
  35 #include <sys/wait.h>
  36 #include <string.h>
  37 #include <sys/mount.h>
  38 #include <sys/reboot.h>
  39 #include <sys/kdaemon.h>
  40 #include <sys/sysmacros.h>
  41 #include <linux/serial.h>       /* for serial_struct */
  42 #include <sys/vt.h>             /* for vt_stat */
  43 #include <sys/ioctl.h>
  44 #ifdef BB_SYSLOGD
  45 #include <sys/syslog.h>
  46 #endif
  47
  48 #if ! defined BB_FEATURE_USE_PROCFS
  49 #error Sorry, I depend on the /proc filesystem right now.
  50 #endif
  51
  52
  53 #define VT_PRIMARY      "/dev/tty1"     /* Primary virtual console */
  54 #define VT_SECONDARY    "/dev/tty2"     /* Virtual console */
  55 #define VT_LOG          "/dev/tty3"     /* Virtual console */
  56 #define SERIAL_CON0     "/dev/ttyS0"    /* Primary serial console */
  57 #define SERIAL_CON1     "/dev/ttyS1"    /* Serial console */
  58 #define SHELL           "/bin/sh"       /* Default shell */
  59 #define INITSCRIPT      "/etc/init.d/rcS"       /* Initscript. */
  60
  61 #define LOG             0x1
  62 #define CONSOLE         0x2
  63 static char *console = _PATH_CONSOLE;
  64 static char *second_console = VT_SECONDARY;
  65 static char *log = VT_LOG;
  66 static int kernel_version = 0;
  67
  68
  69 /* try to open up the specified device */
  70 int device_open(char *device, int mode)
  71 {
  72     int m, f, fd = -1;
  73
  74     m = mode | O_NONBLOCK;
  75
  76     /* Retry up to 5 times */
  77     for (f = 0; f < 5; f++)
  78         if ((fd = open(device, m)) >= 0)
  79             break;
  80     if (fd < 0)
  81         return fd;
  82     /* Reset original flags. */
  83     if (m != mode)
  84         fcntl(fd, F_SETFL, mode);
  85     return fd;
  86 }
  87
  88 /* print a message to the specified device:
  89  * device may be bitwise-or'd from LOG | CONSOLE */
  90 void message(int device, char *fmt, ...)
  91 {
  92     int fd;
  93     va_list arguments;
  94 #ifdef BB_SYSLOGD
  95
  96     /* Log the message to syslogd */
  97     if (device & LOG ) {
  98         char msg[1024];
  99         va_start(arguments, fmt);
 100         vsnprintf(msg, sizeof(msg), fmt, arguments);
 101         va_end(arguments);
 102         syslog(LOG_DAEMON|LOG_NOTICE, msg);
 103     }
 104
 105 #else
 106     static int log_fd=-1;
 107
 108     /* Take full control of the log tty, and never close it.
 109      * It's mine, all mine!  Muhahahaha! */
 110     if (log_fd < 0) {
 111         if (log == NULL) {
 112         /* don't even try to log, because there is no such console */
 113         log_fd = -2;
 114         /* log to main console instead */
 115         device = CONSOLE;
 116     }
 117     else if ((log_fd = device_open(log, O_RDWR|O_NDELAY)) < 0) {
 118             log_fd=-1;
 119             fprintf(stderr, "Bummer, can't write to log on %s!\r\n", log);
 120             fflush(stderr);
 121             return;
 122         }
 123     }
 124     if ( (device & LOG) && (log_fd >= 0) ) {
 125         va_start(arguments, fmt);
 126         vdprintf(log_fd, fmt, arguments);
 127         va_end(arguments);
 128     }
 129 #endif
 130
 131     if (device & CONSOLE) {
 132         /* Always send console messages to /dev/console so people will see them. */
 133         if ((fd = device_open(_PATH_CONSOLE, O_WRONLY|O_NOCTTY|O_NDELAY)) >= 0) {
 134             va_start(arguments, fmt);
 135             vdprintf(fd, fmt, arguments);
 136             va_end(arguments);
 137             close(fd);
 138         } else {
 139             fprintf(stderr, "Bummer, can't print: ");
 140             va_start(arguments, fmt);
 141             vfprintf(stderr, fmt, arguments);
 142             fflush(stderr);
 143             va_end(arguments);
 144         }
 145     }
 146 }
 147
 148
 149 /* Set terminal settings to reasonable defaults */
 150 void set_term( int fd)
 151 {
 152     struct termios tty;
 153 #if 0
 154     static const char control_characters[] = {
 155         '\003', '\034', '\177', '\030', '\004', '\0',
 156         '\1', '\0', '\021', '\023', '\032', '\0', '\022',
 157         '\017', '\027', '\026', '\0'
 158         };
 159 #else
 160     static const char control_characters[] = {
 161         '\003', '\034', '\177', '\025', '\004', '\0',
 162         '\1', '\0', '\021', '\023', '\032', '\0', '\022',
 163         '\017', '\027', '\026', '\0'
 164         };
 165 #endif
 166
 167     tcgetattr(fd, &tty);
 168
 169     /* set control chars */
 170     memcpy(tty.c_cc, control_characters, sizeof(control_characters));
 171
 172     /* use line dicipline 0 */
 173     tty.c_line = 0;
 174
 175     /* Make it be sane */
 176     //tty.c_cflag &= CBAUD|CBAUDEX|CSIZE|CSTOPB|PARENB|PARODD;
 177     //tty.c_cflag |= HUPCL|CLOCAL;
 178
 179     /* input modes */
 180     tty.c_iflag = ICRNL|IXON|IXOFF;
 181
 182     /* output modes */
 183     tty.c_oflag = OPOST|ONLCR;
 184
 185     /* local modes */
 186     tty.c_lflag = ISIG|ICANON|ECHO|ECHOE|ECHOK|ECHOCTL|ECHOKE|IEXTEN;
 187
 188     tcsetattr(fd, TCSANOW, &tty);
 189 }
 190
 191 /* How much memory does this machine have? */
 192 static int mem_total()
 193 {
 194     char s[80];
 195     char *p = "/proc/meminfo";
 196     FILE *f;
 197     const char pattern[] = "MemTotal:";
 198
 199     if ((f = fopen(p, "r")) < 0) {
 200         message(LOG, "Error opening %s: %s\n", p, strerror( errno));
 201         return -1;
 202     }
 203     while (NULL != fgets(s, 79, f)) {
 204         p = strstr(s, pattern);
 205         if (NULL != p) {
 206             fclose(f);
 207             return (atoi(p + strlen(pattern)));
 208         }
 209     }
 210     return -1;
 211 }
 212
 213 static void console_init()
 214 {
 215     int fd;
 216     int tried_devcons = 0;
 217     int tried_vtprimary = 0;
 218     struct serial_struct sr;
 219     char *s;
 220
 221     if ((s = getenv("CONSOLE")) != NULL) {
 222         console = s;
 223     }
 224 #if #cpu(sparc)
 225     /* sparc kernel supports console=tty[ab] parameter which is also
 226      * passed to init, so catch it here */
 227     else if ((s = getenv("console")) != NULL) {
 228         /* remap tty[ab] to /dev/ttyS[01] */
 229         if (strcmp( s, "ttya" )==0)
 230             console = SERIAL_CON0;
 231         else if (strcmp( s, "ttyb" )==0)
 232             console = SERIAL_CON1;
 233     }
 234 #endif
 235     else {
 236         struct vt_stat vt;
 237         static char the_console[13];
 238
 239         console = the_console;
 240         /* 2.2 kernels: identify the real console backend and try to use it */
 241         if (ioctl(0, TIOCGSERIAL, &sr) == 0) {
 242             /* this is a serial console */
 243             snprintf( the_console, sizeof the_console, "/dev/ttyS%d", sr.line );
 244         }
 245         else if (ioctl(0, VT_GETSTATE, &vt) == 0) {
 246             /* this is linux virtual tty */
 247             snprintf( the_console, sizeof the_console, "/dev/tty%d", vt.v_active );
 248         } else {
 249             console = _PATH_CONSOLE;
 250             tried_devcons++;
 251         }
 252     }
 253
 254     while ((fd = open(console, O_RDONLY | O_NONBLOCK)) < 0) {
 255         /* Can't open selected console -- try /dev/console */
 256         if (!tried_devcons) {
 257             tried_devcons++;
 258             console = _PATH_CONSOLE;
 259             continue;
 260         }
 261         /* Can't open selected console -- try vt1 */
 262         if (!tried_vtprimary) {
 263             tried_vtprimary++;
 264             console = VT_PRIMARY;
 265             continue;
 266         }
 267         break;
 268     }
 269     if (fd < 0)
 270         /* Perhaps we should panic here? */
 271         console = "/dev/null";
 272     else {
 273         /* check for serial console and disable logging to tty3 & running a
 274         * shell to tty2 */
 275         if (ioctl(0,TIOCGSERIAL,&sr) == 0) {
 276             message(LOG|CONSOLE, "serial console detected.  Disabling 2nd virtual terminal.\r\n", console );
 277             log = NULL;
 278             second_console = NULL;
 279         }
 280         close(fd);
 281     }
 282     message(LOG, "console=%s\n", console );
 283 }
 284
 285 static int waitfor(int pid)
 286 {
 287     int status, wpid;
 288
 289     message(LOG, "Waiting for process %d.\n", pid);
 290     while ((wpid = wait(&status)) != pid) {
 291         if (wpid > 0)
 292             message(LOG, "pid %d exited, status=0x%x.\n", wpid, status);
 293     }
 294     return wpid;
 295 }
 296
 297
 298 static pid_t run(const char * const* command,
 299         char *terminal, int get_enter)
 300 {
 301     int fd;
 302     pid_t pid;
 303     const char * const* cmd = command+1;
 304     static const char press_enter[] =
 305         "\nPlease press Enter to activate this console. ";
 306
 307     if ((pid = fork()) == 0) {
 308         /* Clean up */
 309         close(0);
 310         close(1);
 311         close(2);
 312         setsid();
 313
 314         /* Reset signal handlers set for parent process */
 315         signal(SIGUSR1, SIG_DFL);
 316         signal(SIGUSR2, SIG_DFL);
 317         signal(SIGINT, SIG_DFL);
 318         signal(SIGTERM, SIG_DFL);
 319
 320         if ((fd = device_open(terminal, O_RDWR)) < 0) {
 321             message(LOG, "Bummer, can't open %s\r\n", terminal);
 322             exit(-1);
 323         }
 324         dup(fd);
 325         dup(fd);
 326         tcsetpgrp(0, getpgrp());
 327         set_term(0);
 328
 329         if (get_enter==TRUE) {
 330             /*
 331              * Save memory by not exec-ing anything large (like a shell)
 332              * before the user wants it. This is critical if swap is not
 333              * enabled and the system has low memory. Generally this will
 334              * be run on the second virtual console, and the first will
 335              * be allowed to start a shell or whatever an init script
 336              * specifies.
 337              */
 338             char c;
 339             message(LOG, "Waiting for enter to start '%s' (pid %d, console %s)\r\n",
 340                     *cmd, getpid(), terminal );
 341             write(1, press_enter, sizeof(press_enter) - 1);
 342             read(0, &c, 1);
 343         }
 344
 345         /* Log the process name and args */
 346         message(LOG, "Starting pid %d, console %s: '", getpid(), terminal);
 347         while ( *cmd) message(LOG, "%s ", *cmd++);
 348         message(LOG, "'\r\n");
 349
 350         /* Now run it.  The new program will take over this PID,
 351          * so nothing further in init.c should be run. */
 352         execvp(*command, (char**)command+1);
 353
 354         message(LOG, "Bummer, could not run '%s'\n", command);
 355         exit(-1);
 356     }
 357     return pid;
 358 }
 359
 360 /* Make sure there is enough memory to do something useful. *
 361  * Calls swapon if needed so be sure /proc is mounted. */
 362 static void check_memory()
 363 {
 364     struct stat statbuf;
 365     const char* const swap_on_cmd[] =
 366             { "/bin/swapon", "swapon", "-a", 0};
 367
 368     if (mem_total() > 3500)
 369         return;
 370
 371     if (stat("/etc/fstab", &statbuf) == 0) {
 372         /* Try to turn on swap */
 373         waitfor(run(swap_on_cmd, log, FALSE));
 374         if (mem_total() < 3500)
 375             goto goodnight;
 376     } else
 377         goto goodnight;
 378     return;
 379
 380 goodnight:
 381         message(CONSOLE, "Sorry, your computer does not have enough memory.\r\n");
 382         while (1) sleep(1);
 383 }
 384
 385 static void shutdown_system(void)
 386 {
 387     const char* const swap_off_cmd[] = { "swapoff", "swapoff", "-a", 0};
 388     const char* const umount_cmd[] = { "umount", "umount", "-a", "-n", 0};
 389
 390 #ifndef DEBUG_INIT
 391     /* Allow Ctrl-Alt-Del to reboot system. */
 392     reboot(RB_ENABLE_CAD);
 393 #endif
 394     message(CONSOLE, "\r\nThe system is going down NOW !!\r\n");
 395     sync();
 396     /* Send signals to every process _except_ pid 1 */
 397     message(CONSOLE, "Sending SIGHUP to all processes.\r\n");
 398 #ifndef DEBUG_INIT
 399     kill(-1, SIGHUP);
 400 #endif
 401     sleep(2);
 402     sync();
 403     message(CONSOLE, "Sending SIGKILL to all processes.\r\n");
 404 #ifndef DEBUG_INIT
 405     kill(-1, SIGKILL);
 406 #endif
 407     sleep(1);
 408     message(CONSOLE, "Disabling swap.\r\n");
 409     waitfor(run( swap_off_cmd, console, FALSE));
 410     message(CONSOLE, "Unmounting filesystems.\r\n");
 411     waitfor(run( umount_cmd, console, FALSE));
 412     sync();
 413     if (kernel_version > 0 && kernel_version <= 2 * 65536 + 2 * 256 + 11) {
 414         /* bdflush, kupdate not needed for kernels >2.2.11 */
 415         message(CONSOLE, "Flushing buffers.\r\n");
 416         bdflush(1, 0);
 417         sync();
 418     }
 419 }
 420
 421 static void halt_signal(int sig)
 422 {
 423     shutdown_system();
 424     message(CONSOLE,
 425             "The system is halted. Press CTRL-ALT-DEL or turn off power\r\n");
 426     sync();
 427 #ifndef DEBUG_INIT
 428     while (1) sleep(1);
 429     reboot(RB_HALT_SYSTEM);
 430     //reboot(RB_POWER_OFF);
 431 #endif
 432     exit(0);
 433 }
 434
 435 static void reboot_signal(int sig)
 436 {
 437     shutdown_system();
 438     message(CONSOLE, "Please stand by while rebooting the system.\r\n");
 439     sync();
 440     while (1) sleep(1);
 441 #ifndef DEBUG_INIT
 442     reboot(RB_AUTOBOOT);
 443 #endif
 444     exit(0);
 445 }
 446
 447 extern int init_main(int argc, char **argv)
 448 {
 449     int run_rc = TRUE;
 450     int wait_for_enter = TRUE;
 451     pid_t pid1 = 0;
 452     pid_t pid2 = 0;
 453     struct stat statbuf;
 454     const char* const rc_script_command[] = { INITSCRIPT, INITSCRIPT, 0};
 455     const char* const shell_command[] = { SHELL, "-" SHELL, 0};
 456     const char* const* tty0_command = shell_command;
 457     const char* const* tty1_command = shell_command;
 458 #ifdef BB_INIT_CMD_IF_RC_SCRIPT_EXITS
 459     const char* const rc_exit_command[] = { "BB_INIT_CMD_IF_RC_SCRIPT_EXITS",
 460                                             "BB_INIT_CMD_IF_RC_SCRIPT_EXITS", 0 };
 461 #endif
 462
 463 #ifdef DEBUG_INIT
 464     char *hello_msg_format =
 465         "init(%d) started:  BusyBox v%s (%s) multi-call binary\r\n";
 466 #else
 467     char *hello_msg_format =
 468         "init started:  BusyBox v%s (%s) multi-call binary\r\n";
 469 #endif
 470
 471
 472 #ifndef DEBUG_INIT
 473     /* Expect to be PID 1 iff we are run as init (not linuxrc) */
 474     if (getpid() != 1 && strstr(argv[0], "init")!=NULL ) {
 475         usage( "init\n\nInit is the parent of all processes.\n\n"
 476                 "This version of init is designed to be run only by the kernel\n");
 477     }
 478 #endif
 479
 480     /* Check if we are supposed to be in single user mode */
 481     if ( argc > 1 && (!strcmp(argv[1], "single") ||
 482                 !strcmp(argv[1], "-s") || !strcmp(argv[1], "1"))) {
 483         run_rc = FALSE;
 484     }
 485
 486
 487     /* Set up sig handlers  -- be sure to
 488      * clear all of these in run() */
 489     signal(SIGUSR1, halt_signal);
 490     signal(SIGUSR2, reboot_signal);
 491     signal(SIGINT, reboot_signal);
 492     signal(SIGTERM, reboot_signal);
 493
 494     /* Turn off rebooting via CTL-ALT-DEL -- we get a
 495      * SIGINT on CAD so we can shut things down gracefully... */
 496 #ifndef DEBUG_INIT
 497     reboot(RB_DISABLE_CAD);
 498 #endif
 499
 500     /* Figure out where the default console should be */
 501     console_init();
 502
 503     /* Close whatever files are open, and reset the console. */
 504     close(0);
 505     close(1);
 506     close(2);
 507     set_term(0);
 508     setsid();
 509
 510     /* Make sure PATH is set to something sane */
 511     putenv(_PATH_STDPATH);
 512
 513
 514     /* Hello world */
 515 #ifndef DEBUG_INIT
 516     message(CONSOLE|LOG, hello_msg_format, BB_VER, BB_BT);
 517 #else
 518     message(CONSOLE|LOG, hello_msg_format, getpid(), BB_VER, BB_BT);
 519 #endif
 520
 521
 522     /* Mount /proc */
 523     if (mount ("proc", "/proc", "proc", 0, 0) == 0) {
 524         message(CONSOLE|LOG, "Mounting /proc: done.\n");
 525         kernel_version = get_kernel_revision();
 526     } else
 527         message(CONSOLE|LOG, "Mounting /proc: failed!\n");
 528
 529     /* Make sure there is enough memory to do something useful. */
 530     check_memory();
 531
 532
 533     /* Make sure an init script exists before trying to run it */
 534     if (run_rc == TRUE && stat(INITSCRIPT, &statbuf)==0) {
 535         wait_for_enter = FALSE;
 536         tty0_command = rc_script_command;
 537     }
 538
 539
 540     /* Ok, now launch the rc script and/or prepare to
 541      * start up some VTs if somebody hits enter...
 542      */
 543     for (;;) {
 544         pid_t wpid;
 545         int status;
 546
 547         if (pid1 == 0 && tty0_command) {
 548             pid1 = run(tty0_command, console, wait_for_enter);
 549         }
 550         if (pid2 == 0 && tty1_command && second_console) {
 551             pid2 = run(tty1_command, second_console, TRUE);
 552         }
 553         wpid = wait(&status);
 554         if (wpid > 0 ) {
 555             message(LOG, "pid %d exited, status=%x.\n", wpid, status);
 556         }
 557         /* Don't respawn init script if it exits */
 558         if (wpid == pid1) {
 559             if (run_rc == FALSE) {
 560                 pid1 = 0;
 561             }
 562 #ifdef BB_INIT_CMD_IF_RC_SCRIPT_EXITS
 563             else {
 564                 pid1 = 0;
 565                 run_rc=FALSE;
 566                 wait_for_enter=TRUE;
 567                 tty0_command=rc_exit_command;
 568             }
 569 #endif
 570         }
 571         if (wpid == pid2) {
 572             pid2 = 0;
 573         }
 574         sleep(1);
 575     }
 576 }