Moved some code and add a coupld #defines to support use of dmalloc.
[oweals/busybox.git] / init.c
diff --git a/init.c b/init.c
index 350ea62a8f4b405e1f06b448c6a83f1f2b0827af..ab4ad56519f2085ebd46226be8cddba1fa5bde87 100644 (file)
--- a/init.c
+++ b/init.c
 
 #include "internal.h"
 #include <stdio.h>
-#include <string.h>
 #include <stdlib.h>
-#include <stdarg.h>
-#include <unistd.h>
 #include <errno.h>
+#include <paths.h>
 #include <signal.h>
+#include <stdarg.h>
+#include <string.h>
 #include <termios.h>
-#include <paths.h>
-#include <sys/types.h>
+#include <unistd.h>
 #include <sys/fcntl.h>
-#include <sys/wait.h>
-#include <string.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 <sys/sysinfo.h>               /* For check_free_memory() */
-#include <linux/version.h>
+#include <sys/mount.h>
+#include <sys/types.h>
+#include <sys/wait.h>
 #ifdef BB_SYSLOGD
-#include <sys/syslog.h>
+# include <sys/syslog.h>
 #endif
 
-#ifndef KERNEL_VERSION
-#define KERNEL_VERSION(a,b,c) (((a) << 16) + ((b) << 8) + (c))
+#define bb_need_full_version
+#define BB_DECLARE_EXTERN
+#include "messages.c"
+
+/* From <linux/vt.h> */
+struct vt_stat {
+       unsigned short v_active;        /* active vt */
+       unsigned short v_signal;        /* signal to send */
+       unsigned short v_state;         /* vt bitmask */
+};
+#define 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];
+};
+
+
+
+#ifndef RB_HALT_SYSTEM
+#define RB_HALT_SYSTEM  0xcdef0123
+#define RB_ENABLE_CAD   0x89abcdef
+#define RB_DISABLE_CAD  0
+#define RB_POWER_OFF    0x4321fedc
+#define RB_AUTOBOOT     0x01234567
+#if defined(__GLIBC__)
+#include <sys/reboot.h>
+  #define init_reboot(magic) reboot(magic)
+#else
+  #define init_reboot(magic) reboot(0xfee1dead, 672274793, magic)
+#endif
+#endif
+
+#ifndef _PATH_STDPATH
+#define _PATH_STDPATH  "/usr/bin:/bin:/usr/sbin:/sbin"
 #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 */
+#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
+
+#define KERNEL_VERSION(a,b,c) (((a) << 16) + ((b) << 8) + (c))
+
+#if defined(__GLIBC__)
+#include <sys/kdaemon.h>
+#else
+static _syscall2(int, bdflush, int, func, int, data);
+#endif                                                 /* __GLIBC__ */
+
+
+#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 {
@@ -84,7 +142,7 @@ typedef enum {
        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;
@@ -97,10 +155,10 @@ static const struct initActionType actions[] = {
        {"wait", WAIT},
        {"once", ONCE},
        {"ctrlaltdel", CTRLALTDEL},
-       {0}
+       {0, 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;
@@ -113,16 +171,19 @@ initAction *initActionList = NULL;
 
 
 static char *secondConsole = VT_SECONDARY;
-static char *log = VT_LOG;
-static int kernelVersion = 0;
-static char termType[32] = "TERM=linux";
-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, ...)
+                  __attribute__ ((format (printf, 2, 3)));
+static void message(int device, char *fmt, ...)
 {
        va_list arguments;
        int fd;
@@ -186,28 +247,29 @@ void message(int device, char *fmt, ...)
        }
 }
 
-
 /* Set terminal settings to reasonable defaults */
 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;
@@ -227,14 +289,28 @@ static int check_free_memory()
 {
        struct sysinfo info;
 
+       /* Pre initialize mem_unit in case this kernel is something prior to
+        * the linux 2.4 kernel (which will actually fill in mem_unit... */
        sysinfo(&info);
        if (sysinfo(&info) != 0) {
-               message(LOG, "Error checking free memory: %s\n", strerror(errno));
+               printf("Error checking free memory: %s\n", strerror(errno));
                return -1;
        }
+       if (info.mem_unit==0) {
+               /* Looks like we have a kernel prior to Linux 2.4.x */
+               info.mem_unit=1024;
+               info.totalram/=info.mem_unit;
+               info.totalswap/=info.mem_unit;
+       } else {
+               /* Bah. Linux 2.4.x completely changed sysinfo. This can in theory
+               overflow a 32 bit unsigned long, but who puts more then 4GiB ram+swap
+               on an embedded system? */
+               info.mem_unit/=1024;
+               info.totalram*=info.mem_unit;
+               info.totalswap*=info.mem_unit;
+       }
 
-       return(info.freeram/1024);
-
+       return(info.totalram+info.totalswap);
 }
 
 static void console_init()
@@ -336,10 +412,6 @@ static pid_t run(char *command, char *terminal, int get_enter)
 
 
        if ((pid = fork()) == 0) {
-#ifdef DEBUG_INIT
-               pid_t shell_pgid = getpid();
-#endif
-
                /* Clean up */
                close(0);
                close(1);
@@ -360,6 +432,7 @@ static pid_t run(char *command, char *terminal, int get_enter)
                dup2(fd, 0);
                dup2(fd, 1);
                dup2(fd, 2);
+               ioctl(0, TIOCSCTTY, 0);
                tcsetpgrp(0, getpgrp());
                set_term(0);
 
@@ -373,8 +446,8 @@ 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
@@ -409,6 +482,18 @@ static pid_t run(char *command, char *terminal, int get_enter)
                        cmd[i] = NULL;
                }
 
+#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. */
                execve(cmd[0], cmd, environment);
@@ -447,8 +532,10 @@ static void check_memory()
                return;
 
        if (stat("/etc/fstab", &statBuf) == 0) {
+               /* swapon -a requires /proc typically */
+               waitfor("mount proc /proc -t proc", console, FALSE);
                /* Try to turn on swap */
-               system("/sbin/swapon -a");
+               waitfor("swapon -a", console, FALSE);
                if (check_free_memory() < 1000)
                        goto goodnight;
        } else
@@ -483,7 +570,7 @@ static void shutdown_system(void)
        signal(SIGHUP, SIG_DFL);
 
        /* Allow Ctrl-Alt-Del to reboot system. */
-       reboot(RB_ENABLE_CAD);
+       init_reboot(RB_ENABLE_CAD);
 
        message(CONSOLE|LOG, "\r\nThe system is going down NOW !!\r\n");
        sync();
@@ -502,7 +589,7 @@ static void shutdown_system(void)
        run_lastAction();
 
        sync();
-       if (kernelVersion > 0 && kernelVersion <= 2 * 65536 + 2 * 256 + 11) {
+       if (kernelVersion > 0 && kernelVersion <= KERNEL_VERSION(2,2,11)) {
                /* bdflush, kupdate not needed for kernels >2.2.11 */
                bdflush(1, 0);
                sync();
@@ -521,12 +608,10 @@ static void halt_signal(int sig)
        /* allow time for last message to reach serial console */
        sleep(2);
 
-#if LINUX_VERSION_CODE >= KERNEL_VERSION(2,2,0)
-       if (sig == SIGUSR2)
-               reboot(RB_POWER_OFF);
+       if (sig == SIGUSR2 && kernelVersion >= KERNEL_VERSION(2,2,0))
+               init_reboot(RB_POWER_OFF);
        else
-#endif
-               reboot(RB_HALT_SYSTEM);
+               init_reboot(RB_HALT_SYSTEM);
        exit(0);
 }
 
@@ -539,7 +624,7 @@ static void reboot_signal(int sig)
        /* allow time for last message to reach serial console */
        sleep(2);
 
-       reboot(RB_AUTOBOOT);
+       init_reboot(RB_AUTOBOOT);
        exit(0);
 }
 
@@ -693,7 +778,7 @@ void parse_inittab(void)
 #ifdef BB_FEATURE_USE_INITTAB
        FILE *file;
        char buf[256], lineAsRead[256], tmpConsole[256];
-       char *p, *q, *r, *s;
+       char *id, *runlev, *action, *process, *eol;
        const struct initActionType *a = actions;
        int foundIt;
 
@@ -720,64 +805,69 @@ void parse_inittab(void)
 
        while (fgets(buf, 255, file) != NULL) {
                foundIt = FALSE;
-               for (p = buf; *p == ' ' || *p == '\t'; p++);
-               if (*p == '#' || *p == '\n')
+               /* Skip leading spaces */
+               for (id = buf; *id == ' ' || *id == '\t'; id++);
+
+               /* Skip the line if it's a comment */
+               if (*id == '#' || *id == '\n')
                        continue;
 
                /* Trim the trailing \n */
-               q = strrchr(p, '\n');
-               if (q != NULL)
-                       *q = '\0';
+               eol = strrchr(id, '\n');
+               if (eol != NULL)
+                       *eol = '\0';
 
                /* 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;
+               /* Separate the ID field from the runlevels */
+               runlev = strchr(id, ':');
+               if (runlev == NULL || *(runlev + 1) == '\0') {
+                       message(LOG | CONSOLE, "Bad inittab entry: %s\n", lineAsRead);
+                       continue;
+               } else {
+                       *runlev = '\0';
+                       ++runlev;
                }
 
-               /* Now peal off the process field from the end
-                * of the string */
-               q = strrchr(p, ':');
-               if (q == NULL || *(q + 1) == '\0') {
+               /* Separate the runlevels from the action */
+               action = strchr(runlev, ':');
+               if (action == NULL || *(action + 1) == '\0') {
                        message(LOG | CONSOLE, "Bad inittab entry: %s\n", lineAsRead);
                        continue;
                } else {
-                       *q = '\0';
-                       ++q;
+                       *action = '\0';
+                       ++action;
                }
 
-               /* Now peal off the action field */
-               r = strrchr(p, ':');
-               if (r == NULL || *(r + 1) == '\0') {
+               /* Separate the action from the process */
+               process = strchr(action, ':');
+               if (process == NULL || *(process + 1) == '\0') {
                        message(LOG | CONSOLE, "Bad inittab entry: %s\n", lineAsRead);
                        continue;
                } else {
-                       ++r;
+                       *process = '\0';
+                       ++process;
                }
 
                /* Ok, now process it */
                a = actions;
                while (a->name != 0) {
-                       if (strcmp(a->name, r) == 0) {
-                               if (*s != '\0') {
+                       if (strcmp(a->name, action) == 0) {
+                               if (*id != '\0') {
                                        struct stat statBuf;
 
                                        strcpy(tmpConsole, "/dev/");
-                                       strncat(tmpConsole, s, 200);
+                                       strncat(tmpConsole, id, 200);
                                        if (stat(tmpConsole, &statBuf) != 0) {
                                                message(LOG | CONSOLE,
                                                                "device '%s' does not exist.  Did you read the directions?\n",
                                                                tmpConsole);
                                                break;
                                        }
-                                       s = tmpConsole;
+                                       id = tmpConsole;
                                }
-                               new_initAction(a->action, q, s);
+                               new_initAction(a->action, process, id);
                                foundIt = TRUE;
                        }
                        a++;
@@ -790,7 +880,7 @@ void parse_inittab(void)
                }
        }
        return;
-#endif
+#endif /* BB_FEATURE_USE_INITTAB */
 }
 
 
@@ -805,7 +895,7 @@ extern int init_main(int argc, char **argv)
        /* 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
+                       && strstr(applet_name, "linuxrc") == NULL
 #endif
                          )
        {
@@ -816,7 +906,7 @@ extern int init_main(int argc, char **argv)
        /* Set up sig handlers  -- be sure to
         * clear all of these in run() */
        signal(SIGUSR1, halt_signal);
-       signal(SIGUSR2, reboot_signal);
+       signal(SIGUSR2, halt_signal);
        signal(SIGINT, reboot_signal);
        signal(SIGTERM, reboot_signal);
 #if defined BB_FEATURE_INIT_CHROOT
@@ -825,9 +915,12 @@ extern int init_main(int argc, char **argv)
 
        /* Turn off rebooting via CTL-ALT-DEL -- we get a 
         * SIGINT on CAD so we can shut things down gracefully... */
-       reboot(RB_DISABLE_CAD);
+       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();
 
@@ -836,6 +929,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 */
@@ -843,13 +937,19 @@ extern int init_main(int argc, char **argv)
 
        /* Hello world */
 #ifndef DEBUG_INIT
-       message(LOG,
-                       "init started:  BusyBox v%s (%s) multi-call binary\r\n",
-                       BB_VER, BB_BT);
+       message(
+#if ! defined BB_FEATURE_EXTRA_QUIET
+                       CONSOLE|
+#endif
+                       LOG,
+                       "init started:  %s\r\n", full_version);
 #else
-       message(LOG,
-                       "init(%d) started:  BusyBox v%s (%s) multi-call binary\r\n",
-                       getpid(), BB_VER, BB_BT);
+       message(
+#if ! defined BB_FEATURE_EXTRA_QUIET
+                       CONSOLE|
+#endif
+                       LOG,
+                       "init(%d) started:  %s\r\n", getpid(), full_version);
 #endif
 
 
@@ -873,7 +973,7 @@ extern int init_main(int argc, char **argv)
                 * of "askfirst" shells */
                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)