1 /* 2 * netifd - network interface daemon 3 * Copyright (C) 2012 Felix Fietkau <nbd@openwrt.org> 4 * 5 * This program is free software; you can redistribute it and/or modify 6 * it under the terms of the GNU General Public License version 2 7 * as published by the Free Software Foundation 8 * 9 * This program is distributed in the hope that it will be useful, 10 * but WITHOUT ANY WARRANTY; without even the implied warranty of 11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 12 * GNU General Public License for more details. 13 */ 14 #include <stdio.h> 15 #include <stdlib.h> 16 #include <string.h> 17 #include <getopt.h> 18 #include <signal.h> 19 #include <stdarg.h> 20 #include <syslog.h> 21 22 #include "netifd.h" 23 #include "ubus.h" 24 #include "config.h" 25 #include "system.h" 26 #include "interface.h" 27 #include "wireless.h" 28 #include "proto.h" 29 30 unsigned int debug_mask = 0; 31 const char *main_path = DEFAULT_MAIN_PATH; 32 const char *config_path = DEFAULT_CONFIG_PATH; 33 const char *resolv_conf = DEFAULT_RESOLV_CONF; 34 static char **global_argv; 35 36 static struct list_head process_list = LIST_HEAD_INIT(process_list); 37 38 #define DEFAULT_LOG_LEVEL L_NOTICE 39 40 static int log_level = DEFAULT_LOG_LEVEL; 41 static const int log_class[] = { 42 [L_CRIT] = LOG_CRIT, 43 [L_WARNING] = LOG_WARNING, 44 [L_NOTICE] = LOG_NOTICE, 45 [L_INFO] = LOG_INFO, 46 [L_DEBUG] = LOG_DEBUG 47 }; 48 49 #ifdef DUMMY_MODE 50 #define use_syslog false 51 #else 52 static bool use_syslog = true; 53 #endif 54 55 56 static void 57 netifd_delete_process(struct netifd_process *proc) 58 { 59 list_del(&proc->list); 60 ustream_free(&proc->log.stream); 61 close(proc->log.fd.fd); 62 } 63 64 void 65 netifd_log_message(int priority, const char *format, ...) 66 { 67 va_list vl; 68 69 if (priority > log_level) 70 return; 71 72 va_start(vl, format); 73 if (use_syslog) 74 vsyslog(log_class[priority], format, vl); 75 else 76 vfprintf(stderr, format, vl); 77 va_end(vl); 78 } 79 80 static void 81 netifd_process_log_read_cb(struct ustream *s, int bytes) 82 { 83 struct netifd_process *proc; 84 const char *log_prefix; 85 char *data; 86 int len = 0; 87 88 proc = container_of(s, struct netifd_process, log.stream); 89 log_prefix = proc->log_prefix; 90 if (!log_prefix) 91 log_prefix = "process"; 92 93 do { 94 char *newline; 95 96 data = ustream_get_read_buf(s, &len); 97 if (!len) 98 break; 99 100 newline = strchr(data, '\n'); 101 102 if (proc->log_overflow) { 103 if (newline) { 104 len = newline + 1 - data; 105 proc->log_overflow = false; 106 } 107 } else if (newline) { 108 *newline = 0; 109 len = newline + 1 - data; 110 netifd_log_message(L_NOTICE, "%s (%d): %s\n", 111 log_prefix, proc->uloop.pid, data); 112 } else if (len == s->r.buffer_len) { 113 netifd_log_message(L_NOTICE, "%s (%d): %s [...]\n", 114 log_prefix, proc->uloop.pid, data); 115 proc->log_overflow = true; 116 } else 117 break; 118 119 ustream_consume(s, len); 120 } while (1); 121 } 122 123 static void 124 netifd_process_cb(struct uloop_process *proc, int ret) 125 { 126 struct netifd_process *np; 127 np = container_of(proc, struct netifd_process, uloop); 128 129 while (ustream_poll(&np->log.stream)); 130 netifd_delete_process(np); 131 return np->cb(np, ret); 132 } 133 134 int 135 netifd_start_process(const char **argv, char **env, struct netifd_process *proc) 136 { 137 int pfds[2]; 138 int pid; 139 140 netifd_kill_process(proc); 141 142 if (pipe(pfds) < 0) 143 return -1; 144 145 if ((pid = fork()) < 0) 146 goto error; 147 148 if (!pid) { 149 int i; 150 151 if (env) { 152 while (*env) { 153 putenv(*env); 154 env++; 155 } 156 } 157 if (proc->dir_fd >= 0) 158 if (fchdir(proc->dir_fd)) {} 159 160 close(pfds[0]); 161 162 for (i = 0; i <= 2; i++) { 163 if (pfds[1] == i) 164 continue; 165 166 dup2(pfds[1], i); 167 } 168 169 if (pfds[1] > 2) 170 close(pfds[1]); 171 172 execvp(argv[0], (char **) argv); 173 exit(127); 174 } 175 176 close(pfds[1]); 177 proc->uloop.cb = netifd_process_cb; 178 proc->uloop.pid = pid; 179 uloop_process_add(&proc->uloop); 180 list_add_tail(&proc->list, &process_list); 181 182 system_fd_set_cloexec(pfds[0]); 183 proc->log.stream.string_data = true; 184 proc->log.stream.notify_read = netifd_process_log_read_cb; 185 ustream_fd_init(&proc->log, pfds[0]); 186 187 return 0; 188 189 error: 190 close(pfds[0]); 191 close(pfds[1]); 192 return -1; 193 } 194 195 void 196 netifd_kill_process(struct netifd_process *proc) 197 { 198 if (!proc->uloop.pending) 199 return; 200 201 kill(proc->uloop.pid, SIGKILL); 202 uloop_process_delete(&proc->uloop); 203 netifd_delete_process(proc); 204 } 205 206 static void netifd_do_restart(struct uloop_timeout *timeout) 207 { 208 execvp(global_argv[0], global_argv); 209 } 210 211 int netifd_reload(void) 212 { 213 return config_init_all(); 214 } 215 216 void netifd_restart(void) 217 { 218 static struct uloop_timeout main_timer = { 219 .cb = netifd_do_restart 220 }; 221 222 interface_set_down(NULL); 223 uloop_timeout_set(&main_timer, 1000); 224 } 225 226 static int usage(const char *progname) 227 { 228 fprintf(stderr, "Usage: %s [options]\n" 229 "Options:\n" 230 " -d <mask>: Mask for debug messages\n" 231 " -s <path>: Path to the ubus socket\n" 232 " -p <path>: Path to netifd addons (default: %s)\n" 233 " -c <path>: Path to UCI configuration\n" 234 " -h <path>: Path to the hotplug script\n" 235 " -r <path>: Path to resolv.conf\n" 236 " -l <level>: Log output level (default: %d)\n" 237 " -S: Use stderr instead of syslog for log messages\n" 238 " (default: "DEFAULT_HOTPLUG_PATH")\n" 239 "\n", progname, main_path, DEFAULT_LOG_LEVEL); 240 241 return 1; 242 } 243 244 static void 245 netifd_handle_signal(int signo) 246 { 247 uloop_end(); 248 } 249 250 static void 251 netifd_setup_signals(void) 252 { 253 struct sigaction s; 254 255 memset(&s, 0, sizeof(s)); 256 s.sa_handler = netifd_handle_signal; 257 s.sa_flags = 0; 258 sigaction(SIGINT, &s, NULL); 259 sigaction(SIGTERM, &s, NULL); 260 sigaction(SIGUSR1, &s, NULL); 261 sigaction(SIGUSR2, &s, NULL); 262 263 s.sa_handler = SIG_IGN; 264 sigaction(SIGPIPE, &s, NULL); 265 } 266 267 static void 268 netifd_kill_processes(void) 269 { 270 struct netifd_process *proc, *tmp; 271 272 list_for_each_entry_safe(proc, tmp, &process_list, list) 273 netifd_kill_process(proc); 274 } 275 276 int main(int argc, char **argv) 277 { 278 const char *socket = NULL; 279 int ch; 280 281 global_argv = argv; 282 283 while ((ch = getopt(argc, argv, "d:s:p:c:h:r:l:S")) != -1) { 284 switch(ch) { 285 case 'd': 286 debug_mask = strtoul(optarg, NULL, 0); 287 break; 288 case 's': 289 socket = optarg; 290 break; 291 case 'p': 292 main_path = optarg; 293 break; 294 case 'c': 295 config_path = optarg; 296 break; 297 case 'h': 298 hotplug_cmd_path = optarg; 299 break; 300 case 'r': 301 resolv_conf = optarg; 302 break; 303 case 'l': 304 log_level = atoi(optarg); 305 if (log_level >= ARRAY_SIZE(log_class)) 306 log_level = ARRAY_SIZE(log_class) - 1; 307 break; 308 #ifndef DUMMY_MODE 309 case 'S': 310 use_syslog = false; 311 break; 312 #endif 313 default: 314 return usage(argv[0]); 315 } 316 } 317 318 if (use_syslog) 319 openlog("netifd", 0, LOG_DAEMON); 320 321 netifd_setup_signals(); 322 if (netifd_ubus_init(socket) < 0) { 323 fprintf(stderr, "Failed to connect to ubus\n"); 324 return 1; 325 } 326 327 proto_shell_init(); 328 wireless_init(); 329 330 if (system_init()) { 331 fprintf(stderr, "Failed to initialize system control\n"); 332 return 1; 333 } 334 335 config_init_all(); 336 337 uloop_run(); 338 netifd_kill_processes(); 339 340 netifd_ubus_done(); 341 342 if (use_syslog) 343 closelog(); 344 345 return 0; 346 } 347
This page was automatically generated by LXR 0.3.1. • OpenWrt