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