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