damnit.
[oweals/busybox.git] / init.c
diff --git a/init.c b/init.c
index 707b1916fa2dd759bc9ef3038e3669ff1d3130b4..3c800b9b2ad8bc7acdd87d1e09865e2affc24903 100644 (file)
--- a/init.c
+++ b/init.c
@@ -29,6 +29,7 @@
 #include <errno.h>
 #include <signal.h>
 #include <termios.h>
+#include <paths.h>
 #include <sys/types.h>
 #include <sys/fcntl.h>
 #include <sys/wait.h>
 #include <linux/serial.h>      /* for serial_struct */
 #include <sys/vt.h>            /* for vt_stat */
 #include <sys/ioctl.h>
+#ifdef BB_SYSLOGD
+#include <sys/syslog.h>
+#endif
+
+#if ! defined BB_FEATURE_USE_PROCFS
+#error Sorry, I depend on the /proc filesystem right now.
+#endif
+
 
-#define VT_CONSOLE      "/dev/console" /* Logical system console */
 #define VT_PRIMARY      "/dev/tty1"    /* Primary virtual console */
 #define VT_SECONDARY    "/dev/tty2"    /* Virtual console */
 #define VT_LOG          "/dev/tty3"    /* Virtual console */
 #define SERIAL_CON0     "/dev/ttyS0"    /* Primary serial console */
 #define SERIAL_CON1     "/dev/ttyS1"    /* Serial console */
+#define GETTY           "/sbin/getty"  /* Default location of getty */
 #define SHELL           "/bin/sh"      /* Default shell */
 #define INITSCRIPT      "/etc/init.d/rcS"      /* Initscript. */
-#define PATH_DEFAULT    "PATH=/usr/local/sbin:/sbin:/bin:/usr/sbin:/usr/bin"
 
 #define LOG             0x1
 #define CONSOLE         0x2
-static char *console = VT_CONSOLE;
+static char *console = _PATH_CONSOLE;
 static char *second_console = VT_SECONDARY;
 static char *log = VT_LOG;
 static int kernel_version = 0;
@@ -83,8 +91,20 @@ int device_open(char *device, int mode)
 void message(int device, char *fmt, ...)
 {
     int fd;
-    static int log_fd=-1;
     va_list arguments;
+#ifdef BB_SYSLOGD
+
+    /* Log the message to syslogd */
+    if (device & LOG ) {
+       char msg[1024];
+       va_start(arguments, fmt);
+       vsnprintf(msg, sizeof(msg), fmt, arguments);
+       va_end(arguments);
+       syslog(LOG_DAEMON|LOG_NOTICE, msg);
+    }
+
+#else
+    static int log_fd=-1;
 
     /* Take full control of the log tty, and never close it.
      * It's mine, all mine!  Muhahahaha! */
@@ -102,14 +122,16 @@ void message(int device, char *fmt, ...)
            return;
        }
     }
-
     if ( (device & LOG) && (log_fd >= 0) ) {
        va_start(arguments, fmt);
        vdprintf(log_fd, fmt, arguments);
        va_end(arguments);
     }
+#endif
+
     if (device & CONSOLE) {
-       if ((fd = device_open(console, O_WRONLY|O_NOCTTY|O_NDELAY)) >= 0) {
+       /* Always send console messages to /dev/console so people will see them. */
+       if ((fd = device_open(_PATH_CONSOLE, O_WRONLY|O_NOCTTY|O_NDELAY)) >= 0) {
            va_start(arguments, fmt);
            vdprintf(fd, fmt, arguments);
            va_end(arguments);
@@ -130,31 +152,31 @@ void set_term( int fd)
 {
     struct termios tty;
     static const char control_characters[] = {
-       '\003', '\034', '\177', '\030', '\004', '\0',
+       '\003', '\034', '\177', '\025', '\004', '\0',
        '\1', '\0', '\021', '\023', '\032', '\0', '\022',
        '\017', '\027', '\026', '\0'
        };
 
     tcgetattr(fd, &tty);
 
-    /* Make it be sane */
-    tty.c_cflag &= CBAUD|CBAUDEX|CSIZE|CSTOPB|PARENB|PARODD;
-    tty.c_cflag |= HUPCL|CLOCAL;
-
-    /* input modes */
-    tty.c_iflag = IGNPAR|ICRNL|IXON|IXOFF|IXANY;
+    /* set control chars */
+    memcpy(tty.c_cc, control_characters, sizeof(control_characters));
 
     /* use line dicipline 0 */
     tty.c_line = 0;
 
+    /* Make it be sane */
+    //tty.c_cflag &= CBAUD|CBAUDEX|CSIZE|CSTOPB|PARENB|PARODD;
+    //tty.c_cflag |= HUPCL|CLOCAL;
+
+    /* input modes */
+    tty.c_iflag = ICRNL|IXON|IXOFF;
+
     /* output modes */
     tty.c_oflag = OPOST|ONLCR;
 
     /* local modes */
-    tty.c_lflag = ISIG|ICANON|ECHO|ECHOE|ECHOK|ECHOCTL|ECHOPRT|ECHOKE|IEXTEN;
-
-    /* control chars */
-    memcpy(tty.c_cc, control_characters, sizeof(control_characters));
+    tty.c_lflag = ISIG|ICANON|ECHO|ECHOE|ECHOK|ECHOCTL|ECHOKE|IEXTEN;
 
     tcsetattr(fd, TCSANOW, &tty);
 }
@@ -192,7 +214,7 @@ static void console_init()
     if ((s = getenv("CONSOLE")) != NULL) {
        console = s;
     }
-#if defined (__sparc__)
+#if #cpu(sparc)
     /* sparc kernel supports console=tty[ab] parameter which is also 
      * passed to init, so catch it here */
     else if ((s = getenv("console")) != NULL) {
@@ -209,7 +231,7 @@ static void console_init()
 
        console = the_console;
        /* 2.2 kernels: identify the real console backend and try to use it */
-       if (ioctl(0,TIOCGSERIAL,&sr) == 0) {
+       if (ioctl(0, TIOCGSERIAL, &sr) == 0) {
            /* this is a serial console */
            snprintf( the_console, sizeof the_console, "/dev/ttyS%d", sr.line );
        }
@@ -217,7 +239,7 @@ static void console_init()
            /* this is linux virtual tty */
            snprintf( the_console, sizeof the_console, "/dev/tty%d", vt.v_active );
        } else {
-           console = VT_CONSOLE;
+           console = _PATH_CONSOLE;
            tried_devcons++;
        }
     }
@@ -226,7 +248,7 @@ static void console_init()
        /* Can't open selected console -- try /dev/console */
        if (!tried_devcons) {
            tried_devcons++;
-           console = VT_CONSOLE;
+           console = _PATH_CONSOLE;
            continue;
        }
        /* Can't open selected console -- try vt1 */
@@ -244,6 +266,7 @@ static void console_init()
        /* check for serial console and disable logging to tty3 & running a
        * shell to tty2 */
        if (ioctl(0,TIOCGSERIAL,&sr) == 0) {
+           message(LOG|CONSOLE, "serial console detected.  Disabling 2nd virtual terminal.\r\n", console );
            log = NULL;
            second_console = NULL;
        }
@@ -313,9 +336,9 @@ static pid_t run(const char * const* command,
        }
 
        /* Log the process name and args */
-       message(LOG, "Starting pid %d, console %s: '", getpid(), terminal);
-       while ( *cmd) message(LOG, "%s ", *cmd++);
-       message(LOG, "'\r\n");
+       message(LOG|CONSOLE, "Starting pid %d, console %s: '", getpid(), terminal);
+       while ( *cmd) message(LOG|CONSOLE, "%s ", *cmd++);
+       message(LOG|CONSOLE, "'\r\n");
        
        /* Now run it.  The new program will take over this PID, 
         * so nothing further in init.c should be run. */
@@ -355,13 +378,13 @@ goodnight:
 static void shutdown_system(void)
 {
     const char* const swap_off_cmd[] = { "swapoff", "swapoff", "-a", 0};
-    const char* const umount_cmd[] = { "umount", "umount", "-a", "-n", 0};
+    const char* const umount_cmd[] = { "umount", "umount", "-a", 0};
 
 #ifndef DEBUG_INIT
     /* Allow Ctrl-Alt-Del to reboot system. */
     reboot(RB_ENABLE_CAD);
 #endif
-    message(CONSOLE, "The system is going down NOW !!\r\n");
+    message(CONSOLE, "\r\nThe system is going down NOW !!\r\n");
     sync();
     /* Send signals to every process _except_ pid 1 */
     message(CONSOLE, "Sending SIGHUP to all processes.\r\n");
@@ -375,7 +398,9 @@ static void shutdown_system(void)
     kill(-1, SIGKILL);
 #endif
     sleep(1);
+    message(CONSOLE, "Disabling swap.\r\n");
     waitfor(run( swap_off_cmd, console, FALSE));
+    message(CONSOLE, "Unmounting filesystems.\r\n");
     waitfor(run( umount_cmd, console, FALSE));
     sync();
     if (kernel_version > 0 && kernel_version <= 2 * 65536 + 2 * 256 + 11) {
@@ -393,8 +418,10 @@ static void halt_signal(int sig)
            "The system is halted. Press CTRL-ALT-DEL or turn off power\r\n");
     sync();
 #ifndef DEBUG_INIT
-    reboot(RB_HALT_SYSTEM);
-    //reboot(RB_POWER_OFF);
+    if (sig == SIGUSR2)
+       reboot(RB_POWER_OFF);
+    else
+       reboot(RB_HALT_SYSTEM);
 #endif
     exit(0);
 }
@@ -412,15 +439,24 @@ static void reboot_signal(int sig)
 
 extern int init_main(int argc, char **argv)
 {
-    int run_rc = TRUE;
-    int wait_for_enter = TRUE;
+    int run_rc = FALSE;
+    int single = FALSE;
+    int wait_for_enter_tty1 = TRUE;
+    int wait_for_enter_tty2 = TRUE;
     pid_t pid1 = 0;
     pid_t pid2 = 0;
     struct stat statbuf;
-    const char* const init_commands[] = { INITSCRIPT, INITSCRIPT, 0};
-    const char* const shell_commands[] = { SHELL, "-" SHELL, 0};
-    const char* const* tty0_commands = shell_commands;
-    const char* const* tty1_commands = shell_commands;
+    const char* const rc_script_command[] = { INITSCRIPT, INITSCRIPT, 0};
+    const char* const getty1_command[] = { GETTY, GETTY, VT_PRIMARY, 0};
+    const char* const getty2_command[] = { GETTY, GETTY, VT_SECONDARY, 0};
+    const char* const shell_command[] = { SHELL, "-" SHELL, 0};
+    const char* const* tty1_command = shell_command;
+    const char* const* tty2_command = shell_command;
+#ifdef BB_INIT_CMD_IF_RC_SCRIPT_EXITS
+    const char* const rc_exit_command[] = { "BB_INIT_CMD_IF_RC_SCRIPT_EXITS", 
+                                           "BB_INIT_CMD_IF_RC_SCRIPT_EXITS", 0 };
+#endif
+
 #ifdef DEBUG_INIT
     char *hello_msg_format =
        "init(%d) started:  BusyBox v%s (%s) multi-call binary\r\n";
@@ -430,13 +466,14 @@ extern int init_main(int argc, char **argv)
 #endif
 
     
-    /* Check if we are supposed to be in single user mode */
-    if ( argc > 1 && (!strcmp(argv[1], "single") || 
-               !strcmp(argv[1], "-s") || !strcmp(argv[1], "1"))) {
-       run_rc = FALSE;
+#ifndef DEBUG_INIT
+    /* Expect to be PID 1 iff we are run as init (not linuxrc) */
+    if (getpid() != 1 && strstr(argv[0], "init")!=NULL ) {
+       usage( "init\n\nInit is the parent of all processes.\n\n"
+               "This version of init is designed to be run only by the kernel\n");
     }
+#endif
 
-    
     /* Set up sig handlers  -- be sure to
      * clear all of these in run() */
     signal(SIGUSR1, halt_signal);
@@ -461,7 +498,7 @@ extern int init_main(int argc, char **argv)
     setsid();
 
     /* Make sure PATH is set to something sane */
-    putenv(PATH_DEFAULT);
+    putenv(_PATH_STDPATH);
 
    
     /* Hello world */
@@ -479,30 +516,63 @@ extern int init_main(int argc, char **argv)
     } else
        message(CONSOLE|LOG, "Mounting /proc: failed!\n");
 
+fprintf(stderr, "got proc\n");
+
     /* Make sure there is enough memory to do something useful. */
     check_memory();
+fprintf(stderr, "got check_memory\n");
 
+    /* Check if we are supposed to be in single user mode */
+    if ( argc > 1 && (!strcmp(argv[1], "single") || 
+               !strcmp(argv[1], "-s") || !strcmp(argv[1], "1"))) {
+       single = TRUE;
+       tty1_command = shell_command;
+       tty2_command = shell_command;
+    }
+fprintf(stderr, "got single\n");
 
     /* Make sure an init script exists before trying to run it */
-    if (run_rc == TRUE && stat(INITSCRIPT, &statbuf)==0) {
-       wait_for_enter = FALSE;
-       tty0_commands = init_commands;
+    if (single==FALSE && stat(INITSCRIPT, &statbuf)==0) {
+       run_rc = TRUE;
+       wait_for_enter_tty1 = FALSE;
+       tty1_command = rc_script_command;
     }
+    
+    /* Make sure /sbin/getty exists before trying to run it */
+    if (stat(GETTY, &statbuf)==0) {
+       char* where;
+fprintf(stderr, "\n");
+       wait_for_enter_tty2 = FALSE;
+       where = strrchr( console, '/');
+       if ( where != NULL) {
+           strcpy( (char*)getty2_command[2], where);
+       }
+       tty2_command = getty2_command;
+       /* Check on hooking a getty onto tty1 */
+       if (run_rc == FALSE && single==FALSE) {
+           wait_for_enter_tty1 = FALSE;
+           where = strrchr( second_console, '/');
+           if ( where != NULL) {
+               strcpy( (char*)getty1_command[2], where);
+           }
+           tty1_command = getty1_command;
+       }
+    }
+    
 
-
-    /* Ok, now launch the rc script and/or prepare to 
-     * start up some VTs if somebody hits enter... 
-     */
+    /* Ok, now launch the tty1_command and tty2_command */
     for (;;) {
        pid_t wpid;
        int status;
 
-       if (pid1 == 0 && tty0_commands) {
-           pid1 = run(tty0_commands, console, wait_for_enter);
+       if (pid1 == 0 && tty1_command) {
+           pid1 = run(tty1_command, console, wait_for_enter_tty1);
        }
-       if (pid2 == 0 && tty1_commands && second_console) {
-           pid2 = run(tty1_commands, second_console, TRUE);
+#ifdef BB_FEATURE_INIT_SECOND_CONSOLE
+       if (pid2 == 0 && tty2_command && second_console) {
+           pid2 = run(tty2_command, second_console, wait_for_enter_tty2);
        }
+#endif
        wpid = wait(&status);
        if (wpid > 0 ) {
            message(LOG, "pid %d exited, status=%x.\n", wpid, status);
@@ -511,19 +581,21 @@ extern int init_main(int argc, char **argv)
        if (wpid == pid1) {
            if (run_rc == FALSE) {
                pid1 = 0;
-           } 
-#if 0
-/* Turn this on to start a shell on the console if the init script exits.... */
+           }
+#ifdef BB_INIT_CMD_IF_RC_SCRIPT_EXITS
            else {
+               pid1 = 0;
                run_rc=FALSE;
-               wait_for_enter=TRUE;
-               tty0_commands=shell_commands;
+               wait_for_enter_tty1=TRUE;
+               tty1_command=rc_exit_command;
            }
 #endif
        }
+#ifdef BB_FEATURE_INIT_SECOND_CONSOLE
        if (wpid == pid2) {
            pid2 = 0;
        }
+#endif
        sleep(1);
     }
 }