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