aboutsummaryrefslogtreecommitdiff
path: root/lib/src
diff options
context:
space:
mode:
Diffstat (limited to 'lib/src')
-rw-r--r--lib/src/split_argv.c109
1 files changed, 60 insertions, 49 deletions
diff --git a/lib/src/split_argv.c b/lib/src/split_argv.c
index 9beec5c..b95720d 100644
--- a/lib/src/split_argv.c
+++ b/lib/src/split_argv.c
@@ -16,78 +16,89 @@
* along with this program. If not, see <https://www.gnu.org/licenses/>.
*/
#include <stdlib.h>
+#include <string.h>
#include <ctype.h>
#include <errno.h>
+#include <stdio.h>
#include "util.h"
-char **split_argv(char *str)
+int pack_argv(char *str)
{
- size_t i = 0, cap = 0, new_cap;
- char **argv = NULL, **new;
- char *ptr;
+ char *dst, *start;
+ int count = 0;
- ptr = str;
+ dst = str;
for (;;) {
- if (*ptr == ' ') {
- ++ptr;
- continue;
- }
-
- if (i == cap) {
- new_cap = cap ? cap * 2 : 16;
- new = realloc(argv, sizeof(argv[0]) * new_cap);
-
- if (new == NULL) {
- free(argv);
- errno = ENOMEM;
- return NULL;
- }
-
- cap = new_cap;
- argv = new;
- }
+ while (*str == ' ')
+ ++str;
- if (*ptr == '\0') {
- argv[i++] = NULL;
+ if (*str == '\0')
break;
- }
-
- argv[i++] = ptr;
- if (*ptr == '"') {
- ++ptr;
- while (*ptr != '\0' && *ptr != '"') {
- if (ptr[0] == '\\' && ptr[1] != '\0')
- ++ptr;
+ if (*str == '"') {
+ start = dst;
+ *(dst++) = *(str++);
- ++ptr;
+ while (*str != '"') {
+ if (*str == '\0')
+ goto fail_str;
+ if (str[0] == '\\' && str[1] != '\0')
+ *(dst++) = *(str++);
+ *(dst++) = *(str++);
}
- if (*ptr == '"')
- ++ptr;
+ *(dst++) = *(str++);
- if (*ptr == ' ') {
- *(ptr++) = '\0';
- } else if (*ptr != '\0') {
+ if (*str != ' ' && *str != '\0')
goto fail_str;
- }
+ if (*str == ' ')
+ ++str;
- if (unescape(argv[i - 1]))
+ *(dst++) = '\0';
+
+ if (unescape(start))
goto fail_str;
- } else {
- while (*ptr != '\0' && *ptr != ' ')
- ++ptr;
- if (*ptr == ' ')
- *(ptr++) = '\0';
+ dst = start + strlen(start) + 1;
+ } else {
+ while (*str != '\0' && *str != ' ')
+ *(dst++) = *(str++);
+ if (*str == ' ') {
+ ++str;
+ *(dst++) = '\0';
+ }
}
+
+ ++count;
}
- return argv;
+ *dst = '\0';
+ return count;
fail_str:
- free(argv);
errno = EINVAL;
- return NULL;
+ return -1;
+}
+
+char **split_argv(char *str)
+{
+ char **argv = NULL;
+ int i, count;
+
+ count = pack_argv(str);
+ if (count <= 0)
+ return NULL;
+
+ argv = malloc(sizeof(argv[0]) * (count + 1));
+ if (argv == NULL)
+ return NULL;
+
+ for (i = 0; i < count; ++i) {
+ argv[i] = str;
+ str += strlen(str) + 1;
+ }
+
+ argv[i] = NULL;
+ return argv;
}