shutdown.c revision 1.54
1/* $NetBSD: shutdown.c,v 1.54 2011/02/16 19:33:48 wiz Exp $ */ 2 3/* 4 * Copyright (c) 1988, 1990, 1993 5 * The Regents of the University of California. All rights reserved. 6 * 7 * Redistribution and use in source and binary forms, with or without 8 * modification, are permitted provided that the following conditions 9 * are met: 10 * 1. Redistributions of source code must retain the above copyright 11 * notice, this list of conditions and the following disclaimer. 12 * 2. Redistributions in binary form must reproduce the above copyright 13 * notice, this list of conditions and the following disclaimer in the 14 * documentation and/or other materials provided with the distribution. 15 * 3. Neither the name of the University nor the names of its contributors 16 * may be used to endorse or promote products derived from this software 17 * without specific prior written permission. 18 * 19 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND 20 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 21 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 22 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE 23 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 24 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 25 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 26 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 27 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 28 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 29 * SUCH DAMAGE. 30 */ 31 32#include <sys/cdefs.h> 33#ifndef lint 34__COPYRIGHT("@(#) Copyright (c) 1988, 1990, 1993\ 35 The Regents of the University of California. All rights reserved."); 36#endif /* not lint */ 37 38#ifndef lint 39#if 0 40static char sccsid[] = "@(#)shutdown.c 8.4 (Berkeley) 4/28/95"; 41#else 42__RCSID("$NetBSD: shutdown.c,v 1.54 2011/02/16 19:33:48 wiz Exp $"); 43#endif 44#endif /* not lint */ 45 46#include <sys/param.h> 47#include <sys/time.h> 48#include <sys/resource.h> 49#include <sys/syslog.h> 50 51#include <ctype.h> 52#include <err.h> 53#include <fcntl.h> 54#include <pwd.h> 55#include <setjmp.h> 56#include <signal.h> 57#include <stdio.h> 58#include <stdlib.h> 59#include <string.h> 60#include <time.h> 61#include <tzfile.h> 62#include <unistd.h> 63#include <errno.h> 64 65#include "pathnames.h" 66 67#ifdef DEBUG 68#undef _PATH_NOLOGIN 69#define _PATH_NOLOGIN "./nologin" 70#undef _PATH_FASTBOOT 71#define _PATH_FASTBOOT "./fastboot" 72#endif 73 74#define H *60*60 75#define M *60 76#define S *1 77#define NOLOG_TIME 5*60 78static const struct interval { 79 time_t timeleft, timetowait; 80} tlist[] = { 81 { 10 H, 5 H }, { 5 H, 3 H }, { 2 H, 1 H }, { 1 H, 30 M }, 82 { 30 M, 10 M }, { 20 M, 10 M }, { 10 M, 5 M }, { 5 M, 3 M }, 83 { 2 M, 1 M }, { 1 M, 30 S }, { 30 S, 30 S }, 84 { 0, 0 } 85}; 86#undef H 87#undef M 88#undef S 89 90static time_t offset, shuttime; 91static int dofast, dohalt, doreboot, killflg, nofork, nosync, dodump; 92static size_t mbuflen; 93static int dopowerdown; 94static int dodebug, dosilent, doverbose; 95static const char *whom; 96static char mbuf[BUFSIZ]; 97static char *bootstr; 98 99static void badtime(void) __dead; 100static void die_you_gravy_sucking_pig_dog(void) __dead; 101static void doitfast(void); 102void dorcshutdown(void); 103static void finish(int) __dead; 104static void getoffset(char *); 105static void loop(void); 106static void nolog(void); 107static void timeout(int); 108static void timewarn(time_t); 109static void usage(void) __dead; 110 111int 112main(int argc, char *argv[]) 113{ 114 char *p, *endp; 115 struct passwd *pw; 116 size_t arglen, len; 117 int ch; 118 119 (void)setprogname(argv[0]); 120#ifndef DEBUG 121 if (geteuid()) 122 errx(1, "%s: Not super-user", strerror(EPERM)); 123#endif 124 while ((ch = getopt(argc, argv, "b:Ddfhknprvxz")) != -1) 125 switch (ch) { 126 case 'b': 127 bootstr = optarg; 128 break; 129 case 'd': 130 dodump = 1; 131 break; 132 case 'D': 133 nofork = 1; 134 break; 135 case 'f': 136 dofast = 1; 137 break; 138 case 'p': 139 dopowerdown = 1; 140 /* FALLTHROUGH */ 141 case 'h': 142 dohalt = 1; 143 break; 144 case 'k': 145 killflg = 1; 146 break; 147 case 'n': 148 nosync = 1; 149 break; 150 case 'r': 151 doreboot = 1; 152 break; 153 case 'v': 154 doverbose = 1; 155 break; 156 case 'x': 157 dodebug = 1; 158 break; 159 case 'z': 160 dosilent = 1; 161 break; 162 case '?': 163 default: 164 usage(); 165 } 166 argc -= optind; 167 argv += optind; 168 169 if (argc < 1) 170 usage(); 171 172 if (dodump && !dohalt && !doreboot) 173 doreboot = 1; 174 175 if (dofast && nosync) { 176 warnx("Incompatible options -f and -n"); 177 usage(); 178 } 179 if (dohalt && doreboot) { 180 const char *which_flag = dopowerdown ? "p" : "h"; 181 182 warnx("Incompatible options -%s and -r", which_flag); 183 usage(); 184 } 185 186 getoffset(*argv++); 187 188 if (argv[0]) { 189 if (strcmp(argv[0], "-") || argv[1]) { 190 for (p = mbuf, len = sizeof(mbuf); *argv; ++argv) { 191 arglen = strlen(*argv); 192 if ((len -= arglen) <= 2) 193 break; 194 if (p != mbuf) 195 *p++ = ' '; 196 (void)memmove(p, *argv, arglen); 197 p += arglen; 198 } 199 *p = '\n'; 200 *++p = '\0'; 201 } else { 202 p = mbuf; 203 endp = mbuf + sizeof(mbuf) - 2; 204 for (;;) { 205 if (!fgets(p, endp - p + 1, stdin)) 206 break; 207 for (; *p && p < endp; ++p); 208 if (p == endp) { 209 *p = '\n'; 210 *++p = '\0'; 211 break; 212 } 213 } 214 } 215 } 216 mbuflen = strlen(mbuf); 217 218 if (offset) 219 (void)printf("Shutdown at %.24s.\n", ctime(&shuttime)); 220 else 221 (void)printf("Shutdown NOW!\n"); 222 223 if (!(whom = getlogin())) 224 whom = (pw = getpwuid(getuid())) ? pw->pw_name : "???"; 225 226#ifdef DEBUG 227 (void)putc('\n', stdout); 228#else 229 (void)setpriority(PRIO_PROCESS, 0, PRIO_MIN); 230 if (nofork == 0) { 231 int forkpid; 232 233 forkpid = fork(); 234 if (forkpid == -1) { 235 perror("shutdown: fork"); 236 exit(1); 237 } 238 if (forkpid) { 239 (void)printf("shutdown: [pid %d]\n", forkpid); 240 exit(0); 241 } 242 (void)setsid(); 243 } 244#endif 245 openlog("shutdown", LOG_CONS, LOG_AUTH); 246 loop(); 247 /* NOTREACHED */ 248#ifdef __GNUC__ 249 return 1; 250#endif 251} 252 253void 254loop(void) 255{ 256 const struct interval *tp; 257 u_int sltime; 258 int logged; 259 260 if (offset <= NOLOG_TIME) { 261 logged = 1; 262 nolog(); 263 } 264 else 265 logged = 0; 266 tp = tlist; 267 if (tp->timeleft < offset) 268 (void)sleep((u_int)(offset - tp->timeleft)); 269 else { 270 while (offset < tp->timeleft) 271 ++tp; 272 /* 273 * Warn now, if going to sleep more than a fifth of 274 * the next wait time. 275 */ 276 if ((sltime = offset - tp->timeleft) != 0) { 277 if (sltime > tp->timetowait / 5) 278 timewarn(offset); 279 (void)sleep(sltime); 280 } 281 } 282 for (;; ++tp) { 283 timewarn(tp->timeleft); 284 if (!logged && tp->timeleft <= NOLOG_TIME) { 285 logged = 1; 286 nolog(); 287 } 288 (void)sleep((u_int)tp->timetowait); 289 if (!tp->timeleft) 290 break; 291 } 292 die_you_gravy_sucking_pig_dog(); 293} 294 295static jmp_buf alarmbuf; 296 297void 298timewarn(time_t timeleft) 299{ 300 static int first; 301 static char hostname[MAXHOSTNAMELEN + 1]; 302 FILE *pf; 303 char wcmd[MAXPATHLEN + 4]; 304 305 if (!first++) { 306 (void)gethostname(hostname, sizeof(hostname)); 307 hostname[sizeof(hostname) - 1] = '\0'; 308 } 309 310 /* undoc -n option to wall suppresses normal wall banner */ 311 (void)snprintf(wcmd, sizeof wcmd, "%s -n", _PATH_WALL); 312 if ((pf = popen(wcmd, "w")) == NULL) { 313 syslog(LOG_ERR, "%s: Can't find `%s' (%m)", getprogname(), 314 _PATH_WALL); 315 return; 316 } 317 318 (void)fprintf(pf, 319 "\007*** %sSystem shutdown message from %s@%s ***\007\n", 320 timeleft ? "": "FINAL ", whom, hostname); 321 322 if (timeleft > 10*60) 323 (void)fprintf(pf, "System going down at %5.5s\n\n", 324 ctime(&shuttime) + 11); 325 else if (timeleft > 59) 326 (void)fprintf(pf, "System going down in %ld minute%s\n\n", 327 (long)timeleft / 60, (timeleft > 60) ? "s" : ""); 328 else if (timeleft) 329 (void)fprintf(pf, "System going down in 30 seconds\n\n"); 330 else 331 (void)fprintf(pf, "System going down IMMEDIATELY\n\n"); 332 333 if (mbuflen) 334 (void)fwrite(mbuf, sizeof(*mbuf), mbuflen, pf); 335 336 /* 337 * play some games, just in case wall doesn't come back 338 * probably unnecessary, given that wall is careful. 339 */ 340 if (!setjmp(alarmbuf)) { 341 (void)signal(SIGALRM, timeout); 342 (void)alarm((u_int)30); 343 (void)pclose(pf); 344 (void)alarm((u_int)0); 345 (void)signal(SIGALRM, SIG_DFL); 346 } 347} 348 349static void 350/*ARGSUSED*/ 351timeout(int signo) 352{ 353 longjmp(alarmbuf, 1); 354} 355 356static void 357die_you_gravy_sucking_pig_dog(void) 358{ 359 const char *what; 360 361 if (doreboot) { 362 what = "reboot"; 363 } else if (dohalt && dopowerdown) { 364 what = "poweroff"; 365 } else if (dohalt) { 366 what = "halt"; 367 } else { 368 what = "shutdown"; 369 } 370 371 syslog(LOG_NOTICE, "%s by %s: %s", what, whom, mbuf); 372 (void)sleep(2); 373 374 (void)printf("\r\nSystem shutdown time has arrived\007\007\r\n"); 375 if (killflg) { 376 (void)printf("\rbut you'll have to do it yourself\r\n"); 377 finish(0); 378 } 379 if (dofast) 380 doitfast(); 381 dorcshutdown(); 382 if (doreboot || dohalt) { 383 const char *args[20]; 384 const char **arg, *path; 385#ifndef DEBUG 386 int serrno; 387#endif 388 389 arg = &args[0]; 390 if (doreboot) { 391 path = _PATH_REBOOT; 392 *arg++ = "reboot"; 393 } else { 394 path = _PATH_HALT; 395 *arg++ = "halt"; 396 } 397 if (doverbose) 398 *arg++ = "-v"; 399 if (dodebug) 400 *arg++ = "-x"; 401 if (dosilent) 402 *arg++ = "-z"; 403 if (dodump) 404 *arg++ = "-d"; 405 if (nosync) 406 *arg++ = "-n"; 407 if (dopowerdown) 408 *arg++ = "-p"; 409 *arg++ = "-l"; 410 if (bootstr) 411 *arg++ = bootstr; 412 *arg++ = 0; 413#ifndef DEBUG 414 (void)unlink(_PATH_NOLOGIN); 415 (void)execve(path, __UNCONST(args), NULL); 416 serrno = errno; 417 syslog(LOG_ERR, "Can't exec `%s' (%m)", path); 418 errno = serrno; 419 warn("Can't exec `%s'", path); 420#else 421 printf("%s", path); 422 for (arg = &args[0]; *arg; arg++) 423 printf(" %s", *arg); 424 printf("\n"); 425#endif 426 } else { 427#ifndef DEBUG 428 (void)kill(1, SIGTERM); /* to single user */ 429#else 430 printf("kill 1\n"); 431#endif 432 } 433 finish(0); 434} 435 436#define ATOI2(s) ((s) += 2, ((s)[-2] - '0') * 10 + ((s)[-1] - '0')) 437 438void 439getoffset(char *timearg) 440{ 441 struct tm *lt; 442 char *p; 443 time_t now; 444 int yearset; 445 446 (void)time(&now); 447 if (!strcasecmp(timearg, "now")) { /* now */ 448 offset = 0; 449 shuttime = now; 450 return; 451 } 452 453 if (*timearg == '+') { /* +minutes */ 454 if (!isdigit((unsigned char)*++timearg)) 455 badtime(); 456 offset = atoi(timearg) * 60; 457 shuttime = now + offset; 458 return; 459 } 460 461 /* handle hh:mm by getting rid of the colon */ 462 for (p = timearg; *p; ++p) 463 if (!isascii(*p) || !isdigit((unsigned char)*p)) { 464 if (*p == ':' && strlen(p) == 3) { 465 p[0] = p[1]; 466 p[1] = p[2]; 467 p[2] = '\0'; 468 } 469 else 470 badtime(); 471 } 472 473 (void)unsetenv("TZ"); /* OUR timezone */ 474 lt = localtime(&now); /* current time val */ 475 476 lt->tm_sec = 0; 477 478 yearset = 0; 479 switch (strlen(timearg)) { 480 case 12: 481 lt->tm_year = ATOI2(timearg) * 100 - TM_YEAR_BASE; 482 yearset = 1; 483 /* FALLTHROUGH */ 484 case 10: 485 if (yearset) { 486 lt->tm_year += ATOI2(timearg); 487 } else { 488 yearset = ATOI2(timearg); 489 if (yearset < 69) 490 lt->tm_year = yearset + 2000 - TM_YEAR_BASE; 491 else 492 lt->tm_year = yearset + 1900 - TM_YEAR_BASE; 493 } 494 /* FALLTHROUGH */ 495 case 8: 496 lt->tm_mon = ATOI2(timearg); 497 --lt->tm_mon; 498 /* FALLTHROUGH */ 499 case 6: 500 lt->tm_mday = ATOI2(timearg); 501 /* FALLTHROUGH */ 502 case 4: 503 lt->tm_hour = ATOI2(timearg); 504 /* FALLTHROUGH */ 505 case 2: 506 lt->tm_min = ATOI2(timearg); 507 break; 508 default: 509 badtime(); 510 } 511 512 if ((shuttime = mktime(lt)) == -1) 513 badtime(); 514 if ((offset = shuttime - now) < 0) 515 errx(1, "time is already past"); 516} 517 518void 519dorcshutdown(void) 520{ 521 (void)printf("\r\nAbout to run shutdown hooks...\r\n"); 522#ifndef DEBUG 523 (void)setuid(0); 524 (void)system(". " _PATH_RCSHUTDOWN); 525#endif 526 (void)sleep(5); /* Give operator a chance to abort this. */ 527 (void)printf("\r\nDone running shutdown hooks.\r\n"); 528} 529 530#define FSMSG "fastboot file for fsck\n" 531void 532doitfast(void) 533{ 534 int fastfd; 535 536 if ((fastfd = open(_PATH_FASTBOOT, O_WRONLY|O_CREAT|O_TRUNC, 537 0664)) >= 0) { 538 (void)write(fastfd, FSMSG, sizeof(FSMSG) - 1); 539 (void)close(fastfd); 540 } 541} 542 543#define NOMSG "\n\nNO LOGINS: System going down at " 544void 545nolog(void) 546{ 547 int logfd; 548 char *ct; 549 550 (void)unlink(_PATH_NOLOGIN); /* in case linked to another file */ 551 (void)signal(SIGINT, finish); 552 (void)signal(SIGHUP, finish); 553 (void)signal(SIGQUIT, finish); 554 (void)signal(SIGTERM, finish); 555 if ((logfd = open(_PATH_NOLOGIN, O_WRONLY|O_CREAT|O_TRUNC, 556 0664)) >= 0) { 557 (void)write(logfd, NOMSG, sizeof(NOMSG) - 1); 558 ct = ctime(&shuttime); 559 (void)write(logfd, ct + 11, 5); 560 (void)write(logfd, "\n\n", 2); 561 (void)write(logfd, mbuf, strlen(mbuf)); 562 (void)close(logfd); 563 } 564} 565 566static void 567/*ARGSUSED*/ 568finish(int signo) 569{ 570 571 if (!killflg) 572 (void)unlink(_PATH_NOLOGIN); 573 exit(0); 574} 575 576static void 577badtime(void) 578{ 579 580 warnx("illegal time format"); 581 usage(); 582} 583 584static void 585usage(void) 586{ 587 588 (void)fprintf(stderr, 589 "Usage: %s [-Ddfhknprvxz] [-b bootstr] time [message ... | -]\n", 590 getprogname()); 591 exit(1); 592} 593