source: MondoRescue/branches/stable/mindi-busybox/init/init.c @ 821

Last change on this file since 821 was 821, checked in by Bruno Cornec, 14 years ago

Addition of busybox 1.2.1 as a mindi-busybox new package
This should avoid delivering binary files in mindi not built there (Fedora and Debian are quite serious about that)

File size: 28.4 KB
Line 
1/* vi: set sw=4 ts=4: */
2/*
3 * Mini init implementation for busybox
4 *
5 * Copyright (C) 1995, 1996 by Bruce Perens <bruce@pixar.com>.
6 * Copyright (C) 1999-2004 by Erik Andersen <andersen@codepoet.org>
7 * Adjusted by so many folks, it's impossible to keep track.
8 *
9 * Licensed under GPLv2 or later, see file LICENSE in this tarball for details.
10 */
11
12#include "busybox.h"
13#include <stdio.h>
14#include <stdlib.h>
15#include <errno.h>
16#include <paths.h>
17#include <signal.h>
18#include <stdarg.h>
19#include <string.h>
20#include <termios.h>
21#include <unistd.h>
22#include <limits.h>
23#include <fcntl.h>
24#include <sys/ioctl.h>
25#include <sys/types.h>
26#include <sys/wait.h>
27#include <sys/reboot.h>
28
29#include "init_shared.h"
30
31
32#ifdef CONFIG_SYSLOGD
33# include <sys/syslog.h>
34#endif
35
36
37#ifdef CONFIG_SELINUX
38# include <selinux/selinux.h>
39#endif /* CONFIG_SELINUX */
40
41
42#define INIT_BUFFS_SIZE 256
43
44/* From <linux/vt.h> */
45struct vt_stat {
46    unsigned short v_active;    /* active vt */
47    unsigned short v_signal;    /* signal to send */
48    unsigned short v_state; /* vt bitmask */
49};
50enum { VT_GETSTATE = 0x5603 };  /* get global vt state info */
51
52/* From <linux/serial.h> */
53struct serial_struct {
54    int type;
55    int line;
56    unsigned int    port;
57    int irq;
58    int flags;
59    int xmit_fifo_size;
60    int custom_divisor;
61    int baud_base;
62    unsigned short  close_delay;
63    char    io_type;
64    char    reserved_char[1];
65    int hub6;
66    unsigned short  closing_wait; /* time to wait before closing */
67    unsigned short  closing_wait2; /* no longer used... */
68    unsigned char   *iomem_base;
69    unsigned short  iomem_reg_shift;
70    unsigned int    port_high;
71    unsigned long   iomap_base; /* cookie passed into ioremap */
72    int reserved[1];
73};
74
75
76#ifndef _PATH_STDPATH
77#define _PATH_STDPATH   "/usr/bin:/bin:/usr/sbin:/sbin"
78#endif
79
80#if defined CONFIG_FEATURE_INIT_COREDUMPS
81/*
82 * When a file named CORE_ENABLE_FLAG_FILE exists, setrlimit is called
83 * before processes are spawned to set core file size as unlimited.
84 * This is for debugging only.  Don't use this is production, unless
85 * you want core dumps lying about....
86 */
87#define CORE_ENABLE_FLAG_FILE "/.init_enable_core"
88#include <sys/resource.h>
89#endif
90
91#define KERNEL_VERSION(a,b,c) (((a) << 16) + ((b) << 8) + (c))
92
93#define INITTAB      "/etc/inittab" /* inittab file location */
94#ifndef INIT_SCRIPT
95#define INIT_SCRIPT  "/etc/init.d/rcS"  /* Default sysinit script. */
96#endif
97
98#define MAXENV  16      /* Number of env. vars */
99
100#define CONSOLE_BUFF_SIZE 32
101
102/* Allowed init action types */
103#define SYSINIT     0x001
104#define RESPAWN     0x002
105#define ASKFIRST    0x004
106#define WAIT        0x008
107#define ONCE        0x010
108#define CTRLALTDEL  0x020
109#define SHUTDOWN    0x040
110#define RESTART     0x080
111
112/* A mapping between "inittab" action name strings and action type codes. */
113struct init_action_type {
114    const char *name;
115    int action;
116};
117
118static const struct init_action_type actions[] = {
119    {"sysinit", SYSINIT},
120    {"respawn", RESPAWN},
121    {"askfirst", ASKFIRST},
122    {"wait", WAIT},
123    {"once", ONCE},
124    {"ctrlaltdel", CTRLALTDEL},
125    {"shutdown", SHUTDOWN},
126    {"restart", RESTART},
127    {0, 0}
128};
129
130/* Set up a linked list of init_actions, to be read from inittab */
131struct init_action {
132    pid_t pid;
133    char command[INIT_BUFFS_SIZE];
134    char terminal[CONSOLE_BUFF_SIZE];
135    struct init_action *next;
136    int action;
137};
138
139/* Static variables */
140static struct init_action *init_action_list = NULL;
141static char console[CONSOLE_BUFF_SIZE] = CONSOLE_DEV;
142
143#ifndef CONFIG_SYSLOGD
144static char *log_console = VC_5;
145#endif
146#if !ENABLE_DEBUG_INIT
147static sig_atomic_t got_cont = 0;
148#endif
149
150enum {
151    LOG = 0x1,
152    CONSOLE = 0x2,
153
154#if defined CONFIG_FEATURE_EXTRA_QUIET
155    MAYBE_CONSOLE = 0x0,
156#else
157    MAYBE_CONSOLE = CONSOLE,
158#endif
159
160#ifndef RB_HALT_SYSTEM
161    RB_HALT_SYSTEM = 0xcdef0123, /* FIXME: this overflows enum */
162    RB_ENABLE_CAD = 0x89abcdef,
163    RB_DISABLE_CAD = 0,
164    RB_POWER_OFF = 0x4321fedc,
165    RB_AUTOBOOT = 0x01234567,
166#endif
167};
168
169static const char * const environment[] = {
170    "HOME=/",
171    "PATH=" _PATH_STDPATH,
172    "SHELL=/bin/sh",
173    "USER=root",
174    NULL
175};
176
177/* Function prototypes */
178static void delete_init_action(struct init_action *a);
179static int waitfor(const struct init_action *a, pid_t pid);
180#if !ENABLE_DEBUG_INIT
181static void shutdown_signal(int sig);
182#endif
183
184static void loop_forever(void)
185{
186    while (1)
187        sleep(1);
188}
189
190/* Print a message to the specified device.
191 * Device may be bitwise-or'd from LOG | CONSOLE */
192#if ENABLE_DEBUG_INIT
193#define messageD message
194#else
195static inline void messageD(int ATTRIBUTE_UNUSED device,
196                const char ATTRIBUTE_UNUSED *fmt, ...)
197{
198}
199#endif
200static void message(int device, const char *fmt, ...)
201    __attribute__ ((format(printf, 2, 3)));
202static void message(int device, const char *fmt, ...)
203{
204    va_list arguments;
205    int l;
206    RESERVE_CONFIG_BUFFER(msg, 1024);
207#ifndef CONFIG_SYSLOGD
208    static int log_fd = -1;
209#endif
210
211    msg[0] = '\r';
212        va_start(arguments, fmt);
213    l = vsnprintf(msg + 1, 1024 - 2, fmt, arguments) + 1;
214        va_end(arguments);
215
216#ifdef CONFIG_SYSLOGD
217    /* Log the message to syslogd */
218    if (device & LOG) {
219        /* don`t out "\r\n" */
220        openlog(bb_applet_name, 0, LOG_DAEMON);
221        syslog(LOG_INFO, "%s", msg + 1);
222        closelog();
223    }
224
225    msg[l++] = '\n';
226    msg[l] = 0;
227#else
228
229    msg[l++] = '\n';
230    msg[l] = 0;
231    /* Take full control of the log tty, and never close it.
232     * It's mine, all mine!  Muhahahaha! */
233    if (log_fd < 0) {
234        if ((log_fd = device_open(log_console, O_RDWR | O_NONBLOCK | O_NOCTTY)) < 0) {
235            log_fd = -2;
236            bb_error_msg("Bummer, can't write to log on %s!", log_console);
237            device = CONSOLE;
238        } else {
239            fcntl(log_fd, F_SETFD, FD_CLOEXEC);
240        }
241    }
242    if ((device & LOG) && (log_fd >= 0)) {
243        bb_full_write(log_fd, msg, l);
244    }
245#endif
246
247    if (device & CONSOLE) {
248        int fd = device_open(CONSOLE_DEV,
249                    O_WRONLY | O_NOCTTY | O_NONBLOCK);
250        /* Always send console messages to /dev/console so people will see them. */
251        if (fd >= 0) {
252            bb_full_write(fd, msg, l);
253            close(fd);
254#if ENABLE_DEBUG_INIT
255        /* all descriptors may be closed */
256        } else {
257            bb_error_msg("Bummer, can't print: ");
258            va_start(arguments, fmt);
259            vfprintf(stderr, fmt, arguments);
260            va_end(arguments);
261#endif
262        }
263    }
264    RELEASE_CONFIG_BUFFER(msg);
265}
266
267/* Set terminal settings to reasonable defaults */
268static void set_term(void)
269{
270    struct termios tty;
271
272    tcgetattr(STDIN_FILENO, &tty);
273
274    /* set control chars */
275    tty.c_cc[VINTR] = 3;    /* C-c */
276    tty.c_cc[VQUIT] = 28;   /* C-\ */
277    tty.c_cc[VERASE] = 127; /* C-? */
278    tty.c_cc[VKILL] = 21;   /* C-u */
279    tty.c_cc[VEOF] = 4; /* C-d */
280    tty.c_cc[VSTART] = 17;  /* C-q */
281    tty.c_cc[VSTOP] = 19;   /* C-s */
282    tty.c_cc[VSUSP] = 26;   /* C-z */
283
284    /* use line dicipline 0 */
285    tty.c_line = 0;
286
287    /* Make it be sane */
288    tty.c_cflag &= CBAUD | CBAUDEX | CSIZE | CSTOPB | PARENB | PARODD;
289    tty.c_cflag |= CREAD | HUPCL | CLOCAL;
290
291
292    /* input modes */
293    tty.c_iflag = ICRNL | IXON | IXOFF;
294
295    /* output modes */
296    tty.c_oflag = OPOST | ONLCR;
297
298    /* local modes */
299    tty.c_lflag =
300        ISIG | ICANON | ECHO | ECHOE | ECHOK | ECHOCTL | ECHOKE | IEXTEN;
301
302    tcsetattr(STDIN_FILENO, TCSANOW, &tty);
303}
304
305static void console_init(void)
306{
307    int fd;
308    int tried = 0;
309    struct vt_stat vt;
310    struct serial_struct sr;
311    char *s;
312
313    if ((s = getenv("CONSOLE")) != NULL || (s = getenv("console")) != NULL) {
314        safe_strncpy(console, s, sizeof(console));
315#if 0 /* #cpu(sparc) */
316    /* sparc kernel supports console=tty[ab] parameter which is also
317     * passed to init, so catch it here */
318        /* remap tty[ab] to /dev/ttyS[01] */
319        if (strcmp(s, "ttya") == 0)
320            safe_strncpy(console, SC_0, sizeof(console));
321        else if (strcmp(s, "ttyb") == 0)
322            safe_strncpy(console, SC_1, sizeof(console));
323#endif
324    } else {
325        /* 2.2 kernels: identify the real console backend and try to use it */
326        if (ioctl(0, TIOCGSERIAL, &sr) == 0) {
327            /* this is a serial console */
328            snprintf(console, sizeof(console) - 1, SC_FORMAT, sr.line);
329        } else if (ioctl(0, VT_GETSTATE, &vt) == 0) {
330            /* this is linux virtual tty */
331            snprintf(console, sizeof(console) - 1, VC_FORMAT, vt.v_active);
332        } else {
333            safe_strncpy(console, CONSOLE_DEV, sizeof(console));
334            tried++;
335        }
336    }
337
338    while ((fd = open(console, O_RDONLY | O_NONBLOCK)) < 0 && tried < 2) {
339        /* Can't open selected console -- try
340            logical system console and VT_MASTER */
341        safe_strncpy(console, (tried == 0 ? CONSOLE_DEV : CURRENT_VC),
342                            sizeof(console));
343        tried++;
344    }
345    if (fd < 0) {
346        /* Perhaps we should panic here? */
347#ifndef CONFIG_SYSLOGD
348        log_console =
349#endif
350        safe_strncpy(console, bb_dev_null, sizeof(console));
351    } else {
352        s = getenv("TERM");
353        /* check for serial console */
354        if (ioctl(fd, TIOCGSERIAL, &sr) == 0) {
355            /* Force the TERM setting to vt102 for serial console --
356             * if TERM is set to linux (the default) */
357            if (s == NULL || strcmp(s, "linux") == 0)
358                putenv("TERM=vt102");
359#ifndef CONFIG_SYSLOGD
360            log_console = console;
361#endif
362        } else {
363            if (s == NULL)
364                putenv("TERM=linux");
365        }
366        close(fd);
367    }
368    messageD(LOG, "console=%s", console);
369}
370
371static void fixup_argv(int argc, char **argv, char *new_argv0)
372{
373    int len;
374
375    /* Fix up argv[0] to be certain we claim to be init */
376    len = strlen(argv[0]);
377    memset(argv[0], 0, len);
378    safe_strncpy(argv[0], new_argv0, len + 1);
379
380    /* Wipe argv[1]-argv[N] so they don't clutter the ps listing */
381    len = 1;
382    while (argc > len) {
383        memset(argv[len], 0, strlen(argv[len]));
384        len++;
385    }
386}
387
388/* Open the new terminal device */
389static void open_new_terminal(const char * const device, const int fail) {
390    struct stat sb;
391
392    if ((device_open(device, O_RDWR)) < 0) {
393        if (stat(device, &sb) != 0) {
394            message(LOG | CONSOLE, "device '%s' does not exist.", device);
395        } else {
396            message(LOG | CONSOLE, "Bummer, can't open %s", device);
397        }
398        if (fail)
399            _exit(1);
400        /* else */
401#if !ENABLE_DEBUG_INIT
402        shutdown_signal(SIGUSR1);
403#else
404        _exit(2);
405#endif
406    }
407}
408
409static pid_t run(const struct init_action *a)
410{
411    int i;
412    pid_t pid;
413    char *s, *tmpCmd, *cmd[INIT_BUFFS_SIZE], *cmdpath;
414    char buf[INIT_BUFFS_SIZE + 6];  /* INIT_BUFFS_SIZE+strlen("exec ")+1 */
415    sigset_t nmask, omask;
416    static const char press_enter[] =
417#ifdef CUSTOMIZED_BANNER
418#include CUSTOMIZED_BANNER
419#endif
420        "\nPlease press Enter to activate this console. ";
421
422    /* Block sigchild while forking.  */
423    sigemptyset(&nmask);
424    sigaddset(&nmask, SIGCHLD);
425    sigprocmask(SIG_BLOCK, &nmask, &omask);
426
427    if ((pid = fork()) == 0) {
428
429        /* Clean up */
430        close(0);
431        close(1);
432        close(2);
433        sigprocmask(SIG_SETMASK, &omask, NULL);
434
435        /* Reset signal handlers that were set by the parent process */
436        signal(SIGUSR1, SIG_DFL);
437        signal(SIGUSR2, SIG_DFL);
438        signal(SIGINT, SIG_DFL);
439        signal(SIGTERM, SIG_DFL);
440        signal(SIGHUP, SIG_DFL);
441        signal(SIGQUIT, SIG_DFL);
442        signal(SIGCONT, SIG_DFL);
443        signal(SIGSTOP, SIG_DFL);
444        signal(SIGTSTP, SIG_DFL);
445
446        /* Create a new session and make ourself the process
447         * group leader */
448        setsid();
449
450        /* Open the new terminal device */
451        open_new_terminal(a->terminal, 1);
452
453        /* Make sure the terminal will act fairly normal for us */
454        set_term();
455        /* Setup stdout, stderr for the new process so
456         * they point to the supplied terminal */
457        dup(0);
458        dup(0);
459
460        /* If the init Action requires us to wait, then force the
461         * supplied terminal to be the controlling tty. */
462        if (a->action & (SYSINIT | WAIT | CTRLALTDEL | SHUTDOWN | RESTART)) {
463
464            /* Now fork off another process to just hang around */
465            if ((pid = fork()) < 0) {
466                message(LOG | CONSOLE, "Can't fork!");
467                _exit(1);
468            }
469
470            if (pid > 0) {
471
472                /* We are the parent -- wait till the child is done */
473                signal(SIGINT, SIG_IGN);
474                signal(SIGTSTP, SIG_IGN);
475                signal(SIGQUIT, SIG_IGN);
476                signal(SIGCHLD, SIG_DFL);
477
478                waitfor(NULL, pid);
479                /* See if stealing the controlling tty back is necessary */
480                if (tcgetpgrp(0) != getpid())
481                    _exit(0);
482
483                /* Use a temporary process to steal the controlling tty. */
484                if ((pid = fork()) < 0) {
485                    message(LOG | CONSOLE, "Can't fork!");
486                    _exit(1);
487                }
488                if (pid == 0) {
489                    setsid();
490                    ioctl(0, TIOCSCTTY, 1);
491                    _exit(0);
492                }
493                waitfor(NULL, pid);
494                _exit(0);
495            }
496
497            /* Now fall though to actually execute things */
498        }
499
500        /* See if any special /bin/sh requiring characters are present */
501        if (strpbrk(a->command, "~`!$^&*()=|\\{}[];\"'<>?") != NULL) {
502            cmd[0] = (char *)DEFAULT_SHELL;
503            cmd[1] = "-c";
504            cmd[2] = strcat(strcpy(buf, "exec "), a->command);
505            cmd[3] = NULL;
506        } else {
507            /* Convert command (char*) into cmd (char**, one word per string) */
508            strcpy(buf, a->command);
509            s = buf;
510            for (tmpCmd = buf, i = 0; (tmpCmd = strsep(&s, " \t")) != NULL;) {
511                if (*tmpCmd != '\0') {
512                    cmd[i] = tmpCmd;
513                    i++;
514                }
515            }
516            cmd[i] = NULL;
517        }
518
519        cmdpath = cmd[0];
520
521        /*
522           Interactive shells want to see a dash in argv[0].  This
523           typically is handled by login, argv will be setup this
524           way if a dash appears at the front of the command path
525           (like "-/bin/sh").
526         */
527
528        if (*cmdpath == '-') {
529
530            /* skip over the dash */
531            ++cmdpath;
532
533            /* find the last component in the command pathname */
534            s = bb_get_last_path_component(cmdpath);
535
536            /* make a new argv[0] */
537            if ((cmd[0] = malloc(strlen(s) + 2)) == NULL) {
538                message(LOG | CONSOLE, bb_msg_memory_exhausted);
539                cmd[0] = cmdpath;
540            } else {
541                cmd[0][0] = '-';
542                strcpy(cmd[0] + 1, s);
543            }
544#ifdef CONFIG_FEATURE_INIT_SCTTY
545            /* Establish this process as session leader and
546             * (attempt) to make the tty (if any) a controlling tty.
547             */
548            (void) setsid();
549            (void) ioctl(0, TIOCSCTTY, 0/*don't steal it*/);
550#endif
551        }
552
553#if !defined(__UCLIBC__) || defined(__ARCH_HAS_MMU__)
554        if (a->action & ASKFIRST) {
555            char c;
556            /*
557             * Save memory by not exec-ing anything large (like a shell)
558             * before the user wants it. This is critical if swap is not
559             * enabled and the system has low memory. Generally this will
560             * be run on the second virtual console, and the first will
561             * be allowed to start a shell or whatever an init script
562             * specifies.
563             */
564            messageD(LOG, "Waiting for enter to start '%s'"
565                        "(pid %d, terminal %s)\n",
566                      cmdpath, getpid(), a->terminal);
567            bb_full_write(1, press_enter, sizeof(press_enter) - 1);
568            while(read(0, &c, 1) == 1 && c != '\n')
569                ;
570        }
571#endif
572
573        /* Log the process name and args */
574        message(LOG, "Starting pid %d, console %s: '%s'",
575                  getpid(), a->terminal, cmdpath);
576
577#if defined CONFIG_FEATURE_INIT_COREDUMPS
578        {
579            struct stat sb;
580            if (stat(CORE_ENABLE_FLAG_FILE, &sb) == 0) {
581                struct rlimit limit;
582
583                limit.rlim_cur = RLIM_INFINITY;
584                limit.rlim_max = RLIM_INFINITY;
585                setrlimit(RLIMIT_CORE, &limit);
586            }
587        }
588#endif
589
590        /* Now run it.  The new program will take over this PID,
591         * so nothing further in init.c should be run. */
592        execv(cmdpath, cmd);
593
594        /* We're still here?  Some error happened. */
595        message(LOG | CONSOLE, "Bummer, could not run '%s': %m", cmdpath);
596        _exit(-1);
597    }
598    sigprocmask(SIG_SETMASK, &omask, NULL);
599    return pid;
600}
601
602static int waitfor(const struct init_action *a, pid_t pid)
603{
604    int runpid;
605    int status, wpid;
606
607    runpid = (NULL == a)? pid : run(a);
608    while (1) {
609        wpid = waitpid(runpid,&status,0);
610        if (wpid == runpid)
611            break;
612        if (wpid == -1 && errno == ECHILD) {
613            /* we missed its termination */
614            break;
615        }
616        /* FIXME other errors should maybe trigger an error, but allow
617         * the program to continue */
618    }
619    return wpid;
620}
621
622/* Run all commands of a particular type */
623static void run_actions(int action)
624{
625    struct init_action *a, *tmp;
626
627    for (a = init_action_list; a; a = tmp) {
628        tmp = a->next;
629        if (a->action == action) {
630            if (access(a->terminal, R_OK | W_OK)) {
631                delete_init_action(a);
632            } else if (a->action & (SYSINIT | WAIT | CTRLALTDEL | SHUTDOWN | RESTART)) {
633                waitfor(a, 0);
634                delete_init_action(a);
635            } else if (a->action & ONCE) {
636                run(a);
637                delete_init_action(a);
638            } else if (a->action & (RESPAWN | ASKFIRST)) {
639                /* Only run stuff with pid==0.  If they have
640                 * a pid, that means it is still running */
641                if (a->pid == 0) {
642                    a->pid = run(a);
643                }
644            }
645        }
646    }
647}
648
649#if !ENABLE_DEBUG_INIT
650static void init_reboot(unsigned long magic)
651{
652    pid_t pid;
653    /* We have to fork here, since the kernel calls do_exit(0) in
654     * linux/kernel/sys.c, which can cause the machine to panic when
655     * the init process is killed.... */
656    if ((pid = fork()) == 0) {
657        reboot(magic);
658        _exit(0);
659    }
660    waitpid (pid, NULL, 0);
661}
662
663static void shutdown_system(void)
664{
665    sigset_t block_signals;
666
667    /* run everything to be run at "shutdown".  This is done _prior_
668     * to killing everything, in case people wish to use scripts to
669     * shut things down gracefully... */
670    run_actions(SHUTDOWN);
671
672    /* first disable all our signals */
673    sigemptyset(&block_signals);
674    sigaddset(&block_signals, SIGHUP);
675    sigaddset(&block_signals, SIGQUIT);
676    sigaddset(&block_signals, SIGCHLD);
677    sigaddset(&block_signals, SIGUSR1);
678    sigaddset(&block_signals, SIGUSR2);
679    sigaddset(&block_signals, SIGINT);
680    sigaddset(&block_signals, SIGTERM);
681    sigaddset(&block_signals, SIGCONT);
682    sigaddset(&block_signals, SIGSTOP);
683    sigaddset(&block_signals, SIGTSTP);
684    sigprocmask(SIG_BLOCK, &block_signals, NULL);
685
686    /* Allow Ctrl-Alt-Del to reboot system. */
687    init_reboot(RB_ENABLE_CAD);
688
689    message(CONSOLE | LOG, "The system is going down NOW !!");
690    sync();
691
692    /* Send signals to every process _except_ pid 1 */
693    message(CONSOLE | LOG, init_sending_format, "TERM");
694    kill(-1, SIGTERM);
695    sleep(1);
696    sync();
697
698    message(CONSOLE | LOG, init_sending_format, "KILL");
699    kill(-1, SIGKILL);
700    sleep(1);
701
702    sync();
703}
704
705static void exec_signal(int sig ATTRIBUTE_UNUSED)
706{
707    struct init_action *a, *tmp;
708    sigset_t unblock_signals;
709
710    for (a = init_action_list; a; a = tmp) {
711        tmp = a->next;
712        if (a->action & RESTART) {
713            shutdown_system();
714
715            /* unblock all signals, blocked in shutdown_system() */
716            sigemptyset(&unblock_signals);
717            sigaddset(&unblock_signals, SIGHUP);
718            sigaddset(&unblock_signals, SIGQUIT);
719            sigaddset(&unblock_signals, SIGCHLD);
720            sigaddset(&unblock_signals, SIGUSR1);
721            sigaddset(&unblock_signals, SIGUSR2);
722            sigaddset(&unblock_signals, SIGINT);
723            sigaddset(&unblock_signals, SIGTERM);
724            sigaddset(&unblock_signals, SIGCONT);
725            sigaddset(&unblock_signals, SIGSTOP);
726            sigaddset(&unblock_signals, SIGTSTP);
727            sigprocmask(SIG_UNBLOCK, &unblock_signals, NULL);
728
729            /* Close whatever files are open. */
730            close(0);
731            close(1);
732            close(2);
733
734            /* Open the new terminal device */
735            open_new_terminal(a->terminal, 0);
736
737            /* Make sure the terminal will act fairly normal for us */
738            set_term();
739            /* Setup stdout, stderr on the supplied terminal */
740            dup(0);
741            dup(0);
742
743            messageD(CONSOLE | LOG, "Trying to re-exec %s", a->command);
744            execl(a->command, a->command, NULL);
745
746            message(CONSOLE | LOG, "exec of '%s' failed: %m",
747                    a->command);
748            sync();
749            sleep(2);
750            init_reboot(RB_HALT_SYSTEM);
751            loop_forever();
752        }
753    }
754}
755
756static void shutdown_signal(int sig)
757{
758    char *m;
759    int rb;
760
761    shutdown_system();
762
763    if (sig == SIGTERM) {
764        m = "reboot";
765        rb = RB_AUTOBOOT;
766    } else if (sig == SIGUSR2) {
767        m = "poweroff";
768        rb = RB_POWER_OFF;
769    } else {
770        m = "halt";
771        rb = RB_HALT_SYSTEM;
772    }
773    message(CONSOLE | LOG, "Requesting system %s.", m);
774    sync();
775
776    /* allow time for last message to reach serial console */
777    sleep(2);
778
779    init_reboot(rb);
780
781    loop_forever();
782}
783
784static void ctrlaltdel_signal(int sig ATTRIBUTE_UNUSED)
785{
786    run_actions(CTRLALTDEL);
787}
788
789/* The SIGSTOP & SIGTSTP handler */
790static void stop_handler(int sig ATTRIBUTE_UNUSED)
791{
792    int saved_errno = errno;
793
794    got_cont = 0;
795    while (!got_cont)
796        pause();
797    got_cont = 0;
798    errno = saved_errno;
799}
800
801/* The SIGCONT handler */
802static void cont_handler(int sig ATTRIBUTE_UNUSED)
803{
804    got_cont = 1;
805}
806
807#endif                          /* ! ENABLE_DEBUG_INIT */
808
809static void new_init_action(int action, const char *command, const char *cons)
810{
811    struct init_action *new_action, *a, *last;
812
813    if (*cons == '\0')
814        cons = console;
815
816    if (strcmp(cons, bb_dev_null) == 0 && (action & ASKFIRST))
817        return;
818
819    new_action = calloc((size_t) (1), sizeof(struct init_action));
820    if (!new_action) {
821        message(LOG | CONSOLE, "Memory allocation failure");
822        loop_forever();
823    }
824
825    /* Append to the end of the list */
826    for (a = last = init_action_list; a; a = a->next) {
827        /* don't enter action if it's already in the list,
828         * but do overwrite existing actions */
829        if ((strcmp(a->command, command) == 0) &&
830            (strcmp(a->terminal, cons) ==0)) {
831            a->action = action;
832            free(new_action);
833            return;
834        }
835        last = a;
836    }
837    if (last) {
838        last->next = new_action;
839    } else {
840        init_action_list = new_action;
841    }
842    strcpy(new_action->command, command);
843    new_action->action = action;
844    strcpy(new_action->terminal, cons);
845#if 0   /* calloc zeroed always */
846    new_action->pid = 0;
847#endif
848    messageD(LOG|CONSOLE, "command='%s' action='%d' terminal='%s'\n",
849        new_action->command, new_action->action, new_action->terminal);
850}
851
852static void delete_init_action(struct init_action *action)
853{
854    struct init_action *a, *b = NULL;
855
856    for (a = init_action_list; a; b = a, a = a->next) {
857        if (a == action) {
858            if (b == NULL) {
859                init_action_list = a->next;
860            } else {
861                b->next = a->next;
862            }
863            free(a);
864            break;
865        }
866    }
867}
868
869/* NOTE that if CONFIG_FEATURE_USE_INITTAB is NOT defined,
870 * then parse_inittab() simply adds in some default
871 * actions(i.e., runs INIT_SCRIPT and then starts a pair
872 * of "askfirst" shells).  If CONFIG_FEATURE_USE_INITTAB
873 * _is_ defined, but /etc/inittab is missing, this
874 * results in the same set of default behaviors.
875 */
876static void parse_inittab(void)
877{
878#ifdef CONFIG_FEATURE_USE_INITTAB
879    FILE *file;
880    char buf[INIT_BUFFS_SIZE], lineAsRead[INIT_BUFFS_SIZE];
881    char tmpConsole[CONSOLE_BUFF_SIZE];
882    char *id, *runlev, *action, *command, *eol;
883    const struct init_action_type *a = actions;
884
885
886    file = fopen(INITTAB, "r");
887    if (file == NULL) {
888        /* No inittab file -- set up some default behavior */
889#endif
890        /* Reboot on Ctrl-Alt-Del */
891        new_init_action(CTRLALTDEL, "/sbin/reboot", "");
892        /* Umount all filesystems on halt/reboot */
893        new_init_action(SHUTDOWN, "/bin/umount -a -r", "");
894        /* Swapoff on halt/reboot */
895        if(ENABLE_SWAPONOFF) new_init_action(SHUTDOWN, "/sbin/swapoff -a", "");
896        /* Prepare to restart init when a HUP is received */
897        new_init_action(RESTART, "/sbin/init", "");
898        /* Askfirst shell on tty1-4 */
899        new_init_action(ASKFIRST, bb_default_login_shell, "");
900        new_init_action(ASKFIRST, bb_default_login_shell, VC_2);
901        new_init_action(ASKFIRST, bb_default_login_shell, VC_3);
902        new_init_action(ASKFIRST, bb_default_login_shell, VC_4);
903        /* sysinit */
904        new_init_action(SYSINIT, INIT_SCRIPT, "");
905
906        return;
907#ifdef CONFIG_FEATURE_USE_INITTAB
908    }
909
910    while (fgets(buf, INIT_BUFFS_SIZE, file) != NULL) {
911        /* Skip leading spaces */
912        for (id = buf; *id == ' ' || *id == '\t'; id++);
913
914        /* Skip the line if it's a comment */
915        if (*id == '#' || *id == '\n')
916            continue;
917
918        /* Trim the trailing \n */
919        eol = strrchr(id, '\n');
920        if (eol != NULL)
921            *eol = '\0';
922
923        /* Keep a copy around for posterity's sake (and error msgs) */
924        strcpy(lineAsRead, buf);
925
926        /* Separate the ID field from the runlevels */
927        runlev = strchr(id, ':');
928        if (runlev == NULL || *(runlev + 1) == '\0') {
929            message(LOG | CONSOLE, "Bad inittab entry: %s", lineAsRead);
930            continue;
931        } else {
932            *runlev = '\0';
933            ++runlev;
934        }
935
936        /* Separate the runlevels from the action */
937        action = strchr(runlev, ':');
938        if (action == NULL || *(action + 1) == '\0') {
939            message(LOG | CONSOLE, "Bad inittab entry: %s", lineAsRead);
940            continue;
941        } else {
942            *action = '\0';
943            ++action;
944        }
945
946        /* Separate the action from the command */
947        command = strchr(action, ':');
948        if (command == NULL || *(command + 1) == '\0') {
949            message(LOG | CONSOLE, "Bad inittab entry: %s", lineAsRead);
950            continue;
951        } else {
952            *command = '\0';
953            ++command;
954        }
955
956        /* Ok, now process it */
957        for (a = actions; a->name != 0; a++) {
958            if (strcmp(a->name, action) == 0) {
959                if (*id != '\0') {
960                    if(strncmp(id, "/dev/", 5) == 0)
961                        id += 5;
962                    strcpy(tmpConsole, "/dev/");
963                    safe_strncpy(tmpConsole + 5, id,
964                        CONSOLE_BUFF_SIZE - 5);
965                    id = tmpConsole;
966                }
967                new_init_action(a->action, command, id);
968                break;
969            }
970        }
971        if (a->name == 0) {
972            /* Choke on an unknown action */
973            message(LOG | CONSOLE, "Bad inittab entry: %s", lineAsRead);
974        }
975    }
976    fclose(file);
977    return;
978#endif                          /* CONFIG_FEATURE_USE_INITTAB */
979}
980
981#ifdef CONFIG_FEATURE_USE_INITTAB
982static void reload_signal(int sig ATTRIBUTE_UNUSED)
983{
984    struct init_action *a, *tmp;
985
986    message(LOG, "Reloading /etc/inittab");
987
988    /* disable old entrys */
989    for (a = init_action_list; a; a = a->next ) {
990        a->action = ONCE;
991    }
992
993    parse_inittab();
994
995    /* remove unused entrys */
996    for (a = init_action_list; a; a = tmp) {
997        tmp = a->next;
998        if (a->action & (ONCE | SYSINIT | WAIT ) &&
999                a->pid == 0 ) {
1000            delete_init_action(a);
1001        }
1002    }
1003    run_actions(RESPAWN);
1004    return;
1005}
1006#endif                          /* CONFIG_FEATURE_USE_INITTAB */
1007
1008int init_main(int argc, char **argv)
1009{
1010    struct init_action *a;
1011    pid_t wpid;
1012
1013    if (argc > 1 && !strcmp(argv[1], "-q")) {
1014        return kill(1,SIGHUP);
1015    }
1016#if !ENABLE_DEBUG_INIT
1017    /* Expect to be invoked as init with PID=1 or be invoked as linuxrc */
1018    if (getpid() != 1 &&
1019        (!ENABLE_FEATURE_INITRD || !strstr(bb_applet_name, "linuxrc")))
1020    {
1021        bb_show_usage();
1022    }
1023    /* Set up sig handlers  -- be sure to
1024     * clear all of these in run() */
1025    signal(SIGHUP, exec_signal);
1026    signal(SIGQUIT, exec_signal);
1027    signal(SIGUSR1, shutdown_signal);
1028    signal(SIGUSR2, shutdown_signal);
1029    signal(SIGINT, ctrlaltdel_signal);
1030    signal(SIGTERM, shutdown_signal);
1031    signal(SIGCONT, cont_handler);
1032    signal(SIGSTOP, stop_handler);
1033    signal(SIGTSTP, stop_handler);
1034
1035    /* Turn off rebooting via CTL-ALT-DEL -- we get a
1036     * SIGINT on CAD so we can shut things down gracefully... */
1037    init_reboot(RB_DISABLE_CAD);
1038#endif
1039
1040    /* Figure out where the default console should be */
1041    console_init();
1042
1043    /* Close whatever files are open, and reset the console. */
1044    close(0);
1045    close(1);
1046    close(2);
1047
1048    if (device_open(console, O_RDWR | O_NOCTTY) == 0) {
1049        set_term();
1050        close(0);
1051    }
1052
1053    chdir("/");
1054    setsid();
1055    {
1056        const char * const *e;
1057        /* Make sure environs is set to something sane */
1058        for(e = environment; *e; e++)
1059            putenv((char *) *e);
1060    }
1061    /* Hello world */
1062    message(MAYBE_CONSOLE | LOG, "init started:  %s", bb_msg_full_version);
1063
1064    /* Make sure there is enough memory to do something useful. */
1065    if (ENABLE_SWAPONOFF) {
1066        struct sysinfo info;
1067
1068        if (!sysinfo(&info) &&
1069            (info.mem_unit ? : 1) * (long long)info.totalram < MEGABYTE)
1070        {
1071            message(CONSOLE,"Low memory: forcing swapon.");
1072            /* swapon -a requires /proc typically */
1073            new_init_action(SYSINIT, "/bin/mount -t proc proc /proc", "");
1074            /* Try to turn on swap */
1075            new_init_action(SYSINIT, "/sbin/swapon -a", "");
1076            run_actions(SYSINIT);   /* wait and removing */
1077        }
1078    }
1079
1080    /* Check if we are supposed to be in single user mode */
1081    if (argc > 1 && (!strcmp(argv[1], "single") ||
1082                     !strcmp(argv[1], "-s") || !strcmp(argv[1], "1"))) {
1083        /* Start a shell on console */
1084        new_init_action(RESPAWN, bb_default_login_shell, "");
1085    } else {
1086        /* Not in single user mode -- see what inittab says */
1087
1088        /* NOTE that if CONFIG_FEATURE_USE_INITTAB is NOT defined,
1089         * then parse_inittab() simply adds in some default
1090         * actions(i.e., runs INIT_SCRIPT and then starts a pair
1091         * of "askfirst" shells */
1092        parse_inittab();
1093    }
1094
1095#ifdef CONFIG_SELINUX
1096    if (getenv("SELINUX_INIT") == NULL) {
1097        int enforce = 0;
1098
1099        putenv("SELINUX_INIT=YES");
1100        if (selinux_init_load_policy(&enforce) == 0) {
1101            execv(argv[0], argv);
1102        } else if (enforce > 0) {
1103            /* SELinux in enforcing mode but load_policy failed */
1104            /* At this point, we probably can't open /dev/console, so log() won't work */
1105            message(CONSOLE,"Unable to load SELinux Policy. Machine is in enforcing mode. Halting now.");
1106            exit(1);
1107        }
1108    }
1109#endif /* CONFIG_SELINUX */
1110
1111    /* Make the command line just say "init"  -- thats all, nothing else */
1112    fixup_argv(argc, argv, "init");
1113
1114    /* Now run everything that needs to be run */
1115
1116    /* First run the sysinit command */
1117    run_actions(SYSINIT);
1118
1119    /* Next run anything that wants to block */
1120    run_actions(WAIT);
1121
1122    /* Next run anything to be run only once */
1123    run_actions(ONCE);
1124
1125#ifdef CONFIG_FEATURE_USE_INITTAB
1126    /* Redefine SIGHUP to reread /etc/inittab */
1127    signal(SIGHUP, reload_signal);
1128#else
1129    signal(SIGHUP, SIG_IGN);
1130#endif /* CONFIG_FEATURE_USE_INITTAB */
1131
1132
1133    /* Now run the looping stuff for the rest of forever */
1134    while (1) {
1135        /* run the respawn stuff */
1136        run_actions(RESPAWN);
1137
1138        /* run the askfirst stuff */
1139        run_actions(ASKFIRST);
1140
1141        /* Don't consume all CPU time -- sleep a bit */
1142        sleep(1);
1143
1144        /* Wait for a child process to exit */
1145        wpid = wait(NULL);
1146        while (wpid > 0) {
1147            /* Find out who died and clean up their corpse */
1148            for (a = init_action_list; a; a = a->next) {
1149                if (a->pid == wpid) {
1150                    /* Set the pid to 0 so that the process gets
1151                     * restarted by run_actions() */
1152                    a->pid = 0;
1153                    message(LOG, "Process '%s' (pid %d) exited.  "
1154                            "Scheduling it for restart.",
1155                            a->command, wpid);
1156                }
1157            }
1158            /* see if anyone else is waiting to be reaped */
1159            wpid = waitpid (-1, NULL, WNOHANG);
1160        }
1161    }
1162}
Note: See TracBrowser for help on using the repository browser.