diff options
Diffstat (limited to 'lib')
| -rw-r--r-- | lib/include/service.h | 11 | ||||
| -rw-r--r-- | lib/src/rdsvc.c | 45 | ||||
| -rw-r--r-- | lib/src/svcscan.c | 4 |
3 files changed, 40 insertions, 20 deletions
diff --git a/lib/include/service.h b/lib/include/service.h index c68ff90..49c5432 100644 --- a/lib/include/service.h +++ b/lib/include/service.h @@ -44,6 +44,13 @@ enum { TGT_MAX }; +enum { + RDSVC_NO_FNAME = 0x01, /* do not store a copy of the filename */ + RDSVC_NO_EXEC = 0x02, /* do not store executable script */ + RDSVC_NO_CTTY = 0x04, /* do not store the controlling tty */ + RDSVC_NO_DEPS = 0x08, /* do not store dependencies */ +}; + typedef struct exec_t { struct exec_t *next; int argc; /* number of elements in argument vector */ @@ -83,7 +90,7 @@ typedef struct { /* Read a service from a file. */ -service_t *rdsvc(int dirfd, const char *filename); +service_t *rdsvc(int dirfd, const char *filename, int flags); void delsvc(service_t *svc); @@ -94,7 +101,7 @@ void delsvc(service_t *svc); Returns 0 on success, -1 on failure. The function takes care of printing error messages on failure. */ -int svcscan(const char *directory, service_list_t *list); +int svcscan(const char *directory, service_list_t *list, int flags); void del_svc_list(service_list_t *list); diff --git a/lib/src/rdsvc.c b/lib/src/rdsvc.c index ef3465f..546962d 100644 --- a/lib/src/rdsvc.c +++ b/lib/src/rdsvc.c @@ -204,15 +204,17 @@ static const struct svc_param { unsigned int allow_block : 1; + int flags; + int (*handle)(service_t *svc, char *arg, rdline_t *rd); } svc_params[] = { - { "description", 0, svc_desc }, - { "exec", 1, svc_exec }, - { "type", 0, svc_type }, - { "target", 0, svc_target }, - { "tty", 0, svc_tty }, - { "before", 0, svc_before }, - { "after", 0, svc_after }, + { "description", 0, 0, svc_desc }, + { "exec", 1, RDSVC_NO_EXEC, svc_exec }, + { "type", 0, 0, svc_type }, + { "target", 0, 0, svc_target }, + { "tty", 0, RDSVC_NO_CTTY, svc_tty }, + { "before", 0, RDSVC_NO_DEPS, svc_before }, + { "after", 0, RDSVC_NO_DEPS, svc_after }, }; static int splitkv(rdline_t *rd, char **k, char **v) @@ -257,7 +259,7 @@ static const struct svc_param *find_param(rdline_t *rd, const char *name) } -service_t *rdsvc(int dirfd, const char *filename) +service_t *rdsvc(int dirfd, const char *filename, int flags) { const struct svc_param *p; const char *arg, *args[1]; @@ -289,9 +291,11 @@ service_t *rdsvc(int dirfd, const char *filename) if (svc == NULL) goto fail_oom; - svc->fname = strdup(filename); - if (svc->fname == NULL) - goto fail_oom; + if (!(flags & RDSVC_NO_FNAME)) { + svc->fname = strdup(filename); + if (svc->fname == NULL) + goto fail_oom; + } memcpy(svc->name, filename, nlen); @@ -306,13 +310,19 @@ service_t *rdsvc(int dirfd, const char *filename) if (p->allow_block && *value == '{') { for (++value; *value == ' '; ++value) ; - if (*value != '\0' && p->handle(svc, value, &rd)) - goto fail; + + if (!(flags & p->flags)) { + if (*value != '\0' && + p->handle(svc, value, &rd)) { + goto fail; + } + } while ((ret = rdline(&rd)) == 0) { if (strcmp(rd.buffer, "}") == 0) break; - + if (flags & p->flags) + continue; if (p->handle(svc, rd.buffer, &rd)) goto fail; } @@ -321,8 +331,11 @@ service_t *rdsvc(int dirfd, const char *filename) goto fail; if (ret > 0) goto fail_bra; - } else if (p->handle(svc, value, &rd)) { - goto fail; + } else { + if (flags & p->flags) + continue; + if (p->handle(svc, value, &rd)) + goto fail; } } diff --git a/lib/src/svcscan.c b/lib/src/svcscan.c index 32c1dee..f4c7ef1 100644 --- a/lib/src/svcscan.c +++ b/lib/src/svcscan.c @@ -27,7 +27,7 @@ #include "service.h" -int svcscan(const char *directory, service_list_t *list) +int svcscan(const char *directory, service_list_t *list, int flags) { int i, dfd, type, ret = 0; struct dirent *ent; @@ -82,7 +82,7 @@ int svcscan(const char *directory, service_list_t *list) if (type != S_IFREG && type != S_IFLNK) continue; - svc = rdsvc(dfd, ent->d_name); + svc = rdsvc(dfd, ent->d_name, flags); if (svc == NULL) { ret = -1; continue; |