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