swayfx/common/stringop.c

331 lines
6.9 KiB
C
Raw Normal View History

#define _POSIX_C_SOURCE 200809L
2019-01-18 03:57:34 +11:00
#include <ctype.h>
#include <stdbool.h>
#include <stdio.h>
2019-01-18 03:57:34 +11:00
#include <stdlib.h>
2015-09-19 00:23:04 +10:00
#include <string.h>
2015-09-08 07:29:40 +10:00
#include <strings.h>
#include <wordexp.h>
#include "list.h"
2019-01-18 03:57:34 +11:00
#include "log.h"
#include "stringop.h"
2015-09-08 07:29:40 +10:00
2018-12-09 22:52:55 +11:00
static const char whitespace[] = " \f\n\r\t\v";
2018-12-09 22:52:55 +11:00
void strip_whitespace(char *str) {
size_t len = strlen(str);
size_t start = strspn(str, whitespace);
memmove(str, &str[start], len + 1 - start);
if (*str) {
for (len -= start + 1; isspace(str[len]); --len) {}
str[len + 1] = '\0';
}
}
2015-09-08 07:29:40 +10:00
void strip_quotes(char *str) {
bool in_str = false;
bool in_chr = false;
bool escaped = false;
char *end = strchr(str,0);
while (*str) {
if (*str == '\'' && !in_str && !escaped) {
in_chr = !in_chr;
goto shift_over;
} else if (*str == '\"' && !in_chr && !escaped) {
in_str = !in_str;
goto shift_over;
} else if (*str == '\\') {
escaped = !escaped;
++str;
continue;
}
escaped = false;
++str;
continue;
shift_over:
memmove(str, str+1, end-- - str);
}
*end = '\0';
}
char *lenient_strcat(char *dest, const char *src) {
if (dest && src) {
return strcat(dest, src);
}
return dest;
}
char *lenient_strncat(char *dest, const char *src, size_t len) {
if (dest && src) {
return strncat(dest, src, len);
}
return dest;
}
2015-11-19 21:52:58 +11:00
// strcmp that also handles null pointers.
int lenient_strcmp(const char *a, const char *b) {
2015-11-19 21:52:58 +11:00
if (a == b) {
return 0;
} else if (!a) {
return -1;
} else if (!b) {
return 1;
} else {
return strcmp(a, b);
}
}
list_t *split_string(const char *str, const char *delims) {
list_t *res = create_list();
2015-09-08 08:03:04 +10:00
char *copy = strdup(str);
2015-09-08 07:29:40 +10:00
2019-01-18 03:56:04 +11:00
char *token = strtok(copy, delims);
while (token) {
list_add(res, strdup(token));
2015-09-08 07:29:40 +10:00
token = strtok(NULL, delims);
}
2015-09-08 07:29:40 +10:00
free(copy);
return res;
}
2015-09-08 07:29:40 +10:00
char **split_args(const char *start, int *argc) {
*argc = 0;
int alloc = 2;
2015-09-09 03:53:15 +10:00
char **argv = malloc(sizeof(char *) * alloc);
2015-09-08 07:29:40 +10:00
bool in_token = false;
bool in_string = false;
bool in_char = false;
2019-11-23 21:38:15 +11:00
bool in_brackets = false; // brackets are used for criteria
2015-09-08 07:29:40 +10:00
bool escaped = false;
const char *end = start;
2015-09-15 12:59:25 +10:00
if (start) {
while (*start) {
if (!in_token) {
start = (end += strspn(end, whitespace));
in_token = true;
}
if (*end == '"' && !in_char && !escaped) {
in_string = !in_string;
} else if (*end == '\'' && !in_string && !escaped) {
in_char = !in_char;
} else if (*end == '[' && !in_string && !in_char && !in_brackets && !escaped) {
in_brackets = true;
} else if (*end == ']' && !in_string && !in_char && in_brackets && !escaped) {
in_brackets = false;
2015-09-15 12:59:25 +10:00
} else if (*end == '\\') {
escaped = !escaped;
} else if (*end == '\0' || (!in_string && !in_char && !in_brackets
&& !escaped && strchr(whitespace, *end))) {
2015-09-15 12:59:25 +10:00
goto add_token;
}
if (*end != '\\') {
escaped = false;
}
++end;
continue;
add_token:
if (end - start > 0) {
char *token = malloc(end - start + 1);
strncpy(token, start, end - start + 1);
token[end - start] = '\0';
argv[*argc] = token;
if (++*argc + 1 == alloc) {
argv = realloc(argv, (alloc *= 2) * sizeof(char *));
}
2015-09-08 07:29:40 +10:00
}
2015-09-15 12:59:25 +10:00
in_token = false;
escaped = false;
2015-09-08 07:29:40 +10:00
}
}
2015-09-09 03:53:15 +10:00
argv[*argc] = NULL;
return argv;
2015-09-08 07:29:40 +10:00
}
void free_argv(int argc, char **argv) {
while (argc-- > 0) {
2015-09-08 07:29:40 +10:00
free(argv[argc]);
}
free(argv);
}
int unescape_string(char *string) {
/* TODO: More C string escapes */
int len = strlen(string);
int i;
for (i = 0; string[i]; ++i) {
if (string[i] == '\\') {
switch (string[++i]) {
case '0':
string[i - 1] = '\0';
2015-09-09 03:27:09 +10:00
return i - 1;
case 'a':
string[i - 1] = '\a';
2015-09-09 03:27:09 +10:00
string[i] = '\0';
break;
case 'b':
string[i - 1] = '\b';
2015-09-09 03:27:09 +10:00
string[i] = '\0';
break;
case 'f':
string[i - 1] = '\f';
2015-09-09 03:27:09 +10:00
string[i] = '\0';
break;
case 'n':
string[i - 1] = '\n';
2015-09-09 03:27:09 +10:00
string[i] = '\0';
break;
case 'r':
string[i - 1] = '\r';
2015-09-09 03:27:09 +10:00
string[i] = '\0';
break;
case 't':
string[i - 1] = '\t';
2015-09-09 03:27:09 +10:00
string[i] = '\0';
break;
case 'v':
string[i - 1] = '\v';
2015-09-09 03:27:09 +10:00
string[i] = '\0';
break;
case '\\':
2015-09-09 03:27:09 +10:00
string[i] = '\0';
break;
case '\'':
string[i - 1] = '\'';
2015-09-09 03:27:09 +10:00
string[i] = '\0';
break;
case '\"':
string[i - 1] = '\"';
2015-09-09 03:27:09 +10:00
string[i] = '\0';
break;
case '?':
string[i - 1] = '?';
2015-09-09 03:27:09 +10:00
string[i] = '\0';
break;
case 'x':
{
unsigned char c = 0;
if (string[i+1] >= '0' && string[i+1] <= '9') {
c = string[i+1] - '0';
if (string[i+2] >= '0' && string[i+2] <= '9') {
c *= 0x10;
c += string[i+2] - '0';
2015-09-09 03:27:09 +10:00
string[i+2] = '\0';
}
2015-09-09 03:27:09 +10:00
string[i+1] = '\0';
}
2015-09-09 03:27:09 +10:00
string[i] = '\0';
string[i - 1] = c;
}
}
}
}
2015-09-09 03:27:09 +10:00
// Shift characters over nullspaces
int shift = 0;
for (i = 0; i < len; ++i) {
if (string[i] == 0) {
shift++;
continue;
}
string[i-shift] = string[i];
}
string[len - shift] = 0;
return len - shift;
}
2015-08-06 12:40:38 +10:00
char *join_args(char **argv, int argc) {
if (!sway_assert(argc > 0, "argc should be positive")) {
return NULL;
}
2015-08-06 12:40:38 +10:00
int len = 0, i;
for (i = 0; i < argc; ++i) {
len += strlen(argv[i]) + 1;
}
char *res = malloc(len);
len = 0;
for (i = 0; i < argc; ++i) {
strcpy(res + len, argv[i]);
len += strlen(argv[i]);
res[len++] = ' ';
}
res[len - 1] = '\0';
return res;
}
static inline char *argsep_next_interesting(const char *src, const char *delim) {
char *special = strpbrk(src, "\"'\\");
char *next_delim = strpbrk(src, delim);
if (!special) {
return next_delim;
}
if (!next_delim) {
return special;
}
return (next_delim < special) ? next_delim : special;
}
char *argsep(char **stringp, const char *delim, char *matched) {
2015-09-15 12:59:25 +10:00
char *start = *stringp;
char *end = start;
bool in_string = false;
bool in_char = false;
bool escaped = false;
char *interesting = NULL;
while ((interesting = argsep_next_interesting(end, delim))) {
if (escaped && interesting != end) {
escaped = false;
}
if (*interesting == '"' && !in_char && !escaped) {
2015-09-15 12:59:25 +10:00
in_string = !in_string;
end = interesting + 1;
} else if (*interesting == '\'' && !in_string && !escaped) {
2015-09-15 12:59:25 +10:00
in_char = !in_char;
end = interesting + 1;
} else if (*interesting == '\\') {
2015-09-15 12:59:25 +10:00
escaped = !escaped;
end = interesting + 1;
} else if (!in_string && !in_char && !escaped) {
// We must have matched a separator
end = interesting;
if (matched) {
*matched = *end;
}
2015-09-15 12:59:25 +10:00
if (end - start) {
*(end++) = 0;
*stringp = end;
2019-01-18 03:56:04 +11:00
break;
2015-09-15 12:59:25 +10:00
} else {
end = ++start;
2015-09-15 12:59:25 +10:00
}
} else {
end++;
2015-09-15 12:59:25 +10:00
}
}
if (!interesting) {
*stringp = NULL;
if (matched) {
*matched = '\0';
2015-09-15 12:59:25 +10:00
}
}
return start;
}
bool expand_path(char **path) {
wordexp_t p = {0};
while (strstr(*path, " ")) {
*path = realloc(*path, strlen(*path) + 2);
char *ptr = strstr(*path, " ") + 1;
memmove(ptr + 1, ptr, strlen(ptr) + 1);
*ptr = '\\';
}
if (wordexp(*path, &p, 0) != 0 || p.we_wordv[0] == NULL) {
wordfree(&p);
return false;
}
free(*path);
*path = join_args(p.we_wordv, p.we_wordc);
wordfree(&p);
return true;
}