1/*
2 * Driver interaction with Linux nl80211/cfg80211 - Capabilities
3 * Copyright (c) 2002-2015, Jouni Malinen <j@w1.fi>
4 * Copyright (c) 2007, Johannes Berg <johannes@sipsolutions.net>
5 * Copyright (c) 2009-2010, Atheros Communications
6 *
7 * This software may be distributed under the terms of the BSD license.
8 * See README for more details.
9 */
10
11#include "includes.h"
12#include <netlink/genl/genl.h>
13
14#include "utils/common.h"
15#include "common/ieee802_11_defs.h"
16#include "common/ieee802_11_common.h"
17#include "common/qca-vendor.h"
18#include "common/qca-vendor-attr.h"
19#include "driver_nl80211.h"
20
21
22static int protocol_feature_handler(struct nl_msg *msg, void *arg)
23{
24	u32 *feat = arg;
25	struct nlattr *tb_msg[NL80211_ATTR_MAX + 1];
26	struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
27
28	nla_parse(tb_msg, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
29		  genlmsg_attrlen(gnlh, 0), NULL);
30
31	if (tb_msg[NL80211_ATTR_PROTOCOL_FEATURES])
32		*feat = nla_get_u32(tb_msg[NL80211_ATTR_PROTOCOL_FEATURES]);
33
34	return NL_SKIP;
35}
36
37
38static u32 get_nl80211_protocol_features(struct wpa_driver_nl80211_data *drv)
39{
40	u32 feat = 0;
41	struct nl_msg *msg;
42
43	msg = nlmsg_alloc();
44	if (!msg)
45		return 0;
46
47	if (!nl80211_cmd(drv, msg, 0, NL80211_CMD_GET_PROTOCOL_FEATURES)) {
48		nlmsg_free(msg);
49		return 0;
50	}
51
52	if (send_and_recv_msgs(drv, msg, protocol_feature_handler, &feat) == 0)
53		return feat;
54
55	return 0;
56}
57
58
59struct wiphy_info_data {
60	struct wpa_driver_nl80211_data *drv;
61	struct wpa_driver_capa *capa;
62
63	unsigned int num_multichan_concurrent;
64
65	unsigned int error:1;
66	unsigned int device_ap_sme:1;
67	unsigned int poll_command_supported:1;
68	unsigned int data_tx_status:1;
69	unsigned int monitor_supported:1;
70	unsigned int auth_supported:1;
71	unsigned int connect_supported:1;
72	unsigned int p2p_go_supported:1;
73	unsigned int p2p_client_supported:1;
74	unsigned int p2p_go_ctwindow_supported:1;
75	unsigned int p2p_concurrent:1;
76	unsigned int channel_switch_supported:1;
77	unsigned int set_qos_map_supported:1;
78	unsigned int have_low_prio_scan:1;
79	unsigned int wmm_ac_supported:1;
80	unsigned int mac_addr_rand_scan_supported:1;
81	unsigned int mac_addr_rand_sched_scan_supported:1;
82};
83
84
85static unsigned int probe_resp_offload_support(int supp_protocols)
86{
87	unsigned int prot = 0;
88
89	if (supp_protocols & NL80211_PROBE_RESP_OFFLOAD_SUPPORT_WPS)
90		prot |= WPA_DRIVER_PROBE_RESP_OFFLOAD_WPS;
91	if (supp_protocols & NL80211_PROBE_RESP_OFFLOAD_SUPPORT_WPS2)
92		prot |= WPA_DRIVER_PROBE_RESP_OFFLOAD_WPS2;
93	if (supp_protocols & NL80211_PROBE_RESP_OFFLOAD_SUPPORT_P2P)
94		prot |= WPA_DRIVER_PROBE_RESP_OFFLOAD_P2P;
95	if (supp_protocols & NL80211_PROBE_RESP_OFFLOAD_SUPPORT_80211U)
96		prot |= WPA_DRIVER_PROBE_RESP_OFFLOAD_INTERWORKING;
97
98	return prot;
99}
100
101
102static void wiphy_info_supported_iftypes(struct wiphy_info_data *info,
103					 struct nlattr *tb)
104{
105	struct nlattr *nl_mode;
106	int i;
107
108	if (tb == NULL)
109		return;
110
111	nla_for_each_nested(nl_mode, tb, i) {
112		switch (nla_type(nl_mode)) {
113		case NL80211_IFTYPE_AP:
114			info->capa->flags |= WPA_DRIVER_FLAGS_AP;
115			break;
116		case NL80211_IFTYPE_MESH_POINT:
117			info->capa->flags |= WPA_DRIVER_FLAGS_MESH;
118			break;
119		case NL80211_IFTYPE_ADHOC:
120			info->capa->flags |= WPA_DRIVER_FLAGS_IBSS;
121			break;
122		case NL80211_IFTYPE_P2P_DEVICE:
123			info->capa->flags |=
124				WPA_DRIVER_FLAGS_DEDICATED_P2P_DEVICE;
125			break;
126		case NL80211_IFTYPE_P2P_GO:
127			info->p2p_go_supported = 1;
128			break;
129		case NL80211_IFTYPE_P2P_CLIENT:
130			info->p2p_client_supported = 1;
131			break;
132		case NL80211_IFTYPE_MONITOR:
133			info->monitor_supported = 1;
134			break;
135		}
136	}
137}
138
139
140static int wiphy_info_iface_comb_process(struct wiphy_info_data *info,
141					 struct nlattr *nl_combi)
142{
143	struct nlattr *tb_comb[NUM_NL80211_IFACE_COMB];
144	struct nlattr *tb_limit[NUM_NL80211_IFACE_LIMIT];
145	struct nlattr *nl_limit, *nl_mode;
146	int err, rem_limit, rem_mode;
147	int combination_has_p2p = 0, combination_has_mgd = 0;
148	static struct nla_policy
149	iface_combination_policy[NUM_NL80211_IFACE_COMB] = {
150		[NL80211_IFACE_COMB_LIMITS] = { .type = NLA_NESTED },
151		[NL80211_IFACE_COMB_MAXNUM] = { .type = NLA_U32 },
152		[NL80211_IFACE_COMB_STA_AP_BI_MATCH] = { .type = NLA_FLAG },
153		[NL80211_IFACE_COMB_NUM_CHANNELS] = { .type = NLA_U32 },
154		[NL80211_IFACE_COMB_RADAR_DETECT_WIDTHS] = { .type = NLA_U32 },
155	},
156	iface_limit_policy[NUM_NL80211_IFACE_LIMIT] = {
157		[NL80211_IFACE_LIMIT_TYPES] = { .type = NLA_NESTED },
158		[NL80211_IFACE_LIMIT_MAX] = { .type = NLA_U32 },
159	};
160
161	err = nla_parse_nested(tb_comb, MAX_NL80211_IFACE_COMB,
162			       nl_combi, iface_combination_policy);
163	if (err || !tb_comb[NL80211_IFACE_COMB_LIMITS] ||
164	    !tb_comb[NL80211_IFACE_COMB_MAXNUM] ||
165	    !tb_comb[NL80211_IFACE_COMB_NUM_CHANNELS])
166		return 0; /* broken combination */
167
168	if (tb_comb[NL80211_IFACE_COMB_RADAR_DETECT_WIDTHS])
169		info->capa->flags |= WPA_DRIVER_FLAGS_RADAR;
170
171	nla_for_each_nested(nl_limit, tb_comb[NL80211_IFACE_COMB_LIMITS],
172			    rem_limit) {
173		err = nla_parse_nested(tb_limit, MAX_NL80211_IFACE_LIMIT,
174				       nl_limit, iface_limit_policy);
175		if (err || !tb_limit[NL80211_IFACE_LIMIT_TYPES])
176			return 0; /* broken combination */
177
178		nla_for_each_nested(nl_mode,
179				    tb_limit[NL80211_IFACE_LIMIT_TYPES],
180				    rem_mode) {
181			int ift = nla_type(nl_mode);
182			if (ift == NL80211_IFTYPE_P2P_GO ||
183			    ift == NL80211_IFTYPE_P2P_CLIENT)
184				combination_has_p2p = 1;
185			if (ift == NL80211_IFTYPE_STATION)
186				combination_has_mgd = 1;
187		}
188		if (combination_has_p2p && combination_has_mgd)
189			break;
190	}
191
192	if (combination_has_p2p && combination_has_mgd) {
193		unsigned int num_channels =
194			nla_get_u32(tb_comb[NL80211_IFACE_COMB_NUM_CHANNELS]);
195
196		info->p2p_concurrent = 1;
197		if (info->num_multichan_concurrent < num_channels)
198			info->num_multichan_concurrent = num_channels;
199	}
200
201	return 0;
202}
203
204
205static void wiphy_info_iface_comb(struct wiphy_info_data *info,
206				  struct nlattr *tb)
207{
208	struct nlattr *nl_combi;
209	int rem_combi;
210
211	if (tb == NULL)
212		return;
213
214	nla_for_each_nested(nl_combi, tb, rem_combi) {
215		if (wiphy_info_iface_comb_process(info, nl_combi) > 0)
216			break;
217	}
218}
219
220
221static void wiphy_info_supp_cmds(struct wiphy_info_data *info,
222				 struct nlattr *tb)
223{
224	struct nlattr *nl_cmd;
225	int i;
226
227	if (tb == NULL)
228		return;
229
230	nla_for_each_nested(nl_cmd, tb, i) {
231		switch (nla_get_u32(nl_cmd)) {
232		case NL80211_CMD_AUTHENTICATE:
233			info->auth_supported = 1;
234			break;
235		case NL80211_CMD_CONNECT:
236			info->connect_supported = 1;
237			break;
238		case NL80211_CMD_START_SCHED_SCAN:
239			info->capa->sched_scan_supported = 1;
240			break;
241		case NL80211_CMD_PROBE_CLIENT:
242			info->poll_command_supported = 1;
243			break;
244		case NL80211_CMD_CHANNEL_SWITCH:
245			info->channel_switch_supported = 1;
246			break;
247		case NL80211_CMD_SET_QOS_MAP:
248			info->set_qos_map_supported = 1;
249			break;
250		}
251	}
252}
253
254
255static void wiphy_info_cipher_suites(struct wiphy_info_data *info,
256				     struct nlattr *tb)
257{
258	int i, num;
259	u32 *ciphers;
260
261	if (tb == NULL)
262		return;
263
264	num = nla_len(tb) / sizeof(u32);
265	ciphers = nla_data(tb);
266	for (i = 0; i < num; i++) {
267		u32 c = ciphers[i];
268
269		wpa_printf(MSG_DEBUG, "nl80211: Supported cipher %02x-%02x-%02x:%d",
270			   c >> 24, (c >> 16) & 0xff,
271			   (c >> 8) & 0xff, c & 0xff);
272		switch (c) {
273		case WLAN_CIPHER_SUITE_CCMP_256:
274			info->capa->enc |= WPA_DRIVER_CAPA_ENC_CCMP_256;
275			break;
276		case WLAN_CIPHER_SUITE_GCMP_256:
277			info->capa->enc |= WPA_DRIVER_CAPA_ENC_GCMP_256;
278			break;
279		case WLAN_CIPHER_SUITE_CCMP:
280			info->capa->enc |= WPA_DRIVER_CAPA_ENC_CCMP;
281			break;
282		case WLAN_CIPHER_SUITE_GCMP:
283			info->capa->enc |= WPA_DRIVER_CAPA_ENC_GCMP;
284			break;
285		case WLAN_CIPHER_SUITE_TKIP:
286			info->capa->enc |= WPA_DRIVER_CAPA_ENC_TKIP;
287			break;
288		case WLAN_CIPHER_SUITE_WEP104:
289			info->capa->enc |= WPA_DRIVER_CAPA_ENC_WEP104;
290			break;
291		case WLAN_CIPHER_SUITE_WEP40:
292			info->capa->enc |= WPA_DRIVER_CAPA_ENC_WEP40;
293			break;
294		case WLAN_CIPHER_SUITE_AES_CMAC:
295			info->capa->enc |= WPA_DRIVER_CAPA_ENC_BIP;
296			break;
297		case WLAN_CIPHER_SUITE_BIP_GMAC_128:
298			info->capa->enc |= WPA_DRIVER_CAPA_ENC_BIP_GMAC_128;
299			break;
300		case WLAN_CIPHER_SUITE_BIP_GMAC_256:
301			info->capa->enc |= WPA_DRIVER_CAPA_ENC_BIP_GMAC_256;
302			break;
303		case WLAN_CIPHER_SUITE_BIP_CMAC_256:
304			info->capa->enc |= WPA_DRIVER_CAPA_ENC_BIP_CMAC_256;
305			break;
306		case WLAN_CIPHER_SUITE_NO_GROUP_ADDR:
307			info->capa->enc |= WPA_DRIVER_CAPA_ENC_GTK_NOT_USED;
308			break;
309		}
310	}
311}
312
313
314static void wiphy_info_max_roc(struct wpa_driver_capa *capa,
315			       struct nlattr *tb)
316{
317	if (tb)
318		capa->max_remain_on_chan = nla_get_u32(tb);
319}
320
321
322static void wiphy_info_tdls(struct wpa_driver_capa *capa, struct nlattr *tdls,
323			    struct nlattr *ext_setup)
324{
325	if (tdls == NULL)
326		return;
327
328	wpa_printf(MSG_DEBUG, "nl80211: TDLS supported");
329	capa->flags |= WPA_DRIVER_FLAGS_TDLS_SUPPORT;
330
331	if (ext_setup) {
332		wpa_printf(MSG_DEBUG, "nl80211: TDLS external setup");
333		capa->flags |= WPA_DRIVER_FLAGS_TDLS_EXTERNAL_SETUP;
334	}
335}
336
337
338static int ext_feature_isset(const u8 *ext_features, int ext_features_len,
339			     enum nl80211_ext_feature_index ftidx)
340{
341	u8 ft_byte;
342
343	if ((int) ftidx / 8 >= ext_features_len)
344		return 0;
345
346	ft_byte = ext_features[ftidx / 8];
347	return (ft_byte & BIT(ftidx % 8)) != 0;
348}
349
350
351static void wiphy_info_ext_feature_flags(struct wiphy_info_data *info,
352					 struct nlattr *tb)
353{
354	struct wpa_driver_capa *capa = info->capa;
355
356	if (tb == NULL)
357		return;
358
359	if (ext_feature_isset(nla_data(tb), nla_len(tb),
360			      NL80211_EXT_FEATURE_VHT_IBSS))
361		capa->flags |= WPA_DRIVER_FLAGS_VHT_IBSS;
362}
363
364
365static void wiphy_info_feature_flags(struct wiphy_info_data *info,
366				     struct nlattr *tb)
367{
368	u32 flags;
369	struct wpa_driver_capa *capa = info->capa;
370
371	if (tb == NULL)
372		return;
373
374	flags = nla_get_u32(tb);
375
376	if (flags & NL80211_FEATURE_SK_TX_STATUS)
377		info->data_tx_status = 1;
378
379	if (flags & NL80211_FEATURE_INACTIVITY_TIMER)
380		capa->flags |= WPA_DRIVER_FLAGS_INACTIVITY_TIMER;
381
382	if (flags & NL80211_FEATURE_SAE)
383		capa->flags |= WPA_DRIVER_FLAGS_SAE;
384
385	if (flags & NL80211_FEATURE_NEED_OBSS_SCAN)
386		capa->flags |= WPA_DRIVER_FLAGS_OBSS_SCAN;
387
388	if (flags & NL80211_FEATURE_AP_MODE_CHAN_WIDTH_CHANGE)
389		capa->flags |= WPA_DRIVER_FLAGS_HT_2040_COEX;
390
391	if (flags & NL80211_FEATURE_TDLS_CHANNEL_SWITCH) {
392		wpa_printf(MSG_DEBUG, "nl80211: TDLS channel switch");
393		capa->flags |= WPA_DRIVER_FLAGS_TDLS_CHANNEL_SWITCH;
394	}
395
396	if (flags & NL80211_FEATURE_P2P_GO_CTWIN)
397		info->p2p_go_ctwindow_supported = 1;
398
399	if (flags & NL80211_FEATURE_LOW_PRIORITY_SCAN)
400		info->have_low_prio_scan = 1;
401
402	if (flags & NL80211_FEATURE_SCAN_RANDOM_MAC_ADDR)
403		info->mac_addr_rand_scan_supported = 1;
404
405	if (flags & NL80211_FEATURE_SCHED_SCAN_RANDOM_MAC_ADDR)
406		info->mac_addr_rand_sched_scan_supported = 1;
407
408	if (flags & NL80211_FEATURE_STATIC_SMPS)
409		capa->smps_modes |= WPA_DRIVER_SMPS_MODE_STATIC;
410
411	if (flags & NL80211_FEATURE_DYNAMIC_SMPS)
412		capa->smps_modes |= WPA_DRIVER_SMPS_MODE_DYNAMIC;
413
414	if (flags & NL80211_FEATURE_SUPPORTS_WMM_ADMISSION)
415		info->wmm_ac_supported = 1;
416
417	if (flags & NL80211_FEATURE_DS_PARAM_SET_IE_IN_PROBES)
418		capa->rrm_flags |= WPA_DRIVER_FLAGS_DS_PARAM_SET_IE_IN_PROBES;
419
420	if (flags & NL80211_FEATURE_WFA_TPC_IE_IN_PROBES)
421		capa->rrm_flags |= WPA_DRIVER_FLAGS_WFA_TPC_IE_IN_PROBES;
422
423	if (flags & NL80211_FEATURE_QUIET)
424		capa->rrm_flags |= WPA_DRIVER_FLAGS_QUIET;
425
426	if (flags & NL80211_FEATURE_TX_POWER_INSERTION)
427		capa->rrm_flags |= WPA_DRIVER_FLAGS_TX_POWER_INSERTION;
428
429	if (flags & NL80211_FEATURE_HT_IBSS)
430		capa->flags |= WPA_DRIVER_FLAGS_HT_IBSS;
431}
432
433
434static void wiphy_info_probe_resp_offload(struct wpa_driver_capa *capa,
435					  struct nlattr *tb)
436{
437	u32 protocols;
438
439	if (tb == NULL)
440		return;
441
442	protocols = nla_get_u32(tb);
443	wpa_printf(MSG_DEBUG, "nl80211: Supports Probe Response offload in AP "
444		   "mode");
445	capa->flags |= WPA_DRIVER_FLAGS_PROBE_RESP_OFFLOAD;
446	capa->probe_resp_offloads = probe_resp_offload_support(protocols);
447}
448
449
450static void wiphy_info_wowlan_triggers(struct wpa_driver_capa *capa,
451				       struct nlattr *tb)
452{
453	struct nlattr *triggers[MAX_NL80211_WOWLAN_TRIG + 1];
454
455	if (tb == NULL)
456		return;
457
458	if (nla_parse_nested(triggers, MAX_NL80211_WOWLAN_TRIG,
459			     tb, NULL))
460		return;
461
462	if (triggers[NL80211_WOWLAN_TRIG_ANY])
463		capa->wowlan_triggers.any = 1;
464	if (triggers[NL80211_WOWLAN_TRIG_DISCONNECT])
465		capa->wowlan_triggers.disconnect = 1;
466	if (triggers[NL80211_WOWLAN_TRIG_MAGIC_PKT])
467		capa->wowlan_triggers.magic_pkt = 1;
468	if (triggers[NL80211_WOWLAN_TRIG_GTK_REKEY_FAILURE])
469		capa->wowlan_triggers.gtk_rekey_failure = 1;
470	if (triggers[NL80211_WOWLAN_TRIG_EAP_IDENT_REQUEST])
471		capa->wowlan_triggers.eap_identity_req = 1;
472	if (triggers[NL80211_WOWLAN_TRIG_4WAY_HANDSHAKE])
473		capa->wowlan_triggers.four_way_handshake = 1;
474	if (triggers[NL80211_WOWLAN_TRIG_RFKILL_RELEASE])
475		capa->wowlan_triggers.rfkill_release = 1;
476}
477
478
479static int wiphy_info_handler(struct nl_msg *msg, void *arg)
480{
481	struct nlattr *tb[NL80211_ATTR_MAX + 1];
482	struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
483	struct wiphy_info_data *info = arg;
484	struct wpa_driver_capa *capa = info->capa;
485	struct wpa_driver_nl80211_data *drv = info->drv;
486
487	nla_parse(tb, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
488		  genlmsg_attrlen(gnlh, 0), NULL);
489
490	if (tb[NL80211_ATTR_WIPHY_NAME])
491		os_strlcpy(drv->phyname,
492			   nla_get_string(tb[NL80211_ATTR_WIPHY_NAME]),
493			   sizeof(drv->phyname));
494	if (tb[NL80211_ATTR_MAX_NUM_SCAN_SSIDS])
495		capa->max_scan_ssids =
496			nla_get_u8(tb[NL80211_ATTR_MAX_NUM_SCAN_SSIDS]);
497
498	if (tb[NL80211_ATTR_MAX_NUM_SCHED_SCAN_SSIDS])
499		capa->max_sched_scan_ssids =
500			nla_get_u8(tb[NL80211_ATTR_MAX_NUM_SCHED_SCAN_SSIDS]);
501
502	if (tb[NL80211_ATTR_MAX_MATCH_SETS])
503		capa->max_match_sets =
504			nla_get_u8(tb[NL80211_ATTR_MAX_MATCH_SETS]);
505
506	if (tb[NL80211_ATTR_MAC_ACL_MAX])
507		capa->max_acl_mac_addrs =
508			nla_get_u8(tb[NL80211_ATTR_MAC_ACL_MAX]);
509
510	wiphy_info_supported_iftypes(info, tb[NL80211_ATTR_SUPPORTED_IFTYPES]);
511	wiphy_info_iface_comb(info, tb[NL80211_ATTR_INTERFACE_COMBINATIONS]);
512	wiphy_info_supp_cmds(info, tb[NL80211_ATTR_SUPPORTED_COMMANDS]);
513	wiphy_info_cipher_suites(info, tb[NL80211_ATTR_CIPHER_SUITES]);
514
515	if (tb[NL80211_ATTR_OFFCHANNEL_TX_OK]) {
516		wpa_printf(MSG_DEBUG, "nl80211: Using driver-based "
517			   "off-channel TX");
518		capa->flags |= WPA_DRIVER_FLAGS_OFFCHANNEL_TX;
519	}
520
521	if (tb[NL80211_ATTR_ROAM_SUPPORT]) {
522		wpa_printf(MSG_DEBUG, "nl80211: Using driver-based roaming");
523		capa->flags |= WPA_DRIVER_FLAGS_BSS_SELECTION;
524	}
525
526	wiphy_info_max_roc(capa,
527			   tb[NL80211_ATTR_MAX_REMAIN_ON_CHANNEL_DURATION]);
528
529	if (tb[NL80211_ATTR_SUPPORT_AP_UAPSD])
530		capa->flags |= WPA_DRIVER_FLAGS_AP_UAPSD;
531
532	wiphy_info_tdls(capa, tb[NL80211_ATTR_TDLS_SUPPORT],
533			tb[NL80211_ATTR_TDLS_EXTERNAL_SETUP]);
534
535	if (tb[NL80211_ATTR_DEVICE_AP_SME])
536		info->device_ap_sme = 1;
537
538	wiphy_info_feature_flags(info, tb[NL80211_ATTR_FEATURE_FLAGS]);
539	wiphy_info_ext_feature_flags(info, tb[NL80211_ATTR_EXT_FEATURES]);
540	wiphy_info_probe_resp_offload(capa,
541				      tb[NL80211_ATTR_PROBE_RESP_OFFLOAD]);
542
543	if (tb[NL80211_ATTR_EXT_CAPA] && tb[NL80211_ATTR_EXT_CAPA_MASK] &&
544	    drv->extended_capa == NULL) {
545		drv->extended_capa =
546			os_malloc(nla_len(tb[NL80211_ATTR_EXT_CAPA]));
547		if (drv->extended_capa) {
548			os_memcpy(drv->extended_capa,
549				  nla_data(tb[NL80211_ATTR_EXT_CAPA]),
550				  nla_len(tb[NL80211_ATTR_EXT_CAPA]));
551			drv->extended_capa_len =
552				nla_len(tb[NL80211_ATTR_EXT_CAPA]);
553		}
554		drv->extended_capa_mask =
555			os_malloc(nla_len(tb[NL80211_ATTR_EXT_CAPA_MASK]));
556		if (drv->extended_capa_mask) {
557			os_memcpy(drv->extended_capa_mask,
558				  nla_data(tb[NL80211_ATTR_EXT_CAPA_MASK]),
559				  nla_len(tb[NL80211_ATTR_EXT_CAPA_MASK]));
560		} else {
561			os_free(drv->extended_capa);
562			drv->extended_capa = NULL;
563			drv->extended_capa_len = 0;
564		}
565	}
566
567	if (tb[NL80211_ATTR_VENDOR_DATA]) {
568		struct nlattr *nl;
569		int rem;
570
571		nla_for_each_nested(nl, tb[NL80211_ATTR_VENDOR_DATA], rem) {
572			struct nl80211_vendor_cmd_info *vinfo;
573			if (nla_len(nl) != sizeof(*vinfo)) {
574				wpa_printf(MSG_DEBUG, "nl80211: Unexpected vendor data info");
575				continue;
576			}
577			vinfo = nla_data(nl);
578			if (vinfo->vendor_id == OUI_QCA) {
579				switch (vinfo->subcmd) {
580				case QCA_NL80211_VENDOR_SUBCMD_TEST:
581					drv->vendor_cmd_test_avail = 1;
582					break;
583				case QCA_NL80211_VENDOR_SUBCMD_ROAMING:
584					drv->roaming_vendor_cmd_avail = 1;
585					break;
586				case QCA_NL80211_VENDOR_SUBCMD_DFS_CAPABILITY:
587					drv->dfs_vendor_cmd_avail = 1;
588					break;
589				case QCA_NL80211_VENDOR_SUBCMD_GET_FEATURES:
590					drv->get_features_vendor_cmd_avail = 1;
591					break;
592				case QCA_NL80211_VENDOR_SUBCMD_DO_ACS:
593					drv->capa.flags |=
594						WPA_DRIVER_FLAGS_ACS_OFFLOAD;
595					break;
596				}
597			}
598
599			wpa_printf(MSG_DEBUG, "nl80211: Supported vendor command: vendor_id=0x%x subcmd=%u",
600				   vinfo->vendor_id, vinfo->subcmd);
601		}
602	}
603
604	if (tb[NL80211_ATTR_VENDOR_EVENTS]) {
605		struct nlattr *nl;
606		int rem;
607
608		nla_for_each_nested(nl, tb[NL80211_ATTR_VENDOR_EVENTS], rem) {
609			struct nl80211_vendor_cmd_info *vinfo;
610			if (nla_len(nl) != sizeof(*vinfo)) {
611				wpa_printf(MSG_DEBUG, "nl80211: Unexpected vendor data info");
612				continue;
613			}
614			vinfo = nla_data(nl);
615			wpa_printf(MSG_DEBUG, "nl80211: Supported vendor event: vendor_id=0x%x subcmd=%u",
616				   vinfo->vendor_id, vinfo->subcmd);
617		}
618	}
619
620	wiphy_info_wowlan_triggers(capa,
621				   tb[NL80211_ATTR_WOWLAN_TRIGGERS_SUPPORTED]);
622
623	if (tb[NL80211_ATTR_MAX_AP_ASSOC_STA])
624		capa->max_stations =
625			nla_get_u32(tb[NL80211_ATTR_MAX_AP_ASSOC_STA]);
626
627	return NL_SKIP;
628}
629
630
631static int wpa_driver_nl80211_get_info(struct wpa_driver_nl80211_data *drv,
632				       struct wiphy_info_data *info)
633{
634	u32 feat;
635	struct nl_msg *msg;
636	int flags = 0;
637
638	os_memset(info, 0, sizeof(*info));
639	info->capa = &drv->capa;
640	info->drv = drv;
641
642	feat = get_nl80211_protocol_features(drv);
643	if (feat & NL80211_PROTOCOL_FEATURE_SPLIT_WIPHY_DUMP)
644		flags = NLM_F_DUMP;
645	msg = nl80211_cmd_msg(drv->first_bss, flags, NL80211_CMD_GET_WIPHY);
646	if (!msg || nla_put_flag(msg, NL80211_ATTR_SPLIT_WIPHY_DUMP)) {
647		nlmsg_free(msg);
648		return -1;
649	}
650
651	if (send_and_recv_msgs(drv, msg, wiphy_info_handler, info))
652		return -1;
653
654	if (info->auth_supported)
655		drv->capa.flags |= WPA_DRIVER_FLAGS_SME;
656	else if (!info->connect_supported) {
657		wpa_printf(MSG_INFO, "nl80211: Driver does not support "
658			   "authentication/association or connect commands");
659		info->error = 1;
660	}
661
662	if (info->p2p_go_supported && info->p2p_client_supported)
663		drv->capa.flags |= WPA_DRIVER_FLAGS_P2P_CAPABLE;
664	if (info->p2p_concurrent) {
665		wpa_printf(MSG_DEBUG, "nl80211: Use separate P2P group "
666			   "interface (driver advertised support)");
667		drv->capa.flags |= WPA_DRIVER_FLAGS_P2P_CONCURRENT;
668		drv->capa.flags |= WPA_DRIVER_FLAGS_P2P_MGMT_AND_NON_P2P;
669	}
670	if (info->num_multichan_concurrent > 1) {
671		wpa_printf(MSG_DEBUG, "nl80211: Enable multi-channel "
672			   "concurrent (driver advertised support)");
673		drv->capa.num_multichan_concurrent =
674			info->num_multichan_concurrent;
675	}
676	if (drv->capa.flags & WPA_DRIVER_FLAGS_DEDICATED_P2P_DEVICE)
677		wpa_printf(MSG_DEBUG, "nl80211: use P2P_DEVICE support");
678
679	/* default to 5000 since early versions of mac80211 don't set it */
680	if (!drv->capa.max_remain_on_chan)
681		drv->capa.max_remain_on_chan = 5000;
682
683	if (info->channel_switch_supported)
684		drv->capa.flags |= WPA_DRIVER_FLAGS_AP_CSA;
685	drv->capa.wmm_ac_supported = info->wmm_ac_supported;
686
687	drv->capa.mac_addr_rand_sched_scan_supported =
688		info->mac_addr_rand_sched_scan_supported;
689	drv->capa.mac_addr_rand_scan_supported =
690		info->mac_addr_rand_scan_supported;
691
692	return 0;
693}
694
695
696static int dfs_info_handler(struct nl_msg *msg, void *arg)
697{
698	struct nlattr *tb[NL80211_ATTR_MAX + 1];
699	struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
700	int *dfs_capability_ptr = arg;
701
702	nla_parse(tb, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
703		  genlmsg_attrlen(gnlh, 0), NULL);
704
705	if (tb[NL80211_ATTR_VENDOR_DATA]) {
706		struct nlattr *nl_vend = tb[NL80211_ATTR_VENDOR_DATA];
707		struct nlattr *tb_vendor[QCA_WLAN_VENDOR_ATTR_MAX + 1];
708
709		nla_parse(tb_vendor, QCA_WLAN_VENDOR_ATTR_MAX,
710			  nla_data(nl_vend), nla_len(nl_vend), NULL);
711
712		if (tb_vendor[QCA_WLAN_VENDOR_ATTR_DFS]) {
713			u32 val;
714			val = nla_get_u32(tb_vendor[QCA_WLAN_VENDOR_ATTR_DFS]);
715			wpa_printf(MSG_DEBUG, "nl80211: DFS offload capability: %u",
716				   val);
717			*dfs_capability_ptr = val;
718		}
719	}
720
721	return NL_SKIP;
722}
723
724
725static void qca_nl80211_check_dfs_capa(struct wpa_driver_nl80211_data *drv)
726{
727	struct nl_msg *msg;
728	int dfs_capability = 0;
729	int ret;
730
731	if (!drv->dfs_vendor_cmd_avail)
732		return;
733
734	if (!(msg = nl80211_drv_msg(drv, 0, NL80211_CMD_VENDOR)) ||
735	    nla_put_u32(msg, NL80211_ATTR_VENDOR_ID, OUI_QCA) ||
736	    nla_put_u32(msg, NL80211_ATTR_VENDOR_SUBCMD,
737			QCA_NL80211_VENDOR_SUBCMD_DFS_CAPABILITY)) {
738		nlmsg_free(msg);
739		return;
740	}
741
742	ret = send_and_recv_msgs(drv, msg, dfs_info_handler, &dfs_capability);
743	if (!ret && dfs_capability)
744		drv->capa.flags |= WPA_DRIVER_FLAGS_DFS_OFFLOAD;
745}
746
747
748struct features_info {
749	u8 *flags;
750	size_t flags_len;
751};
752
753
754static int features_info_handler(struct nl_msg *msg, void *arg)
755{
756	struct nlattr *tb[NL80211_ATTR_MAX + 1];
757	struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
758	struct features_info *info = arg;
759	struct nlattr *nl_vend, *attr;
760
761	nla_parse(tb, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
762		  genlmsg_attrlen(gnlh, 0), NULL);
763
764	nl_vend = tb[NL80211_ATTR_VENDOR_DATA];
765	if (nl_vend) {
766		struct nlattr *tb_vendor[QCA_WLAN_VENDOR_ATTR_MAX + 1];
767
768		nla_parse(tb_vendor, QCA_WLAN_VENDOR_ATTR_MAX,
769			  nla_data(nl_vend), nla_len(nl_vend), NULL);
770
771		attr = tb_vendor[QCA_WLAN_VENDOR_ATTR_FEATURE_FLAGS];
772		if (attr) {
773			info->flags = nla_data(attr);
774			info->flags_len = nla_len(attr);
775		}
776	}
777
778	return NL_SKIP;
779}
780
781
782static int check_feature(enum qca_wlan_vendor_features feature,
783			 struct features_info *info)
784{
785	size_t idx = feature / 8;
786
787	return (idx < info->flags_len) &&
788		(info->flags[idx] & BIT(feature % 8));
789}
790
791
792static void qca_nl80211_get_features(struct wpa_driver_nl80211_data *drv)
793{
794	struct nl_msg *msg;
795	struct features_info info;
796	int ret;
797
798	if (!drv->get_features_vendor_cmd_avail)
799		return;
800
801	if (!(msg = nl80211_drv_msg(drv, 0, NL80211_CMD_VENDOR)) ||
802	    nla_put_u32(msg, NL80211_ATTR_VENDOR_ID, OUI_QCA) ||
803	    nla_put_u32(msg, NL80211_ATTR_VENDOR_SUBCMD,
804			QCA_NL80211_VENDOR_SUBCMD_GET_FEATURES)) {
805		nlmsg_free(msg);
806		return;
807	}
808
809	os_memset(&info, 0, sizeof(info));
810	ret = send_and_recv_msgs(drv, msg, features_info_handler, &info);
811	if (ret || !info.flags)
812		return;
813
814	if (check_feature(QCA_WLAN_VENDOR_FEATURE_KEY_MGMT_OFFLOAD, &info))
815		drv->capa.flags |= WPA_DRIVER_FLAGS_KEY_MGMT_OFFLOAD;
816
817	if (check_feature(QCA_WLAN_VENDOR_FEATURE_SUPPORT_HW_MODE_ANY, &info))
818		drv->capa.flags |= WPA_DRIVER_FLAGS_SUPPORT_HW_MODE_ANY;
819}
820
821
822int wpa_driver_nl80211_capa(struct wpa_driver_nl80211_data *drv)
823{
824	struct wiphy_info_data info;
825	if (wpa_driver_nl80211_get_info(drv, &info))
826		return -1;
827
828	if (info.error)
829		return -1;
830
831	drv->has_capability = 1;
832	drv->capa.key_mgmt = WPA_DRIVER_CAPA_KEY_MGMT_WPA |
833		WPA_DRIVER_CAPA_KEY_MGMT_WPA_PSK |
834		WPA_DRIVER_CAPA_KEY_MGMT_WPA2 |
835		WPA_DRIVER_CAPA_KEY_MGMT_WPA2_PSK |
836		WPA_DRIVER_CAPA_KEY_MGMT_SUITE_B |
837		WPA_DRIVER_CAPA_KEY_MGMT_SUITE_B_192;
838	drv->capa.auth = WPA_DRIVER_AUTH_OPEN |
839		WPA_DRIVER_AUTH_SHARED |
840		WPA_DRIVER_AUTH_LEAP;
841
842	drv->capa.flags |= WPA_DRIVER_FLAGS_SANE_ERROR_CODES;
843	drv->capa.flags |= WPA_DRIVER_FLAGS_SET_KEYS_AFTER_ASSOC_DONE;
844	drv->capa.flags |= WPA_DRIVER_FLAGS_EAPOL_TX_STATUS;
845
846	/*
847	 * As all cfg80211 drivers must support cases where the AP interface is
848	 * removed without the knowledge of wpa_supplicant/hostapd, e.g., in
849	 * case that the user space daemon has crashed, they must be able to
850	 * cleanup all stations and key entries in the AP tear down flow. Thus,
851	 * this flag can/should always be set for cfg80211 drivers.
852	 */
853	drv->capa.flags |= WPA_DRIVER_FLAGS_AP_TEARDOWN_SUPPORT;
854
855	if (!info.device_ap_sme) {
856		drv->capa.flags |= WPA_DRIVER_FLAGS_DEAUTH_TX_STATUS;
857
858		/*
859		 * No AP SME is currently assumed to also indicate no AP MLME
860		 * in the driver/firmware.
861		 */
862		drv->capa.flags |= WPA_DRIVER_FLAGS_AP_MLME;
863	}
864
865	drv->device_ap_sme = info.device_ap_sme;
866	drv->poll_command_supported = info.poll_command_supported;
867	drv->data_tx_status = info.data_tx_status;
868	drv->p2p_go_ctwindow_supported = info.p2p_go_ctwindow_supported;
869	if (info.set_qos_map_supported)
870		drv->capa.flags |= WPA_DRIVER_FLAGS_QOS_MAPPING;
871	drv->have_low_prio_scan = info.have_low_prio_scan;
872
873	/*
874	 * If poll command and tx status are supported, mac80211 is new enough
875	 * to have everything we need to not need monitor interfaces.
876	 */
877	drv->use_monitor = !info.poll_command_supported || !info.data_tx_status;
878
879	if (drv->device_ap_sme && drv->use_monitor) {
880		/*
881		 * Non-mac80211 drivers may not support monitor interface.
882		 * Make sure we do not get stuck with incorrect capability here
883		 * by explicitly testing this.
884		 */
885		if (!info.monitor_supported) {
886			wpa_printf(MSG_DEBUG, "nl80211: Disable use_monitor "
887				   "with device_ap_sme since no monitor mode "
888				   "support detected");
889			drv->use_monitor = 0;
890		}
891	}
892
893	/*
894	 * If we aren't going to use monitor interfaces, but the
895	 * driver doesn't support data TX status, we won't get TX
896	 * status for EAPOL frames.
897	 */
898	if (!drv->use_monitor && !info.data_tx_status)
899		drv->capa.flags &= ~WPA_DRIVER_FLAGS_EAPOL_TX_STATUS;
900
901	qca_nl80211_check_dfs_capa(drv);
902	qca_nl80211_get_features(drv);
903
904	return 0;
905}
906
907
908struct phy_info_arg {
909	u16 *num_modes;
910	struct hostapd_hw_modes *modes;
911	int last_mode, last_chan_idx;
912};
913
914static void phy_info_ht_capa(struct hostapd_hw_modes *mode, struct nlattr *capa,
915			     struct nlattr *ampdu_factor,
916			     struct nlattr *ampdu_density,
917			     struct nlattr *mcs_set)
918{
919	if (capa)
920		mode->ht_capab = nla_get_u16(capa);
921
922	if (ampdu_factor)
923		mode->a_mpdu_params |= nla_get_u8(ampdu_factor) & 0x03;
924
925	if (ampdu_density)
926		mode->a_mpdu_params |= nla_get_u8(ampdu_density) << 2;
927
928	if (mcs_set && nla_len(mcs_set) >= 16) {
929		u8 *mcs;
930		mcs = nla_data(mcs_set);
931		os_memcpy(mode->mcs_set, mcs, 16);
932	}
933}
934
935
936static void phy_info_vht_capa(struct hostapd_hw_modes *mode,
937			      struct nlattr *capa,
938			      struct nlattr *mcs_set)
939{
940	if (capa)
941		mode->vht_capab = nla_get_u32(capa);
942
943	if (mcs_set && nla_len(mcs_set) >= 8) {
944		u8 *mcs;
945		mcs = nla_data(mcs_set);
946		os_memcpy(mode->vht_mcs_set, mcs, 8);
947	}
948}
949
950
951static void phy_info_freq(struct hostapd_hw_modes *mode,
952			  struct hostapd_channel_data *chan,
953			  struct nlattr *tb_freq[])
954{
955	u8 channel;
956	chan->freq = nla_get_u32(tb_freq[NL80211_FREQUENCY_ATTR_FREQ]);
957	chan->flag = 0;
958	chan->dfs_cac_ms = 0;
959	if (ieee80211_freq_to_chan(chan->freq, &channel) != NUM_HOSTAPD_MODES)
960		chan->chan = channel;
961
962	if (tb_freq[NL80211_FREQUENCY_ATTR_DISABLED])
963		chan->flag |= HOSTAPD_CHAN_DISABLED;
964	if (tb_freq[NL80211_FREQUENCY_ATTR_NO_IR])
965		chan->flag |= HOSTAPD_CHAN_NO_IR;
966	if (tb_freq[NL80211_FREQUENCY_ATTR_RADAR])
967		chan->flag |= HOSTAPD_CHAN_RADAR;
968	if (tb_freq[NL80211_FREQUENCY_ATTR_INDOOR_ONLY])
969		chan->flag |= HOSTAPD_CHAN_INDOOR_ONLY;
970	if (tb_freq[NL80211_FREQUENCY_ATTR_GO_CONCURRENT])
971		chan->flag |= HOSTAPD_CHAN_GO_CONCURRENT;
972
973	if (tb_freq[NL80211_FREQUENCY_ATTR_DFS_STATE]) {
974		enum nl80211_dfs_state state =
975			nla_get_u32(tb_freq[NL80211_FREQUENCY_ATTR_DFS_STATE]);
976
977		switch (state) {
978		case NL80211_DFS_USABLE:
979			chan->flag |= HOSTAPD_CHAN_DFS_USABLE;
980			break;
981		case NL80211_DFS_AVAILABLE:
982			chan->flag |= HOSTAPD_CHAN_DFS_AVAILABLE;
983			break;
984		case NL80211_DFS_UNAVAILABLE:
985			chan->flag |= HOSTAPD_CHAN_DFS_UNAVAILABLE;
986			break;
987		}
988	}
989
990	if (tb_freq[NL80211_FREQUENCY_ATTR_DFS_CAC_TIME]) {
991		chan->dfs_cac_ms = nla_get_u32(
992			tb_freq[NL80211_FREQUENCY_ATTR_DFS_CAC_TIME]);
993	}
994}
995
996
997static int phy_info_freqs(struct phy_info_arg *phy_info,
998			  struct hostapd_hw_modes *mode, struct nlattr *tb)
999{
1000	static struct nla_policy freq_policy[NL80211_FREQUENCY_ATTR_MAX + 1] = {
1001		[NL80211_FREQUENCY_ATTR_FREQ] = { .type = NLA_U32 },
1002		[NL80211_FREQUENCY_ATTR_DISABLED] = { .type = NLA_FLAG },
1003		[NL80211_FREQUENCY_ATTR_NO_IR] = { .type = NLA_FLAG },
1004		[NL80211_FREQUENCY_ATTR_RADAR] = { .type = NLA_FLAG },
1005		[NL80211_FREQUENCY_ATTR_MAX_TX_POWER] = { .type = NLA_U32 },
1006		[NL80211_FREQUENCY_ATTR_DFS_STATE] = { .type = NLA_U32 },
1007	};
1008	int new_channels = 0;
1009	struct hostapd_channel_data *channel;
1010	struct nlattr *tb_freq[NL80211_FREQUENCY_ATTR_MAX + 1];
1011	struct nlattr *nl_freq;
1012	int rem_freq, idx;
1013
1014	if (tb == NULL)
1015		return NL_OK;
1016
1017	nla_for_each_nested(nl_freq, tb, rem_freq) {
1018		nla_parse(tb_freq, NL80211_FREQUENCY_ATTR_MAX,
1019			  nla_data(nl_freq), nla_len(nl_freq), freq_policy);
1020		if (!tb_freq[NL80211_FREQUENCY_ATTR_FREQ])
1021			continue;
1022		new_channels++;
1023	}
1024
1025	channel = os_realloc_array(mode->channels,
1026				   mode->num_channels + new_channels,
1027				   sizeof(struct hostapd_channel_data));
1028	if (!channel)
1029		return NL_SKIP;
1030
1031	mode->channels = channel;
1032	mode->num_channels += new_channels;
1033
1034	idx = phy_info->last_chan_idx;
1035
1036	nla_for_each_nested(nl_freq, tb, rem_freq) {
1037		nla_parse(tb_freq, NL80211_FREQUENCY_ATTR_MAX,
1038			  nla_data(nl_freq), nla_len(nl_freq), freq_policy);
1039		if (!tb_freq[NL80211_FREQUENCY_ATTR_FREQ])
1040			continue;
1041		phy_info_freq(mode, &mode->channels[idx], tb_freq);
1042		idx++;
1043	}
1044	phy_info->last_chan_idx = idx;
1045
1046	return NL_OK;
1047}
1048
1049
1050static int phy_info_rates(struct hostapd_hw_modes *mode, struct nlattr *tb)
1051{
1052	static struct nla_policy rate_policy[NL80211_BITRATE_ATTR_MAX + 1] = {
1053		[NL80211_BITRATE_ATTR_RATE] = { .type = NLA_U32 },
1054		[NL80211_BITRATE_ATTR_2GHZ_SHORTPREAMBLE] =
1055		{ .type = NLA_FLAG },
1056	};
1057	struct nlattr *tb_rate[NL80211_BITRATE_ATTR_MAX + 1];
1058	struct nlattr *nl_rate;
1059	int rem_rate, idx;
1060
1061	if (tb == NULL)
1062		return NL_OK;
1063
1064	nla_for_each_nested(nl_rate, tb, rem_rate) {
1065		nla_parse(tb_rate, NL80211_BITRATE_ATTR_MAX,
1066			  nla_data(nl_rate), nla_len(nl_rate),
1067			  rate_policy);
1068		if (!tb_rate[NL80211_BITRATE_ATTR_RATE])
1069			continue;
1070		mode->num_rates++;
1071	}
1072
1073	mode->rates = os_calloc(mode->num_rates, sizeof(int));
1074	if (!mode->rates)
1075		return NL_SKIP;
1076
1077	idx = 0;
1078
1079	nla_for_each_nested(nl_rate, tb, rem_rate) {
1080		nla_parse(tb_rate, NL80211_BITRATE_ATTR_MAX,
1081			  nla_data(nl_rate), nla_len(nl_rate),
1082			  rate_policy);
1083		if (!tb_rate[NL80211_BITRATE_ATTR_RATE])
1084			continue;
1085		mode->rates[idx] = nla_get_u32(
1086			tb_rate[NL80211_BITRATE_ATTR_RATE]);
1087		idx++;
1088	}
1089
1090	return NL_OK;
1091}
1092
1093
1094static int phy_info_band(struct phy_info_arg *phy_info, struct nlattr *nl_band)
1095{
1096	struct nlattr *tb_band[NL80211_BAND_ATTR_MAX + 1];
1097	struct hostapd_hw_modes *mode;
1098	int ret;
1099
1100	if (phy_info->last_mode != nl_band->nla_type) {
1101		mode = os_realloc_array(phy_info->modes,
1102					*phy_info->num_modes + 1,
1103					sizeof(*mode));
1104		if (!mode)
1105			return NL_SKIP;
1106		phy_info->modes = mode;
1107
1108		mode = &phy_info->modes[*(phy_info->num_modes)];
1109		os_memset(mode, 0, sizeof(*mode));
1110		mode->mode = NUM_HOSTAPD_MODES;
1111		mode->flags = HOSTAPD_MODE_FLAG_HT_INFO_KNOWN |
1112			HOSTAPD_MODE_FLAG_VHT_INFO_KNOWN;
1113
1114		/*
1115		 * Unsupported VHT MCS stream is defined as value 3, so the VHT
1116		 * MCS RX/TX map must be initialized with 0xffff to mark all 8
1117		 * possible streams as unsupported. This will be overridden if
1118		 * driver advertises VHT support.
1119		 */
1120		mode->vht_mcs_set[0] = 0xff;
1121		mode->vht_mcs_set[1] = 0xff;
1122		mode->vht_mcs_set[4] = 0xff;
1123		mode->vht_mcs_set[5] = 0xff;
1124
1125		*(phy_info->num_modes) += 1;
1126		phy_info->last_mode = nl_band->nla_type;
1127		phy_info->last_chan_idx = 0;
1128	} else
1129		mode = &phy_info->modes[*(phy_info->num_modes) - 1];
1130
1131	nla_parse(tb_band, NL80211_BAND_ATTR_MAX, nla_data(nl_band),
1132		  nla_len(nl_band), NULL);
1133
1134	phy_info_ht_capa(mode, tb_band[NL80211_BAND_ATTR_HT_CAPA],
1135			 tb_band[NL80211_BAND_ATTR_HT_AMPDU_FACTOR],
1136			 tb_band[NL80211_BAND_ATTR_HT_AMPDU_DENSITY],
1137			 tb_band[NL80211_BAND_ATTR_HT_MCS_SET]);
1138	phy_info_vht_capa(mode, tb_band[NL80211_BAND_ATTR_VHT_CAPA],
1139			  tb_band[NL80211_BAND_ATTR_VHT_MCS_SET]);
1140	ret = phy_info_freqs(phy_info, mode, tb_band[NL80211_BAND_ATTR_FREQS]);
1141	if (ret != NL_OK)
1142		return ret;
1143	ret = phy_info_rates(mode, tb_band[NL80211_BAND_ATTR_RATES]);
1144	if (ret != NL_OK)
1145		return ret;
1146
1147	return NL_OK;
1148}
1149
1150
1151static int phy_info_handler(struct nl_msg *msg, void *arg)
1152{
1153	struct nlattr *tb_msg[NL80211_ATTR_MAX + 1];
1154	struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
1155	struct phy_info_arg *phy_info = arg;
1156	struct nlattr *nl_band;
1157	int rem_band;
1158
1159	nla_parse(tb_msg, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
1160		  genlmsg_attrlen(gnlh, 0), NULL);
1161
1162	if (!tb_msg[NL80211_ATTR_WIPHY_BANDS])
1163		return NL_SKIP;
1164
1165	nla_for_each_nested(nl_band, tb_msg[NL80211_ATTR_WIPHY_BANDS], rem_band)
1166	{
1167		int res = phy_info_band(phy_info, nl_band);
1168		if (res != NL_OK)
1169			return res;
1170	}
1171
1172	return NL_SKIP;
1173}
1174
1175
1176static struct hostapd_hw_modes *
1177wpa_driver_nl80211_postprocess_modes(struct hostapd_hw_modes *modes,
1178				     u16 *num_modes)
1179{
1180	u16 m;
1181	struct hostapd_hw_modes *mode11g = NULL, *nmodes, *mode;
1182	int i, mode11g_idx = -1;
1183
1184	/* heuristic to set up modes */
1185	for (m = 0; m < *num_modes; m++) {
1186		if (!modes[m].num_channels)
1187			continue;
1188		if (modes[m].channels[0].freq < 4000) {
1189			modes[m].mode = HOSTAPD_MODE_IEEE80211B;
1190			for (i = 0; i < modes[m].num_rates; i++) {
1191				if (modes[m].rates[i] > 200) {
1192					modes[m].mode = HOSTAPD_MODE_IEEE80211G;
1193					break;
1194				}
1195			}
1196		} else if (modes[m].channels[0].freq > 50000)
1197			modes[m].mode = HOSTAPD_MODE_IEEE80211AD;
1198		else
1199			modes[m].mode = HOSTAPD_MODE_IEEE80211A;
1200	}
1201
1202	/* If only 802.11g mode is included, use it to construct matching
1203	 * 802.11b mode data. */
1204
1205	for (m = 0; m < *num_modes; m++) {
1206		if (modes[m].mode == HOSTAPD_MODE_IEEE80211B)
1207			return modes; /* 802.11b already included */
1208		if (modes[m].mode == HOSTAPD_MODE_IEEE80211G)
1209			mode11g_idx = m;
1210	}
1211
1212	if (mode11g_idx < 0)
1213		return modes; /* 2.4 GHz band not supported at all */
1214
1215	nmodes = os_realloc_array(modes, *num_modes + 1, sizeof(*nmodes));
1216	if (nmodes == NULL)
1217		return modes; /* Could not add 802.11b mode */
1218
1219	mode = &nmodes[*num_modes];
1220	os_memset(mode, 0, sizeof(*mode));
1221	(*num_modes)++;
1222	modes = nmodes;
1223
1224	mode->mode = HOSTAPD_MODE_IEEE80211B;
1225
1226	mode11g = &modes[mode11g_idx];
1227	mode->num_channels = mode11g->num_channels;
1228	mode->channels = os_malloc(mode11g->num_channels *
1229				   sizeof(struct hostapd_channel_data));
1230	if (mode->channels == NULL) {
1231		(*num_modes)--;
1232		return modes; /* Could not add 802.11b mode */
1233	}
1234	os_memcpy(mode->channels, mode11g->channels,
1235		  mode11g->num_channels * sizeof(struct hostapd_channel_data));
1236
1237	mode->num_rates = 0;
1238	mode->rates = os_malloc(4 * sizeof(int));
1239	if (mode->rates == NULL) {
1240		os_free(mode->channels);
1241		(*num_modes)--;
1242		return modes; /* Could not add 802.11b mode */
1243	}
1244
1245	for (i = 0; i < mode11g->num_rates; i++) {
1246		if (mode11g->rates[i] != 10 && mode11g->rates[i] != 20 &&
1247		    mode11g->rates[i] != 55 && mode11g->rates[i] != 110)
1248			continue;
1249		mode->rates[mode->num_rates] = mode11g->rates[i];
1250		mode->num_rates++;
1251		if (mode->num_rates == 4)
1252			break;
1253	}
1254
1255	if (mode->num_rates == 0) {
1256		os_free(mode->channels);
1257		os_free(mode->rates);
1258		(*num_modes)--;
1259		return modes; /* No 802.11b rates */
1260	}
1261
1262	wpa_printf(MSG_DEBUG, "nl80211: Added 802.11b mode based on 802.11g "
1263		   "information");
1264
1265	return modes;
1266}
1267
1268
1269static void nl80211_set_ht40_mode(struct hostapd_hw_modes *mode, int start,
1270				  int end)
1271{
1272	int c;
1273
1274	for (c = 0; c < mode->num_channels; c++) {
1275		struct hostapd_channel_data *chan = &mode->channels[c];
1276		if (chan->freq - 10 >= start && chan->freq + 10 <= end)
1277			chan->flag |= HOSTAPD_CHAN_HT40;
1278	}
1279}
1280
1281
1282static void nl80211_set_ht40_mode_sec(struct hostapd_hw_modes *mode, int start,
1283				      int end)
1284{
1285	int c;
1286
1287	for (c = 0; c < mode->num_channels; c++) {
1288		struct hostapd_channel_data *chan = &mode->channels[c];
1289		if (!(chan->flag & HOSTAPD_CHAN_HT40))
1290			continue;
1291		if (chan->freq - 30 >= start && chan->freq - 10 <= end)
1292			chan->flag |= HOSTAPD_CHAN_HT40MINUS;
1293		if (chan->freq + 10 >= start && chan->freq + 30 <= end)
1294			chan->flag |= HOSTAPD_CHAN_HT40PLUS;
1295	}
1296}
1297
1298
1299static void nl80211_reg_rule_max_eirp(u32 start, u32 end, u32 max_eirp,
1300				      struct phy_info_arg *results)
1301{
1302	u16 m;
1303
1304	for (m = 0; m < *results->num_modes; m++) {
1305		int c;
1306		struct hostapd_hw_modes *mode = &results->modes[m];
1307
1308		for (c = 0; c < mode->num_channels; c++) {
1309			struct hostapd_channel_data *chan = &mode->channels[c];
1310			if ((u32) chan->freq - 10 >= start &&
1311			    (u32) chan->freq + 10 <= end)
1312				chan->max_tx_power = max_eirp;
1313		}
1314	}
1315}
1316
1317
1318static void nl80211_reg_rule_ht40(u32 start, u32 end,
1319				  struct phy_info_arg *results)
1320{
1321	u16 m;
1322
1323	for (m = 0; m < *results->num_modes; m++) {
1324		if (!(results->modes[m].ht_capab &
1325		      HT_CAP_INFO_SUPP_CHANNEL_WIDTH_SET))
1326			continue;
1327		nl80211_set_ht40_mode(&results->modes[m], start, end);
1328	}
1329}
1330
1331
1332static void nl80211_reg_rule_sec(struct nlattr *tb[],
1333				 struct phy_info_arg *results)
1334{
1335	u32 start, end, max_bw;
1336	u16 m;
1337
1338	if (tb[NL80211_ATTR_FREQ_RANGE_START] == NULL ||
1339	    tb[NL80211_ATTR_FREQ_RANGE_END] == NULL ||
1340	    tb[NL80211_ATTR_FREQ_RANGE_MAX_BW] == NULL)
1341		return;
1342
1343	start = nla_get_u32(tb[NL80211_ATTR_FREQ_RANGE_START]) / 1000;
1344	end = nla_get_u32(tb[NL80211_ATTR_FREQ_RANGE_END]) / 1000;
1345	max_bw = nla_get_u32(tb[NL80211_ATTR_FREQ_RANGE_MAX_BW]) / 1000;
1346
1347	if (max_bw < 20)
1348		return;
1349
1350	for (m = 0; m < *results->num_modes; m++) {
1351		if (!(results->modes[m].ht_capab &
1352		      HT_CAP_INFO_SUPP_CHANNEL_WIDTH_SET))
1353			continue;
1354		nl80211_set_ht40_mode_sec(&results->modes[m], start, end);
1355	}
1356}
1357
1358
1359static void nl80211_set_vht_mode(struct hostapd_hw_modes *mode, int start,
1360				 int end)
1361{
1362	int c;
1363
1364	for (c = 0; c < mode->num_channels; c++) {
1365		struct hostapd_channel_data *chan = &mode->channels[c];
1366		if (chan->freq - 10 >= start && chan->freq + 70 <= end)
1367			chan->flag |= HOSTAPD_CHAN_VHT_10_70;
1368
1369		if (chan->freq - 30 >= start && chan->freq + 50 <= end)
1370			chan->flag |= HOSTAPD_CHAN_VHT_30_50;
1371
1372		if (chan->freq - 50 >= start && chan->freq + 30 <= end)
1373			chan->flag |= HOSTAPD_CHAN_VHT_50_30;
1374
1375		if (chan->freq - 70 >= start && chan->freq + 10 <= end)
1376			chan->flag |= HOSTAPD_CHAN_VHT_70_10;
1377	}
1378}
1379
1380
1381static void nl80211_reg_rule_vht(struct nlattr *tb[],
1382				 struct phy_info_arg *results)
1383{
1384	u32 start, end, max_bw;
1385	u16 m;
1386
1387	if (tb[NL80211_ATTR_FREQ_RANGE_START] == NULL ||
1388	    tb[NL80211_ATTR_FREQ_RANGE_END] == NULL ||
1389	    tb[NL80211_ATTR_FREQ_RANGE_MAX_BW] == NULL)
1390		return;
1391
1392	start = nla_get_u32(tb[NL80211_ATTR_FREQ_RANGE_START]) / 1000;
1393	end = nla_get_u32(tb[NL80211_ATTR_FREQ_RANGE_END]) / 1000;
1394	max_bw = nla_get_u32(tb[NL80211_ATTR_FREQ_RANGE_MAX_BW]) / 1000;
1395
1396	if (max_bw < 80)
1397		return;
1398
1399	for (m = 0; m < *results->num_modes; m++) {
1400		if (!(results->modes[m].ht_capab &
1401		      HT_CAP_INFO_SUPP_CHANNEL_WIDTH_SET))
1402			continue;
1403		/* TODO: use a real VHT support indication */
1404		if (!results->modes[m].vht_capab)
1405			continue;
1406
1407		nl80211_set_vht_mode(&results->modes[m], start, end);
1408	}
1409}
1410
1411
1412static const char * dfs_domain_name(enum nl80211_dfs_regions region)
1413{
1414	switch (region) {
1415	case NL80211_DFS_UNSET:
1416		return "DFS-UNSET";
1417	case NL80211_DFS_FCC:
1418		return "DFS-FCC";
1419	case NL80211_DFS_ETSI:
1420		return "DFS-ETSI";
1421	case NL80211_DFS_JP:
1422		return "DFS-JP";
1423	default:
1424		return "DFS-invalid";
1425	}
1426}
1427
1428
1429static int nl80211_get_reg(struct nl_msg *msg, void *arg)
1430{
1431	struct phy_info_arg *results = arg;
1432	struct nlattr *tb_msg[NL80211_ATTR_MAX + 1];
1433	struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
1434	struct nlattr *nl_rule;
1435	struct nlattr *tb_rule[NL80211_FREQUENCY_ATTR_MAX + 1];
1436	int rem_rule;
1437	static struct nla_policy reg_policy[NL80211_FREQUENCY_ATTR_MAX + 1] = {
1438		[NL80211_ATTR_REG_RULE_FLAGS] = { .type = NLA_U32 },
1439		[NL80211_ATTR_FREQ_RANGE_START] = { .type = NLA_U32 },
1440		[NL80211_ATTR_FREQ_RANGE_END] = { .type = NLA_U32 },
1441		[NL80211_ATTR_FREQ_RANGE_MAX_BW] = { .type = NLA_U32 },
1442		[NL80211_ATTR_POWER_RULE_MAX_ANT_GAIN] = { .type = NLA_U32 },
1443		[NL80211_ATTR_POWER_RULE_MAX_EIRP] = { .type = NLA_U32 },
1444	};
1445
1446	nla_parse(tb_msg, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
1447		  genlmsg_attrlen(gnlh, 0), NULL);
1448	if (!tb_msg[NL80211_ATTR_REG_ALPHA2] ||
1449	    !tb_msg[NL80211_ATTR_REG_RULES]) {
1450		wpa_printf(MSG_DEBUG, "nl80211: No regulatory information "
1451			   "available");
1452		return NL_SKIP;
1453	}
1454
1455	if (tb_msg[NL80211_ATTR_DFS_REGION]) {
1456		enum nl80211_dfs_regions dfs_domain;
1457		dfs_domain = nla_get_u8(tb_msg[NL80211_ATTR_DFS_REGION]);
1458		wpa_printf(MSG_DEBUG, "nl80211: Regulatory information - country=%s (%s)",
1459			   (char *) nla_data(tb_msg[NL80211_ATTR_REG_ALPHA2]),
1460			   dfs_domain_name(dfs_domain));
1461	} else {
1462		wpa_printf(MSG_DEBUG, "nl80211: Regulatory information - country=%s",
1463			   (char *) nla_data(tb_msg[NL80211_ATTR_REG_ALPHA2]));
1464	}
1465
1466	nla_for_each_nested(nl_rule, tb_msg[NL80211_ATTR_REG_RULES], rem_rule)
1467	{
1468		u32 start, end, max_eirp = 0, max_bw = 0, flags = 0;
1469		nla_parse(tb_rule, NL80211_FREQUENCY_ATTR_MAX,
1470			  nla_data(nl_rule), nla_len(nl_rule), reg_policy);
1471		if (tb_rule[NL80211_ATTR_FREQ_RANGE_START] == NULL ||
1472		    tb_rule[NL80211_ATTR_FREQ_RANGE_END] == NULL)
1473			continue;
1474		start = nla_get_u32(tb_rule[NL80211_ATTR_FREQ_RANGE_START]) / 1000;
1475		end = nla_get_u32(tb_rule[NL80211_ATTR_FREQ_RANGE_END]) / 1000;
1476		if (tb_rule[NL80211_ATTR_POWER_RULE_MAX_EIRP])
1477			max_eirp = nla_get_u32(tb_rule[NL80211_ATTR_POWER_RULE_MAX_EIRP]) / 100;
1478		if (tb_rule[NL80211_ATTR_FREQ_RANGE_MAX_BW])
1479			max_bw = nla_get_u32(tb_rule[NL80211_ATTR_FREQ_RANGE_MAX_BW]) / 1000;
1480		if (tb_rule[NL80211_ATTR_REG_RULE_FLAGS])
1481			flags = nla_get_u32(tb_rule[NL80211_ATTR_REG_RULE_FLAGS]);
1482
1483		wpa_printf(MSG_DEBUG, "nl80211: %u-%u @ %u MHz %u mBm%s%s%s%s%s%s%s%s",
1484			   start, end, max_bw, max_eirp,
1485			   flags & NL80211_RRF_NO_OFDM ? " (no OFDM)" : "",
1486			   flags & NL80211_RRF_NO_CCK ? " (no CCK)" : "",
1487			   flags & NL80211_RRF_NO_INDOOR ? " (no indoor)" : "",
1488			   flags & NL80211_RRF_NO_OUTDOOR ? " (no outdoor)" :
1489			   "",
1490			   flags & NL80211_RRF_DFS ? " (DFS)" : "",
1491			   flags & NL80211_RRF_PTP_ONLY ? " (PTP only)" : "",
1492			   flags & NL80211_RRF_PTMP_ONLY ? " (PTMP only)" : "",
1493			   flags & NL80211_RRF_NO_IR ? " (no IR)" : "");
1494		if (max_bw >= 40)
1495			nl80211_reg_rule_ht40(start, end, results);
1496		if (tb_rule[NL80211_ATTR_POWER_RULE_MAX_EIRP])
1497			nl80211_reg_rule_max_eirp(start, end, max_eirp,
1498						  results);
1499	}
1500
1501	nla_for_each_nested(nl_rule, tb_msg[NL80211_ATTR_REG_RULES], rem_rule)
1502	{
1503		nla_parse(tb_rule, NL80211_FREQUENCY_ATTR_MAX,
1504			  nla_data(nl_rule), nla_len(nl_rule), reg_policy);
1505		nl80211_reg_rule_sec(tb_rule, results);
1506	}
1507
1508	nla_for_each_nested(nl_rule, tb_msg[NL80211_ATTR_REG_RULES], rem_rule)
1509	{
1510		nla_parse(tb_rule, NL80211_FREQUENCY_ATTR_MAX,
1511			  nla_data(nl_rule), nla_len(nl_rule), reg_policy);
1512		nl80211_reg_rule_vht(tb_rule, results);
1513	}
1514
1515	return NL_SKIP;
1516}
1517
1518
1519static int nl80211_set_regulatory_flags(struct wpa_driver_nl80211_data *drv,
1520					struct phy_info_arg *results)
1521{
1522	struct nl_msg *msg;
1523
1524	msg = nlmsg_alloc();
1525	if (!msg)
1526		return -ENOMEM;
1527
1528	nl80211_cmd(drv, msg, 0, NL80211_CMD_GET_REG);
1529	return send_and_recv_msgs(drv, msg, nl80211_get_reg, results);
1530}
1531
1532
1533struct hostapd_hw_modes *
1534nl80211_get_hw_feature_data(void *priv, u16 *num_modes, u16 *flags)
1535{
1536	u32 feat;
1537	struct i802_bss *bss = priv;
1538	struct wpa_driver_nl80211_data *drv = bss->drv;
1539	int nl_flags = 0;
1540	struct nl_msg *msg;
1541	struct phy_info_arg result = {
1542		.num_modes = num_modes,
1543		.modes = NULL,
1544		.last_mode = -1,
1545	};
1546
1547	*num_modes = 0;
1548	*flags = 0;
1549
1550	feat = get_nl80211_protocol_features(drv);
1551	if (feat & NL80211_PROTOCOL_FEATURE_SPLIT_WIPHY_DUMP)
1552		nl_flags = NLM_F_DUMP;
1553	if (!(msg = nl80211_cmd_msg(bss, nl_flags, NL80211_CMD_GET_WIPHY)) ||
1554	    nla_put_flag(msg, NL80211_ATTR_SPLIT_WIPHY_DUMP)) {
1555		nlmsg_free(msg);
1556		return NULL;
1557	}
1558
1559	if (send_and_recv_msgs(drv, msg, phy_info_handler, &result) == 0) {
1560		nl80211_set_regulatory_flags(drv, &result);
1561		return wpa_driver_nl80211_postprocess_modes(result.modes,
1562							    num_modes);
1563	}
1564
1565	return NULL;
1566}
1567