xref: /freebsd/contrib/wpa/src/drivers/driver_nl80211_scan.c (revision a90b9d0159070121c221b966469c3e36d912bf82)
1 /*
2  * Driver interaction with Linux nl80211/cfg80211 - Scanning
3  * Copyright(c) 2015 Intel Deutschland GmbH
4  * Copyright (c) 2002-2014, Jouni Malinen <j@w1.fi>
5  * Copyright (c) 2007, Johannes Berg <johannes@sipsolutions.net>
6  * Copyright (c) 2009-2010, Atheros Communications
7  *
8  * This software may be distributed under the terms of the BSD license.
9  * See README for more details.
10  */
11 
12 #include "includes.h"
13 #include <time.h>
14 #include <netlink/genl/genl.h>
15 
16 #include "utils/common.h"
17 #include "utils/eloop.h"
18 #include "common/ieee802_11_defs.h"
19 #include "common/ieee802_11_common.h"
20 #include "common/qca-vendor.h"
21 #include "driver_nl80211.h"
22 
23 
24 #define MAX_NL80211_NOISE_FREQS 100
25 
26 struct nl80211_noise_info {
27 	u32 freq[MAX_NL80211_NOISE_FREQS];
28 	s8 noise[MAX_NL80211_NOISE_FREQS];
29 	unsigned int count;
30 };
31 
get_noise_for_scan_results(struct nl_msg * msg,void * arg)32 static int get_noise_for_scan_results(struct nl_msg *msg, void *arg)
33 {
34 	struct nlattr *tb[NL80211_ATTR_MAX + 1];
35 	struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
36 	struct nlattr *sinfo[NL80211_SURVEY_INFO_MAX + 1];
37 	static struct nla_policy survey_policy[NL80211_SURVEY_INFO_MAX + 1] = {
38 		[NL80211_SURVEY_INFO_FREQUENCY] = { .type = NLA_U32 },
39 		[NL80211_SURVEY_INFO_NOISE] = { .type = NLA_U8 },
40 	};
41 	struct nl80211_noise_info *info = arg;
42 
43 	if (info->count >= MAX_NL80211_NOISE_FREQS)
44 		return NL_SKIP;
45 
46 	nla_parse(tb, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
47 		  genlmsg_attrlen(gnlh, 0), NULL);
48 
49 	if (!tb[NL80211_ATTR_SURVEY_INFO]) {
50 		wpa_printf(MSG_DEBUG, "nl80211: Survey data missing");
51 		return NL_SKIP;
52 	}
53 
54 	if (nla_parse_nested(sinfo, NL80211_SURVEY_INFO_MAX,
55 			     tb[NL80211_ATTR_SURVEY_INFO],
56 			     survey_policy)) {
57 		wpa_printf(MSG_DEBUG, "nl80211: Failed to parse nested "
58 			   "attributes");
59 		return NL_SKIP;
60 	}
61 
62 	if (!sinfo[NL80211_SURVEY_INFO_NOISE])
63 		return NL_SKIP;
64 
65 	if (!sinfo[NL80211_SURVEY_INFO_FREQUENCY])
66 		return NL_SKIP;
67 
68 	info->freq[info->count] =
69 		nla_get_u32(sinfo[NL80211_SURVEY_INFO_FREQUENCY]);
70 	info->noise[info->count] =
71 		(s8) nla_get_u8(sinfo[NL80211_SURVEY_INFO_NOISE]);
72 	info->count++;
73 
74 	return NL_SKIP;
75 }
76 
77 
nl80211_get_noise_for_scan_results(struct wpa_driver_nl80211_data * drv,struct nl80211_noise_info * info)78 static int nl80211_get_noise_for_scan_results(
79 	struct wpa_driver_nl80211_data *drv, struct nl80211_noise_info *info)
80 {
81 	struct nl_msg *msg;
82 
83 	os_memset(info, 0, sizeof(*info));
84 	msg = nl80211_drv_msg(drv, NLM_F_DUMP, NL80211_CMD_GET_SURVEY);
85 	return send_and_recv_resp(drv, msg, get_noise_for_scan_results, info);
86 }
87 
88 
nl80211_abort_scan(struct i802_bss * bss)89 static int nl80211_abort_scan(struct i802_bss *bss)
90 {
91 	int ret;
92 	struct nl_msg *msg;
93 	struct wpa_driver_nl80211_data *drv = bss->drv;
94 
95 	wpa_printf(MSG_DEBUG, "nl80211: Abort scan");
96 	msg = nl80211_cmd_msg(bss, 0, NL80211_CMD_ABORT_SCAN);
97 	ret = send_and_recv_cmd(drv, msg);
98 	if (ret) {
99 		wpa_printf(MSG_DEBUG, "nl80211: Abort scan failed: ret=%d (%s)",
100 			   ret, strerror(-ret));
101 	}
102 	return ret;
103 }
104 
105 
106 #ifdef CONFIG_DRIVER_NL80211_QCA
nl80211_abort_vendor_scan(struct wpa_driver_nl80211_data * drv,u64 scan_cookie)107 static int nl80211_abort_vendor_scan(struct wpa_driver_nl80211_data *drv,
108 				     u64 scan_cookie)
109 {
110 	struct nl_msg *msg;
111 	struct nlattr *params;
112 	int ret;
113 
114 	wpa_printf(MSG_DEBUG, "nl80211: Abort vendor scan with cookie 0x%llx",
115 		   (long long unsigned int) scan_cookie);
116 
117 	msg = nl80211_drv_msg(drv, 0, NL80211_CMD_VENDOR);
118 	if (!msg ||
119 	    nla_put_u32(msg, NL80211_ATTR_VENDOR_ID, OUI_QCA) ||
120 	    nla_put_u32(msg, NL80211_ATTR_VENDOR_SUBCMD,
121 			QCA_NL80211_VENDOR_SUBCMD_ABORT_SCAN) ||
122 	    !(params = nla_nest_start(msg, NL80211_ATTR_VENDOR_DATA)) ||
123 	    nla_put_u64(msg, QCA_WLAN_VENDOR_ATTR_SCAN_COOKIE, scan_cookie))
124 		goto fail;
125 
126 	nla_nest_end(msg, params);
127 
128 	ret = send_and_recv_cmd(drv, msg);
129 	msg = NULL;
130 	if (ret) {
131 		wpa_printf(MSG_INFO,
132 			   "nl80211: Aborting vendor scan with cookie 0x%llx failed: ret=%d (%s)",
133 			   (long long unsigned int) scan_cookie, ret,
134 			   strerror(-ret));
135 		goto fail;
136 	}
137 	return 0;
138 fail:
139 	nlmsg_free(msg);
140 	return -1;
141 }
142 #endif /* CONFIG_DRIVER_NL80211_QCA */
143 
144 
145 /**
146  * wpa_driver_nl80211_scan_timeout - Scan timeout to report scan completion
147  * @eloop_ctx: Driver private data
148  * @timeout_ctx: ctx argument given to wpa_driver_nl80211_init()
149  *
150  * This function can be used as registered timeout when starting a scan to
151  * generate a scan completed event if the driver does not report this.
152  */
wpa_driver_nl80211_scan_timeout(void * eloop_ctx,void * timeout_ctx)153 void wpa_driver_nl80211_scan_timeout(void *eloop_ctx, void *timeout_ctx)
154 {
155 	struct wpa_driver_nl80211_data *drv = eloop_ctx;
156 
157 	wpa_printf(MSG_DEBUG, "nl80211: Scan timeout - try to abort it");
158 #ifdef CONFIG_DRIVER_NL80211_QCA
159 	if (drv->vendor_scan_cookie &&
160 	    nl80211_abort_vendor_scan(drv, drv->vendor_scan_cookie) == 0)
161 		return;
162 #endif /* CONFIG_DRIVER_NL80211_QCA */
163 	if (!drv->vendor_scan_cookie &&
164 	    nl80211_abort_scan(drv->first_bss) == 0)
165 		return;
166 
167 	wpa_printf(MSG_DEBUG, "nl80211: Failed to abort scan");
168 
169 	if (drv->ap_scan_as_station != NL80211_IFTYPE_UNSPECIFIED)
170 		nl80211_restore_ap_mode(drv->first_bss);
171 
172 	wpa_printf(MSG_DEBUG, "nl80211: Try to get scan results");
173 	wpa_supplicant_event(timeout_ctx, EVENT_SCAN_RESULTS, NULL);
174 }
175 
176 
177 static struct nl_msg *
nl80211_scan_common(struct i802_bss * bss,u8 cmd,struct wpa_driver_scan_params * params)178 nl80211_scan_common(struct i802_bss *bss, u8 cmd,
179 		    struct wpa_driver_scan_params *params)
180 {
181 	struct wpa_driver_nl80211_data *drv = bss->drv;
182 	struct nl_msg *msg;
183 	size_t i;
184 	u32 scan_flags = 0;
185 
186 	msg = nl80211_cmd_msg(bss, 0, cmd);
187 	if (!msg)
188 		return NULL;
189 
190 	if (params->num_ssids) {
191 		struct nlattr *ssids;
192 
193 		ssids = nla_nest_start(msg, NL80211_ATTR_SCAN_SSIDS);
194 		if (ssids == NULL)
195 			goto fail;
196 		for (i = 0; i < params->num_ssids; i++) {
197 			wpa_printf(MSG_MSGDUMP, "nl80211: Scan SSID %s",
198 				   wpa_ssid_txt(params->ssids[i].ssid,
199 						params->ssids[i].ssid_len));
200 			if (nla_put(msg, i + 1, params->ssids[i].ssid_len,
201 				    params->ssids[i].ssid))
202 				goto fail;
203 		}
204 		nla_nest_end(msg, ssids);
205 
206 		/*
207 		 * If allowed, scan for 6 GHz APs that are reported by other
208 		 * APs. Note that if the flag is not set and 6 GHz channels are
209 		 * to be scanned, it is highly likely that non-PSC channels
210 		 * would be scanned passively (due to the Probe Request frame
211 		 * transmission restrictions mandated in IEEE Std 802.11ax-2021,
212 		 * 26.17.2.3 (Scanning in the 6 GHz band). Passive scanning of
213 		 * all non-PSC channels would take a significant amount of time.
214 		 */
215 		if (!params->non_coloc_6ghz) {
216 			wpa_printf(MSG_DEBUG,
217 				   "nl80211: Scan co-located APs on 6 GHz");
218 			scan_flags |= NL80211_SCAN_FLAG_COLOCATED_6GHZ;
219 		}
220 	} else {
221 		wpa_printf(MSG_DEBUG, "nl80211: Passive scan requested");
222 	}
223 
224 	if (params->extra_ies) {
225 		wpa_hexdump(MSG_MSGDUMP, "nl80211: Scan extra IEs",
226 			    params->extra_ies, params->extra_ies_len);
227 		if (nla_put(msg, NL80211_ATTR_IE, params->extra_ies_len,
228 			    params->extra_ies))
229 			goto fail;
230 	}
231 
232 	if (params->freqs) {
233 		struct nlattr *freqs;
234 		freqs = nla_nest_start(msg, NL80211_ATTR_SCAN_FREQUENCIES);
235 		if (freqs == NULL)
236 			goto fail;
237 		for (i = 0; params->freqs[i]; i++) {
238 			wpa_printf(MSG_MSGDUMP, "nl80211: Scan frequency %u "
239 				   "MHz", params->freqs[i]);
240 			if (nla_put_u32(msg, i + 1, params->freqs[i]))
241 				goto fail;
242 		}
243 		nla_nest_end(msg, freqs);
244 	}
245 
246 	os_free(drv->filter_ssids);
247 	drv->filter_ssids = params->filter_ssids;
248 	params->filter_ssids = NULL;
249 	drv->num_filter_ssids = params->num_filter_ssids;
250 
251 	if (!drv->hostapd && is_ap_interface(drv->nlmode)) {
252 		wpa_printf(MSG_DEBUG, "nl80211: Add NL80211_SCAN_FLAG_AP");
253 		scan_flags |= NL80211_SCAN_FLAG_AP;
254 	}
255 
256 	if (params->only_new_results) {
257 		wpa_printf(MSG_DEBUG, "nl80211: Add NL80211_SCAN_FLAG_FLUSH");
258 		scan_flags |= NL80211_SCAN_FLAG_FLUSH;
259 	}
260 
261 	if (params->low_priority && drv->have_low_prio_scan) {
262 		wpa_printf(MSG_DEBUG,
263 			   "nl80211: Add NL80211_SCAN_FLAG_LOW_PRIORITY");
264 		scan_flags |= NL80211_SCAN_FLAG_LOW_PRIORITY;
265 	}
266 
267 	if (params->mac_addr_rand) {
268 		wpa_printf(MSG_DEBUG,
269 			   "nl80211: Add NL80211_SCAN_FLAG_RANDOM_ADDR");
270 		scan_flags |= NL80211_SCAN_FLAG_RANDOM_ADDR;
271 
272 		if (params->mac_addr) {
273 			wpa_printf(MSG_DEBUG, "nl80211: MAC address: " MACSTR,
274 				   MAC2STR(params->mac_addr));
275 			if (nla_put(msg, NL80211_ATTR_MAC, ETH_ALEN,
276 				    params->mac_addr))
277 				goto fail;
278 		}
279 
280 		if (params->mac_addr_mask) {
281 			wpa_printf(MSG_DEBUG, "nl80211: MAC address mask: "
282 				   MACSTR, MAC2STR(params->mac_addr_mask));
283 			if (nla_put(msg, NL80211_ATTR_MAC_MASK, ETH_ALEN,
284 				    params->mac_addr_mask))
285 				goto fail;
286 		}
287 	}
288 
289 	if (params->duration) {
290 		if (!(drv->capa.rrm_flags &
291 		      WPA_DRIVER_FLAGS_SUPPORT_SET_SCAN_DWELL) ||
292 		    nla_put_u16(msg, NL80211_ATTR_MEASUREMENT_DURATION,
293 				params->duration))
294 			goto fail;
295 
296 		if (params->duration_mandatory &&
297 		    nla_put_flag(msg,
298 				 NL80211_ATTR_MEASUREMENT_DURATION_MANDATORY))
299 			goto fail;
300 	}
301 
302 	if (params->oce_scan) {
303 		wpa_printf(MSG_DEBUG,
304 			   "nl80211: Add NL80211_SCAN_FLAG_FILS_MAX_CHANNEL_TIME");
305 		wpa_printf(MSG_DEBUG,
306 			   "nl80211: Add NL80211_SCAN_FLAG_ACCEPT_BCAST_PROBE_RESP");
307 		wpa_printf(MSG_DEBUG,
308 			   "nl80211: Add NL80211_SCAN_FLAG_OCE_PROBE_REQ_MIN_TX_RATE");
309 		wpa_printf(MSG_DEBUG,
310 			   "nl80211: Add NL80211_SCAN_FLAG_OCE_PROBE_REQ_DEFERRAL_SUPPRESSION");
311 		scan_flags |= NL80211_SCAN_FLAG_FILS_MAX_CHANNEL_TIME |
312 			NL80211_SCAN_FLAG_ACCEPT_BCAST_PROBE_RESP |
313 			NL80211_SCAN_FLAG_OCE_PROBE_REQ_HIGH_TX_RATE |
314 			NL80211_SCAN_FLAG_OCE_PROBE_REQ_DEFERRAL_SUPPRESSION;
315 	}
316 
317 	if (params->min_probe_req_content) {
318 		if (drv->capa.flags2 & WPA_DRIVER_FLAGS2_SCAN_MIN_PREQ)
319 			scan_flags |= NL80211_SCAN_FLAG_MIN_PREQ_CONTENT;
320 		else
321 			wpa_printf(MSG_DEBUG,
322 				   "nl80211: NL80211_SCAN_FLAG_MIN_PREQ_CONTENT not supported");
323 	}
324 
325 	if (scan_flags &&
326 	    nla_put_u32(msg, NL80211_ATTR_SCAN_FLAGS, scan_flags))
327 		goto fail;
328 
329 	return msg;
330 
331 fail:
332 	nlmsg_free(msg);
333 	return NULL;
334 }
335 
336 
337 /**
338  * wpa_driver_nl80211_scan - Request the driver to initiate scan
339  * @bss: Pointer to private driver data from wpa_driver_nl80211_init()
340  * @params: Scan parameters
341  * Returns: 0 on success, -1 on failure
342  */
wpa_driver_nl80211_scan(struct i802_bss * bss,struct wpa_driver_scan_params * params)343 int wpa_driver_nl80211_scan(struct i802_bss *bss,
344 			    struct wpa_driver_scan_params *params)
345 {
346 	struct wpa_driver_nl80211_data *drv = bss->drv;
347 	int ret = -1, timeout;
348 	struct nl_msg *msg = NULL;
349 
350 	wpa_dbg(drv->ctx, MSG_DEBUG, "nl80211: scan request");
351 	drv->scan_for_auth = 0;
352 
353 	if (TEST_FAIL())
354 		return -1;
355 
356 	msg = nl80211_scan_common(bss, NL80211_CMD_TRIGGER_SCAN, params);
357 	if (!msg)
358 		return -1;
359 
360 	if (params->p2p_probe) {
361 		struct nlattr *rates;
362 
363 		wpa_printf(MSG_DEBUG, "nl80211: P2P probe - mask SuppRates");
364 
365 		rates = nla_nest_start(msg, NL80211_ATTR_SCAN_SUPP_RATES);
366 		if (rates == NULL)
367 			goto fail;
368 
369 		/*
370 		 * Remove 2.4 GHz rates 1, 2, 5.5, 11 Mbps from supported rates
371 		 * by masking out everything else apart from the OFDM rates 6,
372 		 * 9, 12, 18, 24, 36, 48, 54 Mbps from non-MCS rates. All 5 GHz
373 		 * rates are left enabled.
374 		 */
375 		if (nla_put(msg, NL80211_BAND_2GHZ, 8,
376 			    "\x0c\x12\x18\x24\x30\x48\x60\x6c"))
377 			goto fail;
378 		nla_nest_end(msg, rates);
379 
380 		if (nla_put_flag(msg, NL80211_ATTR_TX_NO_CCK_RATE))
381 			goto fail;
382 	}
383 
384 	if (params->bssid) {
385 		wpa_printf(MSG_DEBUG, "nl80211: Scan for a specific BSSID: "
386 			   MACSTR, MAC2STR(params->bssid));
387 		if (nla_put(msg, NL80211_ATTR_BSSID, ETH_ALEN, params->bssid))
388 			goto fail;
389 		/* NL80211_ATTR_MAC was used for this purpose initially and the
390 		 * NL80211_ATTR_BSSID was added in 2016 when MAC address
391 		 * randomization was added. For compatibility with older kernel
392 		 * versions, add the NL80211_ATTR_MAC attribute as well when
393 		 * the conflicting functionality is not in use. */
394 		if (!params->mac_addr_rand &&
395 		    nla_put(msg, NL80211_ATTR_MAC, ETH_ALEN, params->bssid))
396 			goto fail;
397 	}
398 
399 	ret = send_and_recv_cmd(drv, msg);
400 	msg = NULL;
401 	if (ret) {
402 		wpa_printf(MSG_DEBUG, "nl80211: Scan trigger failed: ret=%d "
403 			   "(%s)", ret, strerror(-ret));
404 		if (drv->hostapd && is_ap_interface(drv->nlmode)) {
405 			/*
406 			 * mac80211 does not allow scan requests in AP mode, so
407 			 * try to do this in station mode.
408 			 */
409 			drv->ap_scan_as_station = drv->nlmode;
410 			if (wpa_driver_nl80211_set_mode(
411 				    bss, NL80211_IFTYPE_STATION) ||
412 			    wpa_driver_nl80211_scan(bss, params)) {
413 				nl80211_restore_ap_mode(bss);
414 				goto fail;
415 			}
416 
417 			ret = 0;
418 		} else
419 			goto fail;
420 	}
421 
422 	drv->scan_state = SCAN_REQUESTED;
423 	/* Not all drivers generate "scan completed" wireless event, so try to
424 	 * read results after a timeout. */
425 	timeout = drv->uses_6ghz ? 20 : 10;
426 	if (drv->uses_s1g)
427 		timeout += 5;
428 	if (drv->scan_complete_events) {
429 		/*
430 		 * The driver seems to deliver events to notify when scan is
431 		 * complete, so use longer timeout to avoid race conditions
432 		 * with scanning and following association request.
433 		 */
434 		timeout = 30;
435 	}
436 	wpa_printf(MSG_DEBUG, "Scan requested (ret=%d) - scan timeout %d "
437 		   "seconds", ret, timeout);
438 	eloop_cancel_timeout(wpa_driver_nl80211_scan_timeout, drv, drv->ctx);
439 	eloop_register_timeout(timeout, 0, wpa_driver_nl80211_scan_timeout,
440 			       drv, drv->ctx);
441 	drv->last_scan_cmd = NL80211_CMD_TRIGGER_SCAN;
442 
443 fail:
444 	nlmsg_free(msg);
445 	return ret;
446 }
447 
448 
449 static int
nl80211_sched_scan_add_scan_plans(struct wpa_driver_nl80211_data * drv,struct nl_msg * msg,struct wpa_driver_scan_params * params)450 nl80211_sched_scan_add_scan_plans(struct wpa_driver_nl80211_data *drv,
451 				  struct nl_msg *msg,
452 				  struct wpa_driver_scan_params *params)
453 {
454 	struct nlattr *plans;
455 	struct sched_scan_plan *scan_plans = params->sched_scan_plans;
456 	unsigned int i;
457 
458 	plans = nla_nest_start(msg, NL80211_ATTR_SCHED_SCAN_PLANS);
459 	if (!plans)
460 		return -1;
461 
462 	for (i = 0; i < params->sched_scan_plans_num; i++) {
463 		struct nlattr *plan = nla_nest_start(msg, i + 1);
464 
465 		if (!plan)
466 			return -1;
467 
468 		if (!scan_plans[i].interval ||
469 		    scan_plans[i].interval >
470 		    drv->capa.max_sched_scan_plan_interval) {
471 			wpa_printf(MSG_DEBUG,
472 				   "nl80211: sched scan plan no. %u: Invalid interval: %u",
473 				   i, scan_plans[i].interval);
474 			return -1;
475 		}
476 
477 		if (nla_put_u32(msg, NL80211_SCHED_SCAN_PLAN_INTERVAL,
478 				scan_plans[i].interval))
479 			return -1;
480 
481 		if (scan_plans[i].iterations >
482 		    drv->capa.max_sched_scan_plan_iterations) {
483 			wpa_printf(MSG_DEBUG,
484 				   "nl80211: sched scan plan no. %u: Invalid number of iterations: %u",
485 				   i, scan_plans[i].iterations);
486 			return -1;
487 		}
488 
489 		if (scan_plans[i].iterations &&
490 		    nla_put_u32(msg, NL80211_SCHED_SCAN_PLAN_ITERATIONS,
491 				scan_plans[i].iterations))
492 			return -1;
493 
494 		nla_nest_end(msg, plan);
495 
496 		/*
497 		 * All the scan plans must specify the number of iterations
498 		 * except the last plan, which will run infinitely. So if the
499 		 * number of iterations is not specified, this ought to be the
500 		 * last scan plan.
501 		 */
502 		if (!scan_plans[i].iterations)
503 			break;
504 	}
505 
506 	if (i != params->sched_scan_plans_num - 1) {
507 		wpa_printf(MSG_DEBUG,
508 			   "nl80211: All sched scan plans but the last must specify number of iterations");
509 		return -1;
510 	}
511 
512 	nla_nest_end(msg, plans);
513 	return 0;
514 }
515 
516 
517 /**
518  * wpa_driver_nl80211_sched_scan - Initiate a scheduled scan
519  * @priv: Pointer to private driver data from wpa_driver_nl80211_init()
520  * @params: Scan parameters
521  * Returns: 0 on success, -1 on failure or if not supported
522  */
wpa_driver_nl80211_sched_scan(void * priv,struct wpa_driver_scan_params * params)523 int wpa_driver_nl80211_sched_scan(void *priv,
524 				  struct wpa_driver_scan_params *params)
525 {
526 	struct i802_bss *bss = priv;
527 	struct wpa_driver_nl80211_data *drv = bss->drv;
528 	int ret = -1;
529 	struct nl_msg *msg;
530 	size_t i;
531 
532 	wpa_dbg(drv->ctx, MSG_DEBUG, "nl80211: sched_scan request");
533 
534 #ifdef ANDROID
535 	if (!drv->capa.sched_scan_supported)
536 		return android_pno_start(bss, params);
537 #endif /* ANDROID */
538 
539 	if (!params->sched_scan_plans_num ||
540 	    params->sched_scan_plans_num > drv->capa.max_sched_scan_plans) {
541 		wpa_printf(MSG_ERROR,
542 			   "nl80211: Invalid number of sched scan plans: %u",
543 			   params->sched_scan_plans_num);
544 		return -1;
545 	}
546 
547 	msg = nl80211_scan_common(bss, NL80211_CMD_START_SCHED_SCAN, params);
548 	if (!msg)
549 		goto fail;
550 
551 	if (drv->capa.max_sched_scan_plan_iterations) {
552 		if (nl80211_sched_scan_add_scan_plans(drv, msg, params))
553 			goto fail;
554 	} else {
555 		if (nla_put_u32(msg, NL80211_ATTR_SCHED_SCAN_INTERVAL,
556 				params->sched_scan_plans[0].interval * 1000))
557 			goto fail;
558 	}
559 
560 	if ((drv->num_filter_ssids &&
561 	    (int) drv->num_filter_ssids <= drv->capa.max_match_sets) ||
562 	    params->filter_rssi) {
563 		struct nlattr *match_sets;
564 		match_sets = nla_nest_start(msg, NL80211_ATTR_SCHED_SCAN_MATCH);
565 		if (match_sets == NULL)
566 			goto fail;
567 
568 		for (i = 0; i < drv->num_filter_ssids; i++) {
569 			struct nlattr *match_set_ssid;
570 			wpa_printf(MSG_MSGDUMP,
571 				   "nl80211: Sched scan filter SSID %s",
572 				   wpa_ssid_txt(drv->filter_ssids[i].ssid,
573 						drv->filter_ssids[i].ssid_len));
574 
575 			match_set_ssid = nla_nest_start(msg, i + 1);
576 			if (match_set_ssid == NULL ||
577 			    nla_put(msg, NL80211_ATTR_SCHED_SCAN_MATCH_SSID,
578 				    drv->filter_ssids[i].ssid_len,
579 				    drv->filter_ssids[i].ssid) ||
580 			    (params->filter_rssi &&
581 			     nla_put_u32(msg,
582 					 NL80211_SCHED_SCAN_MATCH_ATTR_RSSI,
583 					 params->filter_rssi)))
584 				goto fail;
585 
586 			nla_nest_end(msg, match_set_ssid);
587 		}
588 
589 		/*
590 		 * Due to backward compatibility code, newer kernels treat this
591 		 * matchset (with only an RSSI filter) as the default for all
592 		 * other matchsets, unless it's the only one, in which case the
593 		 * matchset will actually allow all SSIDs above the RSSI.
594 		 */
595 		if (params->filter_rssi) {
596 			struct nlattr *match_set_rssi;
597 			match_set_rssi = nla_nest_start(msg, 0);
598 			if (match_set_rssi == NULL ||
599 			    nla_put_u32(msg, NL80211_SCHED_SCAN_MATCH_ATTR_RSSI,
600 					params->filter_rssi))
601 				goto fail;
602 			wpa_printf(MSG_MSGDUMP,
603 				   "nl80211: Sched scan RSSI filter %d dBm",
604 				   params->filter_rssi);
605 			nla_nest_end(msg, match_set_rssi);
606 		}
607 
608 		nla_nest_end(msg, match_sets);
609 	}
610 
611 	if (params->relative_rssi_set) {
612 		struct nl80211_bss_select_rssi_adjust rssi_adjust;
613 
614 		os_memset(&rssi_adjust, 0, sizeof(rssi_adjust));
615 		wpa_printf(MSG_DEBUG, "nl80211: Relative RSSI: %d",
616 			   params->relative_rssi);
617 		if (nla_put_u32(msg, NL80211_ATTR_SCHED_SCAN_RELATIVE_RSSI,
618 				params->relative_rssi))
619 			goto fail;
620 
621 		if (params->relative_adjust_rssi) {
622 			int pref_band_set = 1;
623 
624 			switch (params->relative_adjust_band) {
625 			case WPA_SETBAND_5G:
626 				rssi_adjust.band = NL80211_BAND_5GHZ;
627 				break;
628 			case WPA_SETBAND_2G:
629 				rssi_adjust.band = NL80211_BAND_2GHZ;
630 				break;
631 			default:
632 				pref_band_set = 0;
633 				break;
634 			}
635 			rssi_adjust.delta = params->relative_adjust_rssi;
636 
637 			if (pref_band_set &&
638 			    nla_put(msg, NL80211_ATTR_SCHED_SCAN_RSSI_ADJUST,
639 				    sizeof(rssi_adjust), &rssi_adjust))
640 				goto fail;
641 		}
642 	}
643 
644 	if (params->sched_scan_start_delay &&
645 	    nla_put_u32(msg, NL80211_ATTR_SCHED_SCAN_DELAY,
646 			params->sched_scan_start_delay))
647 		goto fail;
648 
649 	ret = send_and_recv_cmd(drv, msg);
650 
651 	/* TODO: if we get an error here, we should fall back to normal scan */
652 
653 	msg = NULL;
654 	if (ret) {
655 		wpa_printf(MSG_DEBUG, "nl80211: Sched scan start failed: "
656 			   "ret=%d (%s)", ret, strerror(-ret));
657 		goto fail;
658 	}
659 
660 	wpa_printf(MSG_DEBUG, "nl80211: Sched scan requested (ret=%d)", ret);
661 
662 fail:
663 	nlmsg_free(msg);
664 	return ret;
665 }
666 
667 
668 /**
669  * wpa_driver_nl80211_stop_sched_scan - Stop a scheduled scan
670  * @priv: Pointer to private driver data from wpa_driver_nl80211_init()
671  * Returns: 0 on success, -1 on failure or if not supported
672  */
wpa_driver_nl80211_stop_sched_scan(void * priv)673 int wpa_driver_nl80211_stop_sched_scan(void *priv)
674 {
675 	struct i802_bss *bss = priv;
676 	struct wpa_driver_nl80211_data *drv = bss->drv;
677 	int ret;
678 	struct nl_msg *msg;
679 
680 #ifdef ANDROID
681 	if (!drv->capa.sched_scan_supported)
682 		return android_pno_stop(bss);
683 #endif /* ANDROID */
684 
685 	msg = nl80211_drv_msg(drv, 0, NL80211_CMD_STOP_SCHED_SCAN);
686 	ret = send_and_recv_cmd(drv, msg);
687 	if (ret) {
688 		wpa_printf(MSG_DEBUG,
689 			   "nl80211: Sched scan stop failed: ret=%d (%s)",
690 			   ret, strerror(-ret));
691 	} else {
692 		wpa_printf(MSG_DEBUG,
693 			   "nl80211: Sched scan stop sent");
694 	}
695 
696 	return ret;
697 }
698 
699 
nl80211_scan_filtered(struct wpa_driver_nl80211_data * drv,const u8 * ie,size_t ie_len)700 static int nl80211_scan_filtered(struct wpa_driver_nl80211_data *drv,
701 				 const u8 *ie, size_t ie_len)
702 {
703 	const u8 *ssid;
704 	size_t i;
705 
706 	if (drv->filter_ssids == NULL)
707 		return 0;
708 
709 	ssid = get_ie(ie, ie_len, WLAN_EID_SSID);
710 	if (ssid == NULL)
711 		return 1;
712 
713 	for (i = 0; i < drv->num_filter_ssids; i++) {
714 		if (ssid[1] == drv->filter_ssids[i].ssid_len &&
715 		    os_memcmp(ssid + 2, drv->filter_ssids[i].ssid, ssid[1]) ==
716 		    0)
717 			return 0;
718 	}
719 
720 	return 1;
721 }
722 
723 
724 static struct wpa_scan_res *
nl80211_parse_bss_info(struct wpa_driver_nl80211_data * drv,struct nl_msg * msg,const u8 * bssid)725 nl80211_parse_bss_info(struct wpa_driver_nl80211_data *drv,
726 		       struct nl_msg *msg, const u8 *bssid)
727 {
728 	struct nlattr *tb[NL80211_ATTR_MAX + 1];
729 	struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
730 	struct nlattr *bss[NL80211_BSS_MAX + 1];
731 	static struct nla_policy bss_policy[NL80211_BSS_MAX + 1] = {
732 		[NL80211_BSS_BSSID] = { .type = NLA_UNSPEC },
733 		[NL80211_BSS_FREQUENCY] = { .type = NLA_U32 },
734 		[NL80211_BSS_TSF] = { .type = NLA_U64 },
735 		[NL80211_BSS_BEACON_INTERVAL] = { .type = NLA_U16 },
736 		[NL80211_BSS_CAPABILITY] = { .type = NLA_U16 },
737 		[NL80211_BSS_INFORMATION_ELEMENTS] = { .type = NLA_UNSPEC },
738 		[NL80211_BSS_SIGNAL_MBM] = { .type = NLA_U32 },
739 		[NL80211_BSS_SIGNAL_UNSPEC] = { .type = NLA_U8 },
740 		[NL80211_BSS_STATUS] = { .type = NLA_U32 },
741 		[NL80211_BSS_SEEN_MS_AGO] = { .type = NLA_U32 },
742 		[NL80211_BSS_BEACON_IES] = { .type = NLA_UNSPEC },
743 		[NL80211_BSS_BEACON_TSF] = { .type = NLA_U64 },
744 		[NL80211_BSS_PARENT_TSF] = { .type = NLA_U64 },
745 		[NL80211_BSS_PARENT_BSSID] = { .type = NLA_UNSPEC },
746 		[NL80211_BSS_LAST_SEEN_BOOTTIME] = { .type = NLA_U64 },
747 	};
748 	struct wpa_scan_res *r;
749 	const u8 *ie, *beacon_ie;
750 	size_t ie_len, beacon_ie_len;
751 	u8 *pos;
752 
753 	nla_parse(tb, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
754 		  genlmsg_attrlen(gnlh, 0), NULL);
755 	if (!tb[NL80211_ATTR_BSS])
756 		return NULL;
757 	if (nla_parse_nested(bss, NL80211_BSS_MAX, tb[NL80211_ATTR_BSS],
758 			     bss_policy))
759 		return NULL;
760 	if (bssid && bss[NL80211_BSS_BSSID] &&
761 	    !ether_addr_equal(bssid, nla_data(bss[NL80211_BSS_BSSID])))
762 		return NULL;
763 	if (bss[NL80211_BSS_INFORMATION_ELEMENTS]) {
764 		ie = nla_data(bss[NL80211_BSS_INFORMATION_ELEMENTS]);
765 		ie_len = nla_len(bss[NL80211_BSS_INFORMATION_ELEMENTS]);
766 	} else {
767 		ie = NULL;
768 		ie_len = 0;
769 	}
770 	if (bss[NL80211_BSS_BEACON_IES]) {
771 		beacon_ie = nla_data(bss[NL80211_BSS_BEACON_IES]);
772 		beacon_ie_len = nla_len(bss[NL80211_BSS_BEACON_IES]);
773 	} else {
774 		beacon_ie = NULL;
775 		beacon_ie_len = 0;
776 	}
777 
778 	if (nl80211_scan_filtered(drv, ie ? ie : beacon_ie,
779 				  ie ? ie_len : beacon_ie_len))
780 		return NULL;
781 
782 	r = os_zalloc(sizeof(*r) + ie_len + beacon_ie_len);
783 	if (r == NULL)
784 		return NULL;
785 	if (bss[NL80211_BSS_BSSID])
786 		os_memcpy(r->bssid, nla_data(bss[NL80211_BSS_BSSID]),
787 			  ETH_ALEN);
788 	if (bss[NL80211_BSS_FREQUENCY])
789 		r->freq = nla_get_u32(bss[NL80211_BSS_FREQUENCY]);
790 	if (bss[NL80211_BSS_BEACON_INTERVAL])
791 		r->beacon_int = nla_get_u16(bss[NL80211_BSS_BEACON_INTERVAL]);
792 	if (bss[NL80211_BSS_CAPABILITY])
793 		r->caps = nla_get_u16(bss[NL80211_BSS_CAPABILITY]);
794 	r->flags |= WPA_SCAN_NOISE_INVALID;
795 	if (bss[NL80211_BSS_SIGNAL_MBM]) {
796 		r->level = nla_get_u32(bss[NL80211_BSS_SIGNAL_MBM]);
797 		r->level /= 100; /* mBm to dBm */
798 		r->flags |= WPA_SCAN_LEVEL_DBM | WPA_SCAN_QUAL_INVALID;
799 	} else if (bss[NL80211_BSS_SIGNAL_UNSPEC]) {
800 		r->level = nla_get_u8(bss[NL80211_BSS_SIGNAL_UNSPEC]);
801 		r->flags |= WPA_SCAN_QUAL_INVALID;
802 	} else
803 		r->flags |= WPA_SCAN_LEVEL_INVALID | WPA_SCAN_QUAL_INVALID;
804 	if (bss[NL80211_BSS_TSF])
805 		r->tsf = nla_get_u64(bss[NL80211_BSS_TSF]);
806 	if (bss[NL80211_BSS_BEACON_TSF]) {
807 		u64 tsf = nla_get_u64(bss[NL80211_BSS_BEACON_TSF]);
808 		if (tsf > r->tsf) {
809 			r->tsf = tsf;
810 			r->beacon_newer = true;
811 		}
812 	}
813 	if (bss[NL80211_BSS_SEEN_MS_AGO])
814 		r->age = nla_get_u32(bss[NL80211_BSS_SEEN_MS_AGO]);
815 	if (bss[NL80211_BSS_LAST_SEEN_BOOTTIME]) {
816 		u64 boottime;
817 		struct timespec ts;
818 
819 #ifndef CLOCK_BOOTTIME
820 #define CLOCK_BOOTTIME 7
821 #endif
822 		if (clock_gettime(CLOCK_BOOTTIME, &ts) == 0) {
823 			/* Use more accurate boottime information to update the
824 			 * scan result age since the driver reports this and
825 			 * CLOCK_BOOTTIME is available. */
826 			boottime = nla_get_u64(
827 				bss[NL80211_BSS_LAST_SEEN_BOOTTIME]);
828 			r->age = ((u64) ts.tv_sec * 1000000000 +
829 				  ts.tv_nsec - boottime) / 1000000;
830 		}
831 	}
832 	r->ie_len = ie_len;
833 	pos = (u8 *) (r + 1);
834 	if (ie) {
835 		os_memcpy(pos, ie, ie_len);
836 		pos += ie_len;
837 	}
838 	r->beacon_ie_len = beacon_ie_len;
839 	if (beacon_ie)
840 		os_memcpy(pos, beacon_ie, beacon_ie_len);
841 
842 	if (bss[NL80211_BSS_STATUS]) {
843 		enum nl80211_bss_status status;
844 		status = nla_get_u32(bss[NL80211_BSS_STATUS]);
845 		switch (status) {
846 		case NL80211_BSS_STATUS_ASSOCIATED:
847 			r->flags |= WPA_SCAN_ASSOCIATED;
848 			break;
849 		default:
850 			break;
851 		}
852 	}
853 
854 	if (bss[NL80211_BSS_PARENT_TSF] && bss[NL80211_BSS_PARENT_BSSID]) {
855 		r->parent_tsf = nla_get_u64(bss[NL80211_BSS_PARENT_TSF]);
856 		os_memcpy(r->tsf_bssid, nla_data(bss[NL80211_BSS_PARENT_BSSID]),
857 			  ETH_ALEN);
858 	}
859 
860 	return r;
861 }
862 
863 
864 struct nl80211_bss_info_arg {
865 	struct wpa_driver_nl80211_data *drv;
866 	struct wpa_scan_results *res;
867 	const u8 *bssid;
868 };
869 
bss_info_handler(struct nl_msg * msg,void * arg)870 static int bss_info_handler(struct nl_msg *msg, void *arg)
871 {
872 	struct nl80211_bss_info_arg *_arg = arg;
873 	struct wpa_scan_results *res = _arg->res;
874 	struct wpa_scan_res **tmp;
875 	struct wpa_scan_res *r;
876 
877 	r = nl80211_parse_bss_info(_arg->drv, msg, _arg->bssid);
878 	if (!r)
879 		return NL_SKIP;
880 
881 	if (!res) {
882 		os_free(r);
883 		return NL_SKIP;
884 	}
885 	tmp = os_realloc_array(res->res, res->num + 1,
886 			       sizeof(struct wpa_scan_res *));
887 	if (tmp == NULL) {
888 		os_free(r);
889 		return NL_SKIP;
890 	}
891 	tmp[res->num++] = r;
892 	res->res = tmp;
893 
894 	return NL_SKIP;
895 }
896 
897 
clear_state_mismatch(struct wpa_driver_nl80211_data * drv,const u8 * addr)898 static void clear_state_mismatch(struct wpa_driver_nl80211_data *drv,
899 				 const u8 *addr)
900 {
901 	if (drv->capa.flags & WPA_DRIVER_FLAGS_SME) {
902 		wpa_printf(MSG_DEBUG, "nl80211: Clear possible state "
903 			   "mismatch (" MACSTR ")", MAC2STR(addr));
904 		wpa_driver_nl80211_mlme(drv, addr,
905 					NL80211_CMD_DEAUTHENTICATE,
906 					WLAN_REASON_PREV_AUTH_NOT_VALID, 1,
907 					drv->first_bss);
908 	}
909 }
910 
911 
nl80211_check_bss_status(struct wpa_driver_nl80211_data * drv,struct wpa_scan_res * r)912 static void nl80211_check_bss_status(struct wpa_driver_nl80211_data *drv,
913 				     struct wpa_scan_res *r)
914 {
915 	if (!(r->flags & WPA_SCAN_ASSOCIATED))
916 		return;
917 
918 	wpa_printf(MSG_DEBUG, "nl80211: Scan results indicate BSS status with "
919 		   MACSTR " as associated", MAC2STR(r->bssid));
920 	if (is_sta_interface(drv->nlmode) && !drv->associated) {
921 		wpa_printf(MSG_DEBUG,
922 			   "nl80211: Local state (not associated) does not match with BSS state");
923 		clear_state_mismatch(drv, r->bssid);
924 	} else if (is_sta_interface(drv->nlmode) &&
925 		   !ether_addr_equal(drv->bssid, r->bssid)) {
926 		wpa_printf(MSG_DEBUG,
927 			   "nl80211: Local state (associated with " MACSTR
928 			   ") does not match with BSS state",
929 			   MAC2STR(drv->bssid));
930 
931 		if (!ether_addr_equal(drv->sta_mlo_info.ap_mld_addr,
932 				      drv->bssid)) {
933 			clear_state_mismatch(drv, r->bssid);
934 
935 			if (!is_zero_ether_addr(drv->sta_mlo_info.ap_mld_addr))
936 				clear_state_mismatch(
937 					drv, drv->sta_mlo_info.ap_mld_addr);
938 			else
939 				clear_state_mismatch(drv, drv->bssid);
940 
941 		} else {
942 			wpa_printf(MSG_DEBUG,
943 				   "nl80211: BSSID is the MLD address");
944 		}
945 	}
946 }
947 
948 
wpa_driver_nl80211_check_bss_status(struct wpa_driver_nl80211_data * drv,struct wpa_scan_results * res)949 static void wpa_driver_nl80211_check_bss_status(
950 	struct wpa_driver_nl80211_data *drv, struct wpa_scan_results *res)
951 {
952 	size_t i;
953 
954 	for (i = 0; i < res->num; i++)
955 		nl80211_check_bss_status(drv, res->res[i]);
956 }
957 
958 
nl80211_update_scan_res_noise(struct wpa_scan_res * res,struct nl80211_noise_info * info)959 static void nl80211_update_scan_res_noise(struct wpa_scan_res *res,
960 					  struct nl80211_noise_info *info)
961 {
962 	unsigned int i;
963 
964 	for (i = 0; res && i < info->count; i++) {
965 		if ((int) info->freq[i] != res->freq ||
966 		    !(res->flags & WPA_SCAN_NOISE_INVALID))
967 			continue;
968 		res->noise = info->noise[i];
969 		res->flags &= ~WPA_SCAN_NOISE_INVALID;
970 	}
971 }
972 
973 
974 static struct wpa_scan_results *
nl80211_get_scan_results(struct wpa_driver_nl80211_data * drv,const u8 * bssid)975 nl80211_get_scan_results(struct wpa_driver_nl80211_data *drv, const u8 *bssid)
976 {
977 	struct nl_msg *msg;
978 	struct wpa_scan_results *res;
979 	int ret;
980 	struct nl80211_bss_info_arg arg;
981 	int count = 0;
982 
983 try_again:
984 	res = os_zalloc(sizeof(*res));
985 	if (res == NULL)
986 		return NULL;
987 	if (!(msg = nl80211_cmd_msg(drv->first_bss, NLM_F_DUMP,
988 				    NL80211_CMD_GET_SCAN))) {
989 		wpa_scan_results_free(res);
990 		return NULL;
991 	}
992 
993 	arg.drv = drv;
994 	arg.res = res;
995 	arg.bssid = bssid;
996 	ret = send_and_recv_resp(drv, msg, bss_info_handler, &arg);
997 	if (ret == -EAGAIN) {
998 		count++;
999 		if (count >= 10) {
1000 			wpa_printf(MSG_INFO,
1001 				   "nl80211: Failed to receive consistent scan result dump");
1002 		} else {
1003 			wpa_printf(MSG_DEBUG,
1004 				   "nl80211: Failed to receive consistent scan result dump - try again");
1005 			wpa_scan_results_free(res);
1006 			goto try_again;
1007 		}
1008 	}
1009 	if (ret == 0) {
1010 		struct nl80211_noise_info info;
1011 
1012 		wpa_printf(MSG_DEBUG, "nl80211: Received scan results (%lu "
1013 			   "BSSes)", (unsigned long) res->num);
1014 		if (nl80211_get_noise_for_scan_results(drv, &info) == 0) {
1015 			size_t i;
1016 
1017 			for (i = 0; i < res->num; ++i)
1018 				nl80211_update_scan_res_noise(res->res[i],
1019 							      &info);
1020 		}
1021 		return res;
1022 	}
1023 	wpa_printf(MSG_DEBUG, "nl80211: Scan result fetch failed: ret=%d "
1024 		   "(%s)", ret, strerror(-ret));
1025 	wpa_scan_results_free(res);
1026 	return NULL;
1027 }
1028 
1029 
1030 /**
1031  * wpa_driver_nl80211_get_scan_results - Fetch the latest scan results
1032  * @priv: Pointer to private nl80211 data from wpa_driver_nl80211_init()
1033  * @bssid: Return results only for the specified BSSID, %NULL for all
1034  * Returns: Scan results on success, -1 on failure
1035  */
wpa_driver_nl80211_get_scan_results(void * priv,const u8 * bssid)1036 struct wpa_scan_results * wpa_driver_nl80211_get_scan_results(void *priv,
1037 							      const u8 *bssid)
1038 {
1039 	struct i802_bss *bss = priv;
1040 	struct wpa_driver_nl80211_data *drv = bss->drv;
1041 	struct wpa_scan_results *res;
1042 
1043 	res = nl80211_get_scan_results(drv, bssid);
1044 	if (res)
1045 		wpa_driver_nl80211_check_bss_status(drv, res);
1046 	return res;
1047 }
1048 
1049 
1050 struct nl80211_dump_scan_ctx {
1051 	struct wpa_driver_nl80211_data *drv;
1052 	int idx;
1053 };
1054 
nl80211_dump_scan_handler(struct nl_msg * msg,void * arg)1055 static int nl80211_dump_scan_handler(struct nl_msg *msg, void *arg)
1056 {
1057 	struct nl80211_dump_scan_ctx *ctx = arg;
1058 	struct wpa_scan_res *r;
1059 
1060 	r = nl80211_parse_bss_info(ctx->drv, msg, NULL);
1061 	if (!r)
1062 		return NL_SKIP;
1063 	wpa_printf(MSG_DEBUG, "nl80211: %d " MACSTR " %d%s",
1064 		   ctx->idx, MAC2STR(r->bssid), r->freq,
1065 		   r->flags & WPA_SCAN_ASSOCIATED ? " [assoc]" : "");
1066 	ctx->idx++;
1067 	os_free(r);
1068 	return NL_SKIP;
1069 }
1070 
1071 
nl80211_dump_scan(struct wpa_driver_nl80211_data * drv)1072 void nl80211_dump_scan(struct wpa_driver_nl80211_data *drv)
1073 {
1074 	struct nl_msg *msg;
1075 	struct nl80211_dump_scan_ctx ctx;
1076 
1077 	wpa_printf(MSG_DEBUG, "nl80211: Scan result dump");
1078 	ctx.drv = drv;
1079 	ctx.idx = 0;
1080 	msg = nl80211_cmd_msg(drv->first_bss, NLM_F_DUMP, NL80211_CMD_GET_SCAN);
1081 	if (msg)
1082 		send_and_recv_resp(drv, msg, nl80211_dump_scan_handler, &ctx);
1083 }
1084 
1085 
wpa_driver_nl80211_abort_scan(void * priv,u64 scan_cookie)1086 int wpa_driver_nl80211_abort_scan(void *priv, u64 scan_cookie)
1087 {
1088 	struct i802_bss *bss = priv;
1089 #ifdef CONFIG_DRIVER_NL80211_QCA
1090 	struct wpa_driver_nl80211_data *drv = bss->drv;
1091 
1092 	/*
1093 	 * If scan_cookie is zero, a normal scan through kernel (cfg80211)
1094 	 * was triggered, hence abort the cfg80211 scan instead of the vendor
1095 	 * scan.
1096 	 */
1097 	if (drv->scan_vendor_cmd_avail && scan_cookie)
1098 		return nl80211_abort_vendor_scan(drv, scan_cookie);
1099 #endif /* CONFIG_DRIVER_NL80211_QCA */
1100 	return nl80211_abort_scan(bss);
1101 }
1102 
1103 
1104 #ifdef CONFIG_DRIVER_NL80211_QCA
1105 
scan_cookie_handler(struct nl_msg * msg,void * arg)1106 static int scan_cookie_handler(struct nl_msg *msg, void *arg)
1107 {
1108 	struct nlattr *tb[NL80211_ATTR_MAX + 1];
1109 	struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
1110 	u64 *cookie = arg;
1111 
1112 	nla_parse(tb, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
1113 		  genlmsg_attrlen(gnlh, 0), NULL);
1114 
1115 	if (tb[NL80211_ATTR_VENDOR_DATA]) {
1116 		struct nlattr *nl_vendor = tb[NL80211_ATTR_VENDOR_DATA];
1117 		struct nlattr *tb_vendor[QCA_WLAN_VENDOR_ATTR_SCAN_MAX + 1];
1118 
1119 		nla_parse(tb_vendor, QCA_WLAN_VENDOR_ATTR_SCAN_MAX,
1120 			  nla_data(nl_vendor), nla_len(nl_vendor), NULL);
1121 
1122 		if (tb_vendor[QCA_WLAN_VENDOR_ATTR_SCAN_COOKIE])
1123 			*cookie = nla_get_u64(
1124 				tb_vendor[QCA_WLAN_VENDOR_ATTR_SCAN_COOKIE]);
1125 	}
1126 
1127 	return NL_SKIP;
1128 }
1129 
1130 
1131 /**
1132  * wpa_driver_nl80211_vendor_scan - Request the driver to initiate a vendor scan
1133  * @bss: Pointer to private driver data from wpa_driver_nl80211_init()
1134  * @params: Scan parameters
1135  * Returns: 0 on success, -1 on failure
1136  */
wpa_driver_nl80211_vendor_scan(struct i802_bss * bss,struct wpa_driver_scan_params * params)1137 int wpa_driver_nl80211_vendor_scan(struct i802_bss *bss,
1138 				   struct wpa_driver_scan_params *params)
1139 {
1140 	struct wpa_driver_nl80211_data *drv = bss->drv;
1141 	struct nl_msg *msg = NULL;
1142 	struct nlattr *attr;
1143 	size_t i;
1144 	u32 scan_flags = 0;
1145 	int ret = -1;
1146 	u64 cookie = 0;
1147 
1148 	wpa_dbg(drv->ctx, MSG_DEBUG, "nl80211: vendor scan request");
1149 	drv->scan_for_auth = 0;
1150 
1151 	if (!(msg = nl80211_drv_msg(drv, 0, NL80211_CMD_VENDOR)) ||
1152 	    nla_put_u32(msg, NL80211_ATTR_VENDOR_ID, OUI_QCA) ||
1153 	    nla_put_u32(msg, NL80211_ATTR_VENDOR_SUBCMD,
1154 			QCA_NL80211_VENDOR_SUBCMD_TRIGGER_SCAN) )
1155 		goto fail;
1156 
1157 	attr = nla_nest_start(msg, NL80211_ATTR_VENDOR_DATA);
1158 	if (attr == NULL)
1159 		goto fail;
1160 
1161 	if (params->num_ssids) {
1162 		struct nlattr *ssids;
1163 
1164 		ssids = nla_nest_start(msg, QCA_WLAN_VENDOR_ATTR_SCAN_SSIDS);
1165 		if (ssids == NULL)
1166 			goto fail;
1167 		for (i = 0; i < params->num_ssids; i++) {
1168 			wpa_printf(MSG_MSGDUMP, "nl80211: Scan SSID %s",
1169 				   wpa_ssid_txt(params->ssids[i].ssid,
1170 						params->ssids[i].ssid_len));
1171 			if (nla_put(msg, i + 1, params->ssids[i].ssid_len,
1172 				    params->ssids[i].ssid))
1173 				goto fail;
1174 		}
1175 		nla_nest_end(msg, ssids);
1176 	}
1177 
1178 	if (params->extra_ies) {
1179 		wpa_hexdump(MSG_MSGDUMP, "nl80211: Scan extra IEs",
1180 			    params->extra_ies, params->extra_ies_len);
1181 		if (nla_put(msg, QCA_WLAN_VENDOR_ATTR_SCAN_IE,
1182 			    params->extra_ies_len, params->extra_ies))
1183 			goto fail;
1184 	}
1185 
1186 	if (params->freqs) {
1187 		struct nlattr *freqs;
1188 
1189 		freqs = nla_nest_start(msg,
1190 				       QCA_WLAN_VENDOR_ATTR_SCAN_FREQUENCIES);
1191 		if (freqs == NULL)
1192 			goto fail;
1193 		for (i = 0; params->freqs[i]; i++) {
1194 			wpa_printf(MSG_MSGDUMP,
1195 				   "nl80211: Scan frequency %u MHz",
1196 				   params->freqs[i]);
1197 			if (nla_put_u32(msg, i + 1, params->freqs[i]))
1198 				goto fail;
1199 		}
1200 		nla_nest_end(msg, freqs);
1201 	}
1202 
1203 	os_free(drv->filter_ssids);
1204 	drv->filter_ssids = params->filter_ssids;
1205 	params->filter_ssids = NULL;
1206 	drv->num_filter_ssids = params->num_filter_ssids;
1207 
1208 	if (params->low_priority && drv->have_low_prio_scan) {
1209 		wpa_printf(MSG_DEBUG,
1210 			   "nl80211: Add NL80211_SCAN_FLAG_LOW_PRIORITY");
1211 		scan_flags |= NL80211_SCAN_FLAG_LOW_PRIORITY;
1212 	}
1213 
1214 	if (params->mac_addr_rand) {
1215 		wpa_printf(MSG_DEBUG,
1216 			   "nl80211: Add NL80211_SCAN_FLAG_RANDOM_ADDR");
1217 		scan_flags |= NL80211_SCAN_FLAG_RANDOM_ADDR;
1218 
1219 		if (params->mac_addr) {
1220 			wpa_printf(MSG_DEBUG, "nl80211: MAC address: " MACSTR,
1221 				   MAC2STR(params->mac_addr));
1222 			if (nla_put(msg, QCA_WLAN_VENDOR_ATTR_SCAN_MAC,
1223 				    ETH_ALEN, params->mac_addr))
1224 				goto fail;
1225 		}
1226 
1227 		if (params->mac_addr_mask) {
1228 			wpa_printf(MSG_DEBUG, "nl80211: MAC address mask: "
1229 				   MACSTR, MAC2STR(params->mac_addr_mask));
1230 			if (nla_put(msg, QCA_WLAN_VENDOR_ATTR_SCAN_MAC_MASK,
1231 				    ETH_ALEN, params->mac_addr_mask))
1232 				goto fail;
1233 		}
1234 	}
1235 
1236 	if (scan_flags &&
1237 	    nla_put_u32(msg, QCA_WLAN_VENDOR_ATTR_SCAN_FLAGS, scan_flags))
1238 		goto fail;
1239 
1240 	if (params->p2p_probe) {
1241 		struct nlattr *rates;
1242 
1243 		wpa_printf(MSG_DEBUG, "nl80211: P2P probe - mask SuppRates");
1244 
1245 		rates = nla_nest_start(msg,
1246 				       QCA_WLAN_VENDOR_ATTR_SCAN_SUPP_RATES);
1247 		if (rates == NULL)
1248 			goto fail;
1249 
1250 		/*
1251 		 * Remove 2.4 GHz rates 1, 2, 5.5, 11 Mbps from supported rates
1252 		 * by masking out everything else apart from the OFDM rates 6,
1253 		 * 9, 12, 18, 24, 36, 48, 54 Mbps from non-MCS rates. All 5 GHz
1254 		 * rates are left enabled.
1255 		 */
1256 		if (nla_put(msg, NL80211_BAND_2GHZ, 8,
1257 			    "\x0c\x12\x18\x24\x30\x48\x60\x6c"))
1258 			goto fail;
1259 		nla_nest_end(msg, rates);
1260 
1261 		if (nla_put_flag(msg, QCA_WLAN_VENDOR_ATTR_SCAN_TX_NO_CCK_RATE))
1262 			goto fail;
1263 	}
1264 
1265 	if (params->bssid) {
1266 		wpa_printf(MSG_DEBUG, "nl80211: Scan for a specific BSSID: "
1267 			   MACSTR, MAC2STR(params->bssid));
1268 		if (nla_put(msg, QCA_WLAN_VENDOR_ATTR_SCAN_BSSID, ETH_ALEN,
1269 			    params->bssid))
1270 			goto fail;
1271 	}
1272 
1273 	if (is_ap_interface(drv->nlmode) &&
1274 	    params->link_id != NL80211_DRV_LINK_ID_NA &&
1275 	    nla_put_u8(msg, QCA_WLAN_VENDOR_ATTR_SCAN_LINK_ID, params->link_id))
1276 		goto fail;
1277 
1278 	nla_nest_end(msg, attr);
1279 
1280 	ret = send_and_recv_resp(drv, msg, scan_cookie_handler, &cookie);
1281 	msg = NULL;
1282 	if (ret) {
1283 		wpa_printf(MSG_DEBUG,
1284 			   "nl80211: Vendor scan trigger failed: ret=%d (%s)",
1285 			   ret, strerror(-ret));
1286 		goto fail;
1287 	}
1288 
1289 	drv->vendor_scan_cookie = cookie;
1290 	drv->scan_state = SCAN_REQUESTED;
1291 	/* Pass the cookie to the caller to help distinguish the scans. */
1292 	params->scan_cookie = cookie;
1293 
1294 	wpa_printf(MSG_DEBUG,
1295 		   "nl80211: Vendor scan requested (ret=%d) - scan timeout 30 seconds, scan cookie:0x%llx",
1296 		   ret, (long long unsigned int) cookie);
1297 	eloop_cancel_timeout(wpa_driver_nl80211_scan_timeout, drv, drv->ctx);
1298 	eloop_register_timeout(30, 0, wpa_driver_nl80211_scan_timeout,
1299 			       drv, drv->ctx);
1300 	drv->last_scan_cmd = NL80211_CMD_VENDOR;
1301 
1302 fail:
1303 	nlmsg_free(msg);
1304 	return ret;
1305 }
1306 
1307 
1308 /**
1309  * nl80211_set_default_scan_ies - Set the scan default IEs to the driver
1310  * @priv: Pointer to private driver data from wpa_driver_nl80211_init()
1311  * @ies: Pointer to IEs buffer
1312  * @ies_len: Length of IEs in bytes
1313  * Returns: 0 on success, -1 on failure
1314  */
nl80211_set_default_scan_ies(void * priv,const u8 * ies,size_t ies_len)1315 int nl80211_set_default_scan_ies(void *priv, const u8 *ies, size_t ies_len)
1316 {
1317 	struct i802_bss *bss = priv;
1318 	struct wpa_driver_nl80211_data *drv = bss->drv;
1319 	struct nl_msg *msg = NULL;
1320 	struct nlattr *attr;
1321 	int ret = -1;
1322 
1323 	if (!drv->set_wifi_conf_vendor_cmd_avail)
1324 		return -1;
1325 
1326 	if (!(msg = nl80211_drv_msg(drv, 0, NL80211_CMD_VENDOR)) ||
1327 	    nla_put_u32(msg, NL80211_ATTR_VENDOR_ID, OUI_QCA) ||
1328 	    nla_put_u32(msg, NL80211_ATTR_VENDOR_SUBCMD,
1329 			QCA_NL80211_VENDOR_SUBCMD_SET_WIFI_CONFIGURATION))
1330 		goto fail;
1331 
1332 	attr = nla_nest_start(msg, NL80211_ATTR_VENDOR_DATA);
1333 	if (attr == NULL)
1334 		goto fail;
1335 
1336 	wpa_hexdump(MSG_MSGDUMP, "nl80211: Scan default IEs", ies, ies_len);
1337 	if (nla_put(msg, QCA_WLAN_VENDOR_ATTR_CONFIG_SCAN_DEFAULT_IES,
1338 		    ies_len, ies))
1339 		goto fail;
1340 
1341 	nla_nest_end(msg, attr);
1342 
1343 	ret = send_and_recv_cmd(drv, msg);
1344 	msg = NULL;
1345 	if (ret) {
1346 		wpa_printf(MSG_ERROR,
1347 			   "nl80211: Set scan default IEs failed: ret=%d (%s)",
1348 			   ret, strerror(-ret));
1349 		goto fail;
1350 	}
1351 
1352 fail:
1353 	nlmsg_free(msg);
1354 	return ret;
1355 }
1356 
1357 #endif /* CONFIG_DRIVER_NL80211_QCA */
1358