Apply patch from Felipe Kellermann to simlify logic of sort functions.
[oweals/busybox.git] / init / init.c
index be91d6a8f075696b10830b61ba78ba46172673bb..2278e521f81b5b5d46dc0c48e3b2fe2a95b4fe6b 100644 (file)
@@ -3,7 +3,7 @@
  * Mini init implementation for busybox
  *
  * Copyright (C) 1995, 1996 by Bruce Perens <bruce@pixar.com>.
- * Copyright (C) 1999-2002 by Erik Andersen <andersee@debian.org>
+ * Copyright (C) 1999-2004 by Erik Andersen <andersen@codepoet.org>
  * Adjusted by so many folks, it's impossible to keep track.
  *
  * This program is free software; you can redistribute it and/or modify
@@ -22,7 +22,7 @@
  *
  */
 
-/* Turn this on to disable all the dangerous 
+/* Turn this on to disable all the dangerous
    rebooting stuff when debugging.
 #define DEBUG_INIT
 */
 #include <sys/mount.h>
 #include <sys/types.h>
 #include <sys/wait.h>
+#include <sys/reboot.h>
 #include "busybox.h"
+
+#include "init_shared.h"
+
+
 #ifdef CONFIG_SYSLOGD
 # include <sys/syslog.h>
 #endif
 
-#if defined(__UCLIBC__) && !defined(__UCLIBC_HAS_MMU__)
-#define fork   vfork
-#endif
 
 #define INIT_BUFFS_SIZE 256
 
@@ -63,37 +65,35 @@ static const int VT_GETSTATE = 0x5603;      /* get global vt state info */
 
 /* From <linux/serial.h> */
 struct serial_struct {
-       int type;
-       int line;
-       int port;
-       int irq;
-       int flags;
-       int xmit_fifo_size;
-       int custom_divisor;
-       int baud_base;
-       unsigned short close_delay;
-       char reserved_char[2];
-       int hub6;
-       unsigned short closing_wait;    /* time to wait before closing */
-       unsigned short closing_wait2;   /* no longer used... */
-       int reserved[4];
+       int     type;
+       int     line;
+       unsigned int    port;
+       int     irq;
+       int     flags;
+       int     xmit_fifo_size;
+       int     custom_divisor;
+       int     baud_base;
+       unsigned short  close_delay;
+       char    io_type;
+       char    reserved_char[1];
+       int     hub6;
+       unsigned short  closing_wait; /* time to wait before closing */
+       unsigned short  closing_wait2; /* no longer used... */
+       unsigned char   *iomem_base;
+       unsigned short  iomem_reg_shift;
+       unsigned int    port_high;
+       unsigned long   iomap_base;     /* cookie passed into ioremap */
+       int     reserved[1];
 };
 
 
-#if (__GNU_LIBRARY__ > 5) || defined(__dietlibc__)
-#include <sys/reboot.h>
-#define init_reboot(magic) reboot(magic)
-#else
-#define init_reboot(magic) reboot(0xfee1dead, 672274793, magic)
-#endif
-
 #ifndef _PATH_STDPATH
 #define _PATH_STDPATH  "/usr/bin:/bin:/usr/sbin:/sbin"
 #endif
 
 #if defined CONFIG_FEATURE_INIT_COREDUMPS
 /*
- * When a file named CORE_ENABLE_FLAG_FILE exists, setrlimit is called 
+ * 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....
@@ -105,14 +105,6 @@ struct serial_struct {
 
 #define KERNEL_VERSION(a,b,c) (((a) << 16) + ((b) << 8) + (c))
 
-#if __GNU_LIBRARY__ > 5
-#include <sys/kdaemon.h>
-#else
-extern int bdflush(int func, long int data);
-#endif
-
-#define SHELL        "/bin/sh" /* Default shell */
-#define LOGIN_SHELL  "-" SHELL /* Default login shell */
 #define INITTAB      "/etc/inittab"    /* inittab file location */
 #ifndef INIT_SCRIPT
 #define INIT_SCRIPT  "/etc/init.d/rcS" /* Default sysinit script. */
@@ -161,7 +153,6 @@ struct init_action {
 
 /* Static variables */
 static struct init_action *init_action_list = NULL;
-static int kernelVersion;
 static char console[CONSOLE_BUFF_SIZE] = _PATH_CONSOLE;
 
 #ifndef CONFIG_SYSLOGD
@@ -188,7 +179,7 @@ static const int RB_AUTOBOOT = 0x01234567;
 static const char * const environment[] = {
        "HOME=/",
        "PATH=" _PATH_STDPATH,
-       "SHELL=" SHELL,
+       "SHELL=/bin/sh",
        "USER=root",
        NULL
 };
@@ -196,6 +187,7 @@ static const char * const environment[] = {
 /* Function prototypes */
 static void delete_init_action(struct init_action *a);
 static int waitfor(const struct init_action *a);
+static void halt_signal(int sig);
 
 
 static void loop_forever(void)
@@ -233,7 +225,9 @@ static void message(int device, const char *fmt, ...)
        /* Log the message to syslogd */
        if (device & LOG) {
                /* don`t out "\r\n" */
-               syslog_msg(LOG_DAEMON, LOG_INFO, msg + 1);
+               openlog(bb_applet_name, 0, LOG_DAEMON);
+               syslog(LOG_INFO, "%s", msg);
+               closelog();
        }
 
        msg[l++] = '\n';
@@ -317,7 +311,7 @@ static void set_term(int fd)
 
 /* How much memory does this machine have?
    Units are kBytes to avoid overflow on 4GB machines */
-static int check_free_memory(void)
+static unsigned int check_free_memory(void)
 {
        struct sysinfo info;
        unsigned int result, u, s = 10;
@@ -337,10 +331,11 @@ static int check_free_memory(void)
                s--;
        }
        result = (info.totalram >> s) + (info.totalswap >> s);
-       result = result * u;
-       if (result < 0)
-               result = INT_MAX;
-       return result;
+       if (((unsigned long long)result * (unsigned long long)u) > UINT_MAX) {
+               return(UINT_MAX);
+       } else {
+               return(result * u);
+       }
 }
 
 static void console_init(void)
@@ -354,7 +349,7 @@ static void console_init(void)
        if ((s = getenv("CONSOLE")) != NULL || (s = getenv("console")) != NULL) {
                safe_strncpy(console, s, sizeof(console));
 #if #cpu(sparc)
-       /* sparc kernel supports console=tty[ab] parameter which is also 
+       /* sparc kernel supports console=tty[ab] parameter which is also
         * passed to init, so catch it here */
                /* remap tty[ab] to /dev/ttyS[01] */
                if (strcmp(s, "ttya") == 0)
@@ -503,7 +498,12 @@ static pid_t run(const struct init_action *a)
                                signal(SIGCHLD, SIG_DFL);
 
                                /* Wait for child to exit */
-                               while ((tmp_pid = waitpid(pid, &junk, 0)) != pid);
+                               while ((tmp_pid = waitpid(pid, &junk, 0)) != pid) {
+                                       if (tmp_pid == -1 && errno == ECHILD) {
+                                               break;
+                                       }
+                                       /* FIXME handle other errors */
+                }
 
                                /* See if stealing the controlling tty back is necessary */
                                pgrp = tcgetpgrp(0);
@@ -532,7 +532,7 @@ static pid_t run(const struct init_action *a)
 
                /* See if any special /bin/sh requiring characters are present */
                if (strpbrk(a->command, "~`!$^&*()=|\\{}[];\"'<>?") != NULL) {
-                       cmd[0] = SHELL;
+                       cmd[0] = (char *)DEFAULT_SHELL;
                        cmd[1] = "-c";
                        cmd[2] = strcat(strcpy(buf, "exec "), a->command);
                        cmd[3] = NULL;
@@ -553,8 +553,8 @@ static pid_t run(const struct init_action *a)
 
                /*
                   Interactive shells want to see a dash in argv[0].  This
-                  typically is handled by login, argv will be setup this 
-                  way if a dash appears at the front of the command path 
+                  typically is handled by login, argv will be setup this
+                  way if a dash appears at the front of the command path
                   (like "-/bin/sh").
                 */
 
@@ -576,6 +576,7 @@ static pid_t run(const struct init_action *a)
                        }
                }
 
+#if !defined(__UCLIBC__) || defined(__ARCH_HAS_MMU__)
                if (a->action & ASKFIRST) {
                        char c;
                        /*
@@ -583,7 +584,7 @@ static pid_t run(const struct init_action *a)
                         * before the user wants it. This is critical if swap is not
                         * enabled and the system has low memory. Generally this will
                         * be run on the second virtual console, and the first will
-                        * be allowed to start a shell or whatever an init script 
+                        * be allowed to start a shell or whatever an init script
                         * specifies.
                         */
                        messageD(LOG, "Waiting for enter to start '%s'"
@@ -593,6 +594,7 @@ static pid_t run(const struct init_action *a)
                        while(read(0, &c, 1) == 1 && c != '\n')
                                ;
                }
+#endif
 
                /* Log the process name and args */
                message(LOG, "Starting pid %d, console %s: '%s'",
@@ -608,7 +610,7 @@ static pid_t run(const struct init_action *a)
                }
 #endif
 
-               /* Now run it.  The new program will take over this PID, 
+               /* Now run it.  The new program will take over this PID,
                 * so nothing further in init.c should be run. */
                execv(cmdpath, cmd);
 
@@ -627,12 +629,15 @@ static int waitfor(const struct init_action *a)
 
        pid = run(a);
        while (1) {
-               wpid = wait(&status);
-               if (wpid > 0 && wpid != pid) {
-                       continue;
-               }
+               wpid = waitpid(pid,&status,0);
                if (wpid == pid)
                        break;
+               if (wpid == -1 && errno == ECHILD) {
+                       /* we missed its termination */
+                       break;
+               }
+               /* FIXME other errors should maybe trigger an error, but allow
+                * the program to continue */
        }
        return wpid;
 }
@@ -663,6 +668,19 @@ static void run_actions(int action)
 }
 
 #ifndef DEBUG_INIT
+static void init_reboot(unsigned long magic)
+{
+       pid_t pid;
+       /* We have to fork here, since the kernel calls do_exit(0) in
+        * linux/kernel/sys.c, which can cause the machine to panic when
+        * the init process is killed.... */
+       if ((pid = fork()) == 0) {
+               reboot(magic);
+               _exit(0);
+       }
+       waitpid (pid, NULL, 0);
+}
+
 static void shutdown_system(void)
 {
        sigset_t block_signals;
@@ -702,11 +720,6 @@ static void shutdown_system(void)
        sleep(1);
 
        sync();
-       if (kernelVersion > 0 && kernelVersion <= KERNEL_VERSION(2, 2, 11)) {
-               /* bdflush, kupdate not needed for kernels >2.2.11 */
-               bdflush(1, 0);
-               sync();
-       }
 }
 
 static void exec_signal(int sig)
@@ -717,6 +730,8 @@ static void exec_signal(int sig)
        for (a = init_action_list; a; a = tmp) {
                tmp = a->next;
                if (a->action & RESTART) {
+                       struct stat sb;
+
                        shutdown_system();
 
                        /* unblock all signals, blocked in shutdown_system() */
@@ -732,6 +747,27 @@ static void exec_signal(int sig)
                        sigaddset(&unblock_signals, SIGTSTP);
                        sigprocmask(SIG_UNBLOCK, &unblock_signals, NULL);
 
+                       /* Close whatever files are open. */
+                       close(0);
+                       close(1);
+                       close(2);
+
+                       /* Open the new terminal device */
+                       if ((device_open(a->terminal, O_RDWR)) < 0) {
+                               if (stat(a->terminal, &sb) != 0) {
+                                       message(LOG | CONSOLE, "device '%s' does not exist.", a->terminal);
+                               } else {
+                                       message(LOG | CONSOLE, "Bummer, can't open %s", a->terminal);
+                               }
+                               halt_signal(SIGUSR1);
+                       }
+
+                       /* Make sure the terminal will act fairly normal for us */
+                       set_term(0);
+                       /* Setup stdout, stderr on the supplied terminal */
+                       dup(0);
+                       dup(0);
+
                        messageD(CONSOLE | LOG, "Trying to re-exec %s", a->command);
                        execl(a->command, a->command, NULL);
 
@@ -761,7 +797,7 @@ static void halt_signal(int sig)
        /* allow time for last message to reach serial console */
        sleep(2);
 
-       if (sig == SIGUSR2 && kernelVersion >= KERNEL_VERSION(2, 2, 0))
+       if (sig == SIGUSR2)
                init_reboot(RB_POWER_OFF);
        else
                init_reboot(RB_HALT_SYSTEM);
@@ -808,7 +844,7 @@ static void cont_handler(int sig)
 
 #endif                                                 /* ! DEBUG_INIT */
 
-static void new_init_action(int action, char *command, const char *cons)
+static void new_init_action(int action, const char *command, const char *cons)
 {
        struct init_action *new_action, *a;
 
@@ -828,7 +864,14 @@ static void new_init_action(int action, char *command, const char *cons)
        }
 
        /* Append to the end of the list */
-       for (a = init_action_list; a && a->next; a = a->next);
+       for (a = init_action_list; a && a->next; a = a->next) {
+               /* don't enter action if it's already in the list */
+               if ((strcmp(a->command, command) == 0) &&
+                   (strcmp(a->terminal, cons) ==0)) {
+                       free(new_action);
+                       return;
+               }
+       }
        if (a) {
                a->next = new_action;
        } else {
@@ -870,7 +913,7 @@ static void check_memory(void)
        if (check_free_memory() > 1000)
                return;
 
-#if !defined(__UCLIBC__) || defined(__UCLIBC_HAS_MMU__)
+#if !defined(__UCLIBC__) || defined(__ARCH_HAS_MMU__)
        if (stat("/etc/fstab", &statBuf) == 0) {
                /* swapon -a requires /proc typically */
                new_init_action(SYSINIT, "/bin/mount -t proc proc /proc", "");
@@ -891,9 +934,9 @@ static void check_memory(void)
 
 /* NOTE that if CONFIG_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 CONFIG_FEATURE_USE_INITTAB 
- * _is_ defined, but /etc/inittab is missing, this 
+ * actions(i.e., runs INIT_SCRIPT and then starts a pair
+ * of "askfirst" shells).  If CONFIG_FEATURE_USE_INITTAB
+ * _is_ defined, but /etc/inittab is missing, this
  * results in the same set of default behaviors.
  */
 static void parse_inittab(void)
@@ -914,17 +957,17 @@ static void parse_inittab(void)
                new_init_action(CTRLALTDEL, "/sbin/reboot", "");
                /* Umount all filesystems on halt/reboot */
                new_init_action(SHUTDOWN, "/bin/umount -a -r", "");
-#if !defined(__UCLIBC__) || defined(__UCLIBC_HAS_MMU__)
+#if !defined(__UCLIBC__) || defined(__ARCH_HAS_MMU__)
                /* Swapoff on halt/reboot */
                new_init_action(SHUTDOWN, "/sbin/swapoff -a", "");
 #endif
                /* Prepare to restart init when a HUP is received */
                new_init_action(RESTART, "/sbin/init", "");
                /* Askfirst shell on tty1-4 */
-               new_init_action(ASKFIRST, LOGIN_SHELL, "");
-               new_init_action(ASKFIRST, LOGIN_SHELL, VC_2);
-               new_init_action(ASKFIRST, LOGIN_SHELL, VC_3);
-               new_init_action(ASKFIRST, LOGIN_SHELL, VC_4);
+               new_init_action(ASKFIRST, bb_default_login_shell, "");
+               new_init_action(ASKFIRST, bb_default_login_shell, VC_2);
+               new_init_action(ASKFIRST, bb_default_login_shell, VC_3);
+               new_init_action(ASKFIRST, bb_default_login_shell, VC_4);
                /* sysinit */
                new_init_action(SYSINIT, INIT_SCRIPT, "");
 
@@ -1003,7 +1046,14 @@ static void parse_inittab(void)
 #endif                                                 /* CONFIG_FEATURE_USE_INITTAB */
 }
 
-
+static void reload_signal(int sig)
+{
+        message(LOG, "Reloading /etc/inittab");
+        parse_inittab();
+       run_actions(RESPAWN);
+        return;
+}
+                                                            
 extern int init_main(int argc, char **argv)
 {
        struct init_action *a;
@@ -1011,15 +1061,7 @@ extern int init_main(int argc, char **argv)
        int status;
 
        if (argc > 1 && !strcmp(argv[1], "-q")) {
-               /* don't assume init's pid == 1 */
-               long *pid = find_pid_by_name("init");
-
-               if (!pid || *pid <= 0) {
-                       pid = find_pid_by_name("linuxrc");
-                       if (!pid || *pid <= 0)
-                               bb_error_msg_and_die("no process killed");
-               }
-               return kill(*pid, SIGHUP);
+               return kill_init(SIGHUP);
        }
 #ifndef DEBUG_INIT
        /* Expect to be invoked as init with PID=1 or be invoked as linuxrc */
@@ -1041,14 +1083,11 @@ extern int init_main(int argc, char **argv)
        signal(SIGSTOP, stop_handler);
        signal(SIGTSTP, stop_handler);
 
-       /* Turn off rebooting via CTL-ALT-DEL -- we get a 
+       /* Turn off rebooting via CTL-ALT-DEL -- we get a
         * SIGINT on CAD so we can shut things down gracefully... */
        init_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();
 
@@ -1080,13 +1119,13 @@ extern int init_main(int argc, char **argv)
        if (argc > 1 && (!strcmp(argv[1], "single") ||
                                         !strcmp(argv[1], "-s") || !strcmp(argv[1], "1"))) {
                /* Start a shell on console */
-               new_init_action(RESPAWN, LOGIN_SHELL, "");
+               new_init_action(RESPAWN, bb_default_login_shell, "");
        } else {
                /* Not in single user mode -- see what inittab says */
 
                /* NOTE that if CONFIG_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 
+                * actions(i.e., runs INIT_SCRIPT and then starts a pair
                 * of "askfirst" shells */
                parse_inittab();
        }
@@ -1105,12 +1144,8 @@ extern int init_main(int argc, char **argv)
        /* Next run anything to be run only once */
        run_actions(ONCE);
 
-       /* If there is nothing else to do, stop */
-       if (init_action_list == NULL) {
-               message(LOG | CONSOLE,
-                               "No more tasks for init -- sleeping forever.");
-               loop_forever();
-       }
+       /* Redefine SIGHUP to reread /etc/inittab */
+       signal(SIGHUP, reload_signal);
 
        /* Now run the looping stuff for the rest of forever */
        while (1) {