diff options
Diffstat (limited to 'util')
| -rw-r--r-- | util/nvmutil/nvmutil.c | 24 |
1 files changed, 15 insertions, 9 deletions
diff --git a/util/nvmutil/nvmutil.c b/util/nvmutil/nvmutil.c index c1882af2..62821278 100644 --- a/util/nvmutil/nvmutil.c +++ b/util/nvmutil/nvmutil.c @@ -13,6 +13,7 @@ #include <string.h> #include <unistd.h> +static void reset_global_state(int, char **); static void set_cmd(int, char **); static void check_cmd_args(int, char **); static void set_io_flags(int, char **); @@ -99,17 +100,9 @@ int main(int argc, char *argv[]) { argv0 = argv[0]; - - mac = NULL; - invert = 0; - part_modified[0] = 0; - part_modified[1] = 0; - fname = ""; - cmd = NULL; - if (argc < 2) usage(); - fname = argv[1]; + reset_global_state(argc, argv); #ifdef __OpenBSD__ err_if(pledge("stdio rpath wpath unveil", NULL) == -1); err_if(unveil("/dev/urandom", "r") == -1); @@ -146,6 +139,19 @@ main(int argc, char *argv[]) } static void +reset_global_state(int argc, char *argv[]) +{ + mac = NULL; + invert = 0; + part_modified[0] = 0; + part_modified[1] = 0; + fname = ""; + cmd = NULL; + + fname = argv[1]; +} + +static void set_cmd(int argc, char *argv[]) { size_t i; |