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