package/network/services/hostapd/src/src/ap/ubus.c

   1 /*
   2  * hostapd / ubus support
   3  * Copyright (c) 2013, Felix Fietkau <nbd@nbd.name>
   4  *
   5  * This software may be distributed under the terms of the BSD license.
   6  * See README for more details.
   7  */
   8
   9 #include "utils/includes.h"
  10 #include "utils/common.h"
  11 #include "utils/eloop.h"
  12 #include "utils/wpabuf.h"
  13 #include "common/ieee802_11_defs.h"
  14 #include "common/hw_features_common.h"
  15 #include "hostapd.h"
  16 #include "neighbor_db.h"
  17 #include "wps_hostapd.h"
  18 #include "sta_info.h"
  19 #include "ubus.h"
  20 #include "ap_drv_ops.h"
  21 #include "beacon.h"
  22 #include "rrm.h"
  23 #include "wnm_ap.h"
  24 #include "taxonomy.h"
  25 #include "airtime_policy.h"
  26 #include "hw_features.h"
  27
  28 static struct ubus_context *ctx;
  29 static struct blob_buf b;
  30 static int ctx_ref;
  31
  32 static inline struct hapd_interfaces *get_hapd_interfaces_from_object(struct ubus_object *obj)
  33 {
  34         return container_of(obj, struct hapd_interfaces, ubus);
  35 }
  36
  37 static inline struct hostapd_data *get_hapd_from_object(struct ubus_object *obj)
  38 {
  39         return container_of(obj, struct hostapd_data, ubus.obj);
  40 }
  41
  42 struct ubus_banned_client {
  43         struct avl_node avl;
  44         u8 addr[ETH_ALEN];
  45 };
  46
  47 static void ubus_receive(int sock, void *eloop_ctx, void *sock_ctx)
  48 {
  49         struct ubus_context *ctx = eloop_ctx;
  50         ubus_handle_event(ctx);
  51 }
  52
  53 static void ubus_reconnect_timeout(void *eloop_data, void *user_ctx)
  54 {
  55         if (ubus_reconnect(ctx, NULL)) {
  56                 eloop_register_timeout(1, 0, ubus_reconnect_timeout, ctx, NULL);
  57                 return;
  58         }
  59
  60         eloop_register_read_sock(ctx->sock.fd, ubus_receive, ctx, NULL);
  61 }
  62
  63 static void hostapd_ubus_connection_lost(struct ubus_context *ctx)
  64 {
  65         eloop_unregister_read_sock(ctx->sock.fd);
  66         eloop_register_timeout(1, 0, ubus_reconnect_timeout, ctx, NULL);
  67 }
  68
  69 static bool hostapd_ubus_init(void)
  70 {
  71         if (ctx)
  72                 return true;
  73
  74         ctx = ubus_connect(NULL);
  75         if (!ctx)
  76                 return false;
  77
  78         ctx->connection_lost = hostapd_ubus_connection_lost;
  79         eloop_register_read_sock(ctx->sock.fd, ubus_receive, ctx, NULL);
  80         return true;
  81 }
  82
  83 static void hostapd_ubus_ref_inc(void)
  84 {
  85         ctx_ref++;
  86 }
  87
  88 static void hostapd_ubus_ref_dec(void)
  89 {
  90         ctx_ref--;
  91         if (!ctx)
  92                 return;
  93
  94         if (ctx_ref)
  95                 return;
  96
  97         eloop_unregister_read_sock(ctx->sock.fd);
  98         ubus_free(ctx);
  99         ctx = NULL;
 100 }
 101
 102 void hostapd_ubus_add_iface(struct hostapd_iface *iface)
 103 {
 104         if (!hostapd_ubus_init())
 105                 return;
 106 }
 107
 108 void hostapd_ubus_free_iface(struct hostapd_iface *iface)
 109 {
 110         if (!ctx)
 111                 return;
 112 }
 113
 114 static void hostapd_notify_ubus(struct ubus_object *obj, char *bssname, char *event)
 115 {
 116         char *event_type;
 117
 118         if (!ctx || !obj)
 119                 return;
 120
 121         if (asprintf(&event_type, "bss.%s", event) < 0)
 122                 return;
 123
 124         blob_buf_init(&b, 0);
 125         blobmsg_add_string(&b, "name", bssname);
 126         ubus_notify(ctx, obj, event_type, b.head, -1);
 127         free(event_type);
 128 }
 129
 130 static void hostapd_send_procd_event(char *bssname, char *event)
 131 {
 132         char *name, *s;
 133         uint32_t id;
 134         void *v;
 135
 136         if (!ctx || ubus_lookup_id(ctx, "service", &id))
 137                 return;
 138
 139         if (asprintf(&name, "hostapd.%s.%s", bssname, event) < 0)
 140                 return;
 141
 142         blob_buf_init(&b, 0);
 143
 144         s = blobmsg_alloc_string_buffer(&b, "type", strlen(name) + 1);
 145         sprintf(s, "%s", name);
 146         blobmsg_add_string_buffer(&b);
 147
 148         v = blobmsg_open_table(&b, "data");
 149         blobmsg_close_table(&b, v);
 150
 151         ubus_invoke(ctx, id, "event", b.head, NULL, NULL, 1000);
 152
 153         free(name);
 154 }
 155
 156 static void hostapd_send_shared_event(struct ubus_object *obj, char *bssname, char *event)
 157 {
 158         hostapd_send_procd_event(bssname, event);
 159         hostapd_notify_ubus(obj, bssname, event);
 160 }
 161
 162 static void
 163 hostapd_bss_del_ban(void *eloop_data, void *user_ctx)
 164 {
 165         struct ubus_banned_client *ban = eloop_data;
 166         struct hostapd_data *hapd = user_ctx;
 167
 168         avl_delete(&hapd->ubus.banned, &ban->avl);
 169         free(ban);
 170 }
 171
 172 static void
 173 hostapd_bss_ban_client(struct hostapd_data *hapd, u8 *addr, int time)
 174 {
 175         struct ubus_banned_client *ban;
 176
 177         if (time < 0)
 178                 time = 0;
 179
 180         ban = avl_find_element(&hapd->ubus.banned, addr, ban, avl);
 181         if (!ban) {
 182                 if (!time)
 183                         return;
 184
 185                 ban = os_zalloc(sizeof(*ban));
 186                 memcpy(ban->addr, addr, sizeof(ban->addr));
 187                 ban->avl.key = ban->addr;
 188                 avl_insert(&hapd->ubus.banned, &ban->avl);
 189         } else {
 190                 eloop_cancel_timeout(hostapd_bss_del_ban, ban, hapd);
 191                 if (!time) {
 192                         hostapd_bss_del_ban(ban, hapd);
 193                         return;
 194                 }
 195         }
 196
 197         eloop_register_timeout(0, time * 1000, hostapd_bss_del_ban, ban, hapd);
 198 }
 199
 200 static int
 201 hostapd_bss_reload(struct ubus_context *ctx, struct ubus_object *obj,
 202                    struct ubus_request_data *req, const char *method,
 203                    struct blob_attr *msg)
 204 {
 205         struct hostapd_data *hapd = container_of(obj, struct hostapd_data, ubus.obj);
 206         int ret = hostapd_reload_config(hapd->iface, 1);
 207
 208         hostapd_send_shared_event(&hapd->iface->interfaces->ubus, hapd->conf->iface, "reload");
 209         return ret;
 210 }
 211
 212
 213 static void
 214 hostapd_parse_vht_map_blobmsg(uint16_t map)
 215 {
 216         char label[4];
 217         int16_t val;
 218         int i;
 219
 220         for (i = 0; i < 8; i++) {
 221                 snprintf(label, 4, "%dss", i + 1);
 222
 223                 val = (map & (BIT(1) | BIT(0))) + 7;
 224                 blobmsg_add_u16(&b, label, val == 10 ? -1 : val);
 225                 map = map >> 2;
 226         }
 227 }
 228
 229 static void
 230 hostapd_parse_vht_capab_blobmsg(struct ieee80211_vht_capabilities *vhtc)
 231 {
 232         void *supported_mcs;
 233         void *map;
 234         int i;
 235
 236         static const struct {
 237                 const char *name;
 238                 uint32_t flag;
 239         } vht_capas[] = {
 240                 { "su_beamformee", VHT_CAP_SU_BEAMFORMEE_CAPABLE },
 241                 { "mu_beamformee", VHT_CAP_MU_BEAMFORMEE_CAPABLE },
 242         };
 243
 244         for (i = 0; i < ARRAY_SIZE(vht_capas); i++)
 245                 blobmsg_add_u8(&b, vht_capas[i].name,
 246                                 !!(vhtc->vht_capabilities_info & vht_capas[i].flag));
 247
 248         supported_mcs = blobmsg_open_table(&b, "mcs_map");
 249
 250         /* RX map */
 251         map = blobmsg_open_table(&b, "rx");
 252         hostapd_parse_vht_map_blobmsg(le_to_host16(vhtc->vht_supported_mcs_set.rx_map));
 253         blobmsg_close_table(&b, map);
 254
 255         /* TX map */
 256         map = blobmsg_open_table(&b, "tx");
 257         hostapd_parse_vht_map_blobmsg(le_to_host16(vhtc->vht_supported_mcs_set.tx_map));
 258         blobmsg_close_table(&b, map);
 259
 260         blobmsg_close_table(&b, supported_mcs);
 261 }
 262
 263 static void
 264 hostapd_parse_capab_blobmsg(struct sta_info *sta)
 265 {
 266         void *r, *v;
 267
 268         v = blobmsg_open_table(&b, "capabilities");
 269
 270         if (sta->vht_capabilities) {
 271                 r = blobmsg_open_table(&b, "vht");
 272                 hostapd_parse_vht_capab_blobmsg(sta->vht_capabilities);
 273                 blobmsg_close_table(&b, r);
 274         }
 275
 276         /* ToDo: Add HT / HE capability parsing */
 277
 278         blobmsg_close_table(&b, v);
 279 }
 280
 281 static int
 282 hostapd_bss_get_clients(struct ubus_context *ctx, struct ubus_object *obj,
 283                         struct ubus_request_data *req, const char *method,
 284                         struct blob_attr *msg)
 285 {
 286         struct hostapd_data *hapd = container_of(obj, struct hostapd_data, ubus.obj);
 287         struct hostap_sta_driver_data sta_driver_data;
 288         struct sta_info *sta;
 289         void *list, *c;
 290         char mac_buf[20];
 291         static const struct {
 292                 const char *name;
 293                 uint32_t flag;
 294         } sta_flags[] = {
 295                 { "auth", WLAN_STA_AUTH },
 296                 { "assoc", WLAN_STA_ASSOC },
 297                 { "authorized", WLAN_STA_AUTHORIZED },
 298                 { "preauth", WLAN_STA_PREAUTH },
 299                 { "wds", WLAN_STA_WDS },
 300                 { "wmm", WLAN_STA_WMM },
 301                 { "ht", WLAN_STA_HT },
 302                 { "vht", WLAN_STA_VHT },
 303                 { "he", WLAN_STA_HE },
 304                 { "wps", WLAN_STA_WPS },
 305                 { "mfp", WLAN_STA_MFP },
 306         };
 307
 308         blob_buf_init(&b, 0);
 309         blobmsg_add_u32(&b, "freq", hapd->iface->freq);
 310         list = blobmsg_open_table(&b, "clients");
 311         for (sta = hapd->sta_list; sta; sta = sta->next) {
 312                 void *r;
 313                 int i;
 314
 315                 sprintf(mac_buf, MACSTR, MAC2STR(sta->addr));
 316                 c = blobmsg_open_table(&b, mac_buf);
 317                 for (i = 0; i < ARRAY_SIZE(sta_flags); i++)
 318                         blobmsg_add_u8(&b, sta_flags[i].name,
 319                                        !!(sta->flags & sta_flags[i].flag));
 320
 321 #ifdef CONFIG_MBO
 322                 blobmsg_add_u8(&b, "mbo", !!(sta->cell_capa));
 323 #endif
 324
 325                 r = blobmsg_open_array(&b, "rrm");
 326                 for (i = 0; i < ARRAY_SIZE(sta->rrm_enabled_capa); i++)
 327                         blobmsg_add_u32(&b, "", sta->rrm_enabled_capa[i]);
 328                 blobmsg_close_array(&b, r);
 329
 330                 r = blobmsg_open_array(&b, "extended_capabilities");
 331                 /* Check if client advertises extended capabilities */
 332                 if (sta->ext_capability && sta->ext_capability[0] > 0) {
 333                         for (i = 0; i < sta->ext_capability[0]; i++) {
 334                                 blobmsg_add_u32(&b, "", sta->ext_capability[1 + i]);
 335                         }
 336                 }
 337                 blobmsg_close_array(&b, r);
 338
 339                 blobmsg_add_u32(&b, "aid", sta->aid);
 340 #ifdef CONFIG_TAXONOMY
 341                 r = blobmsg_alloc_string_buffer(&b, "signature", 1024);
 342                 if (retrieve_sta_taxonomy(hapd, sta, r, 1024) > 0)
 343                         blobmsg_add_string_buffer(&b);
 344 #endif
 345
 346                 /* Driver information */
 347                 if (hostapd_drv_read_sta_data(hapd, &sta_driver_data, sta->addr) >= 0) {
 348                         r = blobmsg_open_table(&b, "bytes");
 349                         blobmsg_add_u64(&b, "rx", sta_driver_data.rx_bytes);
 350                         blobmsg_add_u64(&b, "tx", sta_driver_data.tx_bytes);
 351                         blobmsg_close_table(&b, r);
 352                         r = blobmsg_open_table(&b, "airtime");
 353                         blobmsg_add_u64(&b, "rx", sta_driver_data.rx_airtime);
 354                         blobmsg_add_u64(&b, "tx", sta_driver_data.tx_airtime);
 355                         blobmsg_close_table(&b, r);
 356                         r = blobmsg_open_table(&b, "packets");
 357                         blobmsg_add_u32(&b, "rx", sta_driver_data.rx_packets);
 358                         blobmsg_add_u32(&b, "tx", sta_driver_data.tx_packets);
 359                         blobmsg_close_table(&b, r);
 360                         r = blobmsg_open_table(&b, "rate");
 361                         /* Rate in kbits */
 362                         blobmsg_add_u32(&b, "rx", sta_driver_data.current_rx_rate * 100);
 363                         blobmsg_add_u32(&b, "tx", sta_driver_data.current_tx_rate * 100);
 364                         blobmsg_close_table(&b, r);
 365                         blobmsg_add_u32(&b, "signal", sta_driver_data.signal);
 366                 }
 367
 368                 hostapd_parse_capab_blobmsg(sta);
 369
 370                 blobmsg_close_table(&b, c);
 371         }
 372         blobmsg_close_array(&b, list);
 373         ubus_send_reply(ctx, req, b.head);
 374
 375         return 0;
 376 }
 377
 378 static int
 379 hostapd_bss_get_features(struct ubus_context *ctx, struct ubus_object *obj,
 380                         struct ubus_request_data *req, const char *method,
 381                         struct blob_attr *msg)
 382 {
 383         struct hostapd_data *hapd = container_of(obj, struct hostapd_data, ubus.obj);
 384
 385         blob_buf_init(&b, 0);
 386         blobmsg_add_u8(&b, "ht_supported", ht_supported(hapd->iface->hw_features));
 387         blobmsg_add_u8(&b, "vht_supported", vht_supported(hapd->iface->hw_features));
 388         ubus_send_reply(ctx, req, b.head);
 389
 390         return 0;
 391 }
 392
 393 /* Imported from iw/util.c
 394  *  https://git.kernel.org/pub/scm/linux/kernel/git/jberg/iw.git/tree/util.c?id=4b25ae3537af48dbf9d0abf94132e5ba01b32c18#n200
 395  */
 396 int ieee80211_frequency_to_channel(int freq)
 397 {
 398         /* see 802.11-2007 17.3.8.3.2 and Annex J */
 399         if (freq == 2484)
 400                 return 14;
 401         /* see 802.11ax D6.1 27.3.23.2 and Annex E */
 402         else if (freq == 5935)
 403                 return 2;
 404         else if (freq < 2484)
 405                 return (freq - 2407) / 5;
 406         else if (freq >= 4910 && freq <= 4980)
 407                 return (freq - 4000) / 5;
 408         else if (freq < 5950)
 409                 return (freq - 5000) / 5;
 410         else if (freq <= 45000) /* DMG band lower limit */
 411                 /* see 802.11ax D6.1 27.3.23.2 */
 412                 return (freq - 5950) / 5;
 413         else if (freq >= 58320 && freq <= 70200)
 414                 return (freq - 56160) / 2160;
 415         else
 416                 return 0;
 417 }
 418
 419 static int
 420 hostapd_bss_get_status(struct ubus_context *ctx, struct ubus_object *obj,
 421                        struct ubus_request_data *req, const char *method,
 422                        struct blob_attr *msg)
 423 {
 424         struct hostapd_data *hapd = container_of(obj, struct hostapd_data, ubus.obj);
 425         void *airtime_table, *dfs_table, *rrm_table, *wnm_table;
 426         struct os_reltime now;
 427         char ssid[SSID_MAX_LEN + 1];
 428         char phy_name[17];
 429         size_t ssid_len = SSID_MAX_LEN;
 430         u8 channel = 0, op_class = 0;
 431
 432         if (hapd->conf->ssid.ssid_len < SSID_MAX_LEN)
 433                 ssid_len = hapd->conf->ssid.ssid_len;
 434
 435         ieee80211_freq_to_channel_ext(hapd->iface->freq,
 436                                       hapd->iconf->secondary_channel,
 437                                       hostapd_get_oper_chwidth(hapd->iconf),
 438                                       &op_class, &channel);
 439
 440         blob_buf_init(&b, 0);
 441         blobmsg_add_string(&b, "status", hostapd_state_text(hapd->iface->state));
 442         blobmsg_printf(&b, "bssid", MACSTR, MAC2STR(hapd->conf->bssid));
 443
 444         memset(ssid, 0, SSID_MAX_LEN + 1);
 445         memcpy(ssid, hapd->conf->ssid.ssid, ssid_len);
 446         blobmsg_add_string(&b, "ssid", ssid);
 447
 448         blobmsg_add_u32(&b, "freq", hapd->iface->freq);
 449         blobmsg_add_u32(&b, "channel", channel);
 450         blobmsg_add_u32(&b, "op_class", op_class);
 451         blobmsg_add_u32(&b, "beacon_interval", hapd->iconf->beacon_int);
 452 #ifdef CONFIG_IEEE80211AX
 453         blobmsg_add_u32(&b, "bss_color", hapd->iface->conf->he_op.he_bss_color_disabled ? -1 :
 454                                          hapd->iface->conf->he_op.he_bss_color);
 455 #else
 456         blobmsg_add_u32(&b, "bss_color", -1);
 457 #endif
 458
 459         snprintf(phy_name, 17, "%s", hapd->iface->phy);
 460         blobmsg_add_string(&b, "phy", phy_name);
 461
 462         /* RRM */
 463         rrm_table = blobmsg_open_table(&b, "rrm");
 464         blobmsg_add_u64(&b, "neighbor_report_tx", hapd->openwrt_stats.rrm.neighbor_report_tx);
 465         blobmsg_close_table(&b, rrm_table);
 466
 467         /* WNM */
 468         wnm_table = blobmsg_open_table(&b, "wnm");
 469         blobmsg_add_u64(&b, "bss_transition_query_rx", hapd->openwrt_stats.wnm.bss_transition_query_rx);
 470         blobmsg_add_u64(&b, "bss_transition_request_tx", hapd->openwrt_stats.wnm.bss_transition_request_tx);
 471         blobmsg_add_u64(&b, "bss_transition_response_rx", hapd->openwrt_stats.wnm.bss_transition_response_rx);
 472         blobmsg_close_table(&b, wnm_table);
 473
 474         /* Airtime */
 475         airtime_table = blobmsg_open_table(&b, "airtime");
 476         blobmsg_add_u64(&b, "time", hapd->iface->last_channel_time);
 477         blobmsg_add_u64(&b, "time_busy", hapd->iface->last_channel_time_busy);
 478         blobmsg_add_u16(&b, "utilization", hapd->iface->channel_utilization);
 479         blobmsg_close_table(&b, airtime_table);
 480
 481         /* DFS */
 482         dfs_table = blobmsg_open_table(&b, "dfs");
 483         blobmsg_add_u32(&b, "cac_seconds", hapd->iface->dfs_cac_ms / 1000);
 484         blobmsg_add_u8(&b, "cac_active", !!(hapd->iface->cac_started));
 485         os_reltime_age(&hapd->iface->dfs_cac_start, &now);
 486         blobmsg_add_u32(&b, "cac_seconds_left",
 487                         hapd->iface->cac_started ? hapd->iface->dfs_cac_ms / 1000 - now.sec : 0);
 488         blobmsg_close_table(&b, dfs_table);
 489
 490         ubus_send_reply(ctx, req, b.head);
 491
 492         return 0;
 493 }
 494
 495 enum {
 496         NOTIFY_RESPONSE,
 497         __NOTIFY_MAX
 498 };
 499
 500 static const struct blobmsg_policy notify_policy[__NOTIFY_MAX] = {
 501         [NOTIFY_RESPONSE] = { "notify_response", BLOBMSG_TYPE_INT32 },
 502 };
 503
 504 static int
 505 hostapd_notify_response(struct ubus_context *ctx, struct ubus_object *obj,
 506                         struct ubus_request_data *req, const char *method,
 507                         struct blob_attr *msg)
 508 {
 509         struct blob_attr *tb[__NOTIFY_MAX];
 510         struct hostapd_data *hapd = get_hapd_from_object(obj);
 511         struct wpabuf *elems;
 512         const char *pos;
 513         size_t len;
 514
 515         blobmsg_parse(notify_policy, __NOTIFY_MAX, tb,
 516                       blob_data(msg), blob_len(msg));
 517
 518         if (!tb[NOTIFY_RESPONSE])
 519                 return UBUS_STATUS_INVALID_ARGUMENT;
 520
 521         hapd->ubus.notify_response = blobmsg_get_u32(tb[NOTIFY_RESPONSE]);
 522
 523         return UBUS_STATUS_OK;
 524 }
 525
 526 enum {
 527         DEL_CLIENT_ADDR,
 528         DEL_CLIENT_REASON,
 529         DEL_CLIENT_DEAUTH,
 530         DEL_CLIENT_BAN_TIME,
 531         __DEL_CLIENT_MAX
 532 };
 533
 534 static const struct blobmsg_policy del_policy[__DEL_CLIENT_MAX] = {
 535         [DEL_CLIENT_ADDR] = { "addr", BLOBMSG_TYPE_STRING },
 536         [DEL_CLIENT_REASON] = { "reason", BLOBMSG_TYPE_INT32 },
 537         [DEL_CLIENT_DEAUTH] = { "deauth", BLOBMSG_TYPE_INT8 },
 538         [DEL_CLIENT_BAN_TIME] = { "ban_time", BLOBMSG_TYPE_INT32 },
 539 };
 540
 541 static int
 542 hostapd_bss_del_client(struct ubus_context *ctx, struct ubus_object *obj,
 543                         struct ubus_request_data *req, const char *method,
 544                         struct blob_attr *msg)
 545 {
 546         struct blob_attr *tb[__DEL_CLIENT_MAX];
 547         struct hostapd_data *hapd = container_of(obj, struct hostapd_data, ubus.obj);
 548         struct sta_info *sta;
 549         bool deauth = false;
 550         int reason;
 551         u8 addr[ETH_ALEN];
 552
 553         blobmsg_parse(del_policy, __DEL_CLIENT_MAX, tb, blob_data(msg), blob_len(msg));
 554
 555         if (!tb[DEL_CLIENT_ADDR])
 556                 return UBUS_STATUS_INVALID_ARGUMENT;
 557
 558         if (hwaddr_aton(blobmsg_data(tb[DEL_CLIENT_ADDR]), addr))
 559                 return UBUS_STATUS_INVALID_ARGUMENT;
 560
 561         if (tb[DEL_CLIENT_REASON])
 562                 reason = blobmsg_get_u32(tb[DEL_CLIENT_REASON]);
 563
 564         if (tb[DEL_CLIENT_DEAUTH])
 565                 deauth = blobmsg_get_bool(tb[DEL_CLIENT_DEAUTH]);
 566
 567         sta = ap_get_sta(hapd, addr);
 568         if (sta) {
 569                 if (deauth) {
 570                         hostapd_drv_sta_deauth(hapd, addr, reason);
 571                         ap_sta_deauthenticate(hapd, sta, reason);
 572                 } else {
 573                         hostapd_drv_sta_disassoc(hapd, addr, reason);
 574                         ap_sta_disassociate(hapd, sta, reason);
 575                 }
 576         }
 577
 578         if (tb[DEL_CLIENT_BAN_TIME])
 579                 hostapd_bss_ban_client(hapd, addr, blobmsg_get_u32(tb[DEL_CLIENT_BAN_TIME]));
 580
 581         return 0;
 582 }
 583
 584 static void
 585 blobmsg_add_macaddr(struct blob_buf *buf, const char *name, const u8 *addr)
 586 {
 587         char *s;
 588
 589         s = blobmsg_alloc_string_buffer(buf, name, 20);
 590         sprintf(s, MACSTR, MAC2STR(addr));
 591         blobmsg_add_string_buffer(buf);
 592 }
 593
 594 static int
 595 hostapd_bss_list_bans(struct ubus_context *ctx, struct ubus_object *obj,
 596                       struct ubus_request_data *req, const char *method,
 597                       struct blob_attr *msg)
 598 {
 599         struct hostapd_data *hapd = container_of(obj, struct hostapd_data, ubus.obj);
 600         struct ubus_banned_client *ban;
 601         void *c;
 602
 603         blob_buf_init(&b, 0);
 604         c = blobmsg_open_array(&b, "clients");
 605         avl_for_each_element(&hapd->ubus.banned, ban, avl)
 606                 blobmsg_add_macaddr(&b, NULL, ban->addr);
 607         blobmsg_close_array(&b, c);
 608         ubus_send_reply(ctx, req, b.head);
 609
 610         return 0;
 611 }
 612
 613 #ifdef CONFIG_WPS
 614 static int
 615 hostapd_bss_wps_start(struct ubus_context *ctx, struct ubus_object *obj,
 616                         struct ubus_request_data *req, const char *method,
 617                         struct blob_attr *msg)
 618 {
 619         int rc;
 620         struct hostapd_data *hapd = container_of(obj, struct hostapd_data, ubus.obj);
 621
 622         rc = hostapd_wps_button_pushed(hapd, NULL);
 623
 624         if (rc != 0)
 625                 return UBUS_STATUS_NOT_SUPPORTED;
 626
 627         return 0;
 628 }
 629
 630
 631 static const char * pbc_status_enum_str(enum pbc_status status)
 632 {
 633         switch (status) {
 634         case WPS_PBC_STATUS_DISABLE:
 635                 return "Disabled";
 636         case WPS_PBC_STATUS_ACTIVE:
 637                 return "Active";
 638         case WPS_PBC_STATUS_TIMEOUT:
 639                 return "Timed-out";
 640         case WPS_PBC_STATUS_OVERLAP:
 641                 return "Overlap";
 642         default:
 643                 return "Unknown";
 644         }
 645 }
 646
 647 static int
 648 hostapd_bss_wps_status(struct ubus_context *ctx, struct ubus_object *obj,
 649                         struct ubus_request_data *req, const char *method,
 650                         struct blob_attr *msg)
 651 {
 652         int rc;
 653         struct hostapd_data *hapd = container_of(obj, struct hostapd_data, ubus.obj);
 654
 655         blob_buf_init(&b, 0);
 656
 657         blobmsg_add_string(&b, "pbc_status", pbc_status_enum_str(hapd->wps_stats.pbc_status));
 658         blobmsg_add_string(&b, "last_wps_result",
 659                            (hapd->wps_stats.status == WPS_STATUS_SUCCESS ?
 660                             "Success":
 661                             (hapd->wps_stats.status == WPS_STATUS_FAILURE ?
 662                              "Failed" : "None")));
 663
 664         /* If status == Failure - Add possible Reasons */
 665         if(hapd->wps_stats.status == WPS_STATUS_FAILURE &&
 666            hapd->wps_stats.failure_reason > 0)
 667                 blobmsg_add_string(&b, "reason", wps_ei_str(hapd->wps_stats.failure_reason));
 668
 669         if (hapd->wps_stats.status)
 670                 blobmsg_printf(&b, "peer_address", MACSTR, MAC2STR(hapd->wps_stats.peer_addr));
 671
 672         ubus_send_reply(ctx, req, b.head);
 673
 674         return 0;
 675 }
 676
 677 static int
 678 hostapd_bss_wps_cancel(struct ubus_context *ctx, struct ubus_object *obj,
 679                         struct ubus_request_data *req, const char *method,
 680                         struct blob_attr *msg)
 681 {
 682         int rc;
 683         struct hostapd_data *hapd = container_of(obj, struct hostapd_data, ubus.obj);
 684
 685         rc = hostapd_wps_cancel(hapd);
 686
 687         if (rc != 0)
 688                 return UBUS_STATUS_NOT_SUPPORTED;
 689
 690         return 0;
 691 }
 692 #endif /* CONFIG_WPS */
 693
 694 static int
 695 hostapd_bss_update_beacon(struct ubus_context *ctx, struct ubus_object *obj,
 696                         struct ubus_request_data *req, const char *method,
 697                         struct blob_attr *msg)
 698 {
 699         int rc;
 700         struct hostapd_data *hapd = container_of(obj, struct hostapd_data, ubus.obj);
 701
 702         rc = ieee802_11_set_beacon(hapd);
 703
 704         if (rc != 0)
 705                 return UBUS_STATUS_NOT_SUPPORTED;
 706
 707         return 0;
 708 }
 709
 710 enum {
 711         CONFIG_IFACE,
 712         CONFIG_FILE,
 713         __CONFIG_MAX
 714 };
 715
 716 static const struct blobmsg_policy config_add_policy[__CONFIG_MAX] = {
 717         [CONFIG_IFACE] = { "iface", BLOBMSG_TYPE_STRING },
 718         [CONFIG_FILE] = { "config", BLOBMSG_TYPE_STRING },
 719 };
 720
 721 static int
 722 hostapd_config_add(struct ubus_context *ctx, struct ubus_object *obj,
 723                    struct ubus_request_data *req, const char *method,
 724                    struct blob_attr *msg)
 725 {
 726         struct blob_attr *tb[__CONFIG_MAX];
 727         struct hapd_interfaces *interfaces = get_hapd_interfaces_from_object(obj);
 728         char buf[128];
 729
 730         blobmsg_parse(config_add_policy, __CONFIG_MAX, tb, blob_data(msg), blob_len(msg));
 731
 732         if (!tb[CONFIG_FILE] || !tb[CONFIG_IFACE])
 733                 return UBUS_STATUS_INVALID_ARGUMENT;
 734
 735         snprintf(buf, sizeof(buf), "bss_config=%s:%s",
 736                 blobmsg_get_string(tb[CONFIG_IFACE]),
 737                 blobmsg_get_string(tb[CONFIG_FILE]));
 738
 739         if (hostapd_add_iface(interfaces, buf))
 740                 return UBUS_STATUS_INVALID_ARGUMENT;
 741
 742         blob_buf_init(&b, 0);
 743         blobmsg_add_u32(&b, "pid", getpid());
 744         ubus_send_reply(ctx, req, b.head);
 745
 746         return UBUS_STATUS_OK;
 747 }
 748
 749 enum {
 750         CONFIG_REM_IFACE,
 751         __CONFIG_REM_MAX
 752 };
 753
 754 static const struct blobmsg_policy config_remove_policy[__CONFIG_REM_MAX] = {
 755         [CONFIG_REM_IFACE] = { "iface", BLOBMSG_TYPE_STRING },
 756 };
 757
 758 static int
 759 hostapd_config_remove(struct ubus_context *ctx, struct ubus_object *obj,
 760                       struct ubus_request_data *req, const char *method,
 761                       struct blob_attr *msg)
 762 {
 763         struct blob_attr *tb[__CONFIG_REM_MAX];
 764         struct hapd_interfaces *interfaces = get_hapd_interfaces_from_object(obj);
 765         char buf[128];
 766
 767         blobmsg_parse(config_remove_policy, __CONFIG_REM_MAX, tb, blob_data(msg), blob_len(msg));
 768
 769         if (!tb[CONFIG_REM_IFACE])
 770                 return UBUS_STATUS_INVALID_ARGUMENT;
 771
 772         if (hostapd_remove_iface(interfaces, blobmsg_get_string(tb[CONFIG_REM_IFACE])))
 773                 return UBUS_STATUS_INVALID_ARGUMENT;
 774
 775         return UBUS_STATUS_OK;
 776 }
 777
 778 enum {
 779         CSA_FREQ,
 780         CSA_BCN_COUNT,
 781         CSA_CENTER_FREQ1,
 782         CSA_CENTER_FREQ2,
 783         CSA_BANDWIDTH,
 784         CSA_SEC_CHANNEL_OFFSET,
 785         CSA_HT,
 786         CSA_VHT,
 787         CSA_HE,
 788         CSA_BLOCK_TX,
 789         CSA_FORCE,
 790         __CSA_MAX
 791 };
 792
 793 static const struct blobmsg_policy csa_policy[__CSA_MAX] = {
 794         [CSA_FREQ] = { "freq", BLOBMSG_TYPE_INT32 },
 795         [CSA_BCN_COUNT] = { "bcn_count", BLOBMSG_TYPE_INT32 },
 796         [CSA_CENTER_FREQ1] = { "center_freq1", BLOBMSG_TYPE_INT32 },
 797         [CSA_CENTER_FREQ2] = { "center_freq2", BLOBMSG_TYPE_INT32 },
 798         [CSA_BANDWIDTH] = { "bandwidth", BLOBMSG_TYPE_INT32 },
 799         [CSA_SEC_CHANNEL_OFFSET] = { "sec_channel_offset", BLOBMSG_TYPE_INT32 },
 800         [CSA_HT] = { "ht", BLOBMSG_TYPE_BOOL },
 801         [CSA_VHT] = { "vht", BLOBMSG_TYPE_BOOL },
 802         [CSA_HE] = { "he", BLOBMSG_TYPE_BOOL },
 803         [CSA_BLOCK_TX] = { "block_tx", BLOBMSG_TYPE_BOOL },
 804         [CSA_FORCE] = { "force", BLOBMSG_TYPE_BOOL },
 805 };
 806
 807
 808 static void switch_chan_fallback_cb(void *eloop_data, void *user_ctx)
 809 {
 810         struct hostapd_iface *iface = eloop_data;
 811         struct hostapd_freq_params *freq_params = user_ctx;
 812
 813         hostapd_switch_channel_fallback(iface, freq_params);
 814 }
 815
 816 #ifdef NEED_AP_MLME
 817 static int
 818 hostapd_switch_chan(struct ubus_context *ctx, struct ubus_object *obj,
 819                     struct ubus_request_data *req, const char *method,
 820                     struct blob_attr *msg)
 821 {
 822         struct blob_attr *tb[__CSA_MAX];
 823         struct hostapd_data *hapd = get_hapd_from_object(obj);
 824         struct hostapd_config *iconf = hapd->iface->conf;
 825         struct hostapd_freq_params *freq_params;
 826         struct hostapd_hw_modes *mode = hapd->iface->current_mode;
 827         struct csa_settings css = {
 828                 .freq_params = {
 829                         .ht_enabled = iconf->ieee80211n,
 830                         .vht_enabled = iconf->ieee80211ac,
 831                         .he_enabled = iconf->ieee80211ax,
 832                         .sec_channel_offset = iconf->secondary_channel,
 833                 }
 834         };
 835         u8 chwidth = hostapd_get_oper_chwidth(iconf);
 836         u8 seg0 = 0, seg1 = 0;
 837         int ret = UBUS_STATUS_OK;
 838         int i;
 839
 840         blobmsg_parse(csa_policy, __CSA_MAX, tb, blob_data(msg), blob_len(msg));
 841
 842         if (!tb[CSA_FREQ])
 843                 return UBUS_STATUS_INVALID_ARGUMENT;
 844
 845         switch (iconf->vht_oper_chwidth) {
 846         case CHANWIDTH_USE_HT:
 847                 if (iconf->secondary_channel)
 848                         css.freq_params.bandwidth = 40;
 849                 else
 850                         css.freq_params.bandwidth = 20;
 851                 break;
 852         case CHANWIDTH_160MHZ:
 853                 css.freq_params.bandwidth = 160;
 854                 break;
 855         default:
 856                 css.freq_params.bandwidth = 80;
 857                 break;
 858         }
 859
 860         css.freq_params.freq = blobmsg_get_u32(tb[CSA_FREQ]);
 861
 862 #define SET_CSA_SETTING(name, field, type) \
 863         do { \
 864                 if (tb[name]) \
 865                         css.field = blobmsg_get_ ## type(tb[name]); \
 866         } while(0)
 867
 868         SET_CSA_SETTING(CSA_BCN_COUNT, cs_count, u32);
 869         SET_CSA_SETTING(CSA_CENTER_FREQ1, freq_params.center_freq1, u32);
 870         SET_CSA_SETTING(CSA_CENTER_FREQ2, freq_params.center_freq2, u32);
 871         SET_CSA_SETTING(CSA_BANDWIDTH, freq_params.bandwidth, u32);
 872         SET_CSA_SETTING(CSA_SEC_CHANNEL_OFFSET, freq_params.sec_channel_offset, u32);
 873         SET_CSA_SETTING(CSA_HT, freq_params.ht_enabled, bool);
 874         SET_CSA_SETTING(CSA_VHT, freq_params.vht_enabled, bool);
 875         SET_CSA_SETTING(CSA_HE, freq_params.he_enabled, bool);
 876         SET_CSA_SETTING(CSA_BLOCK_TX, block_tx, bool);
 877
 878         css.freq_params.channel = hostapd_hw_get_channel(hapd, css.freq_params.freq);
 879         if (!css.freq_params.channel)
 880                 return UBUS_STATUS_NOT_SUPPORTED;
 881
 882         switch (css.freq_params.bandwidth) {
 883         case 160:
 884                 chwidth = CHANWIDTH_160MHZ;
 885                 break;
 886         case 80:
 887                 chwidth = css.freq_params.center_freq2 ? CHANWIDTH_80P80MHZ : CHANWIDTH_80MHZ;
 888                 break;
 889         default:
 890                 chwidth = CHANWIDTH_USE_HT;
 891                 break;
 892         }
 893
 894         hostapd_set_freq_params(&css.freq_params, iconf->hw_mode,
 895                                 css.freq_params.freq,
 896                                 css.freq_params.channel, iconf->enable_edmg,
 897                                 iconf->edmg_channel,
 898                                 css.freq_params.ht_enabled,
 899                                 css.freq_params.vht_enabled,
 900                                 css.freq_params.he_enabled,
 901                                 css.freq_params.eht_enabled,
 902                                 css.freq_params.sec_channel_offset,
 903                                 chwidth, seg0, seg1,
 904                                 iconf->vht_capab,
 905                                 mode ? &mode->he_capab[IEEE80211_MODE_AP] :
 906                                 NULL,
 907                                 mode ? &mode->eht_capab[IEEE80211_MODE_AP] :
 908                                 NULL);
 909
 910         for (i = 0; i < hapd->iface->num_bss; i++) {
 911                 struct hostapd_data *bss = hapd->iface->bss[i];
 912
 913                 if (hostapd_switch_channel(bss, &css) != 0)
 914                         ret = UBUS_STATUS_NOT_SUPPORTED;
 915         }
 916
 917         if (!ret || !tb[CSA_FORCE] || !blobmsg_get_bool(tb[CSA_FORCE]))
 918                 return ret;
 919
 920         freq_params = malloc(sizeof(*freq_params));
 921         memcpy(freq_params, &css.freq_params, sizeof(*freq_params));
 922         eloop_register_timeout(0, 1, switch_chan_fallback_cb,
 923                                hapd->iface, freq_params);
 924
 925         return 0;
 926 #undef SET_CSA_SETTING
 927 }
 928 #endif
 929
 930 enum {
 931         VENDOR_ELEMENTS,
 932         __VENDOR_ELEMENTS_MAX
 933 };
 934
 935 static const struct blobmsg_policy ve_policy[__VENDOR_ELEMENTS_MAX] = {
 936         /* vendor elements are provided as hex-string */
 937         [VENDOR_ELEMENTS] = { "vendor_elements", BLOBMSG_TYPE_STRING },
 938 };
 939
 940 static int
 941 hostapd_vendor_elements(struct ubus_context *ctx, struct ubus_object *obj,
 942                         struct ubus_request_data *req, const char *method,
 943                         struct blob_attr *msg)
 944 {
 945         struct blob_attr *tb[__VENDOR_ELEMENTS_MAX];
 946         struct hostapd_data *hapd = get_hapd_from_object(obj);
 947         struct hostapd_bss_config *bss = hapd->conf;
 948         struct wpabuf *elems;
 949         const char *pos;
 950         size_t len;
 951
 952         blobmsg_parse(ve_policy, __VENDOR_ELEMENTS_MAX, tb,
 953                       blob_data(msg), blob_len(msg));
 954
 955         if (!tb[VENDOR_ELEMENTS])
 956                 return UBUS_STATUS_INVALID_ARGUMENT;
 957
 958         pos = blobmsg_data(tb[VENDOR_ELEMENTS]);
 959         len = os_strlen(pos);
 960         if (len & 0x01)
 961                         return UBUS_STATUS_INVALID_ARGUMENT;
 962
 963         len /= 2;
 964         if (len == 0) {
 965                 wpabuf_free(bss->vendor_elements);
 966                 bss->vendor_elements = NULL;
 967                 return 0;
 968         }
 969
 970         elems = wpabuf_alloc(len);
 971         if (elems == NULL)
 972                 return 1;
 973
 974         if (hexstr2bin(pos, wpabuf_put(elems, len), len)) {
 975                 wpabuf_free(elems);
 976                 return UBUS_STATUS_INVALID_ARGUMENT;
 977         }
 978
 979         wpabuf_free(bss->vendor_elements);
 980         bss->vendor_elements = elems;
 981
 982         /* update beacons if vendor elements were set successfully */
 983         if (ieee802_11_update_beacons(hapd->iface) != 0)
 984                 return UBUS_STATUS_NOT_SUPPORTED;
 985         return UBUS_STATUS_OK;
 986 }
 987
 988 static void
 989 hostapd_rrm_print_nr(struct hostapd_neighbor_entry *nr)
 990 {
 991         const u8 *data;
 992         char *str;
 993         int len;
 994
 995         blobmsg_printf(&b, "", MACSTR, MAC2STR(nr->bssid));
 996
 997         str = blobmsg_alloc_string_buffer(&b, "", nr->ssid.ssid_len + 1);
 998         memcpy(str, nr->ssid.ssid, nr->ssid.ssid_len);
 999         str[nr->ssid.ssid_len] = 0;
1000         blobmsg_add_string_buffer(&b);
1001
1002         len = wpabuf_len(nr->nr);
1003         str = blobmsg_alloc_string_buffer(&b, "", 2 * len + 1);
1004         wpa_snprintf_hex(str, 2 * len + 1, wpabuf_head_u8(nr->nr), len);
1005         blobmsg_add_string_buffer(&b);
1006 }
1007
1008 enum {
1009         BSS_MGMT_EN_NEIGHBOR,
1010         BSS_MGMT_EN_BEACON,
1011         BSS_MGMT_EN_LINK_MEASUREMENT,
1012 #ifdef CONFIG_WNM_AP
1013         BSS_MGMT_EN_BSS_TRANSITION,
1014 #endif
1015         __BSS_MGMT_EN_MAX
1016 };
1017
1018 static bool
1019 __hostapd_bss_mgmt_enable_f(struct hostapd_data *hapd, int flag)
1020 {
1021         struct hostapd_bss_config *bss = hapd->conf;
1022         uint32_t flags;
1023
1024         switch (flag) {
1025         case BSS_MGMT_EN_NEIGHBOR:
1026                 if (bss->radio_measurements[0] &
1027                     WLAN_RRM_CAPS_NEIGHBOR_REPORT)
1028                         return false;
1029
1030                 bss->radio_measurements[0] |=
1031                         WLAN_RRM_CAPS_NEIGHBOR_REPORT;
1032                 hostapd_neighbor_set_own_report(hapd);
1033                 return true;
1034         case BSS_MGMT_EN_BEACON:
1035                 flags = WLAN_RRM_CAPS_BEACON_REPORT_PASSIVE |
1036                         WLAN_RRM_CAPS_BEACON_REPORT_ACTIVE |
1037                         WLAN_RRM_CAPS_BEACON_REPORT_TABLE;
1038
1039                 if (bss->radio_measurements[0] & flags == flags)
1040                         return false;
1041
1042                 bss->radio_measurements[0] |= (u8) flags;
1043                 return true;
1044         case BSS_MGMT_EN_LINK_MEASUREMENT:
1045                 flags = WLAN_RRM_CAPS_LINK_MEASUREMENT;
1046
1047                 if (bss->radio_measurements[0] & flags == flags)
1048                         return false;
1049
1050                 bss->radio_measurements[0] |= (u8) flags;
1051                 return true;
1052 #ifdef CONFIG_WNM_AP
1053         case BSS_MGMT_EN_BSS_TRANSITION:
1054                 if (bss->bss_transition)
1055                         return false;
1056
1057                 bss->bss_transition = 1;
1058                 return true;
1059 #endif
1060         }
1061 }
1062
1063 static void
1064 __hostapd_bss_mgmt_enable(struct hostapd_data *hapd, uint32_t flags)
1065 {
1066         bool update = false;
1067         int i;
1068
1069         for (i = 0; i < __BSS_MGMT_EN_MAX; i++) {
1070                 if (!(flags & (1 << i)))
1071                         continue;
1072
1073                 update |= __hostapd_bss_mgmt_enable_f(hapd, i);
1074         }
1075
1076         if (update)
1077                 ieee802_11_update_beacons(hapd->iface);
1078 }
1079
1080
1081 static const struct blobmsg_policy bss_mgmt_enable_policy[__BSS_MGMT_EN_MAX] = {
1082         [BSS_MGMT_EN_NEIGHBOR] = { "neighbor_report", BLOBMSG_TYPE_BOOL },
1083         [BSS_MGMT_EN_BEACON] = { "beacon_report", BLOBMSG_TYPE_BOOL },
1084         [BSS_MGMT_EN_LINK_MEASUREMENT] = { "link_measurement", BLOBMSG_TYPE_BOOL },
1085 #ifdef CONFIG_WNM_AP
1086         [BSS_MGMT_EN_BSS_TRANSITION] = { "bss_transition", BLOBMSG_TYPE_BOOL },
1087 #endif
1088 };
1089
1090 static int
1091 hostapd_bss_mgmt_enable(struct ubus_context *ctx, struct ubus_object *obj,
1092                    struct ubus_request_data *req, const char *method,
1093                    struct blob_attr *msg)
1094
1095 {
1096         struct hostapd_data *hapd = get_hapd_from_object(obj);
1097         struct blob_attr *tb[__BSS_MGMT_EN_MAX];
1098         struct blob_attr *cur;
1099         uint32_t flags = 0;
1100         int i;
1101         bool neigh = false, beacon = false;
1102
1103         blobmsg_parse(bss_mgmt_enable_policy, __BSS_MGMT_EN_MAX, tb, blob_data(msg), blob_len(msg));
1104
1105         for (i = 0; i < ARRAY_SIZE(tb); i++) {
1106                 if (!tb[i] || !blobmsg_get_bool(tb[i]))
1107                         continue;
1108
1109                 flags |= (1 << i);
1110         }
1111
1112         __hostapd_bss_mgmt_enable(hapd, flags);
1113 }
1114
1115
1116 static void
1117 hostapd_rrm_nr_enable(struct hostapd_data *hapd)
1118 {
1119         __hostapd_bss_mgmt_enable(hapd, 1 << BSS_MGMT_EN_NEIGHBOR);
1120 }
1121
1122 static int
1123 hostapd_rrm_nr_get_own(struct ubus_context *ctx, struct ubus_object *obj,
1124                        struct ubus_request_data *req, const char *method,
1125                        struct blob_attr *msg)
1126 {
1127         struct hostapd_data *hapd = get_hapd_from_object(obj);
1128         struct hostapd_neighbor_entry *nr;
1129         void *c;
1130
1131         hostapd_rrm_nr_enable(hapd);
1132
1133         nr = hostapd_neighbor_get(hapd, hapd->own_addr, NULL);
1134         if (!nr)
1135                 return UBUS_STATUS_NOT_FOUND;
1136
1137         blob_buf_init(&b, 0);
1138
1139         c = blobmsg_open_array(&b, "value");
1140         hostapd_rrm_print_nr(nr);
1141         blobmsg_close_array(&b, c);
1142
1143         ubus_send_reply(ctx, req, b.head);
1144
1145         return 0;
1146 }
1147
1148 static int
1149 hostapd_rrm_nr_list(struct ubus_context *ctx, struct ubus_object *obj,
1150                     struct ubus_request_data *req, const char *method,
1151                     struct blob_attr *msg)
1152 {
1153         struct hostapd_data *hapd = get_hapd_from_object(obj);
1154         struct hostapd_neighbor_entry *nr;
1155         void *c;
1156
1157         hostapd_rrm_nr_enable(hapd);
1158         blob_buf_init(&b, 0);
1159
1160         c = blobmsg_open_array(&b, "list");
1161         dl_list_for_each(nr, &hapd->nr_db, struct hostapd_neighbor_entry, list) {
1162                 void *cur;
1163
1164                 if (!memcmp(nr->bssid, hapd->own_addr, ETH_ALEN))
1165                         continue;
1166
1167                 cur = blobmsg_open_array(&b, NULL);
1168                 hostapd_rrm_print_nr(nr);
1169                 blobmsg_close_array(&b, cur);
1170         }
1171         blobmsg_close_array(&b, c);
1172
1173         ubus_send_reply(ctx, req, b.head);
1174
1175         return 0;
1176 }
1177
1178 enum {
1179         NR_SET_LIST,
1180         __NR_SET_LIST_MAX
1181 };
1182
1183 static const struct blobmsg_policy nr_set_policy[__NR_SET_LIST_MAX] = {
1184         [NR_SET_LIST] = { "list", BLOBMSG_TYPE_ARRAY },
1185 };
1186
1187
1188 static void
1189 hostapd_rrm_nr_clear(struct hostapd_data *hapd)
1190 {
1191         struct hostapd_neighbor_entry *nr;
1192
1193 restart:
1194         dl_list_for_each(nr, &hapd->nr_db, struct hostapd_neighbor_entry, list) {
1195                 if (!memcmp(nr->bssid, hapd->own_addr, ETH_ALEN))
1196                         continue;
1197
1198                 hostapd_neighbor_remove(hapd, nr->bssid, &nr->ssid);
1199                 goto restart;
1200         }
1201 }
1202
1203 static int
1204 hostapd_rrm_nr_set(struct ubus_context *ctx, struct ubus_object *obj,
1205                    struct ubus_request_data *req, const char *method,
1206                    struct blob_attr *msg)
1207 {
1208         static const struct blobmsg_policy nr_e_policy[] = {
1209                 { .type = BLOBMSG_TYPE_STRING },
1210                 { .type = BLOBMSG_TYPE_STRING },
1211                 { .type = BLOBMSG_TYPE_STRING },
1212         };
1213         struct hostapd_data *hapd = get_hapd_from_object(obj);
1214         struct blob_attr *tb_l[__NR_SET_LIST_MAX];
1215         struct blob_attr *tb[ARRAY_SIZE(nr_e_policy)];
1216         struct blob_attr *cur;
1217         int rem;
1218
1219         hostapd_rrm_nr_enable(hapd);
1220
1221         blobmsg_parse(nr_set_policy, __NR_SET_LIST_MAX, tb_l, blob_data(msg), blob_len(msg));
1222         if (!tb_l[NR_SET_LIST])
1223                 return UBUS_STATUS_INVALID_ARGUMENT;
1224
1225         hostapd_rrm_nr_clear(hapd);
1226         blobmsg_for_each_attr(cur, tb_l[NR_SET_LIST], rem) {
1227                 struct wpa_ssid_value ssid;
1228                 struct wpabuf *data;
1229                 u8 bssid[ETH_ALEN];
1230                 char *s, *nr_s;
1231
1232                 blobmsg_parse_array(nr_e_policy, ARRAY_SIZE(nr_e_policy), tb, blobmsg_data(cur), blobmsg_data_len(cur));
1233                 if (!tb[0] || !tb[1] || !tb[2])
1234                         goto invalid;
1235
1236                 /* Neighbor Report binary */
1237                 nr_s = blobmsg_get_string(tb[2]);
1238                 data = wpabuf_parse_bin(nr_s);
1239                 if (!data)
1240                         goto invalid;
1241
1242                 /* BSSID */
1243                 s = blobmsg_get_string(tb[0]);
1244                 if (strlen(s) == 0) {
1245                         /* Copy BSSID from neighbor report */
1246                         if (hwaddr_compact_aton(nr_s, bssid))
1247                                 goto invalid;
1248                 } else if (hwaddr_aton(s, bssid)) {
1249                         goto invalid;
1250                 }
1251
1252                 /* SSID */
1253                 s = blobmsg_get_string(tb[1]);
1254                 if (strlen(s) == 0) {
1255                         /* Copy SSID from hostapd BSS conf */
1256                         memcpy(&ssid, &hapd->conf->ssid, sizeof(ssid));
1257                 } else {
1258                         ssid.ssid_len = strlen(s);
1259                         if (ssid.ssid_len > sizeof(ssid.ssid))
1260                                 goto invalid;
1261
1262                         memcpy(&ssid, s, ssid.ssid_len);
1263                 }
1264
1265                 hostapd_neighbor_set(hapd, bssid, &ssid, data, NULL, NULL, 0, 0);
1266                 wpabuf_free(data);
1267                 continue;
1268
1269 invalid:
1270                 return UBUS_STATUS_INVALID_ARGUMENT;
1271         }
1272
1273         return 0;
1274 }
1275
1276 enum {
1277         BEACON_REQ_ADDR,
1278         BEACON_REQ_MODE,
1279         BEACON_REQ_OP_CLASS,
1280         BEACON_REQ_CHANNEL,
1281         BEACON_REQ_DURATION,
1282         BEACON_REQ_BSSID,
1283         BEACON_REQ_SSID,
1284         __BEACON_REQ_MAX,
1285 };
1286
1287 static const struct blobmsg_policy beacon_req_policy[__BEACON_REQ_MAX] = {
1288         [BEACON_REQ_ADDR] = { "addr", BLOBMSG_TYPE_STRING },
1289         [BEACON_REQ_OP_CLASS] { "op_class", BLOBMSG_TYPE_INT32 },
1290         [BEACON_REQ_CHANNEL] { "channel", BLOBMSG_TYPE_INT32 },
1291         [BEACON_REQ_DURATION] { "duration", BLOBMSG_TYPE_INT32 },
1292         [BEACON_REQ_MODE] { "mode", BLOBMSG_TYPE_INT32 },
1293         [BEACON_REQ_BSSID] { "bssid", BLOBMSG_TYPE_STRING },
1294         [BEACON_REQ_SSID] { "ssid", BLOBMSG_TYPE_STRING },
1295 };
1296
1297 static int
1298 hostapd_rrm_beacon_req(struct ubus_context *ctx, struct ubus_object *obj,
1299                        struct ubus_request_data *ureq, const char *method,
1300                        struct blob_attr *msg)
1301 {
1302         struct hostapd_data *hapd = container_of(obj, struct hostapd_data, ubus.obj);
1303         struct blob_attr *tb[__BEACON_REQ_MAX];
1304         struct blob_attr *cur;
1305         struct wpabuf *req;
1306         u8 bssid[ETH_ALEN] = { 0xff, 0xff, 0xff, 0xff, 0xff, 0xff };
1307         u8 addr[ETH_ALEN];
1308         int mode, rem, ret;
1309         int buf_len = 13;
1310
1311         blobmsg_parse(beacon_req_policy, __BEACON_REQ_MAX, tb, blob_data(msg), blob_len(msg));
1312
1313         if (!tb[BEACON_REQ_ADDR] || !tb[BEACON_REQ_MODE] || !tb[BEACON_REQ_DURATION] ||
1314             !tb[BEACON_REQ_OP_CLASS] || !tb[BEACON_REQ_CHANNEL])
1315                 return UBUS_STATUS_INVALID_ARGUMENT;
1316
1317         if (tb[BEACON_REQ_SSID])
1318                 buf_len += blobmsg_data_len(tb[BEACON_REQ_SSID]) + 2 - 1;
1319
1320         mode = blobmsg_get_u32(tb[BEACON_REQ_MODE]);
1321         if (hwaddr_aton(blobmsg_data(tb[BEACON_REQ_ADDR]), addr))
1322                 return UBUS_STATUS_INVALID_ARGUMENT;
1323
1324         if (tb[BEACON_REQ_BSSID] &&
1325             hwaddr_aton(blobmsg_data(tb[BEACON_REQ_BSSID]), bssid))
1326                 return UBUS_STATUS_INVALID_ARGUMENT;
1327
1328         req = wpabuf_alloc(buf_len);
1329         if (!req)
1330                 return UBUS_STATUS_UNKNOWN_ERROR;
1331
1332         /* 1: regulatory class */
1333         wpabuf_put_u8(req, blobmsg_get_u32(tb[BEACON_REQ_OP_CLASS]));
1334
1335         /* 2: channel number */
1336         wpabuf_put_u8(req, blobmsg_get_u32(tb[BEACON_REQ_CHANNEL]));
1337
1338         /* 3-4: randomization interval */
1339         wpabuf_put_le16(req, 0);
1340
1341         /* 5-6: duration */
1342         wpabuf_put_le16(req, blobmsg_get_u32(tb[BEACON_REQ_DURATION]));
1343
1344         /* 7: mode */
1345         wpabuf_put_u8(req, blobmsg_get_u32(tb[BEACON_REQ_MODE]));
1346
1347         /* 8-13: BSSID */
1348         wpabuf_put_data(req, bssid, ETH_ALEN);
1349
1350         if ((cur = tb[BEACON_REQ_SSID]) != NULL) {
1351                 wpabuf_put_u8(req, WLAN_EID_SSID);
1352                 wpabuf_put_u8(req, blobmsg_data_len(cur) - 1);
1353                 wpabuf_put_data(req, blobmsg_data(cur), blobmsg_data_len(cur) - 1);
1354         }
1355
1356         ret = hostapd_send_beacon_req(hapd, addr, 0, req);
1357         if (ret < 0)
1358                 return -ret;
1359
1360         return 0;
1361 }
1362
1363 enum {
1364         LM_REQ_ADDR,
1365         LM_REQ_TX_POWER_USED,
1366         LM_REQ_TX_POWER_MAX,
1367         __LM_REQ_MAX,
1368 };
1369
1370 static const struct blobmsg_policy lm_req_policy[__LM_REQ_MAX] = {
1371         [LM_REQ_ADDR] = { "addr", BLOBMSG_TYPE_STRING },
1372         [LM_REQ_TX_POWER_USED] = { "tx-power-used", BLOBMSG_TYPE_INT32 },
1373         [LM_REQ_TX_POWER_MAX] = { "tx-power-max", BLOBMSG_TYPE_INT32 },
1374 };
1375
1376 static int
1377 hostapd_rrm_lm_req(struct ubus_context *ctx, struct ubus_object *obj,
1378                    struct ubus_request_data *ureq, const char *method,
1379                    struct blob_attr *msg)
1380 {
1381         struct hostapd_data *hapd = container_of(obj, struct hostapd_data, ubus.obj);
1382         struct blob_attr *tb[__LM_REQ_MAX];
1383         struct wpabuf *buf;
1384         u8 addr[ETH_ALEN];
1385         int ret;
1386         int8_t txp_used, txp_max;
1387
1388         txp_used = 0;
1389         txp_max = 0;
1390
1391         blobmsg_parse(lm_req_policy, __LM_REQ_MAX, tb, blob_data(msg), blob_len(msg));
1392
1393         if (!tb[LM_REQ_ADDR])
1394                 return UBUS_STATUS_INVALID_ARGUMENT;
1395
1396         if (tb[LM_REQ_TX_POWER_USED])
1397                 txp_used = (int8_t) blobmsg_get_u32(tb[LM_REQ_TX_POWER_USED]);
1398
1399         if (tb[LM_REQ_TX_POWER_MAX])
1400                 txp_max = (int8_t) blobmsg_get_u32(tb[LM_REQ_TX_POWER_MAX]);
1401
1402         if (hwaddr_aton(blobmsg_data(tb[LM_REQ_ADDR]), addr))
1403                 return UBUS_STATUS_INVALID_ARGUMENT;
1404
1405         buf = wpabuf_alloc(5);
1406         if (!buf)
1407                 return UBUS_STATUS_UNKNOWN_ERROR;
1408
1409         wpabuf_put_u8(buf, WLAN_ACTION_RADIO_MEASUREMENT);
1410         wpabuf_put_u8(buf, WLAN_RRM_LINK_MEASUREMENT_REQUEST);
1411         wpabuf_put_u8(buf, 1);
1412         /* TX-Power used */
1413         wpabuf_put_u8(buf, txp_used);
1414         /* Max TX Power */
1415         wpabuf_put_u8(buf, txp_max);
1416
1417         ret = hostapd_drv_send_action(hapd, hapd->iface->freq, 0, addr,
1418                                       wpabuf_head(buf), wpabuf_len(buf));
1419
1420         wpabuf_free(buf);
1421         if (ret < 0)
1422                 return -ret;
1423
1424         return 0;
1425 }
1426
1427
1428 void hostapd_ubus_handle_link_measurement(struct hostapd_data *hapd, const u8 *data, size_t len)
1429 {
1430         const struct ieee80211_mgmt *mgmt = (const struct ieee80211_mgmt *) data;
1431         const u8 *pos, *end;
1432         u8 token;
1433
1434         end = data + len;
1435         token = mgmt->u.action.u.rrm.dialog_token;
1436         pos = mgmt->u.action.u.rrm.variable;
1437
1438         if (end - pos < 8)
1439                 return;
1440
1441         if (!hapd->ubus.obj.has_subscribers)
1442                 return;
1443
1444         blob_buf_init(&b, 0);
1445         blobmsg_add_macaddr(&b, "address", mgmt->sa);
1446         blobmsg_add_u16(&b, "dialog-token", token);
1447         blobmsg_add_u16(&b, "rx-antenna-id", pos[4]);
1448         blobmsg_add_u16(&b, "tx-antenna-id", pos[5]);
1449         blobmsg_add_u16(&b, "rcpi", pos[6]);
1450         blobmsg_add_u16(&b, "rsni", pos[7]);
1451
1452         ubus_notify(ctx, &hapd->ubus.obj, "link-measurement-report", b.head, -1);
1453 }
1454
1455
1456 #ifdef CONFIG_WNM_AP
1457
1458 static int
1459 hostapd_bss_tr_send(struct hostapd_data *hapd, u8 *addr, bool disassoc_imminent, bool abridged,
1460                     u16 disassoc_timer, u8 validity_period, u8 dialog_token,
1461                     struct blob_attr *neighbors, u8 mbo_reason, u8 cell_pref, u8 reassoc_delay)
1462 {
1463         struct blob_attr *cur;
1464         struct sta_info *sta;
1465         int nr_len = 0;
1466         int rem;
1467         u8 *nr = NULL;
1468         u8 req_mode = 0;
1469         u8 mbo[10];
1470         size_t mbo_len = 0;
1471
1472         sta = ap_get_sta(hapd, addr);
1473         if (!sta)
1474                 return UBUS_STATUS_NOT_FOUND;
1475
1476         if (neighbors) {
1477                 u8 *nr_cur;
1478
1479                 if (blobmsg_check_array(neighbors,
1480                                         BLOBMSG_TYPE_STRING) < 0)
1481                         return UBUS_STATUS_INVALID_ARGUMENT;
1482
1483                 blobmsg_for_each_attr(cur, neighbors, rem) {
1484                         int len = strlen(blobmsg_get_string(cur));
1485
1486                         if (len % 2)
1487                                 return UBUS_STATUS_INVALID_ARGUMENT;
1488
1489                         nr_len += (len / 2) + 2;
1490                 }
1491
1492                 if (nr_len) {
1493                         nr = os_zalloc(nr_len);
1494                         if (!nr)
1495                                 return UBUS_STATUS_UNKNOWN_ERROR;
1496                 }
1497
1498                 nr_cur = nr;
1499                 blobmsg_for_each_attr(cur, neighbors, rem) {
1500                         int len = strlen(blobmsg_get_string(cur)) / 2;
1501
1502                         *nr_cur++ = WLAN_EID_NEIGHBOR_REPORT;
1503                         *nr_cur++ = (u8) len;
1504                         if (hexstr2bin(blobmsg_data(cur), nr_cur, len)) {
1505                                 free(nr);
1506                                 return UBUS_STATUS_INVALID_ARGUMENT;
1507                         }
1508
1509                         nr_cur += len;
1510                 }
1511         }
1512
1513         if (nr)
1514                 req_mode |= WNM_BSS_TM_REQ_PREF_CAND_LIST_INCLUDED;
1515
1516         if (abridged)
1517                 req_mode |= WNM_BSS_TM_REQ_ABRIDGED;
1518
1519         if (disassoc_imminent)
1520                 req_mode |= WNM_BSS_TM_REQ_DISASSOC_IMMINENT;
1521
1522 #ifdef CONFIG_MBO
1523         u8 *mbo_pos = mbo;
1524
1525         if (mbo_reason > MBO_TRANSITION_REASON_PREMIUM_AP)
1526                 return UBUS_STATUS_INVALID_ARGUMENT;
1527
1528         if (cell_pref != 0 && cell_pref != 1 && cell_pref != 255)
1529                 return UBUS_STATUS_INVALID_ARGUMENT;
1530
1531         if (reassoc_delay > 65535 || (reassoc_delay && !disassoc_imminent))
1532                 return UBUS_STATUS_INVALID_ARGUMENT;
1533
1534         *mbo_pos++ = MBO_ATTR_ID_TRANSITION_REASON;
1535         *mbo_pos++ = 1;
1536         *mbo_pos++ = mbo_reason;
1537         *mbo_pos++ = MBO_ATTR_ID_CELL_DATA_PREF;
1538         *mbo_pos++ = 1;
1539         *mbo_pos++ = cell_pref;
1540
1541         if (reassoc_delay) {
1542                 *mbo_pos++ = MBO_ATTR_ID_ASSOC_RETRY_DELAY;
1543                 *mbo_pos++ = 2;
1544                 WPA_PUT_LE16(mbo_pos, reassoc_delay);
1545                 mbo_pos += 2;
1546         }
1547
1548         mbo_len = mbo_pos - mbo;
1549 #endif
1550
1551         if (wnm_send_bss_tm_req(hapd, sta, req_mode, disassoc_timer, validity_period, NULL,
1552                                 dialog_token, NULL, nr, nr_len, mbo_len ? mbo : NULL, mbo_len))
1553                 return UBUS_STATUS_UNKNOWN_ERROR;
1554
1555         return 0;
1556 }
1557
1558 enum {
1559         BSS_TR_ADDR,
1560         BSS_TR_DA_IMMINENT,
1561         BSS_TR_DA_TIMER,
1562         BSS_TR_VALID_PERIOD,
1563         BSS_TR_NEIGHBORS,
1564         BSS_TR_ABRIDGED,
1565         BSS_TR_DIALOG_TOKEN,
1566 #ifdef CONFIG_MBO
1567         BSS_TR_MBO_REASON,
1568         BSS_TR_CELL_PREF,
1569         BSS_TR_REASSOC_DELAY,
1570 #endif
1571         __BSS_TR_DISASSOC_MAX
1572 };
1573
1574 static const struct blobmsg_policy bss_tr_policy[__BSS_TR_DISASSOC_MAX] = {
1575         [BSS_TR_ADDR] = { "addr", BLOBMSG_TYPE_STRING },
1576         [BSS_TR_DA_IMMINENT] = { "disassociation_imminent", BLOBMSG_TYPE_BOOL },
1577         [BSS_TR_DA_TIMER] = { "disassociation_timer", BLOBMSG_TYPE_INT32 },
1578         [BSS_TR_VALID_PERIOD] = { "validity_period", BLOBMSG_TYPE_INT32 },
1579         [BSS_TR_NEIGHBORS] = { "neighbors", BLOBMSG_TYPE_ARRAY },
1580         [BSS_TR_ABRIDGED] = { "abridged", BLOBMSG_TYPE_BOOL },
1581         [BSS_TR_DIALOG_TOKEN] = { "dialog_token", BLOBMSG_TYPE_INT32 },
1582 #ifdef CONFIG_MBO
1583         [BSS_TR_MBO_REASON] = { "mbo_reason", BLOBMSG_TYPE_INT32 },
1584         [BSS_TR_CELL_PREF] = { "cell_pref", BLOBMSG_TYPE_INT32 },
1585         [BSS_TR_REASSOC_DELAY] = { "reassoc_delay", BLOBMSG_TYPE_INT32 },
1586 #endif
1587 };
1588
1589 static int
1590 hostapd_bss_transition_request(struct ubus_context *ctx, struct ubus_object *obj,
1591                                struct ubus_request_data *ureq, const char *method,
1592                                struct blob_attr *msg)
1593 {
1594         struct hostapd_data *hapd = container_of(obj, struct hostapd_data, ubus.obj);
1595         struct blob_attr *tb[__BSS_TR_DISASSOC_MAX];
1596         struct sta_info *sta;
1597         u32 da_timer = 0;
1598         u32 valid_period = 0;
1599         u8 addr[ETH_ALEN];
1600         u32 dialog_token = 1;
1601         bool abridged;
1602         bool da_imminent;
1603         u8 mbo_reason;
1604         u8 cell_pref;
1605         u8 reassoc_delay;
1606
1607         blobmsg_parse(bss_tr_policy, __BSS_TR_DISASSOC_MAX, tb, blob_data(msg), blob_len(msg));
1608
1609         if (!tb[BSS_TR_ADDR])
1610                 return UBUS_STATUS_INVALID_ARGUMENT;
1611
1612         if (hwaddr_aton(blobmsg_data(tb[BSS_TR_ADDR]), addr))
1613                 return UBUS_STATUS_INVALID_ARGUMENT;
1614
1615         if (tb[BSS_TR_DA_TIMER])
1616                 da_timer = blobmsg_get_u32(tb[BSS_TR_DA_TIMER]);
1617
1618         if (tb[BSS_TR_VALID_PERIOD])
1619                 valid_period = blobmsg_get_u32(tb[BSS_TR_VALID_PERIOD]);
1620
1621         if (tb[BSS_TR_DIALOG_TOKEN])
1622                 dialog_token = blobmsg_get_u32(tb[BSS_TR_DIALOG_TOKEN]);
1623
1624         da_imminent = !!(tb[BSS_TR_DA_IMMINENT] && blobmsg_get_bool(tb[BSS_TR_DA_IMMINENT]));
1625         abridged = !!(tb[BSS_TR_ABRIDGED] && blobmsg_get_bool(tb[BSS_TR_ABRIDGED]));
1626
1627 #ifdef CONFIG_MBO
1628         if (tb[BSS_TR_MBO_REASON])
1629                 mbo_reason = blobmsg_get_u32(tb[BSS_TR_MBO_REASON]);
1630
1631         if (tb[BSS_TR_CELL_PREF])
1632                 cell_pref = blobmsg_get_u32(tb[BSS_TR_CELL_PREF]);
1633
1634         if (tb[BSS_TR_REASSOC_DELAY])
1635                 reassoc_delay = blobmsg_get_u32(tb[BSS_TR_REASSOC_DELAY]);
1636 #endif
1637
1638         return hostapd_bss_tr_send(hapd, addr, da_imminent, abridged, da_timer, valid_period,
1639                                    dialog_token, tb[BSS_TR_NEIGHBORS], mbo_reason, cell_pref, reassoc_delay);
1640 }
1641 #endif
1642
1643 #ifdef CONFIG_AIRTIME_POLICY
1644 enum {
1645         UPDATE_AIRTIME_STA,
1646         UPDATE_AIRTIME_WEIGHT,
1647         __UPDATE_AIRTIME_MAX,
1648 };
1649
1650
1651 static const struct blobmsg_policy airtime_policy[__UPDATE_AIRTIME_MAX] = {
1652         [UPDATE_AIRTIME_STA] = { "sta", BLOBMSG_TYPE_STRING },
1653         [UPDATE_AIRTIME_WEIGHT] = { "weight", BLOBMSG_TYPE_INT32 },
1654 };
1655
1656 static int
1657 hostapd_bss_update_airtime(struct ubus_context *ctx, struct ubus_object *obj,
1658                            struct ubus_request_data *ureq, const char *method,
1659                            struct blob_attr *msg)
1660 {
1661         struct hostapd_data *hapd = container_of(obj, struct hostapd_data, ubus.obj);
1662         struct blob_attr *tb[__UPDATE_AIRTIME_MAX];
1663         struct sta_info *sta = NULL;
1664         u8 addr[ETH_ALEN];
1665         int weight;
1666
1667         blobmsg_parse(airtime_policy, __UPDATE_AIRTIME_MAX, tb, blob_data(msg), blob_len(msg));
1668
1669         if (!tb[UPDATE_AIRTIME_WEIGHT])
1670                 return UBUS_STATUS_INVALID_ARGUMENT;
1671
1672         weight = blobmsg_get_u32(tb[UPDATE_AIRTIME_WEIGHT]);
1673
1674         if (!tb[UPDATE_AIRTIME_STA]) {
1675                 if (!weight)
1676                         return UBUS_STATUS_INVALID_ARGUMENT;
1677
1678                 hapd->conf->airtime_weight = weight;
1679                 return 0;
1680         }
1681
1682         if (hwaddr_aton(blobmsg_data(tb[UPDATE_AIRTIME_STA]), addr))
1683                 return UBUS_STATUS_INVALID_ARGUMENT;
1684
1685         sta = ap_get_sta(hapd, addr);
1686         if (!sta)
1687                 return UBUS_STATUS_NOT_FOUND;
1688
1689         sta->dyn_airtime_weight = weight;
1690         airtime_policy_new_sta(hapd, sta);
1691
1692         return 0;
1693 }
1694 #endif
1695
1696
1697 static const struct ubus_method bss_methods[] = {
1698         UBUS_METHOD_NOARG("reload", hostapd_bss_reload),
1699         UBUS_METHOD_NOARG("get_clients", hostapd_bss_get_clients),
1700         UBUS_METHOD_NOARG("get_status", hostapd_bss_get_status),
1701         UBUS_METHOD("del_client", hostapd_bss_del_client, del_policy),
1702 #ifdef CONFIG_AIRTIME_POLICY
1703         UBUS_METHOD("update_airtime", hostapd_bss_update_airtime, airtime_policy),
1704 #endif
1705         UBUS_METHOD_NOARG("list_bans", hostapd_bss_list_bans),
1706 #ifdef CONFIG_WPS
1707         UBUS_METHOD_NOARG("wps_start", hostapd_bss_wps_start),
1708         UBUS_METHOD_NOARG("wps_status", hostapd_bss_wps_status),
1709         UBUS_METHOD_NOARG("wps_cancel", hostapd_bss_wps_cancel),
1710 #endif
1711         UBUS_METHOD_NOARG("update_beacon", hostapd_bss_update_beacon),
1712         UBUS_METHOD_NOARG("get_features", hostapd_bss_get_features),
1713 #ifdef NEED_AP_MLME
1714         UBUS_METHOD("switch_chan", hostapd_switch_chan, csa_policy),
1715 #endif
1716         UBUS_METHOD("set_vendor_elements", hostapd_vendor_elements, ve_policy),
1717         UBUS_METHOD("notify_response", hostapd_notify_response, notify_policy),
1718         UBUS_METHOD("bss_mgmt_enable", hostapd_bss_mgmt_enable, bss_mgmt_enable_policy),
1719         UBUS_METHOD_NOARG("rrm_nr_get_own", hostapd_rrm_nr_get_own),
1720         UBUS_METHOD_NOARG("rrm_nr_list", hostapd_rrm_nr_list),
1721         UBUS_METHOD("rrm_nr_set", hostapd_rrm_nr_set, nr_set_policy),
1722         UBUS_METHOD("rrm_beacon_req", hostapd_rrm_beacon_req, beacon_req_policy),
1723         UBUS_METHOD("link_measurement_req", hostapd_rrm_lm_req, lm_req_policy),
1724 #ifdef CONFIG_WNM_AP
1725         UBUS_METHOD("bss_transition_request", hostapd_bss_transition_request, bss_tr_policy),
1726 #endif
1727 };
1728
1729 static struct ubus_object_type bss_object_type =
1730         UBUS_OBJECT_TYPE("hostapd_bss", bss_methods);
1731
1732 static int avl_compare_macaddr(const void *k1, const void *k2, void *ptr)
1733 {
1734         return memcmp(k1, k2, ETH_ALEN);
1735 }
1736
1737 void hostapd_ubus_add_bss(struct hostapd_data *hapd)
1738 {
1739         struct ubus_object *obj = &hapd->ubus.obj;
1740         char *name;
1741         int ret;
1742
1743 #ifdef CONFIG_MESH
1744         if (hapd->conf->mesh & MESH_ENABLED)
1745                 return;
1746 #endif
1747
1748         if (!hostapd_ubus_init())
1749                 return;
1750
1751         if (asprintf(&name, "hostapd.%s", hapd->conf->iface) < 0)
1752                 return;
1753
1754         avl_init(&hapd->ubus.banned, avl_compare_macaddr, false, NULL);
1755         obj->name = name;
1756         obj->type = &bss_object_type;
1757         obj->methods = bss_object_type.methods;
1758         obj->n_methods = bss_object_type.n_methods;
1759         ret = ubus_add_object(ctx, obj);
1760         hostapd_ubus_ref_inc();
1761
1762         hostapd_send_shared_event(&hapd->iface->interfaces->ubus, hapd->conf->iface, "add");
1763 }
1764
1765 void hostapd_ubus_free_bss(struct hostapd_data *hapd)
1766 {
1767         struct ubus_object *obj = &hapd->ubus.obj;
1768         char *name = (char *) obj->name;
1769
1770 #ifdef CONFIG_MESH
1771         if (hapd->conf->mesh & MESH_ENABLED)
1772                 return;
1773 #endif
1774
1775         if (!ctx)
1776                 return;
1777
1778         hostapd_send_shared_event(&hapd->iface->interfaces->ubus, hapd->conf->iface, "remove");
1779
1780         if (obj->id) {
1781                 ubus_remove_object(ctx, obj);
1782                 hostapd_ubus_ref_dec();
1783         }
1784
1785         free(name);
1786 }
1787
1788 static void
1789 hostapd_ubus_vlan_action(struct hostapd_data *hapd, struct hostapd_vlan *vlan,
1790                          const char *action)
1791 {
1792         struct vlan_description *desc = &vlan->vlan_desc;
1793         void *c;
1794         int i;
1795
1796         if (!hapd->ubus.obj.has_subscribers)
1797                 return;
1798
1799         blob_buf_init(&b, 0);
1800         blobmsg_add_string(&b, "ifname", vlan->ifname);
1801         blobmsg_add_string(&b, "bridge", vlan->bridge);
1802         blobmsg_add_u32(&b, "vlan_id", vlan->vlan_id);
1803
1804         if (desc->notempty) {
1805                 blobmsg_add_u32(&b, "untagged", desc->untagged);
1806                 c = blobmsg_open_array(&b, "tagged");
1807                 for (i = 0; i < ARRAY_SIZE(desc->tagged) && desc->tagged[i]; i++)
1808                         blobmsg_add_u32(&b, "", desc->tagged[i]);
1809                 blobmsg_close_array(&b, c);
1810         }
1811
1812         ubus_notify(ctx, &hapd->ubus.obj, action, b.head, -1);
1813 }
1814
1815 void hostapd_ubus_add_vlan(struct hostapd_data *hapd, struct hostapd_vlan *vlan)
1816 {
1817         hostapd_ubus_vlan_action(hapd, vlan, "vlan_add");
1818 }
1819
1820 void hostapd_ubus_remove_vlan(struct hostapd_data *hapd, struct hostapd_vlan *vlan)
1821 {
1822         hostapd_ubus_vlan_action(hapd, vlan, "vlan_remove");
1823 }
1824
1825 static const struct ubus_method daemon_methods[] = {
1826         UBUS_METHOD("config_add", hostapd_config_add, config_add_policy),
1827         UBUS_METHOD("config_remove", hostapd_config_remove, config_remove_policy),
1828 };
1829
1830 static struct ubus_object_type daemon_object_type =
1831         UBUS_OBJECT_TYPE("hostapd", daemon_methods);
1832
1833 void hostapd_ubus_add(struct hapd_interfaces *interfaces)
1834 {
1835         struct ubus_object *obj = &interfaces->ubus;
1836         int ret;
1837
1838         if (!hostapd_ubus_init())
1839                 return;
1840
1841         obj->name = strdup("hostapd");
1842
1843         obj->type = &daemon_object_type;
1844         obj->methods = daemon_object_type.methods;
1845         obj->n_methods = daemon_object_type.n_methods;
1846         ret = ubus_add_object(ctx, obj);
1847         hostapd_ubus_ref_inc();
1848 }
1849
1850 void hostapd_ubus_free(struct hapd_interfaces *interfaces)
1851 {
1852         struct ubus_object *obj = &interfaces->ubus;
1853         char *name = (char *) obj->name;
1854
1855         if (!ctx)
1856                 return;
1857
1858         if (obj->id) {
1859                 ubus_remove_object(ctx, obj);
1860                 hostapd_ubus_ref_dec();
1861         }
1862
1863         free(name);
1864 }
1865
1866 struct ubus_event_req {
1867         struct ubus_notify_request nreq;
1868         int resp;
1869 };
1870
1871 static void
1872 ubus_event_cb(struct ubus_notify_request *req, int idx, int ret)
1873 {
1874         struct ubus_event_req *ureq = container_of(req, struct ubus_event_req, nreq);
1875
1876         ureq->resp = ret;
1877 }
1878
1879 int hostapd_ubus_handle_event(struct hostapd_data *hapd, struct hostapd_ubus_request *req)
1880 {
1881         struct ubus_banned_client *ban;
1882         const char *types[HOSTAPD_UBUS_TYPE_MAX] = {
1883                 [HOSTAPD_UBUS_PROBE_REQ] = "probe",
1884                 [HOSTAPD_UBUS_AUTH_REQ] = "auth",
1885                 [HOSTAPD_UBUS_ASSOC_REQ] = "assoc",
1886         };
1887         const char *type = "mgmt";
1888         struct ubus_event_req ureq = {};
1889         const u8 *addr;
1890
1891         if (req->mgmt_frame)
1892                 addr = req->mgmt_frame->sa;
1893         else
1894                 addr = req->addr;
1895
1896         ban = avl_find_element(&hapd->ubus.banned, addr, ban, avl);
1897         if (ban)
1898                 return WLAN_STATUS_AP_UNABLE_TO_HANDLE_NEW_STA;
1899
1900         if (!hapd->ubus.obj.has_subscribers)
1901                 return WLAN_STATUS_SUCCESS;
1902
1903         if (req->type < ARRAY_SIZE(types))
1904                 type = types[req->type];
1905
1906         blob_buf_init(&b, 0);
1907         blobmsg_add_macaddr(&b, "address", addr);
1908         if (req->mgmt_frame)
1909                 blobmsg_add_macaddr(&b, "target", req->mgmt_frame->da);
1910         if (req->ssi_signal)
1911                 blobmsg_add_u32(&b, "signal", req->ssi_signal);
1912         blobmsg_add_u32(&b, "freq", hapd->iface->freq);
1913
1914         if (req->elems) {
1915                 if(req->elems->ht_capabilities)
1916                 {
1917                         struct ieee80211_ht_capabilities *ht_capabilities;
1918                         void *ht_cap, *ht_cap_mcs_set, *mcs_set;
1919
1920
1921                         ht_capabilities = (struct ieee80211_ht_capabilities*) req->elems->ht_capabilities;
1922                         ht_cap = blobmsg_open_table(&b, "ht_capabilities");
1923                         blobmsg_add_u16(&b, "ht_capabilities_info", ht_capabilities->ht_capabilities_info);
1924                         ht_cap_mcs_set = blobmsg_open_table(&b, "supported_mcs_set");
1925                         blobmsg_add_u16(&b, "a_mpdu_params", ht_capabilities->a_mpdu_params);
1926                         blobmsg_add_u16(&b, "ht_extended_capabilities", ht_capabilities->ht_extended_capabilities);
1927                         blobmsg_add_u32(&b, "tx_bf_capability_info", ht_capabilities->tx_bf_capability_info);
1928                         blobmsg_add_u16(&b, "asel_capabilities", ht_capabilities->asel_capabilities);
1929                         mcs_set = blobmsg_open_array(&b, "supported_mcs_set");
1930                         for (int i = 0; i < 16; i++) {
1931                                 blobmsg_add_u16(&b, NULL, (u16) ht_capabilities->supported_mcs_set[i]);
1932                         }
1933                         blobmsg_close_array(&b, mcs_set);
1934                         blobmsg_close_table(&b, ht_cap_mcs_set);
1935                         blobmsg_close_table(&b, ht_cap);
1936                 }
1937                 if(req->elems->vht_capabilities)
1938                 {
1939                         struct ieee80211_vht_capabilities *vht_capabilities;
1940                         void *vht_cap, *vht_cap_mcs_set;
1941
1942                         vht_capabilities = (struct ieee80211_vht_capabilities*) req->elems->vht_capabilities;
1943                         vht_cap = blobmsg_open_table(&b, "vht_capabilities");
1944                         blobmsg_add_u32(&b, "vht_capabilities_info", vht_capabilities->vht_capabilities_info);
1945                         vht_cap_mcs_set = blobmsg_open_table(&b, "vht_supported_mcs_set");
1946                         blobmsg_add_u16(&b, "rx_map", vht_capabilities->vht_supported_mcs_set.rx_map);
1947                         blobmsg_add_u16(&b, "rx_highest", vht_capabilities->vht_supported_mcs_set.rx_highest);
1948                         blobmsg_add_u16(&b, "tx_map", vht_capabilities->vht_supported_mcs_set.tx_map);
1949                         blobmsg_add_u16(&b, "tx_highest", vht_capabilities->vht_supported_mcs_set.tx_highest);
1950                         blobmsg_close_table(&b, vht_cap_mcs_set);
1951                         blobmsg_close_table(&b, vht_cap);
1952                 }
1953         }
1954
1955         if (!hapd->ubus.notify_response) {
1956                 ubus_notify(ctx, &hapd->ubus.obj, type, b.head, -1);
1957                 return WLAN_STATUS_SUCCESS;
1958         }
1959
1960         if (ubus_notify_async(ctx, &hapd->ubus.obj, type, b.head, &ureq.nreq))
1961                 return WLAN_STATUS_SUCCESS;
1962
1963         ureq.nreq.status_cb = ubus_event_cb;
1964         ubus_complete_request(ctx, &ureq.nreq.req, 100);
1965
1966         if (ureq.resp)
1967                 return ureq.resp;
1968
1969         return WLAN_STATUS_SUCCESS;
1970 }
1971
1972 void hostapd_ubus_notify(struct hostapd_data *hapd, const char *type, const u8 *addr)
1973 {
1974         if (!hapd->ubus.obj.has_subscribers)
1975                 return;
1976
1977         if (!addr)
1978                 return;
1979
1980         blob_buf_init(&b, 0);
1981         blobmsg_add_macaddr(&b, "address", addr);
1982
1983         ubus_notify(ctx, &hapd->ubus.obj, type, b.head, -1);
1984 }
1985
1986 void hostapd_ubus_notify_beacon_report(
1987         struct hostapd_data *hapd, const u8 *addr, u8 token, u8 rep_mode,
1988         struct rrm_measurement_beacon_report *rep, size_t len)
1989 {
1990         if (!hapd->ubus.obj.has_subscribers)
1991                 return;
1992
1993         if (!addr || !rep)
1994                 return;
1995
1996         blob_buf_init(&b, 0);
1997         blobmsg_add_macaddr(&b, "address", addr);
1998         blobmsg_add_u16(&b, "op-class", rep->op_class);
1999         blobmsg_add_u16(&b, "channel", rep->channel);
2000         blobmsg_add_u64(&b, "start-time", rep->start_time);
2001         blobmsg_add_u16(&b, "duration", rep->duration);
2002         blobmsg_add_u16(&b, "report-info", rep->report_info);
2003         blobmsg_add_u16(&b, "rcpi", rep->rcpi);
2004         blobmsg_add_u16(&b, "rsni", rep->rsni);
2005         blobmsg_add_macaddr(&b, "bssid", rep->bssid);
2006         blobmsg_add_u16(&b, "antenna-id", rep->antenna_id);
2007         blobmsg_add_u16(&b, "parent-tsf", rep->parent_tsf);
2008
2009         ubus_notify(ctx, &hapd->ubus.obj, "beacon-report", b.head, -1);
2010 }
2011
2012 void hostapd_ubus_notify_radar_detected(struct hostapd_iface *iface, int frequency,
2013                                         int chan_width, int cf1, int cf2)
2014 {
2015         struct hostapd_data *hapd;
2016         int i;
2017
2018         blob_buf_init(&b, 0);
2019         blobmsg_add_u16(&b, "frequency", frequency);
2020         blobmsg_add_u16(&b, "width", chan_width);
2021         blobmsg_add_u16(&b, "center1", cf1);
2022         blobmsg_add_u16(&b, "center2", cf2);
2023
2024         for (i = 0; i < iface->num_bss; i++) {
2025                 hapd = iface->bss[i];
2026                 ubus_notify(ctx, &hapd->ubus.obj, "radar-detected", b.head, -1);
2027         }
2028 }
2029
2030 #ifdef CONFIG_WNM_AP
2031 static void hostapd_ubus_notify_bss_transition_add_candidate_list(
2032         const u8 *candidate_list, u16 candidate_list_len)
2033 {
2034         char *cl_str;
2035         int i;
2036
2037         if (candidate_list_len == 0)
2038                 return;
2039
2040         cl_str = blobmsg_alloc_string_buffer(&b, "candidate-list", candidate_list_len * 2 + 1);
2041         for (i = 0; i < candidate_list_len; i++)
2042                 snprintf(&cl_str[i*2], 3, "%02X", candidate_list[i]);
2043         blobmsg_add_string_buffer(&b);
2044
2045 }
2046 #endif
2047
2048 void hostapd_ubus_notify_bss_transition_response(
2049         struct hostapd_data *hapd, const u8 *addr, u8 dialog_token, u8 status_code,
2050         u8 bss_termination_delay, const u8 *target_bssid,
2051         const u8 *candidate_list, u16 candidate_list_len)
2052 {
2053 #ifdef CONFIG_WNM_AP
2054         u16 i;
2055
2056         if (!hapd->ubus.obj.has_subscribers)
2057                 return;
2058
2059         if (!addr)
2060                 return;
2061
2062         blob_buf_init(&b, 0);
2063         blobmsg_add_macaddr(&b, "address", addr);
2064         blobmsg_add_u8(&b, "dialog-token", dialog_token);
2065         blobmsg_add_u8(&b, "status-code", status_code);
2066         blobmsg_add_u8(&b, "bss-termination-delay", bss_termination_delay);
2067         if (target_bssid)
2068                 blobmsg_add_macaddr(&b, "target-bssid", target_bssid);
2069
2070         hostapd_ubus_notify_bss_transition_add_candidate_list(candidate_list, candidate_list_len);
2071
2072         ubus_notify(ctx, &hapd->ubus.obj, "bss-transition-response", b.head, -1);
2073 #endif
2074 }
2075
2076 int hostapd_ubus_notify_bss_transition_query(
2077         struct hostapd_data *hapd, const u8 *addr, u8 dialog_token, u8 reason,
2078         const u8 *candidate_list, u16 candidate_list_len)
2079 {
2080 #ifdef CONFIG_WNM_AP
2081         struct ubus_event_req ureq = {};
2082         char *cl_str;
2083         u16 i;
2084
2085         if (!hapd->ubus.obj.has_subscribers)
2086                 return 0;
2087
2088         if (!addr)
2089                 return 0;
2090
2091         blob_buf_init(&b, 0);
2092         blobmsg_add_macaddr(&b, "address", addr);
2093         blobmsg_add_u8(&b, "dialog-token", dialog_token);
2094         blobmsg_add_u8(&b, "reason", reason);
2095         hostapd_ubus_notify_bss_transition_add_candidate_list(candidate_list, candidate_list_len);
2096
2097         if (!hapd->ubus.notify_response) {
2098                 ubus_notify(ctx, &hapd->ubus.obj, "bss-transition-query", b.head, -1);
2099                 return 0;
2100         }
2101
2102         if (ubus_notify_async(ctx, &hapd->ubus.obj, "bss-transition-query", b.head, &ureq.nreq))
2103                 return 0;
2104
2105         ureq.nreq.status_cb = ubus_event_cb;
2106         ubus_complete_request(ctx, &ureq.nreq.req, 100);
2107
2108         return ureq.resp;
2109 #endif
2110 }