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