From 59731dd69bb4be66c721b3de66fb2ab80bec3e7e Mon Sep 17 00:00:00 2001 From: David Oberhollenzer Date: Sun, 22 Apr 2018 01:02:27 +0200 Subject: Add helper program for running services We no longer need to keep entire scripts in init program (i.e. saving space) and reduce the code and complexity of the init program. The runsvc tool can later be extended to do more complex child setup, such as configuring namespaces or seccomp without adding complexity or memory footprint to init. Signed-off-by: David Oberhollenzer --- cmd/Makemodule.am | 8 ++- cmd/runsvc/env.c | 130 +++++++++++++++++++++++++++++++++++++++++++++++++ cmd/runsvc/runsvc.c | 138 ++++++++++++++++++++++++++++++++++++++++++++++++++++ cmd/runsvc/runsvc.h | 36 ++++++++++++++ 4 files changed, 311 insertions(+), 1 deletion(-) create mode 100644 cmd/runsvc/env.c create mode 100644 cmd/runsvc/runsvc.c create mode 100644 cmd/runsvc/runsvc.h (limited to 'cmd') diff --git a/cmd/Makemodule.am b/cmd/Makemodule.am index f0ae3f9..83d6174 100644 --- a/cmd/Makemodule.am +++ b/cmd/Makemodule.am @@ -10,6 +10,12 @@ reboot_CFLAGS = $(AM_CFLAGS) reboot_LDFLAGS = $(AM_LDFLAGS) reboot_LDADD = libinit.a +runsvc_SOURCES = cmd/runsvc/runsvc.c cmd/runsvc/env.c cmd/runsvc/runsvc.h +runsvc_CPPFLAGS = $(AM_CPPFLAGS) +runsvc_CFLAGS = $(AM_CFLAGS) +runsvc_LDFLAGS = $(AM_LDFLAGS) +runsvc_LDADD = libinit.a + killall5_SOURCES = cmd/killall5.c killall5_CPPFLAGS = $(AM_CPPFLAGS) killall5_CFLAGS = $(AM_CFLAGS) @@ -29,4 +35,4 @@ service_LDADD = libinit.a EXTRA_DIST += $(SRVHEADERS) sbin_PROGRAMS += service reboot shutdown -helper_PROGRAMS += killall5 +helper_PROGRAMS += killall5 runsvc diff --git a/cmd/runsvc/env.c b/cmd/runsvc/env.c new file mode 100644 index 0000000..4fe2368 --- /dev/null +++ b/cmd/runsvc/env.c @@ -0,0 +1,130 @@ +/* SPDX-License-Identifier: GPL-3.0-or-later */ +/* + * Copyright (C) 2018 - David Oberhollenzer + * + * This program is free software: you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation, either version 3 of the License, or + * (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program. If not, see . + */ +#include "runsvc.h" + +struct entry { + struct entry *next; + char data[]; +}; + +extern char **environ; + +static void free_list(struct entry *list) +{ + struct entry *e; + + while (list != NULL) { + e = list; + list = list->next; + free(e); + } +} + +static struct entry *parse_list(rdline_t *rd) +{ + struct entry *e, *list = NULL; + char *ptr; + + while (rdline(rd) == 0) { + ptr = rd->buffer; + + while (*ptr != '\0' && *ptr != ' ' && *ptr != '=') + ++ptr; + + if (*ptr == ' ') + memmove(ptr, ptr + 1, strlen(ptr + 1) + 1); + + if (*(ptr++) != '=') { + fprintf(stderr, "%s: %zu: line is not of the shape " + "'key = value', skipping\n", + rd->filename, rd->lineno); + continue; + } + + if (*ptr == ' ') + memmove(ptr, ptr + 1, strlen(ptr + 1) + 1); + + if (unescape(ptr)) { + fprintf(stderr, "%s: %zu: malformed string constant, " + "skipping\n", + rd->filename, rd->lineno); + continue; + } + + e = calloc(1, sizeof(*e) + strlen(rd->buffer) + 1); + if (e == NULL) + goto fail_oom; + + strcpy(e->data, rd->buffer); + e->next = list; + list = e; + } + + return list; +fail_oom: + fputs("out of memory\n", stderr); + free_list(list); + return NULL; +} + +static struct entry *list_from_file(void) +{ + struct entry *list; + rdline_t rd; + int fd; + + fd = open(ENVFILE, O_RDONLY); + if (fd < 0) { + perror(ENVFILE); + return NULL; + } + + rdline_init(&rd, fd, ENVFILE, 0, NULL); + list = parse_list(&rd); + close(fd); + return list; +} + +int initenv(void) +{ + struct entry *list, *e; + int i, count; + char **envp; + + list = list_from_file(); + if (list == NULL) + return -1; + + for (count = 0, e = list; e != NULL; e = e->next) + ++count; + + envp = malloc((count + 1) * sizeof(char *)); + if (envp == NULL) { + fputs("out of memory\n", stderr); + free_list(list); + return -1; + } + + for (i = 0, e = list; e != NULL; e = e->next) + envp[i] = e->data; + + envp[i] = NULL; + + environ = envp; + return 0; +} diff --git a/cmd/runsvc/runsvc.c b/cmd/runsvc/runsvc.c new file mode 100644 index 0000000..a8e5bb2 --- /dev/null +++ b/cmd/runsvc/runsvc.c @@ -0,0 +1,138 @@ +/* SPDX-License-Identifier: GPL-3.0-or-later */ +/* + * Copyright (C) 2018 - David Oberhollenzer + * + * This program is free software: you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation, either version 3 of the License, or + * (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program. If not, see . + */ +#include "runsvc.h" + +static int setup_tty(const char *ctty) +{ + int fd; + + if (ctty != NULL) { + fd = open(ctty, O_RDWR); + if (fd < 0) { + perror(ctty); + return -1; + } + + close(STDIN_FILENO); + close(STDOUT_FILENO); + close(STDERR_FILENO); + + setsid(); + + dup2(fd, STDIN_FILENO); + dup2(fd, STDOUT_FILENO); + dup2(fd, STDERR_FILENO); + close(fd); + } + + return 0; +} + +/*****************************************************************************/ + +static NORETURN void argv_exec(exec_t *e) +{ + char **argv = alloca(e->argc + 1), *ptr; + int i; + + for (ptr = e->args, i = 0; i < e->argc; ++i, ptr += strlen(ptr) + 1) + argv[i] = ptr; + + argv[i] = NULL; + execvp(argv[0], argv); + perror(argv[0]); + exit(EXIT_FAILURE); +} + +static int runlst_wait(exec_t *list) +{ + pid_t ret, pid; + int status; + + for (; list != NULL; list = list->next) { + pid = fork(); + + if (pid == 0) + argv_exec(list); + + if (pid == -1) { + perror("fork"); + return EXIT_FAILURE; + } + + do { + ret = waitpid(pid, &status, 0); + } while (ret != pid); + + if (!WIFEXITED(status)) + return EXIT_FAILURE; + + if (WEXITSTATUS(status) != EXIT_SUCCESS) + return WEXITSTATUS(status); + } + + return EXIT_SUCCESS; +} + +/*****************************************************************************/ + +int main(int argc, char **argv) +{ + int dirfd, ret = EXIT_FAILURE; + service_t *svc = NULL; + + if (argc != 3) { + fputs("usage: runsvc \n", stderr); + goto out; + } + + if (getppid() != 1) { + fputs("must be run by init!\n", stderr); + goto out; + } + + dirfd = open(argv[1], O_RDONLY | O_DIRECTORY); + if (dirfd < 0) { + perror(argv[1]); + goto out; + } + + svc = rdsvc(dirfd, argv[2], RDSVC_NO_FNAME | RDSVC_NO_DEPS); + close(dirfd); + if (svc == NULL) + goto out; + + if (svc->exec == NULL) { + ret = EXIT_SUCCESS; + goto out; + } + + if (initenv()) + goto out; + + if (setup_tty(svc->ctty)) + goto out; + + if (svc->exec->next == NULL) + argv_exec(svc->exec); + + ret = runlst_wait(svc->exec); +out: + delsvc(svc); + return ret; +} diff --git a/cmd/runsvc/runsvc.h b/cmd/runsvc/runsvc.h new file mode 100644 index 0000000..15785d3 --- /dev/null +++ b/cmd/runsvc/runsvc.h @@ -0,0 +1,36 @@ +/* SPDX-License-Identifier: GPL-3.0-or-later */ +/* + * Copyright (C) 2018 - David Oberhollenzer + * + * This program is free software: you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation, either version 3 of the License, or + * (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program. If not, see . + */ +#ifndef RUNSVC_H +#define RUNSVC_H + +#include +#include +#include +#include +#include +#include +#include + +#include "service.h" +#include "util.h" + +#define ENVFILE ETCPATH "/initd.env" + +int initenv(void); + +#endif /* RUNSVC_H */ -- cgit v1.2.3