use logwtmp()
[unix-history] / usr / src / sbin / reboot / halt.c
index 501d939..f0e7318 100644 (file)
-static char *sccsid = "@(#)halt.c      4.2 (Berkeley) %G%";
+/*
+ * Copyright (c) 1980, 1986 The Regents of the University of California.
+ * All rights reserved.
+ *
+ * Redistribution and use in source and binary forms are permitted
+ * provided that the above copyright notice and this paragraph are
+ * duplicated in all such forms and that any documentation,
+ * advertising materials, and other materials related to such
+ * distribution and use acknowledge that the software was developed
+ * by the University of California, Berkeley.  The name of the
+ * University may not be used to endorse or promote products derived
+ * from this software without specific prior written permission.
+ * THIS SOFTWARE IS PROVIDED ``AS IS'' AND WITHOUT ANY EXPRESS OR
+ * IMPLIED WARRANTIES, INCLUDING, WITHOUT LIMITATION, THE IMPLIED
+ * WARRANTIES OF MERCHANTIBILITY AND FITNESS FOR A PARTICULAR PURPOSE.
+ */
+
+#ifndef lint
+char copyright[] =
+"@(#) Copyright (c) 1980, 1986 The Regents of the University of California.\n\
+ All rights reserved.\n";
+#endif /* not lint */
+
+#ifndef lint
+static char sccsid[] = "@(#)halt.c     5.5 (Berkeley) %G%";
+#endif /* not lint */
+
 /*
  * Halt
  */
 #include <stdio.h>
 #include <sys/reboot.h>
 /*
  * Halt
  */
 #include <stdio.h>
 #include <sys/reboot.h>
+#include <sys/types.h>
+#include <sys/time.h>
+#include <sys/syslog.h>
+#include <errno.h>
+#include <signal.h>
+#include <pwd.h>
 
 main(argc, argv)
        int argc;
        char **argv;
 {
 
 main(argc, argv)
        int argc;
        char **argv;
 {
-       int howto;
-       char *ttyn = (char *)ttyname(2);
+       register int i;
+       register int qflag = 0;
+       struct passwd *pw, *getpwuid();
+       int ch, howto, needlog = 1;
+       char *user, *ttyn, *getlogin(), *ttyname();
 
        howto = RB_HALT;
 
        howto = RB_HALT;
-       argc--, argv++;
-       while (argc > 0) {
-               if (!strcmp(*argv, "-n"))
+       ttyn = ttyname(2);
+       while ((ch = getopt(argc, argv, "lnqy")) != EOF)
+               switch((char)ch) {
+               case 'l':               /* undocumented; for shutdown(8) */
+                       needlog = 0;
+                       break;
+               case 'n':
                        howto |= RB_NOSYNC;
                        howto |= RB_NOSYNC;
-               else if (!strcmp(*argv, "-y"))
+                       break;
+               case 'q':
+                       qflag++;
+                       break;
+               case 'y':
                        ttyn = 0;
                        ttyn = 0;
-               else {
-                       fprintf(stderr, "usage: halt [ -n ]\n");
+                       break;
+               case '?':
+               default:
+                       fprintf(stderr, "usage: halt [-nqy]\n");
                        exit(1);
                }
                        exit(1);
                }
-               argc--, argv++;
-       }
+
        if (ttyn && *(ttyn+strlen("/dev/tty")) == 'd') {
                fprintf(stderr, "halt: dangerous on a dialup; use ``halt -y'' if you are really sure\n");
                exit(1);
        }
        if (ttyn && *(ttyn+strlen("/dev/tty")) == 'd') {
                fprintf(stderr, "halt: dangerous on a dialup; use ``halt -y'' if you are really sure\n");
                exit(1);
        }
+
+       if (needlog) {
+               openlog("halt", 0, LOG_AUTH);
+               if ((user = getlogin()) == NULL)
+                       if ((pw = getpwuid(getuid())))
+                               user = pw->pw_name;
+                       else
+                               user = "???";
+               syslog(LOG_CRIT, "halted by %s", user);
+       }
+
+       signal(SIGHUP, SIG_IGN);                /* for network connections */
+       if (kill(1, SIGTSTP) == -1) {
+               fprintf(stderr, "halt: can't idle init\n");
+               exit(1);
+       }
+       sleep(1);
+       (void) kill(-1, SIGTERM);       /* one chance to catch it */
+       sleep(5);
+
+       if (!qflag) for (i = 1; ; i++) {
+               if (kill(-1, SIGKILL) == -1) {
+                       extern int errno;
+
+                       if (errno == ESRCH)
+                               break;
+
+                       perror("halt: kill");
+                       kill(1, SIGHUP);
+                       exit(1);
+               }
+               if (i > 5) {
+                       fprintf(stderr,
+                           "CAUTION: some process(es) wouldn't die\n");
+                       break;
+               }
+               setalarm(2 * i);
+               pause();
+       }
+
+       if (!qflag && (howto & RB_NOSYNC) == 0) {
+               logwtmp("~", "shutdown", "");
+               sync();
+               setalarm(5);
+               pause();
+       }
        syscall(55, howto);
        syscall(55, howto);
-       perror("reboot");
+       perror("halt");
+}
+
+dingdong()
+{
+       /* RRRIIINNNGGG RRRIIINNNGGG */
+}
+
+setalarm(n)
+{
+       signal(SIGALRM, dingdong);
+       alarm(n);
 }
 }