text: /*
   * hostapd / ubus support
   * Copyright (c) 2013, Felix Fietkau <nbd@openwrt.org>
   *
   * This software may be distributed under the terms of the BSD license.
   * See README for more details.
   */
  
  #include "utils/includes.h"
  #include "utils/common.h"
  #include "utils/eloop.h"
  #include "common/ieee802_11_defs.h"
  #include "hostapd.h"
  #include "sta_info.h"
  #include "ubus.h"
  
  static struct ubus_context *ctx;
  static struct blob_buf b;
  static int ctx_ref;
  
  struct ubus_banned_client {
  	struct avl_node avl;
  	u8 addr[ETH_ALEN];
  };
  
  static void ubus_receive(int sock, void *eloop_ctx, void *sock_ctx)
  {
  	struct ubus_context *ctx = eloop_ctx;
  	ubus_handle_event(ctx);
  }
  
  static void ubus_reconnect_timeout(void *eloop_data, void *user_ctx)
  {
  	if (ubus_reconnect(ctx, NULL)) {
  		eloop_register_timeout(1, 0, ubus_reconnect_timeout, ctx, NULL);
  		return;
  	}
  
  	eloop_register_read_sock(ctx->sock.fd, ubus_receive, ctx, NULL);
  }
  
  static void hostapd_ubus_connection_lost(struct ubus_context *ctx)
  {
  	eloop_unregister_read_sock(ctx->sock.fd);
  	eloop_register_timeout(1, 0, ubus_reconnect_timeout, ctx, NULL);
  }
  
  static bool hostapd_ubus_init(void)
  {
  	if (ctx)
  		return true;
  
  	ctx = ubus_connect(NULL);
  	if (!ctx)
  		return false;
  
  	ctx->connection_lost = hostapd_ubus_connection_lost;
  	eloop_register_read_sock(ctx->sock.fd, ubus_receive, ctx, NULL);
  	return true;
  }
  
  static void hostapd_ubus_ref_inc(void)
  {
  	ctx_ref++;
  }
  
  static void hostapd_ubus_ref_dec(void)
  {
  	ctx_ref--;
  	if (!ctx)
  		return;
  
  	if (ctx_ref)
  		return;
  
  	eloop_unregister_read_sock(ctx->sock.fd);
  	ubus_free(ctx);
  	ctx = NULL;
  }
  
  void hostapd_ubus_add_iface(struct hostapd_iface *iface)
  {
  	if (!hostapd_ubus_init())
  		return;
  }
  
  void hostapd_ubus_free_iface(struct hostapd_iface *iface)
  {
  	if (!ctx)
  		return;
  }
  
  static void
  hostapd_bss_del_ban(void *eloop_data, void *user_ctx)
  {
  	struct ubus_banned_client *ban = eloop_data;
  	struct hostapd_data *hapd = user_ctx;
  
  	avl_delete(&hapd->ubus.banned, &ban->avl);
  	free(ban);
  }
  
  static void
  hostapd_bss_ban_client(struct hostapd_data *hapd, u8 *addr, int time)
  {
  	struct ubus_banned_client *ban;
  
  	if (time < 0)
  		time = 0;
  
  	ban = avl_find_element(&hapd->ubus.banned, addr, ban, avl);
  	if (!ban) {
  		if (!time)
  			return;
  
  		ban = os_zalloc(sizeof(*ban));
  		memcpy(ban->addr, addr, sizeof(ban->addr));
  		ban->avl.key = ban->addr;
  		avl_insert(&hapd->ubus.banned, &ban->avl);
  	} else {
  		eloop_cancel_timeout(hostapd_bss_del_ban, ban, hapd);
  		if (!time) {
  			hostapd_bss_del_ban(ban, hapd);
  			return;
  		}
  	}
  
  	eloop_register_timeout(0, time * 1000, hostapd_bss_del_ban, ban, hapd);
  }
  
  static int
  hostapd_bss_get_clients(struct ubus_context *ctx, struct ubus_object *obj,
  			struct ubus_request_data *req, const char *method,
  			struct blob_attr *msg)
  {
  	struct hostapd_data *hapd = container_of(obj, struct hostapd_data, ubus.obj);
  	struct sta_info *sta;
  	void *list, *c;
  	char mac_buf[20];
  	static const struct {
  		const char *name;
  		uint32_t flag;
  	} sta_flags[] = {
  		{ "auth", WLAN_STA_AUTH },
  		{ "assoc", WLAN_STA_ASSOC },
  		{ "authorized", WLAN_STA_AUTHORIZED },
  		{ "preauth", WLAN_STA_PREAUTH },
  		{ "wds", WLAN_STA_WDS },
  		{ "wmm", WLAN_STA_WMM },
  		{ "ht", WLAN_STA_HT },
  		{ "vht", WLAN_STA_VHT },
  		{ "wps", WLAN_STA_WPS },
  		{ "mfp", WLAN_STA_MFP },
  	};
  
  	blob_buf_init(&b, 0);
  	blobmsg_add_u32(&b, "freq", hapd->iface->freq);
  	list = blobmsg_open_table(&b, "clients");
  	for (sta = hapd->sta_list; sta; sta = sta->next) {
  		int i;
  
  		sprintf(mac_buf, MACSTR, MAC2STR(sta->addr));
  		c = blobmsg_open_table(&b, mac_buf);
  		for (i = 0; i < ARRAY_SIZE(sta_flags); i++)
  			blobmsg_add_u8(&b, sta_flags[i].name,
  				       !!(sta->flags & sta_flags[i].flag));
  		blobmsg_add_u32(&b, "aid", sta->aid);
  		blobmsg_close_table(&b, c);
  	}
  	blobmsg_close_array(&b, list);
  	ubus_send_reply(ctx, req, b.head);
  
  	return 0;
  }
  
  enum {
  	DEL_CLIENT_ADDR,
  	DEL_CLIENT_REASON,
  	DEL_CLIENT_DEAUTH,
  	DEL_CLIENT_BAN_TIME,
  	__DEL_CLIENT_MAX
  };
  
  static const struct blobmsg_policy del_policy[__DEL_CLIENT_MAX] = {
  	[DEL_CLIENT_ADDR] = { "addr", BLOBMSG_TYPE_STRING },
  	[DEL_CLIENT_REASON] = { "reason", BLOBMSG_TYPE_INT32 },
  	[DEL_CLIENT_DEAUTH] = { "deauth", BLOBMSG_TYPE_INT8 },
  	[DEL_CLIENT_BAN_TIME] = { "ban_time", BLOBMSG_TYPE_INT32 },
  };
  
  static int
  hostapd_bss_del_client(struct ubus_context *ctx, struct ubus_object *obj,
  			struct ubus_request_data *req, const char *method,
  			struct blob_attr *msg)
  {
  	struct blob_attr *tb[__DEL_CLIENT_MAX];
  	struct hostapd_data *hapd = container_of(obj, struct hostapd_data, ubus.obj);
  	struct sta_info *sta;
  	bool deauth = false;
  	int reason;
  	u8 addr[ETH_ALEN];
  
  	blobmsg_parse(del_policy, __DEL_CLIENT_MAX, tb, blob_data(msg), blob_len(msg));
  
  	if (!tb[DEL_CLIENT_ADDR])
  		return UBUS_STATUS_INVALID_ARGUMENT;
  
  	if (hwaddr_aton(blobmsg_data(tb[DEL_CLIENT_ADDR]), addr))
  		return UBUS_STATUS_INVALID_ARGUMENT;
  
  	if (tb[DEL_CLIENT_REASON])
  		reason = blobmsg_get_u32(tb[DEL_CLIENT_REASON]);
  
  	if (tb[DEL_CLIENT_DEAUTH])
  		deauth = blobmsg_get_bool(tb[DEL_CLIENT_DEAUTH]);
  
  	sta = ap_get_sta(hapd, addr);
  	if (sta) {
  		if (deauth) {
  			hostapd_drv_sta_deauth(hapd, addr, reason);
  			ap_sta_deauthenticate(hapd, sta, reason);
  		} else {
  			hostapd_drv_sta_disassoc(hapd, addr, reason);
  			ap_sta_disassociate(hapd, sta, reason);
  		}
  	}
  
  	if (tb[DEL_CLIENT_BAN_TIME])
  		hostapd_bss_ban_client(hapd, addr, blobmsg_get_u32(tb[DEL_CLIENT_BAN_TIME]));
  
  	return 0;
  }
  
  static void
  blobmsg_add_macaddr(struct blob_buf *buf, const char *name, const u8 *addr)
  {
  	char *s;
  
  	s = blobmsg_alloc_string_buffer(buf, name, 20);
  	sprintf(s, MACSTR, MAC2STR(addr));
  	blobmsg_add_string_buffer(buf);
  }
  
  static int
  hostapd_bss_list_bans(struct ubus_context *ctx, struct ubus_object *obj,
  		      struct ubus_request_data *req, const char *method,
  		      struct blob_attr *msg)
  {
  	struct hostapd_data *hapd = container_of(obj, struct hostapd_data, ubus.obj);
  	struct ubus_banned_client *ban;
  	void *c;
  
  	blob_buf_init(&b, 0);
  	c = blobmsg_open_array(&b, "clients");
  	avl_for_each_element(&hapd->ubus.banned, ban, avl)
  		blobmsg_add_macaddr(&b, NULL, ban->addr);
  	blobmsg_close_array(&b, c);
  	ubus_send_reply(ctx, req, b.head);
  
  	return 0;
  }
  
  static const struct ubus_method bss_methods[] = {
  	UBUS_METHOD_NOARG("get_clients", hostapd_bss_get_clients),
  	UBUS_METHOD("del_client", hostapd_bss_del_client, del_policy),
  	UBUS_METHOD_NOARG("list_bans", hostapd_bss_list_bans),
  };
  
  static struct ubus_object_type bss_object_type =
  	UBUS_OBJECT_TYPE("hostapd_bss", bss_methods);
  
  static int avl_compare_macaddr(const void *k1, const void *k2, void *ptr)
  {
  	return memcmp(k1, k2, ETH_ALEN);
  }
  
  void hostapd_ubus_add_bss(struct hostapd_data *hapd)
  {
  	struct ubus_object *obj = &hapd->ubus.obj;
  	char *name;
  	int ret;
  
  	if (!hostapd_ubus_init())
  		return;
  
  	if (asprintf(&name, "hostapd.%s", hapd->conf->iface) < 0)
  		return;
  
  	avl_init(&hapd->ubus.banned, avl_compare_macaddr, false, NULL);
  	obj->name = name;
  	obj->type = &bss_object_type;
  	obj->methods = bss_object_type.methods;
  	obj->n_methods = bss_object_type.n_methods;
  	ret = ubus_add_object(ctx, obj);
  	hostapd_ubus_ref_inc();
  }
  
  void hostapd_ubus_free_bss(struct hostapd_data *hapd)
  {
  	struct ubus_object *obj = &hapd->ubus.obj;
  	char *name = (char *) obj->name;
  
  	if (!ctx)
  		return;
  
  	if (obj->id) {
  		ubus_remove_object(ctx, obj);
  		hostapd_ubus_ref_dec();
  	}
  
  	free(name);
  }
  
  struct ubus_event_req {
  	struct ubus_notify_request nreq;
  	bool deny;
  };
  
  static void
  ubus_event_cb(struct ubus_notify_request *req, int idx, int ret)
  {
  	struct ubus_event_req *ureq = container_of(req, struct ubus_event_req, nreq);
  
  	if (ret)
  		ureq->deny = true;
  }
  
  int hostapd_ubus_handle_event(struct hostapd_data *hapd, struct hostapd_ubus_request *req)
  {
  	struct ubus_banned_client *ban;
  	const char *types[HOSTAPD_UBUS_TYPE_MAX] = {
  		[HOSTAPD_UBUS_PROBE_REQ] = "probe",
  		[HOSTAPD_UBUS_AUTH_REQ] = "auth",
  		[HOSTAPD_UBUS_ASSOC_REQ] = "assoc",
  	};
  	const char *type = "mgmt";
  	struct ubus_event_req ureq = {};
  	const u8 *addr;
  
  	if (req->mgmt_frame)
  		addr = req->mgmt_frame->sa;
  	else
  		addr = req->addr;
  
  	ban = avl_find_element(&hapd->ubus.banned, addr, ban, avl);
  	if (ban)
  		return -2;
  
  	if (!hapd->ubus.obj.has_subscribers)
  		return 0;
  
  	if (req->type < ARRAY_SIZE(types))
  		type = types[req->type];
  
  	blob_buf_init(&b, 0);
  	blobmsg_add_macaddr(&b, "address", addr);
  	if (req->mgmt_frame)
  		blobmsg_add_macaddr(&b, "target", req->mgmt_frame->da);
  	if (req->frame_info)
  		blobmsg_add_u32(&b, "signal", req->frame_info->ssi_signal);
  	blobmsg_add_u32(&b, "freq", hapd->iface->freq);
  
  	if (ubus_notify_async(ctx, &hapd->ubus.obj, type, b.head, &ureq.nreq))
  		return 0;
  
  	ureq.nreq.status_cb = ubus_event_cb;
  	ubus_complete_request(ctx, &ureq.nreq.req, 100);
  
  	if (ureq.deny)
  		return -1;
  
  	return 0;
  }
  