linux: unify functions to delete links
[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 }