xref: /dragonfly/sbin/init/init.c (revision 1cef5f30)
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 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 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 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 	 * IFCONSOLE option forces tty off if not the console.
1001 	 */
1002 	if (typ->ty_status & TTY_IFCONSOLE) {
1003 		asprintf(&devpath, "%s%s", _PATH_DEV, typ->ty_name);
1004 		if (stat(devpath, &st) < 0 ||
1005 		    sysctlbyname("kern.console_rdev",
1006 				 &rdev, &rdev_size,
1007 				 NULL, 0) < 0) {
1008 			/* device does not exist or no sysctl, disable */
1009 			typ->ty_status &= ~TTY_ON;
1010 		} else if (rdev != st.st_rdev) {
1011 			typ->ty_status &= ~TTY_ON;
1012 		}
1013 		free(devpath);
1014 	}
1015 }
1016 
1017 /*
1018  * Allocate a new session descriptor.
1019  * Mark it SE_PRESENT.
1020  */
1021 static session_t *
1022 new_session(session_t *sprev, int session_index, struct ttyent *typ)
1023 {
1024 	session_t *sp;
1025 	int fd;
1026 
1027 	if (typ->ty_name == NULL || typ->ty_getty == NULL)
1028 		return 0;
1029 
1030 	if ((typ->ty_status & TTY_ON) == 0)
1031 		return 0;
1032 
1033 	sp = (session_t *) calloc(1, sizeof (session_t));
1034 
1035 	asprintf(&sp->se_device, "%s%s", _PATH_DEV, typ->ty_name);
1036 	sp->se_index = session_index;
1037 	sp->se_flags |= SE_PRESENT;
1038 
1039 	/*
1040 	 * Attempt to open the device, if we get "device not configured"
1041 	 * then don't add the device to the session list.
1042 	 */
1043 	if ((fd = open(sp->se_device, O_RDONLY | O_NONBLOCK, 0)) < 0) {
1044 		if (errno == ENXIO) {
1045 			free_session(sp);
1046 			return (0);
1047 		}
1048 	} else
1049 		close(fd);
1050 
1051 	if (setupargv(sp, typ) == 0) {
1052 		free_session(sp);
1053 		return (0);
1054 	}
1055 
1056 	sp->se_next = NULL;
1057 	if (sprev == NULL) {
1058 		sessions = sp;
1059 		sp->se_prev = NULL;
1060 	} else {
1061 		sprev->se_next = sp;
1062 		sp->se_prev = sprev;
1063 	}
1064 
1065 	return sp;
1066 }
1067 
1068 /*
1069  * Calculate getty and if useful window argv vectors.
1070  */
1071 static int
1072 setupargv(session_t *sp, struct ttyent *typ)
1073 {
1074 
1075 	if (sp->se_getty) {
1076 		free(sp->se_getty);
1077 		free(sp->se_getty_argv_space);
1078 		free(sp->se_getty_argv);
1079 	}
1080 	sp->se_getty = malloc(strlen(typ->ty_getty) + strlen(typ->ty_name) + 2);
1081 	sprintf(sp->se_getty, "%s %s", typ->ty_getty, typ->ty_name);
1082 	sp->se_getty_argv_space = strdup(sp->se_getty);
1083 	sp->se_getty_argv = construct_argv(sp->se_getty_argv_space);
1084 	if (sp->se_getty_argv == NULL) {
1085 		warning("can't parse getty for port %s", sp->se_device);
1086 		free(sp->se_getty);
1087 		free(sp->se_getty_argv_space);
1088 		sp->se_getty = sp->se_getty_argv_space = NULL;
1089 		return (0);
1090 	}
1091 	if (sp->se_window) {
1092 		free(sp->se_window);
1093 		free(sp->se_window_argv_space);
1094 		free(sp->se_window_argv);
1095 	}
1096 	sp->se_window = sp->se_window_argv_space = NULL;
1097 	sp->se_window_argv = NULL;
1098 	if (typ->ty_window) {
1099 		sp->se_window = strdup(typ->ty_window);
1100 		sp->se_window_argv_space = strdup(sp->se_window);
1101 		sp->se_window_argv = construct_argv(sp->se_window_argv_space);
1102 		if (sp->se_window_argv == NULL) {
1103 			warning("can't parse window for port %s",
1104 				sp->se_device);
1105 			free(sp->se_window_argv_space);
1106 			free(sp->se_window);
1107 			sp->se_window = sp->se_window_argv_space = NULL;
1108 			return (0);
1109 		}
1110 	}
1111 	if (sp->se_type)
1112 		free(sp->se_type);
1113 	sp->se_type = typ->ty_type ? strdup(typ->ty_type) : 0;
1114 	return (1);
1115 }
1116 
1117 /*
1118  * Walk the list of ttys and create sessions for each active line.
1119  */
1120 static state_t
1121 f_read_ttys(void)
1122 {
1123 	int session_index = 0;
1124 	session_t *sp, *snext;
1125 	struct ttyent *typ;
1126 
1127 	if (sessions == NULL) {
1128 		struct stat st;
1129 
1130 		make_utmpx("", BOOT_MSG, BOOT_TIME, 0, &boot_time, 0);
1131 
1132 		/*
1133 		 * If wtmpx is not empty, pick the down time from there
1134 		 */
1135 		if (stat(_PATH_WTMPX, &st) != -1 && st.st_size != 0) {
1136 			struct timeval down_time;
1137 
1138 			TIMESPEC_TO_TIMEVAL(&down_time,
1139 			    st.st_atime > st.st_mtime ?
1140 			    &st.st_atimespec : &st.st_mtimespec);
1141 			make_utmpx("", DOWN_MSG, DOWN_TIME, 0, &down_time, 0);
1142 		}
1143 	}
1144 	/*
1145 	 * Destroy any previous session state.
1146 	 * There shouldn't be any, but just in case...
1147 	 */
1148 	for (sp = sessions; sp; sp = snext) {
1149 		if (sp->se_process)
1150 			clear_session_logs(sp, 0);
1151 		snext = sp->se_next;
1152 		free_session(sp);
1153 	}
1154 	sessions = NULL;
1155 	if (start_session_db())
1156 		return single_user;
1157 
1158 	/*
1159 	 * Allocate a session entry for each active port.
1160 	 * Note that sp starts at 0.
1161 	 */
1162 	while ((typ = getttyent()) != NULL) {
1163 		adjttyent(typ);
1164 		if ((snext = new_session(sp, ++session_index, typ)) != NULL)
1165 			sp = snext;
1166 	}
1167 
1168 	endttyent();
1169 
1170 	return multi_user;
1171 }
1172 
1173 /*
1174  * Start a window system running.
1175  */
1176 static void
1177 start_window_system(session_t *sp)
1178 {
1179 	pid_t pid;
1180 	sigset_t mask;
1181 	char term[64], *env[2];
1182 
1183 	if ((pid = fork()) == -1) {
1184 		emergency("can't fork for window system on port %s: %m",
1185 			sp->se_device);
1186 		/* hope that getty fails and we can try again */
1187 		return;
1188 	}
1189 
1190 	if (pid)
1191 		return;
1192 
1193 	sigemptyset(&mask);
1194 	sigprocmask(SIG_SETMASK, &mask, NULL);
1195 
1196 	if (setsid() < 0)
1197 		emergency("setsid failed (window) %m");
1198 
1199 #ifdef LOGIN_CAP
1200 	setprocresources(RESOURCE_WINDOW);
1201 #endif
1202 	if (sp->se_type) {
1203 		/* Don't use malloc after fork */
1204 		strcpy(term, "TERM=");
1205 		strncat(term, sp->se_type, sizeof(term) - 6);
1206 		env[0] = term;
1207 		env[1] = NULL;
1208 	}
1209 	else
1210 		env[0] = NULL;
1211 	execve(sp->se_window_argv[0], sp->se_window_argv, env);
1212 	stall("can't exec window system '%s' for port %s: %m",
1213 		sp->se_window_argv[0], sp->se_device);
1214 	_exit(1);
1215 }
1216 
1217 /*
1218  * Start a login session running.
1219  */
1220 static pid_t
1221 start_getty(session_t *sp)
1222 {
1223 	pid_t pid;
1224 	sigset_t mask;
1225 	time_t current_time = time(NULL);
1226 	int too_quick = 0;
1227 	char term[64], *env[2];
1228 
1229 	if (current_time >= sp->se_started.tv_sec &&
1230 	    current_time - sp->se_started.tv_sec < GETTY_SPACING) {
1231 		if (++sp->se_nspace > GETTY_NSPACE) {
1232 			sp->se_nspace = 0;
1233 			too_quick = 1;
1234 		}
1235 	} else
1236 		sp->se_nspace = 0;
1237 
1238 	/*
1239 	 * fork(), not vfork() -- we can't afford to block.
1240 	 */
1241 	if ((pid = fork()) == -1) {
1242 		emergency("can't fork for getty on port %s: %m", sp->se_device);
1243 		return -1;
1244 	}
1245 
1246 	if (pid)
1247 		return pid;
1248 
1249 	if (too_quick) {
1250 		warning("getty repeating too quickly on port %s, sleeping %d secs",
1251 			sp->se_device, GETTY_SLEEP);
1252 		sleep((unsigned) GETTY_SLEEP);
1253 	}
1254 
1255 	if (sp->se_window) {
1256 		start_window_system(sp);
1257 		sleep(WINDOW_WAIT);
1258 	}
1259 
1260 	sigemptyset(&mask);
1261 	sigprocmask(SIG_SETMASK, &mask, NULL);
1262 
1263 #ifdef LOGIN_CAP
1264 	setprocresources(RESOURCE_GETTY);
1265 #endif
1266 	if (sp->se_type) {
1267 		/* Don't use malloc after fork */
1268 		strcpy(term, "TERM=");
1269 		strncat(term, sp->se_type, sizeof(term) - 6);
1270 		env[0] = term;
1271 		env[1] = NULL;
1272 	}
1273 	else
1274 		env[0] = NULL;
1275 	execve(sp->se_getty_argv[0], sp->se_getty_argv, env);
1276 	stall("can't exec getty '%s' for port %s: %m",
1277 		sp->se_getty_argv[0], sp->se_device);
1278 	_exit(1);
1279 }
1280 
1281 /*
1282  * Collect exit status for a child.
1283  * If an exiting login, start a new login running.
1284  */
1285 static void
1286 collect_child(pid_t pid, int status)
1287 {
1288 	session_t *sp, *sprev, *snext;
1289 
1290 	if (! sessions)
1291 		return;
1292 
1293 	if (! (sp = find_session(pid)))
1294 		return;
1295 
1296 	clear_session_logs(sp, status);
1297 	del_session(sp);
1298 	sp->se_process = 0;
1299 
1300 	if (sp->se_flags & SE_SHUTDOWN) {
1301 		if ((sprev = sp->se_prev) != NULL)
1302 			sprev->se_next = sp->se_next;
1303 		else
1304 			sessions = sp->se_next;
1305 		if ((snext = sp->se_next) != NULL)
1306 			snext->se_prev = sp->se_prev;
1307 		free_session(sp);
1308 		return;
1309 	}
1310 
1311 	if ((pid = start_getty(sp)) == -1) {
1312 		/* serious trouble */
1313 		requested_transition = clean_ttys;
1314 		return;
1315 	}
1316 
1317 	sp->se_process = pid;
1318 	gettimeofday(&sp->se_started, NULL);
1319 	add_session(sp);
1320 }
1321 
1322 /*
1323  * Catch a signal and request a state transition.
1324  */
1325 static void
1326 transition_handler(int sig)
1327 {
1328 
1329 	switch (sig) {
1330 	case SIGHUP:
1331 		requested_transition = clean_ttys;
1332 		break;
1333 	case SIGUSR2:
1334 		howto = RB_POWEROFF;
1335 		/* FALLTHROUGH */
1336 	case SIGUSR1:
1337 		howto |= RB_HALT;
1338 		/* FALLTHROUGH */
1339 	case SIGINT:
1340 		Reboot = TRUE;
1341 		/* FALLTHROUGH */
1342 	case SIGTERM:
1343 		requested_transition = death;
1344 		break;
1345 	case SIGTSTP:
1346 		requested_transition = catatonia;
1347 		break;
1348 	default:
1349 		requested_transition = 0;
1350 		break;
1351 	}
1352 }
1353 
1354 /*
1355  * Take the system multiuser.
1356  */
1357 static state_t
1358 f_multi_user(void)
1359 {
1360 	pid_t pid;
1361 	int status;
1362 	session_t *sp;
1363 
1364 	requested_transition = 0;
1365 
1366 	/*
1367 	 * If the administrator has not set the security level to -1
1368 	 * to indicate that the kernel should not run multiuser in secure
1369 	 * mode, and the run script has not set a higher level of security
1370 	 * than level 1, then put the kernel into secure mode.
1371 	 */
1372 	if (getsecuritylevel() == 0)
1373 		setsecuritylevel(1);
1374 
1375 	for (sp = sessions; sp; sp = sp->se_next) {
1376 		if (sp->se_process)
1377 			continue;
1378 		if ((pid = start_getty(sp)) == -1) {
1379 			/* serious trouble */
1380 			requested_transition = clean_ttys;
1381 			break;
1382 		}
1383 		sp->se_process = pid;
1384 		gettimeofday(&sp->se_started, NULL);
1385 		add_session(sp);
1386 	}
1387 
1388 	while (!requested_transition)
1389 		if ((pid = waitpid(-1, &status, 0)) != -1)
1390 			collect_child(pid, status);
1391 
1392 	return requested_transition;
1393 }
1394 
1395 /*
1396  * This is an (n*2)+(n^2) algorithm.  We hope it isn't run often...
1397  */
1398 static state_t
1399 f_clean_ttys(void)
1400 {
1401 	session_t *sp, *sprev;
1402 	struct ttyent *typ;
1403 	int session_index = 0;
1404 	int devlen;
1405 	char *old_getty, *old_window, *old_type;
1406 
1407 	if (! sessions)
1408 		return multi_user;
1409 
1410 	/*
1411 	 * mark all sessions for death, (!SE_PRESENT)
1412 	 * as we find or create new ones they'll be marked as keepers,
1413 	 * we'll later nuke all the ones not found in /etc/ttys
1414 	 */
1415 	for (sp = sessions; sp != NULL; sp = sp->se_next)
1416 		sp->se_flags &= ~SE_PRESENT;
1417 
1418 	devlen = sizeof(_PATH_DEV) - 1;
1419 	while ((typ = getttyent()) != NULL) {
1420 		++session_index;
1421 
1422 		adjttyent(typ);
1423 		for (sprev = NULL, sp = sessions; sp; sprev = sp, sp = sp->se_next)
1424 			if (strcmp(typ->ty_name, sp->se_device + devlen) == 0)
1425 				break;
1426 
1427 		if (sp) {
1428 			/* we want this one to live */
1429 			sp->se_flags |= SE_PRESENT;
1430 			if (sp->se_index != session_index) {
1431 				warning("port %s changed utmpx index from %d to %d",
1432 				       sp->se_device, sp->se_index,
1433 				       session_index);
1434 				sp->se_index = session_index;
1435 			}
1436 			if ((typ->ty_status & TTY_ON) == 0 ||
1437 			    typ->ty_getty == 0) {
1438 				sp->se_flags |= SE_SHUTDOWN;
1439 				kill(sp->se_process, SIGHUP);
1440 				continue;
1441 			}
1442 			sp->se_flags &= ~SE_SHUTDOWN;
1443 			old_getty = sp->se_getty ? strdup(sp->se_getty) : 0;
1444 			old_window = sp->se_window ? strdup(sp->se_window) : 0;
1445 			old_type = sp->se_type ? strdup(sp->se_type) : 0;
1446 			if (setupargv(sp, typ) == 0) {
1447 				warning("can't parse getty for port %s",
1448 					sp->se_device);
1449 				sp->se_flags |= SE_SHUTDOWN;
1450 				kill(sp->se_process, SIGHUP);
1451 			}
1452 			else if (   !old_getty
1453 				 || (!old_type && sp->se_type)
1454 				 || (old_type && !sp->se_type)
1455 				 || (!old_window && sp->se_window)
1456 				 || (old_window && !sp->se_window)
1457 				 || (strcmp(old_getty, sp->se_getty) != 0)
1458 				 || (old_window && strcmp(old_window, sp->se_window) != 0)
1459 				 || (old_type && strcmp(old_type, sp->se_type) != 0)
1460 				) {
1461 				/* Don't set SE_SHUTDOWN here */
1462 				sp->se_nspace = 0;
1463 				sp->se_started.tv_sec = sp->se_started.tv_usec = 0;
1464 				kill(sp->se_process, SIGHUP);
1465 			}
1466 			if (old_getty)
1467 				free(old_getty);
1468 			if (old_window)
1469 				free(old_window);
1470 			if (old_type)
1471 				free(old_type);
1472 			continue;
1473 		}
1474 
1475 		new_session(sprev, session_index, typ);
1476 	}
1477 
1478 	endttyent();
1479 
1480 	/*
1481 	 * sweep through and kill all deleted sessions
1482 	 * ones who's /etc/ttys line was deleted (SE_PRESENT unset)
1483 	 */
1484 	for (sp = sessions; sp != NULL; sp = sp->se_next) {
1485 		if ((sp->se_flags & SE_PRESENT) == 0) {
1486 			sp->se_flags |= SE_SHUTDOWN;
1487 			kill(sp->se_process, SIGHUP);
1488 		}
1489 	}
1490 
1491 	return multi_user;
1492 }
1493 
1494 /*
1495  * Block further logins.
1496  */
1497 static state_t
1498 f_catatonia(void)
1499 {
1500 	session_t *sp;
1501 
1502 	for (sp = sessions; sp; sp = sp->se_next)
1503 		sp->se_flags |= SE_SHUTDOWN;
1504 
1505 	return multi_user;
1506 }
1507 
1508 /*
1509  * Note SIGALRM.
1510  */
1511 static void
1512 alrm_handler(int sig __unused)
1513 {
1514 	clang = 1;
1515 }
1516 
1517 /*
1518  * Bring the system down to single user.
1519  */
1520 static state_t
1521 f_death(void)
1522 {
1523 	session_t *sp;
1524 	int i;
1525 	pid_t pid;
1526 	int status;
1527 	static const int death_sigs[2] = { SIGTERM, SIGKILL };
1528 
1529 	logwtmpx("~", "shutdown", "", 0, INIT_PROCESS);
1530 
1531 	for (sp = sessions; sp; sp = sp->se_next) {
1532 		sp->se_flags |= SE_SHUTDOWN;
1533 		kill(sp->se_process, SIGHUP);
1534 	}
1535 
1536 	/* Try to run the rc.shutdown script within a period of time */
1537 	runshutdown();
1538 
1539 	for (i = 0; i < 2; ++i) {
1540 		if (kill(-1, death_sigs[i]) == -1 && errno == ESRCH)
1541 			return single_user;
1542 
1543 		clang = 0;
1544 		alarm(DEATH_WATCH);
1545 		do
1546 			if ((pid = waitpid(-1, &status, 0)) != -1)
1547 				collect_child(pid, status);
1548 		while (clang == 0 && errno != ECHILD);
1549 
1550 		if (errno == ECHILD)
1551 			return single_user;
1552 	}
1553 
1554 	warning("some processes would not die; ps axl advised");
1555 
1556 	return single_user;
1557 }
1558 
1559 /*
1560  * Run the system shutdown script.
1561  *
1562  * Exit codes:      XXX I should document more
1563  * -2       shutdown script terminated abnormally
1564  * -1       fatal error - can't run script
1565  * 0        good.
1566  * >0       some error (exit code)
1567  */
1568 static int
1569 runshutdown(void)
1570 {
1571 	pid_t pid, wpid;
1572 	int status;
1573 	int shutdowntimeout;
1574 	size_t len;
1575 	const char *argv[4];
1576 	struct sigaction sa;
1577 	struct stat sb;
1578 
1579 	/*
1580 	 * rc.shutdown is optional, so to prevent any unnecessary
1581 	 * complaints from the shell we simply don't run it if the
1582 	 * file does not exist. If the stat() here fails for other
1583 	 * reasons, we'll let the shell complain.
1584 	 */
1585 	if (stat(_PATH_RUNDOWN, &sb) == -1 && errno == ENOENT)
1586 		return 0;
1587 
1588 	if ((pid = fork()) == 0) {
1589 		int	fd;
1590 
1591 		/* Assume that init already grab console as ctty before */
1592 
1593 		sigemptyset(&sa.sa_mask);
1594 		sa.sa_flags = 0;
1595 		sa.sa_handler = SIG_IGN;
1596 		sigaction(SIGTSTP, &sa, NULL);
1597 		sigaction(SIGHUP, &sa, NULL);
1598 
1599 		if ((fd = open(_PATH_CONSOLE, O_RDWR)) == -1)
1600 		    warning("can't open %s: %m", _PATH_CONSOLE);
1601 		else {
1602 		    dup2(fd, 0);
1603 		    dup2(fd, 1);
1604 		    dup2(fd, 2);
1605 		    if (fd > 2)
1606 			close(fd);
1607 		}
1608 
1609 		/*
1610 		 * Run the shutdown script.
1611 		 */
1612 		argv[0] = "sh";
1613 		argv[1] = _PATH_RUNDOWN;
1614 		if (Reboot)
1615 			argv[2] = "reboot";
1616 		else
1617 			argv[2] = "single";
1618 		argv[3] = NULL;
1619 
1620 		sigprocmask(SIG_SETMASK, &sa.sa_mask, NULL);
1621 
1622 #ifdef LOGIN_CAP
1623 		setprocresources(RESOURCE_RC);
1624 #endif
1625 		execv(_PATH_BSHELL, __DECONST(char **, argv));
1626 		warning("can't exec %s for %s: %m", _PATH_BSHELL, _PATH_RUNDOWN);
1627 		_exit(1);	/* force single user mode */
1628 	}
1629 
1630 	if (pid == -1) {
1631 		emergency("can't fork for %s on %s: %m",
1632 			_PATH_BSHELL, _PATH_RUNDOWN);
1633 		while (waitpid(-1, NULL, WNOHANG) > 0)
1634 			continue;
1635 		sleep(STALL_TIMEOUT);
1636 		return -1;
1637 	}
1638 
1639 	len = sizeof(shutdowntimeout);
1640 	if (sysctlbyname("kern.init_shutdown_timeout",
1641 			 &shutdowntimeout,
1642 			 &len, NULL, 0) == -1 || shutdowntimeout < 2)
1643 	    shutdowntimeout = DEATH_SCRIPT;
1644 	alarm(shutdowntimeout);
1645 	clang = 0;
1646 	/*
1647 	 * Copied from single_user().  This is a bit paranoid.
1648 	 * Use the same ALRM handler.
1649 	 */
1650 	do {
1651 		if ((wpid = waitpid(-1, &status, WUNTRACED)) != -1)
1652 			collect_child(wpid, status);
1653 		if (clang == 1) {
1654 			/* we were waiting for the sub-shell */
1655 			kill(wpid, SIGTERM);
1656 			warning("timeout expired for %s on %s: %m; going to single user mode",
1657 				_PATH_BSHELL, _PATH_RUNDOWN);
1658 			return -1;
1659 		}
1660 		if (wpid == -1) {
1661 			if (errno == EINTR)
1662 				continue;
1663 			warning("wait for %s on %s failed: %m; going to single user mode",
1664 				_PATH_BSHELL, _PATH_RUNDOWN);
1665 			return -1;
1666 		}
1667 		if (wpid == pid && WIFSTOPPED(status)) {
1668 			warning("init: %s on %s stopped, restarting\n",
1669 				_PATH_BSHELL, _PATH_RUNDOWN);
1670 			kill(pid, SIGCONT);
1671 			wpid = -1;
1672 		}
1673 	} while (wpid != pid && !clang);
1674 
1675 	/* Turn off the alarm */
1676 	alarm(0);
1677 
1678 	if (WIFSIGNALED(status) && WTERMSIG(status) == SIGTERM &&
1679 	    requested_transition == catatonia) {
1680 		/*
1681 		 * /etc/rc.shutdown executed /sbin/reboot;
1682 		 * wait for the end quietly
1683 		 */
1684 		sigset_t s;
1685 
1686 		sigfillset(&s);
1687 		for (;;)
1688 			sigsuspend(&s);
1689 	}
1690 
1691 	if (!WIFEXITED(status)) {
1692 		warning("%s on %s terminated abnormally, going to single user mode",
1693 			_PATH_BSHELL, _PATH_RUNDOWN);
1694 		return -2;
1695 	}
1696 
1697 	if ((status = WEXITSTATUS(status)) != 0)
1698 		warning("%s returned status %d", _PATH_RUNDOWN, status);
1699 
1700 	return status;
1701 }
1702 
1703 static char *
1704 strk(char *p)
1705 {
1706     static char *t;
1707     char *q;
1708     int c;
1709 
1710     if (p)
1711 	t = p;
1712     if (!t)
1713 	return 0;
1714 
1715     c = *t;
1716     while (c == ' ' || c == '\t' )
1717 	c = *++t;
1718     if (!c) {
1719 	t = NULL;
1720 	return 0;
1721     }
1722     q = t;
1723     if (c == '\'') {
1724 	c = *++t;
1725 	q = t;
1726 	while (c && c != '\'')
1727 	    c = *++t;
1728 	if (!c)  /* unterminated string */
1729 	    q = t = NULL;
1730 	else
1731 	    *t++ = 0;
1732     } else {
1733 	while (c && c != ' ' && c != '\t' )
1734 	    c = *++t;
1735 	*t++ = 0;
1736 	if (!c)
1737 	    t = NULL;
1738     }
1739     return q;
1740 }
1741 
1742 #ifdef LOGIN_CAP
1743 static void
1744 setprocresources(const char *cname)
1745 {
1746 	login_cap_t *lc;
1747 	if ((lc = login_getclassbyname(cname, NULL)) != NULL) {
1748 		setusercontext(lc, NULL, 0, LOGIN_SETPRIORITY|LOGIN_SETRESOURCES);
1749 		login_close(lc);
1750 	}
1751 }
1752 #endif
1753 
1754 static void
1755 session_utmpx(const session_t *sp, int add)
1756 {
1757 	const char *name = sp->se_getty ? sp->se_getty :
1758 	    (sp->se_window ? sp->se_window : "");
1759 	const char *line = sp->se_device + sizeof(_PATH_DEV) - 1;
1760 
1761 	make_utmpx(name, line, add ? LOGIN_PROCESS : DEAD_PROCESS,
1762 	    sp->se_process, &sp->se_started, sp->se_index);
1763 }
1764 
1765 static void
1766 make_utmpx(const char *name, const char *line, int type, pid_t pid,
1767     const struct timeval *tv, int session)
1768 {
1769 	struct utmpx ut;
1770 	const char *eline;
1771 
1772 	(void)memset(&ut, 0, sizeof(ut));
1773 	(void)strlcpy(ut.ut_name, name, sizeof(ut.ut_name));
1774 	ut.ut_type = type;
1775 	(void)strlcpy(ut.ut_line, line, sizeof(ut.ut_line));
1776 	ut.ut_pid = pid;
1777 	if (tv)
1778 		ut.ut_tv = *tv;
1779 	else
1780 		(void)gettimeofday(&ut.ut_tv, NULL);
1781 	ut.ut_session = session;
1782 
1783 	eline = line + strlen(line);
1784 	if ((size_t)(eline - line) >= sizeof(ut.ut_id))
1785 		line = eline - sizeof(ut.ut_id);
1786 	(void)strncpy(ut.ut_id, line, sizeof(ut.ut_id));
1787 
1788 	if (pututxline(&ut) == NULL)
1789 		warning("can't add utmpx record for `%s': %m", ut.ut_line);
1790 	endutxent();
1791 }
1792 
1793 static char
1794 get_runlevel(const state_t s)
1795 {
1796 	if (s == single_user)
1797 		return SINGLE_USER;
1798 	if (s == runcom)
1799 		return RUNCOM;
1800 	if (s == read_ttys)
1801 		return READ_TTYS;
1802 	if (s == multi_user)
1803 		return MULTI_USER;
1804 	if (s == clean_ttys)
1805 		return CLEAN_TTYS;
1806 	if (s == catatonia)
1807 		return CATATONIA;
1808 	return DEATH;
1809 }
1810 
1811 static void
1812 utmpx_set_runlevel(char old, char new)
1813 {
1814 	struct utmpx ut;
1815 
1816 	/*
1817 	 * Don't record any transitions until we did the first transition
1818 	 * to read ttys, which is when we are guaranteed to have a read-write
1819 	 * /var. Perhaps use a different variable for this?
1820 	 */
1821 	if (sessions == NULL)
1822 		return;
1823 
1824 	(void)memset(&ut, 0, sizeof(ut));
1825 	(void)snprintf(ut.ut_line, sizeof(ut.ut_line), RUNLVL_MSG, new);
1826 	ut.ut_type = RUN_LVL;
1827 	(void)gettimeofday(&ut.ut_tv, NULL);
1828 	ut.ut_exit.e_exit = old;
1829 	ut.ut_exit.e_termination = new;
1830 	if (pututxline(&ut) == NULL)
1831 		warning("can't add utmpx record for `runlevel': %m");
1832 	endutxent();
1833 }
1834