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
  57         *val = strtoul(str, NULL, 10);
  58         if (*val == ULONG_MAX && errno == ERANGE)
  59                 return 1;
  60
  61         if (kilo)
  62                 *val *= get_mult_bytes(str[len - 1]);
  63         else
  64                 *val *= get_mult_time(str[len - 1]);
  65         return 0;
  66 }
  67
  68 static int check_str_bytes(const char *p, unsigned long long *val)
  69 {
  70         return str_to_decimal(p, val, 1);
  71 }
  72
  73 static int check_str_time(const char *p, unsigned long long *val)
  74 {
  75         return str_to_decimal(p, val, 0);
  76 }
  77
  78 void strip_blank_front(char **p)
  79 {
  80         char *s = *p;
  81
  82         while (isspace(*s))
  83                 s++;
  84 }
  85
  86 void strip_blank_end(char *p)
  87 {
  88         char *s = p + strlen(p) - 1;
  89
  90         while (isspace(*s) || iscntrl(*s))
  91                 s--;
  92
  93         *(s + 1) = '\0';
  94 }
  95
  96 static int check_range_bytes(const char *str, unsigned long *val)
  97 {
  98         char suffix;
  99
 100         if (sscanf(str, "%lu%c", val, &suffix) == 2) {
 101                 *val *= get_mult_bytes(suffix);
 102                 return 0;
 103         }
 104
 105         if (sscanf(str, "%lu", val) == 1)
 106                 return 0;
 107
 108         return 1;
 109 }
 110
 111 static int check_int(const char *p, unsigned int *val)
 112 {
 113         if (sscanf(p, "%u", val) == 1)
 114                 return 0;
 115
 116         return 1;
 117 }
 118
 119 static struct fio_option *find_option(struct fio_option *options,
 120                                       const char *opt)
 121 {
 122         struct fio_option *o = &options[0];
 123
 124         while (o->name) {
 125                 if (!strcmp(o->name, opt))
 126                         return o;
 127
 128                 o++;
 129         }
 130
 131         return NULL;
 132 }
 133
 134 static int handle_option(struct fio_option *o, const char *ptr, void *data)
 135 {
 136         unsigned int il, *ilp1, *ilp2;
 137         unsigned long long ull, *ullp;
 138         unsigned long ul1, ul2;
 139         char **cp;
 140         int ret = 0, is_time = 0;
 141
 142         switch (o->type) {
 143         case FIO_OPT_STR: {
 144                 fio_opt_str_fn *fn = o->cb;
 145
 146                 ret = fn(data, ptr);
 147                 break;
 148         }
 149         case FIO_OPT_STR_VAL_TIME:
 150                 is_time = 1;
 151         case FIO_OPT_STR_VAL:
 152         case FIO_OPT_STR_VAL_INT: {
 153                 fio_opt_str_val_fn *fn = o->cb;
 154
 155                 if (is_time)
 156                         ret = check_str_time(ptr, &ull);
 157                 else
 158                         ret = check_str_bytes(ptr, &ull);
 159
 160                 if (ret)
 161                         break;
 162
 163                 if (o->max_val && ull > o->max_val)
 164                         ull = o->max_val;
 165
 166                 if (fn)
 167                         ret = fn(data, &ull);
 168                 else {
 169                         if (o->type == FIO_OPT_STR_VAL_INT) {
 170                                 ilp1 = td_var(data, o->off1);
 171                                 *ilp1 = ull;
 172                         } else {
 173                                 ullp = td_var(data, o->off1);
 174                                 *ullp = ull;
 175                         }
 176                 }
 177                 break;
 178         }
 179         case FIO_OPT_STR_STORE:
 180                 cp = td_var(data, o->off1);
 181                 *cp = strdup(ptr);
 182                 break;
 183         case FIO_OPT_RANGE: {
 184                 char tmp[128];
 185                 char *p1, *p2;
 186
 187                 strncpy(tmp, ptr, sizeof(tmp) - 1);
 188
 189                 p1 = strchr(tmp, '-');
 190                 if (!p1) {
 191                         ret = 1;
 192                         break;
 193                 }
 194
 195                 p2 = p1 + 1;
 196                 *p1 = '\0';
 197                 p1 = tmp;
 198
 199                 ret = 1;
 200                 if (!check_range_bytes(p1, &ul1) && !check_range_bytes(p2, &ul2)) {
 201                         ret = 0;
 202                         ilp1 = td_var(data, o->off1);
 203                         ilp2 = td_var(data, o->off2);
 204                         if (ul1 > ul2) {
 205                                 *ilp1 = ul2;
 206                                 *ilp2 = ul1;
 207                         } else {
 208                                 *ilp2 = ul2;
 209                                 *ilp1 = ul1;
 210                         }
 211                 }
 212
 213                 break;
 214         }
 215         case FIO_OPT_INT: {
 216                 fio_opt_int_fn *fn = o->cb;
 217
 218                 ret = check_int(ptr, &il);
 219                 if (ret)
 220                         break;
 221
 222                 if (o->max_val && il > o->max_val)
 223                         il = o->max_val;
 224
 225                 if (fn)
 226                         ret = fn(data, &il);
 227                 else {
 228                         ilp1 = td_var(data, o->off1);
 229                         *ilp1 = il;
 230                 }
 231                 break;
 232         }
 233         case FIO_OPT_STR_SET: {
 234                 fio_opt_str_set_fn *fn = o->cb;
 235
 236                 if (fn)
 237                         ret = fn(data);
 238                 else {
 239                         ilp1 = td_var(data, o->off1);
 240                         *ilp1 = 1;
 241                 }
 242                 break;
 243         }
 244         default:
 245                 fprintf(stderr, "Bad option type %d\n", o->type);
 246                 ret = 1;
 247         }
 248
 249         return ret;
 250 }
 251
 252 int parse_cmd_option(const char *opt, const char *val,
 253                      struct fio_option *options, void *data)
 254 {
 255         struct fio_option *o;
 256
 257         o = find_option(options, opt);
 258         if (!o) {
 259                 fprintf(stderr, "Bad option %s\n", opt);
 260                 return 1;
 261         }
 262
 263         if (!handle_option(o, val, data))
 264                 return 0;
 265
 266         fprintf(stderr, "fio: failed parsing %s=%s\n", opt, val);
 267         return 1;
 268 }
 269
 270 int parse_option(const char *opt, struct fio_option *options, void *data)
 271 {
 272         struct fio_option *o;
 273         char *pre, *post;
 274         char tmp[64];
 275
 276         strncpy(tmp, opt, sizeof(tmp) - 1);
 277
 278         pre = strchr(tmp, '=');
 279         if (pre) {
 280                 post = pre;
 281                 *pre = '\0';
 282                 pre = tmp;
 283                 post++;
 284                 o = find_option(options, pre);
 285         } else {
 286                 o = find_option(options, tmp);
 287                 post = NULL;
 288         }
 289
 290         if (!o) {
 291                 fprintf(stderr, "Bad option %s\n", tmp);
 292                 return 1;
 293         }
 294
 295         if (!handle_option(o, post, data))
 296                 return 0;
 297
 298         fprintf(stderr, "fio: failed parsing %s\n", opt);
 299         return 1;
 300 }