diff options
author | Cem Keylan <cem@ckyln.com> | 2020-07-20 01:18:11 +0300 |
---|---|---|
committer | Cem Keylan <cem@ckyln.com> | 2020-07-20 01:18:11 +0300 |
commit | 8f2416cbda412ee42b952810f47c5ccef3560932 (patch) | |
tree | a6301f74b165b16f0510b11ebfb0c2db21903c26 /bin/shalt.c | |
parent | acb42d5c8d53331a6257c1a472686faaa5e5dad9 (diff) | |
download | init-8f2416cbda412ee42b952810f47c5ccef3560932.tar.gz |
shalt: change coding style, add response, parse arguments with arg.h
Diffstat (limited to 'bin/shalt.c')
-rw-r--r-- | bin/shalt.c | 37 |
1 files changed, 30 insertions, 7 deletions
diff --git a/bin/shalt.c b/bin/shalt.c index b2e199d..1fa5428 100644 --- a/bin/shalt.c +++ b/bin/shalt.c @@ -1,10 +1,33 @@ -// shalt -- simple halt utility +/* shalt -- simple halt utility */ #include <sys/reboot.h> +#include <stdio.h> +#include <stdlib.h> +#include <string.h> -int main (int argc, char *argv[]) { - switch ((int)argv[argc < 2 ? 0 : 1][0]) { - case 'p': reboot(RB_POWER_OFF); break; - case 'r': reboot(RB_AUTOBOOT); break; - default: return 1; - }; return 0; +static char *argv0; +#include "arg.h" + +static void +usage(void) +{ + fprintf(stderr, "usage: %s [-pr]\n", argv0); + exit(1); +} + +int +main(int argc, char *argv[]) +{ + ARGBEGIN { + case 'p': + reboot(RB_POWER_OFF); + break; + case 'r': + reboot(RB_AUTOBOOT); + break; + default: + usage(); + } ARGEND + + if (argc != 2) usage(); + return 0; } |