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