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