Patch from Jason Schoon to add optional SIGUSR1 support to dd.
[oweals/busybox.git] / init / init.c
index 95d7253ea9c4a2613d6d3a8cc5728cf52c175b41..22fb33dfd8d3a5b8d07215d072a69f97dfa3079e 100644 (file)
@@ -6,20 +6,7 @@
  * 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
- * it under the terms of the GNU General Public License as published by
- * the Free Software Foundation; either version 2 of the License, or
- * (at your option) any later version.
- *
- * This program is distributed in the hope that it will be useful,
- * but WITHOUT ANY WARRANTY; without even the implied warranty of
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
- * General Public License for more details.
- *
- * You should have received a copy of the GNU General Public License
- * along with this program; if not, write to the Free Software
- * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
- *
+ * Licensed under GPLv2 or later, see file LICENSE in this tarball for details.
  */
 
 /* Turn this on to disable all the dangerous
@@ -37,7 +24,7 @@
 #include <termios.h>
 #include <unistd.h>
 #include <limits.h>
-#include <sys/fcntl.h>
+#include <fcntl.h>
 #include <sys/ioctl.h>
 #include <sys/types.h>
 #include <sys/wait.h>
 #endif
 
 
+#ifdef CONFIG_SELINUX
+# include <selinux/selinux.h>
+#endif /* CONFIG_SELINUX */
+
+
 #define INIT_BUFFS_SIZE 256
 
 /* From <linux/vt.h> */
@@ -60,7 +52,7 @@ struct vt_stat {
        unsigned short v_signal;        /* signal to send */
        unsigned short v_state; /* vt bitmask */
 };
-static const int VT_GETSTATE = 0x5603; /* get global vt state info */
+enum { VT_GETSTATE = 0x5603 }; /* get global vt state info */
 
 /* From <linux/serial.h> */
 struct serial_struct {
@@ -158,22 +150,25 @@ static char console[CONSOLE_BUFF_SIZE] = _PATH_CONSOLE;
 static char *log_console = VC_5;
 #endif
 static sig_atomic_t got_cont = 0;
-static const int LOG = 0x1;
-static const int CONSOLE = 0x2;
+
+enum {
+       LOG = 0x1,
+       CONSOLE = 0x2,
 
 #if defined CONFIG_FEATURE_EXTRA_QUIET
-static const int MAYBE_CONSOLE = 0x0;
+       MAYBE_CONSOLE = 0x0,
 #else
-#define MAYBE_CONSOLE  CONSOLE
+       MAYBE_CONSOLE = CONSOLE,
 #endif
-#ifndef RB_HALT_SYSTEM
-static const int RB_HALT_SYSTEM = 0xcdef0123;
-static const int RB_ENABLE_CAD = 0x89abcdef;
-static const int RB_DISABLE_CAD = 0;
 
-#define RB_POWER_OFF    0x4321fedc
-static const int RB_AUTOBOOT = 0x01234567;
+#ifndef RB_HALT_SYSTEM
+       RB_HALT_SYSTEM = 0xcdef0123,
+       RB_ENABLE_CAD = 0x89abcdef,
+       RB_DISABLE_CAD = 0,
+       RB_POWER_OFF = 0x4321fedc,
+       RB_AUTOBOOT = 0x01234567,
 #endif
+};
 
 static const char * const environment[] = {
        "HOME=/",
@@ -198,7 +193,8 @@ static void loop_forever(void)
 /* Print a message to the specified device.
  * Device may be bitwise-or'd from LOG | CONSOLE */
 #ifndef DEBUG_INIT
-static inline void messageD(int device, const char *fmt, ...)
+static inline void messageD(int ATTRIBUTE_UNUSED device, 
+                               const char ATTRIBUTE_UNUSED *fmt, ...)
 {
 }
 #else
@@ -225,7 +221,7 @@ static void message(int device, const char *fmt, ...)
        if (device & LOG) {
                /* don`t out "\r\n" */
                openlog(bb_applet_name, 0, LOG_DAEMON);
-               syslog(LOG_INFO, "%s", msg);
+               syslog(LOG_INFO, "%s", msg + 1);
                closelog();
        }
 
@@ -308,37 +304,6 @@ static void set_term(int fd)
        tcsetattr(fd, TCSANOW, &tty);
 }
 
-#ifdef CONFIG_FEATURE_INIT_SWAPON
-/* How much memory does this machine have?
-   Units are kBytes to avoid overflow on 4GB machines */
-static unsigned int check_free_memory(void)
-{
-       struct sysinfo info;
-       unsigned int result, u, s = 10;
-
-       if (sysinfo(&info) != 0) {
-               bb_perror_msg("Error checking free memory");
-               return -1;
-       }
-
-       /* Kernels 2.0.x and 2.2.x return info.mem_unit==0 with values in bytes.
-        * Kernels 2.4.0 return info.mem_unit in bytes. */
-       u = info.mem_unit;
-       if (u == 0)
-               u = 1;
-       while ((u & 1) == 0 && s > 0) {
-               u >>= 1;
-               s--;
-       }
-       result = (info.totalram >> s) + (info.totalswap >> s);
-       if (((unsigned long long)result * (unsigned long long)u) > UINT_MAX) {
-               return(UINT_MAX);
-       } else {
-               return(result * u);
-       }
-}
-#endif /* CONFIG_FEATURE_INIT_SWAPON */
-
 static void console_init(void)
 {
        int fd;
@@ -384,7 +349,7 @@ static void console_init(void)
 #ifndef CONFIG_SYSLOGD
                log_console =
 #endif
-               safe_strncpy(console, "/dev/null", sizeof(console));
+               safe_strncpy(console, bb_dev_null, sizeof(console));
        } else {
                s = getenv("TERM");
                /* check for serial console */
@@ -424,9 +389,8 @@ static void fixup_argv(int argc, char **argv, char *new_argv0)
 
 static pid_t run(const struct init_action *a)
 {
-       struct stat sb;
        int i, junk;
-       pid_t pid, pgrp, tmp_pid;
+       pid_t pid;
        char *s, *tmpCmd, *cmd[INIT_BUFFS_SIZE], *cmdpath;
        char buf[INIT_BUFFS_SIZE + 6];  /* INIT_BUFFS_SIZE+strlen("exec ")+1 */
        sigset_t nmask, omask;
@@ -442,6 +406,8 @@ static pid_t run(const struct init_action *a)
        sigprocmask(SIG_BLOCK, &nmask, &omask);
 
        if ((pid = fork()) == 0) {
+               struct stat sb;
+
                /* Clean up */
                close(0);
                close(1);
@@ -466,11 +432,10 @@ static pid_t run(const struct init_action *a)
                /* 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);
-                               _exit(1);
+                               message(LOG | CONSOLE, "device '%s' does not exist.", a->terminal);
+                       } else {
+                               message(LOG | CONSOLE, "Bummer, can't open %s", a->terminal);
                        }
-                       message(LOG | CONSOLE, "Bummer, can't open %s", a->terminal);
                        _exit(1);
                }
 
@@ -484,6 +449,7 @@ static pid_t run(const struct init_action *a)
                /* If the init Action requires us to wait, then force the
                 * supplied terminal to be the controlling tty. */
                if (a->action & (SYSINIT | WAIT | CTRLALTDEL | SHUTDOWN | RESTART)) {
+                       pid_t pgrp, tmp_pid;
 
                        /* Now fork off another process to just hang around */
                        if ((pid = fork()) < 0) {
@@ -505,7 +471,7 @@ static pid_t run(const struct init_action *a)
                                                break;
                                        }
                                        /* FIXME handle other errors */
-                }
+               }
 
                                /* See if stealing the controlling tty back is necessary */
                                pgrp = tcgetpgrp(0);
@@ -576,6 +542,13 @@ static pid_t run(const struct init_action *a)
                                cmd[0][0] = '-';
                                strcpy(cmd[0] + 1, s);
                        }
+#ifdef CONFIG_FEATURE_INIT_SCTTY
+                       /* Establish this process as session leader and
+                        * (attempt) to make the tty (if any) a controlling tty.
+                        */
+                       (void) setsid();
+                       (void) ioctl(0, TIOCSCTTY, 0/*don't steal it*/);
+#endif
                }
 
 #if !defined(__UCLIBC__) || defined(__ARCH_HAS_MMU__)
@@ -652,7 +625,9 @@ static void run_actions(int action)
        for (a = init_action_list; a; a = tmp) {
                tmp = a->next;
                if (a->action == action) {
-                       if (a->action & (SYSINIT | WAIT | CTRLALTDEL | SHUTDOWN | RESTART)) {
+                       if (access(a->terminal, R_OK | W_OK)) {
+                               delete_init_action(a);
+                       } else if (a->action & (SYSINIT | WAIT | CTRLALTDEL | SHUTDOWN | RESTART)) {
                                waitfor(a);
                                delete_init_action(a);
                        } else if (a->action & ONCE) {
@@ -725,7 +700,7 @@ static void shutdown_system(void)
        sync();
 }
 
-static void exec_signal(int sig)
+static void exec_signal(int sig ATTRIBUTE_UNUSED)
 {
        struct init_action *a, *tmp;
        sigset_t unblock_signals;
@@ -733,8 +708,6 @@ 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() */
@@ -758,6 +731,7 @@ static void exec_signal(int sig)
 
                        /* Open the new terminal device */
                        if ((device_open(a->terminal, O_RDWR)) < 0) {
+                               struct stat sb;
                                if (stat(a->terminal, &sb) != 0) {
                                        message(LOG | CONSOLE, "device '%s' does not exist.", a->terminal);
                                } else {
@@ -785,17 +759,10 @@ static void exec_signal(int sig)
        }
 }
 
-static void halt_signal(int sig)
+static void halt_signal(int sig ATTRIBUTE_UNUSED)
 {
        shutdown_system();
-       message(CONSOLE | LOG,
-#if #cpu(s390)
-                       /* Seems the s390 console is Wierd(tm). */
-                       "The system is halted. You may reboot now."
-#else
-                       "The system is halted. Press Reset or turn off power"
-#endif
-               );
+       message(CONSOLE | LOG, "The system is halted.");
        sync();
 
        /* allow time for last message to reach serial console */
@@ -809,7 +776,7 @@ static void halt_signal(int sig)
        loop_forever();
 }
 
-static void reboot_signal(int sig)
+static void reboot_signal(int sig ATTRIBUTE_UNUSED)
 {
        shutdown_system();
        message(CONSOLE | LOG, "Please stand by while rebooting the system.");
@@ -823,13 +790,13 @@ static void reboot_signal(int sig)
        loop_forever();
 }
 
-static void ctrlaltdel_signal(int sig)
+static void ctrlaltdel_signal(int sig ATTRIBUTE_UNUSED)
 {
        run_actions(CTRLALTDEL);
 }
 
 /* The SIGSTOP & SIGTSTP handler */
-static void stop_handler(int sig)
+static void stop_handler(int sig ATTRIBUTE_UNUSED)
 {
        int saved_errno = errno;
 
@@ -841,7 +808,7 @@ static void stop_handler(int sig)
 }
 
 /* The SIGCONT handler */
-static void cont_handler(int sig)
+static void cont_handler(int sig ATTRIBUTE_UNUSED)
 {
        got_cont = 1;
 }
@@ -855,10 +822,7 @@ static void new_init_action(int action, const char *command, const char *cons)
        if (*cons == '\0')
                cons = console;
 
-       /* do not run entries if console device is not available */
-       if (access(cons, R_OK | W_OK))
-               return;
-       if (strcmp(cons, "/dev/null") == 0 && (action & ASKFIRST))
+       if (strcmp(cons, bb_dev_null) == 0 && (action & ASKFIRST))
                return;
 
        new_action = calloc((size_t) (1), sizeof(struct init_action));
@@ -911,38 +875,6 @@ static void delete_init_action(struct init_action *action)
        }
 }
 
-#ifdef CONFIG_FEATURE_INIT_SWAPON
-/* Make sure there is enough memory to do something useful. *
- * Calls "swapon -a" if needed so be sure /etc/fstab is present... */
-static void check_memory(void)
-{
-       struct stat statBuf;
-
-       if (check_free_memory() > 1000)
-               return;
-
-#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", "");
-               /* Try to turn on swap */
-               new_init_action(SYSINIT, "/sbin/swapon -a", "");
-               run_actions(SYSINIT);   /* wait and removing */
-               if (check_free_memory() < 1000)
-                       goto goodnight;
-       } else
-               goto goodnight;
-       return;
-#endif
-
-  goodnight:
-       message(CONSOLE, "Sorry, your computer does not have enough memory.");
-       loop_forever();
-}
-#else
-# define check_memory()
-#endif /* CONFIG_FEATURE_INIT_SWAPON */
-
 /* 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
@@ -968,10 +900,8 @@ 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(__ARCH_HAS_MMU__)
                /* Swapoff on halt/reboot */
-               new_init_action(SHUTDOWN, "/sbin/swapoff -a", "");
-#endif
+               if(ENABLE_SWAPONOFF) new_init_action(SHUTDOWN, "/sbin/swapoff -a", "");
                /* Prepare to restart init when a HUP is received */
                new_init_action(RESTART, "/sbin/init", "");
                /* Askfirst shell on tty1-4 */
@@ -1058,7 +988,7 @@ static void parse_inittab(void)
 }
 
 #ifdef CONFIG_FEATURE_USE_INITTAB
-static void reload_signal(int sig)
+static void reload_signal(int sig ATTRIBUTE_UNUSED)
 {
        struct init_action *a, *tmp;
 
@@ -1084,22 +1014,20 @@ static void reload_signal(int sig)
 }
 #endif                                                 /* CONFIG_FEATURE_USE_INITTAB */
 
-extern int init_main(int argc, char **argv)
+int init_main(int argc, char **argv)
 {
        struct init_action *a;
        pid_t wpid;
        int status;
 
        if (argc > 1 && !strcmp(argv[1], "-q")) {
-               return kill_init(SIGHUP);
+               return kill(1,SIGHUP);
        }
 #ifndef DEBUG_INIT
        /* Expect to be invoked as init with PID=1 or be invoked as linuxrc */
-       if (getpid() != 1
-#ifdef CONFIG_FEATURE_INITRD
-               && strstr(bb_applet_name, "linuxrc") == NULL
-#endif
-               ) {
+       if (getpid() != 1 &&
+               (!ENABLE_FEATURE_INITRD || !strstr(bb_applet_name, "linuxrc")))
+       {
                bb_show_usage();
        }
        /* Set up sig handlers  -- be sure to
@@ -1144,7 +1072,20 @@ extern int init_main(int argc, char **argv)
        message(MAYBE_CONSOLE | LOG, "init started:  %s", bb_msg_full_version);
 
        /* Make sure there is enough memory to do something useful. */
-       check_memory();
+       if (ENABLE_SWAPONOFF) {
+               struct sysinfo info;
+
+               if (!sysinfo(&info) &&
+                       (info.mem_unit ? : 1) * (long long)info.totalram < MEGABYTE)
+               {
+                       message(CONSOLE,"Low memory: forcing swapon.");
+                       /* swapon -a requires /proc typically */
+                       new_init_action(SYSINIT, "/bin/mount -t proc proc /proc", "");
+                       /* Try to turn on swap */
+                       new_init_action(SYSINIT, "/sbin/swapon -a", "");
+                       run_actions(SYSINIT);   /* wait and removing */
+               }
+       }
 
        /* Check if we are supposed to be in single user mode */
        if (argc > 1 && (!strcmp(argv[1], "single") ||
@@ -1161,6 +1102,22 @@ extern int init_main(int argc, char **argv)
                parse_inittab();
        }
 
+#ifdef CONFIG_SELINUX
+       if (getenv("SELINUX_INIT") == NULL) {
+               int enforce = 0;
+
+               putenv("SELINUX_INIT=YES");
+               if (selinux_init_load_policy(&enforce) == 0) {
+                       execv(argv[0], argv);
+               } else if (enforce > 0) {
+                       /* SELinux in enforcing mode but load_policy failed */
+                       /* At this point, we probably can't open /dev/console, so log() won't work */
+                       message(CONSOLE,"Unable to load SELinux Policy. Machine is in enforcing mode. Halting now.");
+                       exit(1);
+               }
+       }
+#endif /* CONFIG_SELINUX */
+
        /* Make the command line just say "init"  -- thats all, nothing else */
        fixup_argv(argc, argv, "init");