Making note of my changes
[oweals/busybox.git] / init.c
diff --git a/init.c b/init.c
index 116d0795446b05c548bd07f6a152ea62b5563438..b0a85829dacf51bde94bd2bf455c6d78ca51f98e 100644 (file)
--- a/init.c
+++ b/init.c
 #define VT_LOG          "/dev/tty3"      /* Virtual console */
 #define SERIAL_CON0     "/dev/ttyS0"      /* Primary serial console */
 #define SERIAL_CON1     "/dev/ttyS1"      /* Serial console */
-#define SHELL           "-sh"            /* Default shell */
+#define SHELL           "/bin/sh"        /* Default shell */
 #define INITTAB         "/etc/inittab"   /* inittab file location */
+#ifndef INIT_SCRIPT
 #define INIT_SCRIPT    "/etc/init.d/rcS" /* Default sysinit script. */
+#endif
 
 #define LOG             0x1
 #define CONSOLE         0x2
@@ -97,17 +99,18 @@ typedef struct initActionTag initAction;
 struct initActionTag {
     pid_t pid;
     char process[256];
-    char *console;
+    char console[256];
     initAction *nextPtr;
     initActionEnum action;
 };
 initAction* initActionList = NULL;
 
 
-static char *console = _PATH_CONSOLE;
-static char *second_console = VT_SECONDARY;
+static char *secondConsole = VT_SECONDARY;
 static char *log = VT_LOG;
-static int kernel_version = 0;
+static int kernelVersion = 0;
+static char termType[32] = "TERM=ansi";
+static char console[32] = _PATH_CONSOLE;
 
 
 /* try to open up the specified device */
@@ -144,7 +147,9 @@ void message(int device, char *fmt, ...)
        va_start(arguments, fmt);
        vsnprintf(msg, sizeof(msg), fmt, arguments);
        va_end(arguments);
+       openlog( "init", 0, LOG_DAEMON);
        syslog(LOG_DAEMON|LOG_NOTICE, msg);
+       closelog();
     }
 
 #else
@@ -255,8 +260,12 @@ static void console_init()
     struct serial_struct sr;
     char *s;
 
+    if ((s = getenv("TERM")) != NULL) {
+       snprintf(termType,sizeof(termType)-1,"TERM=%s",s);
+    }
+
     if ((s = getenv("CONSOLE")) != NULL) {
-       console = s;
+       snprintf(console, sizeof(console)-1, "%s",s);
     }
 #if #cpu(sparc)
     /* sparc kernel supports console=tty[ab] parameter which is also 
@@ -264,26 +273,24 @@ static void console_init()
     else if ((s = getenv("console")) != NULL) {
        /* remap tty[ab] to /dev/ttyS[01] */
        if (strcmp( s, "ttya" )==0)
-           console = SERIAL_CON0;
+           snprintf(console, sizeof(console)-1, "%s", SERIAL_CON0);
        else if (strcmp( s, "ttyb" )==0)
-           console = SERIAL_CON1;
+           snprintf(console, sizeof(console)-1, "%s", SERIAL_CON1);
     }
 #endif
     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 );
+           snprintf(console, sizeof(console)-1, "/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 );
+           snprintf(console, sizeof(console)-1, "/dev/tty%d", vt.v_active);
        } else {
-           console = _PATH_CONSOLE;
+           snprintf(console, sizeof(console)-1, "%s", _PATH_CONSOLE);
            tried_devcons++;
        }
     }
@@ -292,58 +299,51 @@ static void console_init()
        /* Can't open selected console -- try /dev/console */
        if (!tried_devcons) {
            tried_devcons++;
-           console = _PATH_CONSOLE;
+           snprintf(console, sizeof(console)-1, "%s", _PATH_CONSOLE);
            continue;
        }
        /* Can't open selected console -- try vt1 */
        if (!tried_vtprimary) {
            tried_vtprimary++;
-           console = VT_PRIMARY;
+           snprintf(console, sizeof(console)-1, "%s", VT_PRIMARY);
            continue;
        }
        break;
     }
-    if (fd < 0)
+    if (fd < 0) {
        /* Perhaps we should panic here? */
-       console = "/dev/null";
-    else {
+       snprintf(console, sizeof(console)-1, "/dev/null");
+    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 );
+           message(LOG|CONSOLE, "serial console detected.  Disabling virtual terminals.\r\n" );
            log = NULL;
-           second_console = NULL;
+           secondConsole = NULL;
        }
        close(fd);
     }
     message(LOG, "console=%s\n", console );
 }
 
-static int waitfor(int pid)
-{
-    int status, wpid;
-
-    while (1) {
-       wpid = wait(&status);
-       if (wpid > 0 ) {
-           message(LOG, "pid %d exited, status=0x%x.\n", wpid, status);
-           break;
-       }
-       if (wpid == pid )
-           break;
-    }
-    return wpid;
-}
-
 static pid_t run(char* command, 
        char *terminal, int get_enter)
 {
-    int i;
+    int i, fd;
     pid_t pid;
     char* tmpCmd;
     char* cmd[255];
     static const char press_enter[] =
        "\nPlease press Enter to activate this console. ";
+    char* environment[] = {
+       "HOME=/",
+       "PATH=/usr/bin:/bin:/usr/sbin:/sbin",
+       "SHELL=/bin/sh",
+       termType,
+       "USER=root",
+       0
+    };
+
 
     if ((pid = fork()) == 0) {
        pid_t shell_pgid = getpid ();
@@ -354,24 +354,20 @@ static pid_t run(char* command,
        close(2);
        setsid();
 
-       if (device_open(terminal, O_RDWR) < 0) {
-           message(LOG|CONSOLE, "Bummer, can't open %s\r\n", terminal);
-           exit(1);
-       }
-       dup(0);
-       dup(0);
-       /* Grab control of the terminal.  */
-       if (tcsetpgrp (0, getpgrp()) < 0) {
-           message(LOG|CONSOLE, "tcsetpgrp error: %s\r\n", strerror(errno));
-       }
-       set_term(0);
-
        /* Reset signal handlers set for parent process */
        signal(SIGUSR1, SIG_DFL);
        signal(SIGUSR2, SIG_DFL);
        signal(SIGINT, SIG_DFL);
        signal(SIGTERM, SIG_DFL);
 
+       if ((fd = device_open(terminal, O_RDWR)) < 0) {
+           message(LOG|CONSOLE, "Bummer, can't open %s\r\n", terminal);
+           exit(1);
+       }
+       dup(fd);
+       dup(fd);
+       tcsetpgrp (0, getpgrp());
+       set_term(0);
 
        if (get_enter==TRUE) {
            /*
@@ -389,23 +385,25 @@ static pid_t run(char* command,
            read(fileno(stdin), &c, 1);
        }
 
+       /* Log the process name and args */
+       message(LOG, "Starting pid %d, console %s: '", 
+               shell_pgid, terminal, command);
+
        /* Convert command (char*) into cmd (char**, one word per string) */
        for (tmpCmd=command, i=0; (tmpCmd=strsep(&command, " \t")) != NULL;) {
            if (*tmpCmd != '\0') {
                cmd[i] = tmpCmd;
+               message(LOG, "%s ", tmpCmd);
                tmpCmd++;
                i++;
            }
        }
        cmd[i] = NULL;
-
-       /* Log the process name and args */
-       message(LOG, "Starting pid %d, console %s: '%s'\r\n", 
-               shell_pgid, terminal, cmd[0]);
+       message(LOG, "'\r\n");
 
        /* Now run it.  The new program will take over this PID, 
         * so nothing further in init.c should be run. */
-       execvp(cmd[0], cmd);
+       execve(cmd[0], cmd, environment);
 
        /* We're still here?  Some error happened. */
        message(LOG|CONSOLE, "Bummer, could not run '%s': %s\n", cmd[0],
@@ -415,18 +413,37 @@ static pid_t run(char* command,
     return pid;
 }
 
+static int waitfor(char* command, 
+       char *terminal, int get_enter)
+{
+    int status, wpid;
+    int pid = run( command, terminal, get_enter);
+
+    while (1) {
+       wpid = wait(&status);
+       if (wpid > 0 ) {
+           message(LOG, "Process '%s' (pid %d) exited.\n", 
+                           command, wpid);
+           break;
+       }
+       if (wpid == pid )
+           break;
+    }
+    return wpid;
+}
+
 /* Make sure there is enough memory to do something useful. *
  * Calls swapon if needed so be sure /proc is mounted. */
 static void check_memory()
 {
-    struct stat statbuf;
+    struct stat statBuf;
 
     if (mem_total() > 3500)
        return;
 
-    if (stat("/etc/fstab", &statbuf) == 0) {
+    if (stat("/etc/fstab", &statBuf) == 0) {
        /* Try to turn on swap */
-       waitfor(run("/bin/swapon swapon -a", log, FALSE));
+       waitfor("/bin/swapon swapon -a", log, FALSE);
        if (mem_total() < 3500)
            goto goodnight;
     } else
@@ -457,13 +474,12 @@ static void shutdown_system(void)
     sleep(1);
 
     message(CONSOLE, "Disabling swap.\r\n");
-    waitfor(run( "swapoff -a", console, FALSE));
+    waitfor( "swapoff -a", console, FALSE);
     message(CONSOLE, "Unmounting filesystems.\r\n");
-    waitfor(run( "umount -a", console, FALSE));
+    waitfor("umount -a", console, FALSE);
     sync();
-    if (kernel_version > 0 && kernel_version <= 2 * 65536 + 2 * 256 + 11) {
+    if (kernelVersion > 0 && kernelVersion <= 2 * 65536 + 2 * 256 + 11) {
        /* bdflush, kupdate not needed for kernels >2.2.11 */
-       message(CONSOLE, "Flushing buffers.\r\n");
        bdflush(1, 0);
        sync();
     }
@@ -493,23 +509,112 @@ static void reboot_signal(int sig)
     exit(0);
 }
 
-#endif
+#if defined BB_FEATURE_INIT_CHROOT
+static void check_chroot(int sig)
+{
+    char *argv_init[2] = { "init", NULL, };
+    char *envp_init[3] = { "HOME=/", "TERM=linux", NULL, };
+    char rootpath[256], *tc;
+    int fd;
+
+    if ((fd = open("/proc/sys/kernel/init-chroot", O_RDONLY)) == -1) {
+       message(CONSOLE, "SIGHUP recived, but could not open proc file\r\n");
+       sleep(2);
+       return;
+    }
+    if (read(fd, rootpath, sizeof(rootpath)) == -1) {
+       message(CONSOLE, "SIGHUP recived, but could not read proc file\r\n");
+       sleep(2);
+       return;
+    }
+    close(fd);
+
+    if (rootpath[0] == '\0') {
+       message(CONSOLE, "SIGHUP recived, but new root is not valid: %s\r\n",
+               rootpath);
+       sleep(2);
+       return;
+    }
+
+    tc = strrchr(rootpath, '\n');
+    *tc = '\0';
+    
+    /* Ok, making it this far means we commit */
+    message(CONSOLE, "Please stand by, changing root to `%s'.\r\n", rootpath);
+
+    /* kill all other programs first */
+    message(CONSOLE, "Sending SIGTERM to all processes.\r\n");
+    kill(-1, SIGTERM);
+    sleep(2);
+    sync();
+
+    message(CONSOLE, "Sending SIGKILL to all processes.\r\n");
+    kill(-1, SIGKILL);
+    sleep(2);
+    sync();
+
+    /* ok, we don't need /proc anymore. we also assume that the signaling
+     * process left the rest of the filesystems alone for us */
+    umount("/proc");
+
+    /* Ok, now we chroot. Hopefully we only have two things mounted, the
+     * new chroot'd mount point, and the old "/" mount. s,
+     * we go ahead and unmount the old "/". This should trigger the kernel
+     * to set things up the Right Way(tm). */
+
+    if (!chroot(rootpath))
+       umount("/dev/root");
+
+    /* If the chroot fails, we are already too far to turn back, so we
+     * continue and hope that executing init below will revive the system */
+
+    /* close all of our descriptors and open new ones */
+    close(0);
+    close(1);
+    close(2);
+    open("/dev/console", O_RDWR, 0);
+    dup(0);
+    dup(0);
+
+    message(CONSOLE, "Executing real init...\r\n");
+    /* execute init in the (hopefully) new root */
+    execve("/sbin/init",argv_init,envp_init);
+
+    message(CONSOLE, "ERROR: Could not exec new init. Hit ctrl+alt+delete to reboot.\r\n");
+    return;
+} 
+#endif /* BB_FEATURE_INIT_CHROOT */
 
-void new_initAction (const struct initActionType *a, 
-       char* process, char* console)
+#endif /* ! DEBUG_INIT */
+
+void new_initAction (initActionEnum action,
+       char* process, char* cons)
 {
     initAction* newAction;
+
+    /* If BusyBox detects that a serial console is in use, 
+     * then entries containing non-empty id fields will _not_ be run.
+     * The exception to this rule is the null device.
+     */
+    if (secondConsole == NULL && (*cons != '\0' || strncmp(cons, "null", 4)))
+       return;
+
     newAction = calloc ((size_t)(1), sizeof(initAction));
     if (!newAction) {
-       fprintf(stderr, "Memory allocation failure\n");
+       message(LOG|CONSOLE,"Memory allocation failure\n");
        while (1) sleep(1);
     }
     newAction->nextPtr = initActionList;
     initActionList = newAction;
     strncpy( newAction->process, process, 255);
-    newAction->action = a->action;
-    newAction->console = console;
+    newAction->action = action;
+    if (*cons != '\0') {
+       strncpy(newAction->console, cons, 255);
+    } else
+       strncpy(newAction->console, console, 255);
     newAction->pid = 0;
+//    message(LOG|CONSOLE, "process='%s' action='%d' console='%s'\n",
+//         newAction->process, newAction->action, newAction->console);
 }
 
 void delete_initAction (initAction *action)
@@ -524,11 +629,19 @@ void delete_initAction (initAction *action)
     }
 }
 
+/* NOTE that if BB_FEATURE_USE_INITTAB is NOT defined,
+ * then parse_inittab() simply adds in some default
+ * actions(i.e runs INIT_SCRIPT and then starts a pair 
+ * of "askfirst" shells).  If BB_FEATURE_USE_INITTAB 
+ * _is_ defined, but /etc/inittab is missing, this 
+ * results in the same set of default behaviors.
+ * */
 void parse_inittab(void) 
 {
+#ifdef BB_FEATURE_USE_INITTAB
     FILE* file;
-    char buf[256];
-    char *p, *q, *r;
+    char buf[256], lineAsRead[256], tmpConsole[256];
+    char *p, *q, *r, *s;
     const struct initActionType *a = actions;
     int foundIt;
 
@@ -536,16 +649,17 @@ void parse_inittab(void)
     file = fopen(INITTAB, "r");
     if (file == NULL) {
        /* No inittab file -- set up some default behavior */
-
+#endif
        /* Askfirst shell on tty1 */
-       new_initAction( &(actions[3]), SHELL, console );
+       new_initAction( ASKFIRST, SHELL, console );
        /* Askfirst shell on tty2 */
-       if (second_console != NULL) 
-           new_initAction( &(actions[3]), SHELL, second_console );
+       if (secondConsole != NULL) 
+           new_initAction( ASKFIRST, SHELL, secondConsole );
        /* sysinit */
-       new_initAction( &(actions[0]), INIT_SCRIPT, console );
+       new_initAction( SYSINIT, INIT_SCRIPT, console );
 
        return;
+#ifdef BB_FEATURE_USE_INITTAB
     }
 
     while ( fgets(buf, 255, file) != NULL) {
@@ -558,15 +672,22 @@ void parse_inittab(void)
        if (q != NULL)
            *q='\0';
 
-       /* Skip past the ID field and the runlevel 
-        * field (both are ignored) */
+       /* Keep a copy around for posterity's sake (and error msgs) */
+       strcpy(lineAsRead, buf);
+
+       /* Grab the ID field */
+       s=p;
        p = strchr( p, ':');
+       if ( p != NULL || *(p+1) != '\0' ) {
+           *p='\0';
+           ++p;
+       }
 
        /* Now peal off the process field from the end
         * of the string */
        q = strrchr( p, ':');
        if ( q == NULL || *(q+1) == '\0' ) {
-           fprintf(stderr, "Bad inittab entry: %s\n", buf);
+           message(LOG|CONSOLE,"Bad inittab entry: %s\n", lineAsRead);
            continue;
        } else {
            *q='\0';
@@ -576,7 +697,7 @@ void parse_inittab(void)
        /* Now peal off the action field */
        r = strrchr( p, ':');
        if ( r == NULL || *(r+1) == '\0') {
-           fprintf(stderr, "Bad inittab entry: %s\n", buf);
+           message(LOG|CONSOLE,"Bad inittab entry: %s\n", lineAsRead);
            continue;
        } else {
            ++r;
@@ -586,7 +707,17 @@ void parse_inittab(void)
        a = actions;
        while (a->name != 0) {
            if (strcmp(a->name, r) == 0) {
-               new_initAction( a, q, NULL);
+               if (*s != '\0') {
+                   struct stat statBuf;
+                   strcpy(tmpConsole, "/dev/");
+                   strncat(tmpConsole, s, 200);
+                   if (stat(tmpConsole, &statBuf) != 0) {
+                       message(LOG|CONSOLE, "device '%s' does not exist.  Did you read the directions?\n", tmpConsole);
+                       break;
+                   }
+                   s = tmpConsole;
+               }
+               new_initAction( a->action, q, s);
                foundIt=TRUE;
            }
            a++;
@@ -595,13 +726,13 @@ void parse_inittab(void)
            continue;
        else {
            /* Choke on an unknown action */
-           fprintf(stderr, "Bad inittab entry: %s\n", buf);
+           message(LOG|CONSOLE, "Bad inittab entry: %s\n", lineAsRead);
        }
     }
     return;
+#endif
 }
 
-
 extern int init_main(int argc, char **argv)
 {
     initAction *a;
@@ -609,26 +740,29 @@ extern int init_main(int argc, char **argv)
     int status;
 
 #ifndef DEBUG_INIT
-    /* Expect to be PID 1 iff we are run as init (not linuxrc) */
+    /* Expect to be PID 1 if 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");
     }
+    /* Fix up argv[0] to be certain we claim to be init */
+    strncpy(argv[0], "init", strlen(argv[0]));
 
-    /* from the controlling terminal */
-    setsid();
-
-    /* Set up sig handlers  -- be sure to clear all of these in run() */
+    /* Set up sig handlers  -- be sure to
+     * clear all of these in run() */
     signal(SIGUSR1, halt_signal);
     signal(SIGUSR2, reboot_signal);
     signal(SIGINT, reboot_signal);
     signal(SIGTERM, reboot_signal);
+#if defined BB_FEATURE_INIT_CHROOT
+    signal(SIGHUP, check_chroot);
+#endif
 
     /* Turn off rebooting via CTL-ALT-DEL -- we get a 
      * SIGINT on CAD so we can shut things down gracefully... */
     reboot(RB_DISABLE_CAD);
 #endif 
-    
+
     /* Figure out where the default console should be */
     console_init();
 
@@ -637,18 +771,18 @@ extern int init_main(int argc, char **argv)
     close(1);
     close(2);
     set_term(0);
+    setsid();
 
     /* Make sure PATH is set to something sane */
     putenv(_PATH_STDPATH);
 
-   
     /* Hello world */
 #ifndef DEBUG_INIT
-    message(CONSOLE|LOG
+    message(LOG|CONSOLE
            "init started:  BusyBox v%s (%s) multi-call binary\r\n", 
            BB_VER, BB_BT);
 #else
-    message(CONSOLE|LOG
+    message(LOG|CONSOLE
            "init(%d) started:  BusyBox v%s (%s) multi-call binary\r\n", 
            getpid(), BB_VER, BB_BT);
 #endif
@@ -656,10 +790,10 @@ extern int init_main(int argc, char **argv)
     
     /* Mount /proc */
     if (mount ("proc", "/proc", "proc", 0, 0) == 0) {
-       message(CONSOLE|LOG, "Mounting /proc: done.\n");
-       kernel_version = get_kernel_revision();
+       message(LOG|CONSOLE, "Mounting /proc: done.\n");
+       kernelVersion = get_kernel_revision();
     } else
-       message(CONSOLE|LOG, "Mounting /proc: failed!\n");
+       message(LOG|CONSOLE, "Mounting /proc: failed!\n");
 
     /* Make sure there is enough memory to do something useful. */
     check_memory();
@@ -669,12 +803,17 @@ extern int init_main(int argc, char **argv)
                !strcmp(argv[1], "-s") || !strcmp(argv[1], "1"))) 
     {
        /* Ask first then start a shell on tty2 */
-       if (second_console != NULL) 
-           new_initAction( &(actions[3]), SHELL, second_console);
+       if (secondConsole != NULL) 
+           new_initAction( ASKFIRST, SHELL, secondConsole);
        /* Ask first then start a shell on tty1 */
-       new_initAction( &(actions[3]), SHELL, console);
+       new_initAction( ASKFIRST, SHELL, console);
     } else {
        /* Not in single user mode -- see what inittab says */
+
+       /* NOTE that if BB_FEATURE_USE_INITTAB is NOT defined,
+        * then parse_inittab() simply adds in some default
+        * actions(i.e runs INIT_SCRIPT and then starts a pair 
+        * of "askfirst" shells */
        parse_inittab();
     }
 
@@ -683,7 +822,7 @@ extern int init_main(int argc, char **argv)
     /* First run the sysinit command */
     for( a=initActionList ; a; a=a->nextPtr) {
        if (a->action == SYSINIT) {
-           waitfor(run(a->process, console, FALSE));
+           waitfor(a->process, a->console, FALSE);
            /* Now remove the "sysinit" entry from the list */
            delete_initAction( a);
        }
@@ -691,7 +830,7 @@ extern int init_main(int argc, char **argv)
     /* Next run anything that wants to block */
     for( a=initActionList ; a; a=a->nextPtr) {
        if (a->action == WAIT) {
-           waitfor(run(a->process, console, FALSE));
+           waitfor(a->process, a->console, FALSE);
            /* Now remove the "wait" entry from the list */
            delete_initAction( a);
        }
@@ -699,11 +838,16 @@ extern int init_main(int argc, char **argv)
     /* Next run anything to be run only once */
     for( a=initActionList ; a; a=a->nextPtr) {
        if (a->action == ONCE) {
-           run(a->process, console, FALSE);
+           run(a->process, a->console, FALSE);
            /* Now remove the "once" entry from the list */
            delete_initAction( a);
        }
     }
+    /* If there is nothing else to do, stop */
+    if (initActionList == NULL) {
+       message(LOG|CONSOLE, "No more tasks for init -- sleeping forever.\n");
+       while (1) sleep(1);
+    }
 
     /* Now run the looping stuff for the rest of forever */
     while (1) {
@@ -714,11 +858,11 @@ extern int init_main(int argc, char **argv)
                switch(a->action) {
                    case RESPAWN:
                        /* run the respawn stuff */
-                       a->pid = run(a->process, console, FALSE);
+                       a->pid = run(a->process, a->console, FALSE);
                        break;
                    case ASKFIRST:
                        /* run the askfirst stuff */
-                       a->pid = run(a->process, console, TRUE);
+                       a->pid = run(a->process, a->console, TRUE);
                        break;
                    /* silence the compiler's incessant whining */
                    default: