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 "common/brcm_vendor.h"
20 #include "driver_nl80211.h"
21 
22 
protocol_feature_handler(struct nl_msg * msg,void * arg)23 static int protocol_feature_handler(struct nl_msg *msg, void *arg)
24 {
25 	u32 *feat = arg;
26 	struct nlattr *tb_msg[NL80211_ATTR_MAX + 1];
27 	struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
28 
29 	nla_parse(tb_msg, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
30 		  genlmsg_attrlen(gnlh, 0), NULL);
31 
32 	if (tb_msg[NL80211_ATTR_PROTOCOL_FEATURES])
33 		*feat = nla_get_u32(tb_msg[NL80211_ATTR_PROTOCOL_FEATURES]);
34 
35 	return NL_SKIP;
36 }
37 
38 
get_nl80211_protocol_features(struct wpa_driver_nl80211_data * drv)39 static u32 get_nl80211_protocol_features(struct wpa_driver_nl80211_data *drv)
40 {
41 	u32 feat = 0;
42 	struct nl_msg *msg;
43 
44 	msg = nlmsg_alloc();
45 	if (!msg)
46 		return 0;
47 
48 	if (!nl80211_cmd(drv, msg, 0, NL80211_CMD_GET_PROTOCOL_FEATURES)) {
49 		nlmsg_free(msg);
50 		return 0;
51 	}
52 
53 	if (send_and_recv_resp(drv, msg, protocol_feature_handler, &feat) == 0)
54 		return feat;
55 
56 	return 0;
57 }
58 
59 
60 struct wiphy_info_data {
61 	struct wpa_driver_nl80211_data *drv;
62 	struct wpa_driver_capa *capa;
63 
64 	unsigned int num_multichan_concurrent;
65 
66 	unsigned int error:1;
67 	unsigned int device_ap_sme:1;
68 	unsigned int poll_command_supported:1;
69 	unsigned int data_tx_status: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 	unsigned int update_ft_ies_supported:1;
83 	unsigned int has_key_mgmt:1;
84 	unsigned int has_key_mgmt_iftype:1;
85 };
86 
87 
probe_resp_offload_support(int supp_protocols)88 static unsigned int probe_resp_offload_support(int supp_protocols)
89 {
90 	unsigned int prot = 0;
91 
92 	if (supp_protocols & NL80211_PROBE_RESP_OFFLOAD_SUPPORT_WPS)
93 		prot |= WPA_DRIVER_PROBE_RESP_OFFLOAD_WPS;
94 	if (supp_protocols & NL80211_PROBE_RESP_OFFLOAD_SUPPORT_WPS2)
95 		prot |= WPA_DRIVER_PROBE_RESP_OFFLOAD_WPS2;
96 	if (supp_protocols & NL80211_PROBE_RESP_OFFLOAD_SUPPORT_P2P)
97 		prot |= WPA_DRIVER_PROBE_RESP_OFFLOAD_P2P;
98 	if (supp_protocols & NL80211_PROBE_RESP_OFFLOAD_SUPPORT_80211U)
99 		prot |= WPA_DRIVER_PROBE_RESP_OFFLOAD_INTERWORKING;
100 
101 	return prot;
102 }
103 
104 
wiphy_info_supported_iftypes(struct wiphy_info_data * info,struct nlattr * tb)105 static void wiphy_info_supported_iftypes(struct wiphy_info_data *info,
106 					 struct nlattr *tb)
107 {
108 	struct nlattr *nl_mode;
109 	int i;
110 
111 	if (tb == NULL)
112 		return;
113 
114 	nla_for_each_nested(nl_mode, tb, i) {
115 		switch (nla_type(nl_mode)) {
116 		case NL80211_IFTYPE_AP:
117 			info->capa->flags |= WPA_DRIVER_FLAGS_AP;
118 			break;
119 		case NL80211_IFTYPE_MESH_POINT:
120 			info->capa->flags |= WPA_DRIVER_FLAGS_MESH;
121 			break;
122 		case NL80211_IFTYPE_ADHOC:
123 			info->capa->flags |= WPA_DRIVER_FLAGS_IBSS;
124 			break;
125 		case NL80211_IFTYPE_P2P_DEVICE:
126 			info->capa->flags |=
127 				WPA_DRIVER_FLAGS_DEDICATED_P2P_DEVICE;
128 			break;
129 		case NL80211_IFTYPE_P2P_GO:
130 			info->p2p_go_supported = 1;
131 			break;
132 		case NL80211_IFTYPE_P2P_CLIENT:
133 			info->p2p_client_supported = 1;
134 			break;
135 		}
136 	}
137 }
138 
139 
wiphy_info_iface_comb_process(struct wiphy_info_data * info,struct nlattr * nl_combi)140 static 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 
wiphy_info_iface_comb(struct wiphy_info_data * info,struct nlattr * tb)205 static 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 
wiphy_info_supp_cmds(struct wiphy_info_data * info,struct nlattr * tb)221 static 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 		case NL80211_CMD_UPDATE_FT_IES:
251 			info->update_ft_ies_supported = 1;
252 			break;
253 		}
254 	}
255 }
256 
257 
get_akm_suites_info(struct nlattr * tb)258 static unsigned int get_akm_suites_info(struct nlattr *tb)
259 {
260 	int i, num;
261 	unsigned int key_mgmt = 0;
262 	u32 *akms;
263 
264 	if (!tb)
265 		return 0;
266 
267 	num = nla_len(tb) / sizeof(u32);
268 	akms = nla_data(tb);
269 	for (i = 0; i < num; i++) {
270 		switch (akms[i]) {
271 		case RSN_AUTH_KEY_MGMT_UNSPEC_802_1X:
272 			key_mgmt |= WPA_DRIVER_CAPA_KEY_MGMT_WPA |
273 				WPA_DRIVER_CAPA_KEY_MGMT_WPA2;
274 			break;
275 		case RSN_AUTH_KEY_MGMT_PSK_OVER_802_1X:
276 			key_mgmt |= WPA_DRIVER_CAPA_KEY_MGMT_WPA_PSK |
277 				WPA_DRIVER_CAPA_KEY_MGMT_WPA2_PSK;
278 			break;
279 		case RSN_AUTH_KEY_MGMT_FT_802_1X:
280 			key_mgmt |= WPA_DRIVER_CAPA_KEY_MGMT_FT;
281 			break;
282 		case RSN_AUTH_KEY_MGMT_FT_PSK:
283 			key_mgmt |= WPA_DRIVER_CAPA_KEY_MGMT_FT_PSK;
284 			break;
285 		case RSN_AUTH_KEY_MGMT_802_1X_SHA256:
286 			key_mgmt |= WPA_DRIVER_CAPA_KEY_MGMT_802_1X_SHA256;
287 			break;
288 		case RSN_AUTH_KEY_MGMT_PSK_SHA256:
289 			key_mgmt |= WPA_DRIVER_CAPA_KEY_MGMT_PSK_SHA256;
290 			break;
291 		case RSN_AUTH_KEY_MGMT_TPK_HANDSHAKE:
292 			key_mgmt |= WPA_DRIVER_CAPA_KEY_MGMT_TPK_HANDSHAKE;
293 			break;
294 		case RSN_AUTH_KEY_MGMT_FT_SAE:
295 			key_mgmt |= WPA_DRIVER_CAPA_KEY_MGMT_FT_SAE;
296 			break;
297 		case RSN_AUTH_KEY_MGMT_FT_SAE_EXT_KEY:
298 			key_mgmt |= WPA_DRIVER_CAPA_KEY_MGMT_FT_SAE_EXT_KEY;
299 			break;
300 		case RSN_AUTH_KEY_MGMT_FT_802_1X_SHA384:
301 			key_mgmt |= WPA_DRIVER_CAPA_KEY_MGMT_FT_802_1X_SHA384;
302 			break;
303 		case RSN_AUTH_KEY_MGMT_CCKM:
304 			key_mgmt |= WPA_DRIVER_CAPA_KEY_MGMT_CCKM;
305 			break;
306 		case RSN_AUTH_KEY_MGMT_OSEN:
307 			key_mgmt |= WPA_DRIVER_CAPA_KEY_MGMT_OSEN;
308 			break;
309 		case RSN_AUTH_KEY_MGMT_802_1X_SUITE_B:
310 			key_mgmt |= WPA_DRIVER_CAPA_KEY_MGMT_SUITE_B;
311 			break;
312 		case RSN_AUTH_KEY_MGMT_802_1X_SUITE_B_192:
313 			key_mgmt |= WPA_DRIVER_CAPA_KEY_MGMT_SUITE_B_192;
314 			break;
315 		case RSN_AUTH_KEY_MGMT_OWE:
316 			key_mgmt |= WPA_DRIVER_CAPA_KEY_MGMT_OWE;
317 			break;
318 		case RSN_AUTH_KEY_MGMT_DPP:
319 			key_mgmt |= WPA_DRIVER_CAPA_KEY_MGMT_DPP;
320 			break;
321 		case RSN_AUTH_KEY_MGMT_FILS_SHA256:
322 			key_mgmt |= WPA_DRIVER_CAPA_KEY_MGMT_FILS_SHA256;
323 			break;
324 		case RSN_AUTH_KEY_MGMT_FILS_SHA384:
325 			key_mgmt |= WPA_DRIVER_CAPA_KEY_MGMT_FILS_SHA384;
326 			break;
327 		case RSN_AUTH_KEY_MGMT_FT_FILS_SHA256:
328 			key_mgmt |= WPA_DRIVER_CAPA_KEY_MGMT_FT_FILS_SHA256;
329 			break;
330 		case RSN_AUTH_KEY_MGMT_FT_FILS_SHA384:
331 			key_mgmt |= WPA_DRIVER_CAPA_KEY_MGMT_FT_FILS_SHA384;
332 			break;
333 		case RSN_AUTH_KEY_MGMT_SAE:
334 			key_mgmt |= WPA_DRIVER_CAPA_KEY_MGMT_SAE;
335 			break;
336 		case RSN_AUTH_KEY_MGMT_SAE_EXT_KEY:
337 			key_mgmt |= WPA_DRIVER_CAPA_KEY_MGMT_SAE_EXT_KEY;
338 			break;
339 		}
340 	}
341 
342 	return key_mgmt;
343 }
344 
345 
get_iface_akm_suites_info(struct wiphy_info_data * info,struct nlattr * nl_akms)346 static void get_iface_akm_suites_info(struct wiphy_info_data *info,
347 					struct nlattr *nl_akms)
348 {
349 	struct nlattr *tb[NL80211_IFTYPE_AKM_ATTR_MAX + 1];
350 	struct nlattr *nl_iftype;
351 	unsigned int key_mgmt;
352 	int i;
353 
354 	if (!nl_akms)
355 		return;
356 
357 	nla_parse(tb, NL80211_IFTYPE_AKM_ATTR_MAX,
358 		  nla_data(nl_akms), nla_len(nl_akms), NULL);
359 
360 	if (!tb[NL80211_IFTYPE_AKM_ATTR_IFTYPES] ||
361 	    !tb[NL80211_IFTYPE_AKM_ATTR_SUITES])
362 		return;
363 
364 	info->has_key_mgmt_iftype = 1;
365 	key_mgmt = get_akm_suites_info(tb[NL80211_IFTYPE_AKM_ATTR_SUITES]);
366 
367 	nla_for_each_nested(nl_iftype, tb[NL80211_IFTYPE_AKM_ATTR_IFTYPES], i) {
368 		switch (nla_type(nl_iftype)) {
369 		case NL80211_IFTYPE_ADHOC:
370 			info->drv->capa.key_mgmt_iftype[WPA_IF_IBSS] = key_mgmt;
371 			break;
372 		case NL80211_IFTYPE_STATION:
373 			info->drv->capa.key_mgmt_iftype[WPA_IF_STATION] =
374 				key_mgmt;
375 			break;
376 		case NL80211_IFTYPE_AP:
377 			info->drv->capa.key_mgmt_iftype[WPA_IF_AP_BSS] =
378 				key_mgmt;
379 			break;
380 		case NL80211_IFTYPE_AP_VLAN:
381 			info->drv->capa.key_mgmt_iftype[WPA_IF_AP_VLAN] =
382 				key_mgmt;
383 			break;
384 		case NL80211_IFTYPE_MESH_POINT:
385 			info->drv->capa.key_mgmt_iftype[WPA_IF_MESH] = key_mgmt;
386 			break;
387 		case NL80211_IFTYPE_P2P_CLIENT:
388 			info->drv->capa.key_mgmt_iftype[WPA_IF_P2P_CLIENT] =
389 				key_mgmt;
390 			break;
391 		case NL80211_IFTYPE_P2P_GO:
392 			info->drv->capa.key_mgmt_iftype[WPA_IF_P2P_GO] =
393 				key_mgmt;
394 			break;
395 		case NL80211_IFTYPE_P2P_DEVICE:
396 			info->drv->capa.key_mgmt_iftype[WPA_IF_P2P_DEVICE] =
397 				key_mgmt;
398 			break;
399 		case NL80211_IFTYPE_NAN:
400 			info->drv->capa.key_mgmt_iftype[WPA_IF_NAN] = key_mgmt;
401 			break;
402 		}
403 		wpa_printf(MSG_DEBUG, "nl80211: %s supported key_mgmt 0x%x",
404 			   nl80211_iftype_str(nla_type(nl_iftype)),
405 			   key_mgmt);
406 	}
407 }
408 
409 
wiphy_info_iftype_akm_suites(struct wiphy_info_data * info,struct nlattr * tb)410 static void wiphy_info_iftype_akm_suites(struct wiphy_info_data *info,
411 					 struct nlattr *tb)
412 {
413 	struct nlattr *nl_if;
414 	int rem_if;
415 
416 	if (!tb)
417 		return;
418 
419 	nla_for_each_nested(nl_if, tb, rem_if)
420 		get_iface_akm_suites_info(info, nl_if);
421 }
422 
423 
wiphy_info_akm_suites(struct wiphy_info_data * info,struct nlattr * tb)424 static void wiphy_info_akm_suites(struct wiphy_info_data *info,
425 				  struct nlattr *tb)
426 {
427 	if (!tb)
428 		return;
429 
430 	info->has_key_mgmt = 1;
431 	info->capa->key_mgmt = get_akm_suites_info(tb);
432 	wpa_printf(MSG_DEBUG, "nl80211: wiphy supported key_mgmt 0x%x",
433 		   info->capa->key_mgmt);
434 }
435 
436 
wiphy_info_cipher_suites(struct wiphy_info_data * info,struct nlattr * tb)437 static void wiphy_info_cipher_suites(struct wiphy_info_data *info,
438 				     struct nlattr *tb)
439 {
440 	int i, num;
441 	u32 *ciphers;
442 
443 	if (tb == NULL)
444 		return;
445 
446 	num = nla_len(tb) / sizeof(u32);
447 	ciphers = nla_data(tb);
448 	for (i = 0; i < num; i++) {
449 		u32 c = ciphers[i];
450 
451 		wpa_printf(MSG_DEBUG, "nl80211: Supported cipher %02x-%02x-%02x:%d",
452 			   c >> 24, (c >> 16) & 0xff,
453 			   (c >> 8) & 0xff, c & 0xff);
454 		switch (c) {
455 		case RSN_CIPHER_SUITE_CCMP_256:
456 			info->capa->enc |= WPA_DRIVER_CAPA_ENC_CCMP_256;
457 			break;
458 		case RSN_CIPHER_SUITE_GCMP_256:
459 			info->capa->enc |= WPA_DRIVER_CAPA_ENC_GCMP_256;
460 			break;
461 		case RSN_CIPHER_SUITE_CCMP:
462 			info->capa->enc |= WPA_DRIVER_CAPA_ENC_CCMP;
463 			break;
464 		case RSN_CIPHER_SUITE_GCMP:
465 			info->capa->enc |= WPA_DRIVER_CAPA_ENC_GCMP;
466 			break;
467 		case RSN_CIPHER_SUITE_TKIP:
468 			info->capa->enc |= WPA_DRIVER_CAPA_ENC_TKIP;
469 			break;
470 		case RSN_CIPHER_SUITE_WEP104:
471 			info->capa->enc |= WPA_DRIVER_CAPA_ENC_WEP104;
472 			break;
473 		case RSN_CIPHER_SUITE_WEP40:
474 			info->capa->enc |= WPA_DRIVER_CAPA_ENC_WEP40;
475 			break;
476 		case RSN_CIPHER_SUITE_AES_128_CMAC:
477 			info->capa->enc |= WPA_DRIVER_CAPA_ENC_BIP;
478 			break;
479 		case RSN_CIPHER_SUITE_BIP_GMAC_128:
480 			info->capa->enc |= WPA_DRIVER_CAPA_ENC_BIP_GMAC_128;
481 			break;
482 		case RSN_CIPHER_SUITE_BIP_GMAC_256:
483 			info->capa->enc |= WPA_DRIVER_CAPA_ENC_BIP_GMAC_256;
484 			break;
485 		case RSN_CIPHER_SUITE_BIP_CMAC_256:
486 			info->capa->enc |= WPA_DRIVER_CAPA_ENC_BIP_CMAC_256;
487 			break;
488 		case RSN_CIPHER_SUITE_NO_GROUP_ADDRESSED:
489 			info->capa->enc |= WPA_DRIVER_CAPA_ENC_GTK_NOT_USED;
490 			break;
491 		}
492 	}
493 }
494 
495 
wiphy_info_max_roc(struct wpa_driver_capa * capa,struct nlattr * tb)496 static void wiphy_info_max_roc(struct wpa_driver_capa *capa,
497 			       struct nlattr *tb)
498 {
499 	if (tb)
500 		capa->max_remain_on_chan = nla_get_u32(tb);
501 }
502 
503 
wiphy_info_tdls(struct wpa_driver_capa * capa,struct nlattr * tdls,struct nlattr * ext_setup)504 static void wiphy_info_tdls(struct wpa_driver_capa *capa, struct nlattr *tdls,
505 			    struct nlattr *ext_setup)
506 {
507 	if (tdls == NULL)
508 		return;
509 
510 	wpa_printf(MSG_DEBUG, "nl80211: TDLS supported");
511 	capa->flags |= WPA_DRIVER_FLAGS_TDLS_SUPPORT;
512 
513 	if (ext_setup) {
514 		wpa_printf(MSG_DEBUG, "nl80211: TDLS external setup");
515 		capa->flags |= WPA_DRIVER_FLAGS_TDLS_EXTERNAL_SETUP;
516 	}
517 }
518 
519 
ext_feature_isset(const u8 * ext_features,int ext_features_len,enum nl80211_ext_feature_index ftidx)520 static int ext_feature_isset(const u8 *ext_features, int ext_features_len,
521 			     enum nl80211_ext_feature_index ftidx)
522 {
523 	u8 ft_byte;
524 
525 	if ((int) ftidx / 8 >= ext_features_len)
526 		return 0;
527 
528 	ft_byte = ext_features[ftidx / 8];
529 	return (ft_byte & BIT(ftidx % 8)) != 0;
530 }
531 
532 
wiphy_info_ext_feature_flags(struct wiphy_info_data * info,struct nlattr * tb)533 static void wiphy_info_ext_feature_flags(struct wiphy_info_data *info,
534 					 struct nlattr *tb)
535 {
536 	struct wpa_driver_capa *capa = info->capa;
537 	u8 *ext_features;
538 	int len;
539 
540 	if (tb == NULL)
541 		return;
542 
543 	ext_features = nla_data(tb);
544 	len = nla_len(tb);
545 
546 	if (ext_feature_isset(ext_features, len, NL80211_EXT_FEATURE_VHT_IBSS))
547 		capa->flags |= WPA_DRIVER_FLAGS_VHT_IBSS;
548 
549 	if (ext_feature_isset(ext_features, len, NL80211_EXT_FEATURE_RRM))
550 		capa->rrm_flags |= WPA_DRIVER_FLAGS_SUPPORT_RRM;
551 
552 	if (ext_feature_isset(ext_features, len, NL80211_EXT_FEATURE_FILS_STA))
553 		capa->flags |= WPA_DRIVER_FLAGS_SUPPORT_FILS;
554 
555 	if (ext_feature_isset(ext_features, len,
556 			      NL80211_EXT_FEATURE_BEACON_RATE_LEGACY))
557 		capa->flags |= WPA_DRIVER_FLAGS_BEACON_RATE_LEGACY;
558 
559 	if (ext_feature_isset(ext_features, len,
560 			      NL80211_EXT_FEATURE_BEACON_RATE_HT))
561 		capa->flags |= WPA_DRIVER_FLAGS_BEACON_RATE_HT;
562 
563 	if (ext_feature_isset(ext_features, len,
564 			      NL80211_EXT_FEATURE_BEACON_RATE_VHT))
565 		capa->flags |= WPA_DRIVER_FLAGS_BEACON_RATE_VHT;
566 
567 	if (ext_feature_isset(ext_features, len,
568 			      NL80211_EXT_FEATURE_BEACON_RATE_HE))
569 		capa->flags2 |= WPA_DRIVER_FLAGS2_BEACON_RATE_HE;
570 
571 	if (ext_feature_isset(ext_features, len,
572 			      NL80211_EXT_FEATURE_SET_SCAN_DWELL))
573 		capa->rrm_flags |= WPA_DRIVER_FLAGS_SUPPORT_SET_SCAN_DWELL;
574 
575 	if (ext_feature_isset(ext_features, len,
576 			      NL80211_EXT_FEATURE_SCAN_START_TIME) &&
577 	    ext_feature_isset(ext_features, len,
578 			      NL80211_EXT_FEATURE_BSS_PARENT_TSF) &&
579 	    ext_feature_isset(ext_features, len,
580 			      NL80211_EXT_FEATURE_SET_SCAN_DWELL))
581 		capa->rrm_flags |= WPA_DRIVER_FLAGS_SUPPORT_BEACON_REPORT;
582 	if (ext_feature_isset(ext_features, len,
583 			      NL80211_EXT_FEATURE_MGMT_TX_RANDOM_TA))
584 		capa->flags |= WPA_DRIVER_FLAGS_MGMT_TX_RANDOM_TA;
585 	if (ext_feature_isset(ext_features, len,
586 			      NL80211_EXT_FEATURE_MGMT_TX_RANDOM_TA_CONNECTED))
587 		capa->flags |= WPA_DRIVER_FLAGS_MGMT_TX_RANDOM_TA_CONNECTED;
588 	if (ext_feature_isset(ext_features, len,
589 			      NL80211_EXT_FEATURE_SCHED_SCAN_RELATIVE_RSSI))
590 		capa->flags |= WPA_DRIVER_FLAGS_SCHED_SCAN_RELATIVE_RSSI;
591 	if (ext_feature_isset(ext_features, len,
592 			      NL80211_EXT_FEATURE_FILS_SK_OFFLOAD))
593 		capa->flags |= WPA_DRIVER_FLAGS_FILS_SK_OFFLOAD;
594 
595 	if (ext_feature_isset(ext_features, len,
596 			      NL80211_EXT_FEATURE_4WAY_HANDSHAKE_STA_PSK))
597 		capa->flags |= WPA_DRIVER_FLAGS_4WAY_HANDSHAKE_PSK;
598 	if (ext_feature_isset(ext_features, len,
599 			      NL80211_EXT_FEATURE_4WAY_HANDSHAKE_STA_1X))
600 		capa->flags |= WPA_DRIVER_FLAGS_4WAY_HANDSHAKE_8021X;
601 
602 	if (ext_feature_isset(ext_features, len,
603 			      NL80211_EXT_FEATURE_SAE_OFFLOAD))
604 		capa->flags2 |= WPA_DRIVER_FLAGS2_SAE_OFFLOAD_STA;
605 
606 	if (ext_feature_isset(ext_features, len,
607 			      NL80211_EXT_FEATURE_MFP_OPTIONAL))
608 		capa->flags |= WPA_DRIVER_FLAGS_MFP_OPTIONAL;
609 
610 	if (ext_feature_isset(ext_features, len,
611 			      NL80211_EXT_FEATURE_DFS_OFFLOAD))
612 		capa->flags |= WPA_DRIVER_FLAGS_DFS_OFFLOAD;
613 
614 #ifdef CONFIG_MBO
615 	if (ext_feature_isset(ext_features, len,
616 			      NL80211_EXT_FEATURE_FILS_MAX_CHANNEL_TIME) &&
617 	    ext_feature_isset(ext_features, len,
618 			      NL80211_EXT_FEATURE_ACCEPT_BCAST_PROBE_RESP) &&
619 	    ext_feature_isset(ext_features, len,
620 			      NL80211_EXT_FEATURE_OCE_PROBE_REQ_HIGH_TX_RATE) &&
621 	    ext_feature_isset(
622 		    ext_features, len,
623 		    NL80211_EXT_FEATURE_OCE_PROBE_REQ_DEFERRAL_SUPPRESSION))
624 		capa->flags |= WPA_DRIVER_FLAGS_OCE_STA;
625 #endif /* CONFIG_MBO */
626 
627 	if (ext_feature_isset(ext_features, len,
628 			      NL80211_EXT_FEATURE_ENABLE_FTM_RESPONDER))
629 		capa->flags |= WPA_DRIVER_FLAGS_FTM_RESPONDER;
630 
631 	if (ext_feature_isset(ext_features, len,
632 			      NL80211_EXT_FEATURE_CONTROL_PORT_OVER_NL80211))
633 		capa->flags |= WPA_DRIVER_FLAGS_CONTROL_PORT;
634 	if (ext_feature_isset(ext_features, len,
635 			      NL80211_EXT_FEATURE_CONTROL_PORT_NO_PREAUTH))
636 		capa->flags2 |= WPA_DRIVER_FLAGS2_CONTROL_PORT_RX;
637 	if (ext_feature_isset(
638 		    ext_features, len,
639 		    NL80211_EXT_FEATURE_CONTROL_PORT_OVER_NL80211_TX_STATUS))
640 		capa->flags2 |= WPA_DRIVER_FLAGS2_CONTROL_PORT_TX_STATUS;
641 
642 	if (ext_feature_isset(ext_features, len,
643 			      NL80211_EXT_FEATURE_VLAN_OFFLOAD))
644 		capa->flags |= WPA_DRIVER_FLAGS_VLAN_OFFLOAD;
645 
646 	if (ext_feature_isset(ext_features, len,
647 			      NL80211_EXT_FEATURE_CAN_REPLACE_PTK0))
648 		capa->flags |= WPA_DRIVER_FLAGS_SAFE_PTK0_REKEYS;
649 
650 	if (ext_feature_isset(ext_features, len,
651 			      NL80211_EXT_FEATURE_BEACON_PROTECTION))
652 		capa->flags |= WPA_DRIVER_FLAGS_BEACON_PROTECTION;
653 
654 	if (ext_feature_isset(ext_features, len,
655 			      NL80211_EXT_FEATURE_EXT_KEY_ID))
656 		capa->flags |= WPA_DRIVER_FLAGS_EXTENDED_KEY_ID;
657 
658 	if (ext_feature_isset(ext_features, len,
659 			      NL80211_EXT_FEATURE_MULTICAST_REGISTRATIONS))
660 		info->drv->multicast_registrations = 1;
661 
662 	if (ext_feature_isset(ext_features, len,
663 			      NL80211_EXT_FEATURE_FILS_DISCOVERY))
664 		info->drv->fils_discovery = 1;
665 
666 	if (ext_feature_isset(ext_features, len,
667 			      NL80211_EXT_FEATURE_UNSOL_BCAST_PROBE_RESP))
668 		info->drv->unsol_bcast_probe_resp = 1;
669 
670 	if (ext_feature_isset(ext_features, len, NL80211_EXT_FEATURE_PUNCT))
671 		info->drv->puncturing = 1;
672 
673 	if (ext_feature_isset(ext_features, len,
674 			      NL80211_EXT_FEATURE_BEACON_PROTECTION_CLIENT))
675 		capa->flags2 |= WPA_DRIVER_FLAGS2_BEACON_PROTECTION_CLIENT;
676 
677 	if (ext_feature_isset(ext_features, len,
678 			      NL80211_EXT_FEATURE_OPERATING_CHANNEL_VALIDATION))
679 		capa->flags2 |= WPA_DRIVER_FLAGS2_OCV;
680 
681 	if (ext_feature_isset(ext_features, len,
682 			      NL80211_EXT_FEATURE_RADAR_BACKGROUND))
683 		capa->flags2 |= WPA_DRIVER_FLAGS2_RADAR_BACKGROUND;
684 
685 	if (ext_feature_isset(ext_features, len,
686 			      NL80211_EXT_FEATURE_SECURE_LTF)) {
687 		capa->flags2 |= WPA_DRIVER_FLAGS2_SEC_LTF_STA;
688 		capa->flags2 |= WPA_DRIVER_FLAGS2_SEC_LTF_AP;
689 	}
690 
691 	if (ext_feature_isset(ext_features, len,
692 			      NL80211_EXT_FEATURE_SECURE_RTT)) {
693 		capa->flags2 |= WPA_DRIVER_FLAGS2_SEC_RTT_STA;
694 		capa->flags2 |= WPA_DRIVER_FLAGS2_SEC_RTT_AP;
695 	}
696 
697 	if (ext_feature_isset(
698 		    ext_features, len,
699 		    NL80211_EXT_FEATURE_PROT_RANGE_NEGO_AND_MEASURE)) {
700 		capa->flags2 |= WPA_DRIVER_FLAGS2_PROT_RANGE_NEG_STA;
701 		capa->flags2 |= WPA_DRIVER_FLAGS2_PROT_RANGE_NEG_AP;
702 	}
703 
704 	if (ext_feature_isset(ext_features, len,
705 			      NL80211_EXT_FEATURE_SCAN_MIN_PREQ_CONTENT))
706 		capa->flags2 |= WPA_DRIVER_FLAGS2_SCAN_MIN_PREQ;
707 
708 	if (ext_feature_isset(ext_features, len,
709 			      NL80211_EXT_FEATURE_4WAY_HANDSHAKE_AP_PSK))
710 		capa->flags2 |= WPA_DRIVER_FLAGS2_4WAY_HANDSHAKE_AP_PSK;
711 
712 	if (ext_feature_isset(ext_features, len,
713 			      NL80211_EXT_FEATURE_OWE_OFFLOAD))
714 		capa->flags2 |= WPA_DRIVER_FLAGS2_OWE_OFFLOAD_STA;
715 
716 	if (ext_feature_isset(ext_features, len,
717 			      NL80211_EXT_FEATURE_OWE_OFFLOAD_AP))
718 		capa->flags2 |= WPA_DRIVER_FLAGS2_OWE_OFFLOAD_AP;
719 
720 	if (ext_feature_isset(ext_features, len,
721 			      NL80211_EXT_FEATURE_SAE_OFFLOAD_AP))
722 		capa->flags2 |= WPA_DRIVER_FLAGS2_SAE_OFFLOAD_AP;
723 }
724 
725 
wiphy_info_feature_flags(struct wiphy_info_data * info,struct nlattr * tb)726 static void wiphy_info_feature_flags(struct wiphy_info_data *info,
727 				     struct nlattr *tb)
728 {
729 	u32 flags;
730 	struct wpa_driver_capa *capa = info->capa;
731 
732 	if (tb == NULL)
733 		return;
734 
735 	flags = nla_get_u32(tb);
736 
737 	if (flags & NL80211_FEATURE_SK_TX_STATUS)
738 		info->data_tx_status = 1;
739 
740 	if (flags & NL80211_FEATURE_INACTIVITY_TIMER)
741 		capa->flags |= WPA_DRIVER_FLAGS_INACTIVITY_TIMER;
742 
743 	if (flags & NL80211_FEATURE_SAE)
744 		capa->flags |= WPA_DRIVER_FLAGS_SAE;
745 
746 	if (flags & NL80211_FEATURE_NEED_OBSS_SCAN)
747 		capa->flags |= WPA_DRIVER_FLAGS_OBSS_SCAN;
748 
749 	if (flags & NL80211_FEATURE_AP_MODE_CHAN_WIDTH_CHANGE)
750 		capa->flags |= WPA_DRIVER_FLAGS_HT_2040_COEX;
751 
752 	if (flags & NL80211_FEATURE_TDLS_CHANNEL_SWITCH) {
753 		wpa_printf(MSG_DEBUG, "nl80211: TDLS channel switch");
754 		capa->flags |= WPA_DRIVER_FLAGS_TDLS_CHANNEL_SWITCH;
755 	}
756 
757 	if (flags & NL80211_FEATURE_P2P_GO_CTWIN)
758 		info->p2p_go_ctwindow_supported = 1;
759 
760 	if (flags & NL80211_FEATURE_LOW_PRIORITY_SCAN)
761 		info->have_low_prio_scan = 1;
762 
763 	if (flags & NL80211_FEATURE_SCAN_RANDOM_MAC_ADDR)
764 		info->mac_addr_rand_scan_supported = 1;
765 
766 	if (flags & NL80211_FEATURE_SCHED_SCAN_RANDOM_MAC_ADDR)
767 		info->mac_addr_rand_sched_scan_supported = 1;
768 
769 	if (flags & NL80211_FEATURE_SUPPORTS_WMM_ADMISSION)
770 		info->wmm_ac_supported = 1;
771 
772 	if (flags & NL80211_FEATURE_DS_PARAM_SET_IE_IN_PROBES)
773 		capa->rrm_flags |= WPA_DRIVER_FLAGS_DS_PARAM_SET_IE_IN_PROBES;
774 
775 	if (flags & NL80211_FEATURE_WFA_TPC_IE_IN_PROBES)
776 		capa->rrm_flags |= WPA_DRIVER_FLAGS_WFA_TPC_IE_IN_PROBES;
777 
778 	if (flags & NL80211_FEATURE_QUIET)
779 		capa->rrm_flags |= WPA_DRIVER_FLAGS_QUIET;
780 
781 	if (flags & NL80211_FEATURE_TX_POWER_INSERTION)
782 		capa->rrm_flags |= WPA_DRIVER_FLAGS_TX_POWER_INSERTION;
783 
784 	if (flags & NL80211_FEATURE_HT_IBSS)
785 		capa->flags |= WPA_DRIVER_FLAGS_HT_IBSS;
786 
787 	if (flags & NL80211_FEATURE_FULL_AP_CLIENT_STATE)
788 		capa->flags |= WPA_DRIVER_FLAGS_FULL_AP_CLIENT_STATE;
789 }
790 
791 
wiphy_info_probe_resp_offload(struct wpa_driver_capa * capa,struct nlattr * tb)792 static void wiphy_info_probe_resp_offload(struct wpa_driver_capa *capa,
793 					  struct nlattr *tb)
794 {
795 	u32 protocols;
796 
797 	if (tb == NULL)
798 		return;
799 
800 	protocols = nla_get_u32(tb);
801 	wpa_printf(MSG_DEBUG, "nl80211: Supports Probe Response offload in AP "
802 		   "mode");
803 	capa->flags |= WPA_DRIVER_FLAGS_PROBE_RESP_OFFLOAD;
804 	capa->probe_resp_offloads = probe_resp_offload_support(protocols);
805 }
806 
807 
wiphy_info_wowlan_triggers(struct wpa_driver_capa * capa,struct nlattr * tb)808 static void wiphy_info_wowlan_triggers(struct wpa_driver_capa *capa,
809 				       struct nlattr *tb)
810 {
811 	struct nlattr *triggers[MAX_NL80211_WOWLAN_TRIG + 1];
812 
813 	if (tb == NULL)
814 		return;
815 
816 	if (nla_parse_nested(triggers, MAX_NL80211_WOWLAN_TRIG,
817 			     tb, NULL))
818 		return;
819 
820 	if (triggers[NL80211_WOWLAN_TRIG_ANY])
821 		capa->wowlan_triggers.any = 1;
822 	if (triggers[NL80211_WOWLAN_TRIG_DISCONNECT])
823 		capa->wowlan_triggers.disconnect = 1;
824 	if (triggers[NL80211_WOWLAN_TRIG_MAGIC_PKT])
825 		capa->wowlan_triggers.magic_pkt = 1;
826 	if (triggers[NL80211_WOWLAN_TRIG_GTK_REKEY_FAILURE])
827 		capa->wowlan_triggers.gtk_rekey_failure = 1;
828 	if (triggers[NL80211_WOWLAN_TRIG_EAP_IDENT_REQUEST])
829 		capa->wowlan_triggers.eap_identity_req = 1;
830 	if (triggers[NL80211_WOWLAN_TRIG_4WAY_HANDSHAKE])
831 		capa->wowlan_triggers.four_way_handshake = 1;
832 	if (triggers[NL80211_WOWLAN_TRIG_RFKILL_RELEASE])
833 		capa->wowlan_triggers.rfkill_release = 1;
834 }
835 
836 
wiphy_info_extended_capab(struct wpa_driver_nl80211_data * drv,struct nlattr * tb)837 static void wiphy_info_extended_capab(struct wpa_driver_nl80211_data *drv,
838 				      struct nlattr *tb)
839 {
840 	int rem = 0, i;
841 	struct nlattr *tb1[NL80211_ATTR_MAX + 1], *attr;
842 
843 	if (!tb || drv->num_iface_capa == NL80211_IFTYPE_MAX)
844 		return;
845 
846 	nla_for_each_nested(attr, tb, rem) {
847 		unsigned int len;
848 		struct drv_nl80211_iface_capa *capa;
849 
850 		nla_parse(tb1, NL80211_ATTR_MAX, nla_data(attr),
851 			  nla_len(attr), NULL);
852 
853 		if (!tb1[NL80211_ATTR_IFTYPE] ||
854 		    !tb1[NL80211_ATTR_EXT_CAPA] ||
855 		    !tb1[NL80211_ATTR_EXT_CAPA_MASK])
856 			continue;
857 
858 		capa = &drv->iface_capa[drv->num_iface_capa];
859 		capa->iftype = nla_get_u32(tb1[NL80211_ATTR_IFTYPE]);
860 		wpa_printf(MSG_DEBUG,
861 			   "nl80211: Driver-advertised extended capabilities for interface type %s",
862 			   nl80211_iftype_str(capa->iftype));
863 
864 		len = nla_len(tb1[NL80211_ATTR_EXT_CAPA]);
865 		capa->ext_capa = os_memdup(nla_data(tb1[NL80211_ATTR_EXT_CAPA]),
866 					   len);
867 		if (!capa->ext_capa)
868 			goto err;
869 
870 		capa->ext_capa_len = len;
871 		wpa_hexdump(MSG_DEBUG, "nl80211: Extended capabilities",
872 			    capa->ext_capa, capa->ext_capa_len);
873 
874 		len = nla_len(tb1[NL80211_ATTR_EXT_CAPA_MASK]);
875 		capa->ext_capa_mask =
876 			os_memdup(nla_data(tb1[NL80211_ATTR_EXT_CAPA_MASK]),
877 				  len);
878 		if (!capa->ext_capa_mask)
879 			goto err;
880 
881 		wpa_hexdump(MSG_DEBUG, "nl80211: Extended capabilities mask",
882 			    capa->ext_capa_mask, capa->ext_capa_len);
883 
884 		if (tb1[NL80211_ATTR_EML_CAPABILITY] &&
885 		    tb1[NL80211_ATTR_MLD_CAPA_AND_OPS]) {
886 			capa->eml_capa =
887 				nla_get_u16(tb1[NL80211_ATTR_EML_CAPABILITY]);
888 			capa->mld_capa_and_ops =
889 				nla_get_u16(tb1[NL80211_ATTR_MLD_CAPA_AND_OPS]);
890 		}
891 
892 		wpa_printf(MSG_DEBUG,
893 			   "nl80211: EML Capability: 0x%x MLD Capability: 0x%x",
894 			   capa->eml_capa, capa->mld_capa_and_ops);
895 
896 		drv->num_iface_capa++;
897 		if (drv->num_iface_capa == NL80211_IFTYPE_MAX)
898 			break;
899 	}
900 
901 	return;
902 
903 err:
904 	/* Cleanup allocated memory on error */
905 	for (i = 0; i < NL80211_IFTYPE_MAX; i++) {
906 		os_free(drv->iface_capa[i].ext_capa);
907 		drv->iface_capa[i].ext_capa = NULL;
908 		os_free(drv->iface_capa[i].ext_capa_mask);
909 		drv->iface_capa[i].ext_capa_mask = NULL;
910 		drv->iface_capa[i].ext_capa_len = 0;
911 	}
912 	drv->num_iface_capa = 0;
913 }
914 
915 
wiphy_info_mbssid(struct wpa_driver_capa * cap,struct nlattr * attr)916 static void wiphy_info_mbssid(struct wpa_driver_capa *cap, struct nlattr *attr)
917 {
918 	struct nlattr *config[NL80211_MBSSID_CONFIG_ATTR_MAX + 1];
919 
920 	if (nla_parse_nested(config, NL80211_MBSSID_CONFIG_ATTR_MAX, attr,
921 			     NULL))
922 		return;
923 
924 	if (!config[NL80211_MBSSID_CONFIG_ATTR_MAX_INTERFACES])
925 		return;
926 
927 	cap->mbssid_max_interfaces =
928 		nla_get_u8(config[NL80211_MBSSID_CONFIG_ATTR_MAX_INTERFACES]);
929 
930 	if (config[NL80211_MBSSID_CONFIG_ATTR_MAX_EMA_PROFILE_PERIODICITY])
931 		cap->ema_max_periodicity =
932 			nla_get_u8(config[NL80211_MBSSID_CONFIG_ATTR_MAX_EMA_PROFILE_PERIODICITY]);
933 
934 	wpa_printf(MSG_DEBUG,
935 		   "mbssid: max interfaces %u, max profile periodicity %u",
936 		   cap->mbssid_max_interfaces, cap->ema_max_periodicity);
937 }
938 
939 
wiphy_info_handler(struct nl_msg * msg,void * arg)940 static int wiphy_info_handler(struct nl_msg *msg, void *arg)
941 {
942 	struct nlattr *tb[NL80211_ATTR_MAX + 1];
943 	struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
944 	struct wiphy_info_data *info = arg;
945 	struct wpa_driver_capa *capa = info->capa;
946 	struct wpa_driver_nl80211_data *drv = info->drv;
947 
948 	nla_parse(tb, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
949 		  genlmsg_attrlen(gnlh, 0), NULL);
950 
951 	if (tb[NL80211_ATTR_WIPHY])
952 		drv->wiphy_idx = nla_get_u32(tb[NL80211_ATTR_WIPHY]);
953 
954 	if (tb[NL80211_ATTR_WIPHY_NAME])
955 		os_strlcpy(drv->phyname,
956 			   nla_get_string(tb[NL80211_ATTR_WIPHY_NAME]),
957 			   sizeof(drv->phyname));
958 	if (tb[NL80211_ATTR_MAX_NUM_SCAN_SSIDS])
959 		capa->max_scan_ssids =
960 			nla_get_u8(tb[NL80211_ATTR_MAX_NUM_SCAN_SSIDS]);
961 
962 	if (tb[NL80211_ATTR_MAX_NUM_SCHED_SCAN_SSIDS])
963 		capa->max_sched_scan_ssids =
964 			nla_get_u8(tb[NL80211_ATTR_MAX_NUM_SCHED_SCAN_SSIDS]);
965 
966 	if (tb[NL80211_ATTR_MAX_NUM_SCHED_SCAN_PLANS] &&
967 	    tb[NL80211_ATTR_MAX_SCAN_PLAN_INTERVAL] &&
968 	    tb[NL80211_ATTR_MAX_SCAN_PLAN_ITERATIONS]) {
969 		capa->max_sched_scan_plans =
970 			nla_get_u32(tb[NL80211_ATTR_MAX_NUM_SCHED_SCAN_PLANS]);
971 
972 		capa->max_sched_scan_plan_interval =
973 			nla_get_u32(tb[NL80211_ATTR_MAX_SCAN_PLAN_INTERVAL]);
974 
975 		capa->max_sched_scan_plan_iterations =
976 			nla_get_u32(tb[NL80211_ATTR_MAX_SCAN_PLAN_ITERATIONS]);
977 	}
978 
979 	if (tb[NL80211_ATTR_MAX_MATCH_SETS])
980 		capa->max_match_sets =
981 			nla_get_u8(tb[NL80211_ATTR_MAX_MATCH_SETS]);
982 
983 	if (tb[NL80211_ATTR_MAC_ACL_MAX])
984 		capa->max_acl_mac_addrs =
985 			nla_get_u32(tb[NL80211_ATTR_MAC_ACL_MAX]);
986 
987 	wiphy_info_supported_iftypes(info, tb[NL80211_ATTR_SUPPORTED_IFTYPES]);
988 	wiphy_info_iface_comb(info, tb[NL80211_ATTR_INTERFACE_COMBINATIONS]);
989 	wiphy_info_supp_cmds(info, tb[NL80211_ATTR_SUPPORTED_COMMANDS]);
990 	wiphy_info_cipher_suites(info, tb[NL80211_ATTR_CIPHER_SUITES]);
991 	wiphy_info_akm_suites(info, tb[NL80211_ATTR_AKM_SUITES]);
992 	wiphy_info_iftype_akm_suites(info, tb[NL80211_ATTR_IFTYPE_AKM_SUITES]);
993 
994 	if (tb[NL80211_ATTR_OFFCHANNEL_TX_OK]) {
995 		wpa_printf(MSG_DEBUG, "nl80211: Using driver-based "
996 			   "off-channel TX");
997 		capa->flags |= WPA_DRIVER_FLAGS_OFFCHANNEL_TX;
998 	}
999 
1000 	if (tb[NL80211_ATTR_ROAM_SUPPORT]) {
1001 		wpa_printf(MSG_DEBUG, "nl80211: Using driver-based roaming");
1002 		capa->flags |= WPA_DRIVER_FLAGS_BSS_SELECTION;
1003 	}
1004 
1005 	wiphy_info_max_roc(capa,
1006 			   tb[NL80211_ATTR_MAX_REMAIN_ON_CHANNEL_DURATION]);
1007 
1008 	if (tb[NL80211_ATTR_SUPPORT_AP_UAPSD])
1009 		capa->flags |= WPA_DRIVER_FLAGS_AP_UAPSD;
1010 
1011 	wiphy_info_tdls(capa, tb[NL80211_ATTR_TDLS_SUPPORT],
1012 			tb[NL80211_ATTR_TDLS_EXTERNAL_SETUP]);
1013 
1014 	if (tb[NL80211_ATTR_DEVICE_AP_SME]) {
1015 		u32 ap_sme_features_flags =
1016 			nla_get_u32(tb[NL80211_ATTR_DEVICE_AP_SME]);
1017 
1018 		if (ap_sme_features_flags & NL80211_AP_SME_SA_QUERY_OFFLOAD)
1019 			capa->flags2 |= WPA_DRIVER_FLAGS2_SA_QUERY_OFFLOAD_AP;
1020 
1021 		info->device_ap_sme = 1;
1022 	}
1023 
1024 	wiphy_info_feature_flags(info, tb[NL80211_ATTR_FEATURE_FLAGS]);
1025 	wiphy_info_ext_feature_flags(info, tb[NL80211_ATTR_EXT_FEATURES]);
1026 	wiphy_info_probe_resp_offload(capa,
1027 				      tb[NL80211_ATTR_PROBE_RESP_OFFLOAD]);
1028 
1029 	if (tb[NL80211_ATTR_EXT_CAPA] && tb[NL80211_ATTR_EXT_CAPA_MASK] &&
1030 	    drv->extended_capa == NULL) {
1031 		drv->extended_capa =
1032 			os_malloc(nla_len(tb[NL80211_ATTR_EXT_CAPA]));
1033 		if (drv->extended_capa) {
1034 			os_memcpy(drv->extended_capa,
1035 				  nla_data(tb[NL80211_ATTR_EXT_CAPA]),
1036 				  nla_len(tb[NL80211_ATTR_EXT_CAPA]));
1037 			drv->extended_capa_len =
1038 				nla_len(tb[NL80211_ATTR_EXT_CAPA]);
1039 			wpa_hexdump(MSG_DEBUG,
1040 				    "nl80211: Driver-advertised extended capabilities (default)",
1041 				    drv->extended_capa, drv->extended_capa_len);
1042 		}
1043 		drv->extended_capa_mask =
1044 			os_malloc(nla_len(tb[NL80211_ATTR_EXT_CAPA_MASK]));
1045 		if (drv->extended_capa_mask) {
1046 			os_memcpy(drv->extended_capa_mask,
1047 				  nla_data(tb[NL80211_ATTR_EXT_CAPA_MASK]),
1048 				  nla_len(tb[NL80211_ATTR_EXT_CAPA_MASK]));
1049 			wpa_hexdump(MSG_DEBUG,
1050 				    "nl80211: Driver-advertised extended capabilities mask (default)",
1051 				    drv->extended_capa_mask,
1052 				    drv->extended_capa_len);
1053 		} else {
1054 			os_free(drv->extended_capa);
1055 			drv->extended_capa = NULL;
1056 			drv->extended_capa_len = 0;
1057 		}
1058 	}
1059 
1060 	wiphy_info_extended_capab(drv, tb[NL80211_ATTR_IFTYPE_EXT_CAPA]);
1061 
1062 	if (tb[NL80211_ATTR_VENDOR_DATA]) {
1063 		struct nlattr *nl;
1064 		int rem;
1065 
1066 		nla_for_each_nested(nl, tb[NL80211_ATTR_VENDOR_DATA], rem) {
1067 			struct nl80211_vendor_cmd_info *vinfo;
1068 			if (nla_len(nl) != sizeof(*vinfo)) {
1069 				wpa_printf(MSG_DEBUG, "nl80211: Unexpected vendor data info");
1070 				continue;
1071 			}
1072 			vinfo = nla_data(nl);
1073 			if (vinfo->vendor_id == OUI_QCA) {
1074 				switch (vinfo->subcmd) {
1075 				case QCA_NL80211_VENDOR_SUBCMD_TEST:
1076 					drv->vendor_cmd_test_avail = 1;
1077 					break;
1078 #ifdef CONFIG_DRIVER_NL80211_QCA
1079 				case QCA_NL80211_VENDOR_SUBCMD_ROAMING:
1080 					drv->roaming_vendor_cmd_avail = 1;
1081 					break;
1082 				case QCA_NL80211_VENDOR_SUBCMD_DFS_CAPABILITY:
1083 					drv->dfs_vendor_cmd_avail = 1;
1084 					break;
1085 				case QCA_NL80211_VENDOR_SUBCMD_GET_FEATURES:
1086 					drv->get_features_vendor_cmd_avail = 1;
1087 					break;
1088 				case QCA_NL80211_VENDOR_SUBCMD_GET_PREFERRED_FREQ_LIST:
1089 					drv->get_pref_freq_list = 1;
1090 					break;
1091 				case QCA_NL80211_VENDOR_SUBCMD_SET_PROBABLE_OPER_CHANNEL:
1092 					drv->set_prob_oper_freq = 1;
1093 					break;
1094 				case QCA_NL80211_VENDOR_SUBCMD_DO_ACS:
1095 					drv->capa.flags |=
1096 						WPA_DRIVER_FLAGS_ACS_OFFLOAD;
1097 					drv->qca_do_acs = 1;
1098 					break;
1099 				case QCA_NL80211_VENDOR_SUBCMD_SETBAND:
1100 					drv->setband_vendor_cmd_avail = 1;
1101 					break;
1102 				case QCA_NL80211_VENDOR_SUBCMD_TRIGGER_SCAN:
1103 					drv->scan_vendor_cmd_avail = 1;
1104 					break;
1105 				case QCA_NL80211_VENDOR_SUBCMD_SET_WIFI_CONFIGURATION:
1106 					drv->set_wifi_conf_vendor_cmd_avail = 1;
1107 					break;
1108 				case QCA_NL80211_VENDOR_SUBCMD_FETCH_BSS_TRANSITION_STATUS:
1109 					drv->fetch_bss_trans_status = 1;
1110 					break;
1111 				case QCA_NL80211_VENDOR_SUBCMD_ROAM:
1112 					drv->roam_vendor_cmd_avail = 1;
1113 					break;
1114 				case QCA_NL80211_VENDOR_SUBCMD_ADD_STA_NODE:
1115 					drv->add_sta_node_vendor_cmd_avail = 1;
1116 					break;
1117 				case QCA_NL80211_VENDOR_SUBCMD_GET_STA_INFO:
1118 					drv->get_sta_info_vendor_cmd_avail = 1;
1119 					break;
1120 				case QCA_NL80211_VENDOR_SUBCMD_SECURE_RANGING_CONTEXT:
1121 					drv->secure_ranging_ctx_vendor_cmd_avail = 1;
1122 					break;
1123 				case QCA_NL80211_VENDOR_SUBCMD_CONNECT_EXT:
1124 					drv->connect_ext_vendor_cmd_avail = 1;
1125 					break;
1126 #endif /* CONFIG_DRIVER_NL80211_QCA */
1127 				}
1128 #ifdef CONFIG_DRIVER_NL80211_BRCM
1129 			} else if (vinfo->vendor_id == OUI_BRCM) {
1130 				switch (vinfo->subcmd) {
1131 				case BRCM_VENDOR_SCMD_ACS:
1132 					drv->capa.flags |=
1133 						WPA_DRIVER_FLAGS_ACS_OFFLOAD;
1134 					wpa_printf(MSG_DEBUG,
1135 						   "Enabled BRCM ACS");
1136 					drv->brcm_do_acs = 1;
1137 					break;
1138 				}
1139 #endif /* CONFIG_DRIVER_NL80211_BRCM */
1140 			}
1141 
1142 			wpa_printf(MSG_DEBUG, "nl80211: Supported vendor command: vendor_id=0x%x subcmd=%u",
1143 				   vinfo->vendor_id, vinfo->subcmd);
1144 		}
1145 	}
1146 
1147 	if (tb[NL80211_ATTR_VENDOR_EVENTS]) {
1148 		struct nlattr *nl;
1149 		int rem;
1150 
1151 		nla_for_each_nested(nl, tb[NL80211_ATTR_VENDOR_EVENTS], rem) {
1152 			struct nl80211_vendor_cmd_info *vinfo;
1153 			if (nla_len(nl) != sizeof(*vinfo)) {
1154 				wpa_printf(MSG_DEBUG, "nl80211: Unexpected vendor data info");
1155 				continue;
1156 			}
1157 			vinfo = nla_data(nl);
1158 			wpa_printf(MSG_DEBUG, "nl80211: Supported vendor event: vendor_id=0x%x subcmd=%u",
1159 				   vinfo->vendor_id, vinfo->subcmd);
1160 		}
1161 	}
1162 
1163 	wiphy_info_wowlan_triggers(capa,
1164 				   tb[NL80211_ATTR_WOWLAN_TRIGGERS_SUPPORTED]);
1165 
1166 	if (tb[NL80211_ATTR_MAX_AP_ASSOC_STA])
1167 		capa->max_stations =
1168 			nla_get_u32(tb[NL80211_ATTR_MAX_AP_ASSOC_STA]);
1169 
1170 	if (tb[NL80211_ATTR_MAX_CSA_COUNTERS])
1171 		capa->max_csa_counters =
1172 			nla_get_u8(tb[NL80211_ATTR_MAX_CSA_COUNTERS]);
1173 
1174 	if (tb[NL80211_ATTR_WIPHY_SELF_MANAGED_REG])
1175 		capa->flags |= WPA_DRIVER_FLAGS_SELF_MANAGED_REGULATORY;
1176 
1177 	if (tb[NL80211_ATTR_MAX_NUM_AKM_SUITES])
1178 		capa->max_num_akms =
1179 			nla_get_u16(tb[NL80211_ATTR_MAX_NUM_AKM_SUITES]);
1180 
1181 	if (tb[NL80211_ATTR_MBSSID_CONFIG])
1182 		wiphy_info_mbssid(capa, tb[NL80211_ATTR_MBSSID_CONFIG]);
1183 
1184 	if (tb[NL80211_ATTR_MLO_SUPPORT])
1185 		capa->flags2 |= WPA_DRIVER_FLAGS2_MLO;
1186 
1187 	return NL_SKIP;
1188 }
1189 
1190 
wpa_driver_nl80211_get_info(struct wpa_driver_nl80211_data * drv,struct wiphy_info_data * info)1191 static int wpa_driver_nl80211_get_info(struct wpa_driver_nl80211_data *drv,
1192 				       struct wiphy_info_data *info)
1193 {
1194 	u32 feat;
1195 	struct nl_msg *msg;
1196 	int flags = 0;
1197 
1198 	os_memset(info, 0, sizeof(*info));
1199 	info->capa = &drv->capa;
1200 	info->drv = drv;
1201 
1202 	feat = get_nl80211_protocol_features(drv);
1203 	if (feat & NL80211_PROTOCOL_FEATURE_SPLIT_WIPHY_DUMP)
1204 		flags = NLM_F_DUMP;
1205 	msg = nl80211_cmd_msg(drv->first_bss, flags, NL80211_CMD_GET_WIPHY);
1206 	if (!msg || nla_put_flag(msg, NL80211_ATTR_SPLIT_WIPHY_DUMP)) {
1207 		nlmsg_free(msg);
1208 		return -1;
1209 	}
1210 
1211 	if (send_and_recv_resp(drv, msg, wiphy_info_handler, info))
1212 		return -1;
1213 
1214 	if (info->auth_supported)
1215 		drv->capa.flags |= WPA_DRIVER_FLAGS_SME;
1216 	else if (!info->connect_supported) {
1217 		wpa_printf(MSG_INFO, "nl80211: Driver does not support "
1218 			   "authentication/association or connect commands");
1219 		info->error = 1;
1220 	}
1221 
1222 	if (info->p2p_go_supported && info->p2p_client_supported)
1223 		drv->capa.flags |= WPA_DRIVER_FLAGS_P2P_CAPABLE;
1224 	if (info->p2p_concurrent) {
1225 		wpa_printf(MSG_DEBUG, "nl80211: Use separate P2P group "
1226 			   "interface (driver advertised support)");
1227 		drv->capa.flags |= WPA_DRIVER_FLAGS_P2P_CONCURRENT;
1228 		drv->capa.flags |= WPA_DRIVER_FLAGS_P2P_MGMT_AND_NON_P2P;
1229 	}
1230 	if (info->num_multichan_concurrent > 1) {
1231 		wpa_printf(MSG_DEBUG, "nl80211: Enable multi-channel "
1232 			   "concurrent (driver advertised support)");
1233 		drv->capa.num_multichan_concurrent =
1234 			info->num_multichan_concurrent;
1235 	}
1236 	if (drv->capa.flags & WPA_DRIVER_FLAGS_DEDICATED_P2P_DEVICE)
1237 		wpa_printf(MSG_DEBUG, "nl80211: use P2P_DEVICE support");
1238 
1239 	/* default to 5000 since early versions of mac80211 don't set it */
1240 	if (!drv->capa.max_remain_on_chan)
1241 		drv->capa.max_remain_on_chan = 5000;
1242 
1243 	drv->capa.wmm_ac_supported = info->wmm_ac_supported;
1244 
1245 	drv->capa.mac_addr_rand_sched_scan_supported =
1246 		info->mac_addr_rand_sched_scan_supported;
1247 	drv->capa.mac_addr_rand_scan_supported =
1248 		info->mac_addr_rand_scan_supported;
1249 
1250 	if (info->channel_switch_supported) {
1251 		drv->capa.flags |= WPA_DRIVER_FLAGS_AP_CSA;
1252 		if (!drv->capa.max_csa_counters)
1253 			drv->capa.max_csa_counters = 1;
1254 	}
1255 
1256 	if (!drv->capa.max_sched_scan_plans) {
1257 		drv->capa.max_sched_scan_plans = 1;
1258 		drv->capa.max_sched_scan_plan_interval = UINT32_MAX;
1259 		drv->capa.max_sched_scan_plan_iterations = 0;
1260 	}
1261 
1262 	if (info->update_ft_ies_supported)
1263 		drv->capa.flags |= WPA_DRIVER_FLAGS_UPDATE_FT_IES;
1264 
1265 	if (!drv->capa.max_num_akms)
1266 		drv->capa.max_num_akms = NL80211_MAX_NR_AKM_SUITES;
1267 
1268 	return 0;
1269 }
1270 
1271 
1272 #ifdef CONFIG_DRIVER_NL80211_QCA
1273 
dfs_info_handler(struct nl_msg * msg,void * arg)1274 static int dfs_info_handler(struct nl_msg *msg, void *arg)
1275 {
1276 	struct nlattr *tb[NL80211_ATTR_MAX + 1];
1277 	struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
1278 	int *dfs_capability_ptr = arg;
1279 
1280 	nla_parse(tb, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
1281 		  genlmsg_attrlen(gnlh, 0), NULL);
1282 
1283 	if (tb[NL80211_ATTR_VENDOR_DATA]) {
1284 		struct nlattr *nl_vend = tb[NL80211_ATTR_VENDOR_DATA];
1285 		struct nlattr *tb_vendor[QCA_WLAN_VENDOR_ATTR_MAX + 1];
1286 
1287 		nla_parse(tb_vendor, QCA_WLAN_VENDOR_ATTR_MAX,
1288 			  nla_data(nl_vend), nla_len(nl_vend), NULL);
1289 
1290 		if (tb_vendor[QCA_WLAN_VENDOR_ATTR_DFS]) {
1291 			u32 val;
1292 			val = nla_get_u32(tb_vendor[QCA_WLAN_VENDOR_ATTR_DFS]);
1293 			wpa_printf(MSG_DEBUG, "nl80211: DFS offload capability: %u",
1294 				   val);
1295 			*dfs_capability_ptr = val;
1296 		}
1297 	}
1298 
1299 	return NL_SKIP;
1300 }
1301 
1302 
qca_nl80211_check_dfs_capa(struct wpa_driver_nl80211_data * drv)1303 static void qca_nl80211_check_dfs_capa(struct wpa_driver_nl80211_data *drv)
1304 {
1305 	struct nl_msg *msg;
1306 	int dfs_capability = 0;
1307 	int ret;
1308 
1309 	if (!drv->dfs_vendor_cmd_avail)
1310 		return;
1311 
1312 	if (!(msg = nl80211_drv_msg(drv, 0, NL80211_CMD_VENDOR)) ||
1313 	    nla_put_u32(msg, NL80211_ATTR_VENDOR_ID, OUI_QCA) ||
1314 	    nla_put_u32(msg, NL80211_ATTR_VENDOR_SUBCMD,
1315 			QCA_NL80211_VENDOR_SUBCMD_DFS_CAPABILITY)) {
1316 		nlmsg_free(msg);
1317 		return;
1318 	}
1319 
1320 	ret = send_and_recv_resp(drv, msg, dfs_info_handler, &dfs_capability);
1321 	if (!ret && dfs_capability)
1322 		drv->capa.flags |= WPA_DRIVER_FLAGS_DFS_OFFLOAD;
1323 }
1324 
1325 
1326 struct features_info {
1327 	u8 *flags;
1328 	size_t flags_len;
1329 	struct wpa_driver_capa *capa;
1330 };
1331 
1332 
features_info_handler(struct nl_msg * msg,void * arg)1333 static int features_info_handler(struct nl_msg *msg, void *arg)
1334 {
1335 	struct nlattr *tb[NL80211_ATTR_MAX + 1];
1336 	struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
1337 	struct features_info *info = arg;
1338 	struct nlattr *nl_vend, *attr;
1339 
1340 	nla_parse(tb, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
1341 		  genlmsg_attrlen(gnlh, 0), NULL);
1342 
1343 	nl_vend = tb[NL80211_ATTR_VENDOR_DATA];
1344 	if (nl_vend) {
1345 		struct nlattr *tb_vendor[QCA_WLAN_VENDOR_ATTR_MAX + 1];
1346 
1347 		nla_parse(tb_vendor, QCA_WLAN_VENDOR_ATTR_MAX,
1348 			  nla_data(nl_vend), nla_len(nl_vend), NULL);
1349 
1350 		attr = tb_vendor[QCA_WLAN_VENDOR_ATTR_FEATURE_FLAGS];
1351 		if (attr) {
1352 			int len = nla_len(attr);
1353 			info->flags = os_malloc(len);
1354 			if (info->flags != NULL) {
1355 				os_memcpy(info->flags, nla_data(attr), len);
1356 				info->flags_len = len;
1357 			}
1358 		}
1359 		attr = tb_vendor[QCA_WLAN_VENDOR_ATTR_CONCURRENCY_CAPA];
1360 		if (attr)
1361 			info->capa->conc_capab = nla_get_u32(attr);
1362 
1363 		attr = tb_vendor[
1364 			QCA_WLAN_VENDOR_ATTR_MAX_CONCURRENT_CHANNELS_2_4_BAND];
1365 		if (attr)
1366 			info->capa->max_conc_chan_2_4 = nla_get_u32(attr);
1367 
1368 		attr = tb_vendor[
1369 			QCA_WLAN_VENDOR_ATTR_MAX_CONCURRENT_CHANNELS_5_0_BAND];
1370 		if (attr)
1371 			info->capa->max_conc_chan_5_0 = nla_get_u32(attr);
1372 	}
1373 
1374 	return NL_SKIP;
1375 }
1376 
1377 
check_feature(enum qca_wlan_vendor_features feature,struct features_info * info)1378 static int check_feature(enum qca_wlan_vendor_features feature,
1379 			 struct features_info *info)
1380 {
1381 	size_t idx = feature / 8;
1382 
1383 	return (idx < info->flags_len) &&
1384 		(info->flags[idx] & BIT(feature % 8));
1385 }
1386 
1387 
qca_nl80211_get_features(struct wpa_driver_nl80211_data * drv)1388 static void qca_nl80211_get_features(struct wpa_driver_nl80211_data *drv)
1389 {
1390 	struct nl_msg *msg;
1391 	struct features_info info;
1392 	int ret;
1393 
1394 	if (!drv->get_features_vendor_cmd_avail)
1395 		return;
1396 
1397 	if (!(msg = nl80211_drv_msg(drv, 0, NL80211_CMD_VENDOR)) ||
1398 	    nla_put_u32(msg, NL80211_ATTR_VENDOR_ID, OUI_QCA) ||
1399 	    nla_put_u32(msg, NL80211_ATTR_VENDOR_SUBCMD,
1400 			QCA_NL80211_VENDOR_SUBCMD_GET_FEATURES)) {
1401 		nlmsg_free(msg);
1402 		return;
1403 	}
1404 
1405 	os_memset(&info, 0, sizeof(info));
1406 	info.capa = &drv->capa;
1407 	ret = send_and_recv_resp(drv, msg, features_info_handler, &info);
1408 	if (ret || !info.flags)
1409 		return;
1410 
1411 	if (check_feature(QCA_WLAN_VENDOR_FEATURE_KEY_MGMT_OFFLOAD, &info))
1412 		drv->capa.flags |= WPA_DRIVER_FLAGS_KEY_MGMT_OFFLOAD;
1413 
1414 	if (check_feature(QCA_WLAN_VENDOR_FEATURE_SUPPORT_HW_MODE_ANY, &info))
1415 		drv->capa.flags |= WPA_DRIVER_FLAGS_SUPPORT_HW_MODE_ANY;
1416 
1417 	if (check_feature(QCA_WLAN_VENDOR_FEATURE_OFFCHANNEL_SIMULTANEOUS,
1418 			  &info))
1419 		drv->capa.flags |= WPA_DRIVER_FLAGS_OFFCHANNEL_SIMULTANEOUS;
1420 	if (check_feature(QCA_WLAN_VENDOR_FEATURE_P2P_LISTEN_OFFLOAD, &info))
1421 		drv->capa.flags |= WPA_DRIVER_FLAGS_P2P_LISTEN_OFFLOAD;
1422 	if (check_feature(QCA_WLAN_VENDOR_FEATURE_OCE_STA, &info))
1423 		drv->capa.flags |= WPA_DRIVER_FLAGS_OCE_STA;
1424 	if (check_feature(QCA_WLAN_VENDOR_FEATURE_OCE_AP, &info))
1425 		drv->capa.flags |= WPA_DRIVER_FLAGS_OCE_AP;
1426 	if (check_feature(QCA_WLAN_VENDOR_FEATURE_OCE_STA_CFON, &info))
1427 		drv->capa.flags |= WPA_DRIVER_FLAGS_OCE_STA_CFON;
1428 	if (check_feature(QCA_WLAN_VENDOR_FEATURE_SECURE_LTF_STA, &info))
1429 		drv->capa.flags2 |= WPA_DRIVER_FLAGS2_SEC_LTF_STA;
1430 	if (check_feature(QCA_WLAN_VENDOR_FEATURE_SECURE_LTF_AP, &info))
1431 		drv->capa.flags2 |= WPA_DRIVER_FLAGS2_SEC_LTF_AP;
1432 	if (check_feature(QCA_WLAN_VENDOR_FEATURE_SECURE_RTT_STA, &info))
1433 		drv->capa.flags2 |= WPA_DRIVER_FLAGS2_SEC_RTT_STA;
1434 	if (check_feature(QCA_WLAN_VENDOR_FEATURE_SECURE_RTT_AP, &info))
1435 		drv->capa.flags2 |= WPA_DRIVER_FLAGS2_SEC_RTT_AP;
1436 	if (check_feature(
1437 		    QCA_WLAN_VENDOR_FEATURE_PROT_RANGE_NEGO_AND_MEASURE_STA,
1438 		    &info))
1439 		drv->capa.flags2 |= WPA_DRIVER_FLAGS2_PROT_RANGE_NEG_STA;
1440 	if (check_feature(
1441 		    QCA_WLAN_VENDOR_FEATURE_PROT_RANGE_NEGO_AND_MEASURE_AP,
1442 		    &info))
1443 		drv->capa.flags2 |= WPA_DRIVER_FLAGS2_PROT_RANGE_NEG_AP;
1444 	if (check_feature(QCA_WLAN_VENDOR_FEATURE_AP_ALLOWED_FREQ_LIST,
1445 			  &info))
1446 		drv->qca_ap_allowed_freqs = 1;
1447 	if (check_feature(QCA_WLAN_VENDOR_FEATURE_HT_VHT_TWT_RESPONDER, &info))
1448 		drv->capa.flags2 |= WPA_DRIVER_FLAGS2_HT_VHT_TWT_RESPONDER;
1449 	if (check_feature(QCA_WLAN_VENDOR_FEATURE_RSN_OVERRIDE_STA, &info)) {
1450 		wpa_printf(MSG_DEBUG,
1451 			   "The driver supports RSN overriding in STA mode");
1452 		drv->capa.flags2 |= WPA_DRIVER_FLAGS2_RSN_OVERRIDE_STA;
1453 	}
1454 
1455 	os_free(info.flags);
1456 }
1457 
1458 #endif /* CONFIG_DRIVER_NL80211_QCA */
1459 
1460 
wpa_driver_nl80211_capa(struct wpa_driver_nl80211_data * drv)1461 int wpa_driver_nl80211_capa(struct wpa_driver_nl80211_data *drv)
1462 {
1463 	struct wiphy_info_data info;
1464 	int i;
1465 
1466 	if (wpa_driver_nl80211_get_info(drv, &info))
1467 		return -1;
1468 
1469 	if (info.error)
1470 		return -1;
1471 
1472 	drv->has_capability = 1;
1473 	drv->has_driver_key_mgmt = info.has_key_mgmt | info.has_key_mgmt_iftype;
1474 
1475 	/* Fallback to hardcoded defaults if the driver does not advertise any
1476 	 * AKM capabilities. */
1477 	if (!drv->has_driver_key_mgmt) {
1478 		drv->capa.key_mgmt = WPA_DRIVER_CAPA_KEY_MGMT_WPA |
1479 			WPA_DRIVER_CAPA_KEY_MGMT_WPA_PSK |
1480 			WPA_DRIVER_CAPA_KEY_MGMT_WPA2 |
1481 			WPA_DRIVER_CAPA_KEY_MGMT_WPA2_PSK |
1482 			WPA_DRIVER_CAPA_KEY_MGMT_PSK_SHA256 |
1483 			WPA_DRIVER_CAPA_KEY_MGMT_SUITE_B |
1484 			WPA_DRIVER_CAPA_KEY_MGMT_OWE |
1485 			WPA_DRIVER_CAPA_KEY_MGMT_DPP;
1486 
1487 		if (drv->capa.enc & (WPA_DRIVER_CAPA_ENC_CCMP_256 |
1488 				     WPA_DRIVER_CAPA_ENC_GCMP_256))
1489 			drv->capa.key_mgmt |=
1490 				WPA_DRIVER_CAPA_KEY_MGMT_SUITE_B_192;
1491 
1492 		if (drv->capa.flags & WPA_DRIVER_FLAGS_SME)
1493 			drv->capa.key_mgmt |=
1494 				WPA_DRIVER_CAPA_KEY_MGMT_FILS_SHA256 |
1495 				WPA_DRIVER_CAPA_KEY_MGMT_FILS_SHA384 |
1496 				WPA_DRIVER_CAPA_KEY_MGMT_FT_FILS_SHA256 |
1497 				WPA_DRIVER_CAPA_KEY_MGMT_FT_FILS_SHA384 |
1498 				WPA_DRIVER_CAPA_KEY_MGMT_SAE_EXT_KEY |
1499 				WPA_DRIVER_CAPA_KEY_MGMT_SAE;
1500 		else if (drv->capa.flags & WPA_DRIVER_FLAGS_FILS_SK_OFFLOAD)
1501 			drv->capa.key_mgmt |=
1502 				WPA_DRIVER_CAPA_KEY_MGMT_FILS_SHA256 |
1503 				WPA_DRIVER_CAPA_KEY_MGMT_FILS_SHA384;
1504 	}
1505 
1506 	if (!info.has_key_mgmt_iftype) {
1507 		/* If the driver does not advertize per interface AKM
1508 		 * capabilities, consider all interfaces to support default AKMs
1509 		 * in key_mgmt. */
1510 		for (i = 0; i < WPA_IF_MAX; i++)
1511 			drv->capa.key_mgmt_iftype[i] = drv->capa.key_mgmt;
1512 	} else if (info.has_key_mgmt_iftype && !info.has_key_mgmt) {
1513 		/* If the driver advertizes only per interface supported AKMs
1514 		 * but does not advertize per wiphy AKM capabilities, consider
1515 		 * the default key_mgmt as a mask of per interface supported
1516 		 * AKMs. */
1517 		drv->capa.key_mgmt = 0;
1518 		for (i = 0; i < WPA_IF_MAX; i++)
1519 			drv->capa.key_mgmt |= drv->capa.key_mgmt_iftype[i];
1520 	} else if (info.has_key_mgmt_iftype && info.has_key_mgmt) {
1521 		/* If the driver advertizes AKM capabilities both per wiphy and
1522 		 * per interface, consider the interfaces for which per
1523 		 * interface AKM capabilities were not received to support the
1524 		 * default key_mgmt capabilities.
1525 		 */
1526 		for (i = 0; i < WPA_IF_MAX; i++)
1527 			if (!drv->capa.key_mgmt_iftype[i])
1528 				drv->capa.key_mgmt_iftype[i] =
1529 					drv->capa.key_mgmt;
1530 	}
1531 
1532 	drv->capa.auth = WPA_DRIVER_AUTH_OPEN |
1533 		WPA_DRIVER_AUTH_SHARED |
1534 		WPA_DRIVER_AUTH_LEAP;
1535 
1536 	drv->capa.flags |= WPA_DRIVER_FLAGS_VALID_ERROR_CODES;
1537 	drv->capa.flags |= WPA_DRIVER_FLAGS_SET_KEYS_AFTER_ASSOC_DONE;
1538 	drv->capa.flags |= WPA_DRIVER_FLAGS_EAPOL_TX_STATUS;
1539 
1540 	/*
1541 	 * As all cfg80211 drivers must support cases where the AP interface is
1542 	 * removed without the knowledge of wpa_supplicant/hostapd, e.g., in
1543 	 * case that the user space daemon has crashed, they must be able to
1544 	 * cleanup all stations and key entries in the AP tear down flow. Thus,
1545 	 * this flag can/should always be set for cfg80211 drivers.
1546 	 */
1547 	drv->capa.flags |= WPA_DRIVER_FLAGS_AP_TEARDOWN_SUPPORT;
1548 
1549 	if (!info.device_ap_sme) {
1550 		drv->capa.flags |= WPA_DRIVER_FLAGS_DEAUTH_TX_STATUS;
1551 		drv->capa.flags2 |= WPA_DRIVER_FLAGS2_AP_SME;
1552 
1553 		/*
1554 		 * No AP SME is currently assumed to also indicate no AP MLME
1555 		 * in the driver/firmware.
1556 		 */
1557 		drv->capa.flags |= WPA_DRIVER_FLAGS_AP_MLME;
1558 	}
1559 
1560 	drv->device_ap_sme = info.device_ap_sme;
1561 	drv->poll_command_supported = info.poll_command_supported;
1562 	drv->data_tx_status = info.data_tx_status;
1563 	drv->p2p_go_ctwindow_supported = info.p2p_go_ctwindow_supported;
1564 	if (info.set_qos_map_supported)
1565 		drv->capa.flags |= WPA_DRIVER_FLAGS_QOS_MAPPING;
1566 	drv->have_low_prio_scan = info.have_low_prio_scan;
1567 
1568 	/*
1569 	 * If poll command and tx status are supported, mac80211 is new enough
1570 	 * to have everything we need to not need monitor interfaces.
1571 	 */
1572 	drv->use_monitor = !info.device_ap_sme &&
1573 		(!info.poll_command_supported || !info.data_tx_status);
1574 
1575 	/*
1576 	 * If we aren't going to use monitor interfaces, but the
1577 	 * driver doesn't support data TX status, we won't get TX
1578 	 * status for EAPOL frames.
1579 	 */
1580 	if (!drv->use_monitor && !info.data_tx_status)
1581 		drv->capa.flags &= ~WPA_DRIVER_FLAGS_EAPOL_TX_STATUS;
1582 
1583 #ifdef CONFIG_DRIVER_NL80211_QCA
1584 	if (!(info.capa->flags & WPA_DRIVER_FLAGS_DFS_OFFLOAD))
1585 		qca_nl80211_check_dfs_capa(drv);
1586 	qca_nl80211_get_features(drv);
1587 
1588 	/*
1589 	 * To enable offchannel simultaneous support in wpa_supplicant, the
1590 	 * underlying driver needs to support the same along with offchannel TX.
1591 	 * Offchannel TX support is needed since remain_on_channel and
1592 	 * action_tx use some common data structures and hence cannot be
1593 	 * scheduled simultaneously.
1594 	 */
1595 	if (!(drv->capa.flags & WPA_DRIVER_FLAGS_OFFCHANNEL_TX))
1596 		drv->capa.flags &= ~WPA_DRIVER_FLAGS_OFFCHANNEL_SIMULTANEOUS;
1597 #endif /* CONFIG_DRIVER_NL80211_QCA */
1598 
1599 	wpa_printf(MSG_DEBUG,
1600 		   "nl80211: key_mgmt=0x%x enc=0x%x auth=0x%x flags=0x%llx flags2=0x%llx rrm_flags=0x%x probe_resp_offloads=0x%x max_stations=%u max_remain_on_chan=%u max_scan_ssids=%d",
1601 		   drv->capa.key_mgmt, drv->capa.enc, drv->capa.auth,
1602 		   (unsigned long long) drv->capa.flags,
1603 		   (unsigned long long) drv->capa.flags2, drv->capa.rrm_flags,
1604 		   drv->capa.probe_resp_offloads, drv->capa.max_stations,
1605 		   drv->capa.max_remain_on_chan, drv->capa.max_scan_ssids);
1606 	return 0;
1607 }
1608 
1609 
1610 struct phy_info_arg {
1611 	u16 *num_modes;
1612 	struct hostapd_hw_modes *modes;
1613 	int last_mode, last_chan_idx;
1614 	int failed;
1615 	u8 dfs_domain;
1616 };
1617 
phy_info_ht_capa(struct hostapd_hw_modes * mode,struct nlattr * capa,struct nlattr * ampdu_factor,struct nlattr * ampdu_density,struct nlattr * mcs_set)1618 static void phy_info_ht_capa(struct hostapd_hw_modes *mode, struct nlattr *capa,
1619 			     struct nlattr *ampdu_factor,
1620 			     struct nlattr *ampdu_density,
1621 			     struct nlattr *mcs_set)
1622 {
1623 	if (capa)
1624 		mode->ht_capab = nla_get_u16(capa);
1625 
1626 	if (ampdu_factor)
1627 		mode->a_mpdu_params |= nla_get_u8(ampdu_factor) & 0x03;
1628 
1629 	if (ampdu_density)
1630 		mode->a_mpdu_params |= nla_get_u8(ampdu_density) << 2;
1631 
1632 	if (mcs_set && nla_len(mcs_set) >= 16) {
1633 		u8 *mcs;
1634 		mcs = nla_data(mcs_set);
1635 		os_memcpy(mode->mcs_set, mcs, 16);
1636 	}
1637 }
1638 
1639 
phy_info_vht_capa(struct hostapd_hw_modes * mode,struct nlattr * capa,struct nlattr * mcs_set)1640 static void phy_info_vht_capa(struct hostapd_hw_modes *mode,
1641 			      struct nlattr *capa,
1642 			      struct nlattr *mcs_set)
1643 {
1644 	if (capa)
1645 		mode->vht_capab = nla_get_u32(capa);
1646 
1647 	if (mcs_set && nla_len(mcs_set) >= 8) {
1648 		u8 *mcs;
1649 		mcs = nla_data(mcs_set);
1650 		os_memcpy(mode->vht_mcs_set, mcs, 8);
1651 	}
1652 }
1653 
1654 
phy_info_edmg_capa(struct hostapd_hw_modes * mode,struct nlattr * bw_config,struct nlattr * channels)1655 static int phy_info_edmg_capa(struct hostapd_hw_modes *mode,
1656 			      struct nlattr *bw_config,
1657 			      struct nlattr *channels)
1658 {
1659 	if (!bw_config || !channels)
1660 		return NL_OK;
1661 
1662 	mode->edmg.bw_config = nla_get_u8(bw_config);
1663 	mode->edmg.channels = nla_get_u8(channels);
1664 
1665 	if (!mode->edmg.channels || !mode->edmg.bw_config)
1666 		return NL_STOP;
1667 
1668 	return NL_OK;
1669 }
1670 
1671 
cw2ecw(unsigned int cw)1672 static int cw2ecw(unsigned int cw)
1673 {
1674 	int bit;
1675 
1676 	if (cw == 0)
1677 		return 0;
1678 
1679 	for (bit = 1; cw != 1; bit++)
1680 		cw >>= 1;
1681 
1682 	return bit;
1683 }
1684 
1685 
phy_info_freq(struct hostapd_hw_modes * mode,struct hostapd_channel_data * chan,struct nlattr * tb_freq[])1686 static void phy_info_freq(struct hostapd_hw_modes *mode,
1687 			  struct hostapd_channel_data *chan,
1688 			  struct nlattr *tb_freq[])
1689 {
1690 	u8 channel;
1691 
1692 	os_memset(chan, 0, sizeof(*chan));
1693 	chan->freq = nla_get_u32(tb_freq[NL80211_FREQUENCY_ATTR_FREQ]);
1694 	chan->flag = 0;
1695 	chan->allowed_bw = ~0;
1696 	chan->dfs_cac_ms = 0;
1697 	if (ieee80211_freq_to_chan(chan->freq, &channel) != NUM_HOSTAPD_MODES)
1698 		chan->chan = channel;
1699 	else
1700 		wpa_printf(MSG_DEBUG,
1701 			   "nl80211: No channel number found for frequency %u MHz",
1702 			   chan->freq);
1703 
1704 	if (tb_freq[NL80211_FREQUENCY_ATTR_DISABLED])
1705 		chan->flag |= HOSTAPD_CHAN_DISABLED;
1706 	if (tb_freq[NL80211_FREQUENCY_ATTR_NO_IR])
1707 		chan->flag |= HOSTAPD_CHAN_NO_IR;
1708 	if (tb_freq[NL80211_FREQUENCY_ATTR_RADAR])
1709 		chan->flag |= HOSTAPD_CHAN_RADAR;
1710 	if (tb_freq[NL80211_FREQUENCY_ATTR_INDOOR_ONLY])
1711 		chan->flag |= HOSTAPD_CHAN_INDOOR_ONLY;
1712 	if (tb_freq[NL80211_FREQUENCY_ATTR_GO_CONCURRENT])
1713 		chan->flag |= HOSTAPD_CHAN_GO_CONCURRENT;
1714 
1715 	if (tb_freq[NL80211_FREQUENCY_ATTR_NO_10MHZ])
1716 		chan->allowed_bw &= ~HOSTAPD_CHAN_WIDTH_10;
1717 	if (tb_freq[NL80211_FREQUENCY_ATTR_NO_20MHZ])
1718 		chan->allowed_bw &= ~HOSTAPD_CHAN_WIDTH_20;
1719 	if (tb_freq[NL80211_FREQUENCY_ATTR_NO_HT40_PLUS])
1720 		chan->allowed_bw &= ~HOSTAPD_CHAN_WIDTH_40P;
1721 	if (tb_freq[NL80211_FREQUENCY_ATTR_NO_HT40_MINUS])
1722 		chan->allowed_bw &= ~HOSTAPD_CHAN_WIDTH_40M;
1723 	if (tb_freq[NL80211_FREQUENCY_ATTR_NO_80MHZ])
1724 		chan->allowed_bw &= ~HOSTAPD_CHAN_WIDTH_80;
1725 	if (tb_freq[NL80211_FREQUENCY_ATTR_NO_160MHZ])
1726 		chan->allowed_bw &= ~HOSTAPD_CHAN_WIDTH_160;
1727 
1728 	if (tb_freq[NL80211_FREQUENCY_ATTR_DFS_STATE]) {
1729 		enum nl80211_dfs_state state =
1730 			nla_get_u32(tb_freq[NL80211_FREQUENCY_ATTR_DFS_STATE]);
1731 
1732 		switch (state) {
1733 		case NL80211_DFS_USABLE:
1734 			chan->flag |= HOSTAPD_CHAN_DFS_USABLE;
1735 			break;
1736 		case NL80211_DFS_AVAILABLE:
1737 			chan->flag |= HOSTAPD_CHAN_DFS_AVAILABLE;
1738 			break;
1739 		case NL80211_DFS_UNAVAILABLE:
1740 			chan->flag |= HOSTAPD_CHAN_DFS_UNAVAILABLE;
1741 			break;
1742 		}
1743 	}
1744 
1745 	if (tb_freq[NL80211_FREQUENCY_ATTR_DFS_CAC_TIME]) {
1746 		chan->dfs_cac_ms = nla_get_u32(
1747 			tb_freq[NL80211_FREQUENCY_ATTR_DFS_CAC_TIME]);
1748 	}
1749 
1750 	chan->wmm_rules_valid = 0;
1751 	if (tb_freq[NL80211_FREQUENCY_ATTR_WMM]) {
1752 		static struct nla_policy wmm_policy[NL80211_WMMR_MAX + 1] = {
1753 			[NL80211_WMMR_CW_MIN] = { .type = NLA_U16 },
1754 			[NL80211_WMMR_CW_MAX] = { .type = NLA_U16 },
1755 			[NL80211_WMMR_AIFSN] = { .type = NLA_U8 },
1756 			[NL80211_WMMR_TXOP] = { .type = NLA_U16 },
1757 		};
1758 		static const u8 wmm_map[4] = {
1759 			[NL80211_AC_BE] = WMM_AC_BE,
1760 			[NL80211_AC_BK] = WMM_AC_BK,
1761 			[NL80211_AC_VI] = WMM_AC_VI,
1762 			[NL80211_AC_VO] = WMM_AC_VO,
1763 		};
1764 		struct nlattr *nl_wmm;
1765 		struct nlattr *tb_wmm[NL80211_WMMR_MAX + 1];
1766 		int rem_wmm, ac, count = 0;
1767 
1768 		nla_for_each_nested(nl_wmm, tb_freq[NL80211_FREQUENCY_ATTR_WMM],
1769 				    rem_wmm) {
1770 			if (nla_parse_nested(tb_wmm, NL80211_WMMR_MAX, nl_wmm,
1771 					     wmm_policy)) {
1772 				wpa_printf(MSG_DEBUG,
1773 					   "nl80211: Failed to parse WMM rules attribute");
1774 				return;
1775 			}
1776 			if (!tb_wmm[NL80211_WMMR_CW_MIN] ||
1777 			    !tb_wmm[NL80211_WMMR_CW_MAX] ||
1778 			    !tb_wmm[NL80211_WMMR_AIFSN] ||
1779 			    !tb_wmm[NL80211_WMMR_TXOP]) {
1780 				wpa_printf(MSG_DEBUG,
1781 					   "nl80211: Channel is missing WMM rule attribute");
1782 				return;
1783 			}
1784 			ac = nl_wmm->nla_type;
1785 			if ((unsigned int) ac >= ARRAY_SIZE(wmm_map)) {
1786 				wpa_printf(MSG_DEBUG,
1787 					   "nl80211: Invalid AC value %d", ac);
1788 				return;
1789 			}
1790 
1791 			ac = wmm_map[ac];
1792 			chan->wmm_rules[ac].min_cwmin =
1793 				cw2ecw(nla_get_u16(
1794 					       tb_wmm[NL80211_WMMR_CW_MIN]));
1795 			chan->wmm_rules[ac].min_cwmax =
1796 				cw2ecw(nla_get_u16(
1797 					       tb_wmm[NL80211_WMMR_CW_MAX]));
1798 			chan->wmm_rules[ac].min_aifs =
1799 				nla_get_u8(tb_wmm[NL80211_WMMR_AIFSN]);
1800 			chan->wmm_rules[ac].max_txop =
1801 				nla_get_u16(tb_wmm[NL80211_WMMR_TXOP]) / 32;
1802 			count++;
1803 		}
1804 
1805 		/* Set valid flag if all the AC rules are present */
1806 		if (count == WMM_AC_NUM)
1807 			chan->wmm_rules_valid = 1;
1808 	}
1809 }
1810 
1811 
phy_info_freqs(struct phy_info_arg * phy_info,struct hostapd_hw_modes * mode,struct nlattr * tb)1812 static int phy_info_freqs(struct phy_info_arg *phy_info,
1813 			  struct hostapd_hw_modes *mode, struct nlattr *tb)
1814 {
1815 	static struct nla_policy freq_policy[NL80211_FREQUENCY_ATTR_MAX + 1] = {
1816 		[NL80211_FREQUENCY_ATTR_FREQ] = { .type = NLA_U32 },
1817 		[NL80211_FREQUENCY_ATTR_DISABLED] = { .type = NLA_FLAG },
1818 		[NL80211_FREQUENCY_ATTR_NO_IR] = { .type = NLA_FLAG },
1819 		[NL80211_FREQUENCY_ATTR_RADAR] = { .type = NLA_FLAG },
1820 		[NL80211_FREQUENCY_ATTR_MAX_TX_POWER] = { .type = NLA_U32 },
1821 		[NL80211_FREQUENCY_ATTR_DFS_STATE] = { .type = NLA_U32 },
1822 		[NL80211_FREQUENCY_ATTR_NO_10MHZ] = { .type = NLA_FLAG },
1823 		[NL80211_FREQUENCY_ATTR_NO_20MHZ] = { .type = NLA_FLAG },
1824 		[NL80211_FREQUENCY_ATTR_NO_HT40_PLUS] = { .type = NLA_FLAG },
1825 		[NL80211_FREQUENCY_ATTR_NO_HT40_MINUS] = { .type = NLA_FLAG },
1826 		[NL80211_FREQUENCY_ATTR_NO_80MHZ] = { .type = NLA_FLAG },
1827 		[NL80211_FREQUENCY_ATTR_NO_160MHZ] = { .type = NLA_FLAG },
1828 	};
1829 	int new_channels = 0;
1830 	struct hostapd_channel_data *channel;
1831 	struct nlattr *tb_freq[NL80211_FREQUENCY_ATTR_MAX + 1];
1832 	struct nlattr *nl_freq;
1833 	int rem_freq, idx;
1834 
1835 	if (tb == NULL)
1836 		return NL_OK;
1837 
1838 	nla_for_each_nested(nl_freq, tb, rem_freq) {
1839 		nla_parse(tb_freq, NL80211_FREQUENCY_ATTR_MAX,
1840 			  nla_data(nl_freq), nla_len(nl_freq), freq_policy);
1841 		if (!tb_freq[NL80211_FREQUENCY_ATTR_FREQ])
1842 			continue;
1843 		new_channels++;
1844 	}
1845 
1846 	channel = os_realloc_array(mode->channels,
1847 				   mode->num_channels + new_channels,
1848 				   sizeof(struct hostapd_channel_data));
1849 	if (!channel)
1850 		return NL_STOP;
1851 
1852 	mode->channels = channel;
1853 	mode->num_channels += new_channels;
1854 
1855 	idx = phy_info->last_chan_idx;
1856 
1857 	nla_for_each_nested(nl_freq, tb, rem_freq) {
1858 		nla_parse(tb_freq, NL80211_FREQUENCY_ATTR_MAX,
1859 			  nla_data(nl_freq), nla_len(nl_freq), freq_policy);
1860 		if (!tb_freq[NL80211_FREQUENCY_ATTR_FREQ])
1861 			continue;
1862 		phy_info_freq(mode, &mode->channels[idx], tb_freq);
1863 		idx++;
1864 	}
1865 	phy_info->last_chan_idx = idx;
1866 
1867 	return NL_OK;
1868 }
1869 
1870 
phy_info_rates(struct hostapd_hw_modes * mode,struct nlattr * tb)1871 static int phy_info_rates(struct hostapd_hw_modes *mode, struct nlattr *tb)
1872 {
1873 	static struct nla_policy rate_policy[NL80211_BITRATE_ATTR_MAX + 1] = {
1874 		[NL80211_BITRATE_ATTR_RATE] = { .type = NLA_U32 },
1875 		[NL80211_BITRATE_ATTR_2GHZ_SHORTPREAMBLE] =
1876 		{ .type = NLA_FLAG },
1877 	};
1878 	struct nlattr *tb_rate[NL80211_BITRATE_ATTR_MAX + 1];
1879 	struct nlattr *nl_rate;
1880 	int rem_rate, idx;
1881 
1882 	if (tb == NULL)
1883 		return NL_OK;
1884 
1885 	nla_for_each_nested(nl_rate, tb, rem_rate) {
1886 		nla_parse(tb_rate, NL80211_BITRATE_ATTR_MAX,
1887 			  nla_data(nl_rate), nla_len(nl_rate),
1888 			  rate_policy);
1889 		if (!tb_rate[NL80211_BITRATE_ATTR_RATE])
1890 			continue;
1891 		mode->num_rates++;
1892 	}
1893 
1894 	mode->rates = os_calloc(mode->num_rates, sizeof(int));
1895 	if (!mode->rates)
1896 		return NL_STOP;
1897 
1898 	idx = 0;
1899 
1900 	nla_for_each_nested(nl_rate, tb, rem_rate) {
1901 		nla_parse(tb_rate, NL80211_BITRATE_ATTR_MAX,
1902 			  nla_data(nl_rate), nla_len(nl_rate),
1903 			  rate_policy);
1904 		if (!tb_rate[NL80211_BITRATE_ATTR_RATE])
1905 			continue;
1906 		mode->rates[idx] = nla_get_u32(
1907 			tb_rate[NL80211_BITRATE_ATTR_RATE]);
1908 		idx++;
1909 	}
1910 
1911 	return NL_OK;
1912 }
1913 
1914 
phy_info_iftype_copy(struct hostapd_hw_modes * mode,enum ieee80211_op_mode opmode,struct nlattr ** tb,struct nlattr ** tb_flags)1915 static void phy_info_iftype_copy(struct hostapd_hw_modes *mode,
1916 				 enum ieee80211_op_mode opmode,
1917 				 struct nlattr **tb, struct nlattr **tb_flags)
1918 {
1919 	enum nl80211_iftype iftype;
1920 	size_t len;
1921 	struct he_capabilities *he_capab = &mode->he_capab[opmode];
1922 	struct eht_capabilities *eht_capab = &mode->eht_capab[opmode];
1923 
1924 	switch (opmode) {
1925 	case IEEE80211_MODE_INFRA:
1926 		iftype = NL80211_IFTYPE_STATION;
1927 		break;
1928 	case IEEE80211_MODE_IBSS:
1929 		iftype = NL80211_IFTYPE_ADHOC;
1930 		break;
1931 	case IEEE80211_MODE_AP:
1932 		iftype = NL80211_IFTYPE_AP;
1933 		break;
1934 	case IEEE80211_MODE_MESH:
1935 		iftype = NL80211_IFTYPE_MESH_POINT;
1936 		break;
1937 	default:
1938 		return;
1939 	}
1940 
1941 	if (!nla_get_flag(tb_flags[iftype]))
1942 		return;
1943 
1944 	he_capab->he_supported = 1;
1945 
1946 	if (tb[NL80211_BAND_IFTYPE_ATTR_HE_CAP_PHY]) {
1947 		len = nla_len(tb[NL80211_BAND_IFTYPE_ATTR_HE_CAP_PHY]);
1948 
1949 		if (len > sizeof(he_capab->phy_cap))
1950 			len = sizeof(he_capab->phy_cap);
1951 		os_memcpy(he_capab->phy_cap,
1952 			  nla_data(tb[NL80211_BAND_IFTYPE_ATTR_HE_CAP_PHY]),
1953 			  len);
1954 	}
1955 
1956 	if (tb[NL80211_BAND_IFTYPE_ATTR_HE_CAP_MAC]) {
1957 		len = nla_len(tb[NL80211_BAND_IFTYPE_ATTR_HE_CAP_MAC]);
1958 
1959 		if (len > sizeof(he_capab->mac_cap))
1960 			len = sizeof(he_capab->mac_cap);
1961 		os_memcpy(he_capab->mac_cap,
1962 			  nla_data(tb[NL80211_BAND_IFTYPE_ATTR_HE_CAP_MAC]),
1963 			  len);
1964 	}
1965 
1966 	if (tb[NL80211_BAND_IFTYPE_ATTR_HE_CAP_MCS_SET]) {
1967 		len = nla_len(tb[NL80211_BAND_IFTYPE_ATTR_HE_CAP_MCS_SET]);
1968 
1969 		if (len > sizeof(he_capab->mcs))
1970 			len = sizeof(he_capab->mcs);
1971 		os_memcpy(he_capab->mcs,
1972 			  nla_data(tb[NL80211_BAND_IFTYPE_ATTR_HE_CAP_MCS_SET]),
1973 			  len);
1974 	}
1975 
1976 	if (tb[NL80211_BAND_IFTYPE_ATTR_HE_CAP_PPE]) {
1977 		len = nla_len(tb[NL80211_BAND_IFTYPE_ATTR_HE_CAP_PPE]);
1978 
1979 		if (len > sizeof(he_capab->ppet))
1980 			len = sizeof(he_capab->ppet);
1981 		os_memcpy(&he_capab->ppet,
1982 			  nla_data(tb[NL80211_BAND_IFTYPE_ATTR_HE_CAP_PPE]),
1983 			  len);
1984 	}
1985 
1986 	if (tb[NL80211_BAND_IFTYPE_ATTR_HE_6GHZ_CAPA]) {
1987 		u16 capa;
1988 
1989 		capa = nla_get_u16(tb[NL80211_BAND_IFTYPE_ATTR_HE_6GHZ_CAPA]);
1990 		he_capab->he_6ghz_capa = le_to_host16(capa);
1991 	}
1992 
1993 	if (!tb[NL80211_BAND_IFTYPE_ATTR_EHT_CAP_MAC] ||
1994 	    !tb[NL80211_BAND_IFTYPE_ATTR_EHT_CAP_PHY])
1995 		return;
1996 
1997 	eht_capab->eht_supported = true;
1998 
1999 	if (tb[NL80211_BAND_IFTYPE_ATTR_EHT_CAP_MAC] &&
2000 	    nla_len(tb[NL80211_BAND_IFTYPE_ATTR_EHT_CAP_MAC]) >= 2) {
2001 		    const u8 *pos;
2002 
2003 		    pos = nla_data(tb[NL80211_BAND_IFTYPE_ATTR_EHT_CAP_MAC]);
2004 		    eht_capab->mac_cap = WPA_GET_LE16(pos);
2005 	}
2006 
2007 	if (tb[NL80211_BAND_IFTYPE_ATTR_EHT_CAP_PHY]) {
2008 		len = nla_len(tb[NL80211_BAND_IFTYPE_ATTR_EHT_CAP_PHY]);
2009 		if (len > sizeof(eht_capab->phy_cap))
2010 			len = sizeof(eht_capab->phy_cap);
2011 		os_memcpy(eht_capab->phy_cap,
2012 			  nla_data(tb[NL80211_BAND_IFTYPE_ATTR_EHT_CAP_PHY]),
2013 			  len);
2014 	}
2015 
2016 	if (tb[NL80211_BAND_IFTYPE_ATTR_EHT_CAP_MCS_SET]) {
2017 		len = nla_len(tb[NL80211_BAND_IFTYPE_ATTR_EHT_CAP_MCS_SET]);
2018 		if (len > sizeof(eht_capab->mcs))
2019 			len = sizeof(eht_capab->mcs);
2020 		os_memcpy(eht_capab->mcs,
2021 			  nla_data(tb[NL80211_BAND_IFTYPE_ATTR_EHT_CAP_MCS_SET]),
2022 			  len);
2023 	}
2024 
2025 	if (tb[NL80211_BAND_IFTYPE_ATTR_EHT_CAP_PPE]) {
2026 		len = nla_len(tb[NL80211_BAND_IFTYPE_ATTR_EHT_CAP_PPE]);
2027 		if (len > sizeof(eht_capab->ppet))
2028 			len = sizeof(eht_capab->ppet);
2029 		os_memcpy(&eht_capab->ppet,
2030 			  nla_data(tb[NL80211_BAND_IFTYPE_ATTR_EHT_CAP_PPE]),
2031 			  len);
2032 	}
2033 }
2034 
2035 
phy_info_iftype(struct hostapd_hw_modes * mode,struct nlattr * nl_iftype)2036 static int phy_info_iftype(struct hostapd_hw_modes *mode,
2037 			   struct nlattr *nl_iftype)
2038 {
2039 	struct nlattr *tb[NL80211_BAND_IFTYPE_ATTR_MAX + 1];
2040 	struct nlattr *tb_flags[NL80211_IFTYPE_MAX + 1];
2041 	unsigned int i;
2042 
2043 	nla_parse(tb, NL80211_BAND_IFTYPE_ATTR_MAX,
2044 		  nla_data(nl_iftype), nla_len(nl_iftype), NULL);
2045 
2046 	if (!tb[NL80211_BAND_IFTYPE_ATTR_IFTYPES])
2047 		return NL_STOP;
2048 
2049 	if (nla_parse_nested(tb_flags, NL80211_IFTYPE_MAX,
2050 			     tb[NL80211_BAND_IFTYPE_ATTR_IFTYPES], NULL))
2051 		return NL_STOP;
2052 
2053 	for (i = 0; i < IEEE80211_MODE_NUM; i++)
2054 		phy_info_iftype_copy(mode, i, tb, tb_flags);
2055 
2056 	return NL_OK;
2057 }
2058 
2059 
phy_info_band(struct phy_info_arg * phy_info,struct nlattr * nl_band)2060 static int phy_info_band(struct phy_info_arg *phy_info, struct nlattr *nl_band)
2061 {
2062 	struct nlattr *tb_band[NL80211_BAND_ATTR_MAX + 1];
2063 	struct hostapd_hw_modes *mode;
2064 	int ret;
2065 
2066 	if (phy_info->last_mode != nl_band->nla_type) {
2067 		mode = os_realloc_array(phy_info->modes,
2068 					*phy_info->num_modes + 1,
2069 					sizeof(*mode));
2070 		if (!mode) {
2071 			phy_info->failed = 1;
2072 			return NL_STOP;
2073 		}
2074 		phy_info->modes = mode;
2075 
2076 		mode = &phy_info->modes[*(phy_info->num_modes)];
2077 		os_memset(mode, 0, sizeof(*mode));
2078 		mode->mode = NUM_HOSTAPD_MODES;
2079 		mode->flags = HOSTAPD_MODE_FLAG_HT_INFO_KNOWN |
2080 			HOSTAPD_MODE_FLAG_VHT_INFO_KNOWN |
2081 			HOSTAPD_MODE_FLAG_HE_INFO_KNOWN;
2082 
2083 		/*
2084 		 * Unsupported VHT MCS stream is defined as value 3, so the VHT
2085 		 * MCS RX/TX map must be initialized with 0xffff to mark all 8
2086 		 * possible streams as unsupported. This will be overridden if
2087 		 * driver advertises VHT support.
2088 		 */
2089 		mode->vht_mcs_set[0] = 0xff;
2090 		mode->vht_mcs_set[1] = 0xff;
2091 		mode->vht_mcs_set[4] = 0xff;
2092 		mode->vht_mcs_set[5] = 0xff;
2093 
2094 		*(phy_info->num_modes) += 1;
2095 		phy_info->last_mode = nl_band->nla_type;
2096 		phy_info->last_chan_idx = 0;
2097 	} else
2098 		mode = &phy_info->modes[*(phy_info->num_modes) - 1];
2099 
2100 	nla_parse(tb_band, NL80211_BAND_ATTR_MAX, nla_data(nl_band),
2101 		  nla_len(nl_band), NULL);
2102 
2103 	phy_info_ht_capa(mode, tb_band[NL80211_BAND_ATTR_HT_CAPA],
2104 			 tb_band[NL80211_BAND_ATTR_HT_AMPDU_FACTOR],
2105 			 tb_band[NL80211_BAND_ATTR_HT_AMPDU_DENSITY],
2106 			 tb_band[NL80211_BAND_ATTR_HT_MCS_SET]);
2107 	phy_info_vht_capa(mode, tb_band[NL80211_BAND_ATTR_VHT_CAPA],
2108 			  tb_band[NL80211_BAND_ATTR_VHT_MCS_SET]);
2109 	ret = phy_info_edmg_capa(mode,
2110 				 tb_band[NL80211_BAND_ATTR_EDMG_BW_CONFIG],
2111 				 tb_band[NL80211_BAND_ATTR_EDMG_CHANNELS]);
2112 	if (ret == NL_OK)
2113 		ret = phy_info_freqs(phy_info, mode,
2114 				     tb_band[NL80211_BAND_ATTR_FREQS]);
2115 	if (ret == NL_OK)
2116 		ret = phy_info_rates(mode, tb_band[NL80211_BAND_ATTR_RATES]);
2117 	if (ret != NL_OK) {
2118 		phy_info->failed = 1;
2119 		return ret;
2120 	}
2121 
2122 	if (tb_band[NL80211_BAND_ATTR_IFTYPE_DATA]) {
2123 		struct nlattr *nl_iftype;
2124 		int rem_band;
2125 
2126 		nla_for_each_nested(nl_iftype,
2127 				    tb_band[NL80211_BAND_ATTR_IFTYPE_DATA],
2128 				    rem_band) {
2129 			ret = phy_info_iftype(mode, nl_iftype);
2130 			if (ret != NL_OK)
2131 				return ret;
2132 		}
2133 	}
2134 
2135 	return NL_OK;
2136 }
2137 
2138 
phy_info_handler(struct nl_msg * msg,void * arg)2139 static int phy_info_handler(struct nl_msg *msg, void *arg)
2140 {
2141 	struct nlattr *tb_msg[NL80211_ATTR_MAX + 1];
2142 	struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
2143 	struct phy_info_arg *phy_info = arg;
2144 	struct nlattr *nl_band;
2145 	int rem_band;
2146 
2147 	nla_parse(tb_msg, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
2148 		  genlmsg_attrlen(gnlh, 0), NULL);
2149 
2150 	if (!tb_msg[NL80211_ATTR_WIPHY_BANDS])
2151 		return NL_SKIP;
2152 
2153 	nla_for_each_nested(nl_band, tb_msg[NL80211_ATTR_WIPHY_BANDS], rem_band)
2154 	{
2155 		int res = phy_info_band(phy_info, nl_band);
2156 		if (res != NL_OK)
2157 			return res;
2158 	}
2159 
2160 	return NL_SKIP;
2161 }
2162 
2163 
2164 static struct hostapd_hw_modes *
wpa_driver_nl80211_postprocess_modes(struct hostapd_hw_modes * modes,u16 * num_modes)2165 wpa_driver_nl80211_postprocess_modes(struct hostapd_hw_modes *modes,
2166 				     u16 *num_modes)
2167 {
2168 	u16 m;
2169 	struct hostapd_hw_modes *mode11g = NULL, *nmodes, *mode;
2170 	int i, mode11g_idx = -1;
2171 
2172 	/* heuristic to set up modes */
2173 	for (m = 0; m < *num_modes; m++) {
2174 		if (!modes[m].num_channels)
2175 			continue;
2176 
2177 		modes[m].is_6ghz = false;
2178 
2179 		if (modes[m].channels[0].freq < 2000) {
2180 			modes[m].num_channels = 0;
2181 			continue;
2182 		} else if (modes[m].channels[0].freq < 4000) {
2183 			modes[m].mode = HOSTAPD_MODE_IEEE80211B;
2184 			for (i = 0; i < modes[m].num_rates; i++) {
2185 				if (modes[m].rates[i] > 200) {
2186 					modes[m].mode = HOSTAPD_MODE_IEEE80211G;
2187 					break;
2188 				}
2189 			}
2190 		} else if (modes[m].channels[0].freq > 50000) {
2191 			modes[m].mode = HOSTAPD_MODE_IEEE80211AD;
2192 		} else if (is_6ghz_freq(modes[m].channels[0].freq)) {
2193 			modes[m].mode = HOSTAPD_MODE_IEEE80211A;
2194 			modes[m].is_6ghz = true;
2195 		} else {
2196 			modes[m].mode = HOSTAPD_MODE_IEEE80211A;
2197 		}
2198 	}
2199 
2200 	/* Remove unsupported bands */
2201 	m = 0;
2202 	while (m < *num_modes) {
2203 		if (modes[m].mode == NUM_HOSTAPD_MODES) {
2204 			wpa_printf(MSG_DEBUG,
2205 				   "nl80211: Remove unsupported mode");
2206 			os_free(modes[m].channels);
2207 			os_free(modes[m].rates);
2208 			if (m + 1 < *num_modes)
2209 				os_memmove(&modes[m], &modes[m + 1],
2210 					   sizeof(struct hostapd_hw_modes) *
2211 					   (*num_modes - (m + 1)));
2212 			(*num_modes)--;
2213 			continue;
2214 		}
2215 		m++;
2216 	}
2217 
2218 	/* If only 802.11g mode is included, use it to construct matching
2219 	 * 802.11b mode data. */
2220 
2221 	for (m = 0; m < *num_modes; m++) {
2222 		if (modes[m].mode == HOSTAPD_MODE_IEEE80211B)
2223 			return modes; /* 802.11b already included */
2224 		if (modes[m].mode == HOSTAPD_MODE_IEEE80211G)
2225 			mode11g_idx = m;
2226 	}
2227 
2228 	if (mode11g_idx < 0)
2229 		return modes; /* 2.4 GHz band not supported at all */
2230 
2231 	nmodes = os_realloc_array(modes, *num_modes + 1, sizeof(*nmodes));
2232 	if (nmodes == NULL)
2233 		return modes; /* Could not add 802.11b mode */
2234 
2235 	mode = &nmodes[*num_modes];
2236 	os_memset(mode, 0, sizeof(*mode));
2237 	(*num_modes)++;
2238 	modes = nmodes;
2239 
2240 	mode->mode = HOSTAPD_MODE_IEEE80211B;
2241 
2242 	mode11g = &modes[mode11g_idx];
2243 	mode->num_channels = mode11g->num_channels;
2244 	mode->channels = os_memdup(mode11g->channels,
2245 				   mode11g->num_channels *
2246 				   sizeof(struct hostapd_channel_data));
2247 	if (mode->channels == NULL) {
2248 		(*num_modes)--;
2249 		return modes; /* Could not add 802.11b mode */
2250 	}
2251 
2252 	mode->num_rates = 0;
2253 	mode->rates = os_malloc(4 * sizeof(int));
2254 	if (mode->rates == NULL) {
2255 		os_free(mode->channels);
2256 		(*num_modes)--;
2257 		return modes; /* Could not add 802.11b mode */
2258 	}
2259 
2260 	for (i = 0; i < mode11g->num_rates; i++) {
2261 		if (mode11g->rates[i] != 10 && mode11g->rates[i] != 20 &&
2262 		    mode11g->rates[i] != 55 && mode11g->rates[i] != 110)
2263 			continue;
2264 		mode->rates[mode->num_rates] = mode11g->rates[i];
2265 		mode->num_rates++;
2266 		if (mode->num_rates == 4)
2267 			break;
2268 	}
2269 
2270 	if (mode->num_rates == 0) {
2271 		os_free(mode->channels);
2272 		os_free(mode->rates);
2273 		(*num_modes)--;
2274 		return modes; /* No 802.11b rates */
2275 	}
2276 
2277 	wpa_printf(MSG_DEBUG, "nl80211: Added 802.11b mode based on 802.11g "
2278 		   "information");
2279 
2280 	return modes;
2281 }
2282 
2283 
nl80211_set_ht40_mode(struct hostapd_hw_modes * mode,int start,int end)2284 static void nl80211_set_ht40_mode(struct hostapd_hw_modes *mode, int start,
2285 				  int end)
2286 {
2287 	int c;
2288 
2289 	for (c = 0; c < mode->num_channels; c++) {
2290 		struct hostapd_channel_data *chan = &mode->channels[c];
2291 		if (chan->freq - 10 >= start && chan->freq + 10 <= end)
2292 			chan->flag |= HOSTAPD_CHAN_HT40;
2293 	}
2294 }
2295 
2296 
nl80211_set_ht40_mode_sec(struct hostapd_hw_modes * mode,int start,int end)2297 static void nl80211_set_ht40_mode_sec(struct hostapd_hw_modes *mode, int start,
2298 				      int end)
2299 {
2300 	int c;
2301 
2302 	for (c = 0; c < mode->num_channels; c++) {
2303 		struct hostapd_channel_data *chan = &mode->channels[c];
2304 		if (!(chan->flag & HOSTAPD_CHAN_HT40))
2305 			continue;
2306 		if (chan->freq - 30 >= start && chan->freq - 10 <= end)
2307 			chan->flag |= HOSTAPD_CHAN_HT40MINUS;
2308 		if (chan->freq + 10 >= start && chan->freq + 30 <= end)
2309 			chan->flag |= HOSTAPD_CHAN_HT40PLUS;
2310 	}
2311 }
2312 
2313 
nl80211_reg_rule_max_eirp(u32 start,u32 end,u32 max_eirp,struct phy_info_arg * results)2314 static void nl80211_reg_rule_max_eirp(u32 start, u32 end, u32 max_eirp,
2315 				      struct phy_info_arg *results)
2316 {
2317 	u16 m;
2318 
2319 	for (m = 0; m < *results->num_modes; m++) {
2320 		int c;
2321 		struct hostapd_hw_modes *mode = &results->modes[m];
2322 
2323 		for (c = 0; c < mode->num_channels; c++) {
2324 			struct hostapd_channel_data *chan = &mode->channels[c];
2325 			if ((u32) chan->freq - 10 >= start &&
2326 			    (u32) chan->freq + 10 <= end)
2327 				chan->max_tx_power = max_eirp;
2328 		}
2329 	}
2330 }
2331 
2332 
nl80211_reg_rule_ht40(u32 start,u32 end,struct phy_info_arg * results)2333 static void nl80211_reg_rule_ht40(u32 start, u32 end,
2334 				  struct phy_info_arg *results)
2335 {
2336 	u16 m;
2337 
2338 	for (m = 0; m < *results->num_modes; m++) {
2339 		if (!(results->modes[m].ht_capab &
2340 		      HT_CAP_INFO_SUPP_CHANNEL_WIDTH_SET))
2341 			continue;
2342 		nl80211_set_ht40_mode(&results->modes[m], start, end);
2343 	}
2344 }
2345 
2346 
nl80211_reg_rule_sec(struct nlattr * tb[],struct phy_info_arg * results)2347 static void nl80211_reg_rule_sec(struct nlattr *tb[],
2348 				 struct phy_info_arg *results)
2349 {
2350 	u32 start, end, max_bw;
2351 	u16 m;
2352 
2353 	if (tb[NL80211_ATTR_FREQ_RANGE_START] == NULL ||
2354 	    tb[NL80211_ATTR_FREQ_RANGE_END] == NULL ||
2355 	    tb[NL80211_ATTR_FREQ_RANGE_MAX_BW] == NULL)
2356 		return;
2357 
2358 	start = nla_get_u32(tb[NL80211_ATTR_FREQ_RANGE_START]) / 1000;
2359 	end = nla_get_u32(tb[NL80211_ATTR_FREQ_RANGE_END]) / 1000;
2360 	max_bw = nla_get_u32(tb[NL80211_ATTR_FREQ_RANGE_MAX_BW]) / 1000;
2361 
2362 	if (max_bw < 20)
2363 		return;
2364 
2365 	for (m = 0; m < *results->num_modes; m++) {
2366 		if (!(results->modes[m].ht_capab &
2367 		      HT_CAP_INFO_SUPP_CHANNEL_WIDTH_SET))
2368 			continue;
2369 		nl80211_set_ht40_mode_sec(&results->modes[m], start, end);
2370 	}
2371 }
2372 
2373 
nl80211_set_vht_mode(struct hostapd_hw_modes * mode,int start,int end,int max_bw)2374 static void nl80211_set_vht_mode(struct hostapd_hw_modes *mode, int start,
2375 				 int end, int max_bw)
2376 {
2377 	int c;
2378 
2379 	for (c = 0; c < mode->num_channels; c++) {
2380 		struct hostapd_channel_data *chan = &mode->channels[c];
2381 
2382 		if (chan->freq - 10 < start || chan->freq + 10 > end)
2383 			continue;
2384 
2385 		if (max_bw >= 80)
2386 			chan->flag |= HOSTAPD_CHAN_VHT_80MHZ_SUBCHANNEL;
2387 
2388 		if (max_bw >= 160)
2389 			chan->flag |= HOSTAPD_CHAN_VHT_160MHZ_SUBCHANNEL;
2390 	}
2391 }
2392 
2393 
nl80211_reg_rule_vht(struct nlattr * tb[],struct phy_info_arg * results)2394 static void nl80211_reg_rule_vht(struct nlattr *tb[],
2395 				 struct phy_info_arg *results)
2396 {
2397 	u32 start, end, max_bw;
2398 	u16 m;
2399 
2400 	if (tb[NL80211_ATTR_FREQ_RANGE_START] == NULL ||
2401 	    tb[NL80211_ATTR_FREQ_RANGE_END] == NULL ||
2402 	    tb[NL80211_ATTR_FREQ_RANGE_MAX_BW] == NULL)
2403 		return;
2404 
2405 	start = nla_get_u32(tb[NL80211_ATTR_FREQ_RANGE_START]) / 1000;
2406 	end = nla_get_u32(tb[NL80211_ATTR_FREQ_RANGE_END]) / 1000;
2407 	max_bw = nla_get_u32(tb[NL80211_ATTR_FREQ_RANGE_MAX_BW]) / 1000;
2408 
2409 	if (max_bw < 80)
2410 		return;
2411 
2412 	for (m = 0; m < *results->num_modes; m++) {
2413 		if (!(results->modes[m].ht_capab &
2414 		      HT_CAP_INFO_SUPP_CHANNEL_WIDTH_SET))
2415 			continue;
2416 		/* TODO: use a real VHT support indication */
2417 		if (!results->modes[m].vht_capab)
2418 			continue;
2419 
2420 		nl80211_set_vht_mode(&results->modes[m], start, end, max_bw);
2421 	}
2422 }
2423 
2424 
nl80211_set_6ghz_mode(struct hostapd_hw_modes * mode,int start,int end,int max_bw)2425 static void nl80211_set_6ghz_mode(struct hostapd_hw_modes *mode, int start,
2426 				  int end, int max_bw)
2427 {
2428 	int c;
2429 
2430 	for (c = 0; c < mode->num_channels; c++) {
2431 		struct hostapd_channel_data *chan = &mode->channels[c];
2432 
2433 		if (chan->freq - 10 < start || chan->freq + 10 > end)
2434 			continue;
2435 
2436 		if (max_bw >= 80)
2437 			chan->flag |= HOSTAPD_CHAN_VHT_80MHZ_SUBCHANNEL;
2438 
2439 		if (max_bw >= 160)
2440 			chan->flag |= HOSTAPD_CHAN_VHT_160MHZ_SUBCHANNEL;
2441 
2442 		if (max_bw >= 320)
2443 			chan->flag |= HOSTAPD_CHAN_EHT_320MHZ_SUBCHANNEL;
2444 	}
2445 }
2446 
2447 
nl80211_reg_rule_6ghz(struct nlattr * tb[],struct phy_info_arg * results)2448 static void nl80211_reg_rule_6ghz(struct nlattr *tb[],
2449 				 struct phy_info_arg *results)
2450 {
2451 	u32 start, end, max_bw;
2452 	u16 m;
2453 
2454 	if (!tb[NL80211_ATTR_FREQ_RANGE_START] ||
2455 	    !tb[NL80211_ATTR_FREQ_RANGE_END] ||
2456 	    !tb[NL80211_ATTR_FREQ_RANGE_MAX_BW])
2457 		return;
2458 
2459 	start = nla_get_u32(tb[NL80211_ATTR_FREQ_RANGE_START]) / 1000;
2460 	end = nla_get_u32(tb[NL80211_ATTR_FREQ_RANGE_END]) / 1000;
2461 	max_bw = nla_get_u32(tb[NL80211_ATTR_FREQ_RANGE_MAX_BW]) / 1000;
2462 
2463 	if (max_bw < 80)
2464 		return;
2465 
2466 	for (m = 0; m < *results->num_modes; m++) {
2467 		if (results->modes[m].num_channels == 0 ||
2468 		    !is_6ghz_freq(results->modes[m].channels[0].freq))
2469 			continue;
2470 
2471 		nl80211_set_6ghz_mode(&results->modes[m], start, end, max_bw);
2472 	}
2473 }
2474 
2475 
nl80211_set_dfs_domain(enum nl80211_dfs_regions region,u8 * dfs_domain)2476 static void nl80211_set_dfs_domain(enum nl80211_dfs_regions region,
2477 				   u8 *dfs_domain)
2478 {
2479 	if (region == NL80211_DFS_FCC)
2480 		*dfs_domain = HOSTAPD_DFS_REGION_FCC;
2481 	else if (region == NL80211_DFS_ETSI)
2482 		*dfs_domain = HOSTAPD_DFS_REGION_ETSI;
2483 	else if (region == NL80211_DFS_JP)
2484 		*dfs_domain = HOSTAPD_DFS_REGION_JP;
2485 	else
2486 		*dfs_domain = 0;
2487 }
2488 
2489 
dfs_domain_name(enum nl80211_dfs_regions region)2490 static const char * dfs_domain_name(enum nl80211_dfs_regions region)
2491 {
2492 	switch (region) {
2493 	case NL80211_DFS_UNSET:
2494 		return "DFS-UNSET";
2495 	case NL80211_DFS_FCC:
2496 		return "DFS-FCC";
2497 	case NL80211_DFS_ETSI:
2498 		return "DFS-ETSI";
2499 	case NL80211_DFS_JP:
2500 		return "DFS-JP";
2501 	default:
2502 		return "DFS-invalid";
2503 	}
2504 }
2505 
2506 
nl80211_get_reg(struct nl_msg * msg,void * arg)2507 static int nl80211_get_reg(struct nl_msg *msg, void *arg)
2508 {
2509 	struct phy_info_arg *results = arg;
2510 	struct nlattr *tb_msg[NL80211_ATTR_MAX + 1];
2511 	struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
2512 	struct nlattr *nl_rule;
2513 	struct nlattr *tb_rule[NL80211_FREQUENCY_ATTR_MAX + 1];
2514 	int rem_rule;
2515 	static struct nla_policy reg_policy[NL80211_FREQUENCY_ATTR_MAX + 1] = {
2516 		[NL80211_ATTR_REG_RULE_FLAGS] = { .type = NLA_U32 },
2517 		[NL80211_ATTR_FREQ_RANGE_START] = { .type = NLA_U32 },
2518 		[NL80211_ATTR_FREQ_RANGE_END] = { .type = NLA_U32 },
2519 		[NL80211_ATTR_FREQ_RANGE_MAX_BW] = { .type = NLA_U32 },
2520 		[NL80211_ATTR_POWER_RULE_MAX_ANT_GAIN] = { .type = NLA_U32 },
2521 		[NL80211_ATTR_POWER_RULE_MAX_EIRP] = { .type = NLA_U32 },
2522 	};
2523 
2524 	nla_parse(tb_msg, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
2525 		  genlmsg_attrlen(gnlh, 0), NULL);
2526 	if (!tb_msg[NL80211_ATTR_REG_ALPHA2] ||
2527 	    !tb_msg[NL80211_ATTR_REG_RULES]) {
2528 		wpa_printf(MSG_DEBUG, "nl80211: No regulatory information "
2529 			   "available");
2530 		return NL_SKIP;
2531 	}
2532 
2533 	if (tb_msg[NL80211_ATTR_DFS_REGION]) {
2534 		enum nl80211_dfs_regions dfs_domain;
2535 		dfs_domain = nla_get_u8(tb_msg[NL80211_ATTR_DFS_REGION]);
2536 		nl80211_set_dfs_domain(dfs_domain, &results->dfs_domain);
2537 		wpa_printf(MSG_DEBUG, "nl80211: Regulatory information - country=%s (%s)",
2538 			   (char *) nla_data(tb_msg[NL80211_ATTR_REG_ALPHA2]),
2539 			   dfs_domain_name(dfs_domain));
2540 	} else {
2541 		wpa_printf(MSG_DEBUG, "nl80211: Regulatory information - country=%s",
2542 			   (char *) nla_data(tb_msg[NL80211_ATTR_REG_ALPHA2]));
2543 	}
2544 
2545 	nla_for_each_nested(nl_rule, tb_msg[NL80211_ATTR_REG_RULES], rem_rule)
2546 	{
2547 		u32 start, end, max_eirp = 0, max_bw = 0, flags = 0;
2548 		nla_parse(tb_rule, NL80211_FREQUENCY_ATTR_MAX,
2549 			  nla_data(nl_rule), nla_len(nl_rule), reg_policy);
2550 		if (tb_rule[NL80211_ATTR_FREQ_RANGE_START] == NULL ||
2551 		    tb_rule[NL80211_ATTR_FREQ_RANGE_END] == NULL)
2552 			continue;
2553 		start = nla_get_u32(tb_rule[NL80211_ATTR_FREQ_RANGE_START]) / 1000;
2554 		end = nla_get_u32(tb_rule[NL80211_ATTR_FREQ_RANGE_END]) / 1000;
2555 		if (tb_rule[NL80211_ATTR_POWER_RULE_MAX_EIRP])
2556 			max_eirp = nla_get_u32(tb_rule[NL80211_ATTR_POWER_RULE_MAX_EIRP]) / 100;
2557 		if (tb_rule[NL80211_ATTR_FREQ_RANGE_MAX_BW])
2558 			max_bw = nla_get_u32(tb_rule[NL80211_ATTR_FREQ_RANGE_MAX_BW]) / 1000;
2559 		if (tb_rule[NL80211_ATTR_REG_RULE_FLAGS])
2560 			flags = nla_get_u32(tb_rule[NL80211_ATTR_REG_RULE_FLAGS]);
2561 
2562 		wpa_printf(MSG_DEBUG, "nl80211: %u-%u @ %u MHz %u mBm%s%s%s%s%s%s%s%s",
2563 			   start, end, max_bw, max_eirp,
2564 			   flags & NL80211_RRF_NO_OFDM ? " (no OFDM)" : "",
2565 			   flags & NL80211_RRF_NO_CCK ? " (no CCK)" : "",
2566 			   flags & NL80211_RRF_NO_INDOOR ? " (no indoor)" : "",
2567 			   flags & NL80211_RRF_NO_OUTDOOR ? " (no outdoor)" :
2568 			   "",
2569 			   flags & NL80211_RRF_DFS ? " (DFS)" : "",
2570 			   flags & NL80211_RRF_PTP_ONLY ? " (PTP only)" : "",
2571 			   flags & NL80211_RRF_PTMP_ONLY ? " (PTMP only)" : "",
2572 			   flags & NL80211_RRF_NO_IR ? " (no IR)" : "");
2573 		if (max_bw >= 40)
2574 			nl80211_reg_rule_ht40(start, end, results);
2575 		if (tb_rule[NL80211_ATTR_POWER_RULE_MAX_EIRP])
2576 			nl80211_reg_rule_max_eirp(start, end, max_eirp,
2577 						  results);
2578 	}
2579 
2580 	nla_for_each_nested(nl_rule, tb_msg[NL80211_ATTR_REG_RULES], rem_rule)
2581 	{
2582 		nla_parse(tb_rule, NL80211_FREQUENCY_ATTR_MAX,
2583 			  nla_data(nl_rule), nla_len(nl_rule), reg_policy);
2584 		nl80211_reg_rule_sec(tb_rule, results);
2585 	}
2586 
2587 	nla_for_each_nested(nl_rule, tb_msg[NL80211_ATTR_REG_RULES], rem_rule)
2588 	{
2589 		nla_parse(tb_rule, NL80211_FREQUENCY_ATTR_MAX,
2590 			  nla_data(nl_rule), nla_len(nl_rule), reg_policy);
2591 		nl80211_reg_rule_vht(tb_rule, results);
2592 	}
2593 
2594 	nla_for_each_nested(nl_rule, tb_msg[NL80211_ATTR_REG_RULES], rem_rule)
2595 	{
2596 		nla_parse(tb_rule, NL80211_FREQUENCY_ATTR_MAX,
2597 			  nla_data(nl_rule), nla_len(nl_rule), reg_policy);
2598 		nl80211_reg_rule_6ghz(tb_rule, results);
2599 	}
2600 
2601 	return NL_SKIP;
2602 }
2603 
2604 
nl80211_set_regulatory_flags(struct wpa_driver_nl80211_data * drv,struct phy_info_arg * results)2605 static int nl80211_set_regulatory_flags(struct wpa_driver_nl80211_data *drv,
2606 					struct phy_info_arg *results)
2607 {
2608 	struct nl_msg *msg;
2609 
2610 	msg = nlmsg_alloc();
2611 	if (!msg)
2612 		return -ENOMEM;
2613 
2614 	nl80211_cmd(drv, msg, 0, NL80211_CMD_GET_REG);
2615 	if (drv->capa.flags & WPA_DRIVER_FLAGS_SELF_MANAGED_REGULATORY) {
2616 		if (nla_put_u32(msg, NL80211_ATTR_WIPHY, drv->wiphy_idx)) {
2617 			nlmsg_free(msg);
2618 			return -1;
2619 		}
2620 	}
2621 
2622 	return send_and_recv_resp(drv, msg, nl80211_get_reg, results);
2623 }
2624 
2625 
modestr(enum hostapd_hw_mode mode)2626 static const char * modestr(enum hostapd_hw_mode mode)
2627 {
2628 	switch (mode) {
2629 	case HOSTAPD_MODE_IEEE80211B:
2630 		return "802.11b";
2631 	case HOSTAPD_MODE_IEEE80211G:
2632 		return "802.11g";
2633 	case HOSTAPD_MODE_IEEE80211A:
2634 		return "802.11a";
2635 	case HOSTAPD_MODE_IEEE80211AD:
2636 		return "802.11ad";
2637 	default:
2638 		return "?";
2639 	}
2640 }
2641 
2642 
nl80211_dump_chan_list(struct wpa_driver_nl80211_data * drv,struct hostapd_hw_modes * modes,u16 num_modes)2643 static void nl80211_dump_chan_list(struct wpa_driver_nl80211_data *drv,
2644 				   struct hostapd_hw_modes *modes,
2645 				   u16 num_modes)
2646 {
2647 	int i;
2648 
2649 	if (!modes)
2650 		return;
2651 
2652 	for (i = 0; i < num_modes; i++) {
2653 		struct hostapd_hw_modes *mode = &modes[i];
2654 		char str[1000];
2655 		char *pos = str;
2656 		char *end = pos + sizeof(str);
2657 		int j, res;
2658 
2659 		for (j = 0; j < mode->num_channels; j++) {
2660 			struct hostapd_channel_data *chan = &mode->channels[j];
2661 
2662 			if (is_6ghz_freq(chan->freq))
2663 				drv->uses_6ghz = true;
2664 			if (chan->freq >= 900 && chan->freq < 1000)
2665 				drv->uses_s1g = true;
2666 			res = os_snprintf(pos, end - pos, " %d%s%s%s",
2667 					  chan->freq,
2668 					  (chan->flag & HOSTAPD_CHAN_DISABLED) ?
2669 					  "[DISABLED]" : "",
2670 					  (chan->flag & HOSTAPD_CHAN_NO_IR) ?
2671 					  "[NO_IR]" : "",
2672 					  (chan->flag & HOSTAPD_CHAN_RADAR) ?
2673 					  "[RADAR]" : "");
2674 			if (os_snprintf_error(end - pos, res))
2675 				break;
2676 			pos += res;
2677 		}
2678 
2679 		*pos = '\0';
2680 		wpa_printf(MSG_DEBUG, "nl80211: Mode IEEE %s:%s",
2681 			   modestr(mode->mode), str);
2682 	}
2683 }
2684 
2685 
2686 struct hostapd_hw_modes *
nl80211_get_hw_feature_data(void * priv,u16 * num_modes,u16 * flags,u8 * dfs_domain)2687 nl80211_get_hw_feature_data(void *priv, u16 *num_modes, u16 *flags,
2688 			    u8 *dfs_domain)
2689 {
2690 	u32 feat;
2691 	struct i802_bss *bss = priv;
2692 	struct wpa_driver_nl80211_data *drv = bss->drv;
2693 	int nl_flags = 0;
2694 	struct nl_msg *msg;
2695 	struct phy_info_arg result = {
2696 		.num_modes = num_modes,
2697 		.modes = NULL,
2698 		.last_mode = -1,
2699 		.failed = 0,
2700 		.dfs_domain = 0,
2701 	};
2702 
2703 	*num_modes = 0;
2704 	*flags = 0;
2705 	*dfs_domain = 0;
2706 
2707 	feat = get_nl80211_protocol_features(drv);
2708 	if (feat & NL80211_PROTOCOL_FEATURE_SPLIT_WIPHY_DUMP)
2709 		nl_flags = NLM_F_DUMP;
2710 	if (!(msg = nl80211_cmd_msg(bss, nl_flags, NL80211_CMD_GET_WIPHY)) ||
2711 	    nla_put_flag(msg, NL80211_ATTR_SPLIT_WIPHY_DUMP)) {
2712 		nlmsg_free(msg);
2713 		return NULL;
2714 	}
2715 
2716 	if (send_and_recv_resp(drv, msg, phy_info_handler, &result) == 0) {
2717 		struct hostapd_hw_modes *modes;
2718 
2719 		nl80211_set_regulatory_flags(drv, &result);
2720 		if (result.failed) {
2721 			int i;
2722 
2723 			for (i = 0; result.modes && i < *num_modes; i++) {
2724 				os_free(result.modes[i].channels);
2725 				os_free(result.modes[i].rates);
2726 			}
2727 			os_free(result.modes);
2728 			*num_modes = 0;
2729 			return NULL;
2730 		}
2731 
2732 		*dfs_domain = result.dfs_domain;
2733 
2734 		modes = wpa_driver_nl80211_postprocess_modes(result.modes,
2735 							     num_modes);
2736 		nl80211_dump_chan_list(drv, modes, *num_modes);
2737 		return modes;
2738 	}
2739 
2740 	return NULL;
2741 }
2742