From 8b048fcdf0861c07124be3f56c80cc3f84d8b65a Mon Sep 17 00:00:00 2001 From: Mathieu Desnoyers Date: Fri, 22 Feb 2013 15:11:06 -0500 Subject: [PATCH] Update cmd line arg parsing Signed-off-by: Mathieu Desnoyers --- test-ssd-write.c | 24 ++++++++---------------- 1 file changed, 8 insertions(+), 16 deletions(-) diff --git a/test-ssd-write.c b/test-ssd-write.c index 2bce0fc..49a4238 100644 --- a/test-ssd-write.c +++ b/test-ssd-write.c @@ -71,7 +71,7 @@ #include enum write_mode { - WRITE_RANDOM, + WRITE_RANDOM = 0, WRITE_ZEROES, }; @@ -236,17 +236,6 @@ int main(int argc, char **argv) printf("Creating file %s of length %zu, random seed %u\n", argv[1], len, seed); - if (strcmp(argv[4], "-r") == 0) { - printf("Generating random data\n"); - write_mode = WRITE_RANDOM; - } else if (strcmp(argv[4], "-z") == 0) { - printf("Filling with zeroes (compressible pattern)\n"); - write_mode = WRITE_ZEROES; - } else { - printf("Invalid argument %s\n", argv[4]); - exit(EXIT_FAILURE); - } - for (i = 4; i < argc; i++) { if (strcmp(argv[i], "-z") == 0) { write_mode = WRITE_ZEROES; @@ -255,19 +244,22 @@ int main(int argc, char **argv) } } - if (verify_mode) { - printf("Verification mode activated.\n"); - } - switch (write_mode) { case WRITE_RANDOM: + printf("Generating random data\n"); + break; case WRITE_ZEROES: + printf("Filling with zeroes (compressible pattern)\n"); break; default: printf("Unsupported write-mode\n"); exit(EXIT_FAILURE); } + if (verify_mode) { + printf("Verification mode activated.\n"); + } + /* Grow file */ pos = lseek(fd, len - 1, SEEK_SET); if (pos < 0) { -- 2.34.1