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