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