1 files changed, 79 insertions, 32 deletions
diff --git a/sway/stringop.c b/sway/stringop.c
index 191e40c8..90f963d6 100644
--- a/sway/stringop.c
+++ b/sway/stringop.c
@@ -105,40 +105,40 @@ char **split_args(const char *start, int *argc) {
 	bool in_char = false;
 	bool escaped = false;
 	const char *end = 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 == '\\') {
-			escaped = !escaped;
-		} else if (*end == '\0' || (!in_string && !in_char && !escaped
-				&& strchr(whitespace, *end))) {
-			goto add_part;
-		}
-		if (*end != '\\') {
-			escaped = false;
-		}
-		++end;
-		continue;
-		add_part:
-		if (end - start > 0) {
-			char *token = malloc(end - start + 1);
-			strncpy(token, start, end - start + 1);
-			token[end - start] = '\0';
-			strip_quotes(token);
-			unescape_string(token);
-			argv[*argc] = token;
-			if (++*argc + 1 == alloc) {
-				argv = realloc(argv, (alloc *= 2) * sizeof(char *));
+	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 == '\\') {
+				escaped = !escaped;
+			} else if (*end == '\0' || (!in_string && !in_char && !escaped
+						&& strchr(whitespace, *end))) {
+				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 *));
+				}
 			}
+			in_token = false;
+			escaped = false;
 		}
-		in_token = false;
-		escaped = false;
 	}
 	argv[*argc] = NULL;
 	return argv;
@@ -311,3 +311,50 @@ char *join_list(list_t *list, char *separator) {
 
 	return res;
 }
+
+char *cmdsep(char **stringp, const char *delim) {
+	char *head = strsep(stringp, delim);
+	// But skip over trailing delims. '3   tokens  here' -> '3' 'tokens  here'
+	if (*stringp) {
+		*stringp += strspn(*stringp, delim);
+		// If skiping over delims brings us to the end of string, set to NULL
+		if (!**stringp) *stringp = NULL;
+	}
+	return head;
+}
+
+char *argsep(char **stringp, const char *delim) {
+	char *start = *stringp;
+	char *end = start;
+	bool in_string = false;
+	bool in_char = false;
+	bool escaped = false;
+	while (1) {
+		if (*end == '"' && !in_char && !escaped) {
+			in_string = !in_string;
+		} else if (*end == '\'' && !in_string && !escaped) {
+			in_char = !in_char;
+		} else if (*end == '\\') {
+			escaped = !escaped;
+		} else if (*end == '\0') {
+			*stringp = NULL;
+			goto found;
+		} else if (!in_string && !in_char && !escaped && strchr(delim, *end)) {
+			if (end - start) {
+				*(end++) = 0;
+				*stringp = end + strspn(end, delim);;
+				if (!**stringp) *stringp = NULL;
+				goto found;
+			} else {
+				++start;
+				end = start;
+			}
+		}
+		if (*end != '\\') {
+			escaped = false;
+		}
+		++end;
+	}
+	found:
+	return start;
+}