work in progress...
[oweals/busybox.git] / init.c
diff --git a/init.c b/init.c
index ce2f237c70928ed895269059896f5f3d032d6c4d..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 DEV_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 = DEV_CONSOLE;
+static char *console = _PATH_CONSOLE;
 static char *second_console = VT_SECONDARY;
 static char *log = VT_LOG;
 static int kernel_version = 0;
@@ -83,8 +98,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,15 +129,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) {
        /* Always send console messages to /dev/console so people will see them. */
-       if ((fd = device_open(DEV_CONSOLE, O_WRONLY|O_NOCTTY|O_NDELAY)) >= 0) {
+       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,19 +158,11 @@ void message(int device, char *fmt, ...)
 void set_term( int fd)
 {
     struct termios tty;
-#if 0
-    static const char control_characters[] = {
-       '\003', '\034', '\177', '\030', '\004', '\0',
-       '\1', '\0', '\021', '\023', '\032', '\0', '\022',
-       '\017', '\027', '\026', '\0'
-       };
-#else  
     static const char control_characters[] = {
        '\003', '\034', '\177', '\025', '\004', '\0',
        '\1', '\0', '\021', '\023', '\032', '\0', '\022',
        '\017', '\027', '\026', '\0'
        };
-#endif
 
     tcgetattr(fd, &tty);
 
@@ -226,7 +246,7 @@ static void console_init()
            /* this is linux virtual tty */
            snprintf( the_console, sizeof the_console, "/dev/tty%d", vt.v_active );
        } else {
-           console = DEV_CONSOLE;
+           console = _PATH_CONSOLE;
            tried_devcons++;
        }
     }
@@ -235,7 +255,7 @@ static void console_init()
        /* Can't open selected console -- try /dev/console */
        if (!tried_devcons) {
            tried_devcons++;
-           console = DEV_CONSOLE;
+           console = _PATH_CONSOLE;
            continue;
        }
        /* Can't open selected console -- try vt1 */
@@ -323,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. */
@@ -365,7 +385,7 @@ 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. */
@@ -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,17 +448,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 rc_script_command[] = { INITSCRIPT, INITSCRIPT, 0};
+    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* tty0_command = shell_command;
     const char* const* tty1_command = shell_command;
-#ifdef BB_CONSOLE_CMD_IF_RC_SCRIPT_EXITS
-    const char* const rc_exit_command[] = BB_CONSOLE_CMD_IF_RC_SCRIPT_EXITS;
+    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
@@ -447,19 +478,13 @@ extern int init_main(int argc, char **argv)
 
     
 #ifndef DEBUG_INIT
-    if (getpid() != 1) {
+    /* 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
 
-    /* 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;
-    }
-
-    
     /* Set up sig handlers  -- be sure to
      * clear all of these in run() */
     signal(SIGUSR1, halt_signal);
@@ -484,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 */
@@ -505,27 +530,59 @@ extern int init_main(int argc, char **argv)
     /* 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_command = rc_script_command;
+    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_command) {
-           pid1 = run(tty0_command, console, wait_for_enter);
+       if (pid1 == 0 && tty1_command) {
+           pid1 = run(tty1_command, console, wait_for_enter_tty1);
        }
-       if (pid2 == 0 && tty1_command && second_console) {
-           pid2 = run(tty1_command, 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);
@@ -535,18 +592,20 @@ extern int init_main(int argc, char **argv)
            if (run_rc == FALSE) {
                pid1 = 0;
            }
-#ifdef BB_CONSOLE_CMD_IF_RC_SCRIPT_EXITS
+#ifdef BB_INIT_CMD_IF_RC_SCRIPT_EXITS
            else {
                pid1 = 0;
                run_rc=FALSE;
-               wait_for_enter=TRUE;
-               tty0_command=rc_exit_command;
+               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);
     }
 }