X-Git-Url: https://git.kernel.dk/?p=fio.git;a=blobdiff_plain;f=parse.c;h=83c59f7026ca7e06b5371d59adfb2b0bd2d39b35;hp=6121dfcdd7e9efb083ada55173d8cac863549042;hb=9bdb9265ed6cfd95fc217a9030b188a5a345cc89;hpb=f0fdbcafc3a62b84250e5ccdcaec47e3ffa93a00 diff --git a/parse.c b/parse.c index 6121dfcd..83c59f70 100644 --- a/parse.c +++ b/parse.c @@ -122,21 +122,48 @@ static void show_option_help(struct fio_option *o, int is_err) show_option_values(o); } -static unsigned long get_mult_time(char c) +static unsigned long long get_mult_time(const char *str, int len, + int is_seconds) { - switch (c) { - case 'm': - case 'M': - return 60; - case 'h': - case 'H': - return 60 * 60; - case 'd': - case 'D': - return 24 * 60 * 60; - default: - return 1; + const char *p = str; + char *c; + unsigned long long mult = 1; + + /* + * Go forward until we hit a non-digit, or +/- sign + */ + while ((p - str) <= len) { + if (!isdigit((int) *p) && (*p != '+') && (*p != '-')) + break; + p++; } + + if (!isalpha((int) *p)) { + if (is_seconds) + return 1000000UL; + else + return 1; + } + + c = strdup(p); + for (int i = 0; i < strlen(c); i++) + c[i] = tolower(c[i]); + + if (!strncmp("us", c, 2) || !strncmp("usec", c, 4)) + mult = 1; + else if (!strncmp("ms", c, 2) || !strncmp("msec", c, 4)) + mult = 1000; + else if (!strcmp("s", c)) + mult = 1000000; + else if (!strcmp("m", c)) + mult = 60 * 1000000UL; + else if (!strcmp("h", c)) + mult = 60 * 60 * 1000000UL; + else if (!strcmp("d", c)) + mult = 24 * 60 * 60 * 1000000UL; + + free(c); + return mult; } static int is_separator(char c) @@ -248,7 +275,8 @@ int str_to_float(const char *str, double *val) /* * convert string into decimal value, noting any size suffix */ -int str_to_decimal(const char *str, long long *val, int kilo, void *data) +int str_to_decimal(const char *str, long long *val, int kilo, void *data, + int is_seconds) { int len, base; @@ -275,19 +303,19 @@ int str_to_decimal(const char *str, long long *val, int kilo, void *data) else *val *= mult; } else - *val *= get_mult_time(str[len - 1]); + *val *= get_mult_time(str, len, is_seconds); return 0; } int check_str_bytes(const char *p, long long *val, void *data) { - return str_to_decimal(p, val, 1, data); + return str_to_decimal(p, val, 1, data, 0); } -int check_str_time(const char *p, long long *val) +int check_str_time(const char *p, long long *val, int is_seconds) { - return str_to_decimal(p, val, 0, NULL); + return str_to_decimal(p, val, 0, NULL, is_seconds); } void strip_blank_front(char **p) @@ -329,7 +357,7 @@ static int check_range_bytes(const char *str, long *val, void *data) { long long __val; - if (!str_to_decimal(str, &__val, 1, data)) { + if (!str_to_decimal(str, &__val, 1, data, 0)) { *val = __val; return 0; } @@ -415,7 +443,7 @@ static int __handle_option(struct fio_option *o, const char *ptr, void *data, if (!strncmp(vp->ival, ptr, str_match_len(vp, ptr))) { ret = 0; if (o->off1) - val_store(ilp, vp->oval, o->off1, vp->or, data, o); + val_store(ilp, vp->oval, o->off1, vp->orval, data, o); continue; } } @@ -439,7 +467,7 @@ static int __handle_option(struct fio_option *o, const char *ptr, void *data, *p = '\0'; if (is_time) - ret = check_str_time(tmp, &ull); + ret = check_str_time(tmp, &ull, o->is_seconds); else ret = check_str_bytes(tmp, &ull, data); @@ -580,6 +608,9 @@ static int __handle_option(struct fio_option *o, const char *ptr, void *data, case FIO_OPT_STR_STORE: { fio_opt_str_fn *fn = o->cb; + if (!strlen(ptr)) + return 1; + if (o->off1) { cp = td_var(data, o, o->off1); *cp = strdup(ptr);