parse.c

   1 /*
   2  * This file contains the ini and command liner parser main.
   3  */
   4 #include <stdio.h>
   5 #include <stdlib.h>
   6 #include <unistd.h>
   7 #include <ctype.h>
   8 #include <string.h>
   9 #include <errno.h>
  10 #include <limits.h>
  11
  12 #include "parse.h"
  13
  14 static unsigned long get_mult_time(char c)
  15 {
  16         switch (c) {
  17                 case 'm':
  18                 case 'M':
  19                         return 60;
  20                 case 'h':
  21                 case 'H':
  22                         return 60 * 60;
  23                 case 'd':
  24                 case 'D':
  25                         return 24 * 60 * 60;
  26                 default:
  27                         return 1;
  28         }
  29 }
  30
  31 static unsigned long get_mult_bytes(char c)
  32 {
  33         switch (c) {
  34                 case 'k':
  35                 case 'K':
  36                         return 1024;
  37                 case 'm':
  38                 case 'M':
  39                         return 1024 * 1024;
  40                 case 'g':
  41                 case 'G':
  42                         return 1024 * 1024 * 1024;
  43                 default:
  44                         return 1;
  45         }
  46 }
  47
  48 /*
  49  * convert string into decimal value, noting any size suffix
  50  */
  51 static int str_to_decimal(const char *str, unsigned long long *val, int kilo)
  52 {
  53         int len;
  54
  55         len = strlen(str);
  56         if (!len)
  57                 return 1;
  58
  59         *val = strtoul(str, NULL, 10);
  60         if (*val == ULONG_MAX && errno == ERANGE)
  61                 return 1;
  62
  63         if (kilo)
  64                 *val *= get_mult_bytes(str[len - 1]);
  65         else
  66                 *val *= get_mult_time(str[len - 1]);
  67
  68         return 0;
  69 }
  70
  71 static int check_str_bytes(const char *p, unsigned long long *val)
  72 {
  73         return str_to_decimal(p, val, 1);
  74 }
  75
  76 static int check_str_time(const char *p, unsigned long long *val)
  77 {
  78         return str_to_decimal(p, val, 0);
  79 }
  80
  81 void strip_blank_front(char **p)
  82 {
  83         char *s = *p;
  84
  85         while (isspace(*s))
  86                 s++;
  87 }
  88
  89 void strip_blank_end(char *p)
  90 {
  91         char *s = p + strlen(p) - 1;
  92
  93         while (isspace(*s) || iscntrl(*s))
  94                 s--;
  95
  96         *(s + 1) = '\0';
  97 }
  98
  99 static int check_range_bytes(const char *str, unsigned long *val)
 100 {
 101         char suffix;
 102
 103         if (!strlen(str))
 104                 return 1;
 105
 106         if (sscanf(str, "%lu%c", val, &suffix) == 2) {
 107                 *val *= get_mult_bytes(suffix);
 108                 return 0;
 109         }
 110
 111         if (sscanf(str, "%lu", val) == 1)
 112                 return 0;
 113
 114         return 1;
 115 }
 116
 117 static int check_int(const char *p, unsigned int *val)
 118 {
 119         if (!strlen(p))
 120                 return 1;
 121         if (sscanf(p, "%u", val) == 1)
 122                 return 0;
 123
 124         return 1;
 125 }
 126
 127 static struct fio_option *find_option(struct fio_option *options,
 128                                       const char *opt)
 129 {
 130         struct fio_option *o = &options[0];
 131
 132         while (o->name) {
 133                 if (!strcmp(o->name, opt))
 134                         return o;
 135
 136                 o++;
 137         }
 138
 139         return NULL;
 140 }
 141
 142 #define val_store(ptr, val, off, data)                  \
 143         do {                                            \
 144                 ptr = td_var((data), (off));            \
 145                 *ptr = (val);                           \
 146         } while (0)
 147
 148 static int __handle_option(struct fio_option *o, const char *ptr, void *data,
 149                            int first, int more)
 150 {
 151         unsigned int il, *ilp;
 152         unsigned long long ull, *ullp;
 153         unsigned long ul1, ul2;
 154         char **cp;
 155         int ret = 0, is_time = 0;
 156
 157         switch (o->type) {
 158         case FIO_OPT_STR: {
 159                 fio_opt_str_fn *fn = o->cb;
 160
 161                 ret = fn(data, ptr);
 162                 break;
 163         }
 164         case FIO_OPT_STR_VAL_TIME:
 165                 is_time = 1;
 166         case FIO_OPT_STR_VAL:
 167         case FIO_OPT_STR_VAL_INT: {
 168                 fio_opt_str_val_fn *fn = o->cb;
 169
 170                 if (is_time)
 171                         ret = check_str_time(ptr, &ull);
 172                 else
 173                         ret = check_str_bytes(ptr, &ull);
 174
 175                 if (ret)
 176                         break;
 177
 178                 if (o->max_val && ull > o->max_val)
 179                         ull = o->max_val;
 180
 181                 if (fn)
 182                         ret = fn(data, &ull);
 183                 else {
 184                         if (o->type == FIO_OPT_STR_VAL_INT) {
 185                                 if (first)
 186                                         val_store(ilp, ull, o->off1, data);
 187                                 if (!more && o->off2)
 188                                         val_store(ilp, ull, o->off2, data);
 189                         } else {
 190                                 if (first)
 191                                         val_store(ullp, ull, o->off1, data);
 192                                 if (!more && o->off2)
 193                                         val_store(ullp, ull, o->off2, data);
 194                         }
 195                 }
 196                 break;
 197         }
 198         case FIO_OPT_STR_STORE:
 199                 cp = td_var(data, o->off1);
 200                 *cp = strdup(ptr);
 201                 break;
 202         case FIO_OPT_RANGE: {
 203                 char tmp[128];
 204                 char *p1, *p2;
 205
 206                 strncpy(tmp, ptr, sizeof(tmp) - 1);
 207
 208                 p1 = strchr(tmp, '-');
 209                 if (!p1) {
 210                         ret = 1;
 211                         break;
 212                 }
 213
 214                 p2 = p1 + 1;
 215                 *p1 = '\0';
 216                 p1 = tmp;
 217
 218                 ret = 1;
 219                 if (!check_range_bytes(p1, &ul1) && !check_range_bytes(p2, &ul2)) {
 220                         ret = 0;
 221                         if (ul1 > ul2) {
 222                                 unsigned long foo = ul1;
 223
 224                                 ul1 = ul2;
 225                                 ul2 = foo;
 226                         }
 227
 228                         if (first) {
 229                                 val_store(ilp, ul1, o->off1, data);
 230                                 val_store(ilp, ul2, o->off2, data);
 231                         }
 232                         if (!more && o->off3 && o->off4) {
 233                                 val_store(ilp, ul1, o->off3, data);
 234                                 val_store(ilp, ul2, o->off4, data);
 235                         }
 236                 }
 237
 238                 break;
 239         }
 240         case FIO_OPT_INT: {
 241                 fio_opt_int_fn *fn = o->cb;
 242
 243                 ret = check_int(ptr, &il);
 244                 if (ret)
 245                         break;
 246
 247                 if (o->max_val && il > o->max_val)
 248                         il = o->max_val;
 249
 250                 if (fn)
 251                         ret = fn(data, &il);
 252                 else {
 253                         if (first)
 254                                 val_store(ilp, il, o->off1, data);
 255                         if (!more && o->off2)
 256                                 val_store(ilp, il, o->off2, data);
 257                 }
 258                 break;
 259         }
 260         case FIO_OPT_STR_SET: {
 261                 fio_opt_str_set_fn *fn = o->cb;
 262
 263                 if (fn)
 264                         ret = fn(data);
 265                 else {
 266                         if (first)
 267                                 val_store(ilp, 1, o->off1, data);
 268                         if (!more && o->off2)
 269                                 val_store(ilp, 1, o->off2, data);
 270                 }
 271                 break;
 272         }
 273         default:
 274                 fprintf(stderr, "Bad option type %d\n", o->type);
 275                 ret = 1;
 276         }
 277
 278         return ret;
 279 }
 280
 281 static int handle_option(struct fio_option *o, const char *ptr, void *data)
 282 {
 283         const char *ptr2;
 284         int r1, r2;
 285
 286         /*
 287          * See if we have a second set of parameters, hidden after a comma.
 288          * Do this before parsing the first round, to check if we should
 289          * copy set 1 options to set 2.
 290          */
 291         ptr2 = strchr(ptr, ',');
 292
 293         /*
 294          * Don't return early if parsing the first option fails - if
 295          * we are doing multiple arguments, we can allow the first one
 296          * being empty.
 297          */
 298         r1 = __handle_option(o, ptr, data, 1, !!ptr2);
 299
 300         if (!ptr2)
 301                 return r1;
 302
 303         ptr2++;
 304         r2 = __handle_option(o, ptr2, data, 0, 0);
 305
 306         return r1 && r2;
 307 }
 308
 309 int parse_cmd_option(const char *opt, const char *val,
 310                      struct fio_option *options, void *data)
 311 {
 312         struct fio_option *o;
 313
 314         o = find_option(options, opt);
 315         if (!o) {
 316                 fprintf(stderr, "Bad option %s\n", opt);
 317                 return 1;
 318         }
 319
 320         if (!handle_option(o, val, data))
 321                 return 0;
 322
 323         fprintf(stderr, "fio: failed parsing %s=%s\n", opt, val);
 324         return 1;
 325 }
 326
 327 int parse_option(const char *opt, struct fio_option *options, void *data)
 328 {
 329         struct fio_option *o;
 330         char *pre, *post;
 331         char tmp[64];
 332
 333         strncpy(tmp, opt, sizeof(tmp) - 1);
 334
 335         pre = strchr(tmp, '=');
 336         if (pre) {
 337                 post = pre;
 338                 *pre = '\0';
 339                 pre = tmp;
 340                 post++;
 341                 o = find_option(options, pre);
 342         } else {
 343                 o = find_option(options, tmp);
 344                 post = NULL;
 345         }
 346
 347         if (!o) {
 348                 fprintf(stderr, "Bad option %s\n", tmp);
 349                 return 1;
 350         }
 351
 352         if (!handle_option(o, post, data))
 353                 return 0;
 354
 355         fprintf(stderr, "fio: failed parsing %s\n", opt);
 356         return 1;
 357 }