bridge: Fix multicast_to_unicast feature by hairpin+isolate
[project/netifd.git] / wireless.c
1 /*
2 * netifd - network interface daemon
3 * Copyright (C) 2013 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 <signal.h>
15 #include "netifd.h"
16 #include "wireless.h"
17 #include "handler.h"
18 #include "ubus.h"
19
20 #define WIRELESS_SETUP_RETRY 3
21
22 struct vlist_tree wireless_devices;
23 struct avl_tree wireless_drivers;
24 static struct blob_buf b;
25 static int drv_fd;
26
27 static const struct blobmsg_policy wdev_policy =
28 { .name = "disabled", .type = BLOBMSG_TYPE_BOOL };
29
30 static const struct uci_blob_param_list wdev_param = {
31 .n_params = 1,
32 .params = &wdev_policy,
33 };
34
35 enum {
36 VIF_ATTR_DISABLED,
37 VIF_ATTR_NETWORK,
38 VIF_ATTR_ISOLATE,
39 __VIF_ATTR_MAX,
40 };
41
42 static const struct blobmsg_policy vif_policy[__VIF_ATTR_MAX] = {
43 [VIF_ATTR_DISABLED] = { .name = "disabled", .type = BLOBMSG_TYPE_BOOL },
44 [VIF_ATTR_NETWORK] = { .name = "network", .type = BLOBMSG_TYPE_ARRAY },
45 [VIF_ATTR_ISOLATE] = { .name = "isolate", .type = BLOBMSG_TYPE_BOOL },
46 };
47
48 static const struct uci_blob_param_list vif_param = {
49 .n_params = ARRAY_SIZE(vif_policy),
50 .params = vif_policy,
51 };
52
53 static void
54 put_container(struct blob_buf *buf, struct blob_attr *attr, const char *name)
55 {
56 void *c = blobmsg_open_table(buf, name);
57 blob_put_raw(buf, blob_data(attr), blob_len(attr));
58 blobmsg_close_table(buf, c);
59 }
60
61 static void
62 vif_config_add_bridge(struct blob_buf *buf, struct blob_attr *networks, bool prepare)
63 {
64 struct interface *iface;
65 struct device *dev = NULL;
66 struct blob_attr *cur;
67 const char *network;
68 int rem;
69
70 if (!networks)
71 return;
72
73 blobmsg_for_each_attr(cur, networks, rem) {
74 network = blobmsg_data(cur);
75
76 iface = vlist_find(&interfaces, network, iface, node);
77 if (!iface)
78 continue;
79
80 dev = iface->main_dev.dev;
81 if (!dev)
82 return;
83
84 if (dev->type != &bridge_device_type)
85 return;
86 }
87
88 if (!dev)
89 return;
90
91 if (dev->hotplug_ops && dev->hotplug_ops->prepare)
92 dev->hotplug_ops->prepare(dev);
93
94 blobmsg_add_string(buf, "bridge", dev->ifname);
95 }
96
97 static void
98 prepare_config(struct wireless_device *wdev, struct blob_buf *buf, bool up)
99 {
100 struct wireless_interface *vif;
101 void *l, *i;
102
103 blob_buf_init(&b, 0);
104 put_container(&b, wdev->config, "config");
105 if (wdev->data)
106 blobmsg_add_blob(&b, wdev->data);
107
108 l = blobmsg_open_table(&b, "interfaces");
109 vlist_for_each_element(&wdev->interfaces, vif, node) {
110 i = blobmsg_open_table(&b, vif->name);
111 vif_config_add_bridge(&b, vif->network, up);
112 put_container(&b, vif->config, "config");
113 if (vif->data)
114 blobmsg_add_blob(&b, vif->data);
115 blobmsg_close_table(&b, i);
116 }
117 blobmsg_close_table(&b, l);
118 }
119
120 static bool
121 wireless_process_check(struct wireless_process *proc)
122 {
123 return check_pid_path(proc->pid, proc->exe);
124 }
125
126 static void
127 wireless_complete_kill_request(struct wireless_device *wdev)
128 {
129 if (!wdev->kill_request)
130 return;
131
132 ubus_complete_deferred_request(ubus_ctx, wdev->kill_request, 0);
133 free(wdev->kill_request);
134 wdev->kill_request = NULL;
135 }
136
137 static void
138 wireless_process_free(struct wireless_device *wdev, struct wireless_process *proc)
139 {
140 D(WIRELESS, "Wireless device '%s' free pid %d\n", wdev->name, proc->pid);
141 list_del(&proc->list);
142 free(proc);
143
144 if (list_empty(&wdev->script_proc))
145 wireless_complete_kill_request(wdev);
146 }
147
148 static void
149 wireless_close_script_proc_fd(struct wireless_device *wdev)
150 {
151 if (wdev->script_proc_fd.fd < 0)
152 return;
153
154 uloop_fd_delete(&wdev->script_proc_fd);
155 close(wdev->script_proc_fd.fd);
156 wdev->script_proc_fd.fd = -1;
157 }
158
159 static void
160 wireless_process_kill_all(struct wireless_device *wdev, int signal, bool free)
161 {
162 struct wireless_process *proc, *tmp;
163
164 list_for_each_entry_safe(proc, tmp, &wdev->script_proc, list) {
165 bool check = wireless_process_check(proc);
166
167 if (check) {
168 D(WIRELESS, "Wireless device '%s' kill pid %d\n", wdev->name, proc->pid);
169 kill(proc->pid, signal);
170 }
171
172 if (free || !check)
173 wireless_process_free(wdev, proc);
174 }
175
176 if (free)
177 wireless_close_script_proc_fd(wdev);
178 }
179
180 static void
181 wireless_device_free_state(struct wireless_device *wdev)
182 {
183 struct wireless_interface *vif;
184
185 uloop_timeout_cancel(&wdev->script_check);
186 uloop_timeout_cancel(&wdev->timeout);
187 wireless_complete_kill_request(wdev);
188 free(wdev->data);
189 wdev->data = NULL;
190 vlist_for_each_element(&wdev->interfaces, vif, node) {
191 free(vif->data);
192 vif->data = NULL;
193 vif->ifname = NULL;
194 }
195 }
196
197 static void wireless_interface_handle_link(struct wireless_interface *vif, bool up)
198 {
199 struct interface *iface;
200 struct blob_attr *cur;
201 const char *network;
202 int rem;
203
204 if (!vif->network || !vif->ifname)
205 return;
206
207 if (up) {
208 struct device *dev = device_get(vif->ifname, 2);
209 if (dev) {
210 dev->wireless_isolate = vif->isolate;
211 dev->wireless = true;
212 }
213 }
214
215 blobmsg_for_each_attr(cur, vif->network, rem) {
216 network = blobmsg_data(cur);
217
218 iface = vlist_find(&interfaces, network, iface, node);
219 if (!iface)
220 continue;
221
222 interface_handle_link(iface, vif->ifname, up, true);
223 }
224 }
225
226 static void
227 wireless_device_setup_cancel(struct wireless_device *wdev)
228 {
229 if (wdev->cancel)
230 return;
231
232 D(WIRELESS, "Cancel wireless device '%s' setup\n", wdev->name);
233 wdev->cancel = true;
234 uloop_timeout_set(&wdev->timeout, 10 * 1000);
235 }
236
237 static void
238 wireless_device_run_handler(struct wireless_device *wdev, bool up)
239 {
240 const char *action = up ? "setup" : "teardown";
241 const char *argv[6];
242 char *config;
243 int i = 0;
244 int fds[2] = { -1, -1 };
245
246 D(WIRELESS, "Wireless device '%s' run %s handler\n", wdev->name, action);
247 if (!up && wdev->prev_config) {
248 config = blobmsg_format_json(wdev->prev_config, true);
249 free(wdev->prev_config);
250 wdev->prev_config = NULL;
251 } else {
252 prepare_config(wdev, &b, up);
253 config = blobmsg_format_json(b.head, true);
254 }
255
256 argv[i++] = wdev->drv->script;
257 argv[i++] = wdev->drv->name;
258 argv[i++] = action;
259 argv[i++] = wdev->name;
260 argv[i++] = config;
261 argv[i] = NULL;
262
263 if (up && pipe(fds) == 0) {
264 wdev->script_proc_fd.fd = fds[0];
265 uloop_fd_add(&wdev->script_proc_fd,
266 ULOOP_READ | ULOOP_EDGE_TRIGGER);
267 }
268
269 netifd_start_process(argv, NULL, &wdev->script_task);
270
271 if (fds[1] >= 0)
272 close(fds[1]);
273
274 free(config);
275 }
276
277 static void
278 __wireless_device_set_up(struct wireless_device *wdev)
279 {
280 if (wdev->disabled)
281 return;
282
283 if (wdev->state != IFS_DOWN || config_init)
284 return;
285
286 free(wdev->prev_config);
287 wdev->prev_config = NULL;
288 wdev->state = IFS_SETUP;
289 wireless_device_run_handler(wdev, true);
290 }
291
292 static void
293 wireless_device_free(struct wireless_device *wdev)
294 {
295 vlist_flush_all(&wdev->interfaces);
296 avl_delete(&wireless_devices.avl, &wdev->node.avl);
297 free(wdev->config);
298 free(wdev->prev_config);
299 free(wdev);
300 }
301
302 static void
303 wdev_handle_config_change(struct wireless_device *wdev)
304 {
305 enum interface_config_state state = wdev->config_state;
306
307 switch(state) {
308 case IFC_NORMAL:
309 case IFC_RELOAD:
310 wdev->config_state = IFC_NORMAL;
311 if (wdev->autostart)
312 __wireless_device_set_up(wdev);
313 break;
314 case IFC_REMOVE:
315 wireless_device_free(wdev);
316 break;
317 }
318 }
319
320 static void
321 wireless_device_mark_down(struct wireless_device *wdev)
322 {
323 struct wireless_interface *vif;
324
325 D(WIRELESS, "Wireless device '%s' is now down\n", wdev->name);
326
327 vlist_for_each_element(&wdev->interfaces, vif, node)
328 wireless_interface_handle_link(vif, false);
329
330 wireless_process_kill_all(wdev, SIGTERM, true);
331
332 wdev->cancel = false;
333 wdev->state = IFS_DOWN;
334 wireless_device_free_state(wdev);
335 wdev_handle_config_change(wdev);
336 }
337
338 static void
339 wireless_device_setup_timeout(struct uloop_timeout *timeout)
340 {
341 struct wireless_device *wdev = container_of(timeout, struct wireless_device, timeout);
342
343 netifd_kill_process(&wdev->script_task);
344 wdev->script_task.cb(&wdev->script_task, -1);
345 wireless_device_mark_down(wdev);
346 }
347
348 void
349 wireless_device_set_up(struct wireless_device *wdev)
350 {
351 wdev->retry = WIRELESS_SETUP_RETRY;
352 wdev->autostart = true;
353 __wireless_device_set_up(wdev);
354 }
355
356 static void
357 __wireless_device_set_down(struct wireless_device *wdev)
358 {
359 if (wdev->state == IFS_TEARDOWN || wdev->state == IFS_DOWN)
360 return;
361
362 if (wdev->script_task.uloop.pending) {
363 wireless_device_setup_cancel(wdev);
364 return;
365 }
366
367 wdev->state = IFS_TEARDOWN;
368 wireless_device_run_handler(wdev, false);
369 }
370
371 static void
372 wireless_device_mark_up(struct wireless_device *wdev)
373 {
374 struct wireless_interface *vif;
375
376 if (wdev->cancel) {
377 wdev->cancel = false;
378 __wireless_device_set_down(wdev);
379 return;
380 }
381
382 D(WIRELESS, "Wireless device '%s' is now up\n", wdev->name);
383 wdev->state = IFS_UP;
384 vlist_for_each_element(&wdev->interfaces, vif, node)
385 wireless_interface_handle_link(vif, true);
386 }
387
388 static void
389 wireless_device_retry_setup(struct wireless_device *wdev)
390 {
391 if (wdev->state == IFS_TEARDOWN || wdev->state == IFS_DOWN || wdev->cancel)
392 return;
393
394 if (--wdev->retry < 0)
395 wdev->autostart = false;
396
397 __wireless_device_set_down(wdev);
398 }
399
400 static void
401 wireless_device_script_task_cb(struct netifd_process *proc, int ret)
402 {
403 struct wireless_device *wdev = container_of(proc, struct wireless_device, script_task);
404
405 switch (wdev->state) {
406 case IFS_SETUP:
407 wireless_device_retry_setup(wdev);
408 break;
409 case IFS_TEARDOWN:
410 wireless_device_mark_down(wdev);
411 break;
412 default:
413 break;
414 }
415 }
416
417 void
418 wireless_device_set_down(struct wireless_device *wdev)
419 {
420 wdev->autostart = false;
421 __wireless_device_set_down(wdev);
422 }
423
424 static void
425 wdev_set_config_state(struct wireless_device *wdev, enum interface_config_state s)
426 {
427 if (wdev->config_state != IFC_NORMAL)
428 return;
429
430 wdev->config_state = s;
431 if (wdev->state == IFS_DOWN)
432 wdev_handle_config_change(wdev);
433 else
434 __wireless_device_set_down(wdev);
435 }
436
437 static void
438 wdev_prepare_prev_config(struct wireless_device *wdev)
439 {
440 if (wdev->prev_config)
441 return;
442
443 prepare_config(wdev, &b, false);
444 wdev->prev_config = blob_memdup(b.head);
445 }
446
447 static void
448 wdev_change_config(struct wireless_device *wdev, struct wireless_device *wd_new)
449 {
450 struct blob_attr *new_config = wd_new->config;
451 bool disabled = wd_new->disabled;
452
453 free(wd_new);
454
455 wdev_prepare_prev_config(wdev);
456 if (blob_attr_equal(wdev->config, new_config) && wdev->disabled == disabled)
457 return;
458
459 D(WIRELESS, "Update configuration of wireless device '%s'\n", wdev->name);
460 free(wdev->config);
461 wdev->config = blob_memdup(new_config);
462 wdev->disabled = disabled;
463 wdev_set_config_state(wdev, IFC_RELOAD);
464 }
465
466 static void
467 wdev_create(struct wireless_device *wdev)
468 {
469 wdev->retry = WIRELESS_SETUP_RETRY;
470 wdev->config = blob_memdup(wdev->config);
471 }
472
473 static void
474 wdev_update(struct vlist_tree *tree, struct vlist_node *node_new,
475 struct vlist_node *node_old)
476 {
477 struct wireless_device *wd_old = container_of(node_old, struct wireless_device, node);
478 struct wireless_device *wd_new = container_of(node_new, struct wireless_device, node);
479
480 if (wd_old && wd_new) {
481 wdev_change_config(wd_old, wd_new);
482 } else if (wd_old) {
483 D(WIRELESS, "Delete wireless device '%s'\n", wd_old->name);
484 wdev_set_config_state(wd_old, IFC_REMOVE);
485 } else if (wd_new) {
486 D(WIRELESS, "Create wireless device '%s'\n", wd_new->name);
487 wdev_create(wd_new);
488 }
489 }
490
491 static void
492 wireless_add_handler(const char *script, const char *name, json_object *obj)
493 {
494 struct wireless_driver *drv;
495 char *name_str, *script_str;
496 json_object *dev_config_obj, *iface_config_obj;
497 struct uci_blob_param_list *dev_config, *iface_config;
498
499 dev_config_obj = json_get_field(obj, "device", json_type_array);
500 iface_config_obj = json_get_field(obj, "iface", json_type_array);
501
502 if (!dev_config_obj || !iface_config_obj)
503 return;
504
505 drv = calloc_a(sizeof(*drv),
506 &dev_config, sizeof(*dev_config) + sizeof(void *),
507 &iface_config, sizeof(*iface_config) + sizeof(void *),
508 &name_str, strlen(name) + 1,
509 &script_str, strlen(script) + 1);
510
511 drv->name = strcpy(name_str, name);
512 drv->script = strcpy(script_str, script);
513
514 dev_config->n_next = 1;
515 dev_config->next[0] = &wdev_param;
516 drv->device.config = dev_config;
517
518 iface_config->n_next = 1;
519 iface_config->next[0] = &vif_param;
520 drv->interface.config = iface_config;
521
522 drv->device.buf = netifd_handler_parse_config(drv->device.config, dev_config_obj);
523 drv->interface.buf = netifd_handler_parse_config(drv->interface.config, iface_config_obj);
524
525 drv->node.key = drv->name;
526 avl_insert(&wireless_drivers, &drv->node);
527 D(WIRELESS, "Add handler for script %s: %s\n", script, name);
528 }
529
530 void wireless_init(void)
531 {
532 vlist_init(&wireless_devices, avl_strcmp, wdev_update);
533 wireless_devices.keep_old = true;
534 wireless_devices.no_delete = true;
535
536 avl_init(&wireless_drivers, avl_strcmp, false, NULL);
537 drv_fd = netifd_open_subdir("wireless");
538 if (drv_fd < 0)
539 return;
540
541 netifd_init_script_handlers(drv_fd, wireless_add_handler);
542 }
543
544 static void
545 wireless_interface_init_config(struct wireless_interface *vif)
546 {
547 struct blob_attr *tb[__VIF_ATTR_MAX];
548 struct blob_attr *cur;
549
550 vif->network = NULL;
551 blobmsg_parse(vif_policy, __VIF_ATTR_MAX, tb, blob_data(vif->config), blob_len(vif->config));
552
553 if ((cur = tb[VIF_ATTR_NETWORK]))
554 vif->network = cur;
555 }
556
557 static void
558 vif_update(struct vlist_tree *tree, struct vlist_node *node_new,
559 struct vlist_node *node_old)
560 {
561 struct wireless_interface *vif_old = container_of(node_old, struct wireless_interface, node);
562 struct wireless_interface *vif_new = container_of(node_new, struct wireless_interface, node);
563 struct wireless_device *wdev;
564
565 if (vif_old)
566 wdev = vif_old->wdev;
567 else
568 wdev = vif_new->wdev;
569
570 if (vif_old && vif_new) {
571 free((void *) vif_old->section);
572 vif_old->section = strdup(vif_new->section);
573 if (blob_attr_equal(vif_old->config, vif_new->config)) {
574 free(vif_new);
575 return;
576 }
577
578 D(WIRELESS, "Update wireless interface %s on device %s\n", vif_new->name, wdev->name);
579 free(vif_old->config);
580 vif_old->config = blob_memdup(vif_new->config);
581 wireless_interface_init_config(vif_old);
582 free(vif_new);
583 } else if (vif_new) {
584 D(WIRELESS, "Create new wireless interface %s on device %s\n", vif_new->name, wdev->name);
585 vif_new->section = strdup(vif_new->section);
586 vif_new->config = blob_memdup(vif_new->config);
587 wireless_interface_init_config(vif_new);
588 } else if (vif_old) {
589 D(WIRELESS, "Delete wireless interface %s on device %s\n", vif_old->name, wdev->name);
590 free((void *) vif_old->section);
591 free(vif_old->config);
592 free(vif_old);
593 }
594
595 wdev_set_config_state(wdev, IFC_RELOAD);
596 }
597
598 static void
599 wireless_proc_poll_fd(struct uloop_fd *fd, unsigned int events)
600 {
601 struct wireless_device *wdev = container_of(fd, struct wireless_device, script_proc_fd);
602 char buf[128];
603
604 while (1) {
605 int b = read(fd->fd, buf, sizeof(buf));
606 if (b < 0) {
607 if (errno == EINTR)
608 continue;
609
610 if (errno == EAGAIN)
611 return;
612
613 goto done;
614 }
615
616 if (!b)
617 goto done;
618 }
619
620 done:
621 uloop_timeout_set(&wdev->script_check, 0);
622 wireless_close_script_proc_fd(wdev);
623 }
624
625 static void
626 wireless_device_check_script_tasks(struct uloop_timeout *timeout)
627 {
628 struct wireless_device *wdev = container_of(timeout, struct wireless_device, script_check);
629 struct wireless_process *proc, *tmp;
630 bool restart = false;
631
632 list_for_each_entry_safe(proc, tmp, &wdev->script_proc, list) {
633 if (wireless_process_check(proc))
634 continue;
635
636 D(WIRELESS, "Wireless device '%s' pid %d has terminated\n", wdev->name, proc->pid);
637 if (proc->required)
638 restart = true;
639
640 wireless_process_free(wdev, proc);
641 }
642
643 if (restart)
644 wireless_device_retry_setup(wdev);
645 else
646 uloop_timeout_set(&wdev->script_check, 1000);
647 }
648
649 void
650 wireless_device_create(struct wireless_driver *drv, const char *name, struct blob_attr *data)
651 {
652 struct wireless_device *wdev;
653 char *name_buf;
654 struct blob_attr *disabled;
655
656 blobmsg_parse(&wdev_policy, 1, &disabled, blob_data(data), blob_len(data));
657
658 wdev = calloc_a(sizeof(*wdev), &name_buf, strlen(name) + 1);
659 if (disabled && blobmsg_get_bool(disabled))
660 wdev->disabled = true;
661 wdev->drv = drv;
662 wdev->state = IFS_DOWN;
663 wdev->config_state = IFC_NORMAL;
664 wdev->name = strcpy(name_buf, name);
665 wdev->config = data;
666 wdev->config_autostart = true;
667 wdev->autostart = wdev->config_autostart;
668 INIT_LIST_HEAD(&wdev->script_proc);
669 vlist_init(&wdev->interfaces, avl_strcmp, vif_update);
670 wdev->interfaces.keep_old = true;
671
672 wdev->timeout.cb = wireless_device_setup_timeout;
673 wdev->script_task.cb = wireless_device_script_task_cb;
674 wdev->script_task.dir_fd = drv_fd;
675 wdev->script_task.log_prefix = wdev->name;
676
677 wdev->script_proc_fd.fd = -1;
678 wdev->script_proc_fd.cb = wireless_proc_poll_fd;
679
680 wdev->script_check.cb = wireless_device_check_script_tasks;
681
682 vlist_add(&wireless_devices, &wdev->node, wdev->name);
683 }
684
685 void wireless_interface_create(struct wireless_device *wdev, struct blob_attr *data, const char *section)
686 {
687 struct wireless_interface *vif;
688 struct blob_attr *tb[__VIF_ATTR_MAX];
689 struct blob_attr *cur;
690 char *name_buf;
691 char name[8];
692
693 blobmsg_parse(vif_policy, __VIF_ATTR_MAX, tb, blob_data(data), blob_len(data));
694
695 cur = tb[VIF_ATTR_DISABLED];
696 if (cur && blobmsg_get_bool(cur))
697 return;
698
699 sprintf(name, "%d", wdev->vif_idx++);
700
701 vif = calloc_a(sizeof(*vif),
702 &name_buf, strlen(name) + 1);
703 vif->name = strcpy(name_buf, name);
704 vif->wdev = wdev;
705 vif->config = data;
706 vif->section = section;
707 vif->isolate = false;
708
709 cur = tb[VIF_ATTR_ISOLATE];
710 if (cur && blobmsg_get_bool(cur))
711 vif->isolate = blobmsg_get_bool(cur);
712
713 vlist_add(&wdev->interfaces, &vif->node, vif->name);
714 }
715
716 static void
717 wireless_interface_status(struct wireless_interface *iface, struct blob_buf *b)
718 {
719 void *i;
720
721 i = blobmsg_open_table(b, NULL);
722 if (iface->section)
723 blobmsg_add_string(b, "section", iface->section);
724 if (iface->ifname)
725 blobmsg_add_string(b, "ifname", iface->ifname);
726 put_container(b, iface->config, "config");
727 blobmsg_close_table(b, i);
728 }
729
730 void
731 wireless_device_status(struct wireless_device *wdev, struct blob_buf *b)
732 {
733 struct wireless_interface *iface;
734 void *c, *i;
735
736 c = blobmsg_open_table(b, wdev->name);
737 blobmsg_add_u8(b, "up", wdev->state == IFS_UP);
738 blobmsg_add_u8(b, "pending", wdev->state == IFS_SETUP || wdev->state == IFS_TEARDOWN);
739 blobmsg_add_u8(b, "autostart", wdev->autostart);
740 blobmsg_add_u8(b, "disabled", wdev->disabled);
741 put_container(b, wdev->config, "config");
742
743 i = blobmsg_open_array(b, "interfaces");
744 vlist_for_each_element(&wdev->interfaces, iface, node)
745 wireless_interface_status(iface, b);
746 blobmsg_close_array(b, i);
747 blobmsg_close_table(b, c);
748 }
749
750 void
751 wireless_device_get_validate(struct wireless_device *wdev, struct blob_buf *b)
752 {
753 struct uci_blob_param_list *p;
754 void *c, *d;
755 int i;
756
757 c = blobmsg_open_table(b, wdev->name);
758
759 d = blobmsg_open_table(b, "device");
760 p = wdev->drv->device.config;
761 for (i = 0; i < p->n_params; i++)
762 blobmsg_add_string(b, p->params[i].name, uci_get_validate_string(p, i));
763 blobmsg_close_table(b, d);
764
765 d = blobmsg_open_table(b, "interface");
766 p = wdev->drv->interface.config;
767 for (i = 0; i < p->n_params; i++)
768 blobmsg_add_string(b, p->params[i].name, uci_get_validate_string(p, i));
769 blobmsg_close_table(b, d);
770
771 blobmsg_close_table(b, c);
772 }
773
774 static void
775 wireless_interface_set_data(struct wireless_interface *vif)
776 {
777 enum {
778 VIF_DATA_IFNAME,
779 __VIF_DATA_MAX,
780 };
781 static const struct blobmsg_policy data_policy[__VIF_DATA_MAX] = {
782 [VIF_DATA_IFNAME] = { .name = "ifname", .type = BLOBMSG_TYPE_STRING },
783 };
784 struct blob_attr *tb[__VIF_DATA_MAX];
785 struct blob_attr *cur;
786
787 blobmsg_parse(data_policy, __VIF_DATA_MAX, tb,
788 blobmsg_data(vif->data), blobmsg_data_len(vif->data));
789
790 if ((cur = tb[VIF_DATA_IFNAME]))
791 vif->ifname = blobmsg_data(cur);
792 }
793
794 static int
795 wireless_device_add_process(struct wireless_device *wdev, struct blob_attr *data)
796 {
797 enum {
798 PROC_ATTR_PID,
799 PROC_ATTR_EXE,
800 PROC_ATTR_REQUIRED,
801 __PROC_ATTR_MAX
802 };
803 static const struct blobmsg_policy proc_policy[__PROC_ATTR_MAX] = {
804 [PROC_ATTR_PID] = { .name = "pid", .type = BLOBMSG_TYPE_INT32 },
805 [PROC_ATTR_EXE] = { .name = "exe", .type = BLOBMSG_TYPE_STRING },
806 [PROC_ATTR_REQUIRED] = { .name = "required", .type = BLOBMSG_TYPE_BOOL },
807 };
808 struct blob_attr *tb[__PROC_ATTR_MAX];
809 struct wireless_process *proc;
810 char *name;
811 int pid;
812
813 if (!data)
814 return UBUS_STATUS_INVALID_ARGUMENT;
815
816 blobmsg_parse(proc_policy, __PROC_ATTR_MAX, tb, blobmsg_data(data), blobmsg_data_len(data));
817 if (!tb[PROC_ATTR_PID] || !tb[PROC_ATTR_EXE])
818 return UBUS_STATUS_INVALID_ARGUMENT;
819
820 pid = blobmsg_get_u32(tb[PROC_ATTR_PID]);
821 if (pid < 2)
822 return UBUS_STATUS_INVALID_ARGUMENT;
823
824 proc = calloc_a(sizeof(*proc),
825 &name, strlen(blobmsg_data(tb[PROC_ATTR_EXE])) + 1);
826
827 proc->pid = pid;
828 proc->exe = strcpy(name, blobmsg_data(tb[PROC_ATTR_EXE]));
829
830 if (tb[PROC_ATTR_REQUIRED])
831 proc->required = blobmsg_get_bool(tb[PROC_ATTR_REQUIRED]);
832
833 D(WIRELESS, "Wireless device '%s' add pid %d\n", wdev->name, proc->pid);
834 list_add(&proc->list, &wdev->script_proc);
835 uloop_timeout_set(&wdev->script_check, 0);
836
837 return 0;
838 }
839
840 static int
841 wireless_device_process_kill_all(struct wireless_device *wdev, struct blob_attr *data,
842 struct ubus_request_data *req)
843 {
844 enum {
845 KILL_ATTR_SIGNAL,
846 KILL_ATTR_IMMEDIATE,
847 __KILL_ATTR_MAX
848 };
849 static const struct blobmsg_policy kill_policy[__KILL_ATTR_MAX] = {
850 [KILL_ATTR_SIGNAL] = { .name = "signal", .type = BLOBMSG_TYPE_INT32 },
851 [KILL_ATTR_IMMEDIATE] = { .name = "immediate", .type = BLOBMSG_TYPE_BOOL },
852 };
853 struct blob_attr *tb[__KILL_ATTR_MAX];
854 struct blob_attr *cur;
855 bool immediate = false;
856 int signal = SIGTERM;
857
858 blobmsg_parse(kill_policy, __KILL_ATTR_MAX, tb, blobmsg_data(data), blobmsg_data_len(data));
859
860 if ((cur = tb[KILL_ATTR_SIGNAL]))
861 signal = blobmsg_get_u32(cur);
862
863 if ((cur = tb[KILL_ATTR_IMMEDIATE]))
864 immediate = blobmsg_get_bool(cur);
865
866 if (wdev->state != IFS_TEARDOWN || wdev->kill_request)
867 return UBUS_STATUS_PERMISSION_DENIED;
868
869 wireless_process_kill_all(wdev, signal, immediate);
870
871 if (list_empty(&wdev->script_proc))
872 return 0;
873
874 wdev->kill_request = calloc(1, sizeof(*wdev->kill_request));
875 ubus_defer_request(ubus_ctx, req, wdev->kill_request);
876
877 return 0;
878 }
879
880 static int
881 wireless_device_set_retry(struct wireless_device *wdev, struct blob_attr *data)
882 {
883 static const struct blobmsg_policy retry_policy = {
884 .name = "retry", .type = BLOBMSG_TYPE_INT32
885 };
886 struct blob_attr *val;
887
888 blobmsg_parse(&retry_policy, 1, &val, blobmsg_data(data), blobmsg_data_len(data));
889 if (!val)
890 return UBUS_STATUS_INVALID_ARGUMENT;
891
892 wdev->retry = blobmsg_get_u32(val);
893 return 0;
894 }
895
896 enum {
897 NOTIFY_CMD_UP = 0,
898 NOTIFY_CMD_SET_DATA = 1,
899 NOTIFY_CMD_PROCESS_ADD = 2,
900 NOTIFY_CMD_PROCESS_KILL_ALL = 3,
901 NOTIFY_CMD_SET_RETRY = 4,
902 };
903
904 int
905 wireless_device_notify(struct wireless_device *wdev, struct blob_attr *data,
906 struct ubus_request_data *req)
907 {
908 enum {
909 NOTIFY_ATTR_COMMAND,
910 NOTIFY_ATTR_VIF,
911 NOTIFY_ATTR_DATA,
912 __NOTIFY_MAX,
913 };
914 static const struct blobmsg_policy notify_policy[__NOTIFY_MAX] = {
915 [NOTIFY_ATTR_COMMAND] = { .name = "command", .type = BLOBMSG_TYPE_INT32 },
916 [NOTIFY_ATTR_VIF] = { .name = "interface", .type = BLOBMSG_TYPE_STRING },
917 [NOTIFY_ATTR_DATA] = { .name = "data", .type = BLOBMSG_TYPE_TABLE },
918 };
919 struct wireless_interface *vif = NULL;
920 struct blob_attr *tb[__NOTIFY_MAX];
921 struct blob_attr *cur, **pdata;
922
923 blobmsg_parse(notify_policy, __NOTIFY_MAX, tb, blob_data(data), blob_len(data));
924
925 if (!tb[NOTIFY_ATTR_COMMAND])
926 return UBUS_STATUS_INVALID_ARGUMENT;
927
928 if ((cur = tb[NOTIFY_ATTR_VIF]) != NULL) {
929 vif = vlist_find(&wdev->interfaces, blobmsg_data(cur), vif, node);
930 if (!vif)
931 return UBUS_STATUS_NOT_FOUND;
932 }
933
934 cur = tb[NOTIFY_ATTR_DATA];
935 if (!cur)
936 return UBUS_STATUS_INVALID_ARGUMENT;
937
938 switch (blobmsg_get_u32(tb[NOTIFY_ATTR_COMMAND])) {
939 case NOTIFY_CMD_UP:
940 if (vif)
941 return UBUS_STATUS_INVALID_ARGUMENT;
942
943 if (wdev->state != IFS_SETUP)
944 return UBUS_STATUS_PERMISSION_DENIED;
945
946 wireless_device_mark_up(wdev);
947 break;
948 case NOTIFY_CMD_SET_DATA:
949 if (vif)
950 pdata = &vif->data;
951 else
952 pdata = &wdev->data;
953
954 if (*pdata)
955 return UBUS_STATUS_INVALID_ARGUMENT;
956
957 *pdata = blob_memdup(cur);
958 if (vif)
959 wireless_interface_set_data(vif);
960 break;
961 case NOTIFY_CMD_PROCESS_ADD:
962 return wireless_device_add_process(wdev, cur);
963 case NOTIFY_CMD_PROCESS_KILL_ALL:
964 return wireless_device_process_kill_all(wdev, cur, req);
965 case NOTIFY_CMD_SET_RETRY:
966 return wireless_device_set_retry(wdev, cur);
967 default:
968 return UBUS_STATUS_INVALID_ARGUMENT;
969 }
970
971 return 0;
972 }
973
974 void
975 wireless_start_pending(void)
976 {
977 struct wireless_device *wdev;
978
979 vlist_for_each_element(&wireless_devices, wdev, node)
980 if (wdev->autostart)
981 __wireless_device_set_up(wdev);
982 }