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  * Licensed under GPLv2 or later, see file LICENSE in this tarball for details.
  10  */
  11
  12 #include "libbb.h"
  13 #include <syslog.h>
  14 #include <paths.h>
  15 #include <sys/reboot.h>
  16
  17 /* Was a CONFIG_xxx option. A lot of people were building
  18  * not fully functional init by switching it on! */
  19 #define DEBUG_INIT 0
  20
  21 #define COMMAND_SIZE 256
  22 #define CONSOLE_NAME_SIZE 32
  23 #define MAXENV  16              /* Number of env. vars */
  24
  25 /*
  26  * When a file named CORE_ENABLE_FLAG_FILE exists, setrlimit is called
  27  * before processes are spawned to set core file size as unlimited.
  28  * This is for debugging only.  Don't use this is production, unless
  29  * you want core dumps lying about....
  30  */
  31 #define CORE_ENABLE_FLAG_FILE "/.init_enable_core"
  32 #include <sys/resource.h>
  33
  34 #define INITTAB      "/etc/inittab"     /* inittab file location */
  35 #ifndef INIT_SCRIPT
  36 #define INIT_SCRIPT  "/etc/init.d/rcS"  /* Default sysinit script. */
  37 #endif
  38
  39 /* Allowed init action types */
  40 #define SYSINIT     0x01
  41 #define RESPAWN     0x02
  42 /* like respawn, but wait for <Enter> to be pressed on tty: */
  43 #define ASKFIRST    0x04
  44 #define WAIT        0x08
  45 #define ONCE        0x10
  46 #define CTRLALTDEL  0x20
  47 #define SHUTDOWN    0x40
  48 #define RESTART     0x80
  49
  50 /* Set up a linked list of init_actions, to be read from inittab */
  51 struct init_action {
  52         struct init_action *next;
  53         pid_t pid;
  54         uint8_t action_type;
  55         char terminal[CONSOLE_NAME_SIZE];
  56         char command[COMMAND_SIZE];
  57 };
  58
  59 /* Static variables */
  60 static struct init_action *init_action_list = NULL;
  61
  62 static const char *log_console = VC_5;
  63 static sig_atomic_t got_cont = 0;
  64
  65 enum {
  66         L_LOG = 0x1,
  67         L_CONSOLE = 0x2,
  68
  69 #if ENABLE_FEATURE_EXTRA_QUIET
  70         MAYBE_CONSOLE = 0x0,
  71 #else
  72         MAYBE_CONSOLE = L_CONSOLE,
  73 #endif
  74
  75 #ifndef RB_HALT_SYSTEM
  76         RB_HALT_SYSTEM = 0xcdef0123, /* FIXME: this overflows enum */
  77         RB_ENABLE_CAD = 0x89abcdef,
  78         RB_DISABLE_CAD = 0,
  79         RB_POWER_OFF = 0x4321fedc,
  80         RB_AUTOBOOT = 0x01234567,
  81 #endif
  82 };
  83
  84 /* Function prototypes */
  85 static void halt_reboot_pwoff(int sig) NORETURN;
  86
  87 static void waitfor(pid_t pid)
  88 {
  89         /* waitfor(run(x)): protect against failed fork inside run() */
  90         if (pid <= 0)
  91                 return;
  92
  93         /* Wait for any child (prevent zombies from exiting orphaned processes)
  94          * but exit the loop only when specified one has exited. */
  95         while (wait(NULL) != pid)
  96                 continue;
  97 }
  98
  99 static void loop_forever(void) NORETURN;
 100 static void loop_forever(void)
 101 {
 102         while (1)
 103                 sleep(1);
 104 }
 105
 106 /* Print a message to the specified device.
 107  * "where" may be bitwise-or'd from L_LOG | L_CONSOLE
 108  * NB: careful, we can be called after vfork!
 109  */
 110 #define messageD(...) do { if (DEBUG_INIT) message(__VA_ARGS__); } while (0)
 111 static void message(int where, const char *fmt, ...)
 112         __attribute__ ((format(printf, 2, 3)));
 113 static void message(int where, const char *fmt, ...)
 114 {
 115         static int log_fd = -1;
 116         va_list arguments;
 117         unsigned l;
 118         char msg[128];
 119
 120         msg[0] = '\r';
 121         va_start(arguments, fmt);
 122         l = vsnprintf(msg + 1, sizeof(msg) - 2, fmt, arguments);
 123         if (l > sizeof(msg) - 2)
 124                 l = sizeof(msg) - 2;
 125         msg[l] = '\0';
 126         va_end(arguments);
 127
 128         if (ENABLE_FEATURE_INIT_SYSLOG) {
 129                 /* Log the message to syslogd */
 130                 if (where & L_LOG) {
 131                         /* don't print out "\r" */
 132                         openlog(applet_name, 0, LOG_DAEMON);
 133                         syslog(LOG_INFO, "init: %s", msg + 1);
 134                         closelog();
 135                 }
 136                 msg[l++] = '\n';
 137                 msg[l] = '\0';
 138         } else {
 139                 msg[l++] = '\n';
 140                 msg[l] = '\0';
 141                 /* Take full control of the log tty, and never close it.
 142                  * It's mine, all mine!  Muhahahaha! */
 143                 if (log_fd < 0) {
 144                         if (!log_console) {
 145                                 log_fd = STDERR_FILENO;
 146                         } else {
 147                                 log_fd = device_open(log_console, O_WRONLY | O_NONBLOCK | O_NOCTTY);
 148                                 if (log_fd < 0) {
 149                                         bb_error_msg("can't log to %s", log_console);
 150                                         where = L_CONSOLE;
 151                                 } else {
 152                                         close_on_exec_on(log_fd);
 153                                 }
 154                         }
 155                 }
 156                 if (where & L_LOG) {
 157                         full_write(log_fd, msg, l);
 158                         if (log_fd == STDERR_FILENO)
 159                                 return; /* don't print dup messages */
 160                 }
 161         }
 162
 163         if (where & L_CONSOLE) {
 164                 /* Send console messages to console so people will see them. */
 165                 full_write(STDERR_FILENO, msg, l);
 166         }
 167 }
 168
 169 /* From <linux/serial.h> */
 170 struct serial_struct {
 171         int     type;
 172         int     line;
 173         unsigned int    port;
 174         int     irq;
 175         int     flags;
 176         int     xmit_fifo_size;
 177         int     custom_divisor;
 178         int     baud_base;
 179         unsigned short  close_delay;
 180         char    io_type;
 181         char    reserved_char[1];
 182         int     hub6;
 183         unsigned short  closing_wait; /* time to wait before closing */
 184         unsigned short  closing_wait2; /* no longer used... */
 185         unsigned char   *iomem_base;
 186         unsigned short  iomem_reg_shift;
 187         unsigned int    port_high;
 188         unsigned long   iomap_base;     /* cookie passed into ioremap */
 189         int     reserved[1];
 190         /* Paranoia (imagine 64bit kernel overwriting 32bit userspace stack) */
 191         uint32_t bbox_reserved[16];
 192 };
 193 static void console_init(void)
 194 {
 195         struct serial_struct sr;
 196         char *s;
 197
 198         s = getenv("CONSOLE");
 199         if (!s)
 200                 s = getenv("console");
 201         if (s) {
 202                 int fd = open(s, O_RDWR | O_NONBLOCK | O_NOCTTY);
 203                 if (fd >= 0) {
 204                         dup2(fd, STDIN_FILENO);
 205                         dup2(fd, STDOUT_FILENO);
 206                         xmove_fd(fd, STDERR_FILENO);
 207                 }
 208                 messageD(L_LOG, "console='%s'", s);
 209         } else {
 210                 /* Make sure fd 0,1,2 are not closed
 211                  * (so that they won't be used by future opens) */
 212                 bb_sanitize_stdio();
 213                 /* Make sure init can't be blocked by writing to stderr */
 214                 fcntl(STDERR_FILENO, F_SETFL, fcntl(STDERR_FILENO, F_GETFL) | O_NONBLOCK);
 215         }
 216
 217         s = getenv("TERM");
 218         if (ioctl(STDIN_FILENO, TIOCGSERIAL, &sr) == 0) {
 219                 /* Force the TERM setting to vt102 for serial console
 220                  * if TERM is set to linux (the default) */
 221                 if (!s || strcmp(s, "linux") == 0)
 222                         putenv((char*)"TERM=vt102");
 223                 if (!ENABLE_FEATURE_INIT_SYSLOG)
 224                         log_console = NULL;
 225         } else if (!s)
 226                 putenv((char*)"TERM=linux");
 227 }
 228
 229 /* Set terminal settings to reasonable defaults.
 230  * NB: careful, we can be called after vfork! */
 231 static void set_sane_term(void)
 232 {
 233         struct termios tty;
 234
 235         tcgetattr(STDIN_FILENO, &tty);
 236
 237         /* set control chars */
 238         tty.c_cc[VINTR] = 3;    /* C-c */
 239         tty.c_cc[VQUIT] = 28;   /* C-\ */
 240         tty.c_cc[VERASE] = 127; /* C-? */
 241         tty.c_cc[VKILL] = 21;   /* C-u */
 242         tty.c_cc[VEOF] = 4;     /* C-d */
 243         tty.c_cc[VSTART] = 17;  /* C-q */
 244         tty.c_cc[VSTOP] = 19;   /* C-s */
 245         tty.c_cc[VSUSP] = 26;   /* C-z */
 246
 247         /* use line discipline 0 */
 248         tty.c_line = 0;
 249
 250         /* Make it be sane */
 251         tty.c_cflag &= CBAUD | CBAUDEX | CSIZE | CSTOPB | PARENB | PARODD;
 252         tty.c_cflag |= CREAD | HUPCL | CLOCAL;
 253
 254         /* input modes */
 255         tty.c_iflag = ICRNL | IXON | IXOFF;
 256
 257         /* output modes */
 258         tty.c_oflag = OPOST | ONLCR;
 259
 260         /* local modes */
 261         tty.c_lflag =
 262                 ISIG | ICANON | ECHO | ECHOE | ECHOK | ECHOCTL | ECHOKE | IEXTEN;
 263
 264         tcsetattr_stdin_TCSANOW(&tty);
 265 }
 266
 267 /* Open the new terminal device.
 268  * NB: careful, we can be called after vfork! */
 269 static void open_stdio_to_tty(const char* tty_name, int exit_on_failure)
 270 {
 271         /* empty tty_name means "use init's tty", else... */
 272         if (tty_name[0]) {
 273                 int fd;
 274                 close(STDIN_FILENO);
 275                 /* fd can be only < 0 or 0: */
 276                 fd = device_open(tty_name, O_RDWR);
 277                 if (fd) {
 278                         message(L_LOG | L_CONSOLE, "can't open %s: %s",
 279                                 tty_name, strerror(errno));
 280                         if (exit_on_failure)
 281                                 _exit(EXIT_FAILURE);
 282                         if (DEBUG_INIT)
 283                                 _exit(2);
 284                         /* NB: we don't reach this if we were called after vfork.
 285                          * Thus halt_reboot_pwoff() itself need not be vfork-safe. */
 286                         halt_reboot_pwoff(SIGUSR1); /* halt the system */
 287                 }
 288                 dup2(STDIN_FILENO, STDOUT_FILENO);
 289                 dup2(STDIN_FILENO, STDERR_FILENO);
 290         }
 291         set_sane_term();
 292 }
 293
 294 /* Wrapper around exec:
 295  * Takes string (max COMMAND_SIZE chars).
 296  * If chars like '>' detected, execs '[-]/bin/sh -c "exec ......."'.
 297  * Otherwise splits words on whitespace, deals with leading dash,
 298  * and uses plain exec().
 299  * NB: careful, we can be called after vfork!
 300  */
 301 static void init_exec(const char *command)
 302 {
 303         char *cmd[COMMAND_SIZE / 2];
 304         char buf[COMMAND_SIZE + 6];  /* COMMAND_SIZE+strlen("exec ")+1 */
 305         int dash = (command[0] == '-' /* maybe? && command[1] == '/' */);
 306
 307         /* See if any special /bin/sh requiring characters are present */
 308         if (strpbrk(command, "~`!$^&*()=|\\{}[];\"'<>?") != NULL) {
 309                 strcpy(buf, "exec ");
 310                 strcpy(buf + 5, command + dash); /* excluding "-" */
 311                 /* NB: LIBBB_DEFAULT_LOGIN_SHELL define has leading dash */
 312                 cmd[0] = (char*)(LIBBB_DEFAULT_LOGIN_SHELL + !dash);
 313                 cmd[1] = (char*)"-c";
 314                 cmd[2] = buf;
 315                 cmd[3] = NULL;
 316         } else {
 317                 /* Convert command (char*) into cmd (char**, one word per string) */
 318                 char *word, *next;
 319                 int i = 0;
 320                 next = strcpy(buf, command); /* including "-" */
 321                 while ((word = strsep(&next, " \t")) != NULL) {
 322                         if (*word != '\0') { /* not two spaces/tabs together? */
 323                                 cmd[i] = word;
 324                                 i++;
 325                         }
 326                 }
 327                 cmd[i] = NULL;
 328         }
 329         /* If we saw leading "-", it is interactive shell.
 330          * Try harder to give it a controlling tty.
 331          * And skip "-" in actual exec call. */
 332         if (dash) {
 333                 /* _Attempt_ to make stdin a controlling tty. */
 334                 if (ENABLE_FEATURE_INIT_SCTTY)
 335                         ioctl(STDIN_FILENO, TIOCSCTTY, 0 /*only try, don't steal*/);
 336         }
 337         BB_EXECVP(cmd[0] + dash, cmd);
 338         message(L_LOG | L_CONSOLE, "cannot run '%s': %s", cmd[0], strerror(errno));
 339         /* returns if execvp fails */
 340 }
 341
 342 /* Used only by run_actions */
 343 static pid_t run(const struct init_action *a)
 344 {
 345         pid_t pid;
 346         sigset_t nmask, omask;
 347
 348         /* Block sigchild while forking (why?) */
 349         sigemptyset(&nmask);
 350         sigaddset(&nmask, SIGCHLD);
 351         sigprocmask(SIG_BLOCK, &nmask, &omask);
 352         if (BB_MMU && (a->action_type & ASKFIRST))
 353                 pid = fork();
 354         else
 355                 pid = vfork();
 356         sigprocmask(SIG_SETMASK, &omask, NULL);
 357
 358         if (pid < 0)
 359                 message(L_LOG | L_CONSOLE, "can't fork");
 360         if (pid)
 361                 return pid;
 362
 363         /* Child */
 364
 365         /* Reset signal handlers that were set by the parent process */
 366         bb_signals(0
 367                 + (1 << SIGUSR1)
 368                 + (1 << SIGUSR2)
 369                 + (1 << SIGINT)
 370                 + (1 << SIGTERM)
 371                 + (1 << SIGHUP)
 372                 + (1 << SIGQUIT)
 373                 + (1 << SIGCONT)
 374                 + (1 << SIGSTOP)
 375                 + (1 << SIGTSTP)
 376                 , SIG_DFL);
 377
 378         /* Create a new session and make ourself the process
 379          * group leader */
 380         setsid();
 381
 382         /* Open the new terminal device */
 383         open_stdio_to_tty(a->terminal, 1 /* - exit if open fails */);
 384
 385 // NB: do not enable unless you change vfork to fork above
 386 #ifdef BUT_RUN_ACTIONS_ALREADY_DOES_WAITING
 387         /* If the init Action requires us to wait, then force the
 388          * supplied terminal to be the controlling tty. */
 389         if (a->action_type & (SYSINIT | WAIT | CTRLALTDEL | SHUTDOWN | RESTART)) {
 390                 /* Now fork off another process to just hang around */
 391                 pid = fork();
 392                 if (pid < 0) {
 393                         message(L_LOG | L_CONSOLE, "can't fork");
 394                         _exit(EXIT_FAILURE);
 395                 }
 396
 397                 if (pid > 0) {
 398                         /* Parent - wait till the child is done */
 399                         bb_signals(0
 400                                 + (1 << SIGINT)
 401                                 + (1 << SIGTSTP)
 402                                 + (1 << SIGQUIT)
 403                                 , SIG_IGN);
 404                         signal(SIGCHLD, SIG_DFL);
 405
 406                         waitfor(pid);
 407                         /* See if stealing the controlling tty back is necessary */
 408                         if (tcgetpgrp(0) != getpid())
 409                                 _exit(EXIT_SUCCESS);
 410
 411                         /* Use a temporary process to steal the controlling tty. */
 412                         pid = fork();
 413                         if (pid < 0) {
 414                                 message(L_LOG | L_CONSOLE, "can't fork");
 415                                 _exit(EXIT_FAILURE);
 416                         }
 417                         if (pid == 0) {
 418                                 setsid();
 419                                 ioctl(0, TIOCSCTTY, 1);
 420                                 _exit(EXIT_SUCCESS);
 421                         }
 422                         waitfor(pid);
 423                         _exit(EXIT_SUCCESS);
 424                 }
 425                 /* Child - fall though to actually execute things */
 426         }
 427 #endif
 428
 429         /* NB: on NOMMU we can't wait for input in child, so
 430          * "askfirst" will work the same as "respawn". */
 431         if (BB_MMU && (a->action_type & ASKFIRST)) {
 432                 static const char press_enter[] ALIGN1 =
 433 #ifdef CUSTOMIZED_BANNER
 434 #include CUSTOMIZED_BANNER
 435 #endif
 436                         "\nPlease press Enter to activate this console. ";
 437                 char c;
 438                 /*
 439                  * Save memory by not exec-ing anything large (like a shell)
 440                  * before the user wants it. This is critical if swap is not
 441                  * enabled and the system has low memory. Generally this will
 442                  * be run on the second virtual console, and the first will
 443                  * be allowed to start a shell or whatever an init script
 444                  * specifies.
 445                  */
 446                 messageD(L_LOG, "waiting for enter to start '%s'"
 447                                         "(pid %d, tty '%s')\n",
 448                                 a->command, getpid(), a->terminal);
 449                 full_write(STDOUT_FILENO, press_enter, sizeof(press_enter) - 1);
 450                 while (safe_read(STDIN_FILENO, &c, 1) == 1 && c != '\n')
 451                         continue;
 452         }
 453
 454         if (ENABLE_FEATURE_INIT_COREDUMPS) {
 455                 struct stat sb;
 456                 if (stat(CORE_ENABLE_FLAG_FILE, &sb) == 0) {
 457                         struct rlimit limit;
 458                         limit.rlim_cur = RLIM_INFINITY;
 459                         limit.rlim_max = RLIM_INFINITY;
 460                         setrlimit(RLIMIT_CORE, &limit);
 461                 }
 462         }
 463
 464         /* Log the process name and args */
 465         message(L_LOG, "starting pid %d, tty '%s': '%s'",
 466                           getpid(), a->terminal, a->command);
 467
 468         /* Now run it.  The new program will take over this PID,
 469          * so nothing further in init.c should be run. */
 470         init_exec(a->command);
 471         /* We're still here?  Some error happened. */
 472         _exit(-1);
 473 }
 474
 475 static void delete_init_action(struct init_action *action)
 476 {
 477         struct init_action *a, *b = NULL;
 478
 479         for (a = init_action_list; a; b = a, a = a->next) {
 480                 if (a == action) {
 481                         if (b == NULL) {
 482                                 init_action_list = a->next;
 483                         } else {
 484                                 b->next = a->next;
 485                         }
 486                         free(a);
 487                         break;
 488                 }
 489         }
 490 }
 491
 492 /* Run all commands of a particular type */
 493 static void run_actions(int action_type)
 494 {
 495         struct init_action *a, *tmp;
 496
 497         for (a = init_action_list; a; a = tmp) {
 498                 tmp = a->next;
 499                 if (a->action_type & action_type) {
 500                         // Pointless: run() will error out if open of device fails.
 501                         ///* a->terminal of "" means "init's console" */
 502                         //if (a->terminal[0] && access(a->terminal, R_OK | W_OK)) {
 503                         //      //message(L_LOG | L_CONSOLE, "Device %s cannot be opened in RW mode", a->terminal /*, strerror(errno)*/);
 504                         //      delete_init_action(a);
 505                         //} else
 506                         if (a->action_type & (SYSINIT | WAIT | CTRLALTDEL | SHUTDOWN | RESTART)) {
 507                                 waitfor(run(a));
 508                                 delete_init_action(a);
 509                         } else if (a->action_type & ONCE) {
 510                                 run(a);
 511                                 delete_init_action(a);
 512                         } else if (a->action_type & (RESPAWN | ASKFIRST)) {
 513                                 /* Only run stuff with pid==0.  If they have
 514                                  * a pid, that means it is still running */
 515                                 if (a->pid == 0) {
 516                                         a->pid = run(a);
 517                                 }
 518                         }
 519                 }
 520         }
 521 }
 522
 523 static void init_reboot(unsigned long magic)
 524 {
 525         pid_t pid;
 526         /* We have to fork here, since the kernel calls do_exit(EXIT_SUCCESS) in
 527          * linux/kernel/sys.c, which can cause the machine to panic when
 528          * the init process is killed.... */
 529         pid = vfork();
 530         if (pid == 0) { /* child */
 531                 reboot(magic);
 532                 _exit(EXIT_SUCCESS);
 533         }
 534         waitfor(pid);
 535 }
 536
 537 static void kill_all_processes(void)
 538 {
 539         /* run everything to be run at "shutdown".  This is done _prior_
 540          * to killing everything, in case people wish to use scripts to
 541          * shut things down gracefully... */
 542         run_actions(SHUTDOWN);
 543
 544         /* first disable all our signals */
 545         sigprocmask_allsigs(SIG_BLOCK);
 546
 547         message(L_CONSOLE | L_LOG, "The system is going down NOW!");
 548
 549         /* Allow Ctrl-Alt-Del to reboot system. */
 550         init_reboot(RB_ENABLE_CAD);
 551
 552         /* Send signals to every process _except_ pid 1 */
 553         message(L_CONSOLE | L_LOG, "Sending SIG%s to all processes", "TERM");
 554         kill(-1, SIGTERM);
 555         sync();
 556         sleep(1);
 557
 558         message(L_CONSOLE | L_LOG, "Sending SIG%s to all processes", "KILL");
 559         kill(-1, SIGKILL);
 560         sync();
 561         sleep(1);
 562 }
 563
 564 static void halt_reboot_pwoff(int sig)
 565 {
 566         const char *m = "halt";
 567         int rb;
 568
 569         kill_all_processes();
 570
 571         rb = RB_HALT_SYSTEM;
 572         if (sig == SIGTERM) {
 573                 m = "reboot";
 574                 rb = RB_AUTOBOOT;
 575         } else if (sig == SIGUSR2) {
 576                 m = "poweroff";
 577                 rb = RB_POWER_OFF;
 578         }
 579         message(L_CONSOLE | L_LOG, "Requesting system %s", m);
 580         /* allow time for last message to reach serial console */
 581         sleep(2);
 582         init_reboot(rb);
 583         loop_forever();
 584 }
 585
 586 /* Handler for QUIT - exec "restart" action,
 587  * else (no such action defined) do nothing */
 588 static void exec_restart_action(int sig UNUSED_PARAM)
 589 {
 590         struct init_action *a;
 591
 592         for (a = init_action_list; a; a = a->next) {
 593                 if (a->action_type & RESTART) {
 594                         kill_all_processes();
 595
 596                         /* unblock all signals (blocked in kill_all_processes()) */
 597                         sigprocmask_allsigs(SIG_UNBLOCK);
 598
 599                         /* Open the new terminal device */
 600                         open_stdio_to_tty(a->terminal, 0 /* - halt if open fails */);
 601
 602                         messageD(L_CONSOLE | L_LOG, "Trying to re-exec %s", a->command);
 603                         init_exec(a->command);
 604                         sleep(2);
 605                         init_reboot(RB_HALT_SYSTEM);
 606                         loop_forever();
 607                 }
 608         }
 609 }
 610
 611 static void ctrlaltdel_signal(int sig UNUSED_PARAM)
 612 {
 613         run_actions(CTRLALTDEL);
 614 }
 615
 616 /* The SIGSTOP & SIGTSTP handler */
 617 static void stop_handler(int sig UNUSED_PARAM)
 618 {
 619         int saved_errno = errno;
 620
 621         got_cont = 0;
 622         while (!got_cont)
 623                 pause();
 624
 625         errno = saved_errno;
 626 }
 627
 628 /* The SIGCONT handler */
 629 static void cont_handler(int sig UNUSED_PARAM)
 630 {
 631         got_cont = 1;
 632 }
 633
 634 static void new_init_action(uint8_t action_type, const char *command, const char *cons)
 635 {
 636         struct init_action *a, *last;
 637
 638 // Why?
 639 //      if (strcmp(cons, bb_dev_null) == 0 && (action & ASKFIRST))
 640 //              return;
 641
 642         /* Append to the end of the list */
 643         for (a = last = init_action_list; a; a = a->next) {
 644                 /* don't enter action if it's already in the list,
 645                  * but do overwrite existing actions */
 646                 if ((strcmp(a->command, command) == 0)
 647                  && (strcmp(a->terminal, cons) == 0)
 648                 ) {
 649                         a->action_type = action_type;
 650                         return;
 651                 }
 652                 last = a;
 653         }
 654
 655         a = xzalloc(sizeof(*a));
 656         if (last) {
 657                 last->next = a;
 658         } else {
 659                 init_action_list = a;
 660         }
 661         a->action_type = action_type;
 662         safe_strncpy(a->command, command, sizeof(a->command));
 663         safe_strncpy(a->terminal, cons, sizeof(a->terminal));
 664         messageD(L_LOG | L_CONSOLE, "command='%s' action=%d tty='%s'\n",
 665                 a->command, a->action_type, a->terminal);
 666 }
 667
 668 /* NOTE that if CONFIG_FEATURE_USE_INITTAB is NOT defined,
 669  * then parse_inittab() simply adds in some default
 670  * actions(i.e., runs INIT_SCRIPT and then starts a pair
 671  * of "askfirst" shells).  If CONFIG_FEATURE_USE_INITTAB
 672  * _is_ defined, but /etc/inittab is missing, this
 673  * results in the same set of default behaviors.
 674  */
 675 static void parse_inittab(void)
 676 {
 677         char *token[4];
 678         /* order must correspond to SYSINIT..RESTART constants */
 679         static const char actions[] ALIGN1 =
 680                 "sysinit\0""respawn\0""askfirst\0""wait\0""once\0"
 681                 "ctrlaltdel\0""shutdown\0""restart\0";
 682
 683         parser_t *parser = config_open2(INITTAB, fopen_for_read);
 684         /* No inittab file -- set up some default behavior */
 685         if (parser == NULL) {
 686                 /* Reboot on Ctrl-Alt-Del */
 687                 new_init_action(CTRLALTDEL, "reboot", "");
 688                 /* Umount all filesystems on halt/reboot */
 689                 new_init_action(SHUTDOWN, "umount -a -r", "");
 690                 /* Swapoff on halt/reboot */
 691                 if (ENABLE_SWAPONOFF)
 692                         new_init_action(SHUTDOWN, "swapoff -a", "");
 693                 /* Prepare to restart init when a QUIT is received */
 694                 new_init_action(RESTART, "init", "");
 695                 /* Askfirst shell on tty1-4 */
 696                 new_init_action(ASKFIRST, bb_default_login_shell, "");
 697 //TODO: VC_1 instead of ""? "" is console -> ctty problems -> angry users
 698                 new_init_action(ASKFIRST, bb_default_login_shell, VC_2);
 699                 new_init_action(ASKFIRST, bb_default_login_shell, VC_3);
 700                 new_init_action(ASKFIRST, bb_default_login_shell, VC_4);
 701                 /* sysinit */
 702                 new_init_action(SYSINIT, INIT_SCRIPT, "");
 703                 return;
 704         }
 705         /* optional_tty:ignored_runlevel:action:command
 706          * Delims are not to be collapsed and need exactly 4 tokens
 707          */
 708         while (config_read(parser, token, 4, 0, "#:",
 709                                 PARSE_NORMAL & ~(PARSE_TRIM | PARSE_COLLAPSE))) {
 710                 int action;
 711                 char *tty = token[0];
 712
 713                 if (!token[3]) /* less than 4 tokens */
 714                         goto bad_entry;
 715                 action = index_in_strings(actions, token[2]);
 716                 if (action < 0 || !token[3][0]) /* token[3]: command */
 717                         goto bad_entry;
 718                 /* turn .*TTY -> /dev/TTY */
 719                 if (tty[0]) {
 720                         if (strncmp(tty, "/dev/", 5) == 0)
 721                                 tty += 5;
 722                         tty = concat_path_file("/dev/", tty);
 723                 }
 724                 new_init_action(1 << action, token[3], tty);
 725                 if (tty[0])
 726                         free(tty);
 727                 continue;
 728  bad_entry:
 729                 message(L_LOG | L_CONSOLE, "Bad inittab entry at line %d",
 730                                 parser->lineno);
 731         }
 732         config_close(parser);
 733 }
 734
 735 #if ENABLE_FEATURE_USE_INITTAB
 736 static void reload_signal(int sig UNUSED_PARAM)
 737 {
 738         struct init_action *a, *tmp;
 739
 740         message(L_LOG, "reloading /etc/inittab");
 741
 742         /* disable old entrys */
 743         for (a = init_action_list; a; a = a->next) {
 744                 a->action_type = ONCE;
 745         }
 746
 747         parse_inittab();
 748
 749         if (ENABLE_FEATURE_KILL_REMOVED) {
 750                 /* Be nice and send SIGTERM first */
 751                 for (a = init_action_list; a; a = a->next) {
 752                         pid_t pid = a->pid;
 753                         if ((a->action_type & ONCE) && pid != 0) {
 754                                 kill(pid, SIGTERM);
 755                         }
 756                 }
 757 #if CONFIG_FEATURE_KILL_DELAY
 758                 /* NB: parent will wait in NOMMU case */
 759                 if ((BB_MMU ? fork() : vfork()) == 0) { /* child */
 760                         sleep(CONFIG_FEATURE_KILL_DELAY);
 761                         for (a = init_action_list; a; a = a->next) {
 762                                 pid_t pid = a->pid;
 763                                 if ((a->action_type & ONCE) && pid != 0) {
 764                                         kill(pid, SIGKILL);
 765                                 }
 766                         }
 767                         _exit(EXIT_SUCCESS);
 768                 }
 769 #endif
 770         }
 771
 772         /* remove unused entrys */
 773         for (a = init_action_list; a; a = tmp) {
 774                 tmp = a->next;
 775                 if ((a->action_type & (ONCE | SYSINIT | WAIT)) && a->pid == 0) {
 776                         delete_init_action(a);
 777                 }
 778         }
 779         run_actions(RESPAWN | ASKFIRST);
 780 }
 781 #endif
 782
 783 int init_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
 784 int init_main(int argc UNUSED_PARAM, char **argv)
 785 {
 786         struct init_action *a;
 787         pid_t wpid;
 788
 789         die_sleep = 30 * 24*60*60; /* if xmalloc will ever die... */
 790
 791         if (argv[1] && !strcmp(argv[1], "-q")) {
 792                 return kill(1, SIGHUP);
 793         }
 794
 795         if (!DEBUG_INIT) {
 796                 /* Expect to be invoked as init with PID=1 or be invoked as linuxrc */
 797                 if (getpid() != 1
 798                  && (!ENABLE_FEATURE_INITRD || !strstr(applet_name, "linuxrc"))
 799                 ) {
 800                         bb_show_usage();
 801                 }
 802                 /* Set up sig handlers  -- be sure to
 803                  * clear all of these in run() */
 804                 signal(SIGQUIT, exec_restart_action);
 805                 bb_signals(0
 806                         + (1 << SIGUSR1)  /* halt */
 807                         + (1 << SIGUSR2)  /* poweroff */
 808                         + (1 << SIGTERM)  /* reboot */
 809                         , halt_reboot_pwoff);
 810                 signal(SIGINT, ctrlaltdel_signal);
 811                 signal(SIGCONT, cont_handler);
 812                 bb_signals(0
 813                         + (1 << SIGSTOP)
 814                         + (1 << SIGTSTP)
 815                         , stop_handler);
 816
 817                 /* Turn off rebooting via CTL-ALT-DEL -- we get a
 818                  * SIGINT on CAD so we can shut things down gracefully... */
 819                 init_reboot(RB_DISABLE_CAD);
 820         }
 821
 822         /* Figure out where the default console should be */
 823         console_init();
 824         set_sane_term();
 825         xchdir("/");
 826         setsid();
 827
 828         /* Make sure environs is set to something sane */
 829         putenv((char *) "HOME=/");
 830         putenv((char *) bb_PATH_root_path);
 831         putenv((char *) "SHELL=/bin/sh");
 832         putenv((char *) "USER=root"); /* needed? why? */
 833
 834         if (argv[1])
 835                 xsetenv("RUNLEVEL", argv[1]);
 836
 837         /* Hello world */
 838         message(MAYBE_CONSOLE | L_LOG, "init started: %s", bb_banner);
 839
 840         /* Make sure there is enough memory to do something useful. */
 841         if (ENABLE_SWAPONOFF) {
 842                 struct sysinfo info;
 843
 844                 if (!sysinfo(&info) &&
 845                         (info.mem_unit ? : 1) * (long long)info.totalram < 1024*1024)
 846                 {
 847                         message(L_CONSOLE, "Low memory, forcing swapon");
 848                         /* swapon -a requires /proc typically */
 849                         new_init_action(SYSINIT, "mount -t proc proc /proc", "");
 850                         /* Try to turn on swap */
 851                         new_init_action(SYSINIT, "swapon -a", "");
 852                         run_actions(SYSINIT);   /* wait and removing */
 853                 }
 854         }
 855
 856         /* Check if we are supposed to be in single user mode */
 857         if (argv[1]
 858          && (!strcmp(argv[1], "single") || !strcmp(argv[1], "-s") || LONE_CHAR(argv[1], '1'))
 859         ) {
 860                 /* ??? shouldn't we set RUNLEVEL="b" here? */
 861                 /* Start a shell on console */
 862                 new_init_action(RESPAWN, bb_default_login_shell, "");
 863         } else {
 864                 /* Not in single user mode -- see what inittab says */
 865
 866                 /* NOTE that if CONFIG_FEATURE_USE_INITTAB is NOT defined,
 867                  * then parse_inittab() simply adds in some default
 868                  * actions(i.e., runs INIT_SCRIPT and then starts a pair
 869                  * of "askfirst" shells */
 870                 parse_inittab();
 871         }
 872
 873 #if ENABLE_SELINUX
 874         if (getenv("SELINUX_INIT") == NULL) {
 875                 int enforce = 0;
 876
 877                 putenv((char*)"SELINUX_INIT=YES");
 878                 if (selinux_init_load_policy(&enforce) == 0) {
 879                         BB_EXECVP(argv[0], argv);
 880                 } else if (enforce > 0) {
 881                         /* SELinux in enforcing mode but load_policy failed */
 882                         message(L_CONSOLE, "cannot load SELinux Policy. "
 883                                 "Machine is in enforcing mode. Halting now.");
 884                         exit(EXIT_FAILURE);
 885                 }
 886         }
 887 #endif /* CONFIG_SELINUX */
 888
 889         /* Make the command line just say "init"  - thats all, nothing else */
 890         strncpy(argv[0], "init", strlen(argv[0]));
 891         /* Wipe argv[1]-argv[N] so they don't clutter the ps listing */
 892         while (*++argv)
 893                 memset(*argv, 0, strlen(*argv));
 894
 895         /* Now run everything that needs to be run */
 896
 897         /* First run the sysinit command */
 898         run_actions(SYSINIT);
 899
 900         /* Next run anything that wants to block */
 901         run_actions(WAIT);
 902
 903         /* Next run anything to be run only once */
 904         run_actions(ONCE);
 905
 906         /* Redefine SIGHUP to reread /etc/inittab */
 907 #if ENABLE_FEATURE_USE_INITTAB
 908         signal(SIGHUP, reload_signal);
 909 #else
 910         signal(SIGHUP, SIG_IGN);
 911 #endif
 912
 913         /* Now run the looping stuff for the rest of forever */
 914         while (1) {
 915                 /* run the respawn/askfirst stuff */
 916                 run_actions(RESPAWN | ASKFIRST);
 917
 918                 /* Don't consume all CPU time -- sleep a bit */
 919                 sleep(1);
 920
 921                 /* Wait for any child process to exit */
 922                 wpid = wait(NULL);
 923                 while (wpid > 0) {
 924                         /* Find out who died and clean up their corpse */
 925                         for (a = init_action_list; a; a = a->next) {
 926                                 if (a->pid == wpid) {
 927                                         /* Set the pid to 0 so that the process gets
 928                                          * restarted by run_actions() */
 929                                         a->pid = 0;
 930                                         message(L_LOG, "process '%s' (pid %d) exited. "
 931                                                         "Scheduling for restart.",
 932                                                         a->command, wpid);
 933                                 }
 934                         }
 935                         /* see if anyone else is waiting to be reaped */
 936                         wpid = wait_any_nohang(NULL);
 937                 }
 938         }
 939 }