From 04a23330e4a2085ee91980c223c5e4f089ebbe97 Mon Sep 17 00:00:00 2001 From: David Oberhollenzer Date: Wed, 4 Apr 2018 14:58:01 +0200 Subject: Merge preprocessing of command lines - Common function for splitting string into argument vector - Preprocess & split command lines while parsing the service file - Specify "before" and "after" dependencies in a single line Signed-off-by: David Oberhollenzer --- lib/include/service.h | 14 ++++-- lib/include/util.h | 2 + lib/src/delsvc.c | 17 ++++--- lib/src/rdsvc.c | 121 ++++++++++++++++++++++++++++++-------------------- lib/src/splitkv.c | 73 ++++++++++++++++++++++++++++++ lib/src/svc_tsort.c | 16 ++++--- 6 files changed, 177 insertions(+), 66 deletions(-) (limited to 'lib') diff --git a/lib/include/service.h b/lib/include/service.h index 76d48ce..593b221 100644 --- a/lib/include/service.h +++ b/lib/include/service.h @@ -44,21 +44,27 @@ enum { TGT_MAX }; +typedef struct exec_t { + char **argv; + char *raw_argv; + + struct exec_t *next; +} exec_t; + typedef struct service_t { int type; /* SVC_* service type */ int target; /* TGT_* service target */ char *name; /* canonical service name */ char *desc; /* description string */ - char **exec; /* command lines to execute */ - size_t num_exec; /* number of command lines */ + exec_t *exec; /* command lines to execute */ char *ctty; /* controlling tty or log file */ int rspwn_limit; /* maximum respawn count */ char **before; /* services that must be executed later */ - size_t num_before; char **after; /* services that must be executed first */ - size_t num_after; + char *raw_after; + char *raw_before; pid_t pid; int status; /* process exit status */ diff --git a/lib/include/util.h b/lib/include/util.h index 0ed7002..a6e96fa 100644 --- a/lib/include/util.h +++ b/lib/include/util.h @@ -73,6 +73,8 @@ char *rdline(int fd, int argc, const char *const *argv); */ int unescape(char *src); +char **split_argv(char *str); + /* Search through an array of enum_map_t entries to resolve a string to a numeric value. The end of the map is indicated by a sentinel entry diff --git a/lib/src/delsvc.c b/lib/src/delsvc.c index 3e077fb..7370653 100644 --- a/lib/src/delsvc.c +++ b/lib/src/delsvc.c @@ -21,16 +21,19 @@ void delsvc(service_t *svc) { - size_t i; + exec_t *e; - for (i = 0; i < svc->num_exec; ++i) - free(svc->exec[i]); + while (svc->exec != NULL) { + e = svc->exec; + svc->exec = e->next; - for (i = 0; i < svc->num_before; ++i) - free(svc->before[i]); + free(e->argv); + free(e->raw_argv); + free(e); + } - for (i = 0; i < svc->num_after; ++i) - free(svc->after[i]); + free(svc->raw_before); + free(svc->raw_after); free(svc->before); free(svc->after); diff --git a/lib/src/rdsvc.c b/lib/src/rdsvc.c index 04cf30f..5058c76 100644 --- a/lib/src/rdsvc.c +++ b/lib/src/rdsvc.c @@ -38,6 +38,26 @@ static int try_unescape(char *arg, const char *filename, size_t lineno) return 0; } +static char **try_split_argv(char *str, const char *filename, size_t lineno) +{ + char **argv = split_argv(str); + + if (argv == NULL) { + switch (errno) { + case EINVAL: + fprintf(stderr, "%s: %zu: malformed string constant\n", + filename, lineno); + break; + default: + fprintf(stderr, "%s: %zu: %s\n", filename, lineno, + strerror(errno)); + break; + } + } + + return argv; +} + static int svc_desc(service_t *svc, char *arg, const char *filename, size_t lineno) { @@ -59,107 +79,108 @@ static int svc_tty(service_t *svc, char *arg, static int svc_exec(service_t *svc, char *arg, const char *filename, size_t lineno) { - char **new = realloc(svc->exec, sizeof(char*) * (svc->num_exec + 1)); + exec_t *e, *end; + char **argv; - if (new == NULL) { + argv = try_split_argv(arg, filename, lineno); + if (argv == NULL) + return -1; + + e = calloc(1, sizeof(*e)); + if (e == NULL) { fprintf(stderr, "%s: %zu: out of memory\n", filename, lineno); - free(arg); + free(argv); return -1; } - svc->exec = new; - svc->exec[svc->num_exec++] = arg; + e->argv = argv; + e->raw_argv = arg; + + if (svc->exec == NULL) { + svc->exec = e; + } else { + for (end = svc->exec; end->next != NULL; end = end->next) + ; + end->next = e; + } return 0; } static int svc_before(service_t *svc, char *arg, const char *filename, size_t lineno) { - char **new; - - if (try_unescape(arg, filename, lineno)) + if (svc->before != NULL) { + fprintf(stderr, "%s: %zu: 'before' dependencies respecified\n", + filename, lineno); return -1; + } - new = realloc(svc->before, sizeof(char*) * (svc->num_before + 1)); - - if (new == NULL) { - fprintf(stderr, "%s: %zu: out of memory\n", filename, lineno); - free(arg); + svc->before = try_split_argv(arg, filename, lineno); + if (svc->before == NULL) return -1; - } - svc->before = new; - svc->before[svc->num_before++] = arg; + svc->raw_before = arg; return 0; } static int svc_after(service_t *svc, char *arg, const char *filename, size_t lineno) { - char **new; - - if (try_unescape(arg, filename, lineno)) + if (svc->after != NULL) { + fprintf(stderr, "%s: %zu: 'after' dependencies respecified\n", + filename, lineno); return -1; + } - new = realloc(svc->after, sizeof(char*) * (svc->num_after + 1)); - - if (new == NULL) { - fprintf(stderr, "%s: %zu: out of memory\n", filename, lineno); - free(arg); + svc->after = try_split_argv(arg, filename, lineno); + if (svc->after == NULL) return -1; - } - svc->after = new; - svc->after[svc->num_after++] = arg; + svc->raw_after = arg; return 0; } static int svc_type(service_t *svc, char *arg, const char *filename, size_t lineno) { - char *extra; - int type; + char **args; + int i, type; - if (try_unescape(arg, filename, lineno)) - return -1; - - for (extra = arg; *extra != ' ' && *extra != '\0'; ++extra) - ; + args = try_split_argv(arg, filename, lineno); - if (*extra == ' ') { - *(extra++) = '\0'; - } else { - extra = NULL; - } + if (args == NULL) + return -1; - type = svc_type_from_string(arg); + type = svc_type_from_string(args[0]); if (type == -1) { fprintf(stderr, "%s: %zu: unknown service type '%s'\n", - filename, lineno, arg); + filename, lineno, args[0]); + free(args); return -1; } - if (extra != NULL) { + if (args[1] != NULL) { switch (type) { case SVC_RESPAWN: - if (strncmp(extra, "limit ", 6) != 0) + if (strcmp(args[1], "limit") != 0) goto fail_limit; - extra += 6; svc->rspwn_limit = 0; - if (!isdigit(*extra)) + if (!isdigit(args[2][0])) goto fail_limit; - while (isdigit(*extra)) { + for (i = 0; isdigit(args[2][i]); ++i) { svc->rspwn_limit *= 10; - svc->rspwn_limit += *(extra++) - '0'; + svc->rspwn_limit += args[2][i] - '0'; } - if (*extra != '\0') + if (args[2][i] != '\0') goto fail_limit; - break; + if (args[3] == NULL) + break; + /* fall-through */ default: fprintf(stderr, "%s: %zu: unexpected extra arguments " "for type '%s'\n", filename, lineno, arg); @@ -168,11 +189,13 @@ static int svc_type(service_t *svc, char *arg, } svc->type = type; + free(args); free(arg); return 0; fail_limit: fprintf(stderr, "%s: %zu: expected 'limit ' after 'respawn'\n", filename, lineno); + free(args); return -1; } diff --git a/lib/src/splitkv.c b/lib/src/splitkv.c index 3631ea8..7d652d3 100644 --- a/lib/src/splitkv.c +++ b/lib/src/splitkv.c @@ -15,7 +15,9 @@ * You should have received a copy of the GNU General Public License * along with this program. If not, see . */ +#include #include +#include #include "util.h" @@ -86,3 +88,74 @@ int unescape(char *src) *(dst++) = '\0'; return 0; } + +char **split_argv(char *str) +{ + size_t i = 0, cap = 0, new_cap; + char **argv = NULL, **new; + char *ptr; + + ptr = 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; + } + + if (*ptr == '\0') { + argv[i++] = NULL; + break; + } + + argv[i++] = ptr; + + if (*ptr == '"') { + ++ptr; + while (*ptr != '\0' && *ptr != '"') { + if (ptr[0] == '\\' && ptr[1] != '\0') + ++ptr; + + ++ptr; + } + + if (*ptr == '"') + ++ptr; + + if (*ptr == ' ') { + *(ptr++) = '\0'; + } else if (*ptr != '\0') { + goto fail_str; + } + + if (unescape(argv[i - 1])) + goto fail_str; + } else { + while (*ptr != '\0' && *ptr != ' ') + ++ptr; + + if (*ptr == ' ') + *(ptr++) = '\0'; + } + } + + return argv; +fail_str: + free(argv); + errno = EINVAL; + return NULL; +} diff --git a/lib/src/svc_tsort.c b/lib/src/svc_tsort.c index 50517f2..dcd1b4c 100644 --- a/lib/src/svc_tsort.c +++ b/lib/src/svc_tsort.c @@ -27,14 +27,18 @@ static bool has_dependencies(service_t *list, service_t *svc) size_t i; while (list != NULL) { - for (i = 0; i < svc->num_after; ++i) { - if (!strcmp(svc->after[i], list->name)) - return true; + if (svc->after != NULL) { + for (i = 0; svc->after[i] != NULL; ++i) { + if (!strcmp(svc->after[i], list->name)) + return true; + } } - for (i = 0; i < list->num_before; ++i) { - if (!strcmp(list->before[i], svc->name)) - return true; + if (list->before != NULL) { + for (i = 0; list->before[i] != NULL; ++i) { + if (!strcmp(list->before[i], svc->name)) + return true; + } } list = list->next; -- cgit v1.2.3