1 /* $NetBSD: shutdown.c,v 1.55 2011/08/27 18:54:39 joerg 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 40 static char sccsid[] = "@(#)shutdown.c 8.4 (Berkeley) 4/28/95"; 41 #else 42 __RCSID("$NetBSD: shutdown.c,v 1.55 2011/08/27 18:54:39 joerg 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 78 static 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 90 static time_t offset, shuttime; 91 static int dofast, dohalt, doreboot, killflg, nofork, nosync, dodump; 92 static size_t mbuflen; 93 static int dopowerdown; 94 static int dodebug, dosilent, doverbose; 95 static const char *whom; 96 static char mbuf[BUFSIZ]; 97 static char *bootstr; 98 99 static void badtime(void) __dead; 100 static void die_you_gravy_sucking_pig_dog(void) __dead; 101 static void doitfast(void); 102 static void dorcshutdown(void); 103 static void finish(int) __dead; 104 static void getoffset(char *); 105 static void loop(void) __dead; 106 static void nolog(void); 107 static void timeout(int) __dead; 108 static void timewarn(time_t); 109 static void usage(void) __dead; 110 111 int 112 main(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 253 static void 254 loop(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 295 static jmp_buf alarmbuf; 296 297 static void 298 timewarn(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 349 static void 350 /*ARGSUSED*/ 351 timeout(int signo) 352 { 353 longjmp(alarmbuf, 1); 354 } 355 356 static void 357 die_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 438 static void 439 getoffset(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 518 static void 519 dorcshutdown(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" 531 static void 532 doitfast(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 " 544 static void 545 nolog(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 566 static void 567 /*ARGSUSED*/ 568 finish(int signo) 569 { 570 571 if (!killflg) 572 (void)unlink(_PATH_NOLOGIN); 573 exit(0); 574 } 575 576 static void 577 badtime(void) 578 { 579 580 warnx("illegal time format"); 581 usage(); 582 } 583 584 static void 585 usage(void) 586 { 587 588 (void)fprintf(stderr, 589 "Usage: %s [-Ddfhknprvxz] [-b bootstr] time [message ... | -]\n", 590 getprogname()); 591 exit(1); 592 } 593