1 #include <errno.h>
2 #include <stdio.h>
3 #include <stdlib.h>
4 #include <sys/reboot.h>
5 #include <unistd.h>
6
reboot_main(int argc,char * argv[])7 int reboot_main(int argc, char *argv[])
8 {
9 int ret;
10 int nosync = 0;
11 int poweroff = 0;
12
13 opterr = 0;
14 do {
15 int c;
16
17 c = getopt(argc, argv, "np");
18
19 if (c == EOF) {
20 break;
21 }
22
23 switch (c) {
24 case 'n':
25 nosync = 1;
26 break;
27 case 'p':
28 poweroff = 1;
29 break;
30 case '?':
31 fprintf(stderr, "usage: %s [-n] [-p] [rebootcommand]\n", argv[0]);
32 exit(EXIT_FAILURE);
33 }
34 } while (1);
35
36 if(argc > optind + 1) {
37 fprintf(stderr, "%s: too many arguments\n", argv[0]);
38 exit(EXIT_FAILURE);
39 }
40
41 if(!nosync)
42 sync();
43
44 if(poweroff)
45 ret = __reboot(LINUX_REBOOT_MAGIC1, LINUX_REBOOT_MAGIC2, LINUX_REBOOT_CMD_POWER_OFF, NULL);
46 else if(argc > optind)
47 ret = __reboot(LINUX_REBOOT_MAGIC1, LINUX_REBOOT_MAGIC2, LINUX_REBOOT_CMD_RESTART2, argv[optind]);
48 else
49 ret = reboot(RB_AUTOBOOT);
50 if(ret < 0) {
51 perror("reboot");
52 exit(EXIT_FAILURE);
53 }
54 fprintf(stderr, "reboot returned\n");
55 return 0;
56 }
57