]>
Commit | Line | Data |
---|---|---|
1 | #include <unistd.h> | |
2 | #include <getopt.h> | |
3 | #include <stdlib.h> | |
4 | #include <stdio.h> | |
5 | #include <smram.h> | |
6 | ||
7 | static struct option longopts[] = { | |
8 | { "open", no_argument, NULL, 'o' }, | |
9 | { "close", no_argument, NULL, 'c' }, | |
10 | { "dump", no_argument, NULL, 'd' }, | |
11 | { NULL, 0, NULL, 0 } | |
12 | }; | |
13 | ||
14 | #define OP_DUMP 1 | |
15 | #define OP_SET 2 | |
16 | ||
17 | void usage(int argc, char **argv) | |
18 | { | |
19 | printf("Usage: %s [ --dump ] [ --open | --close ]\n", | |
20 | argv[0]); | |
21 | exit(1); | |
22 | } | |
23 | ||
24 | int main(int argc, char **argv) | |
25 | { | |
26 | if (geteuid() != 0) | |
27 | { | |
28 | printf("%s: This program must be run as root.\n", argv[0]); | |
29 | return 1; | |
30 | } | |
31 | ||
32 | int ch; | |
33 | int op = 0; | |
34 | int do_open = 0; | |
35 | while ((ch = getopt_long(argc, argv, "ocsd:", longopts, NULL)) != -1) | |
36 | { | |
37 | switch (ch) | |
38 | { | |
39 | case 'o': | |
40 | if (op & OP_SET) usage(argc, argv); | |
41 | op |= OP_SET; | |
42 | do_open = 1; | |
43 | break; | |
44 | case 'c': | |
45 | if (op & OP_SET) usage(argc, argv); | |
46 | op |= OP_SET; | |
47 | break; | |
48 | case 'd': | |
49 | op |= OP_DUMP; | |
50 | break; | |
51 | default: | |
52 | usage(argc, argv); | |
53 | } | |
54 | } | |
55 | ||
56 | if (!op) usage(argc, argv); | |
57 | ||
58 | if (op & OP_DUMP) | |
59 | { | |
60 | smram_aseg_dump(); | |
61 | } | |
62 | ||
63 | if (op & OP_SET) | |
64 | { | |
65 | smram_aseg_set_state(do_open ? SMRAM_ASEG_OPEN : SMRAM_ASEG_SMMONLY); | |
66 | } | |
67 | ||
68 | return 0; | |
69 | } |