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