Oops. Forgot the usleep.c file.
[oweals/busybox.git] / init.c
diff --git a/init.c b/init.c
index 899dca48c4e2a9173c83832314141daffd42494d..d61d1c38eb214aa3dc00dde510447cedd820c281 100644 (file)
--- a/init.c
+++ b/init.c
  *
  */
 
+/* Turn this on to disable all the dangerous 
+   rebooting stuff when debugging.
+#define DEBUG_INIT
+*/
+
 #include "internal.h"
-#include <stdio.h>
-#include <string.h>
-#include <stdlib.h>
-#include <stdarg.h>
-#include <unistd.h>
+#include <asm/types.h>
 #include <errno.h>
-#include <signal.h>
-#include <termios.h>
+#include <linux/serial.h>              /* for serial_struct */
+#include <linux/version.h>
 #include <paths.h>
-#include <sys/types.h>
-#include <sys/fcntl.h>
-#include <sys/wait.h>
+#include <signal.h>
+#include <stdarg.h>
+#include <stdio.h>
+#include <stdlib.h>
 #include <string.h>
+#include <sys/fcntl.h>
+#include <sys/ioctl.h>
+#include <sys/kdaemon.h>
 #include <sys/mount.h>
 #include <sys/reboot.h>
-#include <sys/kdaemon.h>
-#include <sys/sysmacros.h>
-#include <asm/types.h>
-#include <linux/serial.h>              /* for serial_struct */
-#include <sys/vt.h>                            /* for vt_stat */
-#include <sys/ioctl.h>
-#include <linux/version.h>
+#include <sys/sysinfo.h>               /* For check_free_memory() */
 #ifdef BB_SYSLOGD
-#include <sys/syslog.h>
+# include <sys/syslog.h>
 #endif
+#include <sys/sysmacros.h>
+#include <sys/types.h>
+#include <sys/vt.h>                            /* for vt_stat */
+#include <sys/wait.h>
+#include <termios.h>
+#include <unistd.h>
 
-#if ! defined BB_FEATURE_USE_PROCFS
-#error Sorry, I depend on the /proc filesystem right now.
+
+#if defined BB_FEATURE_INIT_COREDUMPS
+/*
+ * When a file named CORE_ENABLE_FLAG_FILE exists, setrlimit is called 
+ * before processes are spawned to set core file size as unlimited.
+ * This is for debugging only.  Don't use this is production, unless
+ * you want core dumps lying about....
+ */
+#define CORE_ENABLE_FLAG_FILE "/.init_enable_core"
+#include <sys/resource.h>
+#include <sys/time.h>
 #endif
 
 #ifndef KERNEL_VERSION
 #endif
 
 
-#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 SHELL           "/bin/sh"      /* Default shell */
-#define INITTAB         "/etc/inittab" /* inittab file location */
+#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 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. */
+#define INIT_SCRIPT  "/etc/init.d/rcS"   /* Default sysinit script. */
 #endif
 
-#define LOG             0x1
-#define CONSOLE         0x2
+#define LOG     0x1
+#define CONSOLE 0x2
 
 /* Allowed init action types */
 typedef enum {
@@ -78,10 +92,11 @@ typedef enum {
        RESPAWN,
        ASKFIRST,
        WAIT,
-       ONCE
+       ONCE,
+       CTRLALTDEL
 } initActionEnum;
 
-/* And now a list of the actions we support in the version of init */
+/* A mapping between "inittab" action name strings and action type codes. */
 typedef struct initActionType {
        const char *name;
        initActionEnum action;
@@ -93,10 +108,11 @@ static const struct initActionType actions[] = {
        {"askfirst", ASKFIRST},
        {"wait", WAIT},
        {"once", ONCE},
+       {"ctrlaltdel", CTRLALTDEL},
        {0}
 };
 
-/* Set up a linked list of initactions, to be read from inittab */
+/* Set up a linked list of initActions, to be read from inittab */
 typedef struct initActionTag initAction;
 struct initActionTag {
        pid_t pid;
@@ -109,15 +125,17 @@ initAction *initActionList = NULL;
 
 
 static char *secondConsole = VT_SECONDARY;
-static char *log = VT_LOG;
-static int kernelVersion = 0;
-static char termType[32] = "TERM=ansi";
-static char console[32] = _PATH_CONSOLE;
+static char *log           = VT_LOG;
+static int  kernelVersion  = 0;
+static char termType[32]   = "TERM=linux";
+static char console[32]    = _PATH_CONSOLE;
+
+static void delete_initAction(initAction * action);
 
 
-/* print a message to the specified device:
- * device may be bitwise-or'd from LOG | CONSOLE */
-void message(int device, char *fmt, ...)
+/* Print a message to the specified device.
+ * Device may be bitwise-or'd from LOG | CONSOLE */
+static void message(int device, char *fmt, ...)
 {
        va_list arguments;
        int fd;
@@ -131,8 +149,8 @@ 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);
+               openlog("init", 0, LOG_USER);
+               syslog(LOG_USER|LOG_INFO, msg);
                closelog();
        }
 #else
@@ -146,11 +164,12 @@ void message(int device, char *fmt, ...)
                        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;
+               } else if ((log_fd = device_open(log, O_RDWR|O_NDELAY)) < 0) {
+                       log_fd = -2;
                        fprintf(stderr, "Bummer, can't write to log on %s!\r\n", log);
                        fflush(stderr);
-                       return;
+                       log = NULL;
+                       device = CONSOLE;
                }
        }
        if ((device & LOG) && (log_fd >= 0)) {
@@ -185,23 +204,25 @@ void message(int device, char *fmt, ...)
 void set_term(int fd)
 {
        struct termios tty;
-       static const char control_characters[] = {
-               '\003', '\034', '\177', '\025', '\004', '\0',
-               '\1', '\0', '\021', '\023', '\032', '\0', '\022',
-               '\017', '\027', '\026', '\0'
-       };
 
        tcgetattr(fd, &tty);
 
        /* set control chars */
-       memcpy(tty.c_cc, control_characters, sizeof(control_characters));
+       tty.c_cc[VINTR]  = 3;   /* C-c */
+       tty.c_cc[VQUIT]  = 28;  /* C-\ */
+       tty.c_cc[VERASE] = 127; /* C-? */
+       tty.c_cc[VKILL]  = 21;  /* C-u */
+       tty.c_cc[VEOF]   = 4;   /* C-d */
+       tty.c_cc[VSTART] = 17;  /* C-q */
+       tty.c_cc[VSTOP]  = 19;  /* C-s */
+       tty.c_cc[VSUSP]  = 26;  /* C-z */
 
        /* 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;
+       tty.c_cflag &= CBAUD|CBAUDEX|CSIZE|CSTOPB|PARENB|PARODD;
+       tty.c_cflag |= HUPCL|CLOCAL;
 
        /* input modes */
        tty.c_iflag = ICRNL | IXON | IXOFF;
@@ -217,25 +238,17 @@ void set_term(int fd)
 }
 
 /* How much memory does this machine have? */
-static int mem_total()
+static int check_free_memory()
 {
-       char s[80];
-       char *p = "/proc/meminfo";
-       FILE *f;
-       const char pattern[] = "MemTotal:";
+       struct sysinfo info;
 
-       if ((f = fopen(p, "r")) < 0) {
-               message(LOG, "Error opening %s: %s\n", p, strerror(errno));
+       sysinfo(&info);
+       if (sysinfo(&info) != 0) {
+               message(LOG, "Error checking free memory: %s\n", strerror(errno));
                return -1;
        }
-       while (NULL != fgets(s, 79, f)) {
-               p = strstr(s, pattern);
-               if (NULL != p) {
-                       fclose(f);
-                       return (atoi(p + strlen(pattern)));
-               }
-       }
-       return -1;
+
+       return((info.totalram+info.totalswap)/1024);
 }
 
 static void console_init()
@@ -243,6 +256,7 @@ static void console_init()
        int fd;
        int tried_devcons = 0;
        int tried_vtprimary = 0;
+       struct vt_stat vt;
        struct serial_struct sr;
        char *s;
 
@@ -265,8 +279,6 @@ static void console_init()
        }
 #endif
        else {
-               struct vt_stat vt;
-
                /* 2.2 kernels: identify the real console backend and try to use it */
                if (ioctl(0, TIOCGSERIAL, &sr) == 0) {
                        /* this is a serial console */
@@ -303,10 +315,14 @@ 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 virtual terminals.\r\n");
                        log = NULL;
                        secondConsole = NULL;
+                       /* Force the TERM setting to vt102 for serial console --
+                        * iff TERM is set to linux (the default) */
+                       if (strcmp( termType, "TERM=linux" ) == 0)
+                               snprintf(termType, sizeof(termType) - 1, "TERM=vt102");
+                       message(LOG | CONSOLE,
+                                       "serial console detected.  Disabling virtual terminals.\r\n");
                }
                close(fd);
        }
@@ -319,6 +335,7 @@ static pid_t run(char *command, char *terminal, int get_enter)
        pid_t pid;
        char *tmpCmd;
        char *cmd[255];
+       char buf[255];
        static const char press_enter[] =
 
                "\nPlease press Enter to activate this console. ";
@@ -333,8 +350,6 @@ static pid_t run(char *command, char *terminal, int get_enter)
 
 
        if ((pid = fork()) == 0) {
-               pid_t shell_pgid = getpid();
-
                /* Clean up */
                close(0);
                close(1);
@@ -368,31 +383,53 @@ static pid_t run(char *command, char *terminal, int get_enter)
                         * specifies.
                         */
                        char c;
-
+#ifdef DEBUG_INIT
+                       pid_t shell_pgid = getpid();
                        message(LOG, "Waiting for enter to start '%s' (pid %d, console %s)\r\n",
                                        command, shell_pgid, terminal);
+#endif
                        write(fileno(stdout), press_enter, sizeof(press_enter) - 1);
                        read(fileno(stdin), &c, 1);
                }
 
+#ifdef DEBUG_INIT
                /* Log the process name and args */
-               message(LOG, "Starting pid %d, console %s: '",
+               message(LOG, "Starting pid %d, console %s: '%s'\r\n",
                                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;
-#ifdef DEBUG_INIT
-                               message(LOG, "%s ", tmpCmd);
 #endif
-                               tmpCmd++;
-                               i++;
+
+               /* See if any special /bin/sh requiring characters are present */
+               if (strpbrk(command, "~`!$^&*()=|\\{}[];\"'<>?") != NULL) {
+                       cmd[0] = SHELL;
+                       cmd[1] = "-c";
+                       strcpy(buf, "exec ");
+                       strncat(buf, command, sizeof(buf) - strlen(buf) - 1);
+                       cmd[2] = buf;
+                       cmd[3] = NULL;
+               } else {
+                       /* 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;
+                                       tmpCmd++;
+                                       i++;
+                               }
                        }
+                       cmd[i] = NULL;
                }
-               cmd[i] = NULL;
-               message(LOG, "'\r\n");
+
+#if defined BB_FEATURE_INIT_COREDUMPS
+               {
+                       struct stat sb;
+                       if (stat (CORE_ENABLE_FLAG_FILE, &sb) == 0) {
+                               struct rlimit limit;
+                               limit.rlim_cur = RLIM_INFINITY;
+                               limit.rlim_max = RLIM_INFINITY;
+                               setrlimit(RLIMIT_CORE, &limit);
+                       }
+               }
+#endif
 
                /* Now run it.  The new program will take over this PID, 
                 * so nothing further in init.c should be run. */
@@ -413,9 +450,8 @@ static int waitfor(char *command, char *terminal, int get_enter)
 
        while (1) {
                wpid = wait(&status);
-               if (wpid > 0) {
-                       message(LOG, "Process '%s' (pid %d) exited.\n", command, wpid);
-                       break;
+               if (wpid > 0 && wpid != pid) {
+                       continue;
                }
                if (wpid == pid)
                        break;
@@ -424,18 +460,18 @@ static int waitfor(char *command, char *terminal, int get_enter)
 }
 
 /* Make sure there is enough memory to do something useful. *
- * Calls swapon if needed so be sure /proc is mounted. */
+ * Calls "swapon -a" if needed so be sure /etc/fstab is present... */
 static void check_memory()
 {
        struct stat statBuf;
 
-       if (mem_total() > 3500)
+       if (check_free_memory() > 1000)
                return;
 
        if (stat("/etc/fstab", &statBuf) == 0) {
                /* Try to turn on swap */
-               waitfor("/bin/swapon swapon -a", log, FALSE);
-               if (mem_total() < 3500)
+               system("/sbin/swapon -a");
+               if (check_free_memory() < 1000)
                        goto goodnight;
        } else
                goto goodnight;
@@ -448,31 +484,45 @@ static void check_memory()
                sleep(1);
 }
 
+/* Run all commands to be run right before halt/reboot */
+static void run_lastAction(void)
+{
+       initAction *a;
+       for (a = initActionList; a; a = a->nextPtr) {
+               if (a->action == CTRLALTDEL) {
+                       waitfor(a->process, a->console, FALSE);
+                       delete_initAction(a);
+               }
+       }
+}
+
+
 #ifndef DEBUG_INIT
 static void shutdown_system(void)
 {
+
        /* first disable our SIGHUP signal */
        signal(SIGHUP, SIG_DFL);
 
        /* Allow Ctrl-Alt-Del to reboot system. */
        reboot(RB_ENABLE_CAD);
-       message(CONSOLE, "\r\nThe system is going down NOW !!\r\n");
+
+       message(CONSOLE|LOG, "\r\nThe system is going down NOW !!\r\n");
        sync();
 
        /* Send signals to every process _except_ pid 1 */
-       message(CONSOLE, "Sending SIGTERM to all processes.\r\n");
+       message(CONSOLE|LOG, "Sending SIGTERM to all processes.\r\n");
        kill(-1, SIGTERM);
-       sleep(5);
+       sleep(1);
        sync();
 
-       message(CONSOLE, "Sending SIGKILL to all processes.\r\n");
+       message(CONSOLE|LOG, "Sending SIGKILL to all processes.\r\n");
        kill(-1, SIGKILL);
-       sleep(5);
+       sleep(1);
+
+       /* run everything to be run at "ctrlaltdel" */
+       run_lastAction();
 
-       message(CONSOLE, "Disabling swap.\r\n");
-       waitfor("swapoff -a", console, FALSE);
-       message(CONSOLE, "Unmounting filesystems.\r\n");
-       waitfor("umount -a -r", console, FALSE);
        sync();
        if (kernelVersion > 0 && kernelVersion <= 2 * 65536 + 2 * 256 + 11) {
                /* bdflush, kupdate not needed for kernels >2.2.11 */
@@ -484,14 +534,14 @@ static void shutdown_system(void)
 static void halt_signal(int sig)
 {
        shutdown_system();
-       message(CONSOLE,
+       message(CONSOLE|LOG,
                        "The system is halted. Press %s or turn off power\r\n",
                        (secondConsole == NULL) /* serial console */
                        ? "Reset" : "CTRL-ALT-DEL");
        sync();
 
        /* allow time for last message to reach serial console */
-       sleep(5);
+       sleep(2);
 
 #if LINUX_VERSION_CODE >= KERNEL_VERSION(2,2,0)
        if (sig == SIGUSR2)
@@ -505,7 +555,7 @@ static void halt_signal(int sig)
 static void reboot_signal(int sig)
 {
        shutdown_system();
-       message(CONSOLE, "Please stand by while rebooting the system.\r\n");
+       message(CONSOLE|LOG, "Please stand by while rebooting the system.\r\n");
        sync();
 
        /* allow time for last message to reach serial console */
@@ -516,6 +566,11 @@ static void reboot_signal(int sig)
 }
 
 #if defined BB_FEATURE_INIT_CHROOT
+
+#if ! defined BB_FEATURE_USE_PROCFS
+#error Sorry, I depend on the /proc filesystem right now.
+#endif
+
 static void check_chroot(int sig)
 {
        char *argv_init[2] = { "init", NULL, };
@@ -631,7 +686,7 @@ void new_initAction(initActionEnum action, char *process, char *cons)
 //      newAction->process, newAction->action, newAction->console);
 }
 
-void delete_initAction(initAction * action)
+static void delete_initAction(initAction * action)
 {
        initAction *a, *b = NULL;
 
@@ -669,6 +724,10 @@ void parse_inittab(void)
        if (file == NULL) {
                /* No inittab file -- set up some default behavior */
 #endif
+               /* Swapoff on halt/reboot */
+               new_initAction(CTRLALTDEL, "/sbin/swapoff -a > /dev/null 2>&1", console);
+               /* Umount all filesystems on halt/reboot */
+               new_initAction(CTRLALTDEL, "/bin/umount -a -r > /dev/null 2>&1", console);
                /* Askfirst shell on tty1 */
                new_initAction(ASKFIRST, SHELL, console);
                /* Askfirst shell on tty2 */
@@ -703,7 +762,7 @@ void parse_inittab(void)
                        ++p;
                }
 
-               /* Now peal off the process field from the end
+               /* Now peel off the process field from the end
                 * of the string */
                q = strrchr(p, ':');
                if (q == NULL || *(q + 1) == '\0') {
@@ -714,7 +773,7 @@ void parse_inittab(void)
                        ++q;
                }
 
-               /* Now peal off the action field */
+               /* Now peel off the action field */
                r = strrchr(p, ':');
                if (r == NULL || *(r + 1) == '\0') {
                        message(LOG | CONSOLE, "Bad inittab entry: %s\n", lineAsRead);
@@ -753,9 +812,11 @@ void parse_inittab(void)
                }
        }
        return;
-#endif
+#endif /* BB_FEATURE_USE_INITTAB */
 }
 
+
+
 extern int init_main(int argc, char **argv)
 {
        initAction *a;
@@ -763,14 +824,17 @@ extern int init_main(int argc, char **argv)
        int status;
 
 #ifndef DEBUG_INIT
-       /* 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");
+       /* Expect to be invoked as init with PID=1 or be invoked as linuxrc */
+       if (getpid() != 1
+#ifdef BB_FEATURE_LINUXRC
+                       && strstr(argv[0], "linuxrc") == NULL
+#endif
+                         )
+       {
+                       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]));
-
        /* Set up sig handlers  -- be sure to
         * clear all of these in run() */
        signal(SIGUSR1, halt_signal);
@@ -786,6 +850,9 @@ extern int init_main(int argc, char **argv)
        reboot(RB_DISABLE_CAD);
 #endif
 
+       /* Figure out what kernel this is running */
+       kernelVersion = get_kernel_revision();
+
        /* Figure out where the default console should be */
        console_init();
 
@@ -794,6 +861,7 @@ extern int init_main(int argc, char **argv)
        close(1);
        close(2);
        set_term(0);
+       chdir("/");
        setsid();
 
        /* Make sure PATH is set to something sane */
@@ -801,23 +869,24 @@ extern int init_main(int argc, char **argv)
 
        /* Hello world */
 #ifndef DEBUG_INIT
-       message(LOG | CONSOLE,
+       message(
+#if ! defined BB_FEATURE_EXTRA_QUIET
+                       CONSOLE|
+#endif
+                       LOG,
                        "init started:  BusyBox v%s (%s) multi-call binary\r\n",
                        BB_VER, BB_BT);
 #else
-       message(LOG | CONSOLE,
+       message(
+#if ! defined BB_FEATURE_EXTRA_QUIET
+                       CONSOLE|
+#endif
+                       LOG,
                        "init(%d) started:  BusyBox v%s (%s) multi-call binary\r\n",
                        getpid(), BB_VER, BB_BT);
 #endif
 
 
-       /* Mount /proc */
-       if (mount("proc", "/proc", "proc", 0, 0) == 0) {
-               message(LOG | CONSOLE, "Mounting /proc: done.\n");
-               kernelVersion = get_kernel_revision();
-       } else
-               message(LOG | CONSOLE, "Mounting /proc: failed!\n");
-
        /* Make sure there is enough memory to do something useful. */
        check_memory();
 
@@ -839,6 +908,11 @@ extern int init_main(int argc, char **argv)
                parse_inittab();
        }
 
+       /* Fix up argv[0] to be certain we claim to be init */
+       strncpy(argv[0], "init", strlen(argv[0])+1);
+       if (argc > 1)
+               strncpy(argv[1], "\0", strlen(argv[1])+1);
+
        /* Now run everything that needs to be run */
 
        /* First run the sysinit command */
@@ -910,3 +984,11 @@ extern int init_main(int argc, char **argv)
                sleep(1);
        }
 }
+
+/*
+Local Variables:
+c-file-style: "linux"
+c-basic-offset: 4
+tab-width: 4
+End:
+*/