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