4 * Copyright (C) 2013 Felix Fietkau <nbd@openwrt.org>
6 * Permission to use, copy, modify, and/or distribute this software for any
7 * purpose with or without fee is hereby granted, provided that the above
8 * copyright notice and this permission notice appear in all copies.
10 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
11 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
12 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
13 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
14 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
15 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
16 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
19 #include <libubox/uloop.h>
20 #include <libubox/runqueue.h>
25 #include <sys/types.h>
30 #include <libubox/ustream.h>
35 static struct runqueue q, r;
39 struct runqueue_process proc;
44 static void pipe_cb(struct ustream *s, int bytes)
46 struct initd *initd = container_of(s, struct initd, fd.stream);
51 str = ustream_get_read_buf(s, NULL);
54 newline = strchr(str, '\n');
58 len = newline + 1 - str;
59 ULOG_NOTE("%s: %s", initd->file, str);
60 #ifdef SHOW_BOOT_ON_CONSOLE
61 fprintf(stderr, "%s: %s\n", initd->file, str);
63 ustream_consume(s, len);
67 static void q_initd_run(struct runqueue *q, struct runqueue_task *t)
69 struct initd *s = container_of(t, struct initd, proc.task);
73 DEBUG(2, "start %s %s \n", s->file, s->param);
74 if (pipe(pipefd) == -1) {
75 ERROR("Failed to create pipe\n");
85 fcntl(pipefd[0], F_SETFD, FD_CLOEXEC);
86 s->fd.stream.string_data = true,
87 s->fd.stream.notify_read = pipe_cb,
88 runqueue_process_add(q, &s->proc, pid);
89 ustream_fd_init(&s->fd, pipefd[0]);
94 int devnull = open("/dev/null", O_RDONLY);
95 dup2(devnull, STDIN_FILENO);
96 dup2(pipefd[1], STDOUT_FILENO);
97 dup2(pipefd[1], STDERR_FILENO);
99 if (devnull > STDERR_FILENO)
102 execlp(s->file, s->file, s->param, NULL);
106 static void q_initd_complete(struct runqueue *q, struct runqueue_task *p)
108 struct initd *s = container_of(p, struct initd, proc.task);
110 DEBUG(2, "stop %s %s \n", s->file, s->param);
111 ustream_free(&s->fd.stream);
116 static void add_initd(struct runqueue *q, char *file, char *param)
118 static const struct runqueue_task_type initd_type = {
120 .cancel = runqueue_process_cancel_cb,
121 .kill = runqueue_process_kill_cb,
126 s = calloc_a(sizeof(*s), &f, strlen(file) + 1, &p, strlen(param) + 1);
128 ERROR("Out of memory in %s.\n", file);
131 s->proc.task.type = &initd_type;
132 s->proc.task.complete = q_initd_complete;
133 if (!strcmp(param, "stop") || !strcmp(param, "shutdown")) {
134 s->proc.task.run_timeout = 15000;
135 s->proc.task.cancel_timeout = 10000;
139 strcpy(s->param, param);
140 strcpy(s->file, file);
141 runqueue_task_add(q, &s->proc.task, false);
144 static int _rc(struct runqueue *q, char *path, const char *file, char *pattern, char *param)
146 char *dir = alloca(2 + strlen(path) + strlen(file) + strlen(pattern));
151 ERROR("Out of memory in %s.\n", file);
155 DEBUG(2, "running %s/%s%s %s\n", path, file, pattern, param);
156 sprintf(dir, "%s/%s%s", path, file, pattern);
157 if (glob(dir, GLOB_NOESCAPE | GLOB_MARK, NULL, &gl)) {
158 DEBUG(2, "glob failed on %s\n", dir);
162 for (j = 0; j < gl.gl_pathc; j++)
163 add_initd(q, gl.gl_pathv[j], param);
170 int rcS(char *pattern, char *param, void (*q_empty)(struct runqueue *))
173 q.empty_cb = q_empty;
174 q.max_running_tasks = 1;
176 return _rc(&q, "/etc/rc.d", pattern, "*", param);
179 int rc(const char *file, char *param)
181 return _rc(&r, "/etc/init.d", file, "", param);
184 static void r_empty(struct runqueue *q)
189 static void __attribute__((constructor)) rc_init() {
191 r.empty_cb = r_empty;
192 r.max_running_tasks = 8;