X-Git-Url: https://git.kernel.dk/?p=fio.git;a=blobdiff_plain;f=parse.c;h=8aaf7992754f1de3e42211a1e3b9c1ef05eddb06;hp=27e7336b49af99720886905928a69058e42f3bf1;hb=3f36aa6b73eb5730d7de716391dcf0735e886d10;hpb=28d7c363e2f82d09cf7f6aa772d255712aff5c5c diff --git a/parse.c b/parse.c index 27e7336b..8aaf7992 100644 --- a/parse.c +++ b/parse.c @@ -199,9 +199,9 @@ static unsigned long long get_mult_bytes(const char *str, int len, void *data, if (len < 2) return __get_mult_bytes(str, data, percent); - /* - * Go forward until we hit a non-digit, or +/- sign - */ + /* + * Go forward until we hit a non-digit, or +/- sign + */ while ((p - str) <= len) { if (!isdigit((int) *p) && (((*p != '+') && (*p != '-')) || digit_seen)) @@ -273,6 +273,8 @@ void strip_blank_front(char **p) { char *s = *p; + if (!strlen(s)) + return; while (isspace((int) *s)) s++; @@ -283,6 +285,9 @@ void strip_blank_end(char *p) { char *start = p, *s; + if (!strlen(p)) + return; + s = strchr(p, ';'); if (s) *s = '\0'; @@ -496,45 +501,54 @@ 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; - posval_sort(o, posval); + if (o->roff1 || o->off1) { + if (o->roff1) + cp = (char **) o->roff1; + else if (o->off1) + cp = td_var(data, o->off1); - if (!o->posval[0].ival) { - vp = NULL; - goto match; + *cp = strdup(ptr); + } else { + cp = NULL; } - ret = 1; - for (i = 0; i < PARSE_MAX_VP; i++) { - vp = &posval[i]; - if (!vp->ival || vp->ival[0] == '\0') - continue; - all_skipped = 0; - if (!strncmp(vp->ival, ptr, opt_len(ptr))) { - char *rest; + if (fn) + ret = fn(data, ptr); + else if (o->posval[0].ival) { + posval_sort(o, posval); - ret = 0; - if (vp->cb) - fn = vp->cb; -match: - if (o->roff1) - cp = (char **) o->roff1; - else - cp = td_var(data, o->off1); - *cp = strdup(ptr); - rest = strstr(*cp, ":"); - if (rest) { - *rest = '\0'; - ptr = rest + 1; - } else if (vp && vp->cb) - ptr = NULL; - break; + ret = 1; + for (i = 0; i < PARSE_MAX_VP; i++) { + vp = &posval[i]; + if (!vp->ival || vp->ival[0] == '\0') + continue; + all_skipped = 0; + if (!strncmp(vp->ival, ptr, opt_len(ptr))) { + char *rest; + + ret = 0; + if (vp->cb) + fn = vp->cb; + rest = strstr(*cp ?: ptr, ":"); + if (rest) { + if (*cp) + *rest = '\0'; + ptr = rest + 1; + } else + ptr = NULL; + break; + } } } - if (ret && !all_skipped) - show_option_values(o); - else if (fn && ptr) - ret = fn(data, ptr); + if (!all_skipped) { + if (ret && !*cp) + show_option_values(o); + else if (ret && *cp) + ret = 0; + else if (fn && ptr) + ret = fn(data, ptr); + } break; } @@ -723,7 +737,7 @@ static int handle_option(struct fio_option *o, const char *__ptr, void *data) return ret; } -static struct fio_option *get_option(const char *opt, +static struct fio_option *get_option(char *opt, struct fio_option *options, char **post) { struct fio_option *o; @@ -733,7 +747,7 @@ static struct fio_option *get_option(const char *opt, if (ret) { *post = ret; *ret = '\0'; - ret = (char *) opt; + ret = opt; (*post)++; strip_blank_end(ret); o = find_option(options, ret); @@ -747,24 +761,27 @@ static struct fio_option *get_option(const char *opt, static int opt_cmp(const void *p1, const void *p2) { - struct fio_option *o1, *o2; - char *s1, *s2, *foo; + struct fio_option *o; + char *s, *foo; int prio1, prio2; - s1 = strdup(*((char **) p1)); - s2 = strdup(*((char **) p2)); - - o1 = get_option(s1, fio_options, &foo); - o2 = get_option(s2, fio_options, &foo); - prio1 = prio2 = 0; - if (o1) - prio1 = o1->prio; - if (o2) - prio2 = o2->prio; - free(s1); - free(s2); + if (*(char **)p1) { + s = strdup(*((char **) p1)); + o = get_option(s, fio_options, &foo); + if (o) + prio1 = o->prio; + free(s); + } + if (*(char **)p2) { + s = strdup(*((char **) p2)); + o = get_option(s, fio_options, &foo); + if (o) + prio2 = o->prio; + free(s); + } + return prio2 - prio1; } @@ -793,83 +810,33 @@ int parse_cmd_option(const char *opt, const char *val, return 1; } -/* - * Return a copy of the input string with substrings of the form ${VARNAME} - * substituted with the value of the environment variable VARNAME. The - * substitution always occurs, even if VARNAME is empty or the corresponding - * environment variable undefined. - */ -static char *option_dup_subs(const char *opt) +int parse_option(char *opt, const char *input, + struct fio_option *options, struct fio_option **o, void *data) { - char out[OPT_LEN_MAX+1]; - char in[OPT_LEN_MAX+1]; - char *outptr = out; - char *inptr = in; - char *ch1, *ch2, *env; - ssize_t nchr = OPT_LEN_MAX; - size_t envlen; - - if (strlen(opt) + 1 > OPT_LEN_MAX) { - log_err("OPT_LEN_MAX (%d) is too small\n", OPT_LEN_MAX); - return NULL; - } - - in[OPT_LEN_MAX] = '\0'; - strncpy(in, opt, OPT_LEN_MAX); - - while (*inptr && nchr > 0) { - if (inptr[0] == '$' && inptr[1] == '{') { - ch2 = strchr(inptr, '}'); - if (ch2 && inptr+1 < ch2) { - ch1 = inptr+2; - inptr = ch2+1; - *ch2 = '\0'; - - env = getenv(ch1); - if (env) { - envlen = strlen(env); - if (envlen <= nchr) { - memcpy(outptr, env, envlen); - outptr += envlen; - nchr -= envlen; - } - } - - continue; - } - } - - *outptr++ = *inptr++; - --nchr; - } + char *post; - *outptr = '\0'; - return strdup(out); -} - -int parse_option(const char *opt, struct fio_option *options, void *data) -{ - struct fio_option *o; - char *post, *tmp; - - tmp = option_dup_subs(opt); - if (!tmp) + if (!opt) { + log_err("fio: failed parsing %s\n", input); + *o = NULL; return 1; + } - o = get_option(tmp, options, &post); - if (!o) { - log_err("Bad option <%s>\n", tmp); - free(tmp); + *o = get_option(opt, options, &post); + if (!*o) { + if (post) { + int len = strlen(opt); + if (opt + len + 1 != post) + memmove(opt + len + 1, post, strlen(post)); + opt[len] = '='; + } return 1; } - if (!handle_option(o, post, data)) { - free(tmp); + if (!handle_option(*o, post, data)) { return 0; } - log_err("fio: failed parsing %s\n", opt); - free(tmp); + log_err("fio: failed parsing %s\n", input); return 1; } @@ -1094,3 +1061,22 @@ void options_init(struct fio_option *options) for (o = &options[0]; o->name; o++) option_init(o); } + +void options_free(struct fio_option *options, void *data) +{ + struct fio_option *o; + char **ptr; + + dprint(FD_PARSE, "free options\n"); + + for (o = &options[0]; o->name; o++) { + if (o->type != FIO_OPT_STR_STORE || !o->off1) + continue; + + ptr = td_var(data, o->off1); + if (*ptr) { + free(*ptr); + *ptr = NULL; + } + } +}