wireless.c

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