mirror of git://git.suckless.org/ubase
52 lines
759 B
C
52 lines
759 B
C
/* See LICENSE file for copyright and license details. */
|
|
#include <sys/syscall.h>
|
|
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <unistd.h>
|
|
|
|
#include "reboot.h"
|
|
#include "util.h"
|
|
|
|
static void
|
|
usage(void)
|
|
{
|
|
eprintf("usage: %s [-pr]\n", argv0);
|
|
}
|
|
|
|
int
|
|
main(int argc, char *argv[])
|
|
{
|
|
int pflag = 0, rflag = 0;
|
|
int cmd = LINUX_REBOOT_CMD_HALT;
|
|
|
|
ARGBEGIN {
|
|
case 'p':
|
|
pflag = 1;
|
|
break;
|
|
case 'r':
|
|
rflag = 1;
|
|
break;
|
|
default:
|
|
usage();
|
|
} ARGEND;
|
|
|
|
if (argc > 0)
|
|
usage();
|
|
|
|
sync();
|
|
|
|
if (pflag && rflag)
|
|
usage();
|
|
|
|
if (pflag)
|
|
cmd = LINUX_REBOOT_CMD_POWER_OFF;
|
|
if (rflag)
|
|
cmd = LINUX_REBOOT_CMD_RESTART;
|
|
|
|
if (syscall(__NR_reboot, LINUX_REBOOT_MAGIC1,
|
|
LINUX_REBOOT_MAGIC2, cmd, NULL) < 0)
|
|
eprintf("reboot:");
|
|
return EXIT_SUCCESS;
|
|
}
|