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