lib/argv_split.c

   1 // SPDX-License-Identifier: GPL-2.0
   2 /*
   3  * Helper function for splitting a string into an argv-like array.
   4  */
   5
   6 #include <linux/kernel.h>
   7 #include <linux/ctype.h>
   8 #include <linux/string.h>
   9 #include <linux/slab.h>
  10 #include <linux/export.h>
  11
  12 static int count_argc(const char *str)
  13 {
  14         int count = 0;
  15         bool was_space;
  16
  17         for (was_space = true; *str; str++) {
  18                 if (isspace(*str)) {
  19                         was_space = true;
  20                 } else if (was_space) {
  21                         was_space = false;
  22                         count++;
  23                 }
  24         }
  25
  26         return count;
  27 }
  28
  29 /**
  30  * argv_free - free an argv
  31  * @argv: the argument vector to be freed
  32  *
  33  * Frees an argv and the strings it points to.
  34  */
  35 void argv_free(char **argv)
  36 {
  37         argv--;
  38         kfree(argv[0]);
  39         kfree(argv);
  40 }
  41 EXPORT_SYMBOL(argv_free);
  42
  43 /**
  44  * argv_split - split a string at whitespace, returning an argv
  45  * @gfp: the GFP mask used to allocate memory
  46  * @str: the string to be split
  47  * @argcp: returned argument count
  48  *
  49  * Returns: an array of pointers to strings which are split out from
  50  * @str.  This is performed by strictly splitting on white-space; no
  51  * quote processing is performed.  Multiple whitespace characters are
  52  * considered to be a single argument separator.  The returned array
  53  * is always NULL-terminated.  Returns NULL on memory allocation
  54  * failure.
  55  *
  56  * The source string at `str' may be undergoing concurrent alteration via
  57  * userspace sysctl activity (at least).  The argv_split() implementation
  58  * attempts to handle this gracefully by taking a local copy to work on.
  59  */
  60 char **argv_split(gfp_t gfp, const char *str, int *argcp)
  61 {
  62         char *argv_str;
  63         bool was_space;
  64         char **argv, **argv_ret;
  65         int argc;
  66
  67         argv_str = kstrndup(str, KMALLOC_MAX_SIZE - 1, gfp);
  68         if (!argv_str)
  69                 return NULL;
  70
  71         argc = count_argc(argv_str);
  72         argv = kmalloc_array(argc + 2, sizeof(*argv), gfp);
  73         if (!argv) {
  74                 kfree(argv_str);
  75                 return NULL;
  76         }
  77
  78         *argv = argv_str;
  79         argv_ret = ++argv;
  80         for (was_space = true; *argv_str; argv_str++) {
  81                 if (isspace(*argv_str)) {
  82                         was_space = true;
  83                         *argv_str = 0;
  84                 } else if (was_space) {
  85                         was_space = false;
  86                         *argv++ = argv_str;
  87                 }
  88         }
  89         *argv = NULL;
  90
  91         if (argcp)
  92                 *argcp = argc;
  93         return argv_ret;
  94 }
  95 EXPORT_SYMBOL(argv_split);