work in progress...
[oweals/busybox.git] / init.c
diff --git a/init.c b/init.c
index 55c5c7318effa9821d11aedd76636cfa126fbb75..561b5fd527186feddd823c743badb27c3b56cd4a 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>
+#include <linux/version.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
+
+#ifndef KERNEL_VERSION
+#define KERNEL_VERSION(a,b,c) (((a) << 16) + ((b) << 8) + (c))
+#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"
+#ifndef BB_INIT_SCRIPT
+#define BB_INIT_SCRIPT "/etc/init.d/rcS"       /* Initscript. */
+#endif
 
 #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;
 
 
 /* try to open up the specified device */
@@ -83,27 +98,47 @@ 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! */
-    if (log_fd==-1) {
-       if ((log_fd = device_open(log, O_RDWR|O_NDELAY)) < 0) {
+    if (log_fd < 0) {
+       if (log == NULL) {
+       /* don't even try to log, because there is no such console */
+       log_fd = -2;
+       /* log to main console instead */
+       device = CONSOLE;
+    }
+    else if ((log_fd = device_open(log, O_RDWR|O_NDELAY)) < 0) {
            log_fd=-1;
            fprintf(stderr, "Bummer, can't write to log on %s!\r\n", log);
            fflush(stderr);
            return;
        }
     }
-
-    if ( (device & LOG) && (log_fd != -1) ) {
+    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);
@@ -123,81 +158,36 @@ void message(int device, char *fmt, ...)
 void set_term( int fd)
 {
     struct termios tty;
-#if 0
     static const char control_characters[] = {
        '\003', '\034', '\177', '\025', '\004', '\0',
        '\1', '\0', '\021', '\023', '\032', '\0', '\022',
        '\017', '\027', '\026', '\0'
        };
-#endif
-    static const char control_characters[] = {
-       '\003', '\034', '\177', '\030', '\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);
 }
 
-/* Set terminal settings to reasonable defaults */
-void set_term_old( int fd)
-{
-    struct termios tty;
-
-    ioctl(fd, TCGETA, &tty);
-
-    tty.c_cflag &= CBAUD|CBAUDEX|CSIZE|CSTOPB|PARENB|PARODD;
-    tty.c_cflag |= HUPCL|CLOCAL;
-
-    tty.c_cc[VINTR]    = 3;
-    tty.c_cc[VQUIT]    = 28;
-    tty.c_cc[VERASE]   = 127;
-    //tty.c_cc[VKILL]    = 21;
-    tty.c_cc[VKILL]    = 24;
-    tty.c_cc[VEOF]     = 4;
-    tty.c_cc[VTIME]    = 0;
-    tty.c_cc[VMIN]     = 1;
-    tty.c_cc[VSWTC]    = 0;
-    tty.c_cc[VSTART]   = 17;
-    tty.c_cc[VSTOP]    = 19;
-    tty.c_cc[VSUSP]    = 26;
-    tty.c_cc[VEOL]     = 0;
-    tty.c_cc[VREPRINT] = 18;
-    tty.c_cc[VDISCARD] = 15;
-    tty.c_cc[VWERASE]  = 23;
-    tty.c_cc[VLNEXT]   = 22;
-    tty.c_cc[VEOL2]    = 0;
-    
-
-    tty.c_line  = 0;
-    tty.c_iflag = IGNPAR|ICRNL|IXON|IXOFF|IXANY;
-    tty.c_oflag = OPOST|ONLCR;
-    tty.c_lflag = ISIG|ICANON|ECHO|ECHOE|ECHOK|ECHOCTL|ECHOPRT|ECHOKE|IEXTEN;
-
-    ioctl(fd, TCSETA, &tty);
-}
-
 /* How much memory does this machine have? */
 static int mem_total()
 {
@@ -225,32 +215,47 @@ static void console_init()
     int fd;
     int tried_devcons = 0;
     int tried_vtprimary = 0;
+    struct serial_struct sr;
     char *s;
 
     if ((s = getenv("CONSOLE")) != NULL) {
        console = s;
-/* Apparently the sparc does wierd things... */
-#if defined (__sparc__)
-       if (strncmp( s, "/dev/tty", 8 )==0) {
-           switch( s[8]) {
-               case 'a':
-                   s=SERIAL_CON0;
-                   break;
-               case 'b':
-                   s=SERIAL_CON1;
-           }
-       }
+    }
+#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) {
+       /* remap tty[ab] to /dev/ttyS[01] */
+       if (strcmp( s, "ttya" )==0)
+           console = SERIAL_CON0;
+       else if (strcmp( s, "ttyb" )==0)
+           console = SERIAL_CON1;
+    }
 #endif
-    } else {
-       console = VT_CONSOLE;
-       tried_devcons++;
+    else {
+       struct vt_stat vt;
+       static char the_console[13];
+
+       console = the_console;
+       /* 2.2 kernels: identify the real console backend and try to use it */
+       if (ioctl(0, TIOCGSERIAL, &sr) == 0) {
+           /* this is a serial console */
+           snprintf( the_console, sizeof the_console, "/dev/ttyS%d", sr.line );
+       }
+       else if (ioctl(0, VT_GETSTATE, &vt) == 0) {
+           /* this is linux virtual tty */
+           snprintf( the_console, sizeof the_console, "/dev/tty%d", vt.v_active );
+       } else {
+           console = _PATH_CONSOLE;
+           tried_devcons++;
+       }
     }
 
     while ((fd = open(console, O_RDONLY | O_NONBLOCK)) < 0) {
        /* 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 */
@@ -264,8 +269,16 @@ static void console_init()
     if (fd < 0)
        /* Perhaps we should panic here? */
        console = "/dev/null";
-    else
+    else {
+       /* 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;
+       }
        close(fd);
+    }
     message(LOG, "console=%s\n", console );
 }
 
@@ -330,9 +343,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. */
@@ -372,13 +385,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");
@@ -392,10 +405,17 @@ 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();
-    bdflush(1, 0);
+    if (kernel_version > 0 && kernel_version <= 2 * 65536 + 2 * 256 + 11) {
+       /* bdflush, kupdate not needed for kernels >2.2.11 */
+       message(CONSOLE, "Flushing buffers.\r\n");
+       bdflush(1, 0);
+       sync();
+    }
 }
 
 static void halt_signal(int sig)
@@ -405,8 +425,12 @@ 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 LINUX_VERSION_CODE >= KERNEL_VERSION(2,2,0)
+    if (sig == SIGUSR2)
+       reboot(RB_POWER_OFF);
+    else
+#endif
+       reboot(RB_HALT_SYSTEM);
 #endif
     exit(0);
 }
@@ -424,15 +448,26 @@ 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;
+    char which_vt1[30];
+    char which_vt2[30];
+    const char* const rc_script_command[] = { BB_INIT_SCRIPT, BB_INIT_SCRIPT, 0};
+    const char* const getty1_command[] = { GETTY, GETTY, "38400", which_vt1, 0};
+    const char* const getty2_command[] = { GETTY, GETTY, "38400", which_vt2, 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";
@@ -442,13 +477,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);
@@ -473,7 +509,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 */
@@ -485,52 +521,91 @@ extern int init_main(int argc, char **argv)
 
     
     /* Mount /proc */
-    if (mount ("proc", "/proc", "proc", 0, 0) == 0)
+    if (mount ("proc", "/proc", "proc", 0, 0) == 0) {
        message(CONSOLE|LOG, "Mounting /proc: done.\n");
-    else
+       kernel_version = get_kernel_revision();
+    } else
        message(CONSOLE|LOG, "Mounting /proc: failed!\n");
 
     /* Make sure there is enough memory to do something useful. */
     check_memory();
 
+    /* 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;
+    }
 
     /* 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(BB_INIT_SCRIPT, &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;
+       /* First do tty2 */
+       wait_for_enter_tty2 = FALSE;
+       where = strrchr( second_console, '/');
+       if ( where != NULL) {
+           where++;
+           strncpy( which_vt2, where, sizeof(which_vt2));
+       }
+       tty2_command = getty2_command;
+
+       /* Check on hooking a getty onto tty1 */
+       if (run_rc == FALSE && single==FALSE) {
+           wait_for_enter_tty1 = FALSE;
+           where = strrchr( console, '/');
+           if ( where != NULL) {
+               where++;
+               strncpy( which_vt1, where, sizeof(which_vt1));
+           }
+           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) {
-           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);
        }
+       /* Don't respawn init script if it exits */
        if (wpid == pid1) {
-           pid1 = 0;
-           if (run_rc == TRUE) {
-               /* Don't respawn init script if it exits,
-                * Start a shell instead. */
+           if (run_rc == FALSE) {
+               pid1 = 0;
+           }
+#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);
     }
 }