e6581880c18ecd381056b029027dd86bb97d0478
[oweals/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                         // Pointless: run() will error out if open of device fails.
517                         ///* a->terminal of "" means "init's console" */
518                         //if (a->terminal[0] && access(a->terminal, R_OK | W_OK)) {
519                         //      //message(L_LOG | L_CONSOLE, "Device %s cannot be opened in RW mode", a->terminal /*, strerror(errno)*/);
520                         //      delete_init_action(a);
521                         //} else
522                         if (a->action & (SYSINIT | WAIT | CTRLALTDEL | SHUTDOWN | RESTART)) {
523                                 waitfor(run(a));
524                                 delete_init_action(a);
525                         } else if (a->action & ONCE) {
526                                 run(a);
527                                 delete_init_action(a);
528                         } else if (a->action & (RESPAWN | ASKFIRST)) {
529                                 /* Only run stuff with pid==0.  If they have
530                                  * a pid, that means it is still running */
531                                 if (a->pid == 0) {
532                                         a->pid = run(a);
533                                 }
534                         }
535                 }
536         }
537 }
538
539 #if !ENABLE_DEBUG_INIT
540 static void init_reboot(unsigned long magic)
541 {
542         pid_t pid;
543         /* We have to fork here, since the kernel calls do_exit(0) in
544          * linux/kernel/sys.c, which can cause the machine to panic when
545          * the init process is killed.... */
546         pid = vfork();
547         if (pid == 0) { /* child */
548                 reboot(magic);
549                 _exit(0);
550         }
551         waitpid(pid, NULL, 0);
552 }
553
554 static void shutdown_system(void)
555 {
556         sigset_t block_signals;
557
558         /* run everything to be run at "shutdown".  This is done _prior_
559          * to killing everything, in case people wish to use scripts to
560          * shut things down gracefully... */
561         run_actions(SHUTDOWN);
562
563         /* first disable all our signals */
564         sigemptyset(&block_signals);
565         sigaddset(&block_signals, SIGHUP);
566         sigaddset(&block_signals, SIGQUIT);
567         sigaddset(&block_signals, SIGCHLD);
568         sigaddset(&block_signals, SIGUSR1);
569         sigaddset(&block_signals, SIGUSR2);
570         sigaddset(&block_signals, SIGINT);
571         sigaddset(&block_signals, SIGTERM);
572         sigaddset(&block_signals, SIGCONT);
573         sigaddset(&block_signals, SIGSTOP);
574         sigaddset(&block_signals, SIGTSTP);
575         sigprocmask(SIG_BLOCK, &block_signals, NULL);
576
577         message(L_CONSOLE | L_LOG, "The system is going down NOW!");
578
579         /* Allow Ctrl-Alt-Del to reboot system. */
580         init_reboot(RB_ENABLE_CAD);
581
582         /* Send signals to every process _except_ pid 1 */
583         message(L_CONSOLE | L_LOG, "Sending SIG%s to all processes", "TERM");
584         kill(-1, SIGTERM);
585         sync();
586         sleep(1);
587
588         message(L_CONSOLE | L_LOG, "Sending SIG%s to all processes", "KILL");
589         kill(-1, SIGKILL);
590         sync();
591         sleep(1);
592 }
593
594 static void shutdown_signal(int sig)
595 {
596         const char *m;
597         int rb;
598
599         shutdown_system();
600
601         m = "halt";
602         rb = RB_HALT_SYSTEM;
603         if (sig == SIGTERM) {
604                 m = "reboot";
605                 rb = RB_AUTOBOOT;
606         } else if (sig == SIGUSR2) {
607                 m = "poweroff";
608                 rb = RB_POWER_OFF;
609         }
610         message(L_CONSOLE | L_LOG, "Requesting system %s", m);
611         /* allow time for last message to reach serial console */
612         sleep(2);
613         init_reboot(rb);
614         loop_forever();
615 }
616
617 static void exec_signal(int sig ATTRIBUTE_UNUSED)
618 {
619         struct init_action *a, *tmp;
620         sigset_t unblock_signals;
621
622         for (a = init_action_list; a; a = tmp) {
623                 tmp = a->next;
624                 if (a->action & RESTART) {
625                         shutdown_system();
626
627                         /* unblock all signals, blocked in shutdown_system() */
628                         sigemptyset(&unblock_signals);
629                         sigaddset(&unblock_signals, SIGHUP);
630                         sigaddset(&unblock_signals, SIGQUIT);
631                         sigaddset(&unblock_signals, SIGCHLD);
632                         sigaddset(&unblock_signals, SIGUSR1);
633                         sigaddset(&unblock_signals, SIGUSR2);
634                         sigaddset(&unblock_signals, SIGINT);
635                         sigaddset(&unblock_signals, SIGTERM);
636                         sigaddset(&unblock_signals, SIGCONT);
637                         sigaddset(&unblock_signals, SIGSTOP);
638                         sigaddset(&unblock_signals, SIGTSTP);
639                         sigprocmask(SIG_UNBLOCK, &unblock_signals, NULL);
640
641                         /* Open the new terminal device */
642                         open_stdio_to_tty(a->terminal, 0 /* - shutdown_signal(SIGUSR1) [halt] if open fails */);
643
644                         messageD(L_CONSOLE | L_LOG, "Trying to re-exec %s", a->command);
645                         BB_EXECLP(a->command, a->command, NULL);
646
647                         message(L_CONSOLE | L_LOG, "Cannot run '%s': %s",
648                                         a->command, strerror(errno));
649                         sleep(2);
650                         init_reboot(RB_HALT_SYSTEM);
651                         loop_forever();
652                 }
653         }
654 }
655
656 static void ctrlaltdel_signal(int sig ATTRIBUTE_UNUSED)
657 {
658         run_actions(CTRLALTDEL);
659 }
660
661 /* The SIGSTOP & SIGTSTP handler */
662 static void stop_handler(int sig ATTRIBUTE_UNUSED)
663 {
664         int saved_errno = errno;
665
666         got_cont = 0;
667         while (!got_cont)
668                 pause();
669
670         errno = saved_errno;
671 }
672
673 /* The SIGCONT handler */
674 static void cont_handler(int sig ATTRIBUTE_UNUSED)
675 {
676         got_cont = 1;
677 }
678
679 #endif  /* !ENABLE_DEBUG_INIT */
680
681 static void new_init_action(uint8_t action, const char *command, const char *cons)
682 {
683         struct init_action *new_action, *a, *last;
684
685         if (strcmp(cons, bb_dev_null) == 0 && (action & ASKFIRST))
686                 return;
687
688         /* Append to the end of the list */
689         for (a = last = init_action_list; a; a = a->next) {
690                 /* don't enter action if it's already in the list,
691                  * but do overwrite existing actions */
692                 if ((strcmp(a->command, command) == 0)
693                  && (strcmp(a->terminal, cons) == 0)
694                 ) {
695                         a->action = action;
696                         return;
697                 }
698                 last = a;
699         }
700
701         new_action = xzalloc(sizeof(struct init_action));
702         if (last) {
703                 last->next = new_action;
704         } else {
705                 init_action_list = new_action;
706         }
707         strcpy(new_action->command, command);
708         new_action->action = action;
709         strcpy(new_action->terminal, cons);
710         messageD(L_LOG | L_CONSOLE, "command='%s' action=%d tty='%s'\n",
711                 new_action->command, new_action->action, new_action->terminal);
712 }
713
714 static void delete_init_action(struct init_action *action)
715 {
716         struct init_action *a, *b = NULL;
717
718         for (a = init_action_list; a; b = a, a = a->next) {
719                 if (a == action) {
720                         if (b == NULL) {
721                                 init_action_list = a->next;
722                         } else {
723                                 b->next = a->next;
724                         }
725                         free(a);
726                         break;
727                 }
728         }
729 }
730
731 /* NOTE that if CONFIG_FEATURE_USE_INITTAB is NOT defined,
732  * then parse_inittab() simply adds in some default
733  * actions(i.e., runs INIT_SCRIPT and then starts a pair
734  * of "askfirst" shells).  If CONFIG_FEATURE_USE_INITTAB
735  * _is_ defined, but /etc/inittab is missing, this
736  * results in the same set of default behaviors.
737  */
738 static void parse_inittab(void)
739 {
740 #if ENABLE_FEATURE_USE_INITTAB
741         FILE *file;
742         char buf[INIT_BUFFS_SIZE];
743
744         file = fopen(INITTAB, "r");
745         if (file == NULL) {
746                 /* No inittab file -- set up some default behavior */
747 #endif
748                 /* Reboot on Ctrl-Alt-Del */
749                 new_init_action(CTRLALTDEL, "reboot", "");
750                 /* Umount all filesystems on halt/reboot */
751                 new_init_action(SHUTDOWN, "umount -a -r", "");
752                 /* Swapoff on halt/reboot */
753                 if (ENABLE_SWAPONOFF)
754                         new_init_action(SHUTDOWN, "swapoff -a", "");
755                 /* Prepare to restart init when a HUP is received */
756                 new_init_action(RESTART, "init", "");
757                 /* Askfirst shell on tty1-4 */
758                 new_init_action(ASKFIRST, bb_default_login_shell, "");
759                 new_init_action(ASKFIRST, bb_default_login_shell, VC_2);
760                 new_init_action(ASKFIRST, bb_default_login_shell, VC_3);
761                 new_init_action(ASKFIRST, bb_default_login_shell, VC_4);
762                 /* sysinit */
763                 new_init_action(SYSINIT, INIT_SCRIPT, "");
764
765                 return;
766 #if ENABLE_FEATURE_USE_INITTAB
767         }
768
769         while (fgets(buf, INIT_BUFFS_SIZE, file) != NULL) {
770                 static const char actions[] =
771                         STR_SYSINIT    "sysinit\0"
772                         STR_RESPAWN    "respawn\0"
773                         STR_ASKFIRST   "askfirst\0"
774                         STR_WAIT       "wait\0"
775                         STR_ONCE       "once\0"
776                         STR_CTRLALTDEL "ctrlaltdel\0"
777                         STR_SHUTDOWN   "shutdown\0"
778                         STR_RESTART    "restart\0"
779                 ;
780                 char tmpConsole[CONSOLE_NAME_SIZE];
781                 char *id, *runlev, *action, *command;
782                 const char *a;
783
784                 /* Skip leading spaces */
785                 id = skip_whitespace(buf);
786                 /* Skip the line if it's a comment */
787                 if (*id == '#' || *id == '\n')
788                         continue;
789                 /* Trim the trailing '\n' */
790                 *strchrnul(id, '\n') = '\0';
791
792                 /* Line is: "id:runlevel_ignored:action:command" */
793                 runlev = strchr(id, ':');
794                 if (runlev == NULL /*|| runlev[1] == '\0' - not needed */)
795                         goto bad_entry;
796                 action = strchr(runlev + 1, ':');
797                 if (action == NULL /*|| action[1] == '\0' - not needed */)
798                         goto bad_entry;
799                 command = strchr(action + 1, ':');
800                 if (command == NULL || command[1] == '\0')
801                         goto bad_entry;
802
803                 *command = '\0'; /* action => ":action\0" now */
804                 for (a = actions; a[0]; a += strlen(a) + 1) {
805                         if (strcmp(a + 1, action + 1) == 0) {
806                                 *runlev = '\0';
807                                 if (*id != '\0') {
808                                         if (strncmp(id, "/dev/", 5) == 0)
809                                                 id += 5;
810                                         strcpy(tmpConsole, "/dev/");
811                                         safe_strncpy(tmpConsole + 5, id,
812                                                 sizeof(tmpConsole) - 5);
813                                         id = tmpConsole;
814                                 }
815                                 new_init_action((uint8_t)a[0], command + 1, id);
816                                 goto next_line;
817                         }
818                 }
819                 *command = ':';
820                 /* Choke on an unknown action */
821  bad_entry:
822                 message(L_LOG | L_CONSOLE, "Bad inittab entry: %s", id);
823  next_line: ;
824         }
825         fclose(file);
826 #endif /* FEATURE_USE_INITTAB */
827 }
828
829 #if ENABLE_FEATURE_USE_INITTAB
830 static void reload_signal(int sig ATTRIBUTE_UNUSED)
831 {
832         struct init_action *a, *tmp;
833
834         message(L_LOG, "reloading /etc/inittab");
835
836         /* disable old entrys */
837         for (a = init_action_list; a; a = a->next) {
838                 a->action = ONCE;
839         }
840
841         parse_inittab();
842
843 #if ENABLE_FEATURE_KILL_REMOVED
844         for (a = init_action_list; a; a = a->next) {
845                 pid_t pid = a->pid;
846                 if ((a->action & ONCE) && pid != 0) {
847                         /* Be nice and send SIGTERM first */
848                         kill(pid, SIGTERM);
849 #if CONFIG_FEATURE_KILL_DELAY
850                         if (fork() == 0) { /* child */
851                                 sleep(CONFIG_FEATURE_KILL_DELAY);
852                                 kill(pid, SIGKILL);
853                                 _exit(0);
854                         }
855 #endif
856                 }
857         }
858 #endif /* FEATURE_KILL_REMOVED */
859
860         /* remove unused entrys */
861         for (a = init_action_list; a; a = tmp) {
862                 tmp = a->next;
863                 if ((a->action & (ONCE | SYSINIT | WAIT)) && a->pid == 0) {
864                         delete_init_action(a);
865                 }
866         }
867         run_actions(RESPAWN);
868 }
869 #endif  /* FEATURE_USE_INITTAB */
870
871 int init_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
872 int init_main(int argc, char **argv)
873 {
874         struct init_action *a;
875         pid_t wpid;
876
877         die_sleep = 30 * 24*60*60; /* if xmalloc will ever die... */
878
879         if (argc > 1 && !strcmp(argv[1], "-q")) {
880                 return kill(1, SIGHUP);
881         }
882 #if !ENABLE_DEBUG_INIT
883         /* Expect to be invoked as init with PID=1 or be invoked as linuxrc */
884         if (getpid() != 1
885          && (!ENABLE_FEATURE_INITRD || !strstr(applet_name, "linuxrc"))
886         ) {
887                 bb_show_usage();
888         }
889         /* Set up sig handlers  -- be sure to
890          * clear all of these in run() */
891         signal(SIGHUP, exec_signal);
892         signal(SIGQUIT, exec_signal);
893         signal(SIGUSR1, shutdown_signal);
894         signal(SIGUSR2, shutdown_signal);
895         signal(SIGINT, ctrlaltdel_signal);
896         signal(SIGTERM, shutdown_signal);
897         signal(SIGCONT, cont_handler);
898         signal(SIGSTOP, stop_handler);
899         signal(SIGTSTP, stop_handler);
900
901         /* Turn off rebooting via CTL-ALT-DEL -- we get a
902          * SIGINT on CAD so we can shut things down gracefully... */
903         init_reboot(RB_DISABLE_CAD);
904 #endif
905
906
907         /* Figure out where the default console should be */
908         console_init();
909         set_sane_term();
910         chdir("/");
911         setsid();
912         {
913                 const char *const *e;
914                 /* Make sure environs is set to something sane */
915                 for (e = environment; *e; e++)
916                         putenv((char *) *e);
917         }
918
919         if (argc > 1) setenv("RUNLEVEL", argv[1], 1);
920
921         /* Hello world */
922         message(MAYBE_CONSOLE | L_LOG, "init started: %s", bb_banner);
923
924         /* Make sure there is enough memory to do something useful. */
925         if (ENABLE_SWAPONOFF) {
926                 struct sysinfo info;
927
928                 if (!sysinfo(&info) &&
929                         (info.mem_unit ? : 1) * (long long)info.totalram < 1024*1024)
930                 {
931                         message(L_CONSOLE, "Low memory, forcing swapon");
932                         /* swapon -a requires /proc typically */
933                         new_init_action(SYSINIT, "mount -t proc proc /proc", "");
934                         /* Try to turn on swap */
935                         new_init_action(SYSINIT, "swapon -a", "");
936                         run_actions(SYSINIT);   /* wait and removing */
937                 }
938         }
939
940         /* Check if we are supposed to be in single user mode */
941         if (argc > 1
942          && (!strcmp(argv[1], "single") || !strcmp(argv[1], "-s") || LONE_CHAR(argv[1], '1'))
943         ) {
944                 /* Start a shell on console */
945                 new_init_action(RESPAWN, bb_default_login_shell, "");
946         } else {
947                 /* Not in single user mode -- see what inittab says */
948
949                 /* NOTE that if CONFIG_FEATURE_USE_INITTAB is NOT defined,
950                  * then parse_inittab() simply adds in some default
951                  * actions(i.e., runs INIT_SCRIPT and then starts a pair
952                  * of "askfirst" shells */
953                 parse_inittab();
954         }
955
956 #if ENABLE_SELINUX
957         if (getenv("SELINUX_INIT") == NULL) {
958                 int enforce = 0;
959
960                 putenv((char*)"SELINUX_INIT=YES");
961                 if (selinux_init_load_policy(&enforce) == 0) {
962                         BB_EXECVP(argv[0], argv);
963                 } else if (enforce > 0) {
964                         /* SELinux in enforcing mode but load_policy failed */
965                         message(L_CONSOLE, "Cannot load SELinux Policy. "
966                                 "Machine is in enforcing mode. Halting now.");
967                         exit(1);
968                 }
969         }
970 #endif /* CONFIG_SELINUX */
971
972         /* Make the command line just say "init"  - thats all, nothing else */
973         strncpy(argv[0], "init", strlen(argv[0]));
974         /* Wipe argv[1]-argv[N] so they don't clutter the ps listing */
975         while (*++argv)
976                 memset(*argv, 0, strlen(*argv));
977
978         /* Now run everything that needs to be run */
979
980         /* First run the sysinit command */
981         run_actions(SYSINIT);
982
983         /* Next run anything that wants to block */
984         run_actions(WAIT);
985
986         /* Next run anything to be run only once */
987         run_actions(ONCE);
988
989 #if ENABLE_FEATURE_USE_INITTAB
990         /* Redefine SIGHUP to reread /etc/inittab */
991         signal(SIGHUP, reload_signal);
992 #else
993         signal(SIGHUP, SIG_IGN);
994 #endif /* FEATURE_USE_INITTAB */
995
996         /* Now run the looping stuff for the rest of forever */
997         while (1) {
998                 /* run the respawn stuff */
999                 run_actions(RESPAWN);
1000
1001                 /* run the askfirst stuff */
1002                 run_actions(ASKFIRST);
1003
1004                 /* Don't consume all CPU time -- sleep a bit */
1005                 sleep(1);
1006
1007                 /* Wait for a child process to exit */
1008                 wpid = wait(NULL);
1009                 while (wpid > 0) {
1010                         /* Find out who died and clean up their corpse */
1011                         for (a = init_action_list; a; a = a->next) {
1012                                 if (a->pid == wpid) {
1013                                         /* Set the pid to 0 so that the process gets
1014                                          * restarted by run_actions() */
1015                                         a->pid = 0;
1016                                         message(L_LOG, "process '%s' (pid %d) exited. "
1017                                                         "Scheduling it for restart.",
1018                                                         a->command, wpid);
1019                                 }
1020                         }
1021                         /* see if anyone else is waiting to be reaped */
1022                         wpid = waitpid(-1, NULL, WNOHANG);
1023                 }
1024         }
1025 }