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