ubus.c

   1 /*
   2  *   This program is free software; you can redistribute it and/or modify
   3  *   it under the terms of the GNU General Public License as published by
   4  *   the Free Software Foundation; either version 2 of the License.
   5  *
   6  *   This program is distributed in the hope that it will be useful,
   7  *   but WITHOUT ANY WARRANTY; without even the implied warranty of
   8  *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
   9  *   GNU General Public License for more details.
  10  *
  11  *   You should have received a copy of the GNU General Public License
  12  *   along with this program; if not, write to the Free Software
  13  *   Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307, USA.
  14  *
  15  *   Copyright (C) 2020 embedd.ch
  16  *   Copyright (C) 2020 Felix Fietkau <nbd@nbd.name>
  17  *   Copyright (C) 2020 John Crispin <john@phrozen.org>
  18  */
  19
  20 #include <sys/types.h>
  21 #include <sys/socket.h>
  22 #include <net/ethernet.h>
  23 #ifdef linux
  24 #include <netinet/ether.h>
  25 #endif
  26
  27 #include "usteer.h"
  28 #include "node.h"
  29 #include "event.h"
  30
  31 static struct blob_buf b;
  32 static KVLIST(host_info, kvlist_blob_len);
  33
  34 static int
  35 usteer_ubus_get_clients(struct ubus_context *ctx, struct ubus_object *obj,
  36                        struct ubus_request_data *req, const char *method,
  37                        struct blob_attr *msg)
  38 {
  39         struct sta_info *si;
  40         struct sta *sta;
  41         char str[20];
  42         void *_s, *_cur_n;
  43
  44         blob_buf_init(&b, 0);
  45         avl_for_each_element(&stations, sta, avl) {
  46                 sprintf(str, MAC_ADDR_FMT, MAC_ADDR_DATA(sta->addr));
  47                 _s = blobmsg_open_table(&b, str);
  48                 list_for_each_entry(si, &sta->nodes, list) {
  49                         _cur_n = blobmsg_open_table(&b, usteer_node_name(si->node));
  50                         blobmsg_add_u8(&b, "connected", si->connected);
  51                         blobmsg_add_u32(&b, "signal", si->signal);
  52                         blobmsg_close_table(&b, _cur_n);
  53                 }
  54                 blobmsg_close_table(&b, _s);
  55         }
  56         ubus_send_reply(ctx, req, b.head);
  57         return 0;
  58 }
  59
  60 static struct blobmsg_policy client_arg[] = {
  61         { .name = "address", .type = BLOBMSG_TYPE_STRING, },
  62 };
  63
  64 static void
  65 usteer_ubus_add_stats(struct sta_info_stats *stats, const char *name)
  66 {
  67         void *s;
  68
  69         s = blobmsg_open_table(&b, name);
  70         blobmsg_add_u32(&b, "requests", stats->requests);
  71         blobmsg_add_u32(&b, "blocked_cur", stats->blocked_cur);
  72         blobmsg_add_u32(&b, "blocked_total", stats->blocked_total);
  73         blobmsg_close_table(&b, s);
  74 }
  75
  76 static int
  77 usteer_ubus_get_client_info(struct ubus_context *ctx, struct ubus_object *obj,
  78                            struct ubus_request_data *req, const char *method,
  79                            struct blob_attr *msg)
  80 {
  81         struct sta_info *si;
  82         struct sta *sta;
  83         struct blob_attr *mac_str;
  84         uint8_t *mac;
  85         void *_n, *_cur_n, *_s;
  86         int i;
  87
  88         blobmsg_parse(client_arg, 1, &mac_str, blob_data(msg), blob_len(msg));
  89         if (!mac_str)
  90                 return UBUS_STATUS_INVALID_ARGUMENT;
  91
  92         mac = (uint8_t *) ether_aton(blobmsg_data(mac_str));
  93         if (!mac)
  94                 return UBUS_STATUS_INVALID_ARGUMENT;
  95
  96         sta = usteer_sta_get(mac, false);
  97         if (!sta)
  98                 return UBUS_STATUS_NOT_FOUND;
  99
 100         blob_buf_init(&b, 0);
 101         blobmsg_add_u8(&b, "2ghz", sta->seen_2ghz);
 102         blobmsg_add_u8(&b, "5ghz", sta->seen_5ghz);
 103         _n = blobmsg_open_table(&b, "nodes");
 104         list_for_each_entry(si, &sta->nodes, list) {
 105                 _cur_n = blobmsg_open_table(&b, usteer_node_name(si->node));
 106                 blobmsg_add_u8(&b, "connected", si->connected);
 107                 blobmsg_add_u32(&b, "signal", si->signal);
 108                 _s = blobmsg_open_table(&b, "stats");
 109                 for (i = 0; i < __EVENT_TYPE_MAX; i++)
 110                         usteer_ubus_add_stats(&si->stats[EVENT_TYPE_PROBE], event_types[i]);
 111                 blobmsg_close_table(&b, _s);
 112                 blobmsg_close_table(&b, _cur_n);
 113         }
 114         blobmsg_close_table(&b, _n);
 115
 116         ubus_send_reply(ctx, req, b.head);
 117
 118         return 0;
 119 }
 120
 121 enum cfg_type {
 122         CFG_BOOL,
 123         CFG_I32,
 124         CFG_U32,
 125         CFG_ARRAY_CB,
 126         CFG_STRING_CB,
 127 };
 128
 129 struct cfg_item {
 130         enum cfg_type type;
 131         union {
 132                 bool *BOOL;
 133                 uint32_t *U32;
 134                 int32_t *I32;
 135                 struct {
 136                         void (*set)(struct blob_attr *data);
 137                         void (*get)(struct blob_buf *buf);
 138                 } CB;
 139         } ptr;
 140 };
 141
 142 #define __config_items \
 143         _cfg(BOOL, syslog), \
 144         _cfg(U32, debug_level), \
 145         _cfg(BOOL, ipv6), \
 146         _cfg(U32, sta_block_timeout), \
 147         _cfg(U32, local_sta_timeout), \
 148         _cfg(U32, local_sta_update), \
 149         _cfg(U32, max_neighbor_reports), \
 150         _cfg(U32, max_retry_band), \
 151         _cfg(U32, seen_policy_timeout), \
 152         _cfg(U32, load_balancing_threshold), \
 153         _cfg(U32, band_steering_threshold), \
 154         _cfg(U32, remote_update_interval), \
 155         _cfg(BOOL, assoc_steering), \
 156         _cfg(I32, min_connect_snr), \
 157         _cfg(I32, min_snr), \
 158         _cfg(U32, roam_process_timeout), \
 159         _cfg(I32, roam_scan_snr), \
 160         _cfg(U32, roam_scan_tries), \
 161         _cfg(U32, roam_scan_interval), \
 162         _cfg(I32, roam_trigger_snr), \
 163         _cfg(U32, roam_trigger_interval), \
 164         _cfg(U32, roam_kick_delay), \
 165         _cfg(U32, signal_diff_threshold), \
 166         _cfg(U32, initial_connect_delay), \
 167         _cfg(BOOL, load_kick_enabled), \
 168         _cfg(U32, load_kick_threshold), \
 169         _cfg(U32, load_kick_delay), \
 170         _cfg(U32, load_kick_min_clients), \
 171         _cfg(U32, load_kick_reason_code), \
 172         _cfg(ARRAY_CB, interfaces), \
 173         _cfg(STRING_CB, node_up_script), \
 174         _cfg(ARRAY_CB, event_log_types), \
 175         _cfg(ARRAY_CB, ssid_list)
 176
 177 enum cfg_items {
 178 #define _cfg(_type, _name) CFG_##_name
 179         __config_items,
 180 #undef _cfg
 181         __CFG_MAX,
 182 };
 183
 184 static const struct blobmsg_policy config_policy[__CFG_MAX] = {
 185 #define _cfg_policy(_type, _name) [CFG_##_name] = { .name = #_name, .type = BLOBMSG_TYPE_ ## _type }
 186 #define _cfg_policy_BOOL(_name) _cfg_policy(BOOL, _name)
 187 #define _cfg_policy_U32(_name) _cfg_policy(INT32, _name)
 188 #define _cfg_policy_I32(_name) _cfg_policy(INT32, _name)
 189 #define _cfg_policy_ARRAY_CB(_name) _cfg_policy(ARRAY, _name)
 190 #define _cfg_policy_STRING_CB(_name) _cfg_policy(STRING, _name)
 191 #define _cfg(_type, _name) _cfg_policy_##_type(_name)
 192         __config_items,
 193 #undef _cfg
 194 };
 195
 196 static const struct cfg_item config_data[__CFG_MAX] = {
 197 #define _cfg_data_BOOL(_name) .ptr.BOOL = &config._name
 198 #define _cfg_data_U32(_name) .ptr.U32 = &config._name
 199 #define _cfg_data_I32(_name) .ptr.I32 = &config._name
 200 #define _cfg_data_ARRAY_CB(_name) .ptr.CB = { .set = config_set_##_name, .get = config_get_##_name }
 201 #define _cfg_data_STRING_CB(_name) .ptr.CB = { .set = config_set_##_name, .get = config_get_##_name }
 202 #define _cfg(_type, _name) [CFG_##_name] = { .type = CFG_##_type, _cfg_data_##_type(_name) }
 203         __config_items,
 204 #undef _cfg
 205 };
 206
 207 static int
 208 usteer_ubus_get_config(struct ubus_context *ctx, struct ubus_object *obj,
 209                       struct ubus_request_data *req, const char *method,
 210                       struct blob_attr *msg)
 211 {
 212         int i;
 213
 214         blob_buf_init(&b, 0);
 215         for (i = 0; i < __CFG_MAX; i++) {
 216                 switch(config_data[i].type) {
 217                 case CFG_BOOL:
 218                         blobmsg_add_u8(&b, config_policy[i].name,
 219                                         *config_data[i].ptr.BOOL);
 220                         break;
 221                 case CFG_I32:
 222                 case CFG_U32:
 223                         blobmsg_add_u32(&b, config_policy[i].name,
 224                                         *config_data[i].ptr.U32);
 225                         break;
 226                 case CFG_ARRAY_CB:
 227                 case CFG_STRING_CB:
 228                         config_data[i].ptr.CB.get(&b);
 229                         break;
 230                 }
 231         }
 232         ubus_send_reply(ctx, req, b.head);
 233         return 0;
 234 }
 235
 236 static int
 237 usteer_ubus_set_config(struct ubus_context *ctx, struct ubus_object *obj,
 238                       struct ubus_request_data *req, const char *method,
 239                       struct blob_attr *msg)
 240 {
 241         struct blob_attr *tb[__CFG_MAX];
 242         int i;
 243
 244         if (!strcmp(method, "set_config"))
 245                 usteer_init_defaults();
 246
 247         blobmsg_parse(config_policy, __CFG_MAX, tb, blob_data(msg), blob_len(msg));
 248         for (i = 0; i < __CFG_MAX; i++) {
 249                 switch(config_data[i].type) {
 250                 case CFG_BOOL:
 251                         if (!tb[i])
 252                                 continue;
 253
 254                         *config_data[i].ptr.BOOL = blobmsg_get_u8(tb[i]);
 255                         break;
 256                 case CFG_I32:
 257                 case CFG_U32:
 258                         if (!tb[i])
 259                                 continue;
 260
 261                         *config_data[i].ptr.U32 = blobmsg_get_u32(tb[i]);
 262                         break;
 263                 case CFG_ARRAY_CB:
 264                 case CFG_STRING_CB:
 265                         config_data[i].ptr.CB.set(tb[i]);
 266                         break;
 267                 }
 268         }
 269
 270         usteer_interface_init();
 271
 272         return 0;
 273 }
 274
 275 void usteer_dump_node(struct blob_buf *buf, struct usteer_node *node)
 276 {
 277         void *c;
 278
 279         c = blobmsg_open_table(buf, usteer_node_name(node));
 280         blobmsg_printf(buf, "bssid", MAC_ADDR_FMT, MAC_ADDR_DATA(node->bssid));
 281         blobmsg_add_u32(buf, "freq", node->freq);
 282         blobmsg_add_u32(buf, "n_assoc", node->n_assoc);
 283         blobmsg_add_u32(buf, "noise", node->noise);
 284         blobmsg_add_u32(buf, "load", node->load);
 285         blobmsg_add_u32(buf, "max_assoc", node->max_assoc);
 286         blobmsg_add_u32(buf, "roam_source", node->roam_source);
 287         blobmsg_add_u32(buf, "roam_destination", node->roam_destination);
 288         if (node->rrm_nr)
 289                 blobmsg_add_field(buf, BLOBMSG_TYPE_ARRAY, "rrm_nr",
 290                                   blobmsg_data(node->rrm_nr),
 291                                   blobmsg_data_len(node->rrm_nr));
 292         if (node->node_info)
 293                 blobmsg_add_field(buf, BLOBMSG_TYPE_TABLE, "node_info",
 294                                   blob_data(node->node_info),
 295                                   blob_len(node->node_info));
 296
 297         blobmsg_close_table(buf, c);
 298 }
 299
 300 void usteer_dump_host(struct blob_buf *buf, struct usteer_remote_host *host)
 301 {
 302         void *c;
 303
 304         c = blobmsg_open_table(buf, host->addr);
 305         blobmsg_add_u32(buf, "id", (uint32_t)(uintptr_t)host->avl.key);
 306         if (host->host_info)
 307                 blobmsg_add_field(buf, BLOBMSG_TYPE_TABLE, "host_info",
 308                                   blobmsg_data(host->host_info),
 309                                   blobmsg_len(host->host_info));
 310         blobmsg_close_table(buf, c);
 311 }
 312
 313 static int
 314 usteer_ubus_local_info(struct ubus_context *ctx, struct ubus_object *obj,
 315                       struct ubus_request_data *req, const char *method,
 316                       struct blob_attr *msg)
 317 {
 318         struct usteer_node *node;
 319
 320         blob_buf_init(&b, 0);
 321
 322         for_each_local_node(node)
 323                 usteer_dump_node(&b, node);
 324
 325         ubus_send_reply(ctx, req, b.head);
 326
 327         return 0;
 328 }
 329
 330 static int
 331 usteer_ubus_remote_hosts(struct ubus_context *ctx, struct ubus_object *obj,
 332                          struct ubus_request_data *req, const char *method,
 333                          struct blob_attr *msg)
 334 {
 335         struct usteer_remote_host *host;
 336
 337         blob_buf_init(&b, 0);
 338
 339         avl_for_each_element(&remote_hosts, host, avl)
 340                 usteer_dump_host(&b, host);
 341
 342         ubus_send_reply(ctx, req, b.head);
 343
 344         return 0;
 345 }
 346
 347 static int
 348 usteer_ubus_remote_info(struct ubus_context *ctx, struct ubus_object *obj,
 349                        struct ubus_request_data *req, const char *method,
 350                        struct blob_attr *msg)
 351 {
 352         struct usteer_remote_node *rn;
 353
 354         blob_buf_init(&b, 0);
 355
 356         for_each_remote_node(rn)
 357                 usteer_dump_node(&b, &rn->node);
 358
 359         ubus_send_reply(ctx, req, b.head);
 360
 361         return 0;
 362 }
 363
 364 enum {
 365         NODE_DATA_NODE,
 366         NODE_DATA_VALUES,
 367         __NODE_DATA_MAX,
 368 };
 369
 370 static const struct blobmsg_policy set_node_data_policy[] = {
 371         [NODE_DATA_NODE] = { "node", BLOBMSG_TYPE_STRING },
 372         [NODE_DATA_VALUES] = { "data", BLOBMSG_TYPE_TABLE },
 373 };
 374
 375 static const struct blobmsg_policy del_node_data_policy[] = {
 376         [NODE_DATA_NODE] = { "node", BLOBMSG_TYPE_STRING },
 377         [NODE_DATA_VALUES] = { "names", BLOBMSG_TYPE_ARRAY },
 378 };
 379
 380 static void
 381 usteer_update_kvlist_data(struct kvlist *kv, struct blob_attr *data,
 382                           bool delete)
 383 {
 384         struct blob_attr *cur;
 385         int rem;
 386
 387         blobmsg_for_each_attr(cur, data, rem) {
 388                 if (delete)
 389                         kvlist_delete(kv, blobmsg_get_string(cur));
 390                 else
 391                         kvlist_set(kv, blobmsg_name(cur), cur);
 392         }
 393 }
 394
 395 static void
 396 usteer_update_kvlist_blob(struct blob_attr **dest, struct kvlist *kv)
 397 {
 398         struct blob_attr *val;
 399         const char *name;
 400
 401         blob_buf_init(&b, 0);
 402         kvlist_for_each(kv, name, val)
 403                 blobmsg_add_field(&b, blobmsg_type(val), name,
 404                                   blobmsg_data(val), blobmsg_len(val));
 405
 406         val = b.head;
 407         if (!blobmsg_len(val))
 408                 val = NULL;
 409
 410         usteer_node_set_blob(dest, val);
 411 }
 412
 413 static int
 414 usteer_ubus_update_node_data(struct ubus_context *ctx, struct ubus_object *obj,
 415                              struct ubus_request_data *req, const char *method,
 416                              struct blob_attr *msg)
 417 {
 418         const struct blobmsg_policy *policy;
 419         struct blob_attr *tb[__NODE_DATA_MAX];
 420         struct usteer_local_node *ln;
 421         struct blob_attr *val;
 422         const char *name;
 423         bool delete;
 424
 425         delete = !strncmp(method, "del", 3);
 426         policy = delete ? del_node_data_policy : set_node_data_policy;
 427
 428         blobmsg_parse(policy, __NODE_DATA_MAX, tb, blob_data(msg), blob_len(msg));
 429         if (!tb[NODE_DATA_NODE] || !tb[NODE_DATA_VALUES])
 430                 return UBUS_STATUS_INVALID_ARGUMENT;
 431
 432         name = blobmsg_get_string(tb[NODE_DATA_NODE]);
 433         val = tb[NODE_DATA_VALUES];
 434         if (delete && blobmsg_check_array(val, BLOBMSG_TYPE_STRING) < 0)
 435                 return UBUS_STATUS_INVALID_ARGUMENT;
 436
 437         if (strcmp(name, "*") != 0) {
 438                 ln = avl_find_element(&local_nodes, name, ln, node.avl);
 439                 if (!ln)
 440                         return UBUS_STATUS_NOT_FOUND;
 441
 442                 usteer_update_kvlist_data(&ln->node_info, val, delete);
 443                 usteer_update_kvlist_blob(&ln->node.node_info, &ln->node_info);
 444
 445                 return 0;
 446         }
 447
 448         usteer_update_kvlist_data(&host_info, val, delete);
 449         usteer_update_kvlist_blob(&host_info_blob, &host_info);
 450
 451         return 0;
 452 }
 453
 454 static const struct ubus_method usteer_methods[] = {
 455         UBUS_METHOD_NOARG("local_info", usteer_ubus_local_info),
 456         UBUS_METHOD_NOARG("remote_hosts", usteer_ubus_remote_hosts),
 457         UBUS_METHOD_NOARG("remote_info", usteer_ubus_remote_info),
 458         UBUS_METHOD_NOARG("get_clients", usteer_ubus_get_clients),
 459         UBUS_METHOD("get_client_info", usteer_ubus_get_client_info, client_arg),
 460         UBUS_METHOD_NOARG("get_config", usteer_ubus_get_config),
 461         UBUS_METHOD("set_config", usteer_ubus_set_config, config_policy),
 462         UBUS_METHOD("update_config", usteer_ubus_set_config, config_policy),
 463         UBUS_METHOD("set_node_data", usteer_ubus_update_node_data, set_node_data_policy),
 464         UBUS_METHOD("delete_node_data", usteer_ubus_update_node_data, del_node_data_policy),
 465 };
 466
 467 static struct ubus_object_type usteer_obj_type =
 468         UBUS_OBJECT_TYPE("usteer", usteer_methods);
 469
 470 struct ubus_object usteer_obj = {
 471         .name = "usteer",
 472         .type = &usteer_obj_type,
 473         .methods = usteer_methods,
 474         .n_methods = ARRAY_SIZE(usteer_methods),
 475 };
 476
 477 static bool
 478 usteer_add_nr_entry(struct usteer_node *ln, struct usteer_node *node)
 479 {
 480         struct blobmsg_policy policy[3] = {
 481                 { .type = BLOBMSG_TYPE_STRING },
 482                 { .type = BLOBMSG_TYPE_STRING },
 483                 { .type = BLOBMSG_TYPE_STRING },
 484         };
 485         struct blob_attr *tb[3];
 486
 487         if (!node->rrm_nr)
 488                 return false;
 489
 490         if (strcmp(ln->ssid, node->ssid) != 0)
 491                 return false;
 492
 493         blobmsg_parse_array(policy, ARRAY_SIZE(tb), tb,
 494                             blobmsg_data(node->rrm_nr),
 495                             blobmsg_data_len(node->rrm_nr));
 496         if (!tb[2])
 497                 return false;
 498
 499         blobmsg_add_field(&b, BLOBMSG_TYPE_STRING, "",
 500                           blobmsg_data(tb[2]),
 501                           blobmsg_data_len(tb[2]));
 502
 503         return true;
 504 }
 505
 506 static void
 507 usteer_ubus_disassoc_add_neighbors(struct sta_info *si)
 508 {
 509         struct usteer_node *node, *last_remote_neighbor = NULL;
 510         int i = 0;
 511         void *c;
 512
 513         c = blobmsg_open_array(&b, "neighbors");
 514         for_each_local_node(node) {
 515                 if (i >= config.max_neighbor_reports)
 516                         break;
 517                 if (usteer_add_nr_entry(si->node, node))
 518                         i++;
 519         }
 520
 521         while (i < config.max_neighbor_reports) {
 522                 node = usteer_node_get_next_neighbor(si->node, last_remote_neighbor);
 523                 if (!node) {
 524                         /* No more nodes available */
 525                         break;
 526                 }
 527
 528                 last_remote_neighbor = node;
 529                 if (usteer_add_nr_entry(si->node, node))
 530                         i++;
 531         }
 532         blobmsg_close_array(&b, c);
 533 }
 534
 535 int usteer_ubus_notify_client_disassoc(struct sta_info *si)
 536 {
 537         struct usteer_local_node *ln = container_of(si->node, struct usteer_local_node, node);
 538
 539         blob_buf_init(&b, 0);
 540         blobmsg_printf(&b, "addr", MAC_ADDR_FMT, MAC_ADDR_DATA(si->sta->addr));
 541         blobmsg_add_u32(&b, "duration", config.roam_kick_delay);
 542         usteer_ubus_disassoc_add_neighbors(si);
 543         return ubus_invoke(ubus_ctx, ln->obj_id, "wnm_disassoc_imminent", b.head, NULL, 0, 100);
 544 }
 545
 546 int usteer_ubus_trigger_client_scan(struct sta_info *si)
 547 {
 548         struct usteer_local_node *ln = container_of(si->node, struct usteer_local_node, node);
 549
 550         si->scan_band = !si->scan_band;
 551
 552         blob_buf_init(&b, 0);
 553         blobmsg_printf(&b, "addr", MAC_ADDR_FMT, MAC_ADDR_DATA(si->sta->addr));
 554         blobmsg_add_u32(&b, "mode", BEACON_MEASUREMENT_ACTIVE);
 555         blobmsg_add_u32(&b, "duration", 65535);
 556         blobmsg_add_u32(&b, "channel", 0);
 557         blobmsg_add_u32(&b, "op_class", si->scan_band ? 1 : 12);
 558         return ubus_invoke(ubus_ctx, ln->obj_id, "rrm_beacon_req", b.head, NULL, 0, 100);
 559 }
 560
 561 void usteer_ubus_kick_client(struct sta_info *si)
 562 {
 563         struct usteer_local_node *ln = container_of(si->node, struct usteer_local_node, node);
 564
 565         blob_buf_init(&b, 0);
 566         blobmsg_printf(&b, "addr", MAC_ADDR_FMT, MAC_ADDR_DATA(si->sta->addr));
 567         blobmsg_add_u32(&b, "reason", config.load_kick_reason_code);
 568         blobmsg_add_u8(&b, "deauth", 1);
 569         ubus_invoke(ubus_ctx, ln->obj_id, "del_client", b.head, NULL, 0, 100);
 570         si->connected = STA_NOT_CONNECTED;
 571         si->roam_kick = current_time;
 572 }
 573
 574 void usteer_ubus_init(struct ubus_context *ctx)
 575 {
 576         ubus_add_object(ctx, &usteer_obj);
 577 }