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