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