1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
|
#include <stdlib.h>
#include <unistd.h>
#include <signal.h>
#include <sys/reboot.h>
#include <klibc/compiler.h>
static __noreturn usage(void)
{
static char mesg[] = "Usage: {halt|reboot|poweroff} [-n] [reboot-arg]\n";
write(2, mesg, sizeof(mesg) - 1);
exit(1);
}
int main(int argc, char *argv[])
{
int cmd = 0; /* initalize to shut gcc up */
int do_sync = 1;
char *ptr, *ptr2;
char *reboot_arg = NULL;
/* Which action (program name)? */
ptr2 = ptr = argv[0];
while (*ptr2)
if (*ptr2++ == '/')
ptr = ptr2;
if (*ptr == 'r')
cmd = LINUX_REBOOT_CMD_RESTART;
else if (*ptr == 'h')
cmd = LINUX_REBOOT_CMD_HALT;
else if (*ptr == 'p')
cmd = LINUX_REBOOT_CMD_POWER_OFF;
else
usage();
/* Walk options */
while (*++argv)
if (**argv == '-') {
switch (*++*argv) {
case 'f':
break; /* -f assumed */
case 'n':
do_sync = 0;
break;
default:
usage();
}
} else if (cmd == LINUX_REBOOT_CMD_RESTART) {
reboot_arg = *argv;
cmd = LINUX_REBOOT_CMD_RESTART2;
} else {
usage(); /* args, not reboot == error */
}
if (do_sync)
sync();
reboot(LINUX_REBOOT_CMD_CAD_OFF, NULL); /* Enable CTRL+ALT+DEL */
if (!reboot(cmd, reboot_arg)) {
/* Success. Currently, CMD_HALT returns, so stop the world */
/* kill(-1, SIGSTOP); */
kill(getpid(), SIGSTOP);
}
write(2, "failed.\n", 8);
return 1;
}
|
