reboot.c revision dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0 
1#include <errno.h>
2#include <stdio.h>
3#include <stdlib.h>
4#include <sys/reboot.h>
5#include <unistd.h>
6
7int 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