xref: /freebsd/sbin/shutdown/shutdown.c (revision 1efe3c6b58ea5d0e5a0d49039d84d00f2370c7de)
18fae3551SRodney W. Grimes /*
28fae3551SRodney W. Grimes  * Copyright (c) 1988, 1990, 1993
38fae3551SRodney W. Grimes  *	The Regents of the University of California.  All rights reserved.
48fae3551SRodney W. Grimes  *
58fae3551SRodney W. Grimes  * Redistribution and use in source and binary forms, with or without
68fae3551SRodney W. Grimes  * modification, are permitted provided that the following conditions
78fae3551SRodney W. Grimes  * are met:
88fae3551SRodney W. Grimes  * 1. Redistributions of source code must retain the above copyright
98fae3551SRodney W. Grimes  *    notice, this list of conditions and the following disclaimer.
108fae3551SRodney W. Grimes  * 2. Redistributions in binary form must reproduce the above copyright
118fae3551SRodney W. Grimes  *    notice, this list of conditions and the following disclaimer in the
128fae3551SRodney W. Grimes  *    documentation and/or other materials provided with the distribution.
138fae3551SRodney W. Grimes  * 4. Neither the name of the University nor the names of its contributors
148fae3551SRodney W. Grimes  *    may be used to endorse or promote products derived from this software
158fae3551SRodney W. Grimes  *    without specific prior written permission.
168fae3551SRodney W. Grimes  *
178fae3551SRodney W. Grimes  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
188fae3551SRodney W. Grimes  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
198fae3551SRodney W. Grimes  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
208fae3551SRodney W. Grimes  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
218fae3551SRodney W. Grimes  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
228fae3551SRodney W. Grimes  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
238fae3551SRodney W. Grimes  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
248fae3551SRodney W. Grimes  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
258fae3551SRodney W. Grimes  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
268fae3551SRodney W. Grimes  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
278fae3551SRodney W. Grimes  * SUCH DAMAGE.
288fae3551SRodney W. Grimes  */
298fae3551SRodney W. Grimes 
30c69284caSDavid E. O'Brien #if 0
318fae3551SRodney W. Grimes #ifndef lint
32d3714863SPhilippe Charnier static const char copyright[] =
338fae3551SRodney W. Grimes "@(#) Copyright (c) 1988, 1990, 1993\n\
348fae3551SRodney W. Grimes 	The Regents of the University of California.  All rights reserved.\n";
358fae3551SRodney W. Grimes #endif /* not lint */
368fae3551SRodney W. Grimes 
378fae3551SRodney W. Grimes #ifndef lint
3839b831afSBruce Evans static char sccsid[] = "@(#)shutdown.c	8.4 (Berkeley) 4/28/95";
398fae3551SRodney W. Grimes #endif /* not lint */
40c69284caSDavid E. O'Brien #endif
41c69284caSDavid E. O'Brien #include <sys/cdefs.h>
42c69284caSDavid E. O'Brien __FBSDID("$FreeBSD$");
438fae3551SRodney W. Grimes 
448fae3551SRodney W. Grimes #include <sys/param.h>
458fae3551SRodney W. Grimes #include <sys/time.h>
468fae3551SRodney W. Grimes #include <sys/resource.h>
478fae3551SRodney W. Grimes #include <sys/syslog.h>
488fae3551SRodney W. Grimes 
498fae3551SRodney W. Grimes #include <ctype.h>
50d3714863SPhilippe Charnier #include <err.h>
518fae3551SRodney W. Grimes #include <fcntl.h>
528d5c19ffSDavid E. O'Brien #include <paths.h>
538fae3551SRodney W. Grimes #include <pwd.h>
548fae3551SRodney W. Grimes #include <setjmp.h>
558fae3551SRodney W. Grimes #include <signal.h>
568fae3551SRodney W. Grimes #include <stdio.h>
578fae3551SRodney W. Grimes #include <stdlib.h>
588fae3551SRodney W. Grimes #include <string.h>
598fae3551SRodney W. Grimes #include <unistd.h>
608fae3551SRodney W. Grimes 
618fae3551SRodney W. Grimes #ifdef DEBUG
628fae3551SRodney W. Grimes #undef _PATH_NOLOGIN
638fae3551SRodney W. Grimes #define	_PATH_NOLOGIN	"./nologin"
648fae3551SRodney W. Grimes #endif
658fae3551SRodney W. Grimes 
668fae3551SRodney W. Grimes #define	H		*60*60
678fae3551SRodney W. Grimes #define	M		*60
688fae3551SRodney W. Grimes #define	S		*1
698fae3551SRodney W. Grimes #define	NOLOG_TIME	5*60
70*1efe3c6bSEd Schouten static struct interval {
718fae3551SRodney W. Grimes 	int timeleft, timetowait;
728fae3551SRodney W. Grimes } tlist[] = {
73f6faa785SAlexander Langer 	{ 10 H,  5 H },
74f6faa785SAlexander Langer 	{  5 H,  3 H },
75f6faa785SAlexander Langer 	{  2 H,  1 H },
76f6faa785SAlexander Langer 	{  1 H, 30 M },
77f6faa785SAlexander Langer 	{ 30 M, 10 M },
78f6faa785SAlexander Langer 	{ 20 M, 10 M },
79f6faa785SAlexander Langer 	{ 10 M,  5 M },
80f6faa785SAlexander Langer 	{  5 M,  3 M },
81f6faa785SAlexander Langer 	{  2 M,  1 M },
82f6faa785SAlexander Langer 	{  1 M, 30 S },
83f6faa785SAlexander Langer 	{ 30 S, 30 S },
84f6faa785SAlexander Langer 	{  0  ,  0   }
858fae3551SRodney W. Grimes };
868fae3551SRodney W. Grimes #undef H
878fae3551SRodney W. Grimes #undef M
888fae3551SRodney W. Grimes #undef S
898fae3551SRodney W. Grimes 
908fae3551SRodney W. Grimes static time_t offset, shuttime;
9135cd460fSRuslan Ermilov static int dohalt, dopower, doreboot, killflg, mbuflen, oflag;
9204983c4fSDima Dorfman static char mbuf[BUFSIZ];
9304983c4fSDima Dorfman static const char *nosync, *whom;
948fae3551SRodney W. Grimes 
959534110fSXin LI static void badtime(void);
9627d434afSEd Schouten static void perform_shutdown(void);
979534110fSXin LI static void finish(int);
989534110fSXin LI static void getoffset(char *);
999534110fSXin LI static void loop(void);
1009534110fSXin LI static void nolog(void);
1019534110fSXin LI static void timeout(int);
1029534110fSXin LI static void timewarn(int);
1039534110fSXin LI static void usage(const char *);
1048fae3551SRodney W. Grimes 
105f906f843SXin LI extern const char **environ;
106f906f843SXin LI 
1078fae3551SRodney W. Grimes int
108f906f843SXin LI main(int argc, char **argv)
1098fae3551SRodney W. Grimes {
11004983c4fSDima Dorfman 	char *p, *endp;
1118fae3551SRodney W. Grimes 	struct passwd *pw;
1128fae3551SRodney W. Grimes 	int arglen, ch, len, readstdin;
1138fae3551SRodney W. Grimes 
1148fae3551SRodney W. Grimes #ifndef DEBUG
115a2bfcdfdSPhilippe Charnier 	if (geteuid())
116a2bfcdfdSPhilippe Charnier 		errx(1, "NOT super-user");
1178fae3551SRodney W. Grimes #endif
1186868734cSPawel Jakub Dawidek 
1198fae3551SRodney W. Grimes 	nosync = NULL;
1208fae3551SRodney W. Grimes 	readstdin = 0;
1216868734cSPawel Jakub Dawidek 
1226868734cSPawel Jakub Dawidek 	/*
1236868734cSPawel Jakub Dawidek 	 * Test for the special case where the utility is called as
1246868734cSPawel Jakub Dawidek 	 * "poweroff", for which it runs 'shutdown -p now'.
1256868734cSPawel Jakub Dawidek 	 */
1266868734cSPawel Jakub Dawidek 	if ((p = rindex(argv[0], '/')) == NULL)
1276868734cSPawel Jakub Dawidek 		p = argv[0];
1286868734cSPawel Jakub Dawidek 	else
1296868734cSPawel Jakub Dawidek 		++p;
1306868734cSPawel Jakub Dawidek 	if (strcmp(p, "poweroff") == 0) {
1316868734cSPawel Jakub Dawidek 		if (getopt(argc, argv, "") != -1)
1326868734cSPawel Jakub Dawidek 			usage((char *)NULL);
1336868734cSPawel Jakub Dawidek 		argc -= optind;
1346868734cSPawel Jakub Dawidek 		argv += optind;
1356868734cSPawel Jakub Dawidek 		if (argc != 0)
1366868734cSPawel Jakub Dawidek 			usage((char *)NULL);
1376868734cSPawel Jakub Dawidek 		dopower = 1;
1386868734cSPawel Jakub Dawidek 		offset = 0;
1396868734cSPawel Jakub Dawidek 		(void)time(&shuttime);
1406868734cSPawel Jakub Dawidek 		goto poweroff;
1416868734cSPawel Jakub Dawidek 	}
1426868734cSPawel Jakub Dawidek 
14335cd460fSRuslan Ermilov 	while ((ch = getopt(argc, argv, "-hknopr")) != -1)
1448fae3551SRodney W. Grimes 		switch (ch) {
1458fae3551SRodney W. Grimes 		case '-':
1468fae3551SRodney W. Grimes 			readstdin = 1;
1478fae3551SRodney W. Grimes 			break;
1488fae3551SRodney W. Grimes 		case 'h':
1498fae3551SRodney W. Grimes 			dohalt = 1;
1508fae3551SRodney W. Grimes 			break;
1518fae3551SRodney W. Grimes 		case 'k':
1528fae3551SRodney W. Grimes 			killflg = 1;
1538fae3551SRodney W. Grimes 			break;
1548fae3551SRodney W. Grimes 		case 'n':
1558fae3551SRodney W. Grimes 			nosync = "-n";
1568fae3551SRodney W. Grimes 			break;
15735cd460fSRuslan Ermilov 		case 'o':
15835cd460fSRuslan Ermilov 			oflag = 1;
15935cd460fSRuslan Ermilov 			break;
1607e2d04d7SMike Smith 		case 'p':
1617e2d04d7SMike Smith 			dopower = 1;
1627e2d04d7SMike Smith 			break;
1638fae3551SRodney W. Grimes 		case 'r':
1648fae3551SRodney W. Grimes 			doreboot = 1;
1658fae3551SRodney W. Grimes 			break;
1668fae3551SRodney W. Grimes 		case '?':
1678fae3551SRodney W. Grimes 		default:
16835cd460fSRuslan Ermilov 			usage((char *)NULL);
1698fae3551SRodney W. Grimes 		}
1708fae3551SRodney W. Grimes 	argc -= optind;
1718fae3551SRodney W. Grimes 	argv += optind;
1728fae3551SRodney W. Grimes 
1738fae3551SRodney W. Grimes 	if (argc < 1)
17435cd460fSRuslan Ermilov 		usage((char *)NULL);
1758fae3551SRodney W. Grimes 
17635cd460fSRuslan Ermilov 	if (killflg + doreboot + dohalt + dopower > 1)
17735cd460fSRuslan Ermilov 		usage("incompatible switches -h, -k, -p and -r");
1781d4f4c0dSJoseph Koshy 
17935cd460fSRuslan Ermilov 	if (oflag && !(dohalt || dopower || doreboot))
18035cd460fSRuslan Ermilov 		usage("-o requires -h, -p or -r");
18135cd460fSRuslan Ermilov 
18235cd460fSRuslan Ermilov 	if (nosync != NULL && !oflag)
18335cd460fSRuslan Ermilov 		usage("-n requires -o");
1841d4f4c0dSJoseph Koshy 
1858fae3551SRodney W. Grimes 	getoffset(*argv++);
1868fae3551SRodney W. Grimes 
1876868734cSPawel Jakub Dawidek poweroff:
1888fae3551SRodney W. Grimes 	if (*argv) {
1898fae3551SRodney W. Grimes 		for (p = mbuf, len = sizeof(mbuf); *argv; ++argv) {
1908fae3551SRodney W. Grimes 			arglen = strlen(*argv);
1918fae3551SRodney W. Grimes 			if ((len -= arglen) <= 2)
1928fae3551SRodney W. Grimes 				break;
1938fae3551SRodney W. Grimes 			if (p != mbuf)
1948fae3551SRodney W. Grimes 				*p++ = ' ';
19539b831afSBruce Evans 			memmove(p, *argv, arglen);
1968fae3551SRodney W. Grimes 			p += arglen;
1978fae3551SRodney W. Grimes 		}
1988fae3551SRodney W. Grimes 		*p = '\n';
1998fae3551SRodney W. Grimes 		*++p = '\0';
2008fae3551SRodney W. Grimes 	}
2018fae3551SRodney W. Grimes 
2028fae3551SRodney W. Grimes 	if (readstdin) {
2038fae3551SRodney W. Grimes 		p = mbuf;
2048fae3551SRodney W. Grimes 		endp = mbuf + sizeof(mbuf) - 2;
2058fae3551SRodney W. Grimes 		for (;;) {
2068fae3551SRodney W. Grimes 			if (!fgets(p, endp - p + 1, stdin))
2078fae3551SRodney W. Grimes 				break;
2088fae3551SRodney W. Grimes 			for (; *p &&  p < endp; ++p);
2098fae3551SRodney W. Grimes 			if (p == endp) {
2108fae3551SRodney W. Grimes 				*p = '\n';
2118fae3551SRodney W. Grimes 				*++p = '\0';
2128fae3551SRodney W. Grimes 				break;
2138fae3551SRodney W. Grimes 			}
2148fae3551SRodney W. Grimes 		}
2158fae3551SRodney W. Grimes 	}
2168fae3551SRodney W. Grimes 	mbuflen = strlen(mbuf);
2178fae3551SRodney W. Grimes 
2188fae3551SRodney W. Grimes 	if (offset)
2198fae3551SRodney W. Grimes 		(void)printf("Shutdown at %.24s.\n", ctime(&shuttime));
2208fae3551SRodney W. Grimes 	else
2218fae3551SRodney W. Grimes 		(void)printf("Shutdown NOW!\n");
2228fae3551SRodney W. Grimes 
2238fae3551SRodney W. Grimes 	if (!(whom = getlogin()))
2248fae3551SRodney W. Grimes 		whom = (pw = getpwuid(getuid())) ? pw->pw_name : "???";
2258fae3551SRodney W. Grimes 
2268fae3551SRodney W. Grimes #ifdef DEBUG
2278fae3551SRodney W. Grimes 	(void)putc('\n', stdout);
2288fae3551SRodney W. Grimes #else
2298fae3551SRodney W. Grimes 	(void)setpriority(PRIO_PROCESS, 0, PRIO_MIN);
2308fae3551SRodney W. Grimes 	{
2318fae3551SRodney W. Grimes 		int forkpid;
2328fae3551SRodney W. Grimes 
2338fae3551SRodney W. Grimes 		forkpid = fork();
234a2bfcdfdSPhilippe Charnier 		if (forkpid == -1)
235a2bfcdfdSPhilippe Charnier 			err(1, "fork");
236a2bfcdfdSPhilippe Charnier 		if (forkpid)
237a2bfcdfdSPhilippe Charnier 			errx(0, "[pid %d]", forkpid);
2388fae3551SRodney W. Grimes 	}
23932c9dffbSJoerg Wunsch 	setsid();
2408fae3551SRodney W. Grimes #endif
2418fae3551SRodney W. Grimes 	openlog("shutdown", LOG_CONS, LOG_AUTH);
2428fae3551SRodney W. Grimes 	loop();
243d3714863SPhilippe Charnier 	return(0);
2448fae3551SRodney W. Grimes }
2458fae3551SRodney W. Grimes 
2469534110fSXin LI static void
2479534110fSXin LI loop(void)
2488fae3551SRodney W. Grimes {
2498fae3551SRodney W. Grimes 	struct interval *tp;
2508fae3551SRodney W. Grimes 	u_int sltime;
2518fae3551SRodney W. Grimes 	int logged;
2528fae3551SRodney W. Grimes 
2538fae3551SRodney W. Grimes 	if (offset <= NOLOG_TIME) {
2548fae3551SRodney W. Grimes 		logged = 1;
2558fae3551SRodney W. Grimes 		nolog();
2568fae3551SRodney W. Grimes 	}
2578fae3551SRodney W. Grimes 	else
2588fae3551SRodney W. Grimes 		logged = 0;
2598fae3551SRodney W. Grimes 	tp = tlist;
2608fae3551SRodney W. Grimes 	if (tp->timeleft < offset)
2618fae3551SRodney W. Grimes 		(void)sleep((u_int)(offset - tp->timeleft));
2628fae3551SRodney W. Grimes 	else {
263966ea00aSRuslan Ermilov 		while (tp->timeleft && offset < tp->timeleft)
2648fae3551SRodney W. Grimes 			++tp;
2658fae3551SRodney W. Grimes 		/*
2668fae3551SRodney W. Grimes 		 * Warn now, if going to sleep more than a fifth of
2678fae3551SRodney W. Grimes 		 * the next wait time.
2688fae3551SRodney W. Grimes 		 */
269f6faa785SAlexander Langer 		if ((sltime = offset - tp->timeleft)) {
27004983c4fSDima Dorfman 			if (sltime > (u_int)(tp->timetowait / 5))
2718fae3551SRodney W. Grimes 				timewarn(offset);
2728fae3551SRodney W. Grimes 			(void)sleep(sltime);
2738fae3551SRodney W. Grimes 		}
2748fae3551SRodney W. Grimes 	}
2758fae3551SRodney W. Grimes 	for (;; ++tp) {
2768fae3551SRodney W. Grimes 		timewarn(tp->timeleft);
2778fae3551SRodney W. Grimes 		if (!logged && tp->timeleft <= NOLOG_TIME) {
2788fae3551SRodney W. Grimes 			logged = 1;
2798fae3551SRodney W. Grimes 			nolog();
2808fae3551SRodney W. Grimes 		}
2818fae3551SRodney W. Grimes 		(void)sleep((u_int)tp->timetowait);
2828fae3551SRodney W. Grimes 		if (!tp->timeleft)
2838fae3551SRodney W. Grimes 			break;
2848fae3551SRodney W. Grimes 	}
28527d434afSEd Schouten 	perform_shutdown();
2868fae3551SRodney W. Grimes }
2878fae3551SRodney W. Grimes 
2888fae3551SRodney W. Grimes static jmp_buf alarmbuf;
2898fae3551SRodney W. Grimes 
29004983c4fSDima Dorfman static const char *restricted_environ[] = {
291f6faa785SAlexander Langer 	"PATH=" _PATH_STDPATH,
292f6faa785SAlexander Langer 	NULL
293f6faa785SAlexander Langer };
294f6faa785SAlexander Langer 
2959534110fSXin LI static void
296f74b6bc2SXin LI timewarn(int timeleft)
2978fae3551SRodney W. Grimes {
2988fae3551SRodney W. Grimes 	static int first;
2998fae3551SRodney W. Grimes 	static char hostname[MAXHOSTNAMELEN + 1];
3008fae3551SRodney W. Grimes 	FILE *pf;
3018fae3551SRodney W. Grimes 	char wcmd[MAXPATHLEN + 4];
3028fae3551SRodney W. Grimes 
3038fae3551SRodney W. Grimes 	if (!first++)
3048fae3551SRodney W. Grimes 		(void)gethostname(hostname, sizeof(hostname));
3058fae3551SRodney W. Grimes 
3068fae3551SRodney W. Grimes 	/* undoc -n option to wall suppresses normal wall banner */
3078fae3551SRodney W. Grimes 	(void)snprintf(wcmd, sizeof(wcmd), "%s -n", _PATH_WALL);
308f6faa785SAlexander Langer 	environ = restricted_environ;
3098fae3551SRodney W. Grimes 	if (!(pf = popen(wcmd, "w"))) {
3108fae3551SRodney W. Grimes 		syslog(LOG_ERR, "shutdown: can't find %s: %m", _PATH_WALL);
3118fae3551SRodney W. Grimes 		return;
3128fae3551SRodney W. Grimes 	}
3138fae3551SRodney W. Grimes 
3148fae3551SRodney W. Grimes 	(void)fprintf(pf,
3158fae3551SRodney W. Grimes 	    "\007*** %sSystem shutdown message from %s@%s ***\007\n",
3168fae3551SRodney W. Grimes 	    timeleft ? "": "FINAL ", whom, hostname);
3178fae3551SRodney W. Grimes 
3188fae3551SRodney W. Grimes 	if (timeleft > 10*60)
3198fae3551SRodney W. Grimes 		(void)fprintf(pf, "System going down at %5.5s\n\n",
3208fae3551SRodney W. Grimes 		    ctime(&shuttime) + 11);
3218fae3551SRodney W. Grimes 	else if (timeleft > 59)
3228fae3551SRodney W. Grimes 		(void)fprintf(pf, "System going down in %d minute%s\n\n",
3238fae3551SRodney W. Grimes 		    timeleft / 60, (timeleft > 60) ? "s" : "");
3248fae3551SRodney W. Grimes 	else if (timeleft)
3258fae3551SRodney W. Grimes 		(void)fprintf(pf, "System going down in 30 seconds\n\n");
3268fae3551SRodney W. Grimes 	else
3278fae3551SRodney W. Grimes 		(void)fprintf(pf, "System going down IMMEDIATELY\n\n");
3288fae3551SRodney W. Grimes 
3298fae3551SRodney W. Grimes 	if (mbuflen)
3308fae3551SRodney W. Grimes 		(void)fwrite(mbuf, sizeof(*mbuf), mbuflen, pf);
3318fae3551SRodney W. Grimes 
3328fae3551SRodney W. Grimes 	/*
3338fae3551SRodney W. Grimes 	 * play some games, just in case wall doesn't come back
334d3714863SPhilippe Charnier 	 * probably unnecessary, given that wall is careful.
3358fae3551SRodney W. Grimes 	 */
3368fae3551SRodney W. Grimes 	if (!setjmp(alarmbuf)) {
3378fae3551SRodney W. Grimes 		(void)signal(SIGALRM, timeout);
3388fae3551SRodney W. Grimes 		(void)alarm((u_int)30);
3398fae3551SRodney W. Grimes 		(void)pclose(pf);
3408fae3551SRodney W. Grimes 		(void)alarm((u_int)0);
3418fae3551SRodney W. Grimes 		(void)signal(SIGALRM, SIG_DFL);
3428fae3551SRodney W. Grimes 	}
3438fae3551SRodney W. Grimes }
3448fae3551SRodney W. Grimes 
3459534110fSXin LI static void
346f74b6bc2SXin LI timeout(int signo __unused)
3478fae3551SRodney W. Grimes {
3488fae3551SRodney W. Grimes 	longjmp(alarmbuf, 1);
3498fae3551SRodney W. Grimes }
3508fae3551SRodney W. Grimes 
3519534110fSXin LI static void
35227d434afSEd Schouten perform_shutdown(void)
3538fae3551SRodney W. Grimes {
354f6faa785SAlexander Langer 	char *empty_environ[] = { NULL };
3558fae3551SRodney W. Grimes 
3568fae3551SRodney W. Grimes 	syslog(LOG_NOTICE, "%s by %s: %s",
3577e2d04d7SMike Smith 	    doreboot ? "reboot" : dohalt ? "halt" : dopower ? "power-down" :
3587e2d04d7SMike Smith 	    "shutdown", whom, mbuf);
3598fae3551SRodney W. Grimes 	(void)sleep(2);
3608fae3551SRodney W. Grimes 
3618fae3551SRodney W. Grimes 	(void)printf("\r\nSystem shutdown time has arrived\007\007\r\n");
3628fae3551SRodney W. Grimes 	if (killflg) {
3638fae3551SRodney W. Grimes 		(void)printf("\rbut you'll have to do it yourself\r\n");
3647dd4667fSAndreas Schulz 		exit(0);
3658fae3551SRodney W. Grimes 	}
3668fae3551SRodney W. Grimes #ifdef DEBUG
3678fae3551SRodney W. Grimes 	if (doreboot)
3688fae3551SRodney W. Grimes 		(void)printf("reboot");
3698fae3551SRodney W. Grimes 	else if (dohalt)
3708fae3551SRodney W. Grimes 		(void)printf("halt");
3717e2d04d7SMike Smith 	else if (dopower)
3727e2d04d7SMike Smith 		(void)printf("power-down");
37335cd460fSRuslan Ermilov 	if (nosync != NULL)
3748fae3551SRodney W. Grimes 		(void)printf(" no sync");
3758fae3551SRodney W. Grimes 	(void)printf("\nkill -HUP 1\n");
3768fae3551SRodney W. Grimes #else
37735cd460fSRuslan Ermilov 	if (!oflag) {
37835cd460fSRuslan Ermilov 		(void)kill(1, doreboot ? SIGINT :	/* reboot */
37935cd460fSRuslan Ermilov 			      dohalt ? SIGUSR1 :	/* halt */
38035cd460fSRuslan Ermilov 			      dopower ? SIGUSR2 :	/* power-down */
38135cd460fSRuslan Ermilov 			      SIGTERM);			/* single-user */
38235cd460fSRuslan Ermilov 	} else {
3838fae3551SRodney W. Grimes 		if (doreboot) {
384f6faa785SAlexander Langer 			execle(_PATH_REBOOT, "reboot", "-l", nosync,
385f6faa785SAlexander Langer 				(char *)NULL, empty_environ);
38635cd460fSRuslan Ermilov 			syslog(LOG_ERR, "shutdown: can't exec %s: %m.",
38735cd460fSRuslan Ermilov 				_PATH_REBOOT);
388f6faa785SAlexander Langer 			warn(_PATH_REBOOT);
3898fae3551SRodney W. Grimes 		}
3908fae3551SRodney W. Grimes 		else if (dohalt) {
391f6faa785SAlexander Langer 			execle(_PATH_HALT, "halt", "-l", nosync,
392f6faa785SAlexander Langer 				(char *)NULL, empty_environ);
39335cd460fSRuslan Ermilov 			syslog(LOG_ERR, "shutdown: can't exec %s: %m.",
39435cd460fSRuslan Ermilov 				_PATH_HALT);
395f6faa785SAlexander Langer 			warn(_PATH_HALT);
3968fae3551SRodney W. Grimes 		}
3977e2d04d7SMike Smith 		else if (dopower) {
3987e2d04d7SMike Smith 			execle(_PATH_HALT, "halt", "-l", "-p", nosync,
3997e2d04d7SMike Smith 				(char *)NULL, empty_environ);
40035cd460fSRuslan Ermilov 			syslog(LOG_ERR, "shutdown: can't exec %s: %m.",
40135cd460fSRuslan Ermilov 				_PATH_HALT);
4027e2d04d7SMike Smith 			warn(_PATH_HALT);
4037e2d04d7SMike Smith 		}
40435cd460fSRuslan Ermilov 		(void)kill(1, SIGTERM);		/* to single-user */
40535cd460fSRuslan Ermilov 	}
4068fae3551SRodney W. Grimes #endif
4078fae3551SRodney W. Grimes 	finish(0);
4088fae3551SRodney W. Grimes }
4098fae3551SRodney W. Grimes 
4108fae3551SRodney W. Grimes #define	ATOI2(p)	(p[0] - '0') * 10 + (p[1] - '0'); p += 2;
4118fae3551SRodney W. Grimes 
4129534110fSXin LI static void
413f74b6bc2SXin LI getoffset(char *timearg)
4148fae3551SRodney W. Grimes {
41504983c4fSDima Dorfman 	struct tm *lt;
41604983c4fSDima Dorfman 	char *p;
4178fae3551SRodney W. Grimes 	time_t now;
4182dc34227SAlexander Langer 	int this_year;
4198fae3551SRodney W. Grimes 
4201d4f4c0dSJoseph Koshy 	(void)time(&now);
4211d4f4c0dSJoseph Koshy 
4228fae3551SRodney W. Grimes 	if (!strcasecmp(timearg, "now")) {		/* now */
4238fae3551SRodney W. Grimes 		offset = 0;
4241d4f4c0dSJoseph Koshy 		shuttime = now;
4258fae3551SRodney W. Grimes 		return;
4268fae3551SRodney W. Grimes 	}
4278fae3551SRodney W. Grimes 
4288fae3551SRodney W. Grimes 	if (*timearg == '+') {				/* +minutes */
4298fae3551SRodney W. Grimes 		if (!isdigit(*++timearg))
4308fae3551SRodney W. Grimes 			badtime();
431966ea00aSRuslan Ermilov 		if ((offset = atoi(timearg) * 60) < 0)
432966ea00aSRuslan Ermilov 			badtime();
4338fae3551SRodney W. Grimes 		shuttime = now + offset;
4348fae3551SRodney W. Grimes 		return;
4358fae3551SRodney W. Grimes 	}
4368fae3551SRodney W. Grimes 
4378fae3551SRodney W. Grimes 	/* handle hh:mm by getting rid of the colon */
4388fae3551SRodney W. Grimes 	for (p = timearg; *p; ++p)
439c1160fe4SBill Fumerola 		if (!isascii(*p) || !isdigit(*p)) {
4408fae3551SRodney W. Grimes 			if (*p == ':' && strlen(p) == 3) {
4418fae3551SRodney W. Grimes 				p[0] = p[1];
4428fae3551SRodney W. Grimes 				p[1] = p[2];
4438fae3551SRodney W. Grimes 				p[2] = '\0';
4448fae3551SRodney W. Grimes 			}
4458fae3551SRodney W. Grimes 			else
4468fae3551SRodney W. Grimes 				badtime();
447c1160fe4SBill Fumerola 		}
4488fae3551SRodney W. Grimes 
4498fae3551SRodney W. Grimes 	unsetenv("TZ");					/* OUR timezone */
4508fae3551SRodney W. Grimes 	lt = localtime(&now);				/* current time val */
4518fae3551SRodney W. Grimes 
4528fae3551SRodney W. Grimes 	switch(strlen(timearg)) {
4538fae3551SRodney W. Grimes 	case 10:
4542dc34227SAlexander Langer 		this_year = lt->tm_year;
4558fae3551SRodney W. Grimes 		lt->tm_year = ATOI2(timearg);
4562dc34227SAlexander Langer 		/*
4572dc34227SAlexander Langer 		 * check if the specified year is in the next century.
4582dc34227SAlexander Langer 		 * allow for one year of user error as many people will
4592dc34227SAlexander Langer 		 * enter n - 1 at the start of year n.
4602dc34227SAlexander Langer 		 */
4612dc34227SAlexander Langer 		if (lt->tm_year < (this_year % 100) - 1)
4622dc34227SAlexander Langer 			lt->tm_year += 100;
46374179785SAlexander Langer 		/* adjust for the year 2000 and beyond */
4642dc34227SAlexander Langer 		lt->tm_year += (this_year - (this_year % 100));
4658fae3551SRodney W. Grimes 		/* FALLTHROUGH */
4668fae3551SRodney W. Grimes 	case 8:
4678fae3551SRodney W. Grimes 		lt->tm_mon = ATOI2(timearg);
4688fae3551SRodney W. Grimes 		if (--lt->tm_mon < 0 || lt->tm_mon > 11)
4698fae3551SRodney W. Grimes 			badtime();
4708fae3551SRodney W. Grimes 		/* FALLTHROUGH */
4718fae3551SRodney W. Grimes 	case 6:
4728fae3551SRodney W. Grimes 		lt->tm_mday = ATOI2(timearg);
4738fae3551SRodney W. Grimes 		if (lt->tm_mday < 1 || lt->tm_mday > 31)
4748fae3551SRodney W. Grimes 			badtime();
4758fae3551SRodney W. Grimes 		/* FALLTHROUGH */
4768fae3551SRodney W. Grimes 	case 4:
4778fae3551SRodney W. Grimes 		lt->tm_hour = ATOI2(timearg);
4788fae3551SRodney W. Grimes 		if (lt->tm_hour < 0 || lt->tm_hour > 23)
4798fae3551SRodney W. Grimes 			badtime();
4808fae3551SRodney W. Grimes 		lt->tm_min = ATOI2(timearg);
4818fae3551SRodney W. Grimes 		if (lt->tm_min < 0 || lt->tm_min > 59)
4828fae3551SRodney W. Grimes 			badtime();
4838fae3551SRodney W. Grimes 		lt->tm_sec = 0;
4848fae3551SRodney W. Grimes 		if ((shuttime = mktime(lt)) == -1)
4858fae3551SRodney W. Grimes 			badtime();
486a2bfcdfdSPhilippe Charnier 		if ((offset = shuttime - now) < 0)
487a2bfcdfdSPhilippe Charnier 			errx(1, "that time is already past.");
4888fae3551SRodney W. Grimes 		break;
4898fae3551SRodney W. Grimes 	default:
4908fae3551SRodney W. Grimes 		badtime();
4918fae3551SRodney W. Grimes 	}
4928fae3551SRodney W. Grimes }
4938fae3551SRodney W. Grimes 
4948fae3551SRodney W. Grimes #define	NOMSG	"\n\nNO LOGINS: System going down at "
4959534110fSXin LI static void
4969534110fSXin LI nolog(void)
4978fae3551SRodney W. Grimes {
4988fae3551SRodney W. Grimes 	int logfd;
4998fae3551SRodney W. Grimes 	char *ct;
5008fae3551SRodney W. Grimes 
5018fae3551SRodney W. Grimes 	(void)unlink(_PATH_NOLOGIN);	/* in case linked to another file */
5028fae3551SRodney W. Grimes 	(void)signal(SIGINT, finish);
5038fae3551SRodney W. Grimes 	(void)signal(SIGHUP, finish);
5048fae3551SRodney W. Grimes 	(void)signal(SIGQUIT, finish);
5058fae3551SRodney W. Grimes 	(void)signal(SIGTERM, finish);
5068fae3551SRodney W. Grimes 	if ((logfd = open(_PATH_NOLOGIN, O_WRONLY|O_CREAT|O_TRUNC,
5078fae3551SRodney W. Grimes 	    0664)) >= 0) {
5088fae3551SRodney W. Grimes 		(void)write(logfd, NOMSG, sizeof(NOMSG) - 1);
5098fae3551SRodney W. Grimes 		ct = ctime(&shuttime);
5108fae3551SRodney W. Grimes 		(void)write(logfd, ct + 11, 5);
5118fae3551SRodney W. Grimes 		(void)write(logfd, "\n\n", 2);
5128fae3551SRodney W. Grimes 		(void)write(logfd, mbuf, strlen(mbuf));
5138fae3551SRodney W. Grimes 		(void)close(logfd);
5148fae3551SRodney W. Grimes 	}
5158fae3551SRodney W. Grimes }
5168fae3551SRodney W. Grimes 
5179534110fSXin LI static void
518f74b6bc2SXin LI finish(int signo __unused)
5198fae3551SRodney W. Grimes {
52039b831afSBruce Evans 	if (!killflg)
5218fae3551SRodney W. Grimes 		(void)unlink(_PATH_NOLOGIN);
5228fae3551SRodney W. Grimes 	exit(0);
5238fae3551SRodney W. Grimes }
5248fae3551SRodney W. Grimes 
5259534110fSXin LI static void
52627d434afSEd Schouten badtime(void)
5278fae3551SRodney W. Grimes {
528d3714863SPhilippe Charnier 	errx(1, "bad time format");
5298fae3551SRodney W. Grimes }
5308fae3551SRodney W. Grimes 
5319534110fSXin LI static void
532f74b6bc2SXin LI usage(const char *cp)
5338fae3551SRodney W. Grimes {
53435cd460fSRuslan Ermilov 	if (cp != NULL)
53535cd460fSRuslan Ermilov 		warnx("%s", cp);
53635cd460fSRuslan Ermilov 	(void)fprintf(stderr,
5376868734cSPawel Jakub Dawidek 	    "usage: shutdown [-] [-h | -p | -r | -k] [-o [-n]] time [warning-message ...]\n"
5386868734cSPawel Jakub Dawidek 	    "       poweroff\n");
5398fae3551SRodney W. Grimes 	exit(1);
5408fae3551SRodney W. Grimes }
541