device_open is common code -- moved into utility.c
[oweals/busybox.git] / init / 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 #include "internal.h"
26 #include <stdio.h>
27 #include <string.h>
28 #include <stdlib.h>
29 #include <stdarg.h>
30 #include <unistd.h>
31 #include <errno.h>
32 #include <signal.h>
33 #include <termios.h>
34 #include <paths.h>
35 #include <sys/types.h>
36 #include <sys/fcntl.h>
37 #include <sys/wait.h>
38 #include <string.h>
39 #include <sys/mount.h>
40 #include <sys/reboot.h>
41 #include <sys/kdaemon.h>
42 #include <sys/sysmacros.h>
43 #include <asm/types.h>
44 #include <linux/serial.h>       /* for serial_struct */
45 #include <sys/vt.h>             /* for vt_stat */
46 #include <sys/ioctl.h>
47 #include <linux/version.h>
48 #ifdef BB_SYSLOGD
49 #include <sys/syslog.h>
50 #endif
51
52 #if ! defined BB_FEATURE_USE_PROCFS
53 #error Sorry, I depend on the /proc filesystem right now.
54 #endif
55
56 #ifndef KERNEL_VERSION
57 #define KERNEL_VERSION(a,b,c) (((a) << 16) + ((b) << 8) + (c))
58 #endif
59
60
61 #define VT_PRIMARY      "/dev/tty1"       /* Primary virtual console */
62 #define VT_SECONDARY    "/dev/tty2"       /* Virtual console */
63 #define VT_LOG          "/dev/tty3"       /* Virtual console */
64 #define SERIAL_CON0     "/dev/ttyS0"      /* Primary serial console */
65 #define SERIAL_CON1     "/dev/ttyS1"      /* Serial console */
66 #define SHELL           "/bin/sh"         /* Default shell */
67 #define INITTAB         "/etc/inittab"    /* inittab file location */
68 #ifndef INIT_SCRIPT
69 #define INIT_SCRIPT     "/etc/init.d/rcS" /* Default sysinit script. */
70 #endif
71
72 #define LOG             0x1
73 #define CONSOLE         0x2
74
75 /* Allowed init action types */
76 typedef enum {
77     SYSINIT=1,
78     RESPAWN,
79     ASKFIRST,
80     WAIT,
81     ONCE
82 } initActionEnum;
83
84 /* And now a list of the actions we support in the version of init */
85 typedef struct initActionType{
86     const char* name;
87     initActionEnum action;
88 } initActionType;
89
90 static const struct initActionType actions[] = {
91     {"sysinit",     SYSINIT},
92     {"respawn",     RESPAWN},
93     {"askfirst",    ASKFIRST},
94     {"wait",        WAIT},
95     {"once",        ONCE},
96     {0}
97 };
98
99 /* Set up a linked list of initactions, to be read from inittab */
100 typedef struct initActionTag initAction;
101 struct initActionTag {
102     pid_t pid;
103     char process[256];
104     char console[256];
105     initAction *nextPtr;
106     initActionEnum action;
107 };
108 initAction* initActionList = NULL;
109
110
111 static char *secondConsole = VT_SECONDARY;
112 static char *log = VT_LOG;
113 static int kernelVersion = 0;
114 static char termType[32] = "TERM=ansi";
115 static char console[32] = _PATH_CONSOLE;
116
117
118 /* print a message to the specified device:
119  * device may be bitwise-or'd from LOG | CONSOLE */
120 void message(int device, char *fmt, ...)
121 {
122     va_list arguments;
123     int fd;
124
125 #ifdef BB_SYSLOGD
126
127     /* Log the message to syslogd */
128     if (device & LOG ) {
129         char msg[1024];
130         va_start(arguments, fmt);
131         vsnprintf(msg, sizeof(msg), fmt, arguments);
132         va_end(arguments);
133         openlog( "init", 0, LOG_DAEMON);
134         syslog(LOG_DAEMON|LOG_NOTICE, msg);
135         closelog();
136     }
137
138 #else
139     static int log_fd = -1;
140
141     /* Take full control of the log tty, and never close it.
142      * It's mine, all mine!  Muhahahaha! */
143     if (log_fd < 0) {
144         if (log == NULL) {
145         /* don't even try to log, because there is no such console */
146         log_fd = -2;
147         /* log to main console instead */
148         device = CONSOLE;
149     }
150     else if ((log_fd = device_open(log, O_RDWR|O_NDELAY)) < 0) {
151             log_fd = -1;
152             fprintf(stderr, "Bummer, can't write to log on %s!\r\n", log);
153             fflush(stderr);
154             return;
155         }
156     }
157     if ( (device & LOG) && (log_fd >= 0) ) {
158         va_start(arguments, fmt);
159         vdprintf(log_fd, fmt, arguments);
160         va_end(arguments);
161     }
162 #endif
163
164     if (device & CONSOLE) {
165         /* Always send console messages to /dev/console so people will see them. */
166         if ((fd = device_open(_PATH_CONSOLE, O_WRONLY|O_NOCTTY|O_NDELAY)) >= 0) {
167             va_start(arguments, fmt);
168             vdprintf(fd, fmt, arguments);
169             va_end(arguments);
170             close(fd);
171         } else {
172             fprintf(stderr, "Bummer, can't print: ");
173             va_start(arguments, fmt);
174             vfprintf(stderr, fmt, arguments);
175             fflush(stderr);
176             va_end(arguments);
177         }
178     }
179 }
180
181
182 /* Set terminal settings to reasonable defaults */
183 void set_term( int fd)
184 {
185     struct termios tty;
186     static const char control_characters[] = {
187         '\003', '\034', '\177', '\025', '\004', '\0',
188         '\1', '\0', '\021', '\023', '\032', '\0', '\022',
189         '\017', '\027', '\026', '\0'
190         };
191
192     tcgetattr(fd, &tty);
193
194     /* set control chars */
195     memcpy(tty.c_cc, control_characters, sizeof(control_characters));
196
197     /* use line dicipline 0 */
198     tty.c_line = 0;
199
200     /* Make it be sane */
201     //tty.c_cflag &= CBAUD|CBAUDEX|CSIZE|CSTOPB|PARENB|PARODD;
202     //tty.c_cflag |= HUPCL|CLOCAL;
203
204     /* input modes */
205     tty.c_iflag = ICRNL|IXON|IXOFF;
206
207     /* output modes */
208     tty.c_oflag = OPOST|ONLCR;
209
210     /* local modes */
211     tty.c_lflag = ISIG|ICANON|ECHO|ECHOE|ECHOK|ECHOCTL|ECHOKE|IEXTEN;
212
213     tcsetattr(fd, TCSANOW, &tty);
214 }
215
216 /* How much memory does this machine have? */
217 static int mem_total()
218 {
219     char s[80];
220     char *p = "/proc/meminfo";
221     FILE *f;
222     const char pattern[] = "MemTotal:";
223
224     if ((f = fopen(p, "r")) < 0) {
225         message(LOG, "Error opening %s: %s\n", p, strerror( errno));
226         return -1;
227     }
228     while (NULL != fgets(s, 79, f)) {
229         p = strstr(s, pattern);
230         if (NULL != p) {
231             fclose(f);
232             return (atoi(p + strlen(pattern)));
233         }
234     }
235     return -1;
236 }
237
238 static void console_init()
239 {
240     int fd;
241     int tried_devcons = 0;
242     int tried_vtprimary = 0;
243     struct serial_struct sr;
244     char *s;
245
246     if ((s = getenv("TERM")) != NULL) {
247         snprintf(termType,sizeof(termType)-1,"TERM=%s",s);
248     }
249
250     if ((s = getenv("CONSOLE")) != NULL) {
251         snprintf(console, sizeof(console)-1, "%s",s);
252     }
253 #if #cpu(sparc)
254     /* sparc kernel supports console=tty[ab] parameter which is also 
255      * passed to init, so catch it here */
256     else if ((s = getenv("console")) != NULL) {
257         /* remap tty[ab] to /dev/ttyS[01] */
258         if (strcmp( s, "ttya" )==0)
259             snprintf(console, sizeof(console)-1, "%s", SERIAL_CON0);
260         else if (strcmp( s, "ttyb" )==0)
261             snprintf(console, sizeof(console)-1, "%s", SERIAL_CON1);
262     }
263 #endif
264     else {
265         struct vt_stat vt;
266
267         /* 2.2 kernels: identify the real console backend and try to use it */
268         if (ioctl(0, TIOCGSERIAL, &sr) == 0) {
269             /* this is a serial console */
270             snprintf(console, sizeof(console)-1, "/dev/ttyS%d", sr.line);
271         }
272         else if (ioctl(0, VT_GETSTATE, &vt) == 0) {
273             /* this is linux virtual tty */
274             snprintf(console, sizeof(console)-1, "/dev/tty%d", vt.v_active);
275         } else {
276             snprintf(console, sizeof(console)-1, "%s", _PATH_CONSOLE);
277             tried_devcons++;
278         }
279     }
280
281     while ((fd = open(console, O_RDONLY | O_NONBLOCK)) < 0) {
282         /* Can't open selected console -- try /dev/console */
283         if (!tried_devcons) {
284             tried_devcons++;
285             snprintf(console, sizeof(console)-1, "%s", _PATH_CONSOLE);
286             continue;
287         }
288         /* Can't open selected console -- try vt1 */
289         if (!tried_vtprimary) {
290             tried_vtprimary++;
291             snprintf(console, sizeof(console)-1, "%s", VT_PRIMARY);
292             continue;
293         }
294         break;
295     }
296     if (fd < 0) {
297         /* Perhaps we should panic here? */
298         snprintf(console, sizeof(console)-1, "/dev/null");
299     } else {
300         /* check for serial console and disable logging to tty3 & running a
301         * shell to tty2 */
302         if (ioctl(0,TIOCGSERIAL,&sr) == 0) {
303             message(LOG|CONSOLE, "serial console detected.  Disabling virtual terminals.\r\n" );
304             log = NULL;
305             secondConsole = NULL;
306         }
307         close(fd);
308     }
309     message(LOG, "console=%s\n", console );
310 }
311
312 static pid_t run(char* command, 
313         char *terminal, int get_enter)
314 {
315     int i, fd;
316     pid_t pid;
317     char* tmpCmd;
318     char* cmd[255];
319     static const char press_enter[] =
320         "\nPlease press Enter to activate this console. ";
321     char* environment[] = {
322         "HOME=/",
323         "PATH=/usr/bin:/bin:/usr/sbin:/sbin",
324         "SHELL=/bin/sh",
325         termType,
326         "USER=root",
327         0
328     };
329
330
331     if ((pid = fork()) == 0) {
332         pid_t shell_pgid = getpid ();
333
334         /* Clean up */
335         close(0);
336         close(1);
337         close(2);
338         setsid();
339
340         /* Reset signal handlers set for parent process */
341         signal(SIGUSR1, SIG_DFL);
342         signal(SIGUSR2, SIG_DFL);
343         signal(SIGINT, SIG_DFL);
344         signal(SIGTERM, SIG_DFL);
345         signal(SIGHUP, SIG_DFL);
346
347         if ((fd = device_open(terminal, O_RDWR)) < 0) {
348             message(LOG|CONSOLE, "Bummer, can't open %s\r\n", terminal);
349             exit(1);
350         }
351         dup2(fd, 0);
352         dup2(fd, 1);
353         dup2(fd, 2);
354         tcsetpgrp (0, getpgrp());
355         set_term(0);
356
357         if (get_enter==TRUE) {
358             /*
359              * Save memory by not exec-ing anything large (like a shell)
360              * before the user wants it. This is critical if swap is not
361              * enabled and the system has low memory. Generally this will
362              * be run on the second virtual console, and the first will
363              * be allowed to start a shell or whatever an init script 
364              * specifies.
365              */
366             char c;
367             message(LOG, "Waiting for enter to start '%s' (pid %d, console %s)\r\n", 
368                     command, shell_pgid, terminal );
369             write(fileno(stdout), press_enter, sizeof(press_enter) - 1);
370             read(fileno(stdin), &c, 1);
371         }
372
373         /* Log the process name and args */
374         message(LOG, "Starting pid %d, console %s: '", 
375                 shell_pgid, terminal, command);
376
377         /* Convert command (char*) into cmd (char**, one word per string) */
378         for (tmpCmd=command, i=0; (tmpCmd=strsep(&command, " \t")) != NULL;) {
379             if (*tmpCmd != '\0') {
380                 cmd[i] = tmpCmd;
381                 message(LOG, "%s ", tmpCmd);
382                 tmpCmd++;
383                 i++;
384             }
385         }
386         cmd[i] = NULL;
387         message(LOG, "'\r\n");
388
389         /* Now run it.  The new program will take over this PID, 
390          * so nothing further in init.c should be run. */
391         execve(cmd[0], cmd, environment);
392
393         /* We're still here?  Some error happened. */
394         message(LOG|CONSOLE, "Bummer, could not run '%s': %s\n", cmd[0],
395                 strerror(errno));
396         exit(-1);
397     }
398     return pid;
399 }
400
401 static int waitfor(char* command, 
402         char *terminal, int get_enter)
403 {
404     int status, wpid;
405     int pid = run( command, terminal, get_enter);
406
407     while (1) {
408         wpid = wait(&status);
409         if (wpid > 0 ) {
410             message(LOG, "Process '%s' (pid %d) exited.\n", 
411                             command, wpid);
412             break;
413         }
414         if (wpid == pid )
415             break;
416     }
417     return wpid;
418 }
419
420 /* Make sure there is enough memory to do something useful. *
421  * Calls swapon if needed so be sure /proc is mounted. */
422 static void check_memory()
423 {
424     struct stat statBuf;
425
426     if (mem_total() > 3500)
427         return;
428
429     if (stat("/etc/fstab", &statBuf) == 0) {
430         /* Try to turn on swap */
431         waitfor("/bin/swapon swapon -a", log, FALSE);
432         if (mem_total() < 3500)
433             goto goodnight;
434     } else
435         goto goodnight;
436     return;
437
438 goodnight:
439         message(CONSOLE, "Sorry, your computer does not have enough memory.\r\n");
440         while (1) sleep(1);
441 }
442
443 #ifndef DEBUG_INIT
444 static void shutdown_system(void)
445 {
446     /* first disable our SIGHUP signal */
447     signal(SIGHUP, SIG_DFL);
448     
449     /* Allow Ctrl-Alt-Del to reboot system. */
450     reboot(RB_ENABLE_CAD);
451     message(CONSOLE, "\r\nThe system is going down NOW !!\r\n");
452     sync();
453
454     /* Send signals to every process _except_ pid 1 */
455     message(CONSOLE, "Sending SIGTERM to all processes.\r\n");
456     kill(-1, SIGTERM);
457     sleep(5);
458     sync();
459
460     message(CONSOLE, "Sending SIGKILL to all processes.\r\n");
461     kill(-1, SIGKILL);
462     sleep(5);
463
464     message(CONSOLE, "Disabling swap.\r\n");
465         waitfor( "swapoff -a", console, FALSE);
466     message(CONSOLE, "Unmounting filesystems.\r\n");
467         waitfor("umount -a -r", console, FALSE);
468     sync();
469     if (kernelVersion > 0 && kernelVersion <= 2 * 65536 + 2 * 256 + 11) {
470         /* bdflush, kupdate not needed for kernels >2.2.11 */
471         bdflush(1, 0);
472         sync();
473     }
474 }
475
476 static void halt_signal(int sig)
477 {
478     shutdown_system();
479     message(CONSOLE, "The system is halted. Press %s or turn off power\r\n",
480         (secondConsole == NULL) /* serial console */
481             ? "Reset" : "CTRL-ALT-DEL");
482     sync();
483
484     /* allow time for last message to reach serial console */
485     sleep(5);
486
487 #if LINUX_VERSION_CODE >= KERNEL_VERSION(2,2,0)
488     if (sig == SIGUSR2)
489         reboot(RB_POWER_OFF);
490     else
491 #endif
492     reboot(RB_HALT_SYSTEM);
493     exit(0);
494 }
495
496 static void reboot_signal(int sig)
497 {
498     shutdown_system();
499     message(CONSOLE, "Please stand by while rebooting the system.\r\n");
500     sync();
501
502     /* allow time for last message to reach serial console */
503     sleep(2);
504
505     reboot(RB_AUTOBOOT);
506     exit(0);
507 }
508
509 #if defined BB_FEATURE_INIT_CHROOT
510 static void check_chroot(int sig)
511 {
512     char *argv_init[2] = { "init", NULL, };
513     char *envp_init[3] = { "HOME=/", "TERM=linux", NULL, };
514     char rootpath[256], *tc;
515     int fd;
516
517     if ((fd = open("/proc/sys/kernel/init-chroot", O_RDONLY)) == -1) {
518         message(CONSOLE, "SIGHUP recived, but could not open proc file\r\n");
519         sleep(2);
520         return;
521     }
522     if (read(fd, rootpath, sizeof(rootpath)) == -1) {
523         message(CONSOLE, "SIGHUP recived, but could not read proc file\r\n");
524         sleep(2);
525         return;
526     }
527     close(fd);
528
529     if (rootpath[0] == '\0') {
530         message(CONSOLE, "SIGHUP recived, but new root is not valid: %s\r\n",
531                 rootpath);
532         sleep(2);
533         return;
534     }
535
536     tc = strrchr(rootpath, '\n');
537     *tc = '\0';
538     
539     /* Ok, making it this far means we commit */
540     message(CONSOLE, "Please stand by, changing root to `%s'.\r\n", rootpath);
541
542     /* kill all other programs first */
543     message(CONSOLE, "Sending SIGTERM to all processes.\r\n");
544     kill(-1, SIGTERM);
545     sleep(2);
546     sync();
547
548     message(CONSOLE, "Sending SIGKILL to all processes.\r\n");
549     kill(-1, SIGKILL);
550     sleep(2);
551     sync();
552
553     /* ok, we don't need /proc anymore. we also assume that the signaling
554      * process left the rest of the filesystems alone for us */
555     umount("/proc");
556
557     /* Ok, now we chroot. Hopefully we only have two things mounted, the
558      * new chroot'd mount point, and the old "/" mount. s,
559      * we go ahead and unmount the old "/". This should trigger the kernel
560      * to set things up the Right Way(tm). */
561
562     if (!chroot(rootpath))
563         umount("/dev/root");
564
565     /* If the chroot fails, we are already too far to turn back, so we
566      * continue and hope that executing init below will revive the system */
567
568     /* close all of our descriptors and open new ones */
569     close(0);
570     close(1);
571     close(2);
572     open("/dev/console", O_RDWR, 0);
573     dup(0);
574     dup(0);
575
576     message(CONSOLE, "Executing real init...\r\n");
577     /* execute init in the (hopefully) new root */
578     execve("/sbin/init",argv_init,envp_init);
579
580     message(CONSOLE, "ERROR: Could not exec new init. Press %s to reboot.\r\n",
581         (secondConsole == NULL) /* serial console */
582             ? "Reset" : "CTRL-ALT-DEL");
583     return;
584
585 #endif /* BB_FEATURE_INIT_CHROOT */
586
587 #endif /* ! DEBUG_INIT */
588
589 void new_initAction (initActionEnum action,
590         char* process, char* cons)
591 {
592     initAction* newAction;
593
594     if (*cons == '\0')
595         cons = console;
596  
597     /* If BusyBox detects that a serial console is in use, 
598      * then entries not refering to the console or null devices will _not_ be run.
599      * The exception to this rule is the null device.
600      */
601     if (secondConsole == NULL && strcmp(cons, console) && strcmp(cons, "/dev/null"))
602         return;
603
604     newAction = calloc ((size_t)(1), sizeof(initAction));
605     if (!newAction) {
606         message(LOG|CONSOLE,"Memory allocation failure\n");
607         while (1) sleep(1);
608     }
609     newAction->nextPtr = initActionList;
610     initActionList = newAction;
611     strncpy( newAction->process, process, 255);
612     newAction->action = action;
613     strncpy(newAction->console, cons, 255);
614     newAction->pid = 0;
615 //    message(LOG|CONSOLE, "process='%s' action='%d' console='%s'\n",
616 //          newAction->process, newAction->action, newAction->console);
617 }
618
619 void delete_initAction (initAction *action)
620 {
621     initAction *a, *b=NULL;
622     for( a=initActionList ; a ; b=a, a=a->nextPtr) {
623         if (a == action) {
624             if (b==NULL) {
625                 initActionList=a->nextPtr;
626             } else {
627                 b->nextPtr=a->nextPtr;
628             }
629             free( a);
630             break;
631         }
632     }
633 }
634
635 /* NOTE that if BB_FEATURE_USE_INITTAB is NOT defined,
636  * then parse_inittab() simply adds in some default
637  * actions(i.e runs INIT_SCRIPT and then starts a pair 
638  * of "askfirst" shells).  If BB_FEATURE_USE_INITTAB 
639  * _is_ defined, but /etc/inittab is missing, this 
640  * results in the same set of default behaviors.
641  * */
642 void parse_inittab(void) 
643 {
644 #ifdef BB_FEATURE_USE_INITTAB
645     FILE* file;
646     char buf[256], lineAsRead[256], tmpConsole[256];
647     char *p, *q, *r, *s;
648     const struct initActionType *a = actions;
649     int foundIt;
650
651
652     file = fopen(INITTAB, "r");
653     if (file == NULL) {
654         /* No inittab file -- set up some default behavior */
655 #endif
656         /* Askfirst shell on tty1 */
657         new_initAction( ASKFIRST, SHELL, console );
658         /* Askfirst shell on tty2 */
659         if (secondConsole != NULL) 
660             new_initAction( ASKFIRST, SHELL, secondConsole );
661         /* sysinit */
662         new_initAction( SYSINIT, INIT_SCRIPT, console );
663
664         return;
665 #ifdef BB_FEATURE_USE_INITTAB
666     }
667
668     while ( fgets(buf, 255, file) != NULL) {
669         foundIt=FALSE;
670         for(p = buf; *p == ' ' || *p == '\t'; p++);
671         if (*p == '#' || *p == '\n') continue;
672
673         /* Trim the trailing \n */
674         q = strrchr( p, '\n');
675         if (q != NULL)
676             *q='\0';
677
678         /* Keep a copy around for posterity's sake (and error msgs) */
679         strcpy(lineAsRead, buf);
680
681         /* Grab the ID field */
682         s=p;
683         p = strchr( p, ':');
684         if ( p != NULL || *(p+1) != '\0' ) {
685             *p='\0';
686             ++p;
687         }
688
689         /* Now peal off the process field from the end
690          * of the string */
691         q = strrchr( p, ':');
692         if ( q == NULL || *(q+1) == '\0' ) {
693             message(LOG|CONSOLE,"Bad inittab entry: %s\n", lineAsRead);
694             continue;
695         } else {
696             *q='\0';
697             ++q;
698         }
699
700         /* Now peal off the action field */
701         r = strrchr( p, ':');
702         if ( r == NULL || *(r+1) == '\0') {
703             message(LOG|CONSOLE,"Bad inittab entry: %s\n", lineAsRead);
704             continue;
705         } else {
706             ++r;
707         }
708
709         /* Ok, now process it */
710         a = actions;
711         while (a->name != 0) {
712             if (strcmp(a->name, r) == 0) {
713                 if (*s != '\0') {
714                     struct stat statBuf;
715                     strcpy(tmpConsole, "/dev/");
716                     strncat(tmpConsole, s, 200);
717                     if (stat(tmpConsole, &statBuf) != 0) {
718                         message(LOG|CONSOLE, "device '%s' does not exist.  Did you read the directions?\n", tmpConsole);
719                         break;
720                     }
721                     s = tmpConsole;
722                 }
723                 new_initAction( a->action, q, s);
724                 foundIt=TRUE;
725             }
726             a++;
727         }
728         if (foundIt==TRUE)
729             continue;
730         else {
731             /* Choke on an unknown action */
732             message(LOG|CONSOLE, "Bad inittab entry: %s\n", lineAsRead);
733         }
734     }
735     return;
736 #endif
737 }
738
739 extern int init_main(int argc, char **argv)
740 {
741     initAction *a;
742     pid_t wpid;
743     int status;
744
745 #ifndef DEBUG_INIT
746     /* Expect to be PID 1 if we are run as init (not linuxrc) */
747     if (getpid() != 1 && strstr(argv[0], "init")!=NULL ) {
748         usage( "init\n\nInit is the parent of all processes.\n\n"
749                 "This version of init is designed to be run only by the kernel\n");
750     }
751     /* Fix up argv[0] to be certain we claim to be init */
752     strncpy(argv[0], "init", strlen(argv[0]));
753
754     /* Set up sig handlers  -- be sure to
755      * clear all of these in run() */
756     signal(SIGUSR1, halt_signal);
757     signal(SIGUSR2, reboot_signal);
758     signal(SIGINT, reboot_signal);
759     signal(SIGTERM, reboot_signal);
760 #if defined BB_FEATURE_INIT_CHROOT
761     signal(SIGHUP, check_chroot);
762 #endif
763
764     /* Turn off rebooting via CTL-ALT-DEL -- we get a 
765      * SIGINT on CAD so we can shut things down gracefully... */
766     reboot(RB_DISABLE_CAD);
767 #endif 
768
769     /* Figure out where the default console should be */
770     console_init();
771
772     /* Close whatever files are open, and reset the console. */
773     close(0);
774     close(1);
775     close(2);
776     set_term(0);
777     setsid();
778
779     /* Make sure PATH is set to something sane */
780     putenv(_PATH_STDPATH);
781
782     /* Hello world */
783 #ifndef DEBUG_INIT
784     message(LOG|CONSOLE, 
785             "init started:  BusyBox v%s (%s) multi-call binary\r\n", 
786             BB_VER, BB_BT);
787 #else
788     message(LOG|CONSOLE, 
789             "init(%d) started:  BusyBox v%s (%s) multi-call binary\r\n", 
790             getpid(), BB_VER, BB_BT);
791 #endif
792
793     
794     /* Mount /proc */
795     if (mount ("proc", "/proc", "proc", 0, 0) == 0) {
796         message(LOG|CONSOLE, "Mounting /proc: done.\n");
797         kernelVersion = get_kernel_revision();
798     } else
799         message(LOG|CONSOLE, "Mounting /proc: failed!\n");
800
801     /* Make sure there is enough memory to do something useful. */
802     check_memory();
803
804     /* Check if we are supposed to be in single user mode */
805     if ( argc > 1 && (!strcmp(argv[1], "single") || 
806                 !strcmp(argv[1], "-s") || !strcmp(argv[1], "1"))) 
807     {
808         /* Ask first then start a shell on tty2 */
809         if (secondConsole != NULL) 
810             new_initAction( ASKFIRST, SHELL, secondConsole);
811         /* Start a shell on tty1 */
812         new_initAction( RESPAWN, SHELL, console);
813     } else {
814         /* Not in single user mode -- see what inittab says */
815
816         /* NOTE that if BB_FEATURE_USE_INITTAB is NOT defined,
817          * then parse_inittab() simply adds in some default
818          * actions(i.e runs INIT_SCRIPT and then starts a pair 
819          * of "askfirst" shells */
820         parse_inittab();
821     }
822
823     /* Now run everything that needs to be run */
824
825     /* First run the sysinit command */
826     for( a=initActionList ; a; a=a->nextPtr) {
827         if (a->action == SYSINIT) {
828             waitfor(a->process, a->console, FALSE);
829             /* Now remove the "sysinit" entry from the list */
830             delete_initAction( a);
831         }
832     }
833     /* Next run anything that wants to block */
834     for( a=initActionList ; a; a=a->nextPtr) {
835         if (a->action == WAIT) {
836             waitfor(a->process, a->console, FALSE);
837             /* Now remove the "wait" entry from the list */
838             delete_initAction( a);
839         }
840     }
841     /* Next run anything to be run only once */
842     for( a=initActionList ; a; a=a->nextPtr) {
843         if (a->action == ONCE) {
844             run(a->process, a->console, FALSE);
845             /* Now remove the "once" entry from the list */
846             delete_initAction( a);
847         }
848     }
849     /* If there is nothing else to do, stop */
850     if (initActionList == NULL) {
851         message(LOG|CONSOLE, "No more tasks for init -- sleeping forever.\n");
852         while (1) sleep(1);
853     }
854
855     /* Now run the looping stuff for the rest of forever */
856     while (1) {
857         for( a=initActionList ; a; a=a->nextPtr) {
858             /* Only run stuff with pid==0.  If they have
859              * a pid, that means they are still running */
860             if (a->pid == 0) {
861                 switch(a->action) {
862                     case RESPAWN:
863                         /* run the respawn stuff */
864                         a->pid = run(a->process, a->console, FALSE);
865                         break;
866                     case ASKFIRST:
867                         /* run the askfirst stuff */
868                         a->pid = run(a->process, a->console, TRUE);
869                         break;
870                     /* silence the compiler's incessant whining */
871                     default:
872                         break;
873                 }
874             }
875         }
876         /* Wait for a child process to exit */
877         wpid = wait(&status);
878         if (wpid > 0 ) {
879             /* Find out who died and clean up their corpse */
880             for( a=initActionList ; a; a=a->nextPtr) {
881                 if (a->pid==wpid) {
882                     a->pid=0;
883                     message(LOG, "Process '%s' (pid %d) exited.  Scheduling it for restart.\n", 
884                             a->process, wpid);
885                 }
886             }
887         }
888         sleep(1);
889     }
890 }
891