Blame view

lib/argv_split.c 1.78 KB
d84d1cc76   Jeremy Fitzhardinge   add argv_split()
1
2
3
4
5
6
  /*
   * Helper function for splitting a string into an argv-like array.
   */
  
  #include <linux/kernel.h>
  #include <linux/ctype.h>
e7d2860b6   André Goddard Rosa   tree-wide: conver...
7
  #include <linux/string.h>
5a56db1c0   Robert P. J. Day   LIB: Replace inap...
8
9
  #include <linux/slab.h>
  #include <linux/module.h>
d84d1cc76   Jeremy Fitzhardinge   add argv_split()
10

d84d1cc76   Jeremy Fitzhardinge   add argv_split()
11
12
13
14
15
16
17
18
19
20
21
22
23
  static const char *skip_arg(const char *cp)
  {
  	while (*cp && !isspace(*cp))
  		cp++;
  
  	return cp;
  }
  
  static int count_argc(const char *str)
  {
  	int count = 0;
  
  	while (*str) {
e7d2860b6   André Goddard Rosa   tree-wide: conver...
24
  		str = skip_spaces(str);
d84d1cc76   Jeremy Fitzhardinge   add argv_split()
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
  		if (*str) {
  			count++;
  			str = skip_arg(str);
  		}
  	}
  
  	return count;
  }
  
  /**
   * argv_free - free an argv
   * @argv - the argument vector to be freed
   *
   * Frees an argv and the strings it points to.
   */
  void argv_free(char **argv)
  {
  	char **p;
  	for (p = argv; *p; p++)
  		kfree(*p);
  
  	kfree(argv);
  }
  EXPORT_SYMBOL(argv_free);
  
  /**
   * argv_split - split a string at whitespace, returning an argv
   * @gfp: the GFP mask used to allocate memory
   * @str: the string to be split
   * @argcp: returned argument count
   *
   * Returns an array of pointers to strings which are split out from
   * @str.  This is performed by strictly splitting on white-space; no
   * quote processing is performed.  Multiple whitespace characters are
   * considered to be a single argument separator.  The returned array
   * is always NULL-terminated.  Returns NULL on memory allocation
   * failure.
   */
  char **argv_split(gfp_t gfp, const char *str, int *argcp)
  {
  	int argc = count_argc(str);
  	char **argv = kzalloc(sizeof(*argv) * (argc+1), gfp);
  	char **argvp;
  
  	if (argv == NULL)
  		goto out;
8e2b70564   Neil Horman   argv_split: allow...
71
72
  	if (argcp)
  		*argcp = argc;
d84d1cc76   Jeremy Fitzhardinge   add argv_split()
73
74
75
  	argvp = argv;
  
  	while (*str) {
e7d2860b6   André Goddard Rosa   tree-wide: conver...
76
  		str = skip_spaces(str);
d84d1cc76   Jeremy Fitzhardinge   add argv_split()
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
  
  		if (*str) {
  			const char *p = str;
  			char *t;
  
  			str = skip_arg(str);
  
  			t = kstrndup(p, str-p, gfp);
  			if (t == NULL)
  				goto fail;
  			*argvp++ = t;
  		}
  	}
  	*argvp = NULL;
  
    out:
  	return argv;
  
    fail:
  	argv_free(argv);
  	return NULL;
  }
  EXPORT_SYMBOL(argv_split);