init/init.c

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