init.c

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