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