diff options
author | Drew DeVault <sir@cmpwn.com> | 2015-11-26 14:31:29 -0500 |
---|---|---|
committer | Drew DeVault <sir@cmpwn.com> | 2015-11-26 14:31:29 -0500 |
commit | 9a15371ba3ab8497ac393d18baeb2f953a3b2762 (patch) | |
tree | f2f797205fd8816af277c3178204dc355cfd4a74 /common/stringop.c | |
parent | Add swaymsg subproject (diff) | |
download | sway-9a15371ba3ab8497ac393d18baeb2f953a3b2762.tar.gz sway-9a15371ba3ab8497ac393d18baeb2f953a3b2762.tar.zst sway-9a15371ba3ab8497ac393d18baeb2f953a3b2762.zip |
Parse command line args for swaymsg
Diffstat (limited to 'common/stringop.c')
-rw-r--r-- | common/stringop.c | 382 |
1 files changed, 382 insertions, 0 deletions
diff --git a/common/stringop.c b/common/stringop.c new file mode 100644 index 00000000..efa3a207 --- /dev/null +++ b/common/stringop.c | |||
@@ -0,0 +1,382 @@ | |||
1 | #include <stdlib.h> | ||
2 | #include <stdio.h> | ||
3 | #include <string.h> | ||
4 | #include <strings.h> | ||
5 | #include <ctype.h> | ||
6 | #include "stringop.h" | ||
7 | #include "log.h" | ||
8 | #include "string.h" | ||
9 | #include "list.h" | ||
10 | |||
11 | const char whitespace[] = " \f\n\r\t\v"; | ||
12 | |||
13 | /* Note: This returns 8 characters for trimmed_start per tab character. */ | ||
14 | char *strip_whitespace(char *_str) { | ||
15 | if (*_str == '\0') | ||
16 | return _str; | ||
17 | char *strold = _str; | ||
18 | while (*_str == ' ' || *_str == '\t') { | ||
19 | _str++; | ||
20 | } | ||
21 | char *str = strdup(_str); | ||
22 | free(strold); | ||
23 | int i; | ||
24 | for (i = 0; str[i] != '\0'; ++i); | ||
25 | do { | ||
26 | i--; | ||
27 | } while (i >= 0 && (str[i] == ' ' || str[i] == '\t')); | ||
28 | str[i + 1] = '\0'; | ||
29 | return str; | ||
30 | } | ||
31 | |||
32 | char *strip_comments(char *str) { | ||
33 | int in_string = 0, in_character = 0; | ||
34 | int i = 0; | ||
35 | while (str[i] != '\0') { | ||
36 | if (str[i] == '"' && !in_character) { | ||
37 | in_string = !in_string; | ||
38 | } else if (str[i] == '\'' && !in_string) { | ||
39 | in_character = !in_character; | ||
40 | } else if (!in_character && !in_string) { | ||
41 | if (str[i] == '#') { | ||
42 | str[i] = '\0'; | ||
43 | break; | ||
44 | } | ||
45 | } | ||
46 | ++i; | ||
47 | } | ||
48 | return str; | ||
49 | } | ||
50 | |||
51 | void strip_quotes(char *str) { | ||
52 | bool in_str = false; | ||
53 | bool in_chr = false; | ||
54 | bool escaped = false; | ||
55 | char *end = strchr(str,0); | ||
56 | while (*str) { | ||
57 | if (*str == '\'' && !in_str && !escaped) { | ||
58 | in_chr = !in_chr; | ||
59 | goto shift_over; | ||
60 | } else if (*str == '\"' && !in_chr && !escaped) { | ||
61 | in_str = !in_str; | ||
62 | goto shift_over; | ||
63 | } else if (*str == '\\') { | ||
64 | escaped = !escaped; | ||
65 | ++str; | ||
66 | continue; | ||
67 | } | ||
68 | escaped = false; | ||
69 | ++str; | ||
70 | continue; | ||
71 | shift_over: | ||
72 | memmove(str, str+1, end-- - str); | ||
73 | } | ||
74 | *end = '\0'; | ||
75 | } | ||
76 | |||
77 | // strcmp that also handles null pointers. | ||
78 | int lenient_strcmp(char *a, char *b) { | ||
79 | if (a == b) { | ||
80 | return 0; | ||
81 | } else if (!a) { | ||
82 | return -1; | ||
83 | } else if (!b) { | ||
84 | return 1; | ||
85 | } else { | ||
86 | return strcmp(a, b); | ||
87 | } | ||
88 | } | ||
89 | |||
90 | list_t *split_string(const char *str, const char *delims) { | ||
91 | list_t *res = create_list(); | ||
92 | char *copy = strdup(str); | ||
93 | char *token; | ||
94 | |||
95 | token = strtok(copy, delims); | ||
96 | while(token) { | ||
97 | token = strdup(token); | ||
98 | list_add(res, token); | ||
99 | token = strtok(NULL, delims); | ||
100 | } | ||
101 | free(copy); | ||
102 | return res; | ||
103 | } | ||
104 | |||
105 | void free_flat_list(list_t *list) { | ||
106 | int i; | ||
107 | for (i = 0; i < list->length; ++i) { | ||
108 | free(list->items[i]); | ||
109 | } | ||
110 | list_free(list); | ||
111 | } | ||
112 | |||
113 | char **split_args(const char *start, int *argc) { | ||
114 | *argc = 0; | ||
115 | int alloc = 2; | ||
116 | char **argv = malloc(sizeof(char *) * alloc); | ||
117 | bool in_token = false; | ||
118 | bool in_string = false; | ||
119 | bool in_char = false; | ||
120 | bool in_brackets = false; // brackets are used for critera | ||
121 | bool escaped = false; | ||
122 | const char *end = start; | ||
123 | if (start) { | ||
124 | while (*start) { | ||
125 | if (!in_token) { | ||
126 | start = (end += strspn(end, whitespace)); | ||
127 | in_token = true; | ||
128 | } | ||
129 | if (*end == '"' && !in_char && !escaped) { | ||
130 | in_string = !in_string; | ||
131 | } else if (*end == '\'' && !in_string && !escaped) { | ||
132 | in_char = !in_char; | ||
133 | } else if (*end == '[' && !in_string && !in_char && !in_brackets && !escaped) { | ||
134 | in_brackets = true; | ||
135 | } else if (*end == ']' && !in_string && !in_char && in_brackets && !escaped) { | ||
136 | in_brackets = false; | ||
137 | } else if (*end == '\\') { | ||
138 | escaped = !escaped; | ||
139 | } else if (*end == '\0' || (!in_string && !in_char && !in_brackets | ||
140 | && !escaped && strchr(whitespace, *end))) { | ||
141 | goto add_token; | ||
142 | } | ||
143 | if (*end != '\\') { | ||
144 | escaped = false; | ||
145 | } | ||
146 | ++end; | ||
147 | continue; | ||
148 | add_token: | ||
149 | if (end - start > 0) { | ||
150 | char *token = malloc(end - start + 1); | ||
151 | strncpy(token, start, end - start + 1); | ||
152 | token[end - start] = '\0'; | ||
153 | argv[*argc] = token; | ||
154 | if (++*argc + 1 == alloc) { | ||
155 | argv = realloc(argv, (alloc *= 2) * sizeof(char *)); | ||
156 | } | ||
157 | } | ||
158 | in_token = false; | ||
159 | escaped = false; | ||
160 | } | ||
161 | } | ||
162 | argv[*argc] = NULL; | ||
163 | return argv; | ||
164 | } | ||
165 | |||
166 | void free_argv(int argc, char **argv) { | ||
167 | while (--argc > 0) { | ||
168 | free(argv[argc]); | ||
169 | } | ||
170 | free(argv); | ||
171 | } | ||
172 | |||
173 | char *code_strstr(const char *haystack, const char *needle) { | ||
174 | /* TODO */ | ||
175 | return strstr(haystack, needle); | ||
176 | } | ||
177 | |||
178 | char *code_strchr(const char *str, char delimiter) { | ||
179 | int in_string = 0, in_character = 0; | ||
180 | int i = 0; | ||
181 | while (str[i] != '\0') { | ||
182 | if (str[i] == '"' && !in_character) { | ||
183 | in_string = !in_string; | ||
184 | } else if (str[i] == '\'' && !in_string) { | ||
185 | in_character = !in_character; | ||
186 | } else if (!in_character && !in_string) { | ||
187 | if (str[i] == delimiter) { | ||
188 | return (char *)str + i; | ||
189 | } | ||
190 | } | ||
191 | ++i; | ||
192 | } | ||
193 | return NULL; | ||
194 | } | ||
195 | |||
196 | int unescape_string(char *string) { | ||
197 | /* TODO: More C string escapes */ | ||
198 | int len = strlen(string); | ||
199 | int i; | ||
200 | for (i = 0; string[i]; ++i) { | ||
201 | if (string[i] == '\\') { | ||
202 | switch (string[++i]) { | ||
203 | case '0': | ||
204 | string[i - 1] = '\0'; | ||
205 | return i - 1; | ||
206 | case 'a': | ||
207 | string[i - 1] = '\a'; | ||
208 | string[i] = '\0'; | ||
209 | break; | ||
210 | case 'b': | ||
211 | string[i - 1] = '\b'; | ||
212 | string[i] = '\0'; | ||
213 | break; | ||
214 | case 'f': | ||
215 | string[i - 1] = '\f'; | ||
216 | string[i] = '\0'; | ||
217 | break; | ||
218 | case 'n': | ||
219 | string[i - 1] = '\n'; | ||
220 | string[i] = '\0'; | ||
221 | break; | ||
222 | case 'r': | ||
223 | string[i - 1] = '\r'; | ||
224 | string[i] = '\0'; | ||
225 | break; | ||
226 | case 't': | ||
227 | string[i - 1] = '\t'; | ||
228 | string[i] = '\0'; | ||
229 | break; | ||
230 | case 'v': | ||
231 | string[i - 1] = '\v'; | ||
232 | string[i] = '\0'; | ||
233 | break; | ||
234 | case '\\': | ||
235 | string[i] = '\0'; | ||
236 | break; | ||
237 | case '\'': | ||
238 | string[i - 1] = '\''; | ||
239 | string[i] = '\0'; | ||
240 | break; | ||
241 | case '\"': | ||
242 | string[i - 1] = '\"'; | ||
243 | string[i] = '\0'; | ||
244 | break; | ||
245 | case '?': | ||
246 | string[i - 1] = '?'; | ||
247 | string[i] = '\0'; | ||
248 | break; | ||
249 | case 'x': | ||
250 | { | ||
251 | unsigned char c = 0; | ||
252 | if (string[i+1] >= '0' && string[i+1] <= '9') { | ||
253 | c = string[i+1] - '0'; | ||
254 | if (string[i+2] >= '0' && string[i+2] <= '9') { | ||
255 | c *= 0x10; | ||
256 | c += string[i+2] - '0'; | ||
257 | string[i+2] = '\0'; | ||
258 | } | ||
259 | string[i+1] = '\0'; | ||
260 | } | ||
261 | string[i] = '\0'; | ||
262 | string[i - 1] = c; | ||
263 | } | ||
264 | } | ||
265 | } | ||
266 | } | ||
267 | // Shift characters over nullspaces | ||
268 | int shift = 0; | ||
269 | for (i = 0; i < len; ++i) { | ||
270 | if (string[i] == 0) { | ||
271 | shift++; | ||
272 | continue; | ||
273 | } | ||
274 | string[i-shift] = string[i]; | ||
275 | } | ||
276 | string[len - shift] = 0; | ||
277 | return len - shift; | ||
278 | } | ||
279 | |||
280 | char *join_args(char **argv, int argc) { | ||
281 | int len = 0, i; | ||
282 | for (i = 0; i < argc; ++i) { | ||
283 | len += strlen(argv[i]) + 1; | ||
284 | } | ||
285 | char *res = malloc(len); | ||
286 | len = 0; | ||
287 | for (i = 0; i < argc; ++i) { | ||
288 | strcpy(res + len, argv[i]); | ||
289 | len += strlen(argv[i]); | ||
290 | res[len++] = ' '; | ||
291 | } | ||
292 | res[len - 1] = '\0'; | ||
293 | return res; | ||
294 | } | ||
295 | |||
296 | /* | ||
297 | * Join a list of strings, adding separator in between. Separator can be NULL. | ||
298 | */ | ||
299 | char *join_list(list_t *list, char *separator) { | ||
300 | if (!sway_assert(list != NULL, "list != NULL") || list->length == 0) { | ||
301 | return NULL; | ||
302 | } | ||
303 | |||
304 | size_t len = 1; // NULL terminator | ||
305 | size_t sep_len = 0; | ||
306 | if (separator != NULL) { | ||
307 | sep_len = strlen(separator); | ||
308 | len += (list->length - 1) * sep_len; | ||
309 | } | ||
310 | |||
311 | for (int i = 0; i < list->length; i++) { | ||
312 | len += strlen(list->items[i]); | ||
313 | } | ||
314 | |||
315 | char *res = malloc(len); | ||
316 | |||
317 | char *p = res + strlen(list->items[0]); | ||
318 | strcpy(res, list->items[0]); | ||
319 | |||
320 | for (int i = 1; i < list->length; i++) { | ||
321 | if (sep_len) { | ||
322 | memcpy(p, separator, sep_len); | ||
323 | p += sep_len; | ||
324 | } | ||
325 | strcpy(p, list->items[i]); | ||
326 | p += strlen(list->items[i]); | ||
327 | } | ||
328 | |||
329 | *p = '\0'; | ||
330 | |||
331 | return res; | ||
332 | } | ||
333 | |||
334 | char *cmdsep(char **stringp, const char *delim) { | ||
335 | // skip over leading delims | ||
336 | char *head = *stringp + strspn(*stringp, delim); | ||
337 | // Find end token | ||
338 | char *tail = *stringp += strcspn(*stringp, delim); | ||
339 | // Set stringp to begining of next token | ||
340 | *stringp += strspn(*stringp, delim); | ||
341 | // Set stringp to null if last token | ||
342 | if (!**stringp) *stringp = NULL; | ||
343 | // Nullify end of first token | ||
344 | *tail = 0; | ||
345 | return head; | ||
346 | } | ||
347 | |||
348 | char *argsep(char **stringp, const char *delim) { | ||
349 | char *start = *stringp; | ||
350 | char *end = start; | ||
351 | bool in_string = false; | ||
352 | bool in_char = false; | ||
353 | bool escaped = false; | ||
354 | while (1) { | ||
355 | if (*end == '"' && !in_char && !escaped) { | ||
356 | in_string = !in_string; | ||
357 | } else if (*end == '\'' && !in_string && !escaped) { | ||
358 | in_char = !in_char; | ||
359 | } else if (*end == '\\') { | ||
360 | escaped = !escaped; | ||
361 | } else if (*end == '\0') { | ||
362 | *stringp = NULL; | ||
363 | goto found; | ||
364 | } else if (!in_string && !in_char && !escaped && strchr(delim, *end)) { | ||
365 | if (end - start) { | ||
366 | *(end++) = 0; | ||
367 | *stringp = end + strspn(end, delim);; | ||
368 | if (!**stringp) *stringp = NULL; | ||
369 | goto found; | ||
370 | } else { | ||
371 | ++start; | ||
372 | end = start; | ||
373 | } | ||
374 | } | ||
375 | if (*end != '\\') { | ||
376 | escaped = false; | ||
377 | } | ||
378 | ++end; | ||
379 | } | ||
380 | found: | ||
381 | return start; | ||
382 | } | ||