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