aboutsummaryrefslogtreecommitdiff
path: root/lib/src
diff options
context:
space:
mode:
Diffstat (limited to 'lib/src')
-rw-r--r--lib/src/rdline.c106
-rw-r--r--lib/src/rdsvc.c191
-rw-r--r--lib/src/unescape.c1
3 files changed, 140 insertions, 158 deletions
diff --git a/lib/src/rdline.c b/lib/src/rdline.c
index 18c9c85..47a9d2a 100644
--- a/lib/src/rdline.c
+++ b/lib/src/rdline.c
@@ -20,22 +20,10 @@
#include <string.h>
#include <errno.h>
#include <ctype.h>
+#include <stdio.h>
#include "util.h"
-typedef struct {
- int fd; /* input file descriptor */
- const char *argstr; /* if not NULL, read from this instead */
-
- size_t i; /* buffer offset */
- size_t bufsiz; /* buffer size */
- char *buffer;
-
- bool string; /* inside a string? */
- bool escape; /* reading an escape sequence? */
- bool comment; /* inside a comment */
-} rdline_t;
-
static int rdline_getc(rdline_t *t)
{
int ret;
@@ -69,9 +57,6 @@ out:
static int rdline_append(rdline_t *t, int c)
{
- size_t newsz;
- char *new;
-
if (t->comment) {
if (c != '\0')
return 0;
@@ -102,67 +87,88 @@ static int rdline_append(rdline_t *t, int c)
t->i -= 1;
}
- if (t->i == t->bufsiz) {
- newsz = t->bufsiz ? t->bufsiz * 2 : 16;
- new = realloc(t->buffer, newsz);
-
- if (new == NULL)
- return -1;
-
- t->buffer = new;
- t->bufsiz = newsz;
- }
+ if (t->i == sizeof(t->buffer))
+ return -1;
t->buffer[t->i++] = c;
return 0;
}
-char *rdline(int fd, int argc, const char *const *argv)
+void rdline_init(rdline_t *t, int fd, const char *filename,
+ int argc, const char *const *argv)
{
- rdline_t rd;
- int c, ret;
+ memset(t, 0, sizeof(*t));
+ t->fd = fd;
+ t->filename = filename;
+ t->argc = argc;
+ t->argv = argv;
+}
- memset(&rd, 0, sizeof(rd));
- rd.fd = fd;
+int rdline(rdline_t *t)
+{
+ const char *errstr;
+ int c;
+retry:
+ t->i = 0;
+ t->argstr = NULL;
+ t->string = t->escape = t->comment = false;
+ t->lineno += 1;
do {
- c = rdline_getc(&rd);
- if (c < 0)
+ errno = 0;
+ c = rdline_getc(t);
+ if (c < 0) {
+ if (errno == 0)
+ return 1;
+ errstr = strerror(errno);
goto fail;
- if (c == 0 && rd.string) {
- errno = EILSEQ;
+ }
+ if (c == 0 && t->string) {
+ errstr = "missing \"";
goto fail;
}
if (c == '%') {
- c = rdline_getc(&rd);
- if (c == 0)
- errno = EILSEQ;
- if (c <= 0)
+ c = rdline_getc(t);
+ if (c == 0) {
+ errstr = "unexpected end of line after '%%'";
goto fail;
+ }
+ if (c < 0) {
+ errstr = strerror(errno);
+ goto fail;
+ }
if (c != '%') {
- if (!isdigit(c) || (c - '0') >= argc) {
- errno = EINVAL;
+ if (!isdigit(c)) {
+ errstr = "exptected digit after '%%'";
goto fail;
}
- if (rd.argstr != NULL) {
- errno = ELOOP;
+ if ((c - '0') >= t->argc) {
+ errstr = "argument out of range";
goto fail;
}
- rd.argstr = argv[c - '0'];
+ if (t->argstr != NULL) {
+ errstr = "recursive argument "
+ "expansion";
+ goto fail;
+ }
+ t->argstr = t->argv[c - '0'];
continue;
}
}
- if (rdline_append(&rd, c))
+ if (rdline_append(t, c)) {
+ errstr = "line too long";
goto fail;
+ }
} while (c != '\0');
- return rd.buffer;
+ if (t->buffer[0] == '\0')
+ goto retry;
+
+ return 0;
fail:
- ret = errno;
- free(rd.buffer);
- errno = ret;
- return NULL;
+ fprintf(stderr, "%s: %zu: %s\n", t->filename, t->lineno, errstr);
+ return -1;
}
diff --git a/lib/src/rdsvc.c b/lib/src/rdsvc.c
index 29c9019..e911724 100644
--- a/lib/src/rdsvc.c
+++ b/lib/src/rdsvc.c
@@ -28,17 +28,17 @@
#include "service.h"
#include "util.h"
-static int try_unescape(char *arg, const char *filename, size_t lineno)
+static int try_unescape(char *arg, rdline_t *rd)
{
if (unescape(arg)) {
fprintf(stderr, "%s: %zu: malformed string constant\n",
- filename, lineno);
+ rd->filename, rd->lineno);
return -1;
}
return 0;
}
-static char **try_split_argv(char *str, const char *filename, size_t lineno)
+static char **try_split_argv(char *str, rdline_t *rd)
{
char **argv = split_argv(str);
@@ -46,11 +46,11 @@ static char **try_split_argv(char *str, const char *filename, size_t lineno)
switch (errno) {
case EINVAL:
fprintf(stderr, "%s: %zu: malformed string constant\n",
- filename, lineno);
+ rd->filename, rd->lineno);
break;
default:
- fprintf(stderr, "%s: %zu: %s\n", filename, lineno,
- strerror(errno));
+ fprintf(stderr, "%s: %zu: %s\n",
+ rd->filename, rd->lineno, strerror(errno));
break;
}
}
@@ -58,43 +58,56 @@ static char **try_split_argv(char *str, const char *filename, size_t lineno)
return argv;
}
-static int svc_desc(service_t *svc, char *arg,
- const char *filename, size_t lineno)
+static char *try_strdup(const char *str, rdline_t *rd)
{
- if (try_unescape(arg, filename, lineno))
+ char *out = strdup(str);
+
+ if (out == NULL) {
+ fprintf(stderr, "%s: %zu: out of memory\n",
+ rd->filename, rd->lineno);
+ }
+ return out;
+}
+
+static int svc_desc(service_t *svc, char *arg, rdline_t *rd)
+{
+ if (try_unescape(arg, rd))
return -1;
- svc->desc = arg;
- return 0;
+ svc->desc = try_strdup(arg, rd);
+ return svc->desc == NULL ? -1 : 0;
}
-static int svc_tty(service_t *svc, char *arg,
- const char *filename, size_t lineno)
+static int svc_tty(service_t *svc, char *arg, rdline_t *rd)
{
- if (try_unescape(arg, filename, lineno))
+ if (try_unescape(arg, rd))
return -1;
- svc->ctty = arg;
- return 0;
+ svc->ctty = try_strdup(arg, rd);
+ return svc->ctty == NULL ? -1 : 0;
}
-static int svc_exec(service_t *svc, char *arg,
- const char *filename, size_t lineno)
+static int svc_exec(service_t *svc, char *arg, rdline_t *rd)
{
exec_t *e, *end;
- char **argv;
-
- 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(argv);
+ fprintf(stderr, "%s: %zu: out of memory\n",
+ rd->filename, rd->lineno);
return -1;
}
- e->argv = argv;
- e->raw_argv = arg;
+ e->raw_argv = try_strdup(arg, rd);
+ if (e->raw_argv == NULL) {
+ free(e);
+ return -1;
+ }
+
+ e->argv = try_split_argv(e->raw_argv, rd);
+ if (e->argv == NULL) {
+ free(e->raw_argv);
+ free(e);
+ return -1;
+ }
if (svc->exec == NULL) {
svc->exec = e;
@@ -106,47 +119,50 @@ static int svc_exec(service_t *svc, char *arg,
return 0;
}
-static int svc_before(service_t *svc, char *arg,
- const char *filename, size_t lineno)
+static int svc_before(service_t *svc, char *arg, rdline_t *rd)
{
if (svc->before != NULL) {
fprintf(stderr, "%s: %zu: 'before' dependencies respecified\n",
- filename, lineno);
+ rd->filename, rd->lineno);
return -1;
}
- svc->before = try_split_argv(arg, filename, lineno);
+ svc->raw_before = try_strdup(arg, rd);
+ if (svc->raw_before == NULL)
+ return -1;
+
+ svc->before = try_split_argv(svc->raw_before, rd);
if (svc->before == NULL)
return -1;
- svc->raw_before = arg;
return 0;
}
-static int svc_after(service_t *svc, char *arg,
- const char *filename, size_t lineno)
+static int svc_after(service_t *svc, char *arg, rdline_t *rd)
{
if (svc->after != NULL) {
fprintf(stderr, "%s: %zu: 'after' dependencies respecified\n",
- filename, lineno);
+ rd->filename, rd->lineno);
return -1;
}
- svc->after = try_split_argv(arg, filename, lineno);
+ svc->raw_after = try_strdup(arg, rd);
+ if (svc->raw_after == NULL)
+ return -1;
+
+ svc->after = try_split_argv(svc->raw_after, rd);
if (svc->after == NULL)
return -1;
- svc->raw_after = arg;
return 0;
}
-static int svc_type(service_t *svc, char *arg,
- const char *filename, size_t lineno)
+static int svc_type(service_t *svc, char *arg, rdline_t *rd)
{
char **args;
int i, type;
- args = try_split_argv(arg, filename, lineno);
+ args = try_split_argv(arg, rd);
if (args == NULL)
return -1;
@@ -155,7 +171,7 @@ static int svc_type(service_t *svc, char *arg,
if (type == -1) {
fprintf(stderr, "%s: %zu: unknown service type '%s'\n",
- filename, lineno, args[0]);
+ rd->filename, rd->lineno, args[0]);
free(args);
return -1;
}
@@ -183,48 +199,45 @@ static int svc_type(service_t *svc, char *arg,
/* fall-through */
default:
fprintf(stderr, "%s: %zu: unexpected extra arguments "
- "for type '%s'\n", filename, lineno, arg);
+ "for type '%s'\n",
+ rd->filename, rd->lineno, arg);
return -1;
}
}
svc->type = type;
free(args);
- free(arg);
return 0;
fail_limit:
fprintf(stderr, "%s: %zu: expected 'limit <value>' after 'respawn'\n",
- filename, lineno);
+ rd->filename, rd->lineno);
free(args);
return -1;
}
-static int svc_target(service_t *svc, char *arg,
- const char *filename, size_t lineno)
+static int svc_target(service_t *svc, char *arg, rdline_t *rd)
{
int target;
- if (try_unescape(arg, filename, lineno))
+ if (try_unescape(arg, rd))
return -1;
target = svc_target_from_string(arg);
if (target == -1) {
fprintf(stderr, "%s: %zu: unknown service target '%s'\n",
- filename, lineno, arg);
+ rd->filename, rd->lineno, arg);
return -1;
}
svc->target = target;
- free(arg);
return 0;
}
static const struct svc_param {
const char *key;
- int (*handle)(service_t *svc, char *arg,
- const char *filename, size_t lineno);
+ int (*handle)(service_t *svc, char *arg, rdline_t *rd);
} svc_params[] = {
{ "description", svc_desc },
{ "exec", svc_exec },
@@ -235,49 +248,15 @@ static const struct svc_param {
{ "after", svc_after },
};
-
-static char *get_line(int fd, const char *filename, size_t *lineno,
- int argc, const char *const *args)
+static int splitkv(rdline_t *rd, char **k, char **v)
{
- const char *error;
- char *line;
- int ret;
-retry:
- errno = 0;
- line = rdline(fd, argc, args);
- ret = errno;
-
- if (line == NULL && errno != 0) {
- switch (errno) {
- case EINVAL: error = "error in argument expansion"; break;
- case ELOOP: error = "recursive argument expansion"; break;
- case EILSEQ: error = "missing \""; break;
- default: error = strerror(errno); break;
- }
-
- fprintf(stderr, "%s: %zu: %s\n", filename, *lineno, error);
- }
-
- if (line != NULL && strlen(line) == 0) {
- free(line);
- (*lineno) += 1;
- goto retry;
- }
-
- errno = ret;
- return line;
-}
-
-static int splitkv(const char *filename, size_t lineno,
- char *line, char **k, char **v)
-{
- char *key = line, *value = line;
+ char *key = rd->buffer, *value = rd->buffer;
while (*value != ' ' && *value != '\0') {
if (!isalpha(*value)) {
fprintf(stderr,
"%s: %zu: unexpected '%c' in keyword\n",
- filename, lineno, *value);
+ rd->filename, rd->lineno, *value);
return -1;
}
++value;
@@ -285,7 +264,7 @@ static int splitkv(const char *filename, size_t lineno,
if (*value != ' ') {
fprintf(stderr, "%s: %zu: expected argument after '%s'\n",
- filename, lineno, key);
+ rd->filename, rd->lineno, key);
return -1;
}
@@ -296,8 +275,7 @@ static int splitkv(const char *filename, size_t lineno,
return 0;
}
-static const struct svc_param *find_param(const char *filename, size_t lineno,
- const char *name)
+static const struct svc_param *find_param(rdline_t *rd, const char *name)
{
size_t i;
@@ -307,19 +285,20 @@ static const struct svc_param *find_param(const char *filename, size_t lineno,
}
fprintf(stderr, "%s: %zu: unknown keyword '%s'\n",
- filename, lineno, name);
+ rd->filename, rd->lineno, name);
return NULL;
}
service_t *rdsvc(int dirfd, const char *filename)
{
- char *line = NULL, *key, *value;
const struct svc_param *p;
const char *arg, *args[1];
service_t *svc = NULL;
- size_t argc, lineno;
- int fd;
+ char *key, *value;
+ size_t argc;
+ rdline_t rd;
+ int fd, ret;
fd = openat(dirfd, filename, O_RDONLY);
if (fd < 0) {
@@ -335,6 +314,8 @@ service_t *rdsvc(int dirfd, const char *filename)
argc = 0;
}
+ rdline_init(&rd, fd, filename, argc, args);
+
svc = calloc(1, sizeof(*svc));
if (svc == NULL)
goto fail_oom;
@@ -348,32 +329,26 @@ service_t *rdsvc(int dirfd, const char *filename)
if (svc->name == NULL)
goto fail_oom;
- for (lineno = 1; ; ++lineno) {
- line = get_line(fd, filename, &lineno, argc, args);
- if (line == NULL) {
- if (errno == 0)
- break;
- goto fail;
- }
-
- if (splitkv(filename, lineno, line, &key, &value))
+ while ((ret = rdline(&rd)) == 0) {
+ if (splitkv(&rd, &key, &value))
goto fail;
- p = find_param(filename, lineno, key);
+ p = find_param(&rd, key);
if (p == NULL)
goto fail;
- memmove(line, value, strlen(value) + 1);
- if (p->handle(svc, line, filename, lineno))
+ if (p->handle(svc, value, &rd))
goto fail;
}
+ if (ret < 0)
+ goto fail;
+
close(fd);
return svc;
fail_oom:
fputs("out of memory\n", stderr);
fail:
- free(line);
delsvc(svc);
close(fd);
return NULL;
diff --git a/lib/src/unescape.c b/lib/src/unescape.c
index 7508e2f..ea5c9e2 100644
--- a/lib/src/unescape.c
+++ b/lib/src/unescape.c
@@ -59,6 +59,7 @@ int unescape(char *src)
case 't': c = '\t'; break;
case '\\':
case '"':
+ case '%':
break;
case 'x':
c = 0;
generated by cgit v1.2.3 (git 2.39.1) at 2024-11-08 05:31:53 +0000