package/network/services/hostapd/patches/600-ubus_support.patch

   1 --- a/hostapd/Makefile
   2 +++ b/hostapd/Makefile
   3 @@ -121,6 +121,11 @@ OBJS += ../src/common/hw_features_common
   4
   5  OBJS += ../src/eapol_auth/eapol_auth_sm.o
   6
   7 +ifdef CONFIG_UBUS
   8 +CFLAGS += -DUBUS_SUPPORT
   9 +OBJS += ../src/ap/ubus.o
  10 +LIBS += -lubox -lubus
  11 +endif
  12
  13  ifdef CONFIG_CODE_COVERAGE
  14  CFLAGS += -O0 -fprofile-arcs -ftest-coverage
  15 --- a/src/ap/hostapd.h
  16 +++ b/src/ap/hostapd.h
  17 @@ -13,6 +13,7 @@
  18  #include "utils/list.h"
  19  #include "ap_config.h"
  20  #include "drivers/driver.h"
  21 +#include "ubus.h"
  22
  23  struct wpa_ctrl_dst;
  24  struct radius_server_data;
  25 @@ -103,6 +104,7 @@ struct hostapd_data {
  26         struct hostapd_iface *iface;
  27         struct hostapd_config *iconf;
  28         struct hostapd_bss_config *conf;
  29 +       struct hostapd_ubus_bss ubus;
  30         int interface_added; /* virtual interface added for this BSS */
  31         unsigned int started:1;
  32         unsigned int disabled:1;
  33 @@ -286,6 +288,8 @@ struct hostapd_iface {
  34         struct hostapd_config *conf;
  35         char phy[16]; /* Name of the PHY (radio) */
  36
  37 +       struct hostapd_ubus_iface ubus;
  38 +
  39         enum hostapd_iface_state {
  40                 HAPD_IFACE_UNINITIALIZED,
  41                 HAPD_IFACE_DISABLED,
  42 --- /dev/null
  43 +++ b/src/ap/ubus.c
  44 @@ -0,0 +1,494 @@
  45 +/*
  46 + * hostapd / ubus support
  47 + * Copyright (c) 2013, Felix Fietkau <nbd@openwrt.org>
  48 + *
  49 + * This software may be distributed under the terms of the BSD license.
  50 + * See README for more details.
  51 + */
  52 +
  53 +#include "utils/includes.h"
  54 +#include "utils/common.h"
  55 +#include "utils/eloop.h"
  56 +#include "common/ieee802_11_defs.h"
  57 +#include "hostapd.h"
  58 +#include "wps_hostapd.h"
  59 +#include "sta_info.h"
  60 +#include "ubus.h"
  61 +
  62 +static struct ubus_context *ctx;
  63 +static struct blob_buf b;
  64 +static int ctx_ref;
  65 +
  66 +static inline struct hostapd_data *get_hapd_from_object(struct ubus_object *obj)
  67 +{
  68 +       return container_of(obj, struct hostapd_data, ubus.obj);
  69 +}
  70 +
  71 +
  72 +struct ubus_banned_client {
  73 +       struct avl_node avl;
  74 +       u8 addr[ETH_ALEN];
  75 +};
  76 +
  77 +static void ubus_receive(int sock, void *eloop_ctx, void *sock_ctx)
  78 +{
  79 +       struct ubus_context *ctx = eloop_ctx;
  80 +       ubus_handle_event(ctx);
  81 +}
  82 +
  83 +static void ubus_reconnect_timeout(void *eloop_data, void *user_ctx)
  84 +{
  85 +       if (ubus_reconnect(ctx, NULL)) {
  86 +               eloop_register_timeout(1, 0, ubus_reconnect_timeout, ctx, NULL);
  87 +               return;
  88 +       }
  89 +
  90 +       eloop_register_read_sock(ctx->sock.fd, ubus_receive, ctx, NULL);
  91 +}
  92 +
  93 +static void hostapd_ubus_connection_lost(struct ubus_context *ctx)
  94 +{
  95 +       eloop_unregister_read_sock(ctx->sock.fd);
  96 +       eloop_register_timeout(1, 0, ubus_reconnect_timeout, ctx, NULL);
  97 +}
  98 +
  99 +static bool hostapd_ubus_init(void)
 100 +{
 101 +       if (ctx)
 102 +               return true;
 103 +
 104 +       ctx = ubus_connect(NULL);
 105 +       if (!ctx)
 106 +               return false;
 107 +
 108 +       ctx->connection_lost = hostapd_ubus_connection_lost;
 109 +       eloop_register_read_sock(ctx->sock.fd, ubus_receive, ctx, NULL);
 110 +       return true;
 111 +}
 112 +
 113 +static void hostapd_ubus_ref_inc(void)
 114 +{
 115 +       ctx_ref++;
 116 +}
 117 +
 118 +static void hostapd_ubus_ref_dec(void)
 119 +{
 120 +       ctx_ref--;
 121 +       if (!ctx)
 122 +               return;
 123 +
 124 +       if (ctx_ref)
 125 +               return;
 126 +
 127 +       eloop_unregister_read_sock(ctx->sock.fd);
 128 +       ubus_free(ctx);
 129 +       ctx = NULL;
 130 +}
 131 +
 132 +void hostapd_ubus_add_iface(struct hostapd_iface *iface)
 133 +{
 134 +       if (!hostapd_ubus_init())
 135 +               return;
 136 +}
 137 +
 138 +void hostapd_ubus_free_iface(struct hostapd_iface *iface)
 139 +{
 140 +       if (!ctx)
 141 +               return;
 142 +}
 143 +
 144 +static void
 145 +hostapd_bss_del_ban(void *eloop_data, void *user_ctx)
 146 +{
 147 +       struct ubus_banned_client *ban = eloop_data;
 148 +       struct hostapd_data *hapd = user_ctx;
 149 +
 150 +       avl_delete(&hapd->ubus.banned, &ban->avl);
 151 +       free(ban);
 152 +}
 153 +
 154 +static void
 155 +hostapd_bss_ban_client(struct hostapd_data *hapd, u8 *addr, int time)
 156 +{
 157 +       struct ubus_banned_client *ban;
 158 +
 159 +       if (time < 0)
 160 +               time = 0;
 161 +
 162 +       ban = avl_find_element(&hapd->ubus.banned, addr, ban, avl);
 163 +       if (!ban) {
 164 +               if (!time)
 165 +                       return;
 166 +
 167 +               ban = os_zalloc(sizeof(*ban));
 168 +               memcpy(ban->addr, addr, sizeof(ban->addr));
 169 +               ban->avl.key = ban->addr;
 170 +               avl_insert(&hapd->ubus.banned, &ban->avl);
 171 +       } else {
 172 +               eloop_cancel_timeout(hostapd_bss_del_ban, ban, hapd);
 173 +               if (!time) {
 174 +                       hostapd_bss_del_ban(ban, hapd);
 175 +                       return;
 176 +               }
 177 +       }
 178 +
 179 +       eloop_register_timeout(0, time * 1000, hostapd_bss_del_ban, ban, hapd);
 180 +}
 181 +
 182 +static int
 183 +hostapd_bss_get_clients(struct ubus_context *ctx, struct ubus_object *obj,
 184 +                       struct ubus_request_data *req, const char *method,
 185 +                       struct blob_attr *msg)
 186 +{
 187 +       struct hostapd_data *hapd = container_of(obj, struct hostapd_data, ubus.obj);
 188 +       struct sta_info *sta;
 189 +       void *list, *c;
 190 +       char mac_buf[20];
 191 +       static const struct {
 192 +               const char *name;
 193 +               uint32_t flag;
 194 +       } sta_flags[] = {
 195 +               { "auth", WLAN_STA_AUTH },
 196 +               { "assoc", WLAN_STA_ASSOC },
 197 +               { "authorized", WLAN_STA_AUTHORIZED },
 198 +               { "preauth", WLAN_STA_PREAUTH },
 199 +               { "wds", WLAN_STA_WDS },
 200 +               { "wmm", WLAN_STA_WMM },
 201 +               { "ht", WLAN_STA_HT },
 202 +               { "vht", WLAN_STA_VHT },
 203 +               { "wps", WLAN_STA_WPS },
 204 +               { "mfp", WLAN_STA_MFP },
 205 +       };
 206 +
 207 +       blob_buf_init(&b, 0);
 208 +       blobmsg_add_u32(&b, "freq", hapd->iface->freq);
 209 +       list = blobmsg_open_table(&b, "clients");
 210 +       for (sta = hapd->sta_list; sta; sta = sta->next) {
 211 +               int i;
 212 +
 213 +               sprintf(mac_buf, MACSTR, MAC2STR(sta->addr));
 214 +               c = blobmsg_open_table(&b, mac_buf);
 215 +               for (i = 0; i < ARRAY_SIZE(sta_flags); i++)
 216 +                       blobmsg_add_u8(&b, sta_flags[i].name,
 217 +                                      !!(sta->flags & sta_flags[i].flag));
 218 +               blobmsg_add_u32(&b, "aid", sta->aid);
 219 +               blobmsg_close_table(&b, c);
 220 +       }
 221 +       blobmsg_close_array(&b, list);
 222 +       ubus_send_reply(ctx, req, b.head);
 223 +
 224 +       return 0;
 225 +}
 226 +
 227 +enum {
 228 +       DEL_CLIENT_ADDR,
 229 +       DEL_CLIENT_REASON,
 230 +       DEL_CLIENT_DEAUTH,
 231 +       DEL_CLIENT_BAN_TIME,
 232 +       __DEL_CLIENT_MAX
 233 +};
 234 +
 235 +static const struct blobmsg_policy del_policy[__DEL_CLIENT_MAX] = {
 236 +       [DEL_CLIENT_ADDR] = { "addr", BLOBMSG_TYPE_STRING },
 237 +       [DEL_CLIENT_REASON] = { "reason", BLOBMSG_TYPE_INT32 },
 238 +       [DEL_CLIENT_DEAUTH] = { "deauth", BLOBMSG_TYPE_INT8 },
 239 +       [DEL_CLIENT_BAN_TIME] = { "ban_time", BLOBMSG_TYPE_INT32 },
 240 +};
 241 +
 242 +static int
 243 +hostapd_bss_del_client(struct ubus_context *ctx, struct ubus_object *obj,
 244 +                       struct ubus_request_data *req, const char *method,
 245 +                       struct blob_attr *msg)
 246 +{
 247 +       struct blob_attr *tb[__DEL_CLIENT_MAX];
 248 +       struct hostapd_data *hapd = container_of(obj, struct hostapd_data, ubus.obj);
 249 +       struct sta_info *sta;
 250 +       bool deauth = false;
 251 +       int reason;
 252 +       u8 addr[ETH_ALEN];
 253 +
 254 +       blobmsg_parse(del_policy, __DEL_CLIENT_MAX, tb, blob_data(msg), blob_len(msg));
 255 +
 256 +       if (!tb[DEL_CLIENT_ADDR])
 257 +               return UBUS_STATUS_INVALID_ARGUMENT;
 258 +
 259 +       if (hwaddr_aton(blobmsg_data(tb[DEL_CLIENT_ADDR]), addr))
 260 +               return UBUS_STATUS_INVALID_ARGUMENT;
 261 +
 262 +       if (tb[DEL_CLIENT_REASON])
 263 +               reason = blobmsg_get_u32(tb[DEL_CLIENT_REASON]);
 264 +
 265 +       if (tb[DEL_CLIENT_DEAUTH])
 266 +               deauth = blobmsg_get_bool(tb[DEL_CLIENT_DEAUTH]);
 267 +
 268 +       sta = ap_get_sta(hapd, addr);
 269 +       if (sta) {
 270 +               if (deauth) {
 271 +                       hostapd_drv_sta_deauth(hapd, addr, reason);
 272 +                       ap_sta_deauthenticate(hapd, sta, reason);
 273 +               } else {
 274 +                       hostapd_drv_sta_disassoc(hapd, addr, reason);
 275 +                       ap_sta_disassociate(hapd, sta, reason);
 276 +               }
 277 +       }
 278 +
 279 +       if (tb[DEL_CLIENT_BAN_TIME])
 280 +               hostapd_bss_ban_client(hapd, addr, blobmsg_get_u32(tb[DEL_CLIENT_BAN_TIME]));
 281 +
 282 +       return 0;
 283 +}
 284 +
 285 +static void
 286 +blobmsg_add_macaddr(struct blob_buf *buf, const char *name, const u8 *addr)
 287 +{
 288 +       char *s;
 289 +
 290 +       s = blobmsg_alloc_string_buffer(buf, name, 20);
 291 +       sprintf(s, MACSTR, MAC2STR(addr));
 292 +       blobmsg_add_string_buffer(buf);
 293 +}
 294 +
 295 +static int
 296 +hostapd_bss_list_bans(struct ubus_context *ctx, struct ubus_object *obj,
 297 +                     struct ubus_request_data *req, const char *method,
 298 +                     struct blob_attr *msg)
 299 +{
 300 +       struct hostapd_data *hapd = container_of(obj, struct hostapd_data, ubus.obj);
 301 +       struct ubus_banned_client *ban;
 302 +       void *c;
 303 +
 304 +       blob_buf_init(&b, 0);
 305 +       c = blobmsg_open_array(&b, "clients");
 306 +       avl_for_each_element(&hapd->ubus.banned, ban, avl)
 307 +               blobmsg_add_macaddr(&b, NULL, ban->addr);
 308 +       blobmsg_close_array(&b, c);
 309 +       ubus_send_reply(ctx, req, b.head);
 310 +
 311 +       return 0;
 312 +}
 313 +
 314 +static int
 315 +hostapd_bss_wps_start(struct ubus_context *ctx, struct ubus_object *obj,
 316 +                       struct ubus_request_data *req, const char *method,
 317 +                       struct blob_attr *msg)
 318 +{
 319 +       int rc;
 320 +       struct hostapd_data *hapd = container_of(obj, struct hostapd_data, ubus.obj);
 321 +
 322 +       rc = hostapd_wps_button_pushed(hapd, NULL);
 323 +
 324 +       if (rc != 0)
 325 +               return UBUS_STATUS_NOT_SUPPORTED;
 326 +
 327 +       return 0;
 328 +}
 329 +
 330 +static int
 331 +hostapd_bss_wps_cancel(struct ubus_context *ctx, struct ubus_object *obj,
 332 +                       struct ubus_request_data *req, const char *method,
 333 +                       struct blob_attr *msg)
 334 +{
 335 +       int rc;
 336 +       struct hostapd_data *hapd = container_of(obj, struct hostapd_data, ubus.obj);
 337 +
 338 +       rc = hostapd_wps_cancel(hapd);
 339 +
 340 +       if (rc != 0)
 341 +               return UBUS_STATUS_NOT_SUPPORTED;
 342 +
 343 +       return 0;
 344 +}
 345 +
 346 +enum {
 347 +       CSA_FREQ,
 348 +       CSA_BCN_COUNT,
 349 +       __CSA_MAX
 350 +};
 351 +
 352 +static const struct blobmsg_policy csa_policy[__CSA_MAX] = {
 353 +       /*
 354 +        * for now, frequency and beacon count are enough, add more
 355 +        * parameters on demand
 356 +        */
 357 +       [CSA_FREQ] = { "freq", BLOBMSG_TYPE_INT32 },
 358 +       [CSA_BCN_COUNT] = { "bcn_count", BLOBMSG_TYPE_INT32 },
 359 +};
 360 +
 361 +#ifdef NEED_AP_MLME
 362 +static int
 363 +hostapd_switch_chan(struct ubus_context *ctx, struct ubus_object *obj,
 364 +                   struct ubus_request_data *req, const char *method,
 365 +                   struct blob_attr *msg)
 366 +{
 367 +       struct blob_attr *tb[__CSA_MAX];
 368 +       struct hostapd_data *hapd = get_hapd_from_object(obj);
 369 +       struct csa_settings css;
 370 +
 371 +       blobmsg_parse(csa_policy, __CSA_MAX, tb, blob_data(msg), blob_len(msg));
 372 +
 373 +       if (!tb[CSA_FREQ])
 374 +               return UBUS_STATUS_INVALID_ARGUMENT;
 375 +
 376 +       memset(&css, 0, sizeof(css));
 377 +       css.freq_params.freq = blobmsg_get_u32(tb[CSA_FREQ]);
 378 +       if (tb[CSA_BCN_COUNT])
 379 +               css.cs_count = blobmsg_get_u32(tb[CSA_BCN_COUNT]);
 380 +
 381 +       if (hostapd_switch_channel(hapd, &css) != 0)
 382 +               return UBUS_STATUS_NOT_SUPPORTED;
 383 +       return UBUS_STATUS_OK;
 384 +}
 385 +#endif
 386 +
 387 +enum {
 388 +       VENDOR_ELEMENTS,
 389 +       __VENDOR_ELEMENTS_MAX
 390 +};
 391 +
 392 +static const struct blobmsg_policy ve_policy[__VENDOR_ELEMENTS_MAX] = {
 393 +       /* vendor elements are provided as hex-string */
 394 +       [VENDOR_ELEMENTS] = { "vendor_elements", BLOBMSG_TYPE_STRING },
 395 +};
 396 +
 397 +static int
 398 +hostapd_vendor_elements(struct ubus_context *ctx, struct ubus_object *obj,
 399 +                       struct ubus_request_data *req, const char *method,
 400 +                       struct blob_attr *msg)
 401 +{
 402 +       struct blob_attr *tb[__VENDOR_ELEMENTS_MAX];
 403 +       struct hostapd_data *hapd = get_hapd_from_object(obj);
 404 +
 405 +       blobmsg_parse(ve_policy, __VENDOR_ELEMENTS_MAX, tb,
 406 +                     blob_data(msg), blob_len(msg));
 407 +
 408 +       if (!tb[VENDOR_ELEMENTS])
 409 +               return UBUS_STATUS_INVALID_ARGUMENT;
 410 +
 411 +       const char *vendor_elements = blobmsg_data(tb[VENDOR_ELEMENTS]);
 412 +       if (hostapd_set_iface(hapd->iconf, hapd->conf, "vendor_elements",
 413 +                             vendor_elements) != 0)
 414 +               return UBUS_STATUS_NOT_SUPPORTED;
 415 +
 416 +       /* update beacons if vendor elements were set successfully */
 417 +       if (ieee802_11_update_beacons(hapd->iface) != 0)
 418 +               return UBUS_STATUS_NOT_SUPPORTED;
 419 +       return UBUS_STATUS_OK;
 420 +}
 421 +
 422 +static const struct ubus_method bss_methods[] = {
 423 +       UBUS_METHOD_NOARG("get_clients", hostapd_bss_get_clients),
 424 +       UBUS_METHOD("del_client", hostapd_bss_del_client, del_policy),
 425 +       UBUS_METHOD_NOARG("list_bans", hostapd_bss_list_bans),
 426 +       UBUS_METHOD_NOARG("wps_start", hostapd_bss_wps_start),
 427 +       UBUS_METHOD_NOARG("wps_cancel", hostapd_bss_wps_cancel),
 428 +#ifdef NEED_AP_MLME
 429 +       UBUS_METHOD("switch_chan", hostapd_switch_chan, csa_policy),
 430 +#endif
 431 +       UBUS_METHOD("set_vendor_elements", hostapd_vendor_elements, ve_policy),
 432 +};
 433 +
 434 +static struct ubus_object_type bss_object_type =
 435 +       UBUS_OBJECT_TYPE("hostapd_bss", bss_methods);
 436 +
 437 +static int avl_compare_macaddr(const void *k1, const void *k2, void *ptr)
 438 +{
 439 +       return memcmp(k1, k2, ETH_ALEN);
 440 +}
 441 +
 442 +void hostapd_ubus_add_bss(struct hostapd_data *hapd)
 443 +{
 444 +       struct ubus_object *obj = &hapd->ubus.obj;
 445 +       char *name;
 446 +       int ret;
 447 +
 448 +       if (!hostapd_ubus_init())
 449 +               return;
 450 +
 451 +       if (asprintf(&name, "hostapd.%s", hapd->conf->iface) < 0)
 452 +               return;
 453 +
 454 +       avl_init(&hapd->ubus.banned, avl_compare_macaddr, false, NULL);
 455 +       obj->name = name;
 456 +       obj->type = &bss_object_type;
 457 +       obj->methods = bss_object_type.methods;
 458 +       obj->n_methods = bss_object_type.n_methods;
 459 +       ret = ubus_add_object(ctx, obj);
 460 +       hostapd_ubus_ref_inc();
 461 +}
 462 +
 463 +void hostapd_ubus_free_bss(struct hostapd_data *hapd)
 464 +{
 465 +       struct ubus_object *obj = &hapd->ubus.obj;
 466 +       char *name = (char *) obj->name;
 467 +
 468 +       if (!ctx)
 469 +               return;
 470 +
 471 +       if (obj->id) {
 472 +               ubus_remove_object(ctx, obj);
 473 +               hostapd_ubus_ref_dec();
 474 +       }
 475 +
 476 +       free(name);
 477 +}
 478 +
 479 +struct ubus_event_req {
 480 +       struct ubus_notify_request nreq;
 481 +       bool deny;
 482 +};
 483 +
 484 +static void
 485 +ubus_event_cb(struct ubus_notify_request *req, int idx, int ret)
 486 +{
 487 +       struct ubus_event_req *ureq = container_of(req, struct ubus_event_req, nreq);
 488 +
 489 +       if (ret)
 490 +               ureq->deny = true;
 491 +}
 492 +
 493 +int hostapd_ubus_handle_event(struct hostapd_data *hapd, struct hostapd_ubus_request *req)
 494 +{
 495 +       struct ubus_banned_client *ban;
 496 +       const char *types[HOSTAPD_UBUS_TYPE_MAX] = {
 497 +               [HOSTAPD_UBUS_PROBE_REQ] = "probe",
 498 +               [HOSTAPD_UBUS_AUTH_REQ] = "auth",
 499 +               [HOSTAPD_UBUS_ASSOC_REQ] = "assoc",
 500 +       };
 501 +       const char *type = "mgmt";
 502 +       struct ubus_event_req ureq = {};
 503 +       const u8 *addr;
 504 +
 505 +       if (req->mgmt_frame)
 506 +               addr = req->mgmt_frame->sa;
 507 +       else
 508 +               addr = req->addr;
 509 +
 510 +       ban = avl_find_element(&hapd->ubus.banned, addr, ban, avl);
 511 +       if (ban)
 512 +               return -2;
 513 +
 514 +       if (!hapd->ubus.obj.has_subscribers)
 515 +               return 0;
 516 +
 517 +       if (req->type < ARRAY_SIZE(types))
 518 +               type = types[req->type];
 519 +
 520 +       blob_buf_init(&b, 0);
 521 +       blobmsg_add_macaddr(&b, "address", addr);
 522 +       if (req->mgmt_frame)
 523 +               blobmsg_add_macaddr(&b, "target", req->mgmt_frame->da);
 524 +       if (req->frame_info)
 525 +               blobmsg_add_u32(&b, "signal", req->frame_info->ssi_signal);
 526 +       blobmsg_add_u32(&b, "freq", hapd->iface->freq);
 527 +
 528 +       if (ubus_notify_async(ctx, &hapd->ubus.obj, type, b.head, &ureq.nreq))
 529 +               return 0;
 530 +
 531 +       ureq.nreq.status_cb = ubus_event_cb;
 532 +       ubus_complete_request(ctx, &ureq.nreq.req, 100);
 533 +
 534 +       if (ureq.deny)
 535 +               return -1;
 536 +
 537 +       return 0;
 538 +}
 539 --- /dev/null
 540 +++ b/src/ap/ubus.h
 541 @@ -0,0 +1,78 @@
 542 +/*
 543 + * hostapd / ubus support
 544 + * Copyright (c) 2013, Felix Fietkau <nbd@openwrt.org>
 545 + *
 546 + * This software may be distributed under the terms of the BSD license.
 547 + * See README for more details.
 548 + */
 549 +#ifndef __HOSTAPD_UBUS_H
 550 +#define __HOSTAPD_UBUS_H
 551 +
 552 +enum hostapd_ubus_event_type {
 553 +       HOSTAPD_UBUS_PROBE_REQ,
 554 +       HOSTAPD_UBUS_AUTH_REQ,
 555 +       HOSTAPD_UBUS_ASSOC_REQ,
 556 +       HOSTAPD_UBUS_TYPE_MAX
 557 +};
 558 +
 559 +struct hostapd_ubus_request {
 560 +       enum hostapd_ubus_event_type type;
 561 +       const struct ieee80211_mgmt *mgmt_frame;
 562 +       const struct hostapd_frame_info *frame_info;
 563 +       const u8 *addr;
 564 +};
 565 +
 566 +struct hostapd_iface;
 567 +struct hostapd_data;
 568 +
 569 +#ifdef UBUS_SUPPORT
 570 +
 571 +#include <libubox/avl.h>
 572 +#include <libubus.h>
 573 +
 574 +struct hostapd_ubus_iface {
 575 +       struct ubus_object obj;
 576 +};
 577 +
 578 +struct hostapd_ubus_bss {
 579 +       struct ubus_object obj;
 580 +       struct avl_tree banned;
 581 +};
 582 +
 583 +void hostapd_ubus_add_iface(struct hostapd_iface *iface);
 584 +void hostapd_ubus_free_iface(struct hostapd_iface *iface);
 585 +void hostapd_ubus_add_bss(struct hostapd_data *hapd);
 586 +void hostapd_ubus_free_bss(struct hostapd_data *hapd);
 587 +
 588 +int hostapd_ubus_handle_event(struct hostapd_data *hapd, struct hostapd_ubus_request *req);
 589 +
 590 +#else
 591 +
 592 +struct hostapd_ubus_iface {};
 593 +
 594 +struct hostapd_ubus_bss {};
 595 +
 596 +static inline void hostapd_ubus_add_iface(struct hostapd_iface *iface)
 597 +{
 598 +}
 599 +
 600 +static inline void hostapd_ubus_free_iface(struct hostapd_iface *iface)
 601 +{
 602 +}
 603 +
 604 +static inline void hostapd_ubus_add_bss(struct hostapd_data *hapd)
 605 +{
 606 +}
 607 +
 608 +static inline void hostapd_ubus_free_bss(struct hostapd_data *hapd)
 609 +{
 610 +}
 611 +
 612 +static inline int hostapd_ubus_handle_event(struct hostapd_data *hapd, struct hostapd_ubus_request *req)
 613 +{
 614 +       return 0;
 615 +}
 616 +
 617 +#endif
 618 +
 619 +#endif
 620 --- a/src/ap/hostapd.c
 621 +++ b/src/ap/hostapd.c
 622 @@ -277,6 +277,7 @@ static void hostapd_free_hapd_data(struc
 623         hapd->started = 0;
 624
 625         wpa_printf(MSG_DEBUG, "%s(%s)", __func__, hapd->conf->iface);
 626 +       hostapd_ubus_free_bss(hapd);
 627         iapp_deinit(hapd->iapp);
 628         hapd->iapp = NULL;
 629         accounting_deinit(hapd);
 630 @@ -1098,6 +1099,8 @@ static int hostapd_setup_bss(struct host
 631         if (hapd->driver && hapd->driver->set_operstate)
 632                 hapd->driver->set_operstate(hapd->drv_priv, 1);
 633
 634 +       hostapd_ubus_add_bss(hapd);
 635 +
 636         return 0;
 637  }
 638
 639 @@ -1384,6 +1387,7 @@ int hostapd_setup_interface_complete(str
 640         if (err)
 641                 goto fail;
 642
 643 +       hostapd_ubus_add_iface(iface);
 644         wpa_printf(MSG_DEBUG, "Completing interface initialization");
 645         if (iface->conf->channel) {
 646  #ifdef NEED_AP_MLME
 647 @@ -1544,6 +1548,7 @@ dfs_offload:
 648
 649  fail:
 650         wpa_printf(MSG_ERROR, "Interface initialization failed");
 651 +       hostapd_ubus_free_iface(iface);
 652         hostapd_set_state(iface, HAPD_IFACE_DISABLED);
 653         wpa_msg(hapd->msg_ctx, MSG_INFO, AP_EVENT_DISABLED);
 654         if (iface->interfaces && iface->interfaces->terminate_on_error)
 655 @@ -1873,6 +1878,7 @@ void hostapd_interface_deinit_free(struc
 656                    (unsigned int) iface->conf->num_bss);
 657         driver = iface->bss[0]->driver;
 658         drv_priv = iface->bss[0]->drv_priv;
 659 +       hostapd_ubus_free_iface(iface);
 660         hostapd_interface_deinit(iface);
 661         wpa_printf(MSG_DEBUG, "%s: driver=%p drv_priv=%p -> hapd_deinit",
 662                    __func__, driver, drv_priv);
 663 --- a/src/ap/ieee802_11.c
 664 +++ b/src/ap/ieee802_11.c
 665 @@ -881,7 +881,8 @@ int auth_sae_init_committed(struct hosta
 666
 667
 668  static void handle_auth(struct hostapd_data *hapd,
 669 -                       const struct ieee80211_mgmt *mgmt, size_t len)
 670 +                       const struct ieee80211_mgmt *mgmt, size_t len,
 671 +                       struct hostapd_frame_info *fi)
 672  {
 673         u16 auth_alg, auth_transaction, status_code;
 674         u16 resp = WLAN_STATUS_SUCCESS;
 675 @@ -897,6 +898,11 @@ static void handle_auth(struct hostapd_d
 676         char *identity = NULL;
 677         char *radius_cui = NULL;
 678         u16 seq_ctrl;
 679 +       struct hostapd_ubus_request req = {
 680 +               .type = HOSTAPD_UBUS_AUTH_REQ,
 681 +               .mgmt_frame = mgmt,
 682 +               .frame_info = fi,
 683 +       };
 684
 685         if (len < IEEE80211_HDRLEN + sizeof(mgmt->u.auth)) {
 686                 wpa_printf(MSG_INFO, "handle_auth - too short payload (len=%lu)",
 687 @@ -983,6 +989,14 @@ static void handle_auth(struct hostapd_d
 688                 resp = WLAN_STATUS_UNSPECIFIED_FAILURE;
 689                 goto fail;
 690         }
 691 +
 692 +       if (hostapd_ubus_handle_event(hapd, &req)) {
 693 +               wpa_printf(MSG_DEBUG, "Station " MACSTR " rejected by ubus handler.\n",
 694 +                      MAC2STR(mgmt->sa));
 695 +               resp = WLAN_STATUS_UNSPECIFIED_FAILURE;
 696 +               goto fail;
 697 +       }
 698 +
 699         if (res == HOSTAPD_ACL_PENDING) {
 700                 wpa_printf(MSG_DEBUG, "Authentication frame from " MACSTR
 701                            " waiting for an external authentication",
 702 @@ -1694,13 +1708,18 @@ static void send_assoc_resp(struct hosta
 703
 704  static void handle_assoc(struct hostapd_data *hapd,
 705                          const struct ieee80211_mgmt *mgmt, size_t len,
 706 -                        int reassoc)
 707 +                        int reassoc, struct hostapd_frame_info *fi)
 708  {
 709         u16 capab_info, listen_interval, seq_ctrl, fc;
 710         u16 resp = WLAN_STATUS_SUCCESS;
 711         const u8 *pos;
 712         int left, i;
 713         struct sta_info *sta;
 714 +       struct hostapd_ubus_request req = {
 715 +               .type = HOSTAPD_UBUS_ASSOC_REQ,
 716 +               .mgmt_frame = mgmt,
 717 +               .frame_info = fi,
 718 +       };
 719
 720         if (len < IEEE80211_HDRLEN + (reassoc ? sizeof(mgmt->u.reassoc_req) :
 721                                       sizeof(mgmt->u.assoc_req))) {
 722 @@ -1820,6 +1839,13 @@ static void handle_assoc(struct hostapd_
 723                 goto fail;
 724         }
 725
 726 +       if (hostapd_ubus_handle_event(hapd, &req)) {
 727 +               wpa_printf(MSG_DEBUG, "Station " MACSTR " assoc rejected by ubus handler.\n",
 728 +                      MAC2STR(mgmt->sa));
 729 +               resp = WLAN_STATUS_UNSPECIFIED_FAILURE;
 730 +               goto fail;
 731 +       }
 732 +
 733         sta->capability = capab_info;
 734         sta->listen_interval = listen_interval;
 735
 736 @@ -2236,7 +2262,7 @@ int ieee802_11_mgmt(struct hostapd_data
 737
 738
 739         if (stype == WLAN_FC_STYPE_PROBE_REQ) {
 740 -               handle_probe_req(hapd, mgmt, len, fi->ssi_signal);
 741 +               handle_probe_req(hapd, mgmt, len, fi);
 742                 return 1;
 743         }
 744
 745 @@ -2251,17 +2277,17 @@ int ieee802_11_mgmt(struct hostapd_data
 746         switch (stype) {
 747         case WLAN_FC_STYPE_AUTH:
 748                 wpa_printf(MSG_DEBUG, "mgmt::auth");
 749 -               handle_auth(hapd, mgmt, len);
 750 +               handle_auth(hapd, mgmt, len, fi);
 751                 ret = 1;
 752                 break;
 753         case WLAN_FC_STYPE_ASSOC_REQ:
 754                 wpa_printf(MSG_DEBUG, "mgmt::assoc_req");
 755 -               handle_assoc(hapd, mgmt, len, 0);
 756 +               handle_assoc(hapd, mgmt, len, 0, fi);
 757                 ret = 1;
 758                 break;
 759         case WLAN_FC_STYPE_REASSOC_REQ:
 760                 wpa_printf(MSG_DEBUG, "mgmt::reassoc_req");
 761 -               handle_assoc(hapd, mgmt, len, 1);
 762 +               handle_assoc(hapd, mgmt, len, 1, fi);
 763                 ret = 1;
 764                 break;
 765         case WLAN_FC_STYPE_DISASSOC:
 766 --- a/src/ap/beacon.c
 767 +++ b/src/ap/beacon.c
 768 @@ -542,7 +542,7 @@ static enum ssid_match_result ssid_match
 769
 770  void handle_probe_req(struct hostapd_data *hapd,
 771                       const struct ieee80211_mgmt *mgmt, size_t len,
 772 -                     int ssi_signal)
 773 +                     struct hostapd_frame_info *fi)
 774  {
 775         u8 *resp;
 776         struct ieee802_11_elems elems;
 777 @@ -550,8 +550,14 @@ void handle_probe_req(struct hostapd_dat
 778         size_t ie_len;
 779         struct sta_info *sta = NULL;
 780         size_t i, resp_len;
 781 +       int ssi_signal = fi->ssi_signal;
 782         int noack;
 783         enum ssid_match_result res;
 784 +       struct hostapd_ubus_request req = {
 785 +               .type = HOSTAPD_UBUS_PROBE_REQ,
 786 +               .mgmt_frame = mgmt,
 787 +               .frame_info = fi,
 788 +       };
 789
 790         ie = mgmt->u.probe_req.variable;
 791         if (len < IEEE80211_HDRLEN + sizeof(mgmt->u.probe_req))
 792 @@ -710,6 +716,12 @@ void handle_probe_req(struct hostapd_dat
 793         }
 794  #endif /* CONFIG_P2P */
 795
 796 +       if (hostapd_ubus_handle_event(hapd, &req)) {
 797 +               wpa_printf(MSG_DEBUG, "Probe request for " MACSTR " rejected by ubus handler.\n",
 798 +                      MAC2STR(mgmt->sa));
 799 +               return;
 800 +       }
 801 +
 802         /* TODO: verify that supp_rates contains at least one matching rate
 803          * with AP configuration */
 804
 805 --- a/src/ap/beacon.h
 806 +++ b/src/ap/beacon.h
 807 @@ -14,7 +14,7 @@ struct ieee80211_mgmt;
 808
 809  void handle_probe_req(struct hostapd_data *hapd,
 810                       const struct ieee80211_mgmt *mgmt, size_t len,
 811 -                     int ssi_signal);
 812 +                     struct hostapd_frame_info *fi);
 813  int ieee802_11_set_beacon(struct hostapd_data *hapd);
 814  int ieee802_11_set_beacons(struct hostapd_iface *iface);
 815  int ieee802_11_update_beacons(struct hostapd_iface *iface);