diff options
Diffstat (limited to 'cmd')
-rw-r--r-- | cmd/Makemodule.am | 8 | ||||
-rw-r--r-- | cmd/runsvc/env.c | 130 | ||||
-rw-r--r-- | cmd/runsvc/runsvc.c | 138 | ||||
-rw-r--r-- | cmd/runsvc/runsvc.h | 36 |
4 files changed, 311 insertions, 1 deletions
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 <https://www.gnu.org/licenses/>. + */ +#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 <https://www.gnu.org/licenses/>. + */ +#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 <directory> <filename>\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 <https://www.gnu.org/licenses/>. + */ +#ifndef RUNSVC_H +#define RUNSVC_H + +#include <sys/types.h> +#include <sys/wait.h> +#include <unistd.h> +#include <stdlib.h> +#include <string.h> +#include <stdio.h> +#include <fcntl.h> + +#include "service.h" +#include "util.h" + +#define ENVFILE ETCPATH "/initd.env" + +int initenv(void); + +#endif /* RUNSVC_H */ |