Vodz' last_patch57:
[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         /* run everything to be run at "shutdown".  This is done _prior_
719          * to killing everything, in case people wish to use scripts to
720          * shut things down gracefully... */
721         run_actions(SHUTDOWN);
722
723         /* first disable all our signals */
724         sigemptyset(&block_signals);
725         sigaddset(&block_signals, SIGHUP);
726         sigaddset(&block_signals, SIGCHLD);
727         sigaddset(&block_signals, SIGUSR1);
728         sigaddset(&block_signals, SIGUSR2);
729         sigaddset(&block_signals, SIGINT);
730         sigaddset(&block_signals, SIGTERM);
731         sigaddset(&block_signals, SIGCONT);
732         sigaddset(&block_signals, SIGSTOP);
733         sigaddset(&block_signals, SIGTSTP);
734         sigprocmask(SIG_BLOCK, &block_signals, NULL);
735
736         /* Allow Ctrl-Alt-Del to reboot system. */
737         init_reboot(RB_ENABLE_CAD);
738
739         message(CONSOLE | LOG, "\n\rThe system is going down NOW !!\n");
740         sync();
741
742         /* Send signals to every process _except_ pid 1 */
743         message(CONSOLE | LOG, "\rSending SIGTERM to all processes.\n");
744         kill(-1, SIGTERM);
745         sleep(1);
746         sync();
747
748         message(CONSOLE | LOG, "\rSending SIGKILL to all processes.\n");
749         kill(-1, SIGKILL);
750         sleep(1);
751
752         sync();
753         if (kernelVersion > 0 && kernelVersion <= KERNEL_VERSION(2, 2, 11)) {
754                 /* bdflush, kupdate not needed for kernels >2.2.11 */
755                 bdflush(1, 0);
756                 sync();
757         }
758 }
759
760 static void exec_signal(int sig)
761 {
762         struct init_action *a, *tmp;
763         sigset_t unblock_signals;
764
765         for (a = init_action_list; a; a = tmp) {
766                 tmp = a->next;
767                 if (a->action & RESTART) {
768                         shutdown_system();
769
770                         /* unblock all signals, blocked in shutdown_system() */
771                         sigemptyset(&unblock_signals);
772                         sigaddset(&unblock_signals, SIGHUP);
773                         sigaddset(&unblock_signals, SIGCHLD);
774                         sigaddset(&unblock_signals, SIGUSR1);
775                         sigaddset(&unblock_signals, SIGUSR2);
776                         sigaddset(&unblock_signals, SIGINT);
777                         sigaddset(&unblock_signals, SIGTERM);
778                         sigaddset(&unblock_signals, SIGCONT);
779                         sigaddset(&unblock_signals, SIGSTOP);
780                         sigaddset(&unblock_signals, SIGTSTP);
781                         sigprocmask(SIG_UNBLOCK, &unblock_signals, NULL);
782
783                         message(CONSOLE | LOG, "\rTrying to re-exec %s\n", a->command);
784                         execl(a->command, a->command, NULL);
785
786                         message(CONSOLE | LOG, "\rexec of '%s' failed: %s\n",
787                                         a->command, strerror(errno));
788                         sync();
789                         sleep(2);
790                         init_reboot(RB_HALT_SYSTEM);
791                         loop_forever();
792                 }
793         }
794 }
795
796 static void halt_signal(int sig)
797 {
798         shutdown_system();
799         message(CONSOLE | LOG,
800 #if #cpu(s390)
801                         /* Seems the s390 console is Wierd(tm). */
802                         "\rThe system is halted. You may reboot now.\n"
803 #else
804                         "\rThe system is halted. Press Reset or turn off power\n"
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)
846                 pause();
847         got_cont = 0;
848         errno = saved_errno;
849 }
850
851 /* The SIGCONT handler */
852 static void cont_handler(int sig)
853 {
854         got_cont = 1;
855 }
856
857 #endif                                                  /* ! DEBUG_INIT */
858
859 static void new_init_action(int action, char *command, char *cons)
860 {
861         struct init_action *new_action, *a;
862
863         if (*cons == '\0')
864                 cons = console;
865
866         /* do not run entries if console device is not available */
867         if (access(cons, R_OK | W_OK))
868                 return;
869         if (strcmp(cons, "/dev/null") == 0 && (action & ASKFIRST))
870                 return;
871
872         new_action = calloc((size_t) (1), sizeof(struct init_action));
873         if (!new_action) {
874                 message(LOG | CONSOLE, "\rMemory allocation failure\n");
875                 loop_forever();
876         }
877
878         /* Append to the end of the list */
879         for (a = init_action_list; a && a->next; a = a->next);
880         if (a) {
881                 a->next = new_action;
882         } else {
883                 init_action_list = new_action;
884         }
885         strcpy(new_action->command, command);
886         new_action->action = action;
887         strcpy(new_action->terminal, cons);
888         new_action->pid = 0;
889 /*    message(LOG|CONSOLE, "command='%s' action='%d' terminal='%s'\n",
890                 new_action->command, new_action->action, new_action->terminal); */
891 }
892
893 static void delete_init_action(struct init_action *action)
894 {
895         struct init_action *a, *b = NULL;
896
897         for (a = init_action_list; a; b = a, a = a->next) {
898                 if (a == action) {
899                         if (b == NULL) {
900                                 init_action_list = a->next;
901                         } else {
902                                 b->next = a->next;
903                         }
904                         free(a);
905                         break;
906                 }
907         }
908 }
909
910 /* NOTE that if CONFIG_FEATURE_USE_INITTAB is NOT defined,
911  * then parse_inittab() simply adds in some default
912  * actions(i.e., runs INIT_SCRIPT and then starts a pair 
913  * of "askfirst" shells).  If CONFIG_FEATURE_USE_INITTAB 
914  * _is_ defined, but /etc/inittab is missing, this 
915  * results in the same set of default behaviors.
916  */
917 static void parse_inittab(void)
918 {
919 #ifdef CONFIG_FEATURE_USE_INITTAB
920         FILE *file;
921         char buf[INIT_BUFFS_SIZE], lineAsRead[INIT_BUFFS_SIZE],
922                 tmpConsole[INIT_BUFFS_SIZE];
923         char *id, *runlev, *action, *command, *eol;
924         const struct init_action_type *a = actions;
925         int foundIt;
926
927
928         file = fopen(INITTAB, "r");
929         if (file == NULL) {
930                 /* No inittab file -- set up some default behavior */
931 #endif
932                 /* Reboot on Ctrl-Alt-Del */
933                 new_init_action(CTRLALTDEL, "/sbin/reboot", console);
934                 /* Umount all filesystems on halt/reboot */
935                 new_init_action(SHUTDOWN, "/bin/umount -a -r", console);
936 #if !defined(__UCLIBC__) || defined(__UCLIBC_HAS_MMU__)
937                 /* Swapoff on halt/reboot */
938                 new_init_action(SHUTDOWN, "/sbin/swapoff -a", console);
939 #endif
940                 /* Prepare to restart init when a HUP is received */
941                 new_init_action(RESTART, "/sbin/init", console);
942                 /* Askfirst shell on tty1-4 */
943                 new_init_action(ASKFIRST, LOGIN_SHELL, console);
944                 new_init_action(ASKFIRST, LOGIN_SHELL, VC_2);
945                 new_init_action(ASKFIRST, LOGIN_SHELL, VC_3);
946                 new_init_action(ASKFIRST, LOGIN_SHELL, VC_4);
947                 /* sysinit */
948                 new_init_action(SYSINIT, INIT_SCRIPT, console);
949
950                 return;
951 #ifdef CONFIG_FEATURE_USE_INITTAB
952         }
953
954         while (fgets(buf, INIT_BUFFS_SIZE, file) != NULL) {
955                 foundIt = FALSE;
956                 /* Skip leading spaces */
957                 for (id = buf; *id == ' ' || *id == '\t'; id++);
958
959                 /* Skip the line if it's a comment */
960                 if (*id == '#' || *id == '\n')
961                         continue;
962
963                 /* Trim the trailing \n */
964                 eol = strrchr(id, '\n');
965                 if (eol != NULL)
966                         *eol = '\0';
967
968                 /* Keep a copy around for posterity's sake (and error msgs) */
969                 strcpy(lineAsRead, buf);
970
971                 /* Separate the ID field from the runlevels */
972                 runlev = strchr(id, ':');
973                 if (runlev == NULL || *(runlev + 1) == '\0') {
974                         message(LOG | CONSOLE, "\rBad inittab entry: %s\n", lineAsRead);
975                         continue;
976                 } else {
977                         *runlev = '\0';
978                         ++runlev;
979                 }
980
981                 /* Separate the runlevels from the action */
982                 action = strchr(runlev, ':');
983                 if (action == NULL || *(action + 1) == '\0') {
984                         message(LOG | CONSOLE, "\rBad inittab entry: %s\n", lineAsRead);
985                         continue;
986                 } else {
987                         *action = '\0';
988                         ++action;
989                 }
990
991                 /* Separate the action from the command */
992                 command = strchr(action, ':');
993                 if (command == NULL || *(command + 1) == '\0') {
994                         message(LOG | CONSOLE, "\rBad inittab entry: %s\n", lineAsRead);
995                         continue;
996                 } else {
997                         *command = '\0';
998                         ++command;
999                 }
1000
1001                 /* Ok, now process it */
1002                 a = actions;
1003                 while (a->name != 0) {
1004                         if (strcmp(a->name, action) == 0) {
1005                                 if (*id != '\0') {
1006                                         strcpy(tmpConsole, "/dev/");
1007                                         strncat(tmpConsole, id, INIT_BUFFS_SIZE - 6);
1008                                         id = tmpConsole;
1009                                 }
1010                                 new_init_action(a->action, command, id);
1011                                 foundIt = TRUE;
1012                         }
1013                         a++;
1014                 }
1015                 if (foundIt == TRUE)
1016                         continue;
1017                 else {
1018                         /* Choke on an unknown action */
1019                         message(LOG | CONSOLE, "\rBad inittab entry: %s\n", lineAsRead);
1020                 }
1021         }
1022         fclose(file);
1023         return;
1024 #endif                                                  /* CONFIG_FEATURE_USE_INITTAB */
1025 }
1026
1027
1028
1029 extern int init_main(int argc, char **argv)
1030 {
1031         struct init_action *a;
1032         pid_t wpid;
1033         int status;
1034
1035         if (argc > 1 && !strcmp(argv[1], "-q")) {
1036                 /* don't assume init's pid == 1 */
1037                 long *pid = find_pid_by_name("init");
1038
1039                 if (!pid || *pid <= 0) {
1040                         pid = find_pid_by_name("linuxrc");
1041                         if (!pid || *pid <= 0)
1042                                 error_msg_and_die("no process killed");
1043                 }
1044                 kill(*pid, SIGHUP);
1045                 exit(0);
1046         }
1047 #ifndef DEBUG_INIT
1048         /* Expect to be invoked as init with PID=1 or be invoked as linuxrc */
1049         if (getpid() != 1
1050 #ifdef CONFIG_FEATURE_INITRD
1051                 && strstr(applet_name, "linuxrc") == NULL
1052 #endif
1053                 ) {
1054                 show_usage();
1055         }
1056         /* Set up sig handlers  -- be sure to
1057          * clear all of these in run() */
1058         signal(SIGHUP, exec_signal);
1059         signal(SIGUSR1, halt_signal);
1060         signal(SIGUSR2, halt_signal);
1061         signal(SIGINT, ctrlaltdel_signal);
1062         signal(SIGTERM, reboot_signal);
1063         signal(SIGCONT, cont_handler);
1064         signal(SIGSTOP, stop_handler);
1065         signal(SIGTSTP, stop_handler);
1066         signal(SIGCHLD, SIG_IGN);
1067
1068         /* Turn off rebooting via CTL-ALT-DEL -- we get a 
1069          * SIGINT on CAD so we can shut things down gracefully... */
1070         init_reboot(RB_DISABLE_CAD);
1071 #endif
1072
1073         /* Figure out what kernel this is running */
1074         kernelVersion = get_kernel_revision();
1075
1076         /* Figure out where the default console should be */
1077         console_init();
1078
1079         /* Close whatever files are open, and reset the console. */
1080         close(0);
1081         close(1);
1082         close(2);
1083
1084         if (device_open(console, O_RDWR | O_NOCTTY) == 0) {
1085                 set_term(0);
1086                 close(0);
1087         }
1088
1089         chdir("/");
1090         setsid();
1091
1092         /* Make sure PATH is set to something sane */
1093         putenv("PATH=" _PATH_STDPATH);
1094
1095         /* Hello world */
1096         message(MAYBE_CONSOLE | LOG, "\rinit started:  %s\n", full_version);
1097
1098         /* Make sure there is enough memory to do something useful. */
1099         check_memory();
1100
1101         /* Check if we are supposed to be in single user mode */
1102         if (argc > 1 && (!strcmp(argv[1], "single") ||
1103                                          !strcmp(argv[1], "-s") || !strcmp(argv[1], "1"))) {
1104                 /* Ask first then start a shell on tty2-4 */
1105                 new_init_action(ASKFIRST, LOGIN_SHELL, VC_2);
1106                 new_init_action(ASKFIRST, LOGIN_SHELL, VC_3);
1107                 new_init_action(ASKFIRST, LOGIN_SHELL, VC_4);
1108                 /* Start a shell on tty1 */
1109                 new_init_action(RESPAWN, LOGIN_SHELL, console);
1110         } else {
1111                 /* Not in single user mode -- see what inittab says */
1112
1113                 /* NOTE that if CONFIG_FEATURE_USE_INITTAB is NOT defined,
1114                  * then parse_inittab() simply adds in some default
1115                  * actions(i.e., runs INIT_SCRIPT and then starts a pair 
1116                  * of "askfirst" shells */
1117                 parse_inittab();
1118         }
1119
1120         /* Make the command line just say "init"  -- thats all, nothing else */
1121         fixup_argv(argc, argv, "init");
1122
1123         /* Now run everything that needs to be run */
1124
1125         /* First run the sysinit command */
1126         run_actions(SYSINIT);
1127
1128         /* Next run anything that wants to block */
1129         run_actions(WAIT);
1130
1131         /* Next run anything to be run only once */
1132         run_actions(ONCE);
1133
1134         /* If there is nothing else to do, stop */
1135         if (init_action_list == NULL) {
1136                 message(LOG | CONSOLE,
1137                                 "\rNo more tasks for init -- sleeping forever.\n");
1138                 loop_forever();
1139         }
1140
1141         /* Now run the looping stuff for the rest of forever */
1142         while (1) {
1143                 /* run the respawn stuff */
1144                 run_actions(RESPAWN);
1145
1146                 /* run the askfirst stuff */
1147                 run_actions(ASKFIRST);
1148
1149                 /* Don't consume all CPU time -- sleep a bit */
1150                 sleep(1);
1151
1152                 /* Wait for a child process to exit */
1153                 wpid = wait(&status);
1154                 if (wpid > 0) {
1155                         /* Find out who died and clean up their corpse */
1156                         for (a = init_action_list; a; a = a->next) {
1157                                 if (a->pid == wpid) {
1158                                         /* Set the pid to 0 so that the process gets
1159                                          * restarted by run_actions() */
1160                                         a->pid = 0;
1161                                         message(LOG, "Process '%s' (pid %d) exited.  "
1162                                                         "Scheduling it for restart.\n", a->command, wpid);
1163                                 }
1164                         }
1165                 }
1166         }
1167 }
1168
1169 /*
1170 Local Variables:
1171 c-file-style: "linux"
1172 c-basic-offset: 4
1173 tab-width: 4
1174 End:
1175 */