Fix warning
[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                 }
236         }
237         if ((device & LOG) && (log_fd >= 0)) {
238                 va_start(arguments, fmt);
239                 vdprintf(log_fd, fmt, arguments);
240                 va_end(arguments);
241         }
242 #endif
243
244         if (device & CONSOLE) {
245                 /* Always send console messages to /dev/console so people will see them. */
246                 if ((fd =
247                          device_open(_PATH_CONSOLE,
248                                                  O_WRONLY | O_NOCTTY | O_NDELAY)) >= 0) {
249                         va_start(arguments, fmt);
250                         vdprintf(fd, fmt, arguments);
251                         va_end(arguments);
252                         close(fd);
253                 } else {
254                         fprintf(stderr, "Bummer, can't print: ");
255                         va_start(arguments, fmt);
256                         vfprintf(stderr, fmt, arguments);
257                         va_end(arguments);
258                 }
259         }
260 }
261
262 /* Set terminal settings to reasonable defaults */
263 static void set_term(int fd)
264 {
265         struct termios tty;
266
267         tcgetattr(fd, &tty);
268
269         /* set control chars */
270         tty.c_cc[VINTR] = 3;    /* C-c */
271         tty.c_cc[VQUIT] = 28;   /* C-\ */
272         tty.c_cc[VERASE] = 127; /* C-? */
273         tty.c_cc[VKILL] = 21;   /* C-u */
274         tty.c_cc[VEOF] = 4;     /* C-d */
275         tty.c_cc[VSTART] = 17;  /* C-q */
276         tty.c_cc[VSTOP] = 19;   /* C-s */
277         tty.c_cc[VSUSP] = 26;   /* C-z */
278
279         /* use line dicipline 0 */
280         tty.c_line = 0;
281
282         /* Make it be sane */
283         tty.c_cflag &= CBAUD | CBAUDEX | CSIZE | CSTOPB | PARENB | PARODD;
284         tty.c_cflag |= CREAD | HUPCL | CLOCAL;
285
286
287         /* input modes */
288         tty.c_iflag = ICRNL | IXON | IXOFF;
289
290         /* output modes */
291         tty.c_oflag = OPOST | ONLCR;
292
293         /* local modes */
294         tty.c_lflag =
295                 ISIG | ICANON | ECHO | ECHOE | ECHOK | ECHOCTL | ECHOKE | IEXTEN;
296
297         tcsetattr(fd, TCSANOW, &tty);
298 }
299
300 /* How much memory does this machine have?
301    Units are kBytes to avoid overflow on 4GB machines */
302 static int check_free_memory(void)
303 {
304         struct sysinfo info;
305         unsigned int result, u, s = 10;
306
307         if (sysinfo(&info) != 0) {
308                 perror_msg("Error checking free memory");
309                 return -1;
310         }
311
312         /* Kernels 2.0.x and 2.2.x return info.mem_unit==0 with values in bytes.
313          * Kernels 2.4.0 return info.mem_unit in bytes. */
314         u = info.mem_unit;
315         if (u == 0)
316                 u = 1;
317         while ((u & 1) == 0 && s > 0) {
318                 u >>= 1;
319                 s--;
320         }
321         result = (info.totalram >> s) + (info.totalswap >> s);
322         result = result * u;
323         if (result < 0)
324                 result = INT_MAX;
325         return result;
326 }
327
328 static void console_init(void)
329 {
330         int fd;
331         int tried_devcons = 0;
332         int tried_vtprimary = 0;
333         struct vt_stat vt;
334         struct serial_struct sr;
335         char *s;
336
337         if ((s = getenv("TERM")) != NULL) {
338                 snprintf(termType, sizeof(termType) - 1, "TERM=%s", s);
339         }
340
341         if ((s = getenv("CONSOLE")) != NULL) {
342                 safe_strncpy(console, s, sizeof(console));
343         }
344 #if #cpu(sparc)
345         /* sparc kernel supports console=tty[ab] parameter which is also 
346          * passed to init, so catch it here */
347         else if ((s = getenv("console")) != NULL) {
348                 /* remap tty[ab] to /dev/ttyS[01] */
349                 if (strcmp(s, "ttya") == 0)
350                         safe_strncpy(console, SC_0, sizeof(console));
351                 else if (strcmp(s, "ttyb") == 0)
352                         safe_strncpy(console, SC_1, sizeof(console));
353         }
354 #endif
355         else {
356                 /* 2.2 kernels: identify the real console backend and try to use it */
357                 if (ioctl(0, TIOCGSERIAL, &sr) == 0) {
358                         /* this is a serial console */
359                         snprintf(console, sizeof(console) - 1, SC_FORMAT, sr.line);
360                 } else if (ioctl(0, VT_GETSTATE, &vt) == 0) {
361                         /* this is linux virtual tty */
362                         snprintf(console, sizeof(console) - 1, VC_FORMAT, vt.v_active);
363                 } else {
364                         safe_strncpy(console, _PATH_CONSOLE, sizeof(console));
365                         tried_devcons++;
366                 }
367         }
368
369         while ((fd = open(console, O_RDONLY | O_NONBLOCK)) < 0) {
370                 /* Can't open selected console -- try /dev/console */
371                 if (!tried_devcons) {
372                         tried_devcons++;
373                         safe_strncpy(console, _PATH_CONSOLE, sizeof(console));
374                         continue;
375                 }
376                 /* Can't open selected console -- try vt1 */
377                 if (!tried_vtprimary) {
378                         tried_vtprimary++;
379                         safe_strncpy(console, VC_1, sizeof(console));
380                         continue;
381                 }
382                 break;
383         }
384         if (fd < 0) {
385                 /* Perhaps we should panic here? */
386                 safe_strncpy(console, "/dev/null", sizeof(console));
387         } else {
388                 /* check for serial console */
389                 if (ioctl(0, TIOCGSERIAL, &sr) == 0) {
390                         /* Force the TERM setting to vt102 for serial console --
391                          * if TERM is set to linux (the default) */
392                         if (strcmp(termType, "TERM=linux") == 0)
393                                 safe_strncpy(termType, "TERM=vt102", sizeof(termType));
394                 }
395                 close(fd);
396         }
397         message(LOG, "console=%s\n", console);
398 }
399
400 static void fixup_argv(int argc, char **argv, char *new_argv0)
401 {
402         int len;
403
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, "\rSorry, your computer does not have enough memory.\n");
441         loop_forever();
442 }
443
444 static pid_t run(struct init_action *a)
445 {
446         struct stat sb;
447         int i, j, junk;
448         pid_t pid, pgrp, tmp_pid;
449         char *s, *tmpCmd, *cmd[INIT_BUFFS_SIZE], *cmdpath;
450         char buf[INIT_BUFFS_SIZE + 6];  /* INIT_BUFFS_SIZE+strlen("exec ")+1 */
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                 /* Clean up */
486                 close(0);
487                 close(1);
488                 close(2);
489                 sigprocmask(SIG_SETMASK, &omask, NULL);
490
491                 /* Reset signal handlers that were set by the parent process */
492                 signal(SIGUSR1, SIG_DFL);
493                 signal(SIGUSR2, SIG_DFL);
494                 signal(SIGINT, SIG_DFL);
495                 signal(SIGTERM, SIG_DFL);
496                 signal(SIGHUP, SIG_DFL);
497                 signal(SIGCONT, SIG_DFL);
498                 signal(SIGSTOP, SIG_DFL);
499                 signal(SIGTSTP, SIG_DFL);
500
501                 /* Create a new session and make ourself the process
502                  * group leader for non-interactive jobs */
503                 if ((a->action & (RESPAWN)) == 0)
504                         setsid();
505
506                 /* Open the new terminal device */
507                 if ((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
517                 /* Non-interactive jobs should not get a controling tty */
518                 if ((a->action & (RESPAWN)) == 0)
519                         (void) ioctl(0, TIOCSCTTY, 0);
520
521                 /* Make sure the terminal will act fairly normal for us */
522                 set_term(0);
523                 /* Setup stdout, stderr for the new process so
524                  * they point to the supplied terminal */
525                 dup(0);
526                 dup(0);
527
528                 /* For interactive jobs, create a new session 
529                  * and become the process group leader */
530                 if ((a->action & (RESPAWN)))
531                         setsid();
532
533                 /* If the init Action requires us to wait, then force the
534                  * supplied terminal to be the controlling tty. */
535                 if (a->action & (SYSINIT | WAIT | CTRLALTDEL | SHUTDOWN | RESTART)) {
536
537                         /* Now fork off another process to just hang around */
538                         if ((pid = fork()) < 0) {
539                                 message(LOG | CONSOLE, "Can't fork!\n");
540                                 _exit(1);
541                         }
542
543                         if (pid > 0) {
544
545                                 /* We are the parent -- wait till the child is done */
546                                 signal(SIGINT, SIG_IGN);
547                                 signal(SIGTSTP, SIG_IGN);
548                                 signal(SIGQUIT, SIG_IGN);
549                                 signal(SIGCHLD, SIG_DFL);
550
551                                 /* Wait for child to exit */
552                                 while ((tmp_pid = waitpid(pid, &junk, 0)) != pid);
553
554                                 /* See if stealing the controlling tty back is necessary */
555                                 pgrp = tcgetpgrp(0);
556                                 if (pgrp != getpid())
557                                         _exit(0);
558
559                                 /* Use a temporary process to steal the controlling tty. */
560                                 if ((pid = fork()) < 0) {
561                                         message(LOG | CONSOLE, "\rCan't fork!\n");
562                                         _exit(1);
563                                 }
564                                 if (pid == 0) {
565                                         setsid();
566                                         ioctl(0, TIOCSCTTY, 1);
567                                         _exit(0);
568                                 }
569                                 while ((tmp_pid = waitpid(pid, &junk, 0)) != pid) {
570                                         if (tmp_pid < 0 && errno == ECHILD)
571                                                 break;
572                                 }
573                                 _exit(0);
574                         }
575
576                         /* Now fall though to actually execute things */
577                 }
578
579                 /* See if any special /bin/sh requiring characters are present */
580                 if (strpbrk(a->command, "~`!$^&*()=|\\{}[];\"'<>?") != NULL) {
581                         cmd[0] = SHELL;
582                         cmd[1] = "-c";
583                         strcat(strcpy(buf, "exec "), a->command);
584                         cmd[2] = buf;
585                         cmd[3] = NULL;
586                 } else {
587                         /* Convert command (char*) into cmd (char**, one word per string) */
588                         strcpy(buf, a->command);
589                         s = buf;
590                         for (tmpCmd = buf, i = 0; (tmpCmd = strsep(&s, " \t")) != NULL;) {
591                                 if (*tmpCmd != '\0') {
592                                         cmd[i] = tmpCmd;
593                                         i++;
594                                 }
595                         }
596                         cmd[i] = NULL;
597                 }
598
599                 cmdpath = cmd[0];
600
601                 /*
602                    Interactive shells want to see a dash in argv[0].  This
603                    typically is handled by login, argv will be setup this 
604                    way if a dash appears at the front of the command path 
605                    (like "-/bin/sh").
606                  */
607
608                 if (*cmdpath == '-') {
609
610                         /* skip over the dash */
611                         ++cmdpath;
612
613                         /* find the last component in the command pathname */
614                         s = get_last_path_component(cmdpath);
615
616                         /* make a new argv[0] */
617                         if ((cmd[0] = malloc(strlen(s) + 2)) == NULL) {
618                                 message(LOG | CONSOLE, "malloc failed");
619                                 cmd[0] = cmdpath;
620                         } else {
621                                 cmd[0][0] = '-';
622                                 strcpy(cmd[0] + 1, s);
623                         }
624                 }
625
626                 if (a->action & ASKFIRST) {
627                         /*
628                          * Save memory by not exec-ing anything large (like a shell)
629                          * before the user wants it. This is critical if swap is not
630                          * enabled and the system has low memory. Generally this will
631                          * be run on the second virtual console, and the first will
632                          * be allowed to start a shell or whatever an init script 
633                          * specifies.
634                          */
635                         messageND(LOG,
636                                           "Waiting for enter to start '%s' (pid %d, terminal %s)\n",
637                                           cmdpath, getpid(), a->terminal);
638                         write(fileno(stdout), press_enter, sizeof(press_enter) - 1);
639                         getc(stdin);
640                 }
641
642                 /* Log the process name and args */
643                 messageND(LOG, "Starting pid %d, console %s: '%s'\n",
644                                   getpid(), a->terminal, cmdpath);
645
646 #if defined CONFIG_FEATURE_INIT_COREDUMPS
647                 if (stat(CORE_ENABLE_FLAG_FILE, &sb) == 0) {
648                         struct rlimit limit;
649
650                         limit.rlim_cur = RLIM_INFINITY;
651                         limit.rlim_max = RLIM_INFINITY;
652                         setrlimit(RLIMIT_CORE, &limit);
653                 }
654 #endif
655
656                 /* Now run it.  The new program will take over this PID, 
657                  * so nothing further in init.c should be run. */
658                 execve(cmdpath, cmd, environment);
659
660                 /* We're still here?  Some error happened. */
661                 message(LOG | CONSOLE, "\rBummer, could not run '%s': %s\n", cmdpath,
662                                 strerror(errno));
663                 _exit(-1);
664         }
665         sigprocmask(SIG_SETMASK, &omask, NULL);
666         return pid;
667 }
668
669 static int waitfor(struct init_action *a)
670 {
671         int pid;
672         int status, wpid;
673
674         pid = run(a);
675         while (1) {
676                 wpid = wait(&status);
677                 if (wpid > 0 && wpid != pid) {
678                         continue;
679                 }
680                 if (wpid == pid)
681                         break;
682         }
683         return wpid;
684 }
685
686 /* Run all commands of a particular type */
687 static void run_actions(int action)
688 {
689         struct init_action *a, *tmp;
690
691         for (a = init_action_list; a; a = tmp) {
692                 tmp = a->next;
693                 if (a->action == action) {
694                         if (a->
695                                 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                         "\rThe system is halted. Press Reset or turn off power\n"
803 #endif
804                 );
805         sync();
806
807         /* allow time for last message to reach serial console */
808         sleep(2);
809
810         if (sig == SIGUSR2 && kernelVersion >= KERNEL_VERSION(2, 2, 0))
811                 init_reboot(RB_POWER_OFF);
812         else
813                 init_reboot(RB_HALT_SYSTEM);
814
815         loop_forever();
816 }
817
818 static void reboot_signal(int sig)
819 {
820         shutdown_system();
821         message(CONSOLE | LOG, "\rPlease stand by while rebooting the system.\n");
822         sync();
823
824         /* allow time for last message to reach serial console */
825         sleep(2);
826
827         init_reboot(RB_AUTOBOOT);
828
829         loop_forever();
830 }
831
832 static void ctrlaltdel_signal(int sig)
833 {
834         run_actions(CTRLALTDEL);
835 }
836
837 /* The SIGSTOP & SIGTSTP handler */
838 static void stop_handler(int sig)
839 {
840         int saved_errno = errno;
841
842         got_cont = 0;
843         while (!got_cont)
844                 pause();
845         got_cont = 0;
846         errno = saved_errno;
847 }
848
849 /* The SIGCONT handler */
850 static void cont_handler(int sig)
851 {
852         got_cont = 1;
853 }
854
855 #endif                                                  /* ! DEBUG_INIT */
856
857 static void new_init_action(int action, char *command, char *cons)
858 {
859         struct init_action *new_action, *a;
860
861         if (*cons == '\0')
862                 cons = console;
863
864         /* do not run entries if console device is not available */
865         if (access(cons, R_OK | W_OK))
866                 return;
867         if (strcmp(cons, "/dev/null") == 0 && (action & ASKFIRST))
868                 return;
869
870         new_action = calloc((size_t) (1), sizeof(struct init_action));
871         if (!new_action) {
872                 message(LOG | CONSOLE, "\rMemory allocation failure\n");
873                 loop_forever();
874         }
875
876         /* Append to the end of the list */
877         for (a = init_action_list; a && a->next; a = a->next);
878         if (a) {
879                 a->next = new_action;
880         } else {
881                 init_action_list = new_action;
882         }
883         strcpy(new_action->command, command);
884         new_action->action = action;
885         strcpy(new_action->terminal, cons);
886         new_action->pid = 0;
887 /*    message(LOG|CONSOLE, "command='%s' action='%d' terminal='%s'\n",
888                 new_action->command, new_action->action, new_action->terminal); */
889 }
890
891 static void delete_init_action(struct init_action *action)
892 {
893         struct init_action *a, *b = NULL;
894
895         for (a = init_action_list; a; b = a, a = a->next) {
896                 if (a == action) {
897                         if (b == NULL) {
898                                 init_action_list = a->next;
899                         } else {
900                                 b->next = a->next;
901                         }
902                         free(a);
903                         break;
904                 }
905         }
906 }
907
908 /* NOTE that if CONFIG_FEATURE_USE_INITTAB is NOT defined,
909  * then parse_inittab() simply adds in some default
910  * actions(i.e., runs INIT_SCRIPT and then starts a pair 
911  * of "askfirst" shells).  If CONFIG_FEATURE_USE_INITTAB 
912  * _is_ defined, but /etc/inittab is missing, this 
913  * results in the same set of default behaviors.
914  */
915 static void parse_inittab(void)
916 {
917 #ifdef CONFIG_FEATURE_USE_INITTAB
918         FILE *file;
919         char buf[INIT_BUFFS_SIZE], lineAsRead[INIT_BUFFS_SIZE],
920                 tmpConsole[INIT_BUFFS_SIZE];
921         char *id, *runlev, *action, *command, *eol;
922         const struct init_action_type *a = actions;
923         int foundIt;
924
925
926         file = fopen(INITTAB, "r");
927         if (file == NULL) {
928                 /* No inittab file -- set up some default behavior */
929 #endif
930                 /* Reboot on Ctrl-Alt-Del */
931                 new_init_action(CTRLALTDEL, "/sbin/reboot", console);
932                 /* Umount all filesystems on halt/reboot */
933                 new_init_action(SHUTDOWN, "/bin/umount -a -r", console);
934 #if !defined(__UCLIBC__) || defined(__UCLIBC_HAS_MMU__)
935                 /* Swapoff on halt/reboot */
936                 new_init_action(SHUTDOWN, "/sbin/swapoff -a", console);
937 #endif
938                 /* Prepare to restart init when a HUP is received */
939                 new_init_action(RESTART, "/sbin/init", console);
940                 /* Askfirst shell on tty1-4 */
941                 new_init_action(ASKFIRST, LOGIN_SHELL, console);
942                 new_init_action(ASKFIRST, LOGIN_SHELL, VC_2);
943                 new_init_action(ASKFIRST, LOGIN_SHELL, VC_3);
944                 new_init_action(ASKFIRST, LOGIN_SHELL, VC_4);
945                 /* sysinit */
946                 new_init_action(SYSINIT, INIT_SCRIPT, console);
947
948                 return;
949 #ifdef CONFIG_FEATURE_USE_INITTAB
950         }
951
952         while (fgets(buf, INIT_BUFFS_SIZE, file) != NULL) {
953                 foundIt = FALSE;
954                 /* Skip leading spaces */
955                 for (id = buf; *id == ' ' || *id == '\t'; id++);
956
957                 /* Skip the line if it's a comment */
958                 if (*id == '#' || *id == '\n')
959                         continue;
960
961                 /* Trim the trailing \n */
962                 eol = strrchr(id, '\n');
963                 if (eol != NULL)
964                         *eol = '\0';
965
966                 /* Keep a copy around for posterity's sake (and error msgs) */
967                 strcpy(lineAsRead, buf);
968
969                 /* Separate the ID field from the runlevels */
970                 runlev = strchr(id, ':');
971                 if (runlev == NULL || *(runlev + 1) == '\0') {
972                         message(LOG | CONSOLE, "\rBad inittab entry: %s\n", lineAsRead);
973                         continue;
974                 } else {
975                         *runlev = '\0';
976                         ++runlev;
977                 }
978
979                 /* Separate the runlevels from the action */
980                 action = strchr(runlev, ':');
981                 if (action == NULL || *(action + 1) == '\0') {
982                         message(LOG | CONSOLE, "\rBad inittab entry: %s\n", lineAsRead);
983                         continue;
984                 } else {
985                         *action = '\0';
986                         ++action;
987                 }
988
989                 /* Separate the action from the command */
990                 command = strchr(action, ':');
991                 if (command == NULL || *(command + 1) == '\0') {
992                         message(LOG | CONSOLE, "\rBad inittab entry: %s\n", lineAsRead);
993                         continue;
994                 } else {
995                         *command = '\0';
996                         ++command;
997                 }
998
999                 /* Ok, now process it */
1000                 a = actions;
1001                 while (a->name != 0) {
1002                         if (strcmp(a->name, action) == 0) {
1003                                 if (*id != '\0') {
1004                                         strcpy(tmpConsole, "/dev/");
1005                                         strncat(tmpConsole, id, INIT_BUFFS_SIZE - 6);
1006                                         id = tmpConsole;
1007                                 }
1008                                 new_init_action(a->action, command, id);
1009                                 foundIt = TRUE;
1010                         }
1011                         a++;
1012                 }
1013                 if (foundIt == TRUE)
1014                         continue;
1015                 else {
1016                         /* Choke on an unknown action */
1017                         message(LOG | CONSOLE, "\rBad inittab entry: %s\n", lineAsRead);
1018                 }
1019         }
1020         fclose(file);
1021         return;
1022 #endif                                                  /* CONFIG_FEATURE_USE_INITTAB */
1023 }
1024
1025
1026
1027 extern int init_main(int argc, char **argv)
1028 {
1029         struct init_action *a;
1030         pid_t wpid;
1031         int status;
1032
1033         if (argc > 1 && !strcmp(argv[1], "-q")) {
1034                 /* don't assume init's pid == 1 */
1035                 long *pid = find_pid_by_name("init");
1036
1037                 if (!pid || *pid <= 0) {
1038                         pid = find_pid_by_name("linuxrc");
1039                         if (!pid || *pid <= 0)
1040                                 error_msg_and_die("no process killed");
1041                 }
1042                 kill(*pid, SIGHUP);
1043                 exit(0);
1044         }
1045 #ifndef DEBUG_INIT
1046         /* Expect to be invoked as init with PID=1 or be invoked as linuxrc */
1047         if (getpid() != 1
1048 #ifdef CONFIG_FEATURE_INITRD
1049                 && strstr(applet_name, "linuxrc") == NULL
1050 #endif
1051                 ) {
1052                 show_usage();
1053         }
1054         /* Set up sig handlers  -- be sure to
1055          * clear all of these in run() */
1056         signal(SIGHUP, exec_signal);
1057         signal(SIGUSR1, halt_signal);
1058         signal(SIGUSR2, halt_signal);
1059         signal(SIGINT, ctrlaltdel_signal);
1060         signal(SIGTERM, reboot_signal);
1061         signal(SIGCONT, cont_handler);
1062         signal(SIGSTOP, stop_handler);
1063         signal(SIGTSTP, stop_handler);
1064
1065         /* Turn off rebooting via CTL-ALT-DEL -- we get a 
1066          * SIGINT on CAD so we can shut things down gracefully... */
1067         init_reboot(RB_DISABLE_CAD);
1068 #endif
1069
1070         /* Figure out what kernel this is running */
1071         kernelVersion = get_kernel_revision();
1072
1073         /* Figure out where the default console should be */
1074         console_init();
1075
1076         /* Close whatever files are open, and reset the console. */
1077         close(0);
1078         close(1);
1079         close(2);
1080
1081         if (device_open(console, O_RDWR | O_NOCTTY) == 0) {
1082                 set_term(0);
1083                 close(0);
1084         }
1085
1086         chdir("/");
1087         setsid();
1088
1089         /* Make sure PATH is set to something sane */
1090         putenv("PATH=" _PATH_STDPATH);
1091
1092         /* Hello world */
1093         message(MAYBE_CONSOLE | LOG, "\rinit started:  %s\n", full_version);
1094
1095         /* Make sure there is enough memory to do something useful. */
1096         check_memory();
1097
1098         /* Check if we are supposed to be in single user mode */
1099         if (argc > 1 && (!strcmp(argv[1], "single") ||
1100                                          !strcmp(argv[1], "-s") || !strcmp(argv[1], "1"))) {
1101                 /* Ask first then start a shell on tty2-4 */
1102                 new_init_action(ASKFIRST, LOGIN_SHELL, VC_2);
1103                 new_init_action(ASKFIRST, LOGIN_SHELL, VC_3);
1104                 new_init_action(ASKFIRST, LOGIN_SHELL, VC_4);
1105                 /* Start a shell on tty1 */
1106                 new_init_action(RESPAWN, LOGIN_SHELL, console);
1107         } else {
1108                 /* Not in single user mode -- see what inittab says */
1109
1110                 /* NOTE that if CONFIG_FEATURE_USE_INITTAB is NOT defined,
1111                  * then parse_inittab() simply adds in some default
1112                  * actions(i.e., runs INIT_SCRIPT and then starts a pair 
1113                  * of "askfirst" shells */
1114                 parse_inittab();
1115         }
1116
1117         /* Make the command line just say "init"  -- thats all, nothing else */
1118         fixup_argv(argc, argv, "init");
1119
1120         /* Now run everything that needs to be run */
1121
1122         /* First run the sysinit command */
1123         run_actions(SYSINIT);
1124
1125         /* Next run anything that wants to block */
1126         run_actions(WAIT);
1127
1128         /* Next run anything to be run only once */
1129         run_actions(ONCE);
1130
1131         /* If there is nothing else to do, stop */
1132         if (init_action_list == NULL) {
1133                 message(LOG | CONSOLE,
1134                                 "\rNo more tasks for init -- sleeping forever.\n");
1135                 loop_forever();
1136         }
1137
1138         /* Now run the looping stuff for the rest of forever */
1139         while (1) {
1140                 /* run the respawn stuff */
1141                 run_actions(RESPAWN);
1142
1143                 /* run the askfirst stuff */
1144                 run_actions(ASKFIRST);
1145
1146                 /* Don't consume all CPU time -- sleep a bit */
1147                 sleep(1);
1148
1149                 /* Wait for a child process to exit */
1150                 wpid = wait(&status);
1151                 if (wpid > 0) {
1152                         /* Find out who died and clean up their corpse */
1153                         for (a = init_action_list; a; a = a->next) {
1154                                 if (a->pid == wpid) {
1155                                         /* Set the pid to 0 so that the process gets
1156                                          * restarted by run_actions() */
1157                                         a->pid = 0;
1158                                         message(LOG, "Process '%s' (pid %d) exited.  "
1159                                                         "Scheduling it for restart.\n", a->command, wpid);
1160                                 }
1161                         }
1162                 }
1163         }
1164 }
1165
1166 /*
1167 Local Variables:
1168 c-file-style: "linux"
1169 c-basic-offset: 4
1170 tab-width: 4
1171 End:
1172 */