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