• source navigation  • diff markup  • identifier search  • freetext search  • 

Sources/procd/rcS.c

  1 /*
  2  * runqueue-example.c
  3  *
  4  * Copyright (C) 2013 Felix Fietkau <nbd@openwrt.org>
  5  *
  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.
  9  *
 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.
 17  */
 18 
 19 #include <libubox/uloop.h>
 20 #include <libubox/runqueue.h>
 21 #include <stdlib.h>
 22 #include <stdio.h>
 23 #include <unistd.h>
 24 
 25 #include <sys/types.h>
 26 #include <sys/stat.h>
 27 #include <fcntl.h>
 28 #include <glob.h>
 29 
 30 #include <libubox/ustream.h>
 31 
 32 #include "procd.h"
 33 #include "rcS.h"
 34 
 35 static struct runqueue q, r;
 36 
 37 struct initd {
 38         struct ustream_fd fd;
 39         struct runqueue_process proc;
 40         struct timespec ts_start;
 41         char *file;
 42         char *param;
 43 };
 44 
 45 static void pipe_cb(struct ustream *s, int bytes)
 46 {
 47         struct initd *initd = container_of(s, struct initd, fd.stream);
 48         char *newline, *str;
 49         int len;
 50 
 51         do {
 52                 str = ustream_get_read_buf(s, NULL);
 53                 if (!str)
 54                         break;
 55                 newline = strchr(str, '\n');
 56                 if (!newline)
 57                         break;
 58                 *newline = 0;
 59                 len = newline + 1 - str;
 60                 ULOG_NOTE("%s: %s", initd->file, str);
 61 #ifdef SHOW_BOOT_ON_CONSOLE
 62                 fprintf(stderr, "%s: %s\n", initd->file, str);
 63 #endif
 64                 ustream_consume(s, len);
 65         } while (1);
 66 }
 67 
 68 static void q_initd_run(struct runqueue *q, struct runqueue_task *t)
 69 {
 70         struct initd *s = container_of(t, struct initd, proc.task);
 71         int pipefd[2];
 72         pid_t pid;
 73 
 74         clock_gettime(CLOCK_MONOTONIC_RAW, &s->ts_start);
 75         DEBUG(2, "start %s %s \n", s->file, s->param);
 76         if (pipe(pipefd) == -1) {
 77                 ERROR("Failed to create pipe: %m\n");
 78                 return;
 79         }
 80 
 81         pid = fork();
 82         if (pid < 0)
 83                 return;
 84 
 85         if (pid) {
 86                 close(pipefd[1]);
 87                 fcntl(pipefd[0], F_SETFD, FD_CLOEXEC);
 88                 s->fd.stream.string_data = true,
 89                 s->fd.stream.notify_read = pipe_cb,
 90                 runqueue_process_add(q, &s->proc, pid);
 91                 ustream_fd_init(&s->fd, pipefd[0]);
 92                 return;
 93         }
 94         close(pipefd[0]);
 95 
 96         int devnull = open("/dev/null", O_RDONLY);
 97         dup2(devnull, STDIN_FILENO);
 98         dup2(pipefd[1], STDOUT_FILENO);
 99         dup2(pipefd[1], STDERR_FILENO);
100 
101         if (devnull > STDERR_FILENO)
102                 close(devnull);
103 
104         execlp(s->file, s->file, s->param, NULL);
105         exit(1);
106 }
107 
108 static void q_initd_complete(struct runqueue *q, struct runqueue_task *p)
109 {
110         struct initd *s = container_of(p, struct initd, proc.task);
111         struct timespec ts_stop, ts_res;
112 
113         clock_gettime(CLOCK_MONOTONIC_RAW, &ts_stop);
114         ts_res.tv_sec = ts_stop.tv_sec - s->ts_start.tv_sec;
115         ts_res.tv_nsec = ts_stop.tv_nsec - s->ts_start.tv_nsec;
116         if (ts_res.tv_nsec < 0) {
117                 --ts_res.tv_sec;
118                 ts_res.tv_nsec += 1000000000;
119         }
120 
121         DEBUG(2, "stop %s %s - took %lu.%09lus\n", s->file, s->param, ts_res.tv_sec, ts_res.tv_nsec);
122         ustream_free(&s->fd.stream);
123         close(s->fd.fd.fd);
124         free(s);
125 }
126 
127 static void add_initd(struct runqueue *q, char *file, char *param)
128 {
129         static const struct runqueue_task_type initd_type = {
130                 .run = q_initd_run,
131                 .cancel = runqueue_process_cancel_cb,
132                 .kill = runqueue_process_kill_cb,
133         };
134         struct initd *s;
135         char *p, *f;
136 
137         s = calloc_a(sizeof(*s), &f, strlen(file) + 1, &p, strlen(param) + 1);
138         if (!s) {
139                 ERROR("Out of memory in %s.\n", file);
140                 return;
141         }
142         s->proc.task.type = &initd_type;
143         s->proc.task.complete = q_initd_complete;
144         if (!strcmp(param, "stop") || !strcmp(param, "shutdown")) {
145                 s->proc.task.run_timeout = 15000;
146                 s->proc.task.cancel_timeout = 10000;
147         }
148         s->param = p;
149         s->file = f;
150         strcpy(s->param, param);
151         strcpy(s->file, file);
152         runqueue_task_add(q, &s->proc.task, false);
153 }
154 
155 static int _rc(struct runqueue *q, char *path, const char *file, char *pattern, char *param)
156 {
157         char *dir = alloca(2 + strlen(path) + strlen(file) + strlen(pattern));
158         glob_t gl;
159         int j;
160 
161         if (!dir) {
162                 ERROR("Out of memory in %s.\n", file);
163                 return -1;
164         }
165 
166         DEBUG(2, "running %s/%s%s %s\n", path, file, pattern, param);
167         sprintf(dir, "%s/%s%s", path, file, pattern);
168         if (glob(dir, GLOB_NOESCAPE | GLOB_MARK, NULL, &gl)) {
169                 DEBUG(2, "glob failed on %s\n", dir);
170                 return -1;
171         }
172 
173         for (j = 0; j < gl.gl_pathc; j++)
174                 add_initd(q, gl.gl_pathv[j], param);
175 
176         globfree(&gl);
177 
178         return 0;
179 }
180 
181 int rcS(char *pattern, char *param, void (*q_empty)(struct runqueue *))
182 {
183         runqueue_init(&q);
184         q.empty_cb = q_empty;
185         q.max_running_tasks = 1;
186 
187         return _rc(&q, "/etc/rc.d", pattern, "*", param);
188 }
189 
190 int rc(const char *file, char *param)
191 {
192         return _rc(&r, "/etc/init.d", file, "", param);
193 }
194 
195 static void r_empty(struct runqueue *q)
196 {
197 
198 }
199 
200 static void __attribute__((constructor)) rc_init() {
201         runqueue_init(&r);
202         r.empty_cb = r_empty;
203         r.max_running_tasks = 8;
204 }
205 

This page was automatically generated by LXR 0.3.1.  •  OpenWrt