xref: /freebsd/contrib/wpa/src/drivers/driver_nl80211_capa.c (revision fa50a3552d1e759e1bb65e54cb0b7e863bcf54d5)
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_common.h"
16 #include "common/wpa_common.h"
17 #include "common/qca-vendor.h"
18 #include "common/qca-vendor-attr.h"
19 #include "driver_nl80211.h"
20 
21 
22 static 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 
38 static 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 
59 struct 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 auth_supported:1;
70 	unsigned int connect_supported:1;
71 	unsigned int p2p_go_supported:1;
72 	unsigned int p2p_client_supported:1;
73 	unsigned int p2p_go_ctwindow_supported:1;
74 	unsigned int p2p_concurrent:1;
75 	unsigned int channel_switch_supported:1;
76 	unsigned int set_qos_map_supported:1;
77 	unsigned int have_low_prio_scan:1;
78 	unsigned int wmm_ac_supported:1;
79 	unsigned int mac_addr_rand_scan_supported:1;
80 	unsigned int mac_addr_rand_sched_scan_supported:1;
81 };
82 
83 
84 static unsigned int probe_resp_offload_support(int supp_protocols)
85 {
86 	unsigned int prot = 0;
87 
88 	if (supp_protocols & NL80211_PROBE_RESP_OFFLOAD_SUPPORT_WPS)
89 		prot |= WPA_DRIVER_PROBE_RESP_OFFLOAD_WPS;
90 	if (supp_protocols & NL80211_PROBE_RESP_OFFLOAD_SUPPORT_WPS2)
91 		prot |= WPA_DRIVER_PROBE_RESP_OFFLOAD_WPS2;
92 	if (supp_protocols & NL80211_PROBE_RESP_OFFLOAD_SUPPORT_P2P)
93 		prot |= WPA_DRIVER_PROBE_RESP_OFFLOAD_P2P;
94 	if (supp_protocols & NL80211_PROBE_RESP_OFFLOAD_SUPPORT_80211U)
95 		prot |= WPA_DRIVER_PROBE_RESP_OFFLOAD_INTERWORKING;
96 
97 	return prot;
98 }
99 
100 
101 static void wiphy_info_supported_iftypes(struct wiphy_info_data *info,
102 					 struct nlattr *tb)
103 {
104 	struct nlattr *nl_mode;
105 	int i;
106 
107 	if (tb == NULL)
108 		return;
109 
110 	nla_for_each_nested(nl_mode, tb, i) {
111 		switch (nla_type(nl_mode)) {
112 		case NL80211_IFTYPE_AP:
113 			info->capa->flags |= WPA_DRIVER_FLAGS_AP;
114 			break;
115 		case NL80211_IFTYPE_MESH_POINT:
116 			info->capa->flags |= WPA_DRIVER_FLAGS_MESH;
117 			break;
118 		case NL80211_IFTYPE_ADHOC:
119 			info->capa->flags |= WPA_DRIVER_FLAGS_IBSS;
120 			break;
121 		case NL80211_IFTYPE_P2P_DEVICE:
122 			info->capa->flags |=
123 				WPA_DRIVER_FLAGS_DEDICATED_P2P_DEVICE;
124 			break;
125 		case NL80211_IFTYPE_P2P_GO:
126 			info->p2p_go_supported = 1;
127 			break;
128 		case NL80211_IFTYPE_P2P_CLIENT:
129 			info->p2p_client_supported = 1;
130 			break;
131 		}
132 	}
133 }
134 
135 
136 static int wiphy_info_iface_comb_process(struct wiphy_info_data *info,
137 					 struct nlattr *nl_combi)
138 {
139 	struct nlattr *tb_comb[NUM_NL80211_IFACE_COMB];
140 	struct nlattr *tb_limit[NUM_NL80211_IFACE_LIMIT];
141 	struct nlattr *nl_limit, *nl_mode;
142 	int err, rem_limit, rem_mode;
143 	int combination_has_p2p = 0, combination_has_mgd = 0;
144 	static struct nla_policy
145 	iface_combination_policy[NUM_NL80211_IFACE_COMB] = {
146 		[NL80211_IFACE_COMB_LIMITS] = { .type = NLA_NESTED },
147 		[NL80211_IFACE_COMB_MAXNUM] = { .type = NLA_U32 },
148 		[NL80211_IFACE_COMB_STA_AP_BI_MATCH] = { .type = NLA_FLAG },
149 		[NL80211_IFACE_COMB_NUM_CHANNELS] = { .type = NLA_U32 },
150 		[NL80211_IFACE_COMB_RADAR_DETECT_WIDTHS] = { .type = NLA_U32 },
151 	},
152 	iface_limit_policy[NUM_NL80211_IFACE_LIMIT] = {
153 		[NL80211_IFACE_LIMIT_TYPES] = { .type = NLA_NESTED },
154 		[NL80211_IFACE_LIMIT_MAX] = { .type = NLA_U32 },
155 	};
156 
157 	err = nla_parse_nested(tb_comb, MAX_NL80211_IFACE_COMB,
158 			       nl_combi, iface_combination_policy);
159 	if (err || !tb_comb[NL80211_IFACE_COMB_LIMITS] ||
160 	    !tb_comb[NL80211_IFACE_COMB_MAXNUM] ||
161 	    !tb_comb[NL80211_IFACE_COMB_NUM_CHANNELS])
162 		return 0; /* broken combination */
163 
164 	if (tb_comb[NL80211_IFACE_COMB_RADAR_DETECT_WIDTHS])
165 		info->capa->flags |= WPA_DRIVER_FLAGS_RADAR;
166 
167 	nla_for_each_nested(nl_limit, tb_comb[NL80211_IFACE_COMB_LIMITS],
168 			    rem_limit) {
169 		err = nla_parse_nested(tb_limit, MAX_NL80211_IFACE_LIMIT,
170 				       nl_limit, iface_limit_policy);
171 		if (err || !tb_limit[NL80211_IFACE_LIMIT_TYPES])
172 			return 0; /* broken combination */
173 
174 		nla_for_each_nested(nl_mode,
175 				    tb_limit[NL80211_IFACE_LIMIT_TYPES],
176 				    rem_mode) {
177 			int ift = nla_type(nl_mode);
178 			if (ift == NL80211_IFTYPE_P2P_GO ||
179 			    ift == NL80211_IFTYPE_P2P_CLIENT)
180 				combination_has_p2p = 1;
181 			if (ift == NL80211_IFTYPE_STATION)
182 				combination_has_mgd = 1;
183 		}
184 		if (combination_has_p2p && combination_has_mgd)
185 			break;
186 	}
187 
188 	if (combination_has_p2p && combination_has_mgd) {
189 		unsigned int num_channels =
190 			nla_get_u32(tb_comb[NL80211_IFACE_COMB_NUM_CHANNELS]);
191 
192 		info->p2p_concurrent = 1;
193 		if (info->num_multichan_concurrent < num_channels)
194 			info->num_multichan_concurrent = num_channels;
195 	}
196 
197 	return 0;
198 }
199 
200 
201 static void wiphy_info_iface_comb(struct wiphy_info_data *info,
202 				  struct nlattr *tb)
203 {
204 	struct nlattr *nl_combi;
205 	int rem_combi;
206 
207 	if (tb == NULL)
208 		return;
209 
210 	nla_for_each_nested(nl_combi, tb, rem_combi) {
211 		if (wiphy_info_iface_comb_process(info, nl_combi) > 0)
212 			break;
213 	}
214 }
215 
216 
217 static void wiphy_info_supp_cmds(struct wiphy_info_data *info,
218 				 struct nlattr *tb)
219 {
220 	struct nlattr *nl_cmd;
221 	int i;
222 
223 	if (tb == NULL)
224 		return;
225 
226 	nla_for_each_nested(nl_cmd, tb, i) {
227 		switch (nla_get_u32(nl_cmd)) {
228 		case NL80211_CMD_AUTHENTICATE:
229 			info->auth_supported = 1;
230 			break;
231 		case NL80211_CMD_CONNECT:
232 			info->connect_supported = 1;
233 			break;
234 		case NL80211_CMD_START_SCHED_SCAN:
235 			info->capa->sched_scan_supported = 1;
236 			break;
237 		case NL80211_CMD_PROBE_CLIENT:
238 			info->poll_command_supported = 1;
239 			break;
240 		case NL80211_CMD_CHANNEL_SWITCH:
241 			info->channel_switch_supported = 1;
242 			break;
243 		case NL80211_CMD_SET_QOS_MAP:
244 			info->set_qos_map_supported = 1;
245 			break;
246 		}
247 	}
248 }
249 
250 
251 static void wiphy_info_cipher_suites(struct wiphy_info_data *info,
252 				     struct nlattr *tb)
253 {
254 	int i, num;
255 	u32 *ciphers;
256 
257 	if (tb == NULL)
258 		return;
259 
260 	num = nla_len(tb) / sizeof(u32);
261 	ciphers = nla_data(tb);
262 	for (i = 0; i < num; i++) {
263 		u32 c = ciphers[i];
264 
265 		wpa_printf(MSG_DEBUG, "nl80211: Supported cipher %02x-%02x-%02x:%d",
266 			   c >> 24, (c >> 16) & 0xff,
267 			   (c >> 8) & 0xff, c & 0xff);
268 		switch (c) {
269 		case RSN_CIPHER_SUITE_CCMP_256:
270 			info->capa->enc |= WPA_DRIVER_CAPA_ENC_CCMP_256;
271 			break;
272 		case RSN_CIPHER_SUITE_GCMP_256:
273 			info->capa->enc |= WPA_DRIVER_CAPA_ENC_GCMP_256;
274 			break;
275 		case RSN_CIPHER_SUITE_CCMP:
276 			info->capa->enc |= WPA_DRIVER_CAPA_ENC_CCMP;
277 			break;
278 		case RSN_CIPHER_SUITE_GCMP:
279 			info->capa->enc |= WPA_DRIVER_CAPA_ENC_GCMP;
280 			break;
281 		case RSN_CIPHER_SUITE_TKIP:
282 			info->capa->enc |= WPA_DRIVER_CAPA_ENC_TKIP;
283 			break;
284 		case RSN_CIPHER_SUITE_WEP104:
285 			info->capa->enc |= WPA_DRIVER_CAPA_ENC_WEP104;
286 			break;
287 		case RSN_CIPHER_SUITE_WEP40:
288 			info->capa->enc |= WPA_DRIVER_CAPA_ENC_WEP40;
289 			break;
290 		case RSN_CIPHER_SUITE_AES_128_CMAC:
291 			info->capa->enc |= WPA_DRIVER_CAPA_ENC_BIP;
292 			break;
293 		case RSN_CIPHER_SUITE_BIP_GMAC_128:
294 			info->capa->enc |= WPA_DRIVER_CAPA_ENC_BIP_GMAC_128;
295 			break;
296 		case RSN_CIPHER_SUITE_BIP_GMAC_256:
297 			info->capa->enc |= WPA_DRIVER_CAPA_ENC_BIP_GMAC_256;
298 			break;
299 		case RSN_CIPHER_SUITE_BIP_CMAC_256:
300 			info->capa->enc |= WPA_DRIVER_CAPA_ENC_BIP_CMAC_256;
301 			break;
302 		case RSN_CIPHER_SUITE_NO_GROUP_ADDRESSED:
303 			info->capa->enc |= WPA_DRIVER_CAPA_ENC_GTK_NOT_USED;
304 			break;
305 		}
306 	}
307 }
308 
309 
310 static void wiphy_info_max_roc(struct wpa_driver_capa *capa,
311 			       struct nlattr *tb)
312 {
313 	if (tb)
314 		capa->max_remain_on_chan = nla_get_u32(tb);
315 }
316 
317 
318 static void wiphy_info_tdls(struct wpa_driver_capa *capa, struct nlattr *tdls,
319 			    struct nlattr *ext_setup)
320 {
321 	if (tdls == NULL)
322 		return;
323 
324 	wpa_printf(MSG_DEBUG, "nl80211: TDLS supported");
325 	capa->flags |= WPA_DRIVER_FLAGS_TDLS_SUPPORT;
326 
327 	if (ext_setup) {
328 		wpa_printf(MSG_DEBUG, "nl80211: TDLS external setup");
329 		capa->flags |= WPA_DRIVER_FLAGS_TDLS_EXTERNAL_SETUP;
330 	}
331 }
332 
333 
334 static int ext_feature_isset(const u8 *ext_features, int ext_features_len,
335 			     enum nl80211_ext_feature_index ftidx)
336 {
337 	u8 ft_byte;
338 
339 	if ((int) ftidx / 8 >= ext_features_len)
340 		return 0;
341 
342 	ft_byte = ext_features[ftidx / 8];
343 	return (ft_byte & BIT(ftidx % 8)) != 0;
344 }
345 
346 
347 static void wiphy_info_ext_feature_flags(struct wiphy_info_data *info,
348 					 struct nlattr *tb)
349 {
350 	struct wpa_driver_capa *capa = info->capa;
351 	u8 *ext_features;
352 	int len;
353 
354 	if (tb == NULL)
355 		return;
356 
357 	ext_features = nla_data(tb);
358 	len = nla_len(tb);
359 
360 	if (ext_feature_isset(ext_features, len, NL80211_EXT_FEATURE_VHT_IBSS))
361 		capa->flags |= WPA_DRIVER_FLAGS_VHT_IBSS;
362 
363 	if (ext_feature_isset(ext_features, len, NL80211_EXT_FEATURE_RRM))
364 		capa->rrm_flags |= WPA_DRIVER_FLAGS_SUPPORT_RRM;
365 
366 	if (ext_feature_isset(ext_features, len, NL80211_EXT_FEATURE_FILS_STA))
367 		capa->flags |= WPA_DRIVER_FLAGS_SUPPORT_FILS;
368 
369 	if (ext_feature_isset(ext_features, len,
370 			      NL80211_EXT_FEATURE_BEACON_RATE_LEGACY))
371 		capa->flags |= WPA_DRIVER_FLAGS_BEACON_RATE_LEGACY;
372 
373 	if (ext_feature_isset(ext_features, len,
374 			      NL80211_EXT_FEATURE_BEACON_RATE_HT))
375 		capa->flags |= WPA_DRIVER_FLAGS_BEACON_RATE_HT;
376 
377 	if (ext_feature_isset(ext_features, len,
378 			      NL80211_EXT_FEATURE_BEACON_RATE_VHT))
379 		capa->flags |= WPA_DRIVER_FLAGS_BEACON_RATE_VHT;
380 
381 	if (ext_feature_isset(ext_features, len,
382 			      NL80211_EXT_FEATURE_SET_SCAN_DWELL))
383 		capa->rrm_flags |= WPA_DRIVER_FLAGS_SUPPORT_SET_SCAN_DWELL;
384 
385 	if (ext_feature_isset(ext_features, len,
386 			      NL80211_EXT_FEATURE_SCAN_START_TIME) &&
387 	    ext_feature_isset(ext_features, len,
388 			      NL80211_EXT_FEATURE_BSS_PARENT_TSF) &&
389 	    ext_feature_isset(ext_features, len,
390 			      NL80211_EXT_FEATURE_SET_SCAN_DWELL))
391 		capa->rrm_flags |= WPA_DRIVER_FLAGS_SUPPORT_BEACON_REPORT;
392 	if (ext_feature_isset(ext_features, len,
393 			      NL80211_EXT_FEATURE_MGMT_TX_RANDOM_TA))
394 		capa->flags |= WPA_DRIVER_FLAGS_MGMT_TX_RANDOM_TA;
395 	if (ext_feature_isset(ext_features, len,
396 			      NL80211_EXT_FEATURE_MGMT_TX_RANDOM_TA_CONNECTED))
397 		capa->flags |= WPA_DRIVER_FLAGS_MGMT_TX_RANDOM_TA_CONNECTED;
398 	if (ext_feature_isset(ext_features, len,
399 			      NL80211_EXT_FEATURE_SCHED_SCAN_RELATIVE_RSSI))
400 		capa->flags |= WPA_DRIVER_FLAGS_SCHED_SCAN_RELATIVE_RSSI;
401 	if (ext_feature_isset(ext_features, len,
402 			      NL80211_EXT_FEATURE_FILS_SK_OFFLOAD))
403 		capa->flags |= WPA_DRIVER_FLAGS_FILS_SK_OFFLOAD;
404 
405 	if (ext_feature_isset(ext_features, len,
406 			      NL80211_EXT_FEATURE_4WAY_HANDSHAKE_STA_PSK) &&
407 	    ext_feature_isset(ext_features, len,
408 			      NL80211_EXT_FEATURE_4WAY_HANDSHAKE_STA_1X))
409 		capa->flags |= WPA_DRIVER_FLAGS_4WAY_HANDSHAKE;
410 
411 	if (ext_feature_isset(ext_features, len,
412 			      NL80211_EXT_FEATURE_MFP_OPTIONAL))
413 		capa->flags |= WPA_DRIVER_FLAGS_MFP_OPTIONAL;
414 
415 	if (ext_feature_isset(ext_features, len,
416 			      NL80211_EXT_FEATURE_DFS_OFFLOAD))
417 		capa->flags |= WPA_DRIVER_FLAGS_DFS_OFFLOAD;
418 
419 #ifdef CONFIG_MBO
420 	if (ext_feature_isset(ext_features, len,
421 			      NL80211_EXT_FEATURE_FILS_MAX_CHANNEL_TIME) &&
422 	    ext_feature_isset(ext_features, len,
423 			      NL80211_EXT_FEATURE_ACCEPT_BCAST_PROBE_RESP) &&
424 	    ext_feature_isset(ext_features, len,
425 			      NL80211_EXT_FEATURE_OCE_PROBE_REQ_HIGH_TX_RATE) &&
426 	    ext_feature_isset(
427 		    ext_features, len,
428 		    NL80211_EXT_FEATURE_OCE_PROBE_REQ_DEFERRAL_SUPPRESSION))
429 		capa->flags |= WPA_DRIVER_FLAGS_OCE_STA;
430 #endif /* CONFIG_MBO */
431 }
432 
433 
434 static void wiphy_info_feature_flags(struct wiphy_info_data *info,
435 				     struct nlattr *tb)
436 {
437 	u32 flags;
438 	struct wpa_driver_capa *capa = info->capa;
439 
440 	if (tb == NULL)
441 		return;
442 
443 	flags = nla_get_u32(tb);
444 
445 	if (flags & NL80211_FEATURE_SK_TX_STATUS)
446 		info->data_tx_status = 1;
447 
448 	if (flags & NL80211_FEATURE_INACTIVITY_TIMER)
449 		capa->flags |= WPA_DRIVER_FLAGS_INACTIVITY_TIMER;
450 
451 	if (flags & NL80211_FEATURE_SAE)
452 		capa->flags |= WPA_DRIVER_FLAGS_SAE;
453 
454 	if (flags & NL80211_FEATURE_NEED_OBSS_SCAN)
455 		capa->flags |= WPA_DRIVER_FLAGS_OBSS_SCAN;
456 
457 	if (flags & NL80211_FEATURE_AP_MODE_CHAN_WIDTH_CHANGE)
458 		capa->flags |= WPA_DRIVER_FLAGS_HT_2040_COEX;
459 
460 	if (flags & NL80211_FEATURE_TDLS_CHANNEL_SWITCH) {
461 		wpa_printf(MSG_DEBUG, "nl80211: TDLS channel switch");
462 		capa->flags |= WPA_DRIVER_FLAGS_TDLS_CHANNEL_SWITCH;
463 	}
464 
465 	if (flags & NL80211_FEATURE_P2P_GO_CTWIN)
466 		info->p2p_go_ctwindow_supported = 1;
467 
468 	if (flags & NL80211_FEATURE_LOW_PRIORITY_SCAN)
469 		info->have_low_prio_scan = 1;
470 
471 	if (flags & NL80211_FEATURE_SCAN_RANDOM_MAC_ADDR)
472 		info->mac_addr_rand_scan_supported = 1;
473 
474 	if (flags & NL80211_FEATURE_SCHED_SCAN_RANDOM_MAC_ADDR)
475 		info->mac_addr_rand_sched_scan_supported = 1;
476 
477 	if (flags & NL80211_FEATURE_STATIC_SMPS)
478 		capa->smps_modes |= WPA_DRIVER_SMPS_MODE_STATIC;
479 
480 	if (flags & NL80211_FEATURE_DYNAMIC_SMPS)
481 		capa->smps_modes |= WPA_DRIVER_SMPS_MODE_DYNAMIC;
482 
483 	if (flags & NL80211_FEATURE_SUPPORTS_WMM_ADMISSION)
484 		info->wmm_ac_supported = 1;
485 
486 	if (flags & NL80211_FEATURE_DS_PARAM_SET_IE_IN_PROBES)
487 		capa->rrm_flags |= WPA_DRIVER_FLAGS_DS_PARAM_SET_IE_IN_PROBES;
488 
489 	if (flags & NL80211_FEATURE_WFA_TPC_IE_IN_PROBES)
490 		capa->rrm_flags |= WPA_DRIVER_FLAGS_WFA_TPC_IE_IN_PROBES;
491 
492 	if (flags & NL80211_FEATURE_QUIET)
493 		capa->rrm_flags |= WPA_DRIVER_FLAGS_QUIET;
494 
495 	if (flags & NL80211_FEATURE_TX_POWER_INSERTION)
496 		capa->rrm_flags |= WPA_DRIVER_FLAGS_TX_POWER_INSERTION;
497 
498 	if (flags & NL80211_FEATURE_HT_IBSS)
499 		capa->flags |= WPA_DRIVER_FLAGS_HT_IBSS;
500 
501 	if (flags & NL80211_FEATURE_FULL_AP_CLIENT_STATE)
502 		capa->flags |= WPA_DRIVER_FLAGS_FULL_AP_CLIENT_STATE;
503 }
504 
505 
506 static void wiphy_info_probe_resp_offload(struct wpa_driver_capa *capa,
507 					  struct nlattr *tb)
508 {
509 	u32 protocols;
510 
511 	if (tb == NULL)
512 		return;
513 
514 	protocols = nla_get_u32(tb);
515 	wpa_printf(MSG_DEBUG, "nl80211: Supports Probe Response offload in AP "
516 		   "mode");
517 	capa->flags |= WPA_DRIVER_FLAGS_PROBE_RESP_OFFLOAD;
518 	capa->probe_resp_offloads = probe_resp_offload_support(protocols);
519 }
520 
521 
522 static void wiphy_info_wowlan_triggers(struct wpa_driver_capa *capa,
523 				       struct nlattr *tb)
524 {
525 	struct nlattr *triggers[MAX_NL80211_WOWLAN_TRIG + 1];
526 
527 	if (tb == NULL)
528 		return;
529 
530 	if (nla_parse_nested(triggers, MAX_NL80211_WOWLAN_TRIG,
531 			     tb, NULL))
532 		return;
533 
534 	if (triggers[NL80211_WOWLAN_TRIG_ANY])
535 		capa->wowlan_triggers.any = 1;
536 	if (triggers[NL80211_WOWLAN_TRIG_DISCONNECT])
537 		capa->wowlan_triggers.disconnect = 1;
538 	if (triggers[NL80211_WOWLAN_TRIG_MAGIC_PKT])
539 		capa->wowlan_triggers.magic_pkt = 1;
540 	if (triggers[NL80211_WOWLAN_TRIG_GTK_REKEY_FAILURE])
541 		capa->wowlan_triggers.gtk_rekey_failure = 1;
542 	if (triggers[NL80211_WOWLAN_TRIG_EAP_IDENT_REQUEST])
543 		capa->wowlan_triggers.eap_identity_req = 1;
544 	if (triggers[NL80211_WOWLAN_TRIG_4WAY_HANDSHAKE])
545 		capa->wowlan_triggers.four_way_handshake = 1;
546 	if (triggers[NL80211_WOWLAN_TRIG_RFKILL_RELEASE])
547 		capa->wowlan_triggers.rfkill_release = 1;
548 }
549 
550 
551 static void wiphy_info_extended_capab(struct wpa_driver_nl80211_data *drv,
552 				      struct nlattr *tb)
553 {
554 	int rem = 0, i;
555 	struct nlattr *tb1[NL80211_ATTR_MAX + 1], *attr;
556 
557 	if (!tb || drv->num_iface_ext_capa == NL80211_IFTYPE_MAX)
558 		return;
559 
560 	nla_for_each_nested(attr, tb, rem) {
561 		unsigned int len;
562 		struct drv_nl80211_ext_capa *capa;
563 
564 		nla_parse(tb1, NL80211_ATTR_MAX, nla_data(attr),
565 			  nla_len(attr), NULL);
566 
567 		if (!tb1[NL80211_ATTR_IFTYPE] ||
568 		    !tb1[NL80211_ATTR_EXT_CAPA] ||
569 		    !tb1[NL80211_ATTR_EXT_CAPA_MASK])
570 			continue;
571 
572 		capa = &drv->iface_ext_capa[drv->num_iface_ext_capa];
573 		capa->iftype = nla_get_u32(tb1[NL80211_ATTR_IFTYPE]);
574 		wpa_printf(MSG_DEBUG,
575 			   "nl80211: Driver-advertised extended capabilities for interface type %s",
576 			   nl80211_iftype_str(capa->iftype));
577 
578 		len = nla_len(tb1[NL80211_ATTR_EXT_CAPA]);
579 		capa->ext_capa = os_memdup(nla_data(tb1[NL80211_ATTR_EXT_CAPA]),
580 					   len);
581 		if (!capa->ext_capa)
582 			goto err;
583 
584 		capa->ext_capa_len = len;
585 		wpa_hexdump(MSG_DEBUG, "nl80211: Extended capabilities",
586 			    capa->ext_capa, capa->ext_capa_len);
587 
588 		len = nla_len(tb1[NL80211_ATTR_EXT_CAPA_MASK]);
589 		capa->ext_capa_mask =
590 			os_memdup(nla_data(tb1[NL80211_ATTR_EXT_CAPA_MASK]),
591 				  len);
592 		if (!capa->ext_capa_mask)
593 			goto err;
594 
595 		wpa_hexdump(MSG_DEBUG, "nl80211: Extended capabilities mask",
596 			    capa->ext_capa_mask, capa->ext_capa_len);
597 
598 		drv->num_iface_ext_capa++;
599 		if (drv->num_iface_ext_capa == NL80211_IFTYPE_MAX)
600 			break;
601 	}
602 
603 	return;
604 
605 err:
606 	/* Cleanup allocated memory on error */
607 	for (i = 0; i < NL80211_IFTYPE_MAX; i++) {
608 		os_free(drv->iface_ext_capa[i].ext_capa);
609 		drv->iface_ext_capa[i].ext_capa = NULL;
610 		os_free(drv->iface_ext_capa[i].ext_capa_mask);
611 		drv->iface_ext_capa[i].ext_capa_mask = NULL;
612 		drv->iface_ext_capa[i].ext_capa_len = 0;
613 	}
614 	drv->num_iface_ext_capa = 0;
615 }
616 
617 
618 static int wiphy_info_handler(struct nl_msg *msg, void *arg)
619 {
620 	struct nlattr *tb[NL80211_ATTR_MAX + 1];
621 	struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
622 	struct wiphy_info_data *info = arg;
623 	struct wpa_driver_capa *capa = info->capa;
624 	struct wpa_driver_nl80211_data *drv = info->drv;
625 
626 	nla_parse(tb, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
627 		  genlmsg_attrlen(gnlh, 0), NULL);
628 
629 	if (tb[NL80211_ATTR_WIPHY])
630 		drv->wiphy_idx = nla_get_u32(tb[NL80211_ATTR_WIPHY]);
631 
632 	if (tb[NL80211_ATTR_WIPHY_NAME])
633 		os_strlcpy(drv->phyname,
634 			   nla_get_string(tb[NL80211_ATTR_WIPHY_NAME]),
635 			   sizeof(drv->phyname));
636 	if (tb[NL80211_ATTR_MAX_NUM_SCAN_SSIDS])
637 		capa->max_scan_ssids =
638 			nla_get_u8(tb[NL80211_ATTR_MAX_NUM_SCAN_SSIDS]);
639 
640 	if (tb[NL80211_ATTR_MAX_NUM_SCHED_SCAN_SSIDS])
641 		capa->max_sched_scan_ssids =
642 			nla_get_u8(tb[NL80211_ATTR_MAX_NUM_SCHED_SCAN_SSIDS]);
643 
644 	if (tb[NL80211_ATTR_MAX_NUM_SCHED_SCAN_PLANS] &&
645 	    tb[NL80211_ATTR_MAX_SCAN_PLAN_INTERVAL] &&
646 	    tb[NL80211_ATTR_MAX_SCAN_PLAN_ITERATIONS]) {
647 		capa->max_sched_scan_plans =
648 			nla_get_u32(tb[NL80211_ATTR_MAX_NUM_SCHED_SCAN_PLANS]);
649 
650 		capa->max_sched_scan_plan_interval =
651 			nla_get_u32(tb[NL80211_ATTR_MAX_SCAN_PLAN_INTERVAL]);
652 
653 		capa->max_sched_scan_plan_iterations =
654 			nla_get_u32(tb[NL80211_ATTR_MAX_SCAN_PLAN_ITERATIONS]);
655 	}
656 
657 	if (tb[NL80211_ATTR_MAX_MATCH_SETS])
658 		capa->max_match_sets =
659 			nla_get_u8(tb[NL80211_ATTR_MAX_MATCH_SETS]);
660 
661 	if (tb[NL80211_ATTR_MAC_ACL_MAX])
662 		capa->max_acl_mac_addrs =
663 			nla_get_u8(tb[NL80211_ATTR_MAC_ACL_MAX]);
664 
665 	wiphy_info_supported_iftypes(info, tb[NL80211_ATTR_SUPPORTED_IFTYPES]);
666 	wiphy_info_iface_comb(info, tb[NL80211_ATTR_INTERFACE_COMBINATIONS]);
667 	wiphy_info_supp_cmds(info, tb[NL80211_ATTR_SUPPORTED_COMMANDS]);
668 	wiphy_info_cipher_suites(info, tb[NL80211_ATTR_CIPHER_SUITES]);
669 
670 	if (tb[NL80211_ATTR_OFFCHANNEL_TX_OK]) {
671 		wpa_printf(MSG_DEBUG, "nl80211: Using driver-based "
672 			   "off-channel TX");
673 		capa->flags |= WPA_DRIVER_FLAGS_OFFCHANNEL_TX;
674 	}
675 
676 	if (tb[NL80211_ATTR_ROAM_SUPPORT]) {
677 		wpa_printf(MSG_DEBUG, "nl80211: Using driver-based roaming");
678 		capa->flags |= WPA_DRIVER_FLAGS_BSS_SELECTION;
679 	}
680 
681 	wiphy_info_max_roc(capa,
682 			   tb[NL80211_ATTR_MAX_REMAIN_ON_CHANNEL_DURATION]);
683 
684 	if (tb[NL80211_ATTR_SUPPORT_AP_UAPSD])
685 		capa->flags |= WPA_DRIVER_FLAGS_AP_UAPSD;
686 
687 	wiphy_info_tdls(capa, tb[NL80211_ATTR_TDLS_SUPPORT],
688 			tb[NL80211_ATTR_TDLS_EXTERNAL_SETUP]);
689 
690 	if (tb[NL80211_ATTR_DEVICE_AP_SME])
691 		info->device_ap_sme = 1;
692 
693 	wiphy_info_feature_flags(info, tb[NL80211_ATTR_FEATURE_FLAGS]);
694 	wiphy_info_ext_feature_flags(info, tb[NL80211_ATTR_EXT_FEATURES]);
695 	wiphy_info_probe_resp_offload(capa,
696 				      tb[NL80211_ATTR_PROBE_RESP_OFFLOAD]);
697 
698 	if (tb[NL80211_ATTR_EXT_CAPA] && tb[NL80211_ATTR_EXT_CAPA_MASK] &&
699 	    drv->extended_capa == NULL) {
700 		drv->extended_capa =
701 			os_malloc(nla_len(tb[NL80211_ATTR_EXT_CAPA]));
702 		if (drv->extended_capa) {
703 			os_memcpy(drv->extended_capa,
704 				  nla_data(tb[NL80211_ATTR_EXT_CAPA]),
705 				  nla_len(tb[NL80211_ATTR_EXT_CAPA]));
706 			drv->extended_capa_len =
707 				nla_len(tb[NL80211_ATTR_EXT_CAPA]);
708 			wpa_hexdump(MSG_DEBUG,
709 				    "nl80211: Driver-advertised extended capabilities (default)",
710 				    drv->extended_capa, drv->extended_capa_len);
711 		}
712 		drv->extended_capa_mask =
713 			os_malloc(nla_len(tb[NL80211_ATTR_EXT_CAPA_MASK]));
714 		if (drv->extended_capa_mask) {
715 			os_memcpy(drv->extended_capa_mask,
716 				  nla_data(tb[NL80211_ATTR_EXT_CAPA_MASK]),
717 				  nla_len(tb[NL80211_ATTR_EXT_CAPA_MASK]));
718 			wpa_hexdump(MSG_DEBUG,
719 				    "nl80211: Driver-advertised extended capabilities mask (default)",
720 				    drv->extended_capa_mask,
721 				    drv->extended_capa_len);
722 		} else {
723 			os_free(drv->extended_capa);
724 			drv->extended_capa = NULL;
725 			drv->extended_capa_len = 0;
726 		}
727 	}
728 
729 	wiphy_info_extended_capab(drv, tb[NL80211_ATTR_IFTYPE_EXT_CAPA]);
730 
731 	if (tb[NL80211_ATTR_VENDOR_DATA]) {
732 		struct nlattr *nl;
733 		int rem;
734 
735 		nla_for_each_nested(nl, tb[NL80211_ATTR_VENDOR_DATA], rem) {
736 			struct nl80211_vendor_cmd_info *vinfo;
737 			if (nla_len(nl) != sizeof(*vinfo)) {
738 				wpa_printf(MSG_DEBUG, "nl80211: Unexpected vendor data info");
739 				continue;
740 			}
741 			vinfo = nla_data(nl);
742 			if (vinfo->vendor_id == OUI_QCA) {
743 				switch (vinfo->subcmd) {
744 				case QCA_NL80211_VENDOR_SUBCMD_TEST:
745 					drv->vendor_cmd_test_avail = 1;
746 					break;
747 #ifdef CONFIG_DRIVER_NL80211_QCA
748 				case QCA_NL80211_VENDOR_SUBCMD_ROAMING:
749 					drv->roaming_vendor_cmd_avail = 1;
750 					break;
751 				case QCA_NL80211_VENDOR_SUBCMD_DFS_CAPABILITY:
752 					drv->dfs_vendor_cmd_avail = 1;
753 					break;
754 				case QCA_NL80211_VENDOR_SUBCMD_GET_FEATURES:
755 					drv->get_features_vendor_cmd_avail = 1;
756 					break;
757 				case QCA_NL80211_VENDOR_SUBCMD_GET_PREFERRED_FREQ_LIST:
758 					drv->get_pref_freq_list = 1;
759 					break;
760 				case QCA_NL80211_VENDOR_SUBCMD_SET_PROBABLE_OPER_CHANNEL:
761 					drv->set_prob_oper_freq = 1;
762 					break;
763 				case QCA_NL80211_VENDOR_SUBCMD_DO_ACS:
764 					drv->capa.flags |=
765 						WPA_DRIVER_FLAGS_ACS_OFFLOAD;
766 					break;
767 				case QCA_NL80211_VENDOR_SUBCMD_SETBAND:
768 					drv->setband_vendor_cmd_avail = 1;
769 					break;
770 				case QCA_NL80211_VENDOR_SUBCMD_TRIGGER_SCAN:
771 					drv->scan_vendor_cmd_avail = 1;
772 					break;
773 				case QCA_NL80211_VENDOR_SUBCMD_SET_WIFI_CONFIGURATION:
774 					drv->set_wifi_conf_vendor_cmd_avail = 1;
775 					break;
776 				case QCA_NL80211_VENDOR_SUBCMD_GET_HE_CAPABILITIES:
777 					drv->he_capab_vendor_cmd_avail = 1;
778 					break;
779 				case QCA_NL80211_VENDOR_SUBCMD_FETCH_BSS_TRANSITION_STATUS:
780 					drv->fetch_bss_trans_status = 1;
781 					break;
782 				case QCA_NL80211_VENDOR_SUBCMD_ROAM:
783 					drv->roam_vendor_cmd_avail = 1;
784 					break;
785 #endif /* CONFIG_DRIVER_NL80211_QCA */
786 				}
787 			}
788 
789 			wpa_printf(MSG_DEBUG, "nl80211: Supported vendor command: vendor_id=0x%x subcmd=%u",
790 				   vinfo->vendor_id, vinfo->subcmd);
791 		}
792 	}
793 
794 	if (tb[NL80211_ATTR_VENDOR_EVENTS]) {
795 		struct nlattr *nl;
796 		int rem;
797 
798 		nla_for_each_nested(nl, tb[NL80211_ATTR_VENDOR_EVENTS], rem) {
799 			struct nl80211_vendor_cmd_info *vinfo;
800 			if (nla_len(nl) != sizeof(*vinfo)) {
801 				wpa_printf(MSG_DEBUG, "nl80211: Unexpected vendor data info");
802 				continue;
803 			}
804 			vinfo = nla_data(nl);
805 			wpa_printf(MSG_DEBUG, "nl80211: Supported vendor event: vendor_id=0x%x subcmd=%u",
806 				   vinfo->vendor_id, vinfo->subcmd);
807 		}
808 	}
809 
810 	wiphy_info_wowlan_triggers(capa,
811 				   tb[NL80211_ATTR_WOWLAN_TRIGGERS_SUPPORTED]);
812 
813 	if (tb[NL80211_ATTR_MAX_AP_ASSOC_STA])
814 		capa->max_stations =
815 			nla_get_u32(tb[NL80211_ATTR_MAX_AP_ASSOC_STA]);
816 
817 	if (tb[NL80211_ATTR_MAX_CSA_COUNTERS])
818 		capa->max_csa_counters =
819 			nla_get_u8(tb[NL80211_ATTR_MAX_CSA_COUNTERS]);
820 
821 	if (tb[NL80211_ATTR_WIPHY_SELF_MANAGED_REG])
822 		capa->flags |= WPA_DRIVER_FLAGS_SELF_MANAGED_REGULATORY;
823 
824 	return NL_SKIP;
825 }
826 
827 
828 static int wpa_driver_nl80211_get_info(struct wpa_driver_nl80211_data *drv,
829 				       struct wiphy_info_data *info)
830 {
831 	u32 feat;
832 	struct nl_msg *msg;
833 	int flags = 0;
834 
835 	os_memset(info, 0, sizeof(*info));
836 	info->capa = &drv->capa;
837 	info->drv = drv;
838 
839 	feat = get_nl80211_protocol_features(drv);
840 	if (feat & NL80211_PROTOCOL_FEATURE_SPLIT_WIPHY_DUMP)
841 		flags = NLM_F_DUMP;
842 	msg = nl80211_cmd_msg(drv->first_bss, flags, NL80211_CMD_GET_WIPHY);
843 	if (!msg || nla_put_flag(msg, NL80211_ATTR_SPLIT_WIPHY_DUMP)) {
844 		nlmsg_free(msg);
845 		return -1;
846 	}
847 
848 	if (send_and_recv_msgs(drv, msg, wiphy_info_handler, info))
849 		return -1;
850 
851 	if (info->auth_supported)
852 		drv->capa.flags |= WPA_DRIVER_FLAGS_SME;
853 	else if (!info->connect_supported) {
854 		wpa_printf(MSG_INFO, "nl80211: Driver does not support "
855 			   "authentication/association or connect commands");
856 		info->error = 1;
857 	}
858 
859 	if (info->p2p_go_supported && info->p2p_client_supported)
860 		drv->capa.flags |= WPA_DRIVER_FLAGS_P2P_CAPABLE;
861 	if (info->p2p_concurrent) {
862 		wpa_printf(MSG_DEBUG, "nl80211: Use separate P2P group "
863 			   "interface (driver advertised support)");
864 		drv->capa.flags |= WPA_DRIVER_FLAGS_P2P_CONCURRENT;
865 		drv->capa.flags |= WPA_DRIVER_FLAGS_P2P_MGMT_AND_NON_P2P;
866 	}
867 	if (info->num_multichan_concurrent > 1) {
868 		wpa_printf(MSG_DEBUG, "nl80211: Enable multi-channel "
869 			   "concurrent (driver advertised support)");
870 		drv->capa.num_multichan_concurrent =
871 			info->num_multichan_concurrent;
872 	}
873 	if (drv->capa.flags & WPA_DRIVER_FLAGS_DEDICATED_P2P_DEVICE)
874 		wpa_printf(MSG_DEBUG, "nl80211: use P2P_DEVICE support");
875 
876 	/* default to 5000 since early versions of mac80211 don't set it */
877 	if (!drv->capa.max_remain_on_chan)
878 		drv->capa.max_remain_on_chan = 5000;
879 
880 	drv->capa.wmm_ac_supported = info->wmm_ac_supported;
881 
882 	drv->capa.mac_addr_rand_sched_scan_supported =
883 		info->mac_addr_rand_sched_scan_supported;
884 	drv->capa.mac_addr_rand_scan_supported =
885 		info->mac_addr_rand_scan_supported;
886 
887 	if (info->channel_switch_supported) {
888 		drv->capa.flags |= WPA_DRIVER_FLAGS_AP_CSA;
889 		if (!drv->capa.max_csa_counters)
890 			drv->capa.max_csa_counters = 1;
891 	}
892 
893 	if (!drv->capa.max_sched_scan_plans) {
894 		drv->capa.max_sched_scan_plans = 1;
895 		drv->capa.max_sched_scan_plan_interval = UINT32_MAX;
896 		drv->capa.max_sched_scan_plan_iterations = 0;
897 	}
898 
899 	return 0;
900 }
901 
902 
903 #ifdef CONFIG_DRIVER_NL80211_QCA
904 
905 static int dfs_info_handler(struct nl_msg *msg, void *arg)
906 {
907 	struct nlattr *tb[NL80211_ATTR_MAX + 1];
908 	struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
909 	int *dfs_capability_ptr = arg;
910 
911 	nla_parse(tb, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
912 		  genlmsg_attrlen(gnlh, 0), NULL);
913 
914 	if (tb[NL80211_ATTR_VENDOR_DATA]) {
915 		struct nlattr *nl_vend = tb[NL80211_ATTR_VENDOR_DATA];
916 		struct nlattr *tb_vendor[QCA_WLAN_VENDOR_ATTR_MAX + 1];
917 
918 		nla_parse(tb_vendor, QCA_WLAN_VENDOR_ATTR_MAX,
919 			  nla_data(nl_vend), nla_len(nl_vend), NULL);
920 
921 		if (tb_vendor[QCA_WLAN_VENDOR_ATTR_DFS]) {
922 			u32 val;
923 			val = nla_get_u32(tb_vendor[QCA_WLAN_VENDOR_ATTR_DFS]);
924 			wpa_printf(MSG_DEBUG, "nl80211: DFS offload capability: %u",
925 				   val);
926 			*dfs_capability_ptr = val;
927 		}
928 	}
929 
930 	return NL_SKIP;
931 }
932 
933 
934 static void qca_nl80211_check_dfs_capa(struct wpa_driver_nl80211_data *drv)
935 {
936 	struct nl_msg *msg;
937 	int dfs_capability = 0;
938 	int ret;
939 
940 	if (!drv->dfs_vendor_cmd_avail)
941 		return;
942 
943 	if (!(msg = nl80211_drv_msg(drv, 0, NL80211_CMD_VENDOR)) ||
944 	    nla_put_u32(msg, NL80211_ATTR_VENDOR_ID, OUI_QCA) ||
945 	    nla_put_u32(msg, NL80211_ATTR_VENDOR_SUBCMD,
946 			QCA_NL80211_VENDOR_SUBCMD_DFS_CAPABILITY)) {
947 		nlmsg_free(msg);
948 		return;
949 	}
950 
951 	ret = send_and_recv_msgs(drv, msg, dfs_info_handler, &dfs_capability);
952 	if (!ret && dfs_capability)
953 		drv->capa.flags |= WPA_DRIVER_FLAGS_DFS_OFFLOAD;
954 }
955 
956 
957 static int qca_nl80211_he_capab_handler(struct nl_msg *msg, void *arg)
958 {
959 	struct nlattr *tb[NL80211_ATTR_MAX + 1];
960 	struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
961 	struct he_capabilities *he_capab = arg;
962 	struct nlattr *nl_vend;
963 	struct nlattr *tb_vendor[QCA_WLAN_VENDOR_ATTR_HE_CAPABILITIES_MAX + 1];
964 	size_t len;
965 
966 	nla_parse(tb, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
967 		  genlmsg_attrlen(gnlh, 0), NULL);
968 
969 	if (!tb[NL80211_ATTR_VENDOR_DATA])
970 		return NL_SKIP;
971 
972 	nl_vend = tb[NL80211_ATTR_VENDOR_DATA];
973 	nla_parse(tb_vendor, QCA_WLAN_VENDOR_ATTR_HE_CAPABILITIES_MAX,
974 		  nla_data(nl_vend), nla_len(nl_vend), NULL);
975 
976 	if (tb_vendor[QCA_WLAN_VENDOR_ATTR_HE_SUPPORTED]) {
977 		u8 he_supported;
978 
979 		he_supported = nla_get_u8(
980 			tb_vendor[QCA_WLAN_VENDOR_ATTR_HE_SUPPORTED]);
981 		wpa_printf(MSG_DEBUG, "nl80211: HE capabilities supported: %u",
982 			   he_supported);
983 		he_capab->he_supported = he_supported;
984 		if (!he_supported)
985 			return NL_SKIP;
986 	}
987 
988 	if (tb_vendor[QCA_WLAN_VENDOR_ATTR_PHY_CAPAB]) {
989 		len = nla_len(tb_vendor[QCA_WLAN_VENDOR_ATTR_PHY_CAPAB]);
990 
991 		if (len > sizeof(he_capab->phy_cap))
992 			len = sizeof(he_capab->phy_cap);
993 		os_memcpy(he_capab->phy_cap,
994 			  nla_data(tb_vendor[QCA_WLAN_VENDOR_ATTR_PHY_CAPAB]),
995 			  len);
996 	}
997 
998 	if (tb_vendor[QCA_WLAN_VENDOR_ATTR_MAC_CAPAB])
999 		he_capab->mac_cap =
1000 			nla_get_u32(tb_vendor[QCA_WLAN_VENDOR_ATTR_MAC_CAPAB]);
1001 
1002 	if (tb_vendor[QCA_WLAN_VENDOR_ATTR_HE_MCS])
1003 		he_capab->mcs =
1004 			nla_get_u32(tb_vendor[QCA_WLAN_VENDOR_ATTR_HE_MCS]);
1005 
1006 	if (tb_vendor[QCA_WLAN_VENDOR_ATTR_NUM_SS])
1007 		he_capab->ppet.numss_m1 =
1008 			nla_get_u32(tb_vendor[QCA_WLAN_VENDOR_ATTR_NUM_SS]);
1009 
1010 	if (tb_vendor[QCA_WLAN_VENDOR_ATTR_RU_IDX_MASK])
1011 		he_capab->ppet.ru_count =
1012 			nla_get_u32(tb_vendor[QCA_WLAN_VENDOR_ATTR_RU_IDX_MASK]);
1013 
1014 	if (tb_vendor[QCA_WLAN_VENDOR_ATTR_PPE_THRESHOLD]) {
1015 		len = nla_len(tb_vendor[QCA_WLAN_VENDOR_ATTR_PPE_THRESHOLD]);
1016 
1017 		if (len > sizeof(he_capab->ppet.ppet16_ppet8_ru3_ru0))
1018 			len = sizeof(he_capab->ppet.ppet16_ppet8_ru3_ru0);
1019 		os_memcpy(he_capab->ppet.ppet16_ppet8_ru3_ru0,
1020 			  nla_data(tb_vendor[QCA_WLAN_VENDOR_ATTR_PPE_THRESHOLD]),
1021 			  len);
1022 	}
1023 
1024 	return NL_SKIP;
1025 }
1026 
1027 
1028 static void qca_nl80211_check_he_capab(struct wpa_driver_nl80211_data *drv)
1029 {
1030 	struct nl_msg *msg;
1031 	int ret;
1032 
1033 	if (!drv->he_capab_vendor_cmd_avail)
1034 		return;
1035 
1036 	if (!(msg = nl80211_drv_msg(drv, 0, NL80211_CMD_VENDOR)) ||
1037 		nla_put_u32(msg, NL80211_ATTR_VENDOR_ID, OUI_QCA) ||
1038 		nla_put_u32(msg, NL80211_ATTR_VENDOR_SUBCMD,
1039 			    QCA_NL80211_VENDOR_SUBCMD_GET_HE_CAPABILITIES)) {
1040 		nlmsg_free(msg);
1041 		return;
1042 	}
1043 
1044 	ret = send_and_recv_msgs(drv, msg, qca_nl80211_he_capab_handler,
1045 				 &drv->he_capab);
1046 	if (!ret && drv->he_capab.he_supported)
1047 		drv->capa.flags |= WPA_DRIVER_FLAGS_HE_CAPABILITIES;
1048 }
1049 
1050 
1051 struct features_info {
1052 	u8 *flags;
1053 	size_t flags_len;
1054 	struct wpa_driver_capa *capa;
1055 };
1056 
1057 
1058 static int features_info_handler(struct nl_msg *msg, void *arg)
1059 {
1060 	struct nlattr *tb[NL80211_ATTR_MAX + 1];
1061 	struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
1062 	struct features_info *info = arg;
1063 	struct nlattr *nl_vend, *attr;
1064 
1065 	nla_parse(tb, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
1066 		  genlmsg_attrlen(gnlh, 0), NULL);
1067 
1068 	nl_vend = tb[NL80211_ATTR_VENDOR_DATA];
1069 	if (nl_vend) {
1070 		struct nlattr *tb_vendor[QCA_WLAN_VENDOR_ATTR_MAX + 1];
1071 
1072 		nla_parse(tb_vendor, QCA_WLAN_VENDOR_ATTR_MAX,
1073 			  nla_data(nl_vend), nla_len(nl_vend), NULL);
1074 
1075 		attr = tb_vendor[QCA_WLAN_VENDOR_ATTR_FEATURE_FLAGS];
1076 		if (attr) {
1077 			int len = nla_len(attr);
1078 			info->flags = os_malloc(len);
1079 			if (info->flags != NULL) {
1080 				os_memcpy(info->flags, nla_data(attr), len);
1081 				info->flags_len = len;
1082 			}
1083 		}
1084 		attr = tb_vendor[QCA_WLAN_VENDOR_ATTR_CONCURRENCY_CAPA];
1085 		if (attr)
1086 			info->capa->conc_capab = nla_get_u32(attr);
1087 
1088 		attr = tb_vendor[
1089 			QCA_WLAN_VENDOR_ATTR_MAX_CONCURRENT_CHANNELS_2_4_BAND];
1090 		if (attr)
1091 			info->capa->max_conc_chan_2_4 = nla_get_u32(attr);
1092 
1093 		attr = tb_vendor[
1094 			QCA_WLAN_VENDOR_ATTR_MAX_CONCURRENT_CHANNELS_5_0_BAND];
1095 		if (attr)
1096 			info->capa->max_conc_chan_5_0 = nla_get_u32(attr);
1097 	}
1098 
1099 	return NL_SKIP;
1100 }
1101 
1102 
1103 static int check_feature(enum qca_wlan_vendor_features feature,
1104 			 struct features_info *info)
1105 {
1106 	size_t idx = feature / 8;
1107 
1108 	return (idx < info->flags_len) &&
1109 		(info->flags[idx] & BIT(feature % 8));
1110 }
1111 
1112 
1113 static void qca_nl80211_get_features(struct wpa_driver_nl80211_data *drv)
1114 {
1115 	struct nl_msg *msg;
1116 	struct features_info info;
1117 	int ret;
1118 
1119 	if (!drv->get_features_vendor_cmd_avail)
1120 		return;
1121 
1122 	if (!(msg = nl80211_drv_msg(drv, 0, NL80211_CMD_VENDOR)) ||
1123 	    nla_put_u32(msg, NL80211_ATTR_VENDOR_ID, OUI_QCA) ||
1124 	    nla_put_u32(msg, NL80211_ATTR_VENDOR_SUBCMD,
1125 			QCA_NL80211_VENDOR_SUBCMD_GET_FEATURES)) {
1126 		nlmsg_free(msg);
1127 		return;
1128 	}
1129 
1130 	os_memset(&info, 0, sizeof(info));
1131 	info.capa = &drv->capa;
1132 	ret = send_and_recv_msgs(drv, msg, features_info_handler, &info);
1133 	if (ret || !info.flags)
1134 		return;
1135 
1136 	if (check_feature(QCA_WLAN_VENDOR_FEATURE_KEY_MGMT_OFFLOAD, &info))
1137 		drv->capa.flags |= WPA_DRIVER_FLAGS_KEY_MGMT_OFFLOAD;
1138 
1139 	if (check_feature(QCA_WLAN_VENDOR_FEATURE_SUPPORT_HW_MODE_ANY, &info))
1140 		drv->capa.flags |= WPA_DRIVER_FLAGS_SUPPORT_HW_MODE_ANY;
1141 
1142 	if (check_feature(QCA_WLAN_VENDOR_FEATURE_OFFCHANNEL_SIMULTANEOUS,
1143 			  &info))
1144 		drv->capa.flags |= WPA_DRIVER_FLAGS_OFFCHANNEL_SIMULTANEOUS;
1145 	if (check_feature(QCA_WLAN_VENDOR_FEATURE_P2P_LISTEN_OFFLOAD, &info))
1146 		drv->capa.flags |= WPA_DRIVER_FLAGS_P2P_LISTEN_OFFLOAD;
1147 	if (check_feature(QCA_WLAN_VENDOR_FEATURE_OCE_STA, &info))
1148 		drv->capa.flags |= WPA_DRIVER_FLAGS_OCE_STA;
1149 	if (check_feature(QCA_WLAN_VENDOR_FEATURE_OCE_AP, &info))
1150 		drv->capa.flags |= WPA_DRIVER_FLAGS_OCE_AP;
1151 	if (check_feature(QCA_WLAN_VENDOR_FEATURE_OCE_STA_CFON, &info))
1152 		drv->capa.flags |= WPA_DRIVER_FLAGS_OCE_STA_CFON;
1153 	os_free(info.flags);
1154 }
1155 
1156 #endif /* CONFIG_DRIVER_NL80211_QCA */
1157 
1158 
1159 int wpa_driver_nl80211_capa(struct wpa_driver_nl80211_data *drv)
1160 {
1161 	struct wiphy_info_data info;
1162 	if (wpa_driver_nl80211_get_info(drv, &info))
1163 		return -1;
1164 
1165 	if (info.error)
1166 		return -1;
1167 
1168 	drv->has_capability = 1;
1169 	drv->capa.key_mgmt = WPA_DRIVER_CAPA_KEY_MGMT_WPA |
1170 		WPA_DRIVER_CAPA_KEY_MGMT_WPA_PSK |
1171 		WPA_DRIVER_CAPA_KEY_MGMT_WPA2 |
1172 		WPA_DRIVER_CAPA_KEY_MGMT_WPA2_PSK |
1173 		WPA_DRIVER_CAPA_KEY_MGMT_SUITE_B |
1174 		WPA_DRIVER_CAPA_KEY_MGMT_SUITE_B_192 |
1175 		WPA_DRIVER_CAPA_KEY_MGMT_OWE |
1176 		WPA_DRIVER_CAPA_KEY_MGMT_DPP;
1177 
1178 	if (drv->capa.flags & WPA_DRIVER_FLAGS_SME)
1179 		drv->capa.key_mgmt |= WPA_DRIVER_CAPA_KEY_MGMT_FILS_SHA256 |
1180 			WPA_DRIVER_CAPA_KEY_MGMT_FILS_SHA384 |
1181 			WPA_DRIVER_CAPA_KEY_MGMT_FT_FILS_SHA256 |
1182 			WPA_DRIVER_CAPA_KEY_MGMT_FT_FILS_SHA384;
1183 	else if (drv->capa.flags & WPA_DRIVER_FLAGS_FILS_SK_OFFLOAD)
1184 		drv->capa.key_mgmt |= WPA_DRIVER_CAPA_KEY_MGMT_FILS_SHA256 |
1185 			WPA_DRIVER_CAPA_KEY_MGMT_FILS_SHA384;
1186 
1187 	drv->capa.auth = WPA_DRIVER_AUTH_OPEN |
1188 		WPA_DRIVER_AUTH_SHARED |
1189 		WPA_DRIVER_AUTH_LEAP;
1190 
1191 	drv->capa.flags |= WPA_DRIVER_FLAGS_SANE_ERROR_CODES;
1192 	drv->capa.flags |= WPA_DRIVER_FLAGS_SET_KEYS_AFTER_ASSOC_DONE;
1193 	drv->capa.flags |= WPA_DRIVER_FLAGS_EAPOL_TX_STATUS;
1194 
1195 	/*
1196 	 * As all cfg80211 drivers must support cases where the AP interface is
1197 	 * removed without the knowledge of wpa_supplicant/hostapd, e.g., in
1198 	 * case that the user space daemon has crashed, they must be able to
1199 	 * cleanup all stations and key entries in the AP tear down flow. Thus,
1200 	 * this flag can/should always be set for cfg80211 drivers.
1201 	 */
1202 	drv->capa.flags |= WPA_DRIVER_FLAGS_AP_TEARDOWN_SUPPORT;
1203 
1204 	if (!info.device_ap_sme) {
1205 		drv->capa.flags |= WPA_DRIVER_FLAGS_DEAUTH_TX_STATUS;
1206 
1207 		/*
1208 		 * No AP SME is currently assumed to also indicate no AP MLME
1209 		 * in the driver/firmware.
1210 		 */
1211 		drv->capa.flags |= WPA_DRIVER_FLAGS_AP_MLME;
1212 	}
1213 
1214 	drv->device_ap_sme = info.device_ap_sme;
1215 	drv->poll_command_supported = info.poll_command_supported;
1216 	drv->data_tx_status = info.data_tx_status;
1217 	drv->p2p_go_ctwindow_supported = info.p2p_go_ctwindow_supported;
1218 	if (info.set_qos_map_supported)
1219 		drv->capa.flags |= WPA_DRIVER_FLAGS_QOS_MAPPING;
1220 	drv->have_low_prio_scan = info.have_low_prio_scan;
1221 
1222 	/*
1223 	 * If poll command and tx status are supported, mac80211 is new enough
1224 	 * to have everything we need to not need monitor interfaces.
1225 	 */
1226 	drv->use_monitor = !info.device_ap_sme &&
1227 		(!info.poll_command_supported || !info.data_tx_status);
1228 
1229 	/*
1230 	 * If we aren't going to use monitor interfaces, but the
1231 	 * driver doesn't support data TX status, we won't get TX
1232 	 * status for EAPOL frames.
1233 	 */
1234 	if (!drv->use_monitor && !info.data_tx_status)
1235 		drv->capa.flags &= ~WPA_DRIVER_FLAGS_EAPOL_TX_STATUS;
1236 
1237 #ifdef CONFIG_DRIVER_NL80211_QCA
1238 	if (!(info.capa->flags & WPA_DRIVER_FLAGS_DFS_OFFLOAD))
1239 		qca_nl80211_check_dfs_capa(drv);
1240 	qca_nl80211_get_features(drv);
1241 	qca_nl80211_check_he_capab(drv);
1242 
1243 	/*
1244 	 * To enable offchannel simultaneous support in wpa_supplicant, the
1245 	 * underlying driver needs to support the same along with offchannel TX.
1246 	 * Offchannel TX support is needed since remain_on_channel and
1247 	 * action_tx use some common data structures and hence cannot be
1248 	 * scheduled simultaneously.
1249 	 */
1250 	if (!(drv->capa.flags & WPA_DRIVER_FLAGS_OFFCHANNEL_TX))
1251 		drv->capa.flags &= ~WPA_DRIVER_FLAGS_OFFCHANNEL_SIMULTANEOUS;
1252 #endif /* CONFIG_DRIVER_NL80211_QCA */
1253 
1254 	return 0;
1255 }
1256 
1257 
1258 struct phy_info_arg {
1259 	u16 *num_modes;
1260 	struct hostapd_hw_modes *modes;
1261 	int last_mode, last_chan_idx;
1262 	int failed;
1263 	u8 dfs_domain;
1264 };
1265 
1266 static void phy_info_ht_capa(struct hostapd_hw_modes *mode, struct nlattr *capa,
1267 			     struct nlattr *ampdu_factor,
1268 			     struct nlattr *ampdu_density,
1269 			     struct nlattr *mcs_set)
1270 {
1271 	if (capa)
1272 		mode->ht_capab = nla_get_u16(capa);
1273 
1274 	if (ampdu_factor)
1275 		mode->a_mpdu_params |= nla_get_u8(ampdu_factor) & 0x03;
1276 
1277 	if (ampdu_density)
1278 		mode->a_mpdu_params |= nla_get_u8(ampdu_density) << 2;
1279 
1280 	if (mcs_set && nla_len(mcs_set) >= 16) {
1281 		u8 *mcs;
1282 		mcs = nla_data(mcs_set);
1283 		os_memcpy(mode->mcs_set, mcs, 16);
1284 	}
1285 }
1286 
1287 
1288 static void phy_info_vht_capa(struct hostapd_hw_modes *mode,
1289 			      struct nlattr *capa,
1290 			      struct nlattr *mcs_set)
1291 {
1292 	if (capa)
1293 		mode->vht_capab = nla_get_u32(capa);
1294 
1295 	if (mcs_set && nla_len(mcs_set) >= 8) {
1296 		u8 *mcs;
1297 		mcs = nla_data(mcs_set);
1298 		os_memcpy(mode->vht_mcs_set, mcs, 8);
1299 	}
1300 }
1301 
1302 
1303 static void phy_info_freq(struct hostapd_hw_modes *mode,
1304 			  struct hostapd_channel_data *chan,
1305 			  struct nlattr *tb_freq[])
1306 {
1307 	u8 channel;
1308 	chan->freq = nla_get_u32(tb_freq[NL80211_FREQUENCY_ATTR_FREQ]);
1309 	chan->flag = 0;
1310 	chan->dfs_cac_ms = 0;
1311 	if (ieee80211_freq_to_chan(chan->freq, &channel) != NUM_HOSTAPD_MODES)
1312 		chan->chan = channel;
1313 
1314 	if (tb_freq[NL80211_FREQUENCY_ATTR_DISABLED])
1315 		chan->flag |= HOSTAPD_CHAN_DISABLED;
1316 	if (tb_freq[NL80211_FREQUENCY_ATTR_NO_IR])
1317 		chan->flag |= HOSTAPD_CHAN_NO_IR;
1318 	if (tb_freq[NL80211_FREQUENCY_ATTR_RADAR])
1319 		chan->flag |= HOSTAPD_CHAN_RADAR;
1320 	if (tb_freq[NL80211_FREQUENCY_ATTR_INDOOR_ONLY])
1321 		chan->flag |= HOSTAPD_CHAN_INDOOR_ONLY;
1322 	if (tb_freq[NL80211_FREQUENCY_ATTR_GO_CONCURRENT])
1323 		chan->flag |= HOSTAPD_CHAN_GO_CONCURRENT;
1324 
1325 	if (tb_freq[NL80211_FREQUENCY_ATTR_DFS_STATE]) {
1326 		enum nl80211_dfs_state state =
1327 			nla_get_u32(tb_freq[NL80211_FREQUENCY_ATTR_DFS_STATE]);
1328 
1329 		switch (state) {
1330 		case NL80211_DFS_USABLE:
1331 			chan->flag |= HOSTAPD_CHAN_DFS_USABLE;
1332 			break;
1333 		case NL80211_DFS_AVAILABLE:
1334 			chan->flag |= HOSTAPD_CHAN_DFS_AVAILABLE;
1335 			break;
1336 		case NL80211_DFS_UNAVAILABLE:
1337 			chan->flag |= HOSTAPD_CHAN_DFS_UNAVAILABLE;
1338 			break;
1339 		}
1340 	}
1341 
1342 	if (tb_freq[NL80211_FREQUENCY_ATTR_DFS_CAC_TIME]) {
1343 		chan->dfs_cac_ms = nla_get_u32(
1344 			tb_freq[NL80211_FREQUENCY_ATTR_DFS_CAC_TIME]);
1345 	}
1346 }
1347 
1348 
1349 static int phy_info_freqs(struct phy_info_arg *phy_info,
1350 			  struct hostapd_hw_modes *mode, struct nlattr *tb)
1351 {
1352 	static struct nla_policy freq_policy[NL80211_FREQUENCY_ATTR_MAX + 1] = {
1353 		[NL80211_FREQUENCY_ATTR_FREQ] = { .type = NLA_U32 },
1354 		[NL80211_FREQUENCY_ATTR_DISABLED] = { .type = NLA_FLAG },
1355 		[NL80211_FREQUENCY_ATTR_NO_IR] = { .type = NLA_FLAG },
1356 		[NL80211_FREQUENCY_ATTR_RADAR] = { .type = NLA_FLAG },
1357 		[NL80211_FREQUENCY_ATTR_MAX_TX_POWER] = { .type = NLA_U32 },
1358 		[NL80211_FREQUENCY_ATTR_DFS_STATE] = { .type = NLA_U32 },
1359 	};
1360 	int new_channels = 0;
1361 	struct hostapd_channel_data *channel;
1362 	struct nlattr *tb_freq[NL80211_FREQUENCY_ATTR_MAX + 1];
1363 	struct nlattr *nl_freq;
1364 	int rem_freq, idx;
1365 
1366 	if (tb == NULL)
1367 		return NL_OK;
1368 
1369 	nla_for_each_nested(nl_freq, tb, rem_freq) {
1370 		nla_parse(tb_freq, NL80211_FREQUENCY_ATTR_MAX,
1371 			  nla_data(nl_freq), nla_len(nl_freq), freq_policy);
1372 		if (!tb_freq[NL80211_FREQUENCY_ATTR_FREQ])
1373 			continue;
1374 		new_channels++;
1375 	}
1376 
1377 	channel = os_realloc_array(mode->channels,
1378 				   mode->num_channels + new_channels,
1379 				   sizeof(struct hostapd_channel_data));
1380 	if (!channel)
1381 		return NL_STOP;
1382 
1383 	mode->channels = channel;
1384 	mode->num_channels += new_channels;
1385 
1386 	idx = phy_info->last_chan_idx;
1387 
1388 	nla_for_each_nested(nl_freq, tb, rem_freq) {
1389 		nla_parse(tb_freq, NL80211_FREQUENCY_ATTR_MAX,
1390 			  nla_data(nl_freq), nla_len(nl_freq), freq_policy);
1391 		if (!tb_freq[NL80211_FREQUENCY_ATTR_FREQ])
1392 			continue;
1393 		phy_info_freq(mode, &mode->channels[idx], tb_freq);
1394 		idx++;
1395 	}
1396 	phy_info->last_chan_idx = idx;
1397 
1398 	return NL_OK;
1399 }
1400 
1401 
1402 static int phy_info_rates(struct hostapd_hw_modes *mode, struct nlattr *tb)
1403 {
1404 	static struct nla_policy rate_policy[NL80211_BITRATE_ATTR_MAX + 1] = {
1405 		[NL80211_BITRATE_ATTR_RATE] = { .type = NLA_U32 },
1406 		[NL80211_BITRATE_ATTR_2GHZ_SHORTPREAMBLE] =
1407 		{ .type = NLA_FLAG },
1408 	};
1409 	struct nlattr *tb_rate[NL80211_BITRATE_ATTR_MAX + 1];
1410 	struct nlattr *nl_rate;
1411 	int rem_rate, idx;
1412 
1413 	if (tb == NULL)
1414 		return NL_OK;
1415 
1416 	nla_for_each_nested(nl_rate, tb, rem_rate) {
1417 		nla_parse(tb_rate, NL80211_BITRATE_ATTR_MAX,
1418 			  nla_data(nl_rate), nla_len(nl_rate),
1419 			  rate_policy);
1420 		if (!tb_rate[NL80211_BITRATE_ATTR_RATE])
1421 			continue;
1422 		mode->num_rates++;
1423 	}
1424 
1425 	mode->rates = os_calloc(mode->num_rates, sizeof(int));
1426 	if (!mode->rates)
1427 		return NL_STOP;
1428 
1429 	idx = 0;
1430 
1431 	nla_for_each_nested(nl_rate, tb, rem_rate) {
1432 		nla_parse(tb_rate, NL80211_BITRATE_ATTR_MAX,
1433 			  nla_data(nl_rate), nla_len(nl_rate),
1434 			  rate_policy);
1435 		if (!tb_rate[NL80211_BITRATE_ATTR_RATE])
1436 			continue;
1437 		mode->rates[idx] = nla_get_u32(
1438 			tb_rate[NL80211_BITRATE_ATTR_RATE]);
1439 		idx++;
1440 	}
1441 
1442 	return NL_OK;
1443 }
1444 
1445 
1446 static int phy_info_band(struct phy_info_arg *phy_info, struct nlattr *nl_band)
1447 {
1448 	struct nlattr *tb_band[NL80211_BAND_ATTR_MAX + 1];
1449 	struct hostapd_hw_modes *mode;
1450 	int ret;
1451 
1452 	if (phy_info->last_mode != nl_band->nla_type) {
1453 		mode = os_realloc_array(phy_info->modes,
1454 					*phy_info->num_modes + 1,
1455 					sizeof(*mode));
1456 		if (!mode) {
1457 			phy_info->failed = 1;
1458 			return NL_STOP;
1459 		}
1460 		phy_info->modes = mode;
1461 
1462 		mode = &phy_info->modes[*(phy_info->num_modes)];
1463 		os_memset(mode, 0, sizeof(*mode));
1464 		mode->mode = NUM_HOSTAPD_MODES;
1465 		mode->flags = HOSTAPD_MODE_FLAG_HT_INFO_KNOWN |
1466 			HOSTAPD_MODE_FLAG_VHT_INFO_KNOWN;
1467 
1468 		/*
1469 		 * Unsupported VHT MCS stream is defined as value 3, so the VHT
1470 		 * MCS RX/TX map must be initialized with 0xffff to mark all 8
1471 		 * possible streams as unsupported. This will be overridden if
1472 		 * driver advertises VHT support.
1473 		 */
1474 		mode->vht_mcs_set[0] = 0xff;
1475 		mode->vht_mcs_set[1] = 0xff;
1476 		mode->vht_mcs_set[4] = 0xff;
1477 		mode->vht_mcs_set[5] = 0xff;
1478 
1479 		*(phy_info->num_modes) += 1;
1480 		phy_info->last_mode = nl_band->nla_type;
1481 		phy_info->last_chan_idx = 0;
1482 	} else
1483 		mode = &phy_info->modes[*(phy_info->num_modes) - 1];
1484 
1485 	nla_parse(tb_band, NL80211_BAND_ATTR_MAX, nla_data(nl_band),
1486 		  nla_len(nl_band), NULL);
1487 
1488 	phy_info_ht_capa(mode, tb_band[NL80211_BAND_ATTR_HT_CAPA],
1489 			 tb_band[NL80211_BAND_ATTR_HT_AMPDU_FACTOR],
1490 			 tb_band[NL80211_BAND_ATTR_HT_AMPDU_DENSITY],
1491 			 tb_band[NL80211_BAND_ATTR_HT_MCS_SET]);
1492 	phy_info_vht_capa(mode, tb_band[NL80211_BAND_ATTR_VHT_CAPA],
1493 			  tb_band[NL80211_BAND_ATTR_VHT_MCS_SET]);
1494 	ret = phy_info_freqs(phy_info, mode, tb_band[NL80211_BAND_ATTR_FREQS]);
1495 	if (ret == NL_OK)
1496 		ret = phy_info_rates(mode, tb_band[NL80211_BAND_ATTR_RATES]);
1497 	if (ret != NL_OK) {
1498 		phy_info->failed = 1;
1499 		return ret;
1500 	}
1501 
1502 	return NL_OK;
1503 }
1504 
1505 
1506 static int phy_info_handler(struct nl_msg *msg, void *arg)
1507 {
1508 	struct nlattr *tb_msg[NL80211_ATTR_MAX + 1];
1509 	struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
1510 	struct phy_info_arg *phy_info = arg;
1511 	struct nlattr *nl_band;
1512 	int rem_band;
1513 
1514 	nla_parse(tb_msg, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
1515 		  genlmsg_attrlen(gnlh, 0), NULL);
1516 
1517 	if (!tb_msg[NL80211_ATTR_WIPHY_BANDS])
1518 		return NL_SKIP;
1519 
1520 	nla_for_each_nested(nl_band, tb_msg[NL80211_ATTR_WIPHY_BANDS], rem_band)
1521 	{
1522 		int res = phy_info_band(phy_info, nl_band);
1523 		if (res != NL_OK)
1524 			return res;
1525 	}
1526 
1527 	return NL_SKIP;
1528 }
1529 
1530 
1531 static struct hostapd_hw_modes *
1532 wpa_driver_nl80211_postprocess_modes(struct hostapd_hw_modes *modes,
1533 				     u16 *num_modes)
1534 {
1535 	u16 m;
1536 	struct hostapd_hw_modes *mode11g = NULL, *nmodes, *mode;
1537 	int i, mode11g_idx = -1;
1538 
1539 	/* heuristic to set up modes */
1540 	for (m = 0; m < *num_modes; m++) {
1541 		if (!modes[m].num_channels)
1542 			continue;
1543 		if (modes[m].channels[0].freq < 4000) {
1544 			modes[m].mode = HOSTAPD_MODE_IEEE80211B;
1545 			for (i = 0; i < modes[m].num_rates; i++) {
1546 				if (modes[m].rates[i] > 200) {
1547 					modes[m].mode = HOSTAPD_MODE_IEEE80211G;
1548 					break;
1549 				}
1550 			}
1551 		} else if (modes[m].channels[0].freq > 50000)
1552 			modes[m].mode = HOSTAPD_MODE_IEEE80211AD;
1553 		else
1554 			modes[m].mode = HOSTAPD_MODE_IEEE80211A;
1555 	}
1556 
1557 	/* If only 802.11g mode is included, use it to construct matching
1558 	 * 802.11b mode data. */
1559 
1560 	for (m = 0; m < *num_modes; m++) {
1561 		if (modes[m].mode == HOSTAPD_MODE_IEEE80211B)
1562 			return modes; /* 802.11b already included */
1563 		if (modes[m].mode == HOSTAPD_MODE_IEEE80211G)
1564 			mode11g_idx = m;
1565 	}
1566 
1567 	if (mode11g_idx < 0)
1568 		return modes; /* 2.4 GHz band not supported at all */
1569 
1570 	nmodes = os_realloc_array(modes, *num_modes + 1, sizeof(*nmodes));
1571 	if (nmodes == NULL)
1572 		return modes; /* Could not add 802.11b mode */
1573 
1574 	mode = &nmodes[*num_modes];
1575 	os_memset(mode, 0, sizeof(*mode));
1576 	(*num_modes)++;
1577 	modes = nmodes;
1578 
1579 	mode->mode = HOSTAPD_MODE_IEEE80211B;
1580 
1581 	mode11g = &modes[mode11g_idx];
1582 	mode->num_channels = mode11g->num_channels;
1583 	mode->channels = os_memdup(mode11g->channels,
1584 				   mode11g->num_channels *
1585 				   sizeof(struct hostapd_channel_data));
1586 	if (mode->channels == NULL) {
1587 		(*num_modes)--;
1588 		return modes; /* Could not add 802.11b mode */
1589 	}
1590 
1591 	mode->num_rates = 0;
1592 	mode->rates = os_malloc(4 * sizeof(int));
1593 	if (mode->rates == NULL) {
1594 		os_free(mode->channels);
1595 		(*num_modes)--;
1596 		return modes; /* Could not add 802.11b mode */
1597 	}
1598 
1599 	for (i = 0; i < mode11g->num_rates; i++) {
1600 		if (mode11g->rates[i] != 10 && mode11g->rates[i] != 20 &&
1601 		    mode11g->rates[i] != 55 && mode11g->rates[i] != 110)
1602 			continue;
1603 		mode->rates[mode->num_rates] = mode11g->rates[i];
1604 		mode->num_rates++;
1605 		if (mode->num_rates == 4)
1606 			break;
1607 	}
1608 
1609 	if (mode->num_rates == 0) {
1610 		os_free(mode->channels);
1611 		os_free(mode->rates);
1612 		(*num_modes)--;
1613 		return modes; /* No 802.11b rates */
1614 	}
1615 
1616 	wpa_printf(MSG_DEBUG, "nl80211: Added 802.11b mode based on 802.11g "
1617 		   "information");
1618 
1619 	return modes;
1620 }
1621 
1622 
1623 static void nl80211_set_ht40_mode(struct hostapd_hw_modes *mode, int start,
1624 				  int end)
1625 {
1626 	int c;
1627 
1628 	for (c = 0; c < mode->num_channels; c++) {
1629 		struct hostapd_channel_data *chan = &mode->channels[c];
1630 		if (chan->freq - 10 >= start && chan->freq + 10 <= end)
1631 			chan->flag |= HOSTAPD_CHAN_HT40;
1632 	}
1633 }
1634 
1635 
1636 static void nl80211_set_ht40_mode_sec(struct hostapd_hw_modes *mode, int start,
1637 				      int end)
1638 {
1639 	int c;
1640 
1641 	for (c = 0; c < mode->num_channels; c++) {
1642 		struct hostapd_channel_data *chan = &mode->channels[c];
1643 		if (!(chan->flag & HOSTAPD_CHAN_HT40))
1644 			continue;
1645 		if (chan->freq - 30 >= start && chan->freq - 10 <= end)
1646 			chan->flag |= HOSTAPD_CHAN_HT40MINUS;
1647 		if (chan->freq + 10 >= start && chan->freq + 30 <= end)
1648 			chan->flag |= HOSTAPD_CHAN_HT40PLUS;
1649 	}
1650 }
1651 
1652 
1653 static void nl80211_reg_rule_max_eirp(u32 start, u32 end, u32 max_eirp,
1654 				      struct phy_info_arg *results)
1655 {
1656 	u16 m;
1657 
1658 	for (m = 0; m < *results->num_modes; m++) {
1659 		int c;
1660 		struct hostapd_hw_modes *mode = &results->modes[m];
1661 
1662 		for (c = 0; c < mode->num_channels; c++) {
1663 			struct hostapd_channel_data *chan = &mode->channels[c];
1664 			if ((u32) chan->freq - 10 >= start &&
1665 			    (u32) chan->freq + 10 <= end)
1666 				chan->max_tx_power = max_eirp;
1667 		}
1668 	}
1669 }
1670 
1671 
1672 static void nl80211_reg_rule_ht40(u32 start, u32 end,
1673 				  struct phy_info_arg *results)
1674 {
1675 	u16 m;
1676 
1677 	for (m = 0; m < *results->num_modes; m++) {
1678 		if (!(results->modes[m].ht_capab &
1679 		      HT_CAP_INFO_SUPP_CHANNEL_WIDTH_SET))
1680 			continue;
1681 		nl80211_set_ht40_mode(&results->modes[m], start, end);
1682 	}
1683 }
1684 
1685 
1686 static void nl80211_reg_rule_sec(struct nlattr *tb[],
1687 				 struct phy_info_arg *results)
1688 {
1689 	u32 start, end, max_bw;
1690 	u16 m;
1691 
1692 	if (tb[NL80211_ATTR_FREQ_RANGE_START] == NULL ||
1693 	    tb[NL80211_ATTR_FREQ_RANGE_END] == NULL ||
1694 	    tb[NL80211_ATTR_FREQ_RANGE_MAX_BW] == NULL)
1695 		return;
1696 
1697 	start = nla_get_u32(tb[NL80211_ATTR_FREQ_RANGE_START]) / 1000;
1698 	end = nla_get_u32(tb[NL80211_ATTR_FREQ_RANGE_END]) / 1000;
1699 	max_bw = nla_get_u32(tb[NL80211_ATTR_FREQ_RANGE_MAX_BW]) / 1000;
1700 
1701 	if (max_bw < 20)
1702 		return;
1703 
1704 	for (m = 0; m < *results->num_modes; m++) {
1705 		if (!(results->modes[m].ht_capab &
1706 		      HT_CAP_INFO_SUPP_CHANNEL_WIDTH_SET))
1707 			continue;
1708 		nl80211_set_ht40_mode_sec(&results->modes[m], start, end);
1709 	}
1710 }
1711 
1712 
1713 static void nl80211_set_vht_mode(struct hostapd_hw_modes *mode, int start,
1714 				 int end, int max_bw)
1715 {
1716 	int c;
1717 
1718 	for (c = 0; c < mode->num_channels; c++) {
1719 		struct hostapd_channel_data *chan = &mode->channels[c];
1720 		if (chan->freq - 10 >= start && chan->freq + 70 <= end)
1721 			chan->flag |= HOSTAPD_CHAN_VHT_10_70;
1722 
1723 		if (chan->freq - 30 >= start && chan->freq + 50 <= end)
1724 			chan->flag |= HOSTAPD_CHAN_VHT_30_50;
1725 
1726 		if (chan->freq - 50 >= start && chan->freq + 30 <= end)
1727 			chan->flag |= HOSTAPD_CHAN_VHT_50_30;
1728 
1729 		if (chan->freq - 70 >= start && chan->freq + 10 <= end)
1730 			chan->flag |= HOSTAPD_CHAN_VHT_70_10;
1731 
1732 		if (max_bw >= 160) {
1733 			if (chan->freq - 10 >= start && chan->freq + 150 <= end)
1734 				chan->flag |= HOSTAPD_CHAN_VHT_10_150;
1735 
1736 			if (chan->freq - 30 >= start && chan->freq + 130 <= end)
1737 				chan->flag |= HOSTAPD_CHAN_VHT_30_130;
1738 
1739 			if (chan->freq - 50 >= start && chan->freq + 110 <= end)
1740 				chan->flag |= HOSTAPD_CHAN_VHT_50_110;
1741 
1742 			if (chan->freq - 70 >= start && chan->freq + 90 <= end)
1743 				chan->flag |= HOSTAPD_CHAN_VHT_70_90;
1744 
1745 			if (chan->freq - 90 >= start && chan->freq + 70 <= end)
1746 				chan->flag |= HOSTAPD_CHAN_VHT_90_70;
1747 
1748 			if (chan->freq - 110 >= start && chan->freq + 50 <= end)
1749 				chan->flag |= HOSTAPD_CHAN_VHT_110_50;
1750 
1751 			if (chan->freq - 130 >= start && chan->freq + 30 <= end)
1752 				chan->flag |= HOSTAPD_CHAN_VHT_130_30;
1753 
1754 			if (chan->freq - 150 >= start && chan->freq + 10 <= end)
1755 				chan->flag |= HOSTAPD_CHAN_VHT_150_10;
1756 		}
1757 	}
1758 }
1759 
1760 
1761 static void nl80211_reg_rule_vht(struct nlattr *tb[],
1762 				 struct phy_info_arg *results)
1763 {
1764 	u32 start, end, max_bw;
1765 	u16 m;
1766 
1767 	if (tb[NL80211_ATTR_FREQ_RANGE_START] == NULL ||
1768 	    tb[NL80211_ATTR_FREQ_RANGE_END] == NULL ||
1769 	    tb[NL80211_ATTR_FREQ_RANGE_MAX_BW] == NULL)
1770 		return;
1771 
1772 	start = nla_get_u32(tb[NL80211_ATTR_FREQ_RANGE_START]) / 1000;
1773 	end = nla_get_u32(tb[NL80211_ATTR_FREQ_RANGE_END]) / 1000;
1774 	max_bw = nla_get_u32(tb[NL80211_ATTR_FREQ_RANGE_MAX_BW]) / 1000;
1775 
1776 	if (max_bw < 80)
1777 		return;
1778 
1779 	for (m = 0; m < *results->num_modes; m++) {
1780 		if (!(results->modes[m].ht_capab &
1781 		      HT_CAP_INFO_SUPP_CHANNEL_WIDTH_SET))
1782 			continue;
1783 		/* TODO: use a real VHT support indication */
1784 		if (!results->modes[m].vht_capab)
1785 			continue;
1786 
1787 		nl80211_set_vht_mode(&results->modes[m], start, end, max_bw);
1788 	}
1789 }
1790 
1791 
1792 static void nl80211_set_dfs_domain(enum nl80211_dfs_regions region,
1793 				   u8 *dfs_domain)
1794 {
1795 	if (region == NL80211_DFS_FCC)
1796 		*dfs_domain = HOSTAPD_DFS_REGION_FCC;
1797 	else if (region == NL80211_DFS_ETSI)
1798 		*dfs_domain = HOSTAPD_DFS_REGION_ETSI;
1799 	else if (region == NL80211_DFS_JP)
1800 		*dfs_domain = HOSTAPD_DFS_REGION_JP;
1801 	else
1802 		*dfs_domain = 0;
1803 }
1804 
1805 
1806 static const char * dfs_domain_name(enum nl80211_dfs_regions region)
1807 {
1808 	switch (region) {
1809 	case NL80211_DFS_UNSET:
1810 		return "DFS-UNSET";
1811 	case NL80211_DFS_FCC:
1812 		return "DFS-FCC";
1813 	case NL80211_DFS_ETSI:
1814 		return "DFS-ETSI";
1815 	case NL80211_DFS_JP:
1816 		return "DFS-JP";
1817 	default:
1818 		return "DFS-invalid";
1819 	}
1820 }
1821 
1822 
1823 static int nl80211_get_reg(struct nl_msg *msg, void *arg)
1824 {
1825 	struct phy_info_arg *results = arg;
1826 	struct nlattr *tb_msg[NL80211_ATTR_MAX + 1];
1827 	struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
1828 	struct nlattr *nl_rule;
1829 	struct nlattr *tb_rule[NL80211_FREQUENCY_ATTR_MAX + 1];
1830 	int rem_rule;
1831 	static struct nla_policy reg_policy[NL80211_FREQUENCY_ATTR_MAX + 1] = {
1832 		[NL80211_ATTR_REG_RULE_FLAGS] = { .type = NLA_U32 },
1833 		[NL80211_ATTR_FREQ_RANGE_START] = { .type = NLA_U32 },
1834 		[NL80211_ATTR_FREQ_RANGE_END] = { .type = NLA_U32 },
1835 		[NL80211_ATTR_FREQ_RANGE_MAX_BW] = { .type = NLA_U32 },
1836 		[NL80211_ATTR_POWER_RULE_MAX_ANT_GAIN] = { .type = NLA_U32 },
1837 		[NL80211_ATTR_POWER_RULE_MAX_EIRP] = { .type = NLA_U32 },
1838 	};
1839 
1840 	nla_parse(tb_msg, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
1841 		  genlmsg_attrlen(gnlh, 0), NULL);
1842 	if (!tb_msg[NL80211_ATTR_REG_ALPHA2] ||
1843 	    !tb_msg[NL80211_ATTR_REG_RULES]) {
1844 		wpa_printf(MSG_DEBUG, "nl80211: No regulatory information "
1845 			   "available");
1846 		return NL_SKIP;
1847 	}
1848 
1849 	if (tb_msg[NL80211_ATTR_DFS_REGION]) {
1850 		enum nl80211_dfs_regions dfs_domain;
1851 		dfs_domain = nla_get_u8(tb_msg[NL80211_ATTR_DFS_REGION]);
1852 		nl80211_set_dfs_domain(dfs_domain, &results->dfs_domain);
1853 		wpa_printf(MSG_DEBUG, "nl80211: Regulatory information - country=%s (%s)",
1854 			   (char *) nla_data(tb_msg[NL80211_ATTR_REG_ALPHA2]),
1855 			   dfs_domain_name(dfs_domain));
1856 	} else {
1857 		wpa_printf(MSG_DEBUG, "nl80211: Regulatory information - country=%s",
1858 			   (char *) nla_data(tb_msg[NL80211_ATTR_REG_ALPHA2]));
1859 	}
1860 
1861 	nla_for_each_nested(nl_rule, tb_msg[NL80211_ATTR_REG_RULES], rem_rule)
1862 	{
1863 		u32 start, end, max_eirp = 0, max_bw = 0, flags = 0;
1864 		nla_parse(tb_rule, NL80211_FREQUENCY_ATTR_MAX,
1865 			  nla_data(nl_rule), nla_len(nl_rule), reg_policy);
1866 		if (tb_rule[NL80211_ATTR_FREQ_RANGE_START] == NULL ||
1867 		    tb_rule[NL80211_ATTR_FREQ_RANGE_END] == NULL)
1868 			continue;
1869 		start = nla_get_u32(tb_rule[NL80211_ATTR_FREQ_RANGE_START]) / 1000;
1870 		end = nla_get_u32(tb_rule[NL80211_ATTR_FREQ_RANGE_END]) / 1000;
1871 		if (tb_rule[NL80211_ATTR_POWER_RULE_MAX_EIRP])
1872 			max_eirp = nla_get_u32(tb_rule[NL80211_ATTR_POWER_RULE_MAX_EIRP]) / 100;
1873 		if (tb_rule[NL80211_ATTR_FREQ_RANGE_MAX_BW])
1874 			max_bw = nla_get_u32(tb_rule[NL80211_ATTR_FREQ_RANGE_MAX_BW]) / 1000;
1875 		if (tb_rule[NL80211_ATTR_REG_RULE_FLAGS])
1876 			flags = nla_get_u32(tb_rule[NL80211_ATTR_REG_RULE_FLAGS]);
1877 
1878 		wpa_printf(MSG_DEBUG, "nl80211: %u-%u @ %u MHz %u mBm%s%s%s%s%s%s%s%s",
1879 			   start, end, max_bw, max_eirp,
1880 			   flags & NL80211_RRF_NO_OFDM ? " (no OFDM)" : "",
1881 			   flags & NL80211_RRF_NO_CCK ? " (no CCK)" : "",
1882 			   flags & NL80211_RRF_NO_INDOOR ? " (no indoor)" : "",
1883 			   flags & NL80211_RRF_NO_OUTDOOR ? " (no outdoor)" :
1884 			   "",
1885 			   flags & NL80211_RRF_DFS ? " (DFS)" : "",
1886 			   flags & NL80211_RRF_PTP_ONLY ? " (PTP only)" : "",
1887 			   flags & NL80211_RRF_PTMP_ONLY ? " (PTMP only)" : "",
1888 			   flags & NL80211_RRF_NO_IR ? " (no IR)" : "");
1889 		if (max_bw >= 40)
1890 			nl80211_reg_rule_ht40(start, end, results);
1891 		if (tb_rule[NL80211_ATTR_POWER_RULE_MAX_EIRP])
1892 			nl80211_reg_rule_max_eirp(start, end, max_eirp,
1893 						  results);
1894 	}
1895 
1896 	nla_for_each_nested(nl_rule, tb_msg[NL80211_ATTR_REG_RULES], rem_rule)
1897 	{
1898 		nla_parse(tb_rule, NL80211_FREQUENCY_ATTR_MAX,
1899 			  nla_data(nl_rule), nla_len(nl_rule), reg_policy);
1900 		nl80211_reg_rule_sec(tb_rule, results);
1901 	}
1902 
1903 	nla_for_each_nested(nl_rule, tb_msg[NL80211_ATTR_REG_RULES], rem_rule)
1904 	{
1905 		nla_parse(tb_rule, NL80211_FREQUENCY_ATTR_MAX,
1906 			  nla_data(nl_rule), nla_len(nl_rule), reg_policy);
1907 		nl80211_reg_rule_vht(tb_rule, results);
1908 	}
1909 
1910 	return NL_SKIP;
1911 }
1912 
1913 
1914 static int nl80211_set_regulatory_flags(struct wpa_driver_nl80211_data *drv,
1915 					struct phy_info_arg *results)
1916 {
1917 	struct nl_msg *msg;
1918 
1919 	msg = nlmsg_alloc();
1920 	if (!msg)
1921 		return -ENOMEM;
1922 
1923 	nl80211_cmd(drv, msg, 0, NL80211_CMD_GET_REG);
1924 	if (drv->capa.flags & WPA_DRIVER_FLAGS_SELF_MANAGED_REGULATORY) {
1925 		if (nla_put_u32(msg, NL80211_ATTR_WIPHY, drv->wiphy_idx)) {
1926 			nlmsg_free(msg);
1927 			return -1;
1928 		}
1929 	}
1930 
1931 	return send_and_recv_msgs(drv, msg, nl80211_get_reg, results);
1932 }
1933 
1934 
1935 struct hostapd_hw_modes *
1936 nl80211_get_hw_feature_data(void *priv, u16 *num_modes, u16 *flags,
1937 			    u8 *dfs_domain)
1938 {
1939 	u32 feat;
1940 	struct i802_bss *bss = priv;
1941 	struct wpa_driver_nl80211_data *drv = bss->drv;
1942 	int nl_flags = 0;
1943 	struct nl_msg *msg;
1944 	struct phy_info_arg result = {
1945 		.num_modes = num_modes,
1946 		.modes = NULL,
1947 		.last_mode = -1,
1948 		.failed = 0,
1949 		.dfs_domain = 0,
1950 	};
1951 
1952 	*num_modes = 0;
1953 	*flags = 0;
1954 	*dfs_domain = 0;
1955 
1956 	feat = get_nl80211_protocol_features(drv);
1957 	if (feat & NL80211_PROTOCOL_FEATURE_SPLIT_WIPHY_DUMP)
1958 		nl_flags = NLM_F_DUMP;
1959 	if (!(msg = nl80211_cmd_msg(bss, nl_flags, NL80211_CMD_GET_WIPHY)) ||
1960 	    nla_put_flag(msg, NL80211_ATTR_SPLIT_WIPHY_DUMP)) {
1961 		nlmsg_free(msg);
1962 		return NULL;
1963 	}
1964 
1965 	if (send_and_recv_msgs(drv, msg, phy_info_handler, &result) == 0) {
1966 		nl80211_set_regulatory_flags(drv, &result);
1967 		if (result.failed) {
1968 			int i;
1969 
1970 			for (i = 0; result.modes && i < *num_modes; i++) {
1971 				os_free(result.modes[i].channels);
1972 				os_free(result.modes[i].rates);
1973 			}
1974 			os_free(result.modes);
1975 			*num_modes = 0;
1976 			return NULL;
1977 		}
1978 
1979 		*dfs_domain = result.dfs_domain;
1980 
1981 		return wpa_driver_nl80211_postprocess_modes(result.modes,
1982 							    num_modes);
1983 	}
1984 
1985 	return NULL;
1986 }
1987