hc
2024-03-22 a0752693d998599af469473b8dc239ef973a012f
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
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
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
// SPDX-License-Identifier: GPL-2.0
/*
 * Helper function for splitting a string into an argv-like array.
 */
 
#include <stdlib.h>
#include <linux/kernel.h>
#include <linux/ctype.h>
#include <linux/string.h>
 
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) {
       str = skip_spaces(str);
       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++) {
       free(*p);
       *p = NULL;
   }
 
   free(argv);
}
 
/**
 * argv_split - split a string at whitespace, returning an argv
 * @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(const char *str, int *argcp)
{
   int argc = count_argc(str);
   char **argv = calloc(argc + 1, sizeof(*argv));
   char **argvp;
 
   if (argv == NULL)
       goto out;
 
   if (argcp)
       *argcp = argc;
 
   argvp = argv;
 
   while (*str) {
       str = skip_spaces(str);
 
       if (*str) {
           const char *p = str;
           char *t;
 
           str = skip_arg(str);
 
           t = strndup(p, str-p);
           if (t == NULL)
               goto fail;
           *argvp++ = t;
       }
   }
   *argvp = NULL;
 
out:
   return argv;
 
fail:
   argv_free(argv);
   return NULL;
}