kernel/params.c

   1 /* Helpers for initial module or kernel cmdline parsing
   2    Copyright (C) 2001 Rusty Russell.
   3
   4     This program is free software; you can redistribute it and/or modify
   5     it under the terms of the GNU General Public License as published by
   6     the Free Software Foundation; either version 2 of the License, or
   7     (at your option) any later version.
   8
   9     This program is distributed in the hope that it will be useful,
  10     but WITHOUT ANY WARRANTY; without even the implied warranty of
  11     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12     GNU General Public License for more details.
  13
  14     You should have received a copy of the GNU General Public License
  15     along with this program; if not, write to the Free Software
  16     Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  17 */
  18 #include <linux/moduleparam.h>
  19 #include <linux/kernel.h>
  20 #include <linux/string.h>
  21 #include <linux/errno.h>
  22 #include <linux/module.h>
  23
  24 #if 0
  25 #define DEBUGP printk
  26 #else
  27 #define DEBUGP(fmt, a...)
  28 #endif
  29
  30 static inline int dash2underscore(char c)
  31 {
  32         if (c == '-')
  33                 return '_';
  34         return c;
  35 }
  36
  37 static inline int parameq(const char *input, const char *paramname)
  38 {
  39         unsigned int i;
  40         for (i = 0; dash2underscore(input[i]) == paramname[i]; i++)
  41                 if (input[i] == '\0')
  42                         return 1;
  43         return 0;
  44 }
  45
  46 static int parse_one(char *param,
  47                      char *val,
  48                      struct kernel_param *params,
  49                      unsigned num_params,
  50                      int (*handle_unknown)(char *param, char *val))
  51 {
  52         unsigned int i;
  53
  54         /* Find parameter */
  55         for (i = 0; i < num_params; i++) {
  56                 if (parameq(param, params[i].name)) {
  57                         DEBUGP("They are equal!  Calling %p\n",
  58                                params[i].set);
  59                         return params[i].set(val, &params[i]);
  60                 }
  61         }
  62
  63         if (handle_unknown) {
  64                 DEBUGP("Unknown argument: calling %p\n", handle_unknown);
  65                 return handle_unknown(param, val);
  66         }
  67
  68         DEBUGP("Unknown argument `%s'\n", param);
  69         return -ENOENT;
  70 }
  71
  72 /* You can use " around spaces, but can't escape ". */
  73 /* Hyphens and underscores equivalent in parameter names. */
  74 static char *next_arg(char *args, char **param, char **val)
  75 {
  76         unsigned int i, equals = 0;
  77         int in_quote = 0;
  78
  79         /* Chew any extra spaces */
  80         while (*args == ' ') args++;
  81
  82         for (i = 0; args[i]; i++) {
  83                 if (args[i] == ' ' && !in_quote)
  84                         break;
  85                 if (equals == 0) {
  86                         if (args[i] == '=')
  87                                 equals = i;
  88                 }
  89                 if (args[i] == '"')
  90                         in_quote = !in_quote;
  91         }
  92
  93         *param = args;
  94         if (!equals)
  95                 *val = NULL;
  96         else {
  97                 args[equals] = '\0';
  98                 *val = args + equals + 1;
  99
 100                 /* Don't include quotes in value. */
 101                 if (**val == '"') {
 102                         (*val)++;
 103                         if (args[i-1] == '"')
 104                                 args[i-1] = '\0';
 105                 }
 106         }
 107
 108         if (args[i]) {
 109                 args[i] = '\0';
 110                 return args + i + 1;
 111         } else
 112                 return args + i;
 113 }
 114
 115 /* Args looks like "foo=bar,bar2 baz=fuz wiz". */
 116 int parse_args(const char *name,
 117                char *args,
 118                struct kernel_param *params,
 119                unsigned num,
 120                int (*unknown)(char *param, char *val))
 121 {
 122         char *param, *val;
 123
 124         DEBUGP("Parsing ARGS: %s\n", args);
 125
 126         while (*args) {
 127                 int ret;
 128
 129                 args = next_arg(args, &param, &val);
 130                 ret = parse_one(param, val, params, num, unknown);
 131                 switch (ret) {
 132                 case -ENOENT:
 133                         printk(KERN_ERR "%s: Unknown parameter `%s'\n",
 134                                name, param);
 135                         return ret;
 136                 case -ENOSPC:
 137                         printk(KERN_ERR
 138                                "%s: `%s' too large for parameter `%s'\n",
 139                                name, val ?: "", param);
 140                         return ret;
 141                 case 0:
 142                         break;
 143                 default:
 144                         printk(KERN_ERR
 145                                "%s: `%s' invalid for parameter `%s'\n",
 146                                name, val ?: "", param);
 147                         return ret;
 148                 }
 149         }
 150
 151         /* All parsed OK. */
 152         return 0;
 153 }
 154
 155 /* Lazy bastard, eh? */
 156 #define STANDARD_PARAM_DEF(name, type, format, tmptype, strtolfn)       \
 157         int param_set_##name(const char *val, struct kernel_param *kp)  \
 158         {                                                               \
 159                 char *endp;                                             \
 160                 tmptype l;                                              \
 161                                                                         \
 162                 if (!val) return -EINVAL;                               \
 163                 l = strtolfn(val, &endp, 0);                            \
 164                 if (endp == val || ((type)l != l))                      \
 165                         return -EINVAL;                                 \
 166                 *((type *)kp->arg) = l;                                 \
 167                 return 0;                                               \
 168         }                                                               \
 169         int param_get_##name(char *buffer, struct kernel_param *kp)     \
 170         {                                                               \
 171                 return sprintf(buffer, format, *((type *)kp->arg));     \
 172         }
 173
 174 STANDARD_PARAM_DEF(short, short, "%hi", long, simple_strtol);
 175 STANDARD_PARAM_DEF(ushort, unsigned short, "%hu", unsigned long, simple_strtoul);
 176 STANDARD_PARAM_DEF(int, int, "%i", long, simple_strtol);
 177 STANDARD_PARAM_DEF(uint, unsigned int, "%u", unsigned long, simple_strtoul);
 178 STANDARD_PARAM_DEF(long, long, "%li", long, simple_strtol);
 179 STANDARD_PARAM_DEF(ulong, unsigned long, "%lu", unsigned long, simple_strtoul);
 180
 181 int param_set_charp(const char *val, struct kernel_param *kp)
 182 {
 183         if (!val) {
 184                 printk(KERN_ERR "%s: string parameter expected\n",
 185                        kp->name);
 186                 return -EINVAL;
 187         }
 188
 189         if (strlen(val) > 1024) {
 190                 printk(KERN_ERR "%s: string parameter too long\n",
 191                        kp->name);
 192                 return -ENOSPC;
 193         }
 194
 195         *(char **)kp->arg = (char *)val;
 196         return 0;
 197 }
 198
 199 int param_get_charp(char *buffer, struct kernel_param *kp)
 200 {
 201         return sprintf(buffer, "%s", *((char **)kp->arg));
 202 }
 203
 204 int param_set_bool(const char *val, struct kernel_param *kp)
 205 {
 206         /* No equals means "set"... */
 207         if (!val) val = "1";
 208
 209         /* One of =[yYnN01] */
 210         switch (val[0]) {
 211         case 'y': case 'Y': case '1':
 212                 *(int *)kp->arg = 1;
 213                 return 0;
 214         case 'n': case 'N': case '0':
 215                 *(int *)kp->arg = 0;
 216                 return 0;
 217         }
 218         return -EINVAL;
 219 }
 220
 221 int param_get_bool(char *buffer, struct kernel_param *kp)
 222 {
 223         /* Y and N chosen as being relatively non-coder friendly */
 224         return sprintf(buffer, "%c", (*(int *)kp->arg) ? 'Y' : 'N');
 225 }
 226
 227 int param_set_invbool(const char *val, struct kernel_param *kp)
 228 {
 229         int boolval, ret;
 230         struct kernel_param dummy = { .arg = &boolval };
 231
 232         ret = param_set_bool(val, &dummy);
 233         if (ret == 0)
 234                 *(int *)kp->arg = !boolval;
 235         return ret;
 236 }
 237
 238 int param_get_invbool(char *buffer, struct kernel_param *kp)
 239 {
 240         int val;
 241         struct kernel_param dummy = { .arg = &val };
 242
 243         val = !*(int *)kp->arg;
 244         return param_get_bool(buffer, &dummy);
 245 }
 246
 247 /* We cheat here and temporarily mangle the string. */
 248 int param_array(const char *name,
 249                 const char *val,
 250                 unsigned int min, unsigned int max,
 251                 void *elem, int elemsize,
 252                 int (*set)(const char *, struct kernel_param *kp),
 253                 int *num)
 254 {
 255         int ret;
 256         struct kernel_param kp;
 257         char save;
 258
 259         /* Get the name right for errors. */
 260         kp.name = name;
 261         kp.arg = elem;
 262
 263         /* No equals sign? */
 264         if (!val) {
 265                 printk(KERN_ERR "%s: expects arguments\n", name);
 266                 return -EINVAL;
 267         }
 268
 269         *num = 0;
 270         /* We expect a comma-separated list of values. */
 271         do {
 272                 int len;
 273
 274                 if (*num == max) {
 275                         printk(KERN_ERR "%s: can only take %i arguments\n",
 276                                name, max);
 277                         return -EINVAL;
 278                 }
 279                 len = strcspn(val, ",");
 280
 281                 /* nul-terminate and parse */
 282                 save = val[len];
 283                 ((char *)val)[len] = '\0';
 284                 ret = set(val, &kp);
 285
 286                 if (ret != 0)
 287                         return ret;
 288                 kp.arg += elemsize;
 289                 val += len+1;
 290                 (*num)++;
 291         } while (save == ',');
 292
 293         if (*num < min) {
 294                 printk(KERN_ERR "%s: needs at least %i arguments\n",
 295                        name, min);
 296                 return -EINVAL;
 297         }
 298         return 0;
 299 }
 300
 301 int param_array_set(const char *val, struct kernel_param *kp)
 302 {
 303         struct kparam_array *arr = kp->arg;
 304
 305         return param_array(kp->name, val, 1, arr->max, arr->elem,
 306                            arr->elemsize, arr->set, arr->num);
 307 }
 308
 309 int param_array_get(char *buffer, struct kernel_param *kp)
 310 {
 311         int i, off, ret;
 312         struct kparam_array *arr = kp->arg;
 313         struct kernel_param p;
 314
 315         p = *kp;
 316         for (i = off = 0; i < *arr->num; i++) {
 317                 if (i)
 318                         buffer[off++] = ',';
 319                 p.arg = arr->elem + arr->elemsize * i;
 320                 ret = arr->get(buffer + off, &p);
 321                 if (ret < 0)
 322                         return ret;
 323                 off += ret;
 324         }
 325         buffer[off] = '\0';
 326         return off;
 327 }
 328
 329 int param_set_copystring(const char *val, struct kernel_param *kp)
 330 {
 331         struct kparam_string *kps = kp->arg;
 332
 333         if (strlen(val)+1 > kps->maxlen) {
 334                 printk(KERN_ERR "%s: string doesn't fit in %u chars.\n",
 335                        kp->name, kps->maxlen-1);
 336                 return -ENOSPC;
 337         }
 338         strcpy(kps->string, val);
 339         return 0;
 340 }
 341
 342 EXPORT_SYMBOL(param_set_short);
 343 EXPORT_SYMBOL(param_get_short);
 344 EXPORT_SYMBOL(param_set_ushort);
 345 EXPORT_SYMBOL(param_get_ushort);
 346 EXPORT_SYMBOL(param_set_int);
 347 EXPORT_SYMBOL(param_get_int);
 348 EXPORT_SYMBOL(param_set_uint);
 349 EXPORT_SYMBOL(param_get_uint);
 350 EXPORT_SYMBOL(param_set_long);
 351 EXPORT_SYMBOL(param_get_long);
 352 EXPORT_SYMBOL(param_set_ulong);
 353 EXPORT_SYMBOL(param_get_ulong);
 354 EXPORT_SYMBOL(param_set_charp);
 355 EXPORT_SYMBOL(param_get_charp);
 356 EXPORT_SYMBOL(param_set_bool);
 357 EXPORT_SYMBOL(param_get_bool);
 358 EXPORT_SYMBOL(param_set_invbool);
 359 EXPORT_SYMBOL(param_get_invbool);
 360 EXPORT_SYMBOL(param_array_set);
 361 EXPORT_SYMBOL(param_array_get);
 362 EXPORT_SYMBOL(param_set_copystring);