GNU Linux-libre 4.19.211-gnu1
[releases.git] / tools / perf / util / string.c
1 // SPDX-License-Identifier: GPL-2.0
2 #include "string2.h"
3 #include <linux/kernel.h>
4 #include <linux/string.h>
5 #include <stdlib.h>
6
7 #include "sane_ctype.h"
8
9 #define K 1024LL
10 /*
11  * perf_atoll()
12  * Parse (\d+)(b|B|kb|KB|mb|MB|gb|GB|tb|TB) (e.g. "256MB")
13  * and return its numeric value
14  */
15 s64 perf_atoll(const char *str)
16 {
17         s64 length;
18         char *p;
19         char c;
20
21         if (!isdigit(str[0]))
22                 goto out_err;
23
24         length = strtoll(str, &p, 10);
25         switch (c = *p++) {
26                 case 'b': case 'B':
27                         if (*p)
28                                 goto out_err;
29
30                         __fallthrough;
31                 case '\0':
32                         return length;
33                 default:
34                         goto out_err;
35                 /* two-letter suffices */
36                 case 'k': case 'K':
37                         length <<= 10;
38                         break;
39                 case 'm': case 'M':
40                         length <<= 20;
41                         break;
42                 case 'g': case 'G':
43                         length <<= 30;
44                         break;
45                 case 't': case 'T':
46                         length <<= 40;
47                         break;
48         }
49         /* we want the cases to match */
50         if (islower(c)) {
51                 if (strcmp(p, "b") != 0)
52                         goto out_err;
53         } else {
54                 if (strcmp(p, "B") != 0)
55                         goto out_err;
56         }
57         return length;
58
59 out_err:
60         return -1;
61 }
62
63 /*
64  * Helper function for splitting a string into an argv-like array.
65  * originally copied from lib/argv_split.c
66  */
67 static const char *skip_sep(const char *cp)
68 {
69         while (*cp && isspace(*cp))
70                 cp++;
71
72         return cp;
73 }
74
75 static const char *skip_arg(const char *cp)
76 {
77         while (*cp && !isspace(*cp))
78                 cp++;
79
80         return cp;
81 }
82
83 static int count_argc(const char *str)
84 {
85         int count = 0;
86
87         while (*str) {
88                 str = skip_sep(str);
89                 if (*str) {
90                         count++;
91                         str = skip_arg(str);
92                 }
93         }
94
95         return count;
96 }
97
98 /**
99  * argv_free - free an argv
100  * @argv - the argument vector to be freed
101  *
102  * Frees an argv and the strings it points to.
103  */
104 void argv_free(char **argv)
105 {
106         char **p;
107         for (p = argv; *p; p++) {
108                 free(*p);
109                 *p = NULL;
110         }
111
112         free(argv);
113 }
114
115 /**
116  * argv_split - split a string at whitespace, returning an argv
117  * @str: the string to be split
118  * @argcp: returned argument count
119  *
120  * Returns an array of pointers to strings which are split out from
121  * @str.  This is performed by strictly splitting on white-space; no
122  * quote processing is performed.  Multiple whitespace characters are
123  * considered to be a single argument separator.  The returned array
124  * is always NULL-terminated.  Returns NULL on memory allocation
125  * failure.
126  */
127 char **argv_split(const char *str, int *argcp)
128 {
129         int argc = count_argc(str);
130         char **argv = calloc(argc + 1, sizeof(*argv));
131         char **argvp;
132
133         if (argv == NULL)
134                 goto out;
135
136         if (argcp)
137                 *argcp = argc;
138
139         argvp = argv;
140
141         while (*str) {
142                 str = skip_sep(str);
143
144                 if (*str) {
145                         const char *p = str;
146                         char *t;
147
148                         str = skip_arg(str);
149
150                         t = strndup(p, str-p);
151                         if (t == NULL)
152                                 goto fail;
153                         *argvp++ = t;
154                 }
155         }
156         *argvp = NULL;
157
158 out:
159         return argv;
160
161 fail:
162         argv_free(argv);
163         return NULL;
164 }
165
166 /* Character class matching */
167 static bool __match_charclass(const char *pat, char c, const char **npat)
168 {
169         bool complement = false, ret = true;
170
171         if (*pat == '!') {
172                 complement = true;
173                 pat++;
174         }
175         if (*pat++ == c)        /* First character is special */
176                 goto end;
177
178         while (*pat && *pat != ']') {   /* Matching */
179                 if (*pat == '-' && *(pat + 1) != ']') { /* Range */
180                         if (*(pat - 1) <= c && c <= *(pat + 1))
181                                 goto end;
182                         if (*(pat - 1) > *(pat + 1))
183                                 goto error;
184                         pat += 2;
185                 } else if (*pat++ == c)
186                         goto end;
187         }
188         if (!*pat)
189                 goto error;
190         ret = false;
191
192 end:
193         while (*pat && *pat != ']')     /* Searching closing */
194                 pat++;
195         if (!*pat)
196                 goto error;
197         *npat = pat + 1;
198         return complement ? !ret : ret;
199
200 error:
201         return false;
202 }
203
204 /* Glob/lazy pattern matching */
205 static bool __match_glob(const char *str, const char *pat, bool ignore_space,
206                         bool case_ins)
207 {
208         while (*str && *pat && *pat != '*') {
209                 if (ignore_space) {
210                         /* Ignore spaces for lazy matching */
211                         if (isspace(*str)) {
212                                 str++;
213                                 continue;
214                         }
215                         if (isspace(*pat)) {
216                                 pat++;
217                                 continue;
218                         }
219                 }
220                 if (*pat == '?') {      /* Matches any single character */
221                         str++;
222                         pat++;
223                         continue;
224                 } else if (*pat == '[') /* Character classes/Ranges */
225                         if (__match_charclass(pat + 1, *str, &pat)) {
226                                 str++;
227                                 continue;
228                         } else
229                                 return false;
230                 else if (*pat == '\\') /* Escaped char match as normal char */
231                         pat++;
232                 if (case_ins) {
233                         if (tolower(*str) != tolower(*pat))
234                                 return false;
235                 } else if (*str != *pat)
236                         return false;
237                 str++;
238                 pat++;
239         }
240         /* Check wild card */
241         if (*pat == '*') {
242                 while (*pat == '*')
243                         pat++;
244                 if (!*pat)      /* Tail wild card matches all */
245                         return true;
246                 while (*str)
247                         if (__match_glob(str++, pat, ignore_space, case_ins))
248                                 return true;
249         }
250         return !*str && !*pat;
251 }
252
253 /**
254  * strglobmatch - glob expression pattern matching
255  * @str: the target string to match
256  * @pat: the pattern string to match
257  *
258  * This returns true if the @str matches @pat. @pat can includes wildcards
259  * ('*','?') and character classes ([CHARS], complementation and ranges are
260  * also supported). Also, this supports escape character ('\') to use special
261  * characters as normal character.
262  *
263  * Note: if @pat syntax is broken, this always returns false.
264  */
265 bool strglobmatch(const char *str, const char *pat)
266 {
267         return __match_glob(str, pat, false, false);
268 }
269
270 bool strglobmatch_nocase(const char *str, const char *pat)
271 {
272         return __match_glob(str, pat, false, true);
273 }
274
275 /**
276  * strlazymatch - matching pattern strings lazily with glob pattern
277  * @str: the target string to match
278  * @pat: the pattern string to match
279  *
280  * This is similar to strglobmatch, except this ignores spaces in
281  * the target string.
282  */
283 bool strlazymatch(const char *str, const char *pat)
284 {
285         return __match_glob(str, pat, true, false);
286 }
287
288 /**
289  * strtailcmp - Compare the tail of two strings
290  * @s1: 1st string to be compared
291  * @s2: 2nd string to be compared
292  *
293  * Return 0 if whole of either string is same as another's tail part.
294  */
295 int strtailcmp(const char *s1, const char *s2)
296 {
297         int i1 = strlen(s1);
298         int i2 = strlen(s2);
299         while (--i1 >= 0 && --i2 >= 0) {
300                 if (s1[i1] != s2[i2])
301                         return s1[i1] - s2[i2];
302         }
303         return 0;
304 }
305
306 /**
307  * strxfrchar - Locate and replace character in @s
308  * @s:    The string to be searched/changed.
309  * @from: Source character to be replaced.
310  * @to:   Destination character.
311  *
312  * Return pointer to the changed string.
313  */
314 char *strxfrchar(char *s, char from, char to)
315 {
316         char *p = s;
317
318         while ((p = strchr(p, from)) != NULL)
319                 *p++ = to;
320
321         return s;
322 }
323
324 /**
325  * ltrim - Removes leading whitespace from @s.
326  * @s: The string to be stripped.
327  *
328  * Return pointer to the first non-whitespace character in @s.
329  */
330 char *ltrim(char *s)
331 {
332         while (isspace(*s))
333                 s++;
334
335         return s;
336 }
337
338 /**
339  * rtrim - Removes trailing whitespace from @s.
340  * @s: The string to be stripped.
341  *
342  * Note that the first trailing whitespace is replaced with a %NUL-terminator
343  * in the given string @s. Returns @s.
344  */
345 char *rtrim(char *s)
346 {
347         size_t size = strlen(s);
348         char *end;
349
350         if (!size)
351                 return s;
352
353         end = s + size - 1;
354         while (end >= s && isspace(*end))
355                 end--;
356         *(end + 1) = '\0';
357
358         return s;
359 }
360
361 char *asprintf_expr_inout_ints(const char *var, bool in, size_t nints, int *ints)
362 {
363         /*
364          * FIXME: replace this with an expression using log10() when we
365          * find a suitable implementation, maybe the one in the dvb drivers...
366          *
367          * "%s == %d || " = log10(MAXINT) * 2 + 8 chars for the operators
368          */
369         size_t size = nints * 28 + 1; /* \0 */
370         size_t i, printed = 0;
371         char *expr = malloc(size);
372
373         if (expr) {
374                 const char *or_and = "||", *eq_neq = "==";
375                 char *e = expr;
376
377                 if (!in) {
378                         or_and = "&&";
379                         eq_neq = "!=";
380                 }
381
382                 for (i = 0; i < nints; ++i) {
383                         if (printed == size)
384                                 goto out_err_overflow;
385
386                         if (i > 0)
387                                 printed += scnprintf(e + printed, size - printed, " %s ", or_and);
388                         printed += scnprintf(e + printed, size - printed,
389                                              "%s %s %d", var, eq_neq, ints[i]);
390                 }
391         }
392
393         return expr;
394
395 out_err_overflow:
396         free(expr);
397         return NULL;
398 }
399
400 /* Like strpbrk(), but not break if it is right after a backslash (escaped) */
401 char *strpbrk_esc(char *str, const char *stopset)
402 {
403         char *ptr;
404
405         do {
406                 ptr = strpbrk(str, stopset);
407                 if (ptr == str ||
408                     (ptr == str + 1 && *(ptr - 1) != '\\'))
409                         break;
410                 str = ptr + 1;
411         } while (ptr && *(ptr - 1) == '\\' && *(ptr - 2) != '\\');
412
413         return ptr;
414 }
415
416 /* Like strdup, but do not copy a single backslash */
417 char *strdup_esc(const char *str)
418 {
419         char *s, *d, *p, *ret = strdup(str);
420
421         if (!ret)
422                 return NULL;
423
424         d = strchr(ret, '\\');
425         if (!d)
426                 return ret;
427
428         s = d + 1;
429         do {
430                 if (*s == '\0') {
431                         *d = '\0';
432                         break;
433                 }
434                 p = strchr(s + 1, '\\');
435                 if (p) {
436                         memmove(d, s, p - s);
437                         d += p - s;
438                         s = p + 1;
439                 } else
440                         memmove(d, s, strlen(s) + 1);
441         } while (p);
442
443         return ret;
444 }