#include "parse.h"
#include "debug.h"
+#include "options.h"
static struct fio_option *fio_options;
+extern unsigned int fio_get_kb_base(void *);
static int vp_cmp(const void *p1, const void *p2)
{
}
}
-static unsigned long get_mult_bytes(char c)
+static unsigned long long get_mult_bytes(char c, void *data)
{
+ unsigned int kb_base = fio_get_kb_base(data);
+ unsigned long long ret = 1;
+
switch (c) {
- case 'k':
- case 'K':
- return 1024;
- case 'm':
- case 'M':
- return 1024 * 1024;
+ default:
+ break;
+ case 'p':
+ case 'P':
+ ret *= (unsigned long long) kb_base;
+ case 't':
+ case 'T':
+ ret *= (unsigned long long) kb_base;
case 'g':
case 'G':
- return 1024 * 1024 * 1024;
- case 'e':
- case 'E':
- return 1024 * 1024 * 1024 * 1024UL;
- default:
- return 1;
+ ret *= (unsigned long long) kb_base;
+ case 'm':
+ case 'M':
+ ret *= (unsigned long long) kb_base;
+ case 'k':
+ case 'K':
+ ret *= (unsigned long long) kb_base;
+ break;
}
+
+ return ret;
}
/*
* convert string into decimal value, noting any size suffix
*/
-int str_to_decimal(const char *str, long long *val, int kilo)
+int str_to_decimal(const char *str, long long *val, int kilo, void *data)
{
int len, base;
if (*val == LONG_MAX && errno == ERANGE)
return 1;
- if (kilo)
- *val *= get_mult_bytes(str[len - 1]);
- else
+ if (kilo) {
+ const char *p;
+ /*
+ * if the last char is 'b' or 'B', the user likely used
+ * "1gb" instead of just "1g". If the second to last is also
+ * a letter, adjust.
+ */
+ p = str + len - 1;
+ if ((*p == 'b' || *p == 'B') && isalpha(*(p - 1)))
+ --p;
+
+ *val *= get_mult_bytes(*p, data);
+ } else
*val *= get_mult_time(str[len - 1]);
return 0;
}
-static int check_str_bytes(const char *p, long long *val)
+static int check_str_bytes(const char *p, long long *val, void *data)
{
- return str_to_decimal(p, val, 1);
+ return str_to_decimal(p, val, 1, data);
}
static int check_str_time(const char *p, long long *val)
{
- return str_to_decimal(p, val, 0);
+ return str_to_decimal(p, val, 0, NULL);
}
void strip_blank_front(char **p)
*(s + 1) = '\0';
}
-static int check_range_bytes(const char *str, long *val)
+static int check_range_bytes(const char *str, long *val, void *data)
{
char suffix;
return 1;
if (sscanf(str, "%lu%c", val, &suffix) == 2) {
- *val *= get_mult_bytes(suffix);
+ *val *= get_mult_bytes(suffix, data);
return 0;
}
return 1;
}
+static inline int o_match(struct fio_option *o, const char *opt)
+{
+ if (!strcmp(o->name, opt))
+ return 1;
+ else if (o->alias && !strcmp(o->alias, opt))
+ return 1;
+
+ return 0;
+}
+
static struct fio_option *find_option(struct fio_option *options,
- const char *opt)
+ struct flist_head *eops, const char *opt)
{
+ struct flist_head *n;
struct fio_option *o;
- for (o = &options[0]; o->name; o++) {
- if (!strcmp(o->name, opt))
- return o;
- else if (o->alias && !strcmp(o->alias, opt))
+ for (o = &options[0]; o->name; o++)
+ if (o_match(o, opt))
return o;
+
+ if (!eops)
+ return NULL;
+
+ flist_for_each(n, eops) {
+ struct ext_option *eopt;
+
+ eopt = flist_entry(n, struct ext_option, list);
+ if (o_match(&eopt->o, opt))
+ return &eopt->o;
}
return NULL;
if (is_time)
ret = check_str_time(ptr, &ull);
else
- ret = check_str_bytes(ptr, &ull);
+ ret = check_str_bytes(ptr, &ull, data);
if (ret)
break;
p1 = tmp;
ret = 1;
- if (!check_range_bytes(p1, &ul1) &&
- !check_range_bytes(p2, &ul2)) {
+ if (!check_range_bytes(p1, &ul1, data) &&
+ !check_range_bytes(p2, &ul2, data)) {
ret = 0;
if (ul1 > ul2) {
unsigned long foo = ul1;
}
static struct fio_option *get_option(const char *opt,
- struct fio_option *options, char **post)
+ struct fio_option *options,
+ struct flist_head *eops, char **post)
{
struct fio_option *o;
char *ret;
ret = (char *) opt;
(*post)++;
strip_blank_end(ret);
- o = find_option(options, ret);
+ o = find_option(options, eops, ret);
} else {
- o = find_option(options, opt);
+ o = find_option(options, eops, opt);
*post = NULL;
}
s1 = strdup(*((char **) p1));
s2 = strdup(*((char **) p2));
- o1 = get_option(s1, fio_options, &foo);
- o2 = get_option(s2, fio_options, &foo);
-
+ o1 = get_option(s1, fio_options, NULL, &foo);
+ o2 = get_option(s2, fio_options, NULL, &foo);
+
prio1 = prio2 = 0;
if (o1)
prio1 = o1->prio;
}
int parse_cmd_option(const char *opt, const char *val,
- struct fio_option *options, void *data)
+ struct fio_option *options, struct flist_head *eops,
+ void *data)
{
struct fio_option *o;
- o = find_option(options, opt);
+ o = find_option(options, eops, opt);
if (!o) {
fprintf(stderr, "Bad option <%s>\n", opt);
return 1;
ssize_t nchr = OPT_LEN_MAX;
size_t envlen;
+ if (strlen(in) + 1 > OPT_LEN_MAX) {
+ fprintf(stderr, "OPT_LEN_MAX (%d) is too small\n", OPT_LEN_MAX);
+ return NULL;
+ }
+
in[OPT_LEN_MAX] = '\0';
strncpy(in, opt, OPT_LEN_MAX);
return strdup(out);
}
-int parse_option(const char *opt, struct fio_option *options, void *data)
+int parse_option(const char *opt, struct fio_option *options,
+ struct flist_head *ext_opt_list, void *data)
{
struct fio_option *o;
char *post, *tmp;
tmp = option_dup_subs(opt);
+ if (!tmp)
+ return 1;
- o = get_option(tmp, options, &post);
+ o = get_option(tmp, options, ext_opt_list, &post);
if (!o) {
fprintf(stderr, "Bad option <%s>\n", tmp);
free(tmp);
handle_option(o, o->def, data);
}
+void option_init(struct fio_option *o)
+{
+ if (o->type == FIO_OPT_DEPRECATED)
+ return;
+ if (o->type == FIO_OPT_BOOL) {
+ o->minval = 0;
+ o->maxval = 1;
+ }
+ if (o->type == FIO_OPT_STR_SET && o->def) {
+ fprintf(stderr, "Option %s: string set option with"
+ " default will always be true\n", o->name);
+ }
+ if (!o->cb && !o->off1) {
+ fprintf(stderr, "Option %s: neither cb nor offset given\n",
+ o->name);
+ }
+ if (o->type == FIO_OPT_STR || o->type == FIO_OPT_STR_STORE)
+ return;
+ if (o->cb && (o->off1 || o->off2 || o->off3 || o->off4)) {
+ fprintf(stderr, "Option %s: both cb and offset given\n",
+ o->name);
+ }
+}
+
/*
* Sanitize the options structure. For now it just sets min/max for bool
* values and whether both callback and offsets are given.
dprint(FD_PARSE, "init options\n");
- for (o = &options[0]; o->name; o++) {
- if (o->type == FIO_OPT_DEPRECATED)
- continue;
- if (o->type == FIO_OPT_BOOL) {
- o->minval = 0;
- o->maxval = 1;
- }
- if (o->type == FIO_OPT_STR_SET && o->def) {
- fprintf(stderr, "Option %s: string set option with"
- " default will always be true\n",
- o->name);
- }
- if (!o->cb && !o->off1) {
- fprintf(stderr, "Option %s: neither cb nor offset"
- " given\n", o->name);
- }
- if (o->type == FIO_OPT_STR || o->type == FIO_OPT_STR_STORE)
- continue;
- if (o->cb && (o->off1 || o->off2 || o->off3 || o->off4)) {
- fprintf(stderr, "Option %s: both cb and offset given\n",
- o->name);
- }
- }
+ for (o = &options[0]; o->name; o++)
+ option_init(o);
}