use values from <sys/syscall.h> (4.3BSD/etc/150)

[unix-history] / usr / src / sbin / reboot / reboot.c

/*
* 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[] = "@(#)reboot.c    5.7 (Berkeley) %G%";
#endif /* not lint */

/*
* Reboot
*/
#include <stdio.h>
#include <sys/reboot.h>
#include <errno.h>
#include <signal.h>
#include <pwd.h>
#include <sys/types.h>
#include <sys/time.h>
#include <sys/syslog.h>
#include <sys/syscall.h>

main(argc, argv)
       int argc;
       char **argv;
{
       int howto;
       register char *argp;
       register i;
       register ok = 0;
       register qflag = 0;
       int needlog = 1;
       char *user, *getlogin();
       struct passwd *pw, *getpwuid();

       openlog("reboot", 0, LOG_AUTH);
       argc--, argv++;
       howto = 0;
       while (argc > 0) {
               if (!strcmp(*argv, "-q"))
                       qflag++;
               else if (!strcmp(*argv, "-n"))
                       howto |= RB_NOSYNC;
               else if (!strcmp(*argv, "-l"))
                       needlog = 0;
               else {
                       fprintf(stderr,
                           "usage: reboot [ -n ][ -q ]\n");
                       exit(1);
               }
               argc--, argv++;
       }

       if (needlog) {
               user = getlogin();
               if (user == (char *)0 && (pw = getpwuid(getuid())))
                       user = pw->pw_name;
               if (user == (char *)0)
                       user = "root";
               syslog(LOG_CRIT, "rebooted by %s", user);
       }

       signal(SIGHUP, SIG_IGN);        /* for remote connections */
       if (kill(1, SIGTSTP) == -1) {
               fprintf(stderr, "reboot: 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("reboot: 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(SYS_reboot, howto);
       perror("reboot");
       kill(1, SIGHUP);
       exit(1);
}

dingdong()
{
       /* RRRIIINNNGGG RRRIIINNNGGG */
}

setalarm(n)
{
       signal(SIGALRM, dingdong);
       alarm(n);
}