sbin/init/init.c

   1 /*-
   2  * Copyright (c) 1991, 1993
   3  *      The Regents of the University of California.  All rights reserved.
   4  *
   5  * This code is derived from software contributed to Berkeley by
   6  * Donn Seeley at Berkeley Software Design, Inc.
   7  *
   8  * Redistribution and use in source and binary forms, with or without
   9  * modification, are permitted provided that the following conditions
  10  * are met:
  11  * 1. Redistributions of source code must retain the above copyright
  12  *    notice, this list of conditions and the following disclaimer.
  13  * 2. Redistributions in binary form must reproduce the above copyright
  14  *    notice, this list of conditions and the following disclaimer in the
  15  *    documentation and/or other materials provided with the distribution.
  16  * 4. Neither the name of the University nor the names of its contributors
  17  *    may be used to endorse or promote products derived from this software
  18  *    without specific prior written permission.
  19  *
  20  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
  21  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  22  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  23  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
  24  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  25  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  26  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  27  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  28  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  29  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  30  * SUCH DAMAGE.
  31  */
  32
  33 #ifndef lint
  34 static const char copyright[] =
  35 "@(#) Copyright (c) 1991, 1993\n\
  36         The Regents of the University of California.  All rights reserved.\n";
  37 #endif /* not lint */
  38
  39 #ifndef lint
  40 #if 0
  41 static char sccsid[] = "@(#)init.c      8.1 (Berkeley) 7/15/93";
  42 #endif
  43 static const char rcsid[] =
  44   "$FreeBSD$";
  45 #endif /* not lint */
  46
  47 #include <sys/param.h>
  48 #include <sys/ioctl.h>
  49 #include <sys/mman.h>
  50 #include <sys/mount.h>
  51 #include <sys/sysctl.h>
  52 #include <sys/wait.h>
  53 #include <sys/stat.h>
  54 #include <sys/uio.h>
  55
  56 #include <db.h>
  57 #include <errno.h>
  58 #include <fcntl.h>
  59 #include <kenv.h>
  60 #include <libutil.h>
  61 #include <paths.h>
  62 #include <signal.h>
  63 #include <stdio.h>
  64 #include <stdlib.h>
  65 #include <string.h>
  66 #include <syslog.h>
  67 #include <time.h>
  68 #include <ttyent.h>
  69 #include <unistd.h>
  70 #include <sys/reboot.h>
  71 #include <err.h>
  72
  73 #include <stdarg.h>
  74
  75 #ifdef SECURE
  76 #include <pwd.h>
  77 #endif
  78
  79 #ifdef LOGIN_CAP
  80 #include <login_cap.h>
  81 #endif
  82
  83 #include "mntopts.h"
  84 #include "pathnames.h"
  85
  86 /*
  87  * Sleep times; used to prevent thrashing.
  88  */
  89 #define GETTY_SPACING            5      /* N secs minimum getty spacing */
  90 #define GETTY_SLEEP             30      /* sleep N secs after spacing problem */
  91 #define GETTY_NSPACE             3      /* max. spacing count to bring reaction */
  92 #define WINDOW_WAIT              3      /* wait N secs after starting window */
  93 #define STALL_TIMEOUT           30      /* wait N secs after warning */
  94 #define DEATH_WATCH             10      /* wait N secs for procs to die */
  95 #define DEATH_SCRIPT            120     /* wait for 2min for /etc/rc.shutdown */
  96 #define RESOURCE_RC             "daemon"
  97 #define RESOURCE_WINDOW         "default"
  98 #define RESOURCE_GETTY          "default"
  99
 100 static void handle(sig_t, ...);
 101 static void delset(sigset_t *, ...);
 102
 103 static void stall(const char *, ...) __printflike(1, 2);
 104 static void warning(const char *, ...) __printflike(1, 2);
 105 static void emergency(const char *, ...) __printflike(1, 2);
 106 static void disaster(int);
 107 static void badsys(int);
 108 static void revoke_ttys(void);
 109 static int  runshutdown(void);
 110 static char *strk(char *);
 111
 112 /*
 113  * We really need a recursive typedef...
 114  * The following at least guarantees that the return type of (*state_t)()
 115  * is sufficiently wide to hold a function pointer.
 116  */
 117 typedef long (*state_func_t)(void);
 118 typedef state_func_t (*state_t)(void);
 119
 120 static state_func_t single_user(void);
 121 static state_func_t runcom(void);
 122 static state_func_t read_ttys(void);
 123 static state_func_t multi_user(void);
 124 static state_func_t clean_ttys(void);
 125 static state_func_t catatonia(void);
 126 static state_func_t death(void);
 127 static state_func_t death_single(void);
 128 static state_func_t reroot(void);
 129 static state_func_t reroot_phase_two(void);
 130
 131 static state_func_t run_script(const char *);
 132
 133 static enum { AUTOBOOT, FASTBOOT } runcom_mode = AUTOBOOT;
 134 #define FALSE   0
 135 #define TRUE    1
 136
 137 static int Reboot = FALSE;
 138 static int howto = RB_AUTOBOOT;
 139
 140 static int devfs;
 141 static char *init_path_argv0;
 142
 143 static void transition(state_t);
 144 static state_t requested_transition;
 145 static state_t current_state = death_single;
 146
 147 static void open_console(void);
 148 static const char *get_shell(void);
 149 static void write_stderr(const char *message);
 150
 151 typedef struct init_session {
 152         pid_t   se_process;             /* controlling process */
 153         time_t  se_started;             /* used to avoid thrashing */
 154         int     se_flags;               /* status of session */
 155 #define SE_SHUTDOWN     0x1             /* session won't be restarted */
 156 #define SE_PRESENT      0x2             /* session is in /etc/ttys */
 157         int     se_nspace;              /* spacing count */
 158         char    *se_device;             /* filename of port */
 159         char    *se_getty;              /* what to run on that port */
 160         char    *se_getty_argv_space;   /* pre-parsed argument array space */
 161         char    **se_getty_argv;        /* pre-parsed argument array */
 162         char    *se_window;             /* window system (started only once) */
 163         char    *se_window_argv_space;  /* pre-parsed argument array space */
 164         char    **se_window_argv;       /* pre-parsed argument array */
 165         char    *se_type;               /* default terminal type */
 166         struct  init_session *se_prev;
 167         struct  init_session *se_next;
 168 } session_t;
 169
 170 static void free_session(session_t *);
 171 static session_t *new_session(session_t *, struct ttyent *);
 172 static session_t *sessions;
 173
 174 static char **construct_argv(char *);
 175 static void start_window_system(session_t *);
 176 static void collect_child(pid_t);
 177 static pid_t start_getty(session_t *);
 178 static void transition_handler(int);
 179 static void alrm_handler(int);
 180 static void setsecuritylevel(int);
 181 static int getsecuritylevel(void);
 182 static int setupargv(session_t *, struct ttyent *);
 183 #ifdef LOGIN_CAP
 184 static void setprocresources(const char *);
 185 #endif
 186 static int clang;
 187
 188 static int start_session_db(void);
 189 static void add_session(session_t *);
 190 static void del_session(session_t *);
 191 static session_t *find_session(pid_t);
 192 static DB *session_db;
 193
 194 /*
 195  * The mother of all processes.
 196  */
 197 int
 198 main(int argc, char *argv[])
 199 {
 200         state_t initial_transition = runcom;
 201         char kenv_value[PATH_MAX];
 202         int c, error;
 203         struct sigaction sa;
 204         sigset_t mask;
 205
 206         /* Dispose of random users. */
 207         if (getuid() != 0)
 208                 errx(1, "%s", strerror(EPERM));
 209
 210         /* System V users like to reexec init. */
 211         if (getpid() != 1) {
 212 #ifdef COMPAT_SYSV_INIT
 213                 /* So give them what they want */
 214                 if (argc > 1) {
 215                         if (strlen(argv[1]) == 1) {
 216                                 char runlevel = *argv[1];
 217                                 int sig;
 218
 219                                 switch (runlevel) {
 220                                 case '0': /* halt + poweroff */
 221                                         sig = SIGUSR2;
 222                                         break;
 223                                 case '1': /* single-user */
 224                                         sig = SIGTERM;
 225                                         break;
 226                                 case '6': /* reboot */
 227                                         sig = SIGINT;
 228                                         break;
 229                                 case 'c': /* block further logins */
 230                                         sig = SIGTSTP;
 231                                         break;
 232                                 case 'q': /* rescan /etc/ttys */
 233                                         sig = SIGHUP;
 234                                         break;
 235                                 case 'r': /* remount root */
 236                                         sig = SIGEMT;
 237                                         break;
 238                                 default:
 239                                         goto invalid;
 240                                 }
 241                                 kill(1, sig);
 242                                 _exit(0);
 243                         } else
 244 invalid:
 245                                 errx(1, "invalid run-level ``%s''", argv[1]);
 246                 } else
 247 #endif
 248                         errx(1, "already running");
 249         }
 250
 251         init_path_argv0 = strdup(argv[0]);
 252         if (init_path_argv0 == NULL)
 253                 err(1, "strdup");
 254
 255         /*
 256          * Note that this does NOT open a file...
 257          * Does 'init' deserve its own facility number?
 258          */
 259         openlog("init", LOG_CONS, LOG_AUTH);
 260
 261         /*
 262          * Create an initial session.
 263          */
 264         if (setsid() < 0 && (errno != EPERM || getsid(0) != 1))
 265                 warning("initial setsid() failed: %m");
 266
 267         /*
 268          * Establish an initial user so that programs running
 269          * single user do not freak out and die (like passwd).
 270          */
 271         if (setlogin("root") < 0)
 272                 warning("setlogin() failed: %m");
 273
 274         /*
 275          * This code assumes that we always get arguments through flags,
 276          * never through bits set in some random machine register.
 277          */
 278         while ((c = getopt(argc, argv, "dsfr")) != -1)
 279                 switch (c) {
 280                 case 'd':
 281                         devfs = 1;
 282                         break;
 283                 case 's':
 284                         initial_transition = single_user;
 285                         break;
 286                 case 'f':
 287                         runcom_mode = FASTBOOT;
 288                         break;
 289                 case 'r':
 290                         initial_transition = reroot_phase_two;
 291                         break;
 292                 default:
 293                         warning("unrecognized flag '-%c'", c);
 294                         break;
 295                 }
 296
 297         if (optind != argc)
 298                 warning("ignoring excess arguments");
 299
 300         /*
 301          * We catch or block signals rather than ignore them,
 302          * so that they get reset on exec.
 303          */
 304         handle(badsys, SIGSYS, 0);
 305         handle(disaster, SIGABRT, SIGFPE, SIGILL, SIGSEGV, SIGBUS, SIGXCPU,
 306             SIGXFSZ, 0);
 307         handle(transition_handler, SIGHUP, SIGINT, SIGEMT, SIGTERM, SIGTSTP,
 308             SIGUSR1, SIGUSR2, 0);
 309         handle(alrm_handler, SIGALRM, 0);
 310         sigfillset(&mask);
 311         delset(&mask, SIGABRT, SIGFPE, SIGILL, SIGSEGV, SIGBUS, SIGSYS,
 312             SIGXCPU, SIGXFSZ, SIGHUP, SIGINT, SIGEMT, SIGTERM, SIGTSTP,
 313             SIGALRM, SIGUSR1, SIGUSR2, 0);
 314         sigprocmask(SIG_SETMASK, &mask, (sigset_t *) 0);
 315         sigemptyset(&sa.sa_mask);
 316         sa.sa_flags = 0;
 317         sa.sa_handler = SIG_IGN;
 318         sigaction(SIGTTIN, &sa, (struct sigaction *)0);
 319         sigaction(SIGTTOU, &sa, (struct sigaction *)0);
 320
 321         /*
 322          * Paranoia.
 323          */
 324         close(0);
 325         close(1);
 326         close(2);
 327
 328         if (kenv(KENV_GET, "init_script", kenv_value, sizeof(kenv_value)) > 0) {
 329                 state_func_t next_transition;
 330
 331                 if ((next_transition = run_script(kenv_value)) != NULL)
 332                         initial_transition = (state_t) next_transition;
 333         }
 334
 335         if (kenv(KENV_GET, "init_chroot", kenv_value, sizeof(kenv_value)) > 0) {
 336                 if (chdir(kenv_value) != 0 || chroot(".") != 0)
 337                         warning("Can't chroot to %s: %m", kenv_value);
 338         }
 339
 340         /*
 341          * Additional check if devfs needs to be mounted:
 342          * If "/" and "/dev" have the same device number,
 343          * then it hasn't been mounted yet.
 344          */
 345         if (!devfs) {
 346                 struct stat stst;
 347                 dev_t root_devno;
 348
 349                 stat("/", &stst);
 350                 root_devno = stst.st_dev;
 351                 if (stat("/dev", &stst) != 0)
 352                         warning("Can't stat /dev: %m");
 353                 else if (stst.st_dev == root_devno)
 354                         devfs++;
 355         }
 356
 357         if (devfs) {
 358                 struct iovec iov[4];
 359                 char *s;
 360                 int i;
 361
 362                 char _fstype[]  = "fstype";
 363                 char _devfs[]   = "devfs";
 364                 char _fspath[]  = "fspath";
 365                 char _path_dev[]= _PATH_DEV;
 366
 367                 iov[0].iov_base = _fstype;
 368                 iov[0].iov_len = sizeof(_fstype);
 369                 iov[1].iov_base = _devfs;
 370                 iov[1].iov_len = sizeof(_devfs);
 371                 iov[2].iov_base = _fspath;
 372                 iov[2].iov_len = sizeof(_fspath);
 373                 /*
 374                  * Try to avoid the trailing slash in _PATH_DEV.
 375                  * Be *very* defensive.
 376                  */
 377                 s = strdup(_PATH_DEV);
 378                 if (s != NULL) {
 379                         i = strlen(s);
 380                         if (i > 0 && s[i - 1] == '/')
 381                                 s[i - 1] = '\0';
 382                         iov[3].iov_base = s;
 383                         iov[3].iov_len = strlen(s) + 1;
 384                 } else {
 385                         iov[3].iov_base = _path_dev;
 386                         iov[3].iov_len = sizeof(_path_dev);
 387                 }
 388                 nmount(iov, 4, 0);
 389                 if (s != NULL)
 390                         free(s);
 391         }
 392
 393         if (initial_transition != reroot_phase_two) {
 394                 /*
 395                  * Unmount reroot leftovers.  This runs after init(8)
 396                  * gets reexecuted after reroot_phase_two() is done.
 397                  */
 398                 error = unmount(_PATH_REROOT, MNT_FORCE);
 399                 if (error != 0 && errno != EINVAL)
 400                         warning("Cannot unmount %s: %m", _PATH_REROOT);
 401         }
 402
 403         /*
 404          * Start the state machine.
 405          */
 406         transition(initial_transition);
 407
 408         /*
 409          * Should never reach here.
 410          */
 411         return 1;
 412 }
 413
 414 /*
 415  * Associate a function with a signal handler.
 416  */
 417 static void
 418 handle(sig_t handler, ...)
 419 {
 420         int sig;
 421         struct sigaction sa;
 422         sigset_t mask_everything;
 423         va_list ap;
 424         va_start(ap, handler);
 425
 426         sa.sa_handler = handler;
 427         sigfillset(&mask_everything);
 428
 429         while ((sig = va_arg(ap, int)) != 0) {
 430                 sa.sa_mask = mask_everything;
 431                 /* XXX SA_RESTART? */
 432                 sa.sa_flags = sig == SIGCHLD ? SA_NOCLDSTOP : 0;
 433                 sigaction(sig, &sa, (struct sigaction *) 0);
 434         }
 435         va_end(ap);
 436 }
 437
 438 /*
 439  * Delete a set of signals from a mask.
 440  */
 441 static void
 442 delset(sigset_t *maskp, ...)
 443 {
 444         int sig;
 445         va_list ap;
 446         va_start(ap, maskp);
 447
 448         while ((sig = va_arg(ap, int)) != 0)
 449                 sigdelset(maskp, sig);
 450         va_end(ap);
 451 }
 452
 453 /*
 454  * Log a message and sleep for a while (to give someone an opportunity
 455  * to read it and to save log or hardcopy output if the problem is chronic).
 456  * NB: should send a message to the session logger to avoid blocking.
 457  */
 458 static void
 459 stall(const char *message, ...)
 460 {
 461         va_list ap;
 462         va_start(ap, message);
 463
 464         vsyslog(LOG_ALERT, message, ap);
 465         va_end(ap);
 466         sleep(STALL_TIMEOUT);
 467 }
 468
 469 /*
 470  * Like stall(), but doesn't sleep.
 471  * If cpp had variadic macros, the two functions could be #defines for another.
 472  * NB: should send a message to the session logger to avoid blocking.
 473  */
 474 static void
 475 warning(const char *message, ...)
 476 {
 477         va_list ap;
 478         va_start(ap, message);
 479
 480         vsyslog(LOG_ALERT, message, ap);
 481         va_end(ap);
 482 }
 483
 484 /*
 485  * Log an emergency message.
 486  * NB: should send a message to the session logger to avoid blocking.
 487  */
 488 static void
 489 emergency(const char *message, ...)
 490 {
 491         va_list ap;
 492         va_start(ap, message);
 493
 494         vsyslog(LOG_EMERG, message, ap);
 495         va_end(ap);
 496 }
 497
 498 /*
 499  * Catch a SIGSYS signal.
 500  *
 501  * These may arise if a system does not support sysctl.
 502  * We tolerate up to 25 of these, then throw in the towel.
 503  */
 504 static void
 505 badsys(int sig)
 506 {
 507         static int badcount = 0;
 508
 509         if (badcount++ < 25)
 510                 return;
 511         disaster(sig);
 512 }
 513
 514 /*
 515  * Catch an unexpected signal.
 516  */
 517 static void
 518 disaster(int sig)
 519 {
 520
 521         emergency("fatal signal: %s",
 522             (unsigned)sig < NSIG ? sys_siglist[sig] : "unknown signal");
 523
 524         sleep(STALL_TIMEOUT);
 525         _exit(sig);             /* reboot */
 526 }
 527
 528 /*
 529  * Get the security level of the kernel.
 530  */
 531 static int
 532 getsecuritylevel(void)
 533 {
 534 #ifdef KERN_SECURELVL
 535         int name[2], curlevel;
 536         size_t len;
 537
 538         name[0] = CTL_KERN;
 539         name[1] = KERN_SECURELVL;
 540         len = sizeof curlevel;
 541         if (sysctl(name, 2, &curlevel, &len, NULL, 0) == -1) {
 542                 emergency("cannot get kernel security level: %s",
 543                     strerror(errno));
 544                 return (-1);
 545         }
 546         return (curlevel);
 547 #else
 548         return (-1);
 549 #endif
 550 }
 551
 552 /*
 553  * Set the security level of the kernel.
 554  */
 555 static void
 556 setsecuritylevel(int newlevel)
 557 {
 558 #ifdef KERN_SECURELVL
 559         int name[2], curlevel;
 560
 561         curlevel = getsecuritylevel();
 562         if (newlevel == curlevel)
 563                 return;
 564         name[0] = CTL_KERN;
 565         name[1] = KERN_SECURELVL;
 566         if (sysctl(name, 2, NULL, NULL, &newlevel, sizeof newlevel) == -1) {
 567                 emergency(
 568                     "cannot change kernel security level from %d to %d: %s",
 569                     curlevel, newlevel, strerror(errno));
 570                 return;
 571         }
 572 #ifdef SECURE
 573         warning("kernel security level changed from %d to %d",
 574             curlevel, newlevel);
 575 #endif
 576 #endif
 577 }
 578
 579 /*
 580  * Change states in the finite state machine.
 581  * The initial state is passed as an argument.
 582  */
 583 static void
 584 transition(state_t s)
 585 {
 586
 587         current_state = s;
 588         for (;;)
 589                 current_state = (state_t) (*current_state)();
 590 }
 591
 592 /*
 593  * Start a session and allocate a controlling terminal.
 594  * Only called by children of init after forking.
 595  */
 596 static void
 597 open_console(void)
 598 {
 599         int fd;
 600
 601         /*
 602          * Try to open /dev/console.  Open the device with O_NONBLOCK to
 603          * prevent potential blocking on a carrier.
 604          */
 605         revoke(_PATH_CONSOLE);
 606         if ((fd = open(_PATH_CONSOLE, O_RDWR | O_NONBLOCK)) != -1) {
 607                 (void)fcntl(fd, F_SETFL, fcntl(fd, F_GETFL) & ~O_NONBLOCK);
 608                 if (login_tty(fd) == 0)
 609                         return;
 610                 close(fd);
 611         }
 612
 613         /* No luck.  Log output to file if possible. */
 614         if ((fd = open(_PATH_DEVNULL, O_RDWR)) == -1) {
 615                 stall("cannot open null device.");
 616                 _exit(1);
 617         }
 618         if (fd != STDIN_FILENO) {
 619                 dup2(fd, STDIN_FILENO);
 620                 close(fd);
 621         }
 622         fd = open(_PATH_INITLOG, O_WRONLY | O_APPEND | O_CREAT, 0644);
 623         if (fd == -1)
 624                 dup2(STDIN_FILENO, STDOUT_FILENO);
 625         else if (fd != STDOUT_FILENO) {
 626                 dup2(fd, STDOUT_FILENO);
 627                 close(fd);
 628         }
 629         dup2(STDOUT_FILENO, STDERR_FILENO);
 630 }
 631
 632 static const char *
 633 get_shell(void)
 634 {
 635         static char kenv_value[PATH_MAX];
 636
 637         if (kenv(KENV_GET, "init_shell", kenv_value, sizeof(kenv_value)) > 0)
 638                 return kenv_value;
 639         else
 640                 return _PATH_BSHELL;
 641 }
 642
 643 static void
 644 write_stderr(const char *message)
 645 {
 646
 647         write(STDERR_FILENO, message, strlen(message));
 648 }
 649
 650 static int
 651 read_file(const char *path, void **bufp, size_t *bufsizep)
 652 {
 653         struct stat sb;
 654         size_t bufsize;
 655         void *buf;
 656         ssize_t nbytes;
 657         int error, fd;
 658
 659         fd = open(path, O_RDONLY);
 660         if (fd < 0) {
 661                 emergency("%s: %s", path, strerror(errno));
 662                 return (-1);
 663         }
 664
 665         error = fstat(fd, &sb);
 666         if (error != 0) {
 667                 emergency("fstat: %s", strerror(errno));
 668                 close(fd);
 669                 return (error);
 670         }
 671
 672         bufsize = sb.st_size;
 673         buf = malloc(bufsize);
 674         if (buf == NULL) {
 675                 emergency("malloc: %s", strerror(errno));
 676                 close(fd);
 677                 return (error);
 678         }
 679
 680         nbytes = read(fd, buf, bufsize);
 681         if (nbytes != (ssize_t)bufsize) {
 682                 emergency("read: %s", strerror(errno));
 683                 close(fd);
 684                 free(buf);
 685                 return (error);
 686         }
 687
 688         error = close(fd);
 689         if (error != 0) {
 690                 emergency("close: %s", strerror(errno));
 691                 free(buf);
 692                 return (error);
 693         }
 694
 695         *bufp = buf;
 696         *bufsizep = bufsize;
 697
 698         return (0);
 699 }
 700
 701 static int
 702 create_file(const char *path, const void *buf, size_t bufsize)
 703 {
 704         ssize_t nbytes;
 705         int error, fd;
 706
 707         fd = open(path, O_WRONLY | O_CREAT | O_EXCL, 0700);
 708         if (fd < 0) {
 709                 emergency("%s: %s", path, strerror(errno));
 710                 return (-1);
 711         }
 712
 713         nbytes = write(fd, buf, bufsize);
 714         if (nbytes != (ssize_t)bufsize) {
 715                 emergency("write: %s", strerror(errno));
 716                 close(fd);
 717                 return (-1);
 718         }
 719
 720         error = close(fd);
 721         if (error != 0) {
 722                 emergency("close: %s", strerror(errno));
 723                 return (-1);
 724         }
 725
 726         return (0);
 727 }
 728
 729 static int
 730 mount_tmpfs(const char *fspath)
 731 {
 732         struct iovec *iov;
 733         char errmsg[255];
 734         int error, iovlen;
 735
 736         iov = NULL;
 737         iovlen = 0;
 738         memset(errmsg, 0, sizeof(errmsg));
 739         build_iovec(&iov, &iovlen, "fstype",
 740             __DECONST(void *, "tmpfs"), (size_t)-1);
 741         build_iovec(&iov, &iovlen, "fspath",
 742             __DECONST(void *, fspath), (size_t)-1);
 743         build_iovec(&iov, &iovlen, "errmsg",
 744             errmsg, sizeof(errmsg));
 745
 746         error = nmount(iov, iovlen, 0);
 747         if (error != 0) {
 748                 if (*errmsg != '\0') {
 749                         emergency("cannot mount tmpfs on %s: %s: %s",
 750                             fspath, errmsg, strerror(errno));
 751                 } else {
 752                         emergency("cannot mount tmpfs on %s: %s",
 753                             fspath, strerror(errno));
 754                 }
 755                 return (error);
 756         }
 757         return (0);
 758 }
 759
 760 static state_func_t
 761 reroot(void)
 762 {
 763         void *buf;
 764         size_t bufsize;
 765         int error;
 766
 767         buf = NULL;
 768         bufsize = 0;
 769
 770         revoke_ttys();
 771         runshutdown();
 772
 773         /*
 774          * Make sure nobody can interfere with our scheme.
 775          * Ignore ESRCH, which can apparently happen when
 776          * there are no processes to kill.
 777          */
 778         error = kill(-1, SIGKILL);
 779         if (error != 0 && errno != ESRCH) {
 780                 emergency("kill(2) failed: %s", strerror(errno));
 781                 goto out;
 782         }
 783
 784         /*
 785          * Copy the init binary into tmpfs, so that we can unmount
 786          * the old rootfs without committing suicide.
 787          */
 788         error = read_file(init_path_argv0, &buf, &bufsize);
 789         if (error != 0)
 790                 goto out;
 791         error = mount_tmpfs(_PATH_REROOT);
 792         if (error != 0)
 793                 goto out;
 794         error = create_file(_PATH_REROOT_INIT, buf, bufsize);
 795         if (error != 0)
 796                 goto out;
 797
 798         /*
 799          * Execute the temporary init.
 800          */
 801         execl(_PATH_REROOT_INIT, _PATH_REROOT_INIT, "-r", NULL);
 802         emergency("cannot exec %s: %s", _PATH_REROOT_INIT, strerror(errno));
 803
 804 out:
 805         emergency("reroot failed; going to single user mode");
 806         free(buf);
 807         return (state_func_t) single_user;
 808 }
 809
 810 static state_func_t
 811 reroot_phase_two(void)
 812 {
 813         char init_path[PATH_MAX], *path, *path_component;
 814         size_t init_path_len;
 815         int nbytes, error;
 816
 817         /*
 818          * Ask the kernel to mount the new rootfs.
 819          */
 820         error = reboot(RB_REROOT);
 821         if (error != 0) {
 822                 emergency("RB_REBOOT failed: %s", strerror(errno));
 823                 goto out;
 824         }
 825
 826         /*
 827          * Figure out where the destination init(8) binary is.  Note that
 828          * the path could be different than what we've started with.  Use
 829          * the value from kenv, if set, or the one from sysctl otherwise.
 830          * The latter defaults to a hardcoded value, but can be overridden
 831          * by a build time option.
 832          */
 833         nbytes = kenv(KENV_GET, "init_path", init_path, sizeof(init_path));
 834         if (nbytes <= 0) {
 835                 init_path_len = sizeof(init_path);
 836                 error = sysctlbyname("kern.init_path",
 837                     init_path, &init_path_len, NULL, 0);
 838                 if (error != 0) {
 839                         emergency("failed to retrieve kern.init_path: %s",
 840                             strerror(errno));
 841                         goto out;
 842                 }
 843         }
 844
 845         /*
 846          * Repeat the init search logic from sys/kern/init_path.c
 847          */
 848         path_component = init_path;
 849         while ((path = strsep(&path_component, ":")) != NULL) {
 850                 /*
 851                  * Execute init(8) from the new rootfs.
 852                  */
 853                 execl(path, path, NULL);
 854         }
 855         emergency("cannot exec init from %s: %s", init_path, strerror(errno));
 856
 857 out:
 858         emergency("reroot failed; going to single user mode");
 859         return (state_func_t) single_user;
 860 }
 861
 862 /*
 863  * Bring the system up single user.
 864  */
 865 static state_func_t
 866 single_user(void)
 867 {
 868         pid_t pid, wpid;
 869         int status;
 870         sigset_t mask;
 871         const char *shell;
 872         char *argv[2];
 873         struct timeval tv, tn;
 874 #ifdef SECURE
 875         struct ttyent *typ;
 876         struct passwd *pp;
 877         static const char banner[] =
 878                 "Enter root password, or ^D to go multi-user\n";
 879         char *clear, *password;
 880 #endif
 881 #ifdef DEBUGSHELL
 882         char altshell[128];
 883 #endif
 884
 885         if (Reboot) {
 886                 /* Instead of going single user, let's reboot the machine */
 887                 sync();
 888                 if (reboot(howto) == -1) {
 889                         emergency("reboot(%#x) failed, %s", howto,
 890                             strerror(errno));
 891                         _exit(1); /* panic and reboot */
 892                 }
 893                 warning("reboot(%#x) returned", howto);
 894                 _exit(0); /* panic as well */
 895         }
 896
 897         shell = get_shell();
 898
 899         if ((pid = fork()) == 0) {
 900                 /*
 901                  * Start the single user session.
 902                  */
 903                 open_console();
 904
 905 #ifdef SECURE
 906                 /*
 907                  * Check the root password.
 908                  * We don't care if the console is 'on' by default;
 909                  * it's the only tty that can be 'off' and 'secure'.
 910                  */
 911                 typ = getttynam("console");
 912                 pp = getpwnam("root");
 913                 if (typ && (typ->ty_status & TTY_SECURE) == 0 &&
 914                     pp && *pp->pw_passwd) {
 915                         write_stderr(banner);
 916                         for (;;) {
 917                                 clear = getpass("Password:");
 918                                 if (clear == NULL || *clear == '\0')
 919                                         _exit(0);
 920                                 password = crypt(clear, pp->pw_passwd);
 921                                 bzero(clear, _PASSWORD_LEN);
 922                                 if (password == NULL ||
 923                                     strcmp(password, pp->pw_passwd) == 0)
 924                                         break;
 925                                 warning("single-user login failed\n");
 926                         }
 927                 }
 928                 endttyent();
 929                 endpwent();
 930 #endif /* SECURE */
 931
 932 #ifdef DEBUGSHELL
 933                 {
 934                         char *cp = altshell;
 935                         int num;
 936
 937 #define SHREQUEST "Enter full pathname of shell or RETURN for "
 938                         write_stderr(SHREQUEST);
 939                         write_stderr(shell);
 940                         write_stderr(": ");
 941                         while ((num = read(STDIN_FILENO, cp, 1)) != -1 &&
 942                             num != 0 && *cp != '\n' && cp < &altshell[127])
 943                                 cp++;
 944                         *cp = '\0';
 945                         if (altshell[0] != '\0')
 946                                 shell = altshell;
 947                 }
 948 #endif /* DEBUGSHELL */
 949
 950                 /*
 951                  * Unblock signals.
 952                  * We catch all the interesting ones,
 953                  * and those are reset to SIG_DFL on exec.
 954                  */
 955                 sigemptyset(&mask);
 956                 sigprocmask(SIG_SETMASK, &mask, (sigset_t *) 0);
 957
 958                 /*
 959                  * Fire off a shell.
 960                  * If the default one doesn't work, try the Bourne shell.
 961                  */
 962
 963                 char name[] = "-sh";
 964
 965                 argv[0] = name;
 966                 argv[1] = 0;
 967                 execv(shell, argv);
 968                 emergency("can't exec %s for single user: %m", shell);
 969                 execv(_PATH_BSHELL, argv);
 970                 emergency("can't exec %s for single user: %m", _PATH_BSHELL);
 971                 sleep(STALL_TIMEOUT);
 972                 _exit(1);
 973         }
 974
 975         if (pid == -1) {
 976                 /*
 977                  * We are seriously hosed.  Do our best.
 978                  */
 979                 emergency("can't fork single-user shell, trying again");
 980                 while (waitpid(-1, (int *) 0, WNOHANG) > 0)
 981                         continue;
 982                 return (state_func_t) single_user;
 983         }
 984
 985         requested_transition = 0;
 986         do {
 987                 if ((wpid = waitpid(-1, &status, WUNTRACED)) != -1)
 988                         collect_child(wpid);
 989                 if (wpid == -1) {
 990                         if (errno == EINTR)
 991                                 continue;
 992                         warning("wait for single-user shell failed: %m; restarting");
 993                         return (state_func_t) single_user;
 994                 }
 995                 if (wpid == pid && WIFSTOPPED(status)) {
 996                         warning("init: shell stopped, restarting\n");
 997                         kill(pid, SIGCONT);
 998                         wpid = -1;
 999                 }
1000         } while (wpid != pid && !requested_transition);
1001
1002         if (requested_transition)
1003                 return (state_func_t) requested_transition;
1004
1005         if (!WIFEXITED(status)) {
1006                 if (WTERMSIG(status) == SIGKILL) {
1007                         /*
1008                          *  reboot(8) killed shell?
1009                          */
1010                         warning("single user shell terminated.");
1011                         gettimeofday(&tv, NULL);
1012                         tn = tv;
1013                         tv.tv_sec += STALL_TIMEOUT;
1014                         while (tv.tv_sec > tn.tv_sec || (tv.tv_sec ==
1015                             tn.tv_sec && tv.tv_usec > tn.tv_usec)) {
1016                                 sleep(1);
1017                                 gettimeofday(&tn, NULL);
1018                         }
1019                         _exit(0);
1020                 } else {
1021                         warning("single user shell terminated, restarting");
1022                         return (state_func_t) single_user;
1023                 }
1024         }
1025
1026         runcom_mode = FASTBOOT;
1027         return (state_func_t) runcom;
1028 }
1029
1030 /*
1031  * Run the system startup script.
1032  */
1033 static state_func_t
1034 runcom(void)
1035 {
1036         state_func_t next_transition;
1037
1038         if ((next_transition = run_script(_PATH_RUNCOM)) != NULL)
1039                 return next_transition;
1040
1041         runcom_mode = AUTOBOOT;         /* the default */
1042         return (state_func_t) read_ttys;
1043 }
1044
1045 /*
1046  * Run a shell script.
1047  * Returns 0 on success, otherwise the next transition to enter:
1048  *  - single_user if fork/execv/waitpid failed, or if the script
1049  *    terminated with a signal or exit code != 0.
1050  *  - death_single if a SIGTERM was delivered to init(8).
1051  */
1052 static state_func_t
1053 run_script(const char *script)
1054 {
1055         pid_t pid, wpid;
1056         int status;
1057         char *argv[4];
1058         const char *shell;
1059         struct sigaction sa;
1060
1061         shell = get_shell();
1062
1063         if ((pid = fork()) == 0) {
1064                 sigemptyset(&sa.sa_mask);
1065                 sa.sa_flags = 0;
1066                 sa.sa_handler = SIG_IGN;
1067                 sigaction(SIGTSTP, &sa, (struct sigaction *)0);
1068                 sigaction(SIGHUP, &sa, (struct sigaction *)0);
1069
1070                 open_console();
1071
1072                 char _sh[]              = "sh";
1073                 char _autoboot[]        = "autoboot";
1074
1075                 argv[0] = _sh;
1076                 argv[1] = __DECONST(char *, script);
1077                 argv[2] = runcom_mode == AUTOBOOT ? _autoboot : 0;
1078                 argv[3] = 0;
1079
1080                 sigprocmask(SIG_SETMASK, &sa.sa_mask, (sigset_t *) 0);
1081
1082 #ifdef LOGIN_CAP
1083                 setprocresources(RESOURCE_RC);
1084 #endif
1085                 execv(shell, argv);
1086                 stall("can't exec %s for %s: %m", shell, script);
1087                 _exit(1);       /* force single user mode */
1088         }
1089
1090         if (pid == -1) {
1091                 emergency("can't fork for %s on %s: %m", shell, script);
1092                 while (waitpid(-1, (int *) 0, WNOHANG) > 0)
1093                         continue;
1094                 sleep(STALL_TIMEOUT);
1095                 return (state_func_t) single_user;
1096         }
1097
1098         /*
1099          * Copied from single_user().  This is a bit paranoid.
1100          */
1101         requested_transition = 0;
1102         do {
1103                 if ((wpid = waitpid(-1, &status, WUNTRACED)) != -1)
1104                         collect_child(wpid);
1105                 if (wpid == -1) {
1106                         if (requested_transition == death_single ||
1107                             requested_transition == reroot)
1108                                 return (state_func_t) requested_transition;
1109                         if (errno == EINTR)
1110                                 continue;
1111                         warning("wait for %s on %s failed: %m; going to "
1112                             "single user mode", shell, script);
1113                         return (state_func_t) single_user;
1114                 }
1115                 if (wpid == pid && WIFSTOPPED(status)) {
1116                         warning("init: %s on %s stopped, restarting\n",
1117                             shell, script);
1118                         kill(pid, SIGCONT);
1119                         wpid = -1;
1120                 }
1121         } while (wpid != pid);
1122
1123         if (WIFSIGNALED(status) && WTERMSIG(status) == SIGTERM &&
1124             requested_transition == catatonia) {
1125                 /* /etc/rc executed /sbin/reboot; wait for the end quietly */
1126                 sigset_t s;
1127
1128                 sigfillset(&s);
1129                 for (;;)
1130                         sigsuspend(&s);
1131         }
1132
1133         if (!WIFEXITED(status)) {
1134                 warning("%s on %s terminated abnormally, going to single "
1135                     "user mode", shell, script);
1136                 return (state_func_t) single_user;
1137         }
1138
1139         if (WEXITSTATUS(status))
1140                 return (state_func_t) single_user;
1141
1142         return (state_func_t) 0;
1143 }
1144
1145 /*
1146  * Open the session database.
1147  *
1148  * NB: We could pass in the size here; is it necessary?
1149  */
1150 static int
1151 start_session_db(void)
1152 {
1153         if (session_db && (*session_db->close)(session_db))
1154                 emergency("session database close: %s", strerror(errno));
1155         if ((session_db = dbopen(NULL, O_RDWR, 0, DB_HASH, NULL)) == NULL) {
1156                 emergency("session database open: %s", strerror(errno));
1157                 return (1);
1158         }
1159         return (0);
1160
1161 }
1162
1163 /*
1164  * Add a new login session.
1165  */
1166 static void
1167 add_session(session_t *sp)
1168 {
1169         DBT key;
1170         DBT data;
1171
1172         key.data = &sp->se_process;
1173         key.size = sizeof sp->se_process;
1174         data.data = &sp;
1175         data.size = sizeof sp;
1176
1177         if ((*session_db->put)(session_db, &key, &data, 0))
1178                 emergency("insert %d: %s", sp->se_process, strerror(errno));
1179 }
1180
1181 /*
1182  * Delete an old login session.
1183  */
1184 static void
1185 del_session(session_t *sp)
1186 {
1187         DBT key;
1188
1189         key.data = &sp->se_process;
1190         key.size = sizeof sp->se_process;
1191
1192         if ((*session_db->del)(session_db, &key, 0))
1193                 emergency("delete %d: %s", sp->se_process, strerror(errno));
1194 }
1195
1196 /*
1197  * Look up a login session by pid.
1198  */
1199 static session_t *
1200 find_session(pid_t pid)
1201 {
1202         DBT key;
1203         DBT data;
1204         session_t *ret;
1205
1206         key.data = &pid;
1207         key.size = sizeof pid;
1208         if ((*session_db->get)(session_db, &key, &data, 0) != 0)
1209                 return 0;
1210         bcopy(data.data, (char *)&ret, sizeof(ret));
1211         return ret;
1212 }
1213
1214 /*
1215  * Construct an argument vector from a command line.
1216  */
1217 static char **
1218 construct_argv(char *command)
1219 {
1220         int argc = 0;
1221         char **argv = (char **) malloc(((strlen(command) + 1) / 2 + 1)
1222                                                 * sizeof (char *));
1223
1224         if ((argv[argc++] = strk(command)) == NULL) {
1225                 free(argv);
1226                 return (NULL);
1227         }
1228         while ((argv[argc++] = strk((char *) 0)) != NULL)
1229                 continue;
1230         return argv;
1231 }
1232
1233 /*
1234  * Deallocate a session descriptor.
1235  */
1236 static void
1237 free_session(session_t *sp)
1238 {
1239         free(sp->se_device);
1240         if (sp->se_getty) {
1241                 free(sp->se_getty);
1242                 free(sp->se_getty_argv_space);
1243                 free(sp->se_getty_argv);
1244         }
1245         if (sp->se_window) {
1246                 free(sp->se_window);
1247                 free(sp->se_window_argv_space);
1248                 free(sp->se_window_argv);
1249         }
1250         if (sp->se_type)
1251                 free(sp->se_type);
1252         free(sp);
1253 }
1254
1255 /*
1256  * Allocate a new session descriptor.
1257  * Mark it SE_PRESENT.
1258  */
1259 static session_t *
1260 new_session(session_t *sprev, struct ttyent *typ)
1261 {
1262         session_t *sp;
1263         int fd;
1264
1265         if ((typ->ty_status & TTY_ON) == 0 ||
1266             typ->ty_name == 0 ||
1267             typ->ty_getty == 0)
1268                 return 0;
1269
1270         sp = (session_t *) calloc(1, sizeof (session_t));
1271
1272         sp->se_flags |= SE_PRESENT;
1273
1274         sp->se_device = malloc(sizeof(_PATH_DEV) + strlen(typ->ty_name));
1275         sprintf(sp->se_device, "%s%s", _PATH_DEV, typ->ty_name);
1276
1277         /*
1278          * Attempt to open the device, if we get "device not configured"
1279          * then don't add the device to the session list.
1280          */
1281         if ((fd = open(sp->se_device, O_RDONLY | O_NONBLOCK, 0)) < 0) {
1282                 if (errno == ENXIO) {
1283                         free_session(sp);
1284                         return (0);
1285                 }
1286         } else
1287                 close(fd);
1288
1289         if (setupargv(sp, typ) == 0) {
1290                 free_session(sp);
1291                 return (0);
1292         }
1293
1294         sp->se_next = 0;
1295         if (sprev == NULL) {
1296                 sessions = sp;
1297                 sp->se_prev = 0;
1298         } else {
1299                 sprev->se_next = sp;
1300                 sp->se_prev = sprev;
1301         }
1302
1303         return sp;
1304 }
1305
1306 /*
1307  * Calculate getty and if useful window argv vectors.
1308  */
1309 static int
1310 setupargv(session_t *sp, struct ttyent *typ)
1311 {
1312
1313         if (sp->se_getty) {
1314                 free(sp->se_getty);
1315                 free(sp->se_getty_argv_space);
1316                 free(sp->se_getty_argv);
1317         }
1318         sp->se_getty = malloc(strlen(typ->ty_getty) + strlen(typ->ty_name) + 2);
1319         sprintf(sp->se_getty, "%s %s", typ->ty_getty, typ->ty_name);
1320         sp->se_getty_argv_space = strdup(sp->se_getty);
1321         sp->se_getty_argv = construct_argv(sp->se_getty_argv_space);
1322         if (sp->se_getty_argv == NULL) {
1323                 warning("can't parse getty for port %s", sp->se_device);
1324                 free(sp->se_getty);
1325                 free(sp->se_getty_argv_space);
1326                 sp->se_getty = sp->se_getty_argv_space = 0;
1327                 return (0);
1328         }
1329         if (sp->se_window) {
1330                 free(sp->se_window);
1331                 free(sp->se_window_argv_space);
1332                 free(sp->se_window_argv);
1333         }
1334         sp->se_window = sp->se_window_argv_space = 0;
1335         sp->se_window_argv = 0;
1336         if (typ->ty_window) {
1337                 sp->se_window = strdup(typ->ty_window);
1338                 sp->se_window_argv_space = strdup(sp->se_window);
1339                 sp->se_window_argv = construct_argv(sp->se_window_argv_space);
1340                 if (sp->se_window_argv == NULL) {
1341                         warning("can't parse window for port %s",
1342                             sp->se_device);
1343                         free(sp->se_window_argv_space);
1344                         free(sp->se_window);
1345                         sp->se_window = sp->se_window_argv_space = 0;
1346                         return (0);
1347                 }
1348         }
1349         if (sp->se_type)
1350                 free(sp->se_type);
1351         sp->se_type = typ->ty_type ? strdup(typ->ty_type) : 0;
1352         return (1);
1353 }
1354
1355 /*
1356  * Walk the list of ttys and create sessions for each active line.
1357  */
1358 static state_func_t
1359 read_ttys(void)
1360 {
1361         session_t *sp, *snext;
1362         struct ttyent *typ;
1363
1364         /*
1365          * Destroy any previous session state.
1366          * There shouldn't be any, but just in case...
1367          */
1368         for (sp = sessions; sp; sp = snext) {
1369                 snext = sp->se_next;
1370                 free_session(sp);
1371         }
1372         sessions = 0;
1373         if (start_session_db())
1374                 return (state_func_t) single_user;
1375
1376         /*
1377          * Allocate a session entry for each active port.
1378          * Note that sp starts at 0.
1379          */
1380         while ((typ = getttyent()) != NULL)
1381                 if ((snext = new_session(sp, typ)) != NULL)
1382                         sp = snext;
1383
1384         endttyent();
1385
1386         return (state_func_t) multi_user;
1387 }
1388
1389 /*
1390  * Start a window system running.
1391  */
1392 static void
1393 start_window_system(session_t *sp)
1394 {
1395         pid_t pid;
1396         sigset_t mask;
1397         char term[64], *env[2];
1398         int status;
1399
1400         if ((pid = fork()) == -1) {
1401                 emergency("can't fork for window system on port %s: %m",
1402                     sp->se_device);
1403                 /* hope that getty fails and we can try again */
1404                 return;
1405         }
1406         if (pid) {
1407                 waitpid(-1, &status, 0);
1408                 return;
1409         }
1410
1411         /* reparent window process to the init to not make a zombie on exit */
1412         if ((pid = fork()) == -1) {
1413                 emergency("can't fork for window system on port %s: %m",
1414                     sp->se_device);
1415                 _exit(1);
1416         }
1417         if (pid)
1418                 _exit(0);
1419
1420         sigemptyset(&mask);
1421         sigprocmask(SIG_SETMASK, &mask, (sigset_t *) 0);
1422
1423         if (setsid() < 0)
1424                 emergency("setsid failed (window) %m");
1425
1426 #ifdef LOGIN_CAP
1427         setprocresources(RESOURCE_WINDOW);
1428 #endif
1429         if (sp->se_type) {
1430                 /* Don't use malloc after fork */
1431                 strcpy(term, "TERM=");
1432                 strncat(term, sp->se_type, sizeof(term) - 6);
1433                 env[0] = term;
1434                 env[1] = 0;
1435         }
1436         else
1437                 env[0] = 0;
1438         execve(sp->se_window_argv[0], sp->se_window_argv, env);
1439         stall("can't exec window system '%s' for port %s: %m",
1440                 sp->se_window_argv[0], sp->se_device);
1441         _exit(1);
1442 }
1443
1444 /*
1445  * Start a login session running.
1446  */
1447 static pid_t
1448 start_getty(session_t *sp)
1449 {
1450         pid_t pid;
1451         sigset_t mask;
1452         time_t current_time = time((time_t *) 0);
1453         int too_quick = 0;
1454         char term[64], *env[2];
1455
1456         if (current_time >= sp->se_started &&
1457             current_time - sp->se_started < GETTY_SPACING) {
1458                 if (++sp->se_nspace > GETTY_NSPACE) {
1459                         sp->se_nspace = 0;
1460                         too_quick = 1;
1461                 }
1462         } else
1463                 sp->se_nspace = 0;
1464
1465         /*
1466          * fork(), not vfork() -- we can't afford to block.
1467          */
1468         if ((pid = fork()) == -1) {
1469                 emergency("can't fork for getty on port %s: %m", sp->se_device);
1470                 return -1;
1471         }
1472
1473         if (pid)
1474                 return pid;
1475
1476         if (too_quick) {
1477                 warning("getty repeating too quickly on port %s, sleeping %d secs",
1478                     sp->se_device, GETTY_SLEEP);
1479                 sleep((unsigned) GETTY_SLEEP);
1480         }
1481
1482         if (sp->se_window) {
1483                 start_window_system(sp);
1484                 sleep(WINDOW_WAIT);
1485         }
1486
1487         sigemptyset(&mask);
1488         sigprocmask(SIG_SETMASK, &mask, (sigset_t *) 0);
1489
1490 #ifdef LOGIN_CAP
1491         setprocresources(RESOURCE_GETTY);
1492 #endif
1493         if (sp->se_type) {
1494                 /* Don't use malloc after fork */
1495                 strcpy(term, "TERM=");
1496                 strncat(term, sp->se_type, sizeof(term) - 6);
1497                 env[0] = term;
1498                 env[1] = 0;
1499         } else
1500                 env[0] = 0;
1501         execve(sp->se_getty_argv[0], sp->se_getty_argv, env);
1502         stall("can't exec getty '%s' for port %s: %m",
1503                 sp->se_getty_argv[0], sp->se_device);
1504         _exit(1);
1505 }
1506
1507 /*
1508  * Collect exit status for a child.
1509  * If an exiting login, start a new login running.
1510  */
1511 static void
1512 collect_child(pid_t pid)
1513 {
1514         session_t *sp, *sprev, *snext;
1515
1516         if (! sessions)
1517                 return;
1518
1519         if (! (sp = find_session(pid)))
1520                 return;
1521
1522         del_session(sp);
1523         sp->se_process = 0;
1524
1525         if (sp->se_flags & SE_SHUTDOWN) {
1526                 if ((sprev = sp->se_prev) != NULL)
1527                         sprev->se_next = sp->se_next;
1528                 else
1529                         sessions = sp->se_next;
1530                 if ((snext = sp->se_next) != NULL)
1531                         snext->se_prev = sp->se_prev;
1532                 free_session(sp);
1533                 return;
1534         }
1535
1536         if ((pid = start_getty(sp)) == -1) {
1537                 /* serious trouble */
1538                 requested_transition = clean_ttys;
1539                 return;
1540         }
1541
1542         sp->se_process = pid;
1543         sp->se_started = time((time_t *) 0);
1544         add_session(sp);
1545 }
1546
1547 /*
1548  * Catch a signal and request a state transition.
1549  */
1550 static void
1551 transition_handler(int sig)
1552 {
1553
1554         switch (sig) {
1555         case SIGHUP:
1556                 if (current_state == read_ttys || current_state == multi_user ||
1557                     current_state == clean_ttys || current_state == catatonia)
1558                         requested_transition = clean_ttys;
1559                 break;
1560         case SIGUSR2:
1561                 howto = RB_POWEROFF;
1562         case SIGUSR1:
1563                 howto |= RB_HALT;
1564         case SIGINT:
1565                 Reboot = TRUE;
1566         case SIGTERM:
1567                 if (current_state == read_ttys || current_state == multi_user ||
1568                     current_state == clean_ttys || current_state == catatonia)
1569                         requested_transition = death;
1570                 else
1571                         requested_transition = death_single;
1572                 break;
1573         case SIGTSTP:
1574                 if (current_state == runcom || current_state == read_ttys ||
1575                     current_state == clean_ttys ||
1576                     current_state == multi_user || current_state == catatonia)
1577                         requested_transition = catatonia;
1578                 break;
1579         case SIGEMT:
1580                 requested_transition = reroot;
1581                 break;
1582         default:
1583                 requested_transition = 0;
1584                 break;
1585         }
1586 }
1587
1588 /*
1589  * Take the system multiuser.
1590  */
1591 static state_func_t
1592 multi_user(void)
1593 {
1594         pid_t pid;
1595         session_t *sp;
1596
1597         requested_transition = 0;
1598
1599         /*
1600          * If the administrator has not set the security level to -1
1601          * to indicate that the kernel should not run multiuser in secure
1602          * mode, and the run script has not set a higher level of security
1603          * than level 1, then put the kernel into secure mode.
1604          */
1605         if (getsecuritylevel() == 0)
1606                 setsecuritylevel(1);
1607
1608         for (sp = sessions; sp; sp = sp->se_next) {
1609                 if (sp->se_process)
1610                         continue;
1611                 if ((pid = start_getty(sp)) == -1) {
1612                         /* serious trouble */
1613                         requested_transition = clean_ttys;
1614                         break;
1615                 }
1616                 sp->se_process = pid;
1617                 sp->se_started = time((time_t *) 0);
1618                 add_session(sp);
1619         }
1620
1621         while (!requested_transition)
1622                 if ((pid = waitpid(-1, (int *) 0, 0)) != -1)
1623                         collect_child(pid);
1624
1625         return (state_func_t) requested_transition;
1626 }
1627
1628 /*
1629  * This is an (n*2)+(n^2) algorithm.  We hope it isn't run often...
1630  */
1631 static state_func_t
1632 clean_ttys(void)
1633 {
1634         session_t *sp, *sprev;
1635         struct ttyent *typ;
1636         int devlen;
1637         char *old_getty, *old_window, *old_type;
1638
1639         /*
1640          * mark all sessions for death, (!SE_PRESENT)
1641          * as we find or create new ones they'll be marked as keepers,
1642          * we'll later nuke all the ones not found in /etc/ttys
1643          */
1644         for (sp = sessions; sp != NULL; sp = sp->se_next)
1645                 sp->se_flags &= ~SE_PRESENT;
1646
1647         devlen = sizeof(_PATH_DEV) - 1;
1648         while ((typ = getttyent()) != NULL) {
1649                 for (sprev = 0, sp = sessions; sp; sprev = sp, sp = sp->se_next)
1650                         if (strcmp(typ->ty_name, sp->se_device + devlen) == 0)
1651                                 break;
1652
1653                 if (sp) {
1654                         /* we want this one to live */
1655                         sp->se_flags |= SE_PRESENT;
1656                         if ((typ->ty_status & TTY_ON) == 0 ||
1657                             typ->ty_getty == 0) {
1658                                 sp->se_flags |= SE_SHUTDOWN;
1659                                 kill(sp->se_process, SIGHUP);
1660                                 continue;
1661                         }
1662                         sp->se_flags &= ~SE_SHUTDOWN;
1663                         old_getty = sp->se_getty ? strdup(sp->se_getty) : 0;
1664                         old_window = sp->se_window ? strdup(sp->se_window) : 0;
1665                         old_type = sp->se_type ? strdup(sp->se_type) : 0;
1666                         if (setupargv(sp, typ) == 0) {
1667                                 warning("can't parse getty for port %s",
1668                                         sp->se_device);
1669                                 sp->se_flags |= SE_SHUTDOWN;
1670                                 kill(sp->se_process, SIGHUP);
1671                         }
1672                         else if (   !old_getty
1673                                  || (!old_type && sp->se_type)
1674                                  || (old_type && !sp->se_type)
1675                                  || (!old_window && sp->se_window)
1676                                  || (old_window && !sp->se_window)
1677                                  || (strcmp(old_getty, sp->se_getty) != 0)
1678                                  || (old_window && strcmp(old_window, sp->se_window) != 0)
1679                                  || (old_type && strcmp(old_type, sp->se_type) != 0)
1680                                 ) {
1681                                 /* Don't set SE_SHUTDOWN here */
1682                                 sp->se_nspace = 0;
1683                                 sp->se_started = 0;
1684                                 kill(sp->se_process, SIGHUP);
1685                         }
1686                         if (old_getty)
1687                                 free(old_getty);
1688                         if (old_window)
1689                                 free(old_window);
1690                         if (old_type)
1691                                 free(old_type);
1692                         continue;
1693                 }
1694
1695                 new_session(sprev, typ);
1696         }
1697
1698         endttyent();
1699
1700         /*
1701          * sweep through and kill all deleted sessions
1702          * ones who's /etc/ttys line was deleted (SE_PRESENT unset)
1703          */
1704         for (sp = sessions; sp != NULL; sp = sp->se_next) {
1705                 if ((sp->se_flags & SE_PRESENT) == 0) {
1706                         sp->se_flags |= SE_SHUTDOWN;
1707                         kill(sp->se_process, SIGHUP);
1708                 }
1709         }
1710
1711         return (state_func_t) multi_user;
1712 }
1713
1714 /*
1715  * Block further logins.
1716  */
1717 static state_func_t
1718 catatonia(void)
1719 {
1720         session_t *sp;
1721
1722         for (sp = sessions; sp; sp = sp->se_next)
1723                 sp->se_flags |= SE_SHUTDOWN;
1724
1725         return (state_func_t) multi_user;
1726 }
1727
1728 /*
1729  * Note SIGALRM.
1730  */
1731 static void
1732 alrm_handler(int sig)
1733 {
1734
1735         (void)sig;
1736         clang = 1;
1737 }
1738
1739 /*
1740  * Bring the system down to single user.
1741  */
1742 static state_func_t
1743 death(void)
1744 {
1745         int block, blocked;
1746         size_t len;
1747
1748         /* Temporarily block suspend. */
1749         len = sizeof(blocked);
1750         block = 1;
1751         if (sysctlbyname("kern.suspend_blocked", &blocked, &len,
1752             &block, sizeof(block)) == -1)
1753                 blocked = 0;
1754
1755         /*
1756          * Also revoke the TTY here.  Because runshutdown() may reopen
1757          * the TTY whose getty we're killing here, there is no guarantee
1758          * runshutdown() will perform the initial open() call, causing
1759          * the terminal attributes to be misconfigured.
1760          */
1761         revoke_ttys();
1762
1763         /* Try to run the rc.shutdown script within a period of time */
1764         runshutdown();
1765
1766         /* Unblock suspend if we blocked it. */
1767         if (!blocked)
1768                 sysctlbyname("kern.suspend_blocked", NULL, NULL,
1769                     &blocked, sizeof(blocked));
1770
1771         return (state_func_t) death_single;
1772 }
1773
1774 /*
1775  * Do what is necessary to reinitialize single user mode or reboot
1776  * from an incomplete state.
1777  */
1778 static state_func_t
1779 death_single(void)
1780 {
1781         int i;
1782         pid_t pid;
1783         static const int death_sigs[2] = { SIGTERM, SIGKILL };
1784
1785         revoke(_PATH_CONSOLE);
1786
1787         for (i = 0; i < 2; ++i) {
1788                 if (kill(-1, death_sigs[i]) == -1 && errno == ESRCH)
1789                         return (state_func_t) single_user;
1790
1791                 clang = 0;
1792                 alarm(DEATH_WATCH);
1793                 do
1794                         if ((pid = waitpid(-1, (int *)0, 0)) != -1)
1795                                 collect_child(pid);
1796                 while (clang == 0 && errno != ECHILD);
1797
1798                 if (errno == ECHILD)
1799                         return (state_func_t) single_user;
1800         }
1801
1802         warning("some processes would not die; ps axl advised");
1803
1804         return (state_func_t) single_user;
1805 }
1806
1807 static void
1808 revoke_ttys(void)
1809 {
1810         session_t *sp;
1811
1812         for (sp = sessions; sp; sp = sp->se_next) {
1813                 sp->se_flags |= SE_SHUTDOWN;
1814                 kill(sp->se_process, SIGHUP);
1815                 revoke(sp->se_device);
1816         }
1817 }
1818
1819 /*
1820  * Run the system shutdown script.
1821  *
1822  * Exit codes:      XXX I should document more
1823  * -2       shutdown script terminated abnormally
1824  * -1       fatal error - can't run script
1825  * 0        good.
1826  * >0       some error (exit code)
1827  */
1828 static int
1829 runshutdown(void)
1830 {
1831         pid_t pid, wpid;
1832         int status;
1833         int shutdowntimeout;
1834         size_t len;
1835         char *argv[4];
1836         const char *shell;
1837         struct sigaction sa;
1838         struct stat sb;
1839
1840         /*
1841          * rc.shutdown is optional, so to prevent any unnecessary
1842          * complaints from the shell we simply don't run it if the
1843          * file does not exist. If the stat() here fails for other
1844          * reasons, we'll let the shell complain.
1845          */
1846         if (stat(_PATH_RUNDOWN, &sb) == -1 && errno == ENOENT)
1847                 return 0;
1848
1849         shell = get_shell();
1850
1851         if ((pid = fork()) == 0) {
1852                 sigemptyset(&sa.sa_mask);
1853                 sa.sa_flags = 0;
1854                 sa.sa_handler = SIG_IGN;
1855                 sigaction(SIGTSTP, &sa, (struct sigaction *)0);
1856                 sigaction(SIGHUP, &sa, (struct sigaction *)0);
1857
1858                 open_console();
1859
1860                 char _sh[]      = "sh";
1861                 char _reboot[]  = "reboot";
1862                 char _single[]  = "single";
1863                 char _path_rundown[] = _PATH_RUNDOWN;
1864
1865                 argv[0] = _sh;
1866                 argv[1] = _path_rundown;
1867                 argv[2] = Reboot ? _reboot : _single;
1868                 argv[3] = 0;
1869
1870                 sigprocmask(SIG_SETMASK, &sa.sa_mask, (sigset_t *) 0);
1871
1872 #ifdef LOGIN_CAP
1873                 setprocresources(RESOURCE_RC);
1874 #endif
1875                 execv(shell, argv);
1876                 warning("can't exec %s for %s: %m", shell, _PATH_RUNDOWN);
1877                 _exit(1);       /* force single user mode */
1878         }
1879
1880         if (pid == -1) {
1881                 emergency("can't fork for %s on %s: %m", shell, _PATH_RUNDOWN);
1882                 while (waitpid(-1, (int *) 0, WNOHANG) > 0)
1883                         continue;
1884                 sleep(STALL_TIMEOUT);
1885                 return -1;
1886         }
1887
1888         len = sizeof(shutdowntimeout);
1889         if (sysctlbyname("kern.init_shutdown_timeout", &shutdowntimeout, &len,
1890             NULL, 0) == -1 || shutdowntimeout < 2)
1891                 shutdowntimeout = DEATH_SCRIPT;
1892         alarm(shutdowntimeout);
1893         clang = 0;
1894         /*
1895          * Copied from single_user().  This is a bit paranoid.
1896          * Use the same ALRM handler.
1897          */
1898         do {
1899                 if ((wpid = waitpid(-1, &status, WUNTRACED)) != -1)
1900                         collect_child(wpid);
1901                 if (clang == 1) {
1902                         /* we were waiting for the sub-shell */
1903                         kill(wpid, SIGTERM);
1904                         warning("timeout expired for %s on %s: %m; going to "
1905                             "single user mode", shell, _PATH_RUNDOWN);
1906                         return -1;
1907                 }
1908                 if (wpid == -1) {
1909                         if (errno == EINTR)
1910                                 continue;
1911                         warning("wait for %s on %s failed: %m; going to "
1912                             "single user mode", shell, _PATH_RUNDOWN);
1913                         return -1;
1914                 }
1915                 if (wpid == pid && WIFSTOPPED(status)) {
1916                         warning("init: %s on %s stopped, restarting\n",
1917                                 shell, _PATH_RUNDOWN);
1918                         kill(pid, SIGCONT);
1919                         wpid = -1;
1920                 }
1921         } while (wpid != pid && !clang);
1922
1923         /* Turn off the alarm */
1924         alarm(0);
1925
1926         if (WIFSIGNALED(status) && WTERMSIG(status) == SIGTERM &&
1927             requested_transition == catatonia) {
1928                 /*
1929                  * /etc/rc.shutdown executed /sbin/reboot;
1930                  * wait for the end quietly
1931                  */
1932                 sigset_t s;
1933
1934                 sigfillset(&s);
1935                 for (;;)
1936                         sigsuspend(&s);
1937         }
1938
1939         if (!WIFEXITED(status)) {
1940                 warning("%s on %s terminated abnormally, going to "
1941                     "single user mode", shell, _PATH_RUNDOWN);
1942                 return -2;
1943         }
1944
1945         if ((status = WEXITSTATUS(status)) != 0)
1946                 warning("%s returned status %d", _PATH_RUNDOWN, status);
1947
1948         return status;
1949 }
1950
1951 static char *
1952 strk(char *p)
1953 {
1954         static char *t;
1955         char *q;
1956         int c;
1957
1958         if (p)
1959                 t = p;
1960         if (!t)
1961                 return 0;
1962
1963         c = *t;
1964         while (c == ' ' || c == '\t' )
1965                 c = *++t;
1966         if (!c) {
1967                 t = 0;
1968                 return 0;
1969         }
1970         q = t;
1971         if (c == '\'') {
1972                 c = *++t;
1973                 q = t;
1974                 while (c && c != '\'')
1975                         c = *++t;
1976                 if (!c)  /* unterminated string */
1977                         q = t = 0;
1978                 else
1979                         *t++ = 0;
1980         } else {
1981                 while (c && c != ' ' && c != '\t' )
1982                         c = *++t;
1983                 *t++ = 0;
1984                 if (!c)
1985                         t = 0;
1986         }
1987         return q;
1988 }
1989
1990 #ifdef LOGIN_CAP
1991 static void
1992 setprocresources(const char *cname)
1993 {
1994         login_cap_t *lc;
1995         if ((lc = login_getclassbyname(cname, NULL)) != NULL) {
1996                 setusercontext(lc, (struct passwd*)NULL, 0,
1997                     LOGIN_SETPRIORITY | LOGIN_SETRESOURCES |
1998                     LOGIN_SETLOGINCLASS | LOGIN_SETCPUMASK);
1999                 login_close(lc);
2000         }
2001 }
2002 #endif