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
  15 /* The wireless configuration is projected on the following objects
  16  *
  17  * 1. wireless device
  18  * 2. wireless interface
  19  * 3. wireless vlan
  20  * 4. wireless station
  21  *
  22  * A wireless device is a phy or simplified a wireless card.
  23  * A wireless interface is a SSID on a phy.
  24  * A wireless vlan can be assigned to a wireless interface. A wireless interface can
  25  *   have multiple vlans.
  26  * A wireless station is a client connected to an wireless interface.
  27  */
  28
  29 #include <signal.h>
  30 #include "netifd.h"
  31 #include "wireless.h"
  32 #include "handler.h"
  33 #include "ubus.h"
  34
  35 #define WIRELESS_SETUP_RETRY    3
  36
  37 struct vlist_tree wireless_devices;
  38 struct avl_tree wireless_drivers;
  39 static struct blob_buf b;
  40 static int drv_fd;
  41 static LIST_HEAD(handlers);
  42 static bool handler_pending;
  43
  44 enum {
  45         WDEV_ATTR_DISABLED,
  46         WDEV_ATTR_RECONF,
  47         WDEV_ATTR_SERIALIZE,
  48         __WDEV_ATTR_MAX,
  49 };
  50
  51 static const struct blobmsg_policy wdev_policy[__WDEV_ATTR_MAX] = {
  52         [WDEV_ATTR_DISABLED] = { .name = "disabled", .type = BLOBMSG_TYPE_BOOL },
  53         [WDEV_ATTR_RECONF] = { .name = "reconf", .type = BLOBMSG_TYPE_BOOL },
  54         [WDEV_ATTR_SERIALIZE] = { .name = "serialize", .type = BLOBMSG_TYPE_BOOL },
  55 };
  56
  57 static const struct uci_blob_param_list wdev_param = {
  58         .n_params = ARRAY_SIZE(wdev_policy),
  59         .params = wdev_policy,
  60 };
  61
  62 enum {
  63         VIF_ATTR_DISABLED,
  64         VIF_ATTR_NETWORK,
  65         VIF_ATTR_ISOLATE,
  66         VIF_ATTR_MODE,
  67         __VIF_ATTR_MAX,
  68 };
  69
  70 static const struct blobmsg_policy vif_policy[__VIF_ATTR_MAX] = {
  71         [VIF_ATTR_DISABLED] = { .name = "disabled", .type = BLOBMSG_TYPE_BOOL },
  72         [VIF_ATTR_NETWORK] = { .name = "network", .type = BLOBMSG_TYPE_ARRAY },
  73         [VIF_ATTR_ISOLATE] = { .name = "isolate", .type = BLOBMSG_TYPE_BOOL },
  74         [VIF_ATTR_MODE] = { .name = "mode", .type = BLOBMSG_TYPE_STRING },
  75 };
  76
  77 static const struct uci_blob_param_list vif_param = {
  78         .n_params = ARRAY_SIZE(vif_policy),
  79         .params = vif_policy,
  80 };
  81
  82 enum {
  83         VLAN_ATTR_DISABLED,
  84         VLAN_ATTR_NETWORK,
  85         VLAN_ATTR_ISOLATE,
  86         __VLAN_ATTR_MAX,
  87 };
  88
  89 static const struct blobmsg_policy vlan_policy[__VLAN_ATTR_MAX] = {
  90         [VLAN_ATTR_DISABLED] = { .name = "disabled", .type = BLOBMSG_TYPE_BOOL },
  91         [VLAN_ATTR_NETWORK] = { .name = "network", .type = BLOBMSG_TYPE_ARRAY },
  92         [VLAN_ATTR_ISOLATE] = { .name = "isolate", .type = BLOBMSG_TYPE_BOOL },
  93 };
  94
  95 static const struct uci_blob_param_list vlan_param = {
  96         .n_params = ARRAY_SIZE(vlan_policy),
  97         .params = vlan_policy,
  98 };
  99
 100 enum {
 101         STA_ATTR_DISABLED,
 102         __STA_ATTR_MAX,
 103 };
 104
 105 static const struct blobmsg_policy sta_policy[__STA_ATTR_MAX] = {
 106         [STA_ATTR_DISABLED] = { .name = "disabled", .type = BLOBMSG_TYPE_BOOL },
 107 };
 108
 109 static const struct uci_blob_param_list station_param = {
 110         .n_params = ARRAY_SIZE(sta_policy),
 111         .params = sta_policy,
 112 };
 113
 114 static void
 115 wireless_handler_stop(struct wireless_device *wdev)
 116 {
 117         if (wdev->handler_pending) {
 118                 wdev->handler_pending = false;
 119                 list_del(&wdev->handler);
 120         }
 121 }
 122
 123 static void
 124 put_container(struct blob_buf *buf, struct blob_attr *attr, const char *name)
 125 {
 126         void *c = blobmsg_open_table(buf, name);
 127         blob_put_raw(buf, blob_data(attr), blob_len(attr));
 128         blobmsg_close_table(buf, c);
 129 }
 130
 131 static void
 132 vif_config_add_bridge(struct blob_buf *buf, struct blob_attr *networks, bool prepare)
 133 {
 134         struct interface *iface;
 135         struct device *dev = NULL, *orig_dev;
 136         struct blob_attr *cur;
 137         const char *network;
 138         int rem;
 139
 140         if (!networks)
 141                 return;
 142
 143         blobmsg_for_each_attr(cur, networks, rem) {
 144                 network = blobmsg_data(cur);
 145
 146                 iface = vlist_find(&interfaces, network, iface, node);
 147                 if (!iface)
 148                         continue;
 149
 150                 dev = iface->main_dev.dev;
 151                 if (!dev)
 152                         return;
 153
 154                 if (!dev->hotplug_ops)
 155                         return;
 156         }
 157
 158         if (!dev)
 159                 return;
 160
 161         orig_dev = dev;
 162         if (dev->hotplug_ops && dev->hotplug_ops->prepare)
 163                 dev->hotplug_ops->prepare(dev, &dev);
 164
 165         if (!dev || !dev->type->bridge_capability)
 166                 return;
 167
 168         blobmsg_add_string(buf, "bridge", dev->ifname);
 169         blobmsg_add_string(buf, "bridge-ifname", orig_dev->ifname);
 170
 171         if (dev->settings.flags & DEV_OPT_MULTICAST_TO_UNICAST)
 172                 blobmsg_add_u8(buf, "multicast_to_unicast",
 173                                dev->settings.multicast_to_unicast);
 174 }
 175
 176 static void
 177 prepare_config(struct wireless_device *wdev, struct blob_buf *buf, bool up)
 178 {
 179         struct wireless_interface *vif;
 180         struct wireless_vlan *vlan;
 181         struct wireless_station *sta;
 182         void *l, *i, *j, *k;
 183
 184         blob_buf_init(&b, 0);
 185         put_container(&b, wdev->config, "config");
 186         if (wdev->data)
 187                 blobmsg_add_blob(&b, wdev->data);
 188
 189         l = blobmsg_open_table(&b, "interfaces");
 190         vlist_for_each_element(&wdev->interfaces, vif, node) {
 191                 i = blobmsg_open_table(&b, vif->name);
 192                 vif_config_add_bridge(&b, vif->network, up);
 193                 put_container(&b, vif->config, "config");
 194                 if (vif->data)
 195                         blobmsg_add_blob(&b, vif->data);
 196
 197                 j = blobmsg_open_table(&b, "vlans");
 198                 vlist_for_each_element(&wdev->vlans, vlan, node) {
 199                         if (strcmp(vlan->vif, vif->name))
 200                                 continue;
 201                         k = blobmsg_open_table(&b, vlan->name);
 202                         vif_config_add_bridge(&b, vlan->network, up);
 203                         put_container(&b, vlan->config, "config");
 204                         if (vlan->data)
 205                                 blobmsg_add_blob(&b, vlan->data);
 206                         blobmsg_close_table(&b, k);
 207                 }
 208                 blobmsg_close_table(&b, j);
 209
 210                 j = blobmsg_open_table(&b, "stas");
 211                 vlist_for_each_element(&wdev->stations, sta, node) {
 212                         if (strcmp(sta->vif, vif->name))
 213                                 continue;
 214                         k = blobmsg_open_table(&b, sta->name);
 215                         put_container(&b, sta->config, "config");
 216                         if (sta->data)
 217                                 blobmsg_add_blob(&b, sta->data);
 218                         blobmsg_close_table(&b, k);
 219                 }
 220                 blobmsg_close_table(&b, j);
 221                 blobmsg_close_table(&b, i);
 222         }
 223         blobmsg_close_table(&b, l);
 224
 225 }
 226
 227 static bool
 228 wireless_process_check(struct wireless_process *proc)
 229 {
 230         return check_pid_path(proc->pid, proc->exe);
 231 }
 232
 233 static void
 234 wireless_complete_kill_request(struct wireless_device *wdev)
 235 {
 236         if (!wdev->kill_request)
 237                 return;
 238
 239         ubus_complete_deferred_request(ubus_ctx, wdev->kill_request, 0);
 240         free(wdev->kill_request);
 241         wdev->kill_request = NULL;
 242 }
 243
 244 static void
 245 wireless_process_free(struct wireless_device *wdev, struct wireless_process *proc)
 246 {
 247         D(WIRELESS, "Wireless device '%s' free pid %d\n", wdev->name, proc->pid);
 248         list_del(&proc->list);
 249         free(proc);
 250
 251         if (list_empty(&wdev->script_proc))
 252                 wireless_complete_kill_request(wdev);
 253 }
 254
 255 static void
 256 wireless_close_script_proc_fd(struct wireless_device *wdev)
 257 {
 258         if (wdev->script_proc_fd.fd < 0)
 259                 return;
 260
 261         uloop_fd_delete(&wdev->script_proc_fd);
 262         close(wdev->script_proc_fd.fd);
 263         wdev->script_proc_fd.fd = -1;
 264 }
 265
 266 static void
 267 wireless_process_kill_all(struct wireless_device *wdev, int signal, bool free)
 268 {
 269         struct wireless_process *proc, *tmp;
 270
 271         list_for_each_entry_safe(proc, tmp, &wdev->script_proc, list) {
 272                 bool check = wireless_process_check(proc);
 273
 274                 if (check && !proc->keep) {
 275                         D(WIRELESS, "Wireless device '%s' kill pid %d\n", wdev->name, proc->pid);
 276                         kill(proc->pid, signal);
 277                 }
 278
 279                 if (free || !check)
 280                         wireless_process_free(wdev, proc);
 281         }
 282
 283         if (free)
 284                 wireless_close_script_proc_fd(wdev);
 285 }
 286
 287 static void
 288 wireless_device_free_state(struct wireless_device *wdev)
 289 {
 290         struct wireless_interface *vif;
 291         struct wireless_vlan *vlan;
 292         struct wireless_station *sta;
 293
 294         wireless_handler_stop(wdev);
 295         uloop_timeout_cancel(&wdev->script_check);
 296         uloop_timeout_cancel(&wdev->timeout);
 297         wireless_complete_kill_request(wdev);
 298         free(wdev->data);
 299         wdev->data = NULL;
 300         vlist_for_each_element(&wdev->interfaces, vif, node) {
 301                 free(vif->data);
 302                 vif->data = NULL;
 303                 vif->ifname = NULL;
 304         }
 305         vlist_for_each_element(&wdev->vlans, vlan, node) {
 306                 free(vlan->data);
 307                 vlan->data = NULL;
 308                 vlan->ifname = NULL;
 309         }
 310         vlist_for_each_element(&wdev->stations, sta, node) {
 311                 free(sta->data);
 312                 sta->data = NULL;
 313         }
 314 }
 315
 316 static void wireless_interface_handle_link(struct wireless_interface *vif, const char *ifname, bool up)
 317 {
 318         struct interface *iface;
 319         struct blob_attr *cur;
 320         const char *network;
 321         int rem;
 322
 323         if (!vif->network || !vif->ifname)
 324                 return;
 325
 326         if (!ifname)
 327                 ifname = vif->ifname;
 328
 329         if (up) {
 330                 struct device *dev = device_get(ifname, 2);
 331                 if (dev) {
 332                         dev->wireless_isolate = vif->isolate;
 333                         dev->wireless = true;
 334                         dev->wireless_ap = vif->ap_mode;
 335                         dev->bpdu_filter = dev->wireless_ap && ifname == vif->ifname;
 336                 }
 337         }
 338
 339         blobmsg_for_each_attr(cur, vif->network, rem) {
 340                 network = blobmsg_data(cur);
 341
 342                 iface = vlist_find(&interfaces, network, iface, node);
 343                 if (!iface)
 344                         continue;
 345
 346                 interface_handle_link(iface, ifname, NULL, up, true);
 347         }
 348 }
 349
 350 static void wireless_vlan_handle_link(struct wireless_vlan *vlan, bool up)
 351 {
 352         struct interface *iface;
 353         struct blob_attr *cur;
 354         const char *network;
 355         int rem;
 356
 357         if (!vlan->network || !vlan->ifname)
 358                 return;
 359
 360         if (up) {
 361                 struct device *dev = device_get(vlan->ifname, 2);
 362                 if (dev) {
 363                         dev->wireless_isolate = vlan->isolate;
 364                         dev->wireless = true;
 365                         dev->wireless_ap = true;
 366                         dev->bpdu_filter = true;
 367                 }
 368         }
 369
 370         blobmsg_for_each_attr(cur, vlan->network, rem) {
 371                 network = blobmsg_data(cur);
 372
 373                 iface = vlist_find(&interfaces, network, iface, node);
 374                 if (!iface)
 375                         continue;
 376
 377                 interface_handle_link(iface, vlan->ifname, NULL, up, true);
 378         }
 379 }
 380
 381 static void
 382 wireless_device_setup_cancel(struct wireless_device *wdev)
 383 {
 384         if (wdev->cancel)
 385                 return;
 386
 387         wireless_handler_stop(wdev);
 388         D(WIRELESS, "Cancel wireless device '%s' setup\n", wdev->name);
 389         wdev->cancel = true;
 390         uloop_timeout_set(&wdev->timeout, 10 * 1000);
 391 }
 392
 393 static void
 394 wireless_device_run_handler(struct wireless_device *wdev, bool up)
 395 {
 396         const char *action = up ? "setup" : "teardown";
 397         const char *argv[6];
 398         char *config;
 399         int i = 0;
 400         int fds[2] = { -1, -1 };
 401
 402         wireless_handler_stop(wdev);
 403
 404         if (handler_pending && wdev->serialize) {
 405                 wdev->handler_action = up;
 406                 wdev->handler_pending = true;
 407                 list_add_tail(&wdev->handler, &handlers);
 408                 return;
 409         }
 410         if (wdev->serialize)
 411                 handler_pending = true;
 412
 413         D(WIRELESS, "Wireless device '%s' run %s handler\n", wdev->name, action);
 414         if (!up && wdev->prev_config) {
 415                 config = blobmsg_format_json(wdev->prev_config, true);
 416                 free(wdev->prev_config);
 417                 wdev->prev_config = NULL;
 418         } else {
 419                 prepare_config(wdev, &b, up);
 420                 config = blobmsg_format_json(b.head, true);
 421         }
 422
 423         argv[i++] = wdev->drv->script;
 424         argv[i++] = wdev->drv->name;
 425         argv[i++] = action;
 426         argv[i++] = wdev->name;
 427         argv[i++] = config;
 428         argv[i] = NULL;
 429
 430         if (up && pipe(fds) == 0) {
 431                 wdev->script_proc_fd.fd = fds[0];
 432                 uloop_fd_add(&wdev->script_proc_fd,
 433                              ULOOP_READ | ULOOP_EDGE_TRIGGER);
 434         }
 435
 436         netifd_start_process(argv, NULL, &wdev->script_task);
 437
 438         if (fds[1] >= 0)
 439                 close(fds[1]);
 440
 441         free(config);
 442 }
 443
 444 static void
 445 wireless_handler_next(void)
 446 {
 447         struct wireless_device *wdev;
 448
 449         if (handler_pending)
 450                 return;
 451         if (list_empty(&handlers))
 452                 return;
 453         wdev = list_first_entry(&handlers, struct wireless_device, handler);
 454         list_del(&wdev->handler);
 455         wdev->handler_pending = false;
 456         wireless_device_run_handler(wdev, wdev->handler_action);
 457 }
 458
 459 static void
 460 __wireless_device_set_up(struct wireless_device *wdev, int force)
 461 {
 462         if (wdev->disabled)
 463                 return;
 464
 465         if (wdev->retry_setup_failed)
 466                 return;
 467
 468         if (!wdev->autostart)
 469                 return;
 470
 471         if (!force && (wdev->state != IFS_DOWN || config_init))
 472                 return;
 473
 474         free(wdev->prev_config);
 475         wdev->prev_config = NULL;
 476         wdev->state = IFS_SETUP;
 477         wireless_device_run_handler(wdev, true);
 478 }
 479
 480 static void
 481 wireless_device_free(struct wireless_device *wdev)
 482 {
 483         wireless_handler_stop(wdev);
 484         vlist_flush_all(&wdev->interfaces);
 485         vlist_flush_all(&wdev->vlans);
 486         vlist_flush_all(&wdev->stations);
 487         avl_delete(&wireless_devices.avl, &wdev->node.avl);
 488         free(wdev->config);
 489         free(wdev->prev_config);
 490         free(wdev);
 491 }
 492
 493 static void
 494 wdev_handle_config_change(struct wireless_device *wdev)
 495 {
 496         enum interface_config_state state = wdev->config_state;
 497
 498         switch(state) {
 499         case IFC_NORMAL:
 500         case IFC_RELOAD:
 501                 __wireless_device_set_up(wdev, 0);
 502
 503                 wdev->config_state = IFC_NORMAL;
 504                 break;
 505         case IFC_REMOVE:
 506                 wireless_device_free(wdev);
 507                 break;
 508         }
 509 }
 510
 511 static void
 512 wireless_device_mark_down(struct wireless_device *wdev)
 513 {
 514         struct wireless_interface *vif;
 515         struct wireless_vlan *vlan;
 516
 517         D(WIRELESS, "Wireless device '%s' is now down\n", wdev->name);
 518
 519         vlist_for_each_element(&wdev->vlans, vlan, node)
 520                 wireless_vlan_handle_link(vlan, false);
 521
 522         vlist_for_each_element(&wdev->interfaces, vif, node)
 523                 wireless_interface_handle_link(vif, NULL, false);
 524
 525         wireless_process_kill_all(wdev, SIGTERM, true);
 526
 527         wdev->cancel = false;
 528         wdev->state = IFS_DOWN;
 529         wireless_device_free_state(wdev);
 530         wdev_handle_config_change(wdev);
 531 }
 532
 533 /* timeout callback to protect the tear down */
 534 static void
 535 wireless_device_setup_timeout(struct uloop_timeout *timeout)
 536 {
 537         struct wireless_device *wdev = container_of(timeout, struct wireless_device, timeout);
 538
 539         if (wdev->handler_pending) {
 540                 wdev->handler_pending = false;
 541                 list_del(&wdev->handler);
 542         }
 543         netifd_kill_process(&wdev->script_task);
 544         wdev->script_task.cb(&wdev->script_task, -1);
 545         wireless_device_mark_down(wdev);
 546 }
 547
 548 void
 549 wireless_device_set_up(struct wireless_device *wdev)
 550 {
 551         wdev->retry = WIRELESS_SETUP_RETRY;
 552         wdev->autostart = true;
 553         __wireless_device_set_up(wdev, 0);
 554 }
 555
 556 void
 557 wireless_device_reconf(struct wireless_device *wdev)
 558 {
 559         wdev->retry = WIRELESS_SETUP_RETRY;
 560         wdev->autostart = true;
 561         __wireless_device_set_up(wdev, wdev->reconf && (wdev->state == IFS_UP));
 562 }
 563
 564 static void
 565 __wireless_device_set_down(struct wireless_device *wdev)
 566 {
 567         if (wdev->state == IFS_TEARDOWN || wdev->state == IFS_DOWN)
 568                 return;
 569
 570         if (wdev->script_task.uloop.pending) {
 571                 wireless_device_setup_cancel(wdev);
 572                 return;
 573         }
 574
 575         wdev->state = IFS_TEARDOWN;
 576         wireless_device_run_handler(wdev, false);
 577 }
 578
 579 /* ubus callback network.wireless.notify, command = up */
 580 static void
 581 wireless_device_mark_up(struct wireless_device *wdev)
 582 {
 583         struct wireless_interface *vif;
 584         struct wireless_vlan *vlan;
 585
 586         if (wdev->cancel) {
 587                 wdev->cancel = false;
 588                 __wireless_device_set_down(wdev);
 589                 return;
 590         }
 591
 592         D(WIRELESS, "Wireless device '%s' is now up\n", wdev->name);
 593         wdev->state = IFS_UP;
 594         vlist_for_each_element(&wdev->interfaces, vif, node)
 595                 wireless_interface_handle_link(vif, NULL, true);
 596         vlist_for_each_element(&wdev->vlans, vlan, node)
 597                 wireless_vlan_handle_link(vlan, true);
 598 }
 599
 600 static void
 601 wireless_device_retry_setup(struct wireless_device *wdev)
 602 {
 603         if (wdev->state == IFS_TEARDOWN || wdev->state == IFS_DOWN || wdev->cancel)
 604                 return;
 605
 606         if (--wdev->retry < 0)
 607                 wdev->retry_setup_failed = true;
 608
 609         __wireless_device_set_down(wdev);
 610 }
 611
 612 static void
 613 wireless_device_script_task_cb(struct netifd_process *proc, int ret)
 614 {
 615         struct wireless_device *wdev = container_of(proc, struct wireless_device, script_task);
 616
 617         switch (wdev->state) {
 618         case IFS_SETUP:
 619                 wireless_device_retry_setup(wdev);
 620                 break;
 621         case IFS_TEARDOWN:
 622                 wireless_device_mark_down(wdev);
 623                 break;
 624         default:
 625                 break;
 626         }
 627
 628         if (wdev->serialize) {
 629                 handler_pending = false;
 630                 wireless_handler_next();
 631         }
 632 }
 633
 634 void
 635 wireless_device_set_down(struct wireless_device *wdev)
 636 {
 637         wdev->retry_setup_failed = false;
 638         wdev->autostart = false;
 639         __wireless_device_set_down(wdev);
 640 }
 641
 642 static void
 643 wdev_set_config_state(struct wireless_device *wdev, enum interface_config_state s)
 644 {
 645         if (wdev->config_state != IFC_NORMAL)
 646                 return;
 647
 648         wdev->config_state = s;
 649         if (wdev->state == IFS_DOWN)
 650                 wdev_handle_config_change(wdev);
 651         else if (!wdev->reconf || wdev->state != IFS_UP)
 652                 __wireless_device_set_down(wdev);
 653 }
 654
 655 static void
 656 wdev_prepare_prev_config(struct wireless_device *wdev)
 657 {
 658         if (wdev->prev_config)
 659                 return;
 660
 661         prepare_config(wdev, &b, false);
 662         wdev->prev_config = blob_memdup(b.head);
 663 }
 664
 665 static void
 666 wdev_change_config(struct wireless_device *wdev, struct wireless_device *wd_new)
 667 {
 668         struct blob_attr *new_config = wd_new->config;
 669         bool disabled = wd_new->disabled;
 670
 671         free(wd_new);
 672
 673         wdev_prepare_prev_config(wdev);
 674         if (blob_attr_equal(wdev->config, new_config) && wdev->disabled == disabled)
 675                 return;
 676
 677         D(WIRELESS, "Update configuration of wireless device '%s'\n", wdev->name);
 678         free(wdev->config);
 679         wdev->config = blob_memdup(new_config);
 680         wdev->disabled = disabled;
 681         wdev->retry_setup_failed = false;
 682         wdev_set_config_state(wdev, IFC_RELOAD);
 683 }
 684
 685 static void
 686 wdev_create(struct wireless_device *wdev)
 687 {
 688         wdev->retry = WIRELESS_SETUP_RETRY;
 689         wdev->config = blob_memdup(wdev->config);
 690 }
 691
 692 /* vlist update call for wireless device list */
 693 static void
 694 wdev_update(struct vlist_tree *tree, struct vlist_node *node_new,
 695             struct vlist_node *node_old)
 696 {
 697         struct wireless_device *wd_old = container_of(node_old, struct wireless_device, node);
 698         struct wireless_device *wd_new = container_of(node_new, struct wireless_device, node);
 699
 700         if (wd_old && wd_new) {
 701                 D(WIRELESS, "Update wireless device '%s'\n", wd_old->name);
 702                 wdev_change_config(wd_old, wd_new);
 703         } else if (wd_old) {
 704                 D(WIRELESS, "Delete wireless device '%s'\n", wd_old->name);
 705                 wdev_set_config_state(wd_old, IFC_REMOVE);
 706         } else if (wd_new) {
 707                 D(WIRELESS, "Create wireless device '%s'\n", wd_new->name);
 708                 wdev_create(wd_new);
 709         }
 710 }
 711
 712 /* wireless netifd script handler */
 713 static void
 714 wireless_add_handler(const char *script, const char *name, json_object *obj)
 715 {
 716         struct wireless_driver *drv;
 717         char *name_str, *script_str;
 718         json_object *dev_config_obj, *iface_config_obj, *vlan_config_obj, *station_config_obj;
 719         struct uci_blob_param_list *dev_config, *iface_config, *vlan_config, *station_config;
 720
 721         dev_config_obj = json_get_field(obj, "device", json_type_array);
 722         iface_config_obj = json_get_field(obj, "iface", json_type_array);
 723         vlan_config_obj = json_get_field(obj, "vlan", json_type_array);
 724         station_config_obj = json_get_field(obj, "station", json_type_array);
 725
 726         if (!dev_config_obj || !iface_config_obj || !vlan_config_obj || !station_config_obj)
 727                 return;
 728
 729         drv = calloc_a(sizeof(*drv),
 730                 &dev_config, sizeof(*dev_config) + sizeof(void *),
 731                 &iface_config, sizeof(*iface_config) + sizeof(void *),
 732                 &vlan_config, sizeof(*vlan_config) + sizeof(void *),
 733                 &station_config, sizeof(*station_config) + sizeof(void *),
 734                 &name_str, strlen(name) + 1,
 735                 &script_str, strlen(script) + 1);
 736
 737         drv->name = strcpy(name_str, name);
 738         drv->script = strcpy(script_str, script);
 739
 740         dev_config->n_next = 1;
 741         dev_config->next[0] = &wdev_param;
 742         drv->device.config = dev_config;
 743
 744         iface_config->n_next = 1;
 745         iface_config->next[0] = &vif_param;
 746         drv->interface.config = iface_config;
 747
 748         vlan_config->n_next = 1;
 749         vlan_config->next[0] = &vlan_param;
 750         drv->vlan.config = vlan_config;
 751
 752         station_config->n_next = 1;
 753         station_config->next[0] = &station_param;
 754         drv->station.config = station_config;
 755
 756         drv->device.buf = netifd_handler_parse_config(drv->device.config, dev_config_obj);
 757         drv->interface.buf = netifd_handler_parse_config(drv->interface.config, iface_config_obj);
 758         drv->vlan.buf = netifd_handler_parse_config(drv->vlan.config, vlan_config_obj);
 759         drv->station.buf = netifd_handler_parse_config(drv->station.config, station_config_obj);
 760
 761         drv->node.key = drv->name;
 762         avl_insert(&wireless_drivers, &drv->node);
 763         D(WIRELESS, "Add handler for script %s: %s\n", script, name);
 764 }
 765
 766 void wireless_init(void)
 767 {
 768         vlist_init(&wireless_devices, avl_strcmp, wdev_update);
 769         wireless_devices.keep_old = true;
 770         wireless_devices.no_delete = true;
 771
 772         avl_init(&wireless_drivers, avl_strcmp, false, NULL);
 773         drv_fd = netifd_open_subdir("wireless");
 774         if (drv_fd < 0)
 775                 return;
 776
 777         netifd_init_script_handlers(drv_fd, wireless_add_handler);
 778 }
 779
 780 /* parse blob config into the wireless interface object */
 781 static void
 782 wireless_interface_init_config(struct wireless_interface *vif)
 783 {
 784         struct blob_attr *tb[__VIF_ATTR_MAX];
 785         struct blob_attr *cur;
 786
 787         vif->network = NULL;
 788         blobmsg_parse(vif_policy, __VIF_ATTR_MAX, tb, blob_data(vif->config), blob_len(vif->config));
 789
 790         if ((cur = tb[VIF_ATTR_NETWORK]))
 791                 vif->network = cur;
 792
 793         cur = tb[VIF_ATTR_ISOLATE];
 794         if (cur)
 795                 vif->isolate = blobmsg_get_bool(cur);
 796
 797         cur = tb[VIF_ATTR_MODE];
 798         if (cur)
 799                 vif->ap_mode = !strcmp(blobmsg_get_string(cur), "ap");
 800 }
 801
 802 /* vlist update call for wireless interface list */
 803 static void
 804 vif_update(struct vlist_tree *tree, struct vlist_node *node_new,
 805            struct vlist_node *node_old)
 806 {
 807         struct wireless_interface *vif_old = container_of(node_old, struct wireless_interface, node);
 808         struct wireless_interface *vif_new = container_of(node_new, struct wireless_interface, node);
 809         struct wireless_device *wdev;
 810
 811         if (vif_old)
 812                 wdev = vif_old->wdev;
 813         else
 814                 wdev = vif_new->wdev;
 815
 816         if (vif_old && vif_new) {
 817                 free((void *) vif_old->section);
 818                 vif_old->section = strdup(vif_new->section);
 819                 if (blob_attr_equal(vif_old->config, vif_new->config)) {
 820                         free(vif_new);
 821                         return;
 822                 }
 823
 824                 D(WIRELESS, "Update wireless interface %s on device %s\n", vif_new->name, wdev->name);
 825                 wireless_interface_handle_link(vif_old, NULL, false);
 826                 free(vif_old->config);
 827                 vif_old->config = blob_memdup(vif_new->config);
 828                 vif_old->isolate = vif_new->isolate;
 829                 vif_old->ap_mode = vif_new->ap_mode;
 830                 wireless_interface_init_config(vif_old);
 831                 free(vif_new);
 832         } else if (vif_new) {
 833                 D(WIRELESS, "Create new wireless interface %s on device %s\n", vif_new->name, wdev->name);
 834                 vif_new->section = strdup(vif_new->section);
 835                 vif_new->config = blob_memdup(vif_new->config);
 836                 wireless_interface_init_config(vif_new);
 837         } else if (vif_old) {
 838                 D(WIRELESS, "Delete wireless interface %s on device %s\n", vif_old->name, wdev->name);
 839                 wireless_interface_handle_link(vif_old, NULL, false);
 840                 free((void *) vif_old->section);
 841                 free(vif_old->config);
 842                 free(vif_old);
 843         }
 844
 845         wdev_set_config_state(wdev, IFC_RELOAD);
 846 }
 847
 848 /* parse blob config into the vlan object */
 849 static void
 850 wireless_vlan_init_config(struct wireless_vlan *vlan)
 851 {
 852         struct blob_attr *tb[__VLAN_ATTR_MAX];
 853         struct blob_attr *cur;
 854
 855         vlan->network = NULL;
 856         blobmsg_parse(vlan_policy, __VLAN_ATTR_MAX, tb, blob_data(vlan->config), blob_len(vlan->config));
 857
 858         if ((cur = tb[VLAN_ATTR_NETWORK]))
 859                 vlan->network = cur;
 860
 861         cur = tb[VLAN_ATTR_ISOLATE];
 862         if (cur)
 863                 vlan->isolate = blobmsg_get_bool(cur);
 864 }
 865
 866 /* vlist update call for vlan list */
 867 static void
 868 vlan_update(struct vlist_tree *tree, struct vlist_node *node_new,
 869             struct vlist_node *node_old)
 870 {
 871         struct wireless_vlan *vlan_old = container_of(node_old, struct wireless_vlan, node);
 872         struct wireless_vlan *vlan_new = container_of(node_new, struct wireless_vlan, node);
 873         struct wireless_device *wdev;
 874
 875         if (vlan_old)
 876                 wdev = vlan_old->wdev;
 877         else
 878                 wdev = vlan_new->wdev;
 879
 880         if (vlan_old && vlan_new) {
 881                 free((void *) vlan_old->section);
 882                 vlan_old->section = strdup(vlan_new->section);
 883                 if (blob_attr_equal(vlan_old->config, vlan_new->config)) {
 884                         free(vlan_new);
 885                         return;
 886                 }
 887
 888                 D(WIRELESS, "Update wireless vlan %s on device %s\n", vlan_new->name, wdev->name);
 889                 wireless_vlan_handle_link(vlan_old, false);
 890                 free(vlan_old->config);
 891                 vlan_old->config = blob_memdup(vlan_new->config);
 892                 vlan_old->isolate = vlan_new->isolate;
 893                 wireless_vlan_init_config(vlan_old);
 894                 free(vlan_new);
 895         } else if (vlan_new) {
 896                 D(WIRELESS, "Create new wireless vlan %s on device %s\n", vlan_new->name, wdev->name);
 897                 vlan_new->section = strdup(vlan_new->section);
 898                 vlan_new->config = blob_memdup(vlan_new->config);
 899                 wireless_vlan_init_config(vlan_new);
 900         } else if (vlan_old) {
 901                 D(WIRELESS, "Delete wireless interface %s on device %s\n", vlan_old->name, wdev->name);
 902                 wireless_vlan_handle_link(vlan_old, false);
 903                 free((void *) vlan_old->section);
 904                 free(vlan_old->config);
 905                 free(vlan_old);
 906         }
 907
 908         wdev_set_config_state(wdev, IFC_RELOAD);
 909 }
 910
 911 /* vlist update call for station list */
 912 static void
 913 station_update(struct vlist_tree *tree, struct vlist_node *node_new,
 914                struct vlist_node *node_old)
 915 {
 916         struct wireless_station *sta_old = container_of(node_old, struct wireless_station, node);
 917         struct wireless_station *sta_new = container_of(node_new, struct wireless_station, node);
 918         struct wireless_device *wdev;
 919
 920         if (sta_old)
 921                 wdev = sta_old->wdev;
 922         else
 923                 wdev = sta_new->wdev;
 924
 925         if (sta_old && sta_new) {
 926                 free((void *) sta_old->section);
 927                 sta_old->section = strdup(sta_new->section);
 928                 if (blob_attr_equal(sta_old->config, sta_new->config)) {
 929                         free(sta_new);
 930                         return;
 931                 }
 932
 933                 D(WIRELESS, "Update wireless station %s on device %s\n", sta_new->name, wdev->name);
 934                 free(sta_old->config);
 935                 sta_old->config = blob_memdup(sta_new->config);
 936                 free(sta_new);
 937         } else if (sta_new) {
 938                 D(WIRELESS, "Create new wireless station %s on device %s\n", sta_new->name, wdev->name);
 939                 sta_new->section = strdup(sta_new->section);
 940                 sta_new->config = blob_memdup(sta_new->config);
 941         } else if (sta_old) {
 942                 D(WIRELESS, "Delete wireless station %s on device %s\n", sta_old->name, wdev->name);
 943                 free((void *) sta_old->section);
 944                 free(sta_old->config);
 945                 free(sta_old);
 946         }
 947
 948         wdev_set_config_state(wdev, IFC_RELOAD);
 949 }
 950
 951 static void
 952 wireless_proc_poll_fd(struct uloop_fd *fd, unsigned int events)
 953 {
 954         struct wireless_device *wdev = container_of(fd, struct wireless_device, script_proc_fd);
 955         char buf[128];
 956
 957         while (1) {
 958                 int b = read(fd->fd, buf, sizeof(buf));
 959                 if (b < 0) {
 960                         if (errno == EINTR)
 961                                 continue;
 962
 963                         if (errno == EAGAIN)
 964                                 return;
 965
 966                         goto done;
 967                 }
 968
 969                 if (!b)
 970                         goto done;
 971         }
 972
 973 done:
 974         uloop_timeout_set(&wdev->script_check, 0);
 975         wireless_close_script_proc_fd(wdev);
 976 }
 977
 978 /* watchdog and garbage collector for wireless processes.
 979  * It cleans up terminated processes. If a process is a requirement for the wireless device, it retries the setup */
 980 static void
 981 wireless_device_check_script_tasks(struct uloop_timeout *timeout)
 982 {
 983         struct wireless_device *wdev = container_of(timeout, struct wireless_device, script_check);
 984         struct wireless_process *proc, *tmp;
 985         bool restart = false;
 986
 987         list_for_each_entry_safe(proc, tmp, &wdev->script_proc, list) {
 988                 if (wireless_process_check(proc))
 989                         continue;
 990
 991                 D(WIRELESS, "Wireless device '%s' pid %d has terminated\n", wdev->name, proc->pid);
 992                 if (proc->required)
 993                         restart = true;
 994
 995                 wireless_process_free(wdev, proc);
 996         }
 997
 998         if (restart)
 999                 wireless_device_retry_setup(wdev);
1000         else
1001                 uloop_timeout_set(&wdev->script_check, 1000);
1002 }
1003
1004 /* creates a wireless device object. Called by config */
1005 void
1006 wireless_device_create(struct wireless_driver *drv, const char *name, struct blob_attr *data)
1007 {
1008         struct wireless_device *wdev;
1009         char *name_buf;
1010         struct blob_attr *tb[__WDEV_ATTR_MAX];
1011         struct blob_attr *cur;
1012
1013         blobmsg_parse(wdev_policy, __WDEV_ATTR_MAX, tb, blob_data(data), blob_len(data));
1014
1015         wdev = calloc_a(sizeof(*wdev), &name_buf, strlen(name) + 1);
1016
1017         cur = tb[WDEV_ATTR_DISABLED];
1018         wdev->disabled = cur && blobmsg_get_bool(cur);
1019
1020         wdev->drv = drv;
1021         wdev->state = IFS_DOWN;
1022         wdev->config_state = IFC_NORMAL;
1023         wdev->name = strcpy(name_buf, name);
1024         wdev->config = data;
1025         wdev->handler_pending = false;
1026
1027         cur = tb[WDEV_ATTR_SERIALIZE];
1028         wdev->serialize = cur && blobmsg_get_bool(cur);
1029
1030         cur = tb[WDEV_ATTR_RECONF];
1031         wdev->reconf = cur && blobmsg_get_bool(cur);
1032
1033         wdev->retry_setup_failed = false;
1034         wdev->autostart = true;
1035         INIT_LIST_HEAD(&wdev->script_proc);
1036         vlist_init(&wdev->interfaces, avl_strcmp, vif_update);
1037         wdev->interfaces.keep_old = true;
1038         vlist_init(&wdev->vlans, avl_strcmp, vlan_update);
1039         wdev->vlans.keep_old = true;
1040         vlist_init(&wdev->stations, avl_strcmp, station_update);
1041         wdev->stations.keep_old = true;
1042
1043         wdev->timeout.cb = wireless_device_setup_timeout;
1044         wdev->script_task.cb = wireless_device_script_task_cb;
1045         wdev->script_task.dir_fd = drv_fd;
1046         wdev->script_task.log_prefix = wdev->name;
1047
1048         wdev->script_proc_fd.fd = -1;
1049         wdev->script_proc_fd.cb = wireless_proc_poll_fd;
1050
1051         wdev->script_check.cb = wireless_device_check_script_tasks;
1052
1053         vlist_add(&wireless_devices, &wdev->node, wdev->name);
1054 }
1055
1056 /* creates a wireless station object. Called by config */
1057 void
1058 wireless_station_create(struct wireless_device *wdev, char *vif, struct blob_attr *data, const char *section)
1059 {
1060         struct wireless_station *sta;
1061         struct blob_attr *tb[__STA_ATTR_MAX];
1062         struct blob_attr *cur;
1063         char *name_buf, *vif_buf;
1064         char name[8];
1065
1066         blobmsg_parse(sta_policy, __STA_ATTR_MAX, tb, blob_data(data), blob_len(data));
1067
1068         cur = tb[STA_ATTR_DISABLED];
1069         if (cur && blobmsg_get_bool(cur))
1070                 return;
1071
1072         sprintf(name, "%d", wdev->vlan_idx++);
1073
1074         sta = calloc_a(sizeof(*sta),
1075                        &name_buf, strlen(name) + 1,
1076                        &vif_buf, strlen(vif) + 1);
1077         sta->name = strcpy(name_buf, name);
1078         sta->vif = strcpy(vif_buf, vif);
1079         sta->wdev = wdev;
1080         sta->config = data;
1081         sta->section = section;
1082
1083         vlist_add(&wdev->stations, &sta->node, sta->name);
1084 }
1085
1086 /* ubus callback network.wireless.status, runs for every interface, encode the station */
1087 static void
1088 wireless_station_status(struct wireless_station *sta, struct blob_buf *b)
1089 {
1090         void *i;
1091
1092         i = blobmsg_open_table(b, NULL);
1093         if (sta->section)
1094                 blobmsg_add_string(b, "section", sta->section);
1095         put_container(b, sta->config, "config");
1096         blobmsg_close_table(b, i);
1097 }
1098
1099 /* create a vlan object. Called by config */
1100 void
1101 wireless_vlan_create(struct wireless_device *wdev, char *vif, struct blob_attr *data, const char *section)
1102 {
1103         struct wireless_vlan *vlan;
1104         struct blob_attr *tb[__VLAN_ATTR_MAX];
1105         struct blob_attr *cur;
1106         char *name_buf, *vif_buf;
1107         char name[8];
1108
1109         blobmsg_parse(vlan_policy, __VLAN_ATTR_MAX, tb, blob_data(data), blob_len(data));
1110
1111         cur = tb[VLAN_ATTR_DISABLED];
1112         if (cur && blobmsg_get_bool(cur))
1113                 return;
1114
1115         sprintf(name, "%d", wdev->vlan_idx++);
1116
1117         vlan = calloc_a(sizeof(*vlan),
1118                        &name_buf, strlen(name) + 1,
1119                        &vif_buf, strlen(vif) + 1);
1120         vlan->name = strcpy(name_buf, name);
1121         vlan->vif = strcpy(vif_buf, vif);
1122         vlan->wdev = wdev;
1123         vlan->config = data;
1124         vlan->section = section;
1125         vlan->isolate = false;
1126
1127         vlist_add(&wdev->vlans, &vlan->node, vlan->name);
1128 }
1129
1130 /* ubus callback network.wireless.status, runs for every interface, encode the vlan informations */
1131 static void
1132 wireless_vlan_status(struct wireless_vlan *vlan, struct blob_buf *b)
1133 {
1134         void *i;
1135
1136         i = blobmsg_open_table(b, NULL);
1137         if (vlan->section)
1138                 blobmsg_add_string(b, "section", vlan->section);
1139         if (vlan->ifname)
1140                 blobmsg_add_string(b, "ifname", vlan->ifname);
1141         put_container(b, vlan->config, "config");
1142         blobmsg_close_table(b, i);
1143 }
1144
1145 /* create a wireless interface object. Called by config */
1146 struct wireless_interface* wireless_interface_create(struct wireless_device *wdev, struct blob_attr *data, const char *section)
1147 {
1148         struct wireless_interface *vif;
1149         struct blob_attr *tb[__VIF_ATTR_MAX];
1150         struct blob_attr *cur;
1151         char *name_buf;
1152         char name[8];
1153
1154         blobmsg_parse(vif_policy, __VIF_ATTR_MAX, tb, blob_data(data), blob_len(data));
1155
1156         cur = tb[VIF_ATTR_DISABLED];
1157         if (cur && blobmsg_get_bool(cur))
1158                 return NULL;
1159
1160         sprintf(name, "%d", wdev->vif_idx++);
1161
1162         vif = calloc_a(sizeof(*vif),
1163                        &name_buf, strlen(name) + 1);
1164         vif->name = strcpy(name_buf, name);
1165         vif->wdev = wdev;
1166         vif->config = data;
1167         vif->section = section;
1168         vif->isolate = false;
1169
1170         vlist_add(&wdev->interfaces, &vif->node, vif->name);
1171
1172         return vlist_find(&wdev->interfaces, name, vif, node);
1173 }
1174
1175 /* ubus callback network.wireless.status, runs for every interface */
1176 static void
1177 wireless_interface_status(struct wireless_interface *iface, struct blob_buf *b)
1178 {
1179         struct wireless_vlan *vlan;
1180         struct wireless_station *sta;
1181         void *i, *j;
1182
1183         i = blobmsg_open_table(b, NULL);
1184         if (iface->section)
1185                 blobmsg_add_string(b, "section", iface->section);
1186         if (iface->ifname)
1187                 blobmsg_add_string(b, "ifname", iface->ifname);
1188         put_container(b, iface->config, "config");
1189         j = blobmsg_open_array(b, "vlans");
1190         vlist_for_each_element(&iface->wdev->vlans, vlan, node)
1191                 if (!strcmp(iface->name, vlan->vif))
1192                         wireless_vlan_status(vlan, b);
1193         blobmsg_close_array(b, j);
1194         j = blobmsg_open_array(b, "stations");
1195         vlist_for_each_element(&iface->wdev->stations, sta, node)
1196                 if (!strcmp(iface->name, sta->vif))
1197                         wireless_station_status(sta, b);
1198         blobmsg_close_array(b, j);
1199         blobmsg_close_table(b, i);
1200 }
1201
1202 /* ubus callback network.wireless.status */
1203 void
1204 wireless_device_status(struct wireless_device *wdev, struct blob_buf *b)
1205 {
1206         struct wireless_interface *iface;
1207         void *c, *i;
1208
1209         c = blobmsg_open_table(b, wdev->name);
1210         blobmsg_add_u8(b, "up", wdev->state == IFS_UP);
1211         blobmsg_add_u8(b, "pending", wdev->state == IFS_SETUP || wdev->state == IFS_TEARDOWN);
1212         blobmsg_add_u8(b, "autostart", wdev->autostart);
1213         blobmsg_add_u8(b, "disabled", wdev->disabled);
1214         blobmsg_add_u8(b, "retry_setup_failed", wdev->retry_setup_failed);
1215         put_container(b, wdev->config, "config");
1216
1217         i = blobmsg_open_array(b, "interfaces");
1218         vlist_for_each_element(&wdev->interfaces, iface, node)
1219                 wireless_interface_status(iface, b);
1220         blobmsg_close_array(b, i);
1221         blobmsg_close_table(b, c);
1222 }
1223
1224 /* ubus callback network.wireless.get_validate */
1225 void
1226 wireless_device_get_validate(struct wireless_device *wdev, struct blob_buf *b)
1227 {
1228         struct uci_blob_param_list *p;
1229         void *c, *d;
1230         int i;
1231
1232         c = blobmsg_open_table(b, wdev->name);
1233
1234         d = blobmsg_open_table(b, "device");
1235         p = wdev->drv->device.config;
1236         for (i = 0; i < p->n_params; i++)
1237                 blobmsg_add_string(b, p->params[i].name, uci_get_validate_string(p, i));
1238         blobmsg_close_table(b, d);
1239
1240         d = blobmsg_open_table(b, "interface");
1241         p = wdev->drv->interface.config;
1242         for (i = 0; i < p->n_params; i++)
1243                 blobmsg_add_string(b, p->params[i].name, uci_get_validate_string(p, i));
1244         blobmsg_close_table(b, d);
1245
1246         blobmsg_close_table(b, c);
1247 }
1248
1249 /* ubus callback network.wireless.notify, command = set data, for vif */
1250 static void
1251 wireless_interface_set_data(struct wireless_interface *vif)
1252 {
1253         enum {
1254                 VIF_DATA_IFNAME,
1255                 __VIF_DATA_MAX,
1256         };
1257         static const struct blobmsg_policy data_policy[__VIF_DATA_MAX] = {
1258                 [VIF_DATA_IFNAME] = { .name = "ifname", .type = BLOBMSG_TYPE_STRING },
1259         };
1260         struct blob_attr *tb[__VIF_DATA_MAX];
1261         struct blob_attr *cur;
1262
1263         blobmsg_parse(data_policy, __VIF_DATA_MAX, tb,
1264                       blobmsg_data(vif->data), blobmsg_data_len(vif->data));
1265
1266         if ((cur = tb[VIF_DATA_IFNAME]))
1267                 vif->ifname = blobmsg_data(cur);
1268 }
1269
1270 /* ubus callback network.wireless.notify, command = set data, for vlan */
1271 static void
1272 wireless_vlan_set_data(struct wireless_vlan *vlan)
1273 {
1274         enum {
1275                 VLAN_DATA_IFNAME,
1276                 __VLAN_DATA_MAX,
1277         };
1278         static const struct blobmsg_policy data_policy[__VLAN_DATA_MAX] = {
1279                 [VLAN_DATA_IFNAME] = { .name = "ifname", .type = BLOBMSG_TYPE_STRING },
1280         };
1281         struct blob_attr *tb[__VLAN_DATA_MAX];
1282         struct blob_attr *cur;
1283
1284         blobmsg_parse(data_policy, __VLAN_DATA_MAX, tb,
1285                       blobmsg_data(vlan->data), blobmsg_data_len(vlan->data));
1286
1287         if ((cur = tb[VLAN_DATA_IFNAME]))
1288                 vlan->ifname = blobmsg_data(cur);
1289 }
1290
1291 /* ubus callback network.wireless.notify, command = process add */
1292 static int
1293 wireless_device_add_process(struct wireless_device *wdev, struct blob_attr *data)
1294 {
1295         enum {
1296                 PROC_ATTR_PID,
1297                 PROC_ATTR_EXE,
1298                 PROC_ATTR_REQUIRED,
1299                 PROC_ATTR_KEEP,
1300                 __PROC_ATTR_MAX
1301         };
1302         static const struct blobmsg_policy proc_policy[__PROC_ATTR_MAX] = {
1303                 [PROC_ATTR_PID] = { .name = "pid", .type = BLOBMSG_TYPE_INT32 },
1304                 [PROC_ATTR_EXE] = { .name = "exe", .type = BLOBMSG_TYPE_STRING },
1305                 [PROC_ATTR_REQUIRED] = { .name = "required", .type = BLOBMSG_TYPE_BOOL },
1306                 [PROC_ATTR_KEEP] = { .name = "keep", .type = BLOBMSG_TYPE_BOOL },
1307         };
1308         struct blob_attr *tb[__PROC_ATTR_MAX];
1309         struct wireless_process *proc;
1310         char *name;
1311         int pid;
1312
1313         if (!data)
1314                 return UBUS_STATUS_INVALID_ARGUMENT;
1315
1316         blobmsg_parse(proc_policy, __PROC_ATTR_MAX, tb, blobmsg_data(data), blobmsg_data_len(data));
1317         if (!tb[PROC_ATTR_PID] || !tb[PROC_ATTR_EXE])
1318                 return UBUS_STATUS_INVALID_ARGUMENT;
1319
1320         pid = blobmsg_get_u32(tb[PROC_ATTR_PID]);
1321         if (pid < 2)
1322                 return UBUS_STATUS_INVALID_ARGUMENT;
1323
1324         proc = calloc_a(sizeof(*proc),
1325                 &name, strlen(blobmsg_data(tb[PROC_ATTR_EXE])) + 1);
1326
1327         proc->pid = pid;
1328         proc->exe = strcpy(name, blobmsg_data(tb[PROC_ATTR_EXE]));
1329
1330         if (tb[PROC_ATTR_REQUIRED])
1331                 proc->required = blobmsg_get_bool(tb[PROC_ATTR_REQUIRED]);
1332
1333         if (tb[PROC_ATTR_KEEP])
1334                 proc->keep = blobmsg_get_bool(tb[PROC_ATTR_KEEP]);
1335
1336         D(WIRELESS, "Wireless device '%s' add pid %d\n", wdev->name, proc->pid);
1337         list_add(&proc->list, &wdev->script_proc);
1338         uloop_timeout_set(&wdev->script_check, 0);
1339
1340         return 0;
1341 }
1342
1343 /* ubus callback network.wireless.notify, command = process kill all */
1344 static int
1345 wireless_device_process_kill_all(struct wireless_device *wdev, struct blob_attr *data,
1346                                  struct ubus_request_data *req)
1347 {
1348         enum {
1349                 KILL_ATTR_SIGNAL,
1350                 KILL_ATTR_IMMEDIATE,
1351                 __KILL_ATTR_MAX
1352         };
1353         static const struct blobmsg_policy kill_policy[__KILL_ATTR_MAX] = {
1354                 [KILL_ATTR_SIGNAL] = { .name = "signal", .type = BLOBMSG_TYPE_INT32 },
1355                 [KILL_ATTR_IMMEDIATE] = { .name = "immediate", .type = BLOBMSG_TYPE_BOOL },
1356         };
1357         struct blob_attr *tb[__KILL_ATTR_MAX];
1358         struct blob_attr *cur;
1359         bool immediate = false;
1360         int signal = SIGTERM;
1361
1362         blobmsg_parse(kill_policy, __KILL_ATTR_MAX, tb, blobmsg_data(data), blobmsg_data_len(data));
1363
1364         if ((cur = tb[KILL_ATTR_SIGNAL]))
1365                 signal = blobmsg_get_u32(cur);
1366
1367         if ((cur = tb[KILL_ATTR_IMMEDIATE]))
1368                 immediate = blobmsg_get_bool(cur);
1369
1370         if (wdev->state != IFS_TEARDOWN || wdev->kill_request)
1371                 return UBUS_STATUS_PERMISSION_DENIED;
1372
1373         wireless_process_kill_all(wdev, signal, immediate);
1374
1375         if (list_empty(&wdev->script_proc))
1376                 return 0;
1377
1378         wdev->kill_request = calloc(1, sizeof(*wdev->kill_request));
1379         ubus_defer_request(ubus_ctx, req, wdev->kill_request);
1380
1381         return 0;
1382 }
1383
1384 /* ubus callback network.wireless.notify, command = set_retry */
1385 static int
1386 wireless_device_set_retry(struct wireless_device *wdev, struct blob_attr *data)
1387 {
1388         static const struct blobmsg_policy retry_policy = {
1389                 .name = "retry", .type = BLOBMSG_TYPE_INT32
1390         };
1391         struct blob_attr *val;
1392
1393         blobmsg_parse(&retry_policy, 1, &val, blobmsg_data(data), blobmsg_data_len(data));
1394         if (!val)
1395                 return UBUS_STATUS_INVALID_ARGUMENT;
1396
1397         wdev->retry = blobmsg_get_u32(val);
1398         return 0;
1399 }
1400
1401 enum {
1402         NOTIFY_CMD_UP = 0,
1403         NOTIFY_CMD_SET_DATA = 1,
1404         NOTIFY_CMD_PROCESS_ADD = 2,
1405         NOTIFY_CMD_PROCESS_KILL_ALL = 3,
1406         NOTIFY_CMD_SET_RETRY = 4,
1407 };
1408
1409 /* ubus callback network.wireless.notify */
1410 int
1411 wireless_device_notify(struct wireless_device *wdev, struct blob_attr *data,
1412                        struct ubus_request_data *req)
1413 {
1414         enum {
1415                 NOTIFY_ATTR_COMMAND,
1416                 NOTIFY_ATTR_VIF,
1417                 NOTIFY_ATTR_VLAN,
1418                 NOTIFY_ATTR_DATA,
1419                 __NOTIFY_MAX,
1420         };
1421         static const struct blobmsg_policy notify_policy[__NOTIFY_MAX] = {
1422                 [NOTIFY_ATTR_COMMAND] = { .name = "command", .type = BLOBMSG_TYPE_INT32 },
1423                 [NOTIFY_ATTR_VIF] = { .name = "interface", .type = BLOBMSG_TYPE_STRING },
1424                 [NOTIFY_ATTR_VLAN] = { .name = "vlan", .type = BLOBMSG_TYPE_STRING },
1425                 [NOTIFY_ATTR_DATA] = { .name = "data", .type = BLOBMSG_TYPE_TABLE },
1426         };
1427         struct wireless_interface *vif = NULL;
1428         struct wireless_vlan *vlan = NULL;
1429         struct blob_attr *tb[__NOTIFY_MAX];
1430         struct blob_attr *cur, **pdata;
1431
1432         blobmsg_parse(notify_policy, __NOTIFY_MAX, tb, blob_data(data), blob_len(data));
1433
1434         if (!tb[NOTIFY_ATTR_COMMAND])
1435                 return UBUS_STATUS_INVALID_ARGUMENT;
1436
1437         if ((cur = tb[NOTIFY_ATTR_VIF]) != NULL) {
1438                 vif = vlist_find(&wdev->interfaces, blobmsg_data(cur), vif, node);
1439                 if (!vif)
1440                         return UBUS_STATUS_NOT_FOUND;
1441         }
1442
1443         if ((cur = tb[NOTIFY_ATTR_VLAN]) != NULL) {
1444                 vlan = vlist_find(&wdev->vlans, blobmsg_data(cur), vlan, node);
1445                 if (!vlan)
1446                         return UBUS_STATUS_NOT_FOUND;
1447         }
1448
1449         cur = tb[NOTIFY_ATTR_DATA];
1450         if (!cur)
1451                 return UBUS_STATUS_INVALID_ARGUMENT;
1452
1453         switch (blobmsg_get_u32(tb[NOTIFY_ATTR_COMMAND])) {
1454         case NOTIFY_CMD_UP:
1455                 if (vif || vlan)
1456                         return UBUS_STATUS_INVALID_ARGUMENT;
1457
1458                 if (wdev->state != IFS_SETUP)
1459                         return UBUS_STATUS_PERMISSION_DENIED;
1460
1461                 wireless_device_mark_up(wdev);
1462                 break;
1463         case NOTIFY_CMD_SET_DATA:
1464                 if (vif)
1465                         pdata = &vif->data;
1466                 else if (vlan)
1467                         pdata = &vlan->data;
1468                 else
1469                         pdata = &wdev->data;
1470
1471                 if (*pdata)
1472                         return UBUS_STATUS_INVALID_ARGUMENT;
1473
1474                 *pdata = blob_memdup(cur);
1475                 if (vif)
1476                         wireless_interface_set_data(vif);
1477                 else if (vlan)
1478                         wireless_vlan_set_data(vlan);
1479                 break;
1480         case NOTIFY_CMD_PROCESS_ADD:
1481                 return wireless_device_add_process(wdev, cur);
1482         case NOTIFY_CMD_PROCESS_KILL_ALL:
1483                 return wireless_device_process_kill_all(wdev, cur, req);
1484         case NOTIFY_CMD_SET_RETRY:
1485                 return wireless_device_set_retry(wdev, cur);
1486         default:
1487                 return UBUS_STATUS_INVALID_ARGUMENT;
1488         }
1489
1490         return 0;
1491 }
1492
1493 /* called on startup and by netifd reload() */
1494 void
1495 wireless_start_pending(void)
1496 {
1497         struct wireless_device *wdev;
1498
1499         vlist_for_each_element(&wireless_devices, wdev, node)
1500                 __wireless_device_set_up(wdev, 0);
1501 }
1502
1503 void wireless_device_hotplug_event(const char *name, bool add)
1504 {
1505         struct wireless_interface *vif;
1506         struct wireless_device *wdev;
1507         const char *s;
1508         int len;
1509
1510         s = strstr(name, ".sta");
1511         if (s) {
1512                 if (strchr(s + 4, '.'))
1513                         return;
1514
1515                 len = s - name;
1516         } else if (!device_find(name)) {
1517                 len = strlen(name);
1518         } else {
1519                 return;
1520         }
1521
1522         vlist_for_each_element(&wireless_devices, wdev, node) {
1523                 vlist_for_each_element(&wdev->interfaces, vif, node) {
1524                         if (!vif->ifname)
1525                                 continue;
1526
1527                         if (strlen(vif->ifname) != len ||
1528                             strncmp(vif->ifname, name, len) != 0)
1529                                 continue;
1530
1531                         wireless_interface_handle_link(vif, name, add);
1532                 }
1533         }
1534 }