wireless.c

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