1 /*-
2  * Copyright (c) 2002-2008 Sam Leffler, Errno Consulting
3  * All rights reserved.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions
7  * are met:
8  * 1. Redistributions of source code must retain the above copyright
9  *    notice, this list of conditions and the following disclaimer.
10  * 2. Redistributions in binary form must reproduce the above copyright
11  *    notice, this list of conditions and the following disclaimer in the
12  *    documentation and/or other materials provided with the distribution.
13  *
14  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
15  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
16  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
17  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
18  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
19  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
20  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
21  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
22  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
23  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
24  */
25 
26 #include <sys/cdefs.h>
27 __FBSDID("$FreeBSD$");
28 
29 /*
30  * IEEE 802.11 scanning support.
31  */
32 #include "opt_wlan.h"
33 
34 #include <sys/param.h>
35 #include <sys/systm.h>
36 #include <sys/proc.h>
37 #include <sys/kernel.h>
38 #include <sys/malloc.h>
39 #include <sys/condvar.h>
40 
41 #include <sys/socket.h>
42 
43 #include <net/if.h>
44 #include <net/if_var.h>
45 #include <net/if_media.h>
46 #include <net/ethernet.h>
47 
48 #include <netproto/802_11/ieee80211_var.h>
49 
50 #include <netproto/802_11/ieee80211_scan_sw.h>
51 
52 #include <net/bpf.h>
53 
54 struct scan_state {
55 	struct ieee80211_scan_state base;	/* public state */
56 
57 	u_int			ss_iflags;	/* flags used internally */
58 #define	ISCAN_MINDWELL 		0x0001		/* min dwell time reached */
59 #define	ISCAN_DISCARD		0x0002		/* discard rx'd frames */
60 #define	ISCAN_CANCEL		0x0004		/* cancel current scan */
61 #define	ISCAN_ABORT		0x0008		/* end the scan immediately */
62 #define	ISCAN_RUNNING		0x0010		/* scan was started */
63 
64 	unsigned long		ss_chanmindwell;  /* min dwell on curchan */
65 	unsigned long		ss_scanend;	/* time scan must stop */
66 	u_int			ss_duration;	/* duration for next scan */
67 	struct task		ss_scan_start;	/* scan start */
68 	struct timeout_task	ss_scan_curchan;  /* scan execution */
69 };
70 #define	SCAN_PRIVATE(ss)	((struct scan_state *) ss)
71 
72 /*
73  * Amount of time to go off-channel during a background
74  * scan.  This value should be large enough to catch most
75  * ap's but short enough that we can return on-channel
76  * before our listen interval expires.
77  *
78  * XXX tunable
79  * XXX check against configured listen interval
80  */
81 #define	IEEE80211_SCAN_OFFCHANNEL	msecs_to_ticks(150)
82 
83 /*
84  * Roaming-related defaults.  RSSI thresholds are as returned by the
85  * driver (.5dBm).  Transmit rate thresholds are IEEE rate codes (i.e
86  * .5M units) or MCS.
87  */
88 /* rssi thresholds */
89 #define	ROAM_RSSI_11A_DEFAULT		14	/* 11a bss */
90 #define	ROAM_RSSI_11B_DEFAULT		14	/* 11b bss */
91 #define	ROAM_RSSI_11BONLY_DEFAULT	14	/* 11b-only bss */
92 /* transmit rate thresholds */
93 #define	ROAM_RATE_11A_DEFAULT		2*12	/* 11a bss */
94 #define	ROAM_RATE_11B_DEFAULT		2*5	/* 11b bss */
95 #define	ROAM_RATE_11BONLY_DEFAULT	2*1	/* 11b-only bss */
96 #define	ROAM_RATE_HALF_DEFAULT		2*6	/* half-width 11a/g bss */
97 #define	ROAM_RATE_QUARTER_DEFAULT	2*3	/* quarter-width 11a/g bss */
98 #define	ROAM_MCS_11N_DEFAULT		(1 | IEEE80211_RATE_MCS) /* 11n bss */
99 
100 static	void scan_curchan(struct ieee80211_scan_state *, unsigned long);
101 static	void scan_mindwell(struct ieee80211_scan_state *);
102 static	void scan_signal(struct ieee80211_scan_state *, int);
103 static	void scan_signal_locked(struct ieee80211_scan_state *, int);
104 static	void scan_start(void *, int);
105 static	void scan_curchan_task(void *, int);
106 static	void scan_end(struct ieee80211_scan_state *, int);
107 static	void scan_done(struct ieee80211_scan_state *, int);
108 
109 MALLOC_DEFINE(M_80211_SCAN, "80211scan", "802.11 scan state");
110 
111 static void
112 ieee80211_swscan_detach(struct ieee80211com *ic)
113 {
114 	struct ieee80211_scan_state *ss = ic->ic_scan;
115 
116 	if (ss != NULL) {
117 		scan_signal(ss, ISCAN_ABORT);
118 		ieee80211_draintask(ic, &SCAN_PRIVATE(ss)->ss_scan_start);
119 		taskqueue_drain_timeout(ic->ic_tq,
120 		    &SCAN_PRIVATE(ss)->ss_scan_curchan);
121 		KASSERT((ic->ic_flags & IEEE80211_F_SCAN) == 0,
122 		    ("scan still running"));
123 
124 		/*
125 		 * For now, do the ss_ops detach here rather
126 		 * than ieee80211_scan_detach().
127 		 *
128 		 * I'll figure out how to cleanly split things up
129 		 * at a later date.
130 		 */
131 		if (ss->ss_ops != NULL) {
132 			ss->ss_ops->scan_detach(ss);
133 			ss->ss_ops = NULL;
134 		}
135 		ic->ic_scan = NULL;
136 		IEEE80211_FREE(SCAN_PRIVATE(ss), M_80211_SCAN);
137 	}
138 }
139 
140 static void
141 ieee80211_swscan_vattach(struct ieee80211vap *vap)
142 {
143 	/* nothing to do for now */
144 	/*
145 	 * TODO: all of the vap scan calls should be methods!
146 	 */
147 
148 }
149 
150 static void
151 ieee80211_swscan_vdetach(struct ieee80211vap *vap)
152 {
153 	struct ieee80211com *ic = vap->iv_ic;
154 	struct ieee80211_scan_state *ss = ic->ic_scan;
155 
156 	IEEE80211_LOCK_ASSERT(ic);
157 
158 	if (ss != NULL && ss->ss_vap == vap &&
159 	    (ic->ic_flags & IEEE80211_F_SCAN))
160 		scan_signal_locked(ss, ISCAN_ABORT);
161 }
162 
163 static void
164 ieee80211_swscan_set_scan_duration(struct ieee80211vap *vap, u_int duration)
165 {
166 	struct ieee80211com *ic = vap->iv_ic;
167 	struct ieee80211_scan_state *ss = ic->ic_scan;
168 
169 	IEEE80211_LOCK_ASSERT(ic);
170 
171 	/* NB: flush frames rx'd before 1st channel change */
172 	SCAN_PRIVATE(ss)->ss_iflags |= ISCAN_DISCARD;
173 	SCAN_PRIVATE(ss)->ss_duration = duration;
174 }
175 
176 /*
177  * Start a scan unless one is already going.
178  */
179 static int
180 ieee80211_swscan_start_scan_locked(const struct ieee80211_scanner *scan,
181     struct ieee80211vap *vap, int flags, u_int duration,
182     u_int mindwell, u_int maxdwell,
183     u_int nssid, const struct ieee80211_scan_ssid ssids[])
184 {
185 	struct ieee80211com *ic = vap->iv_ic;
186 	struct ieee80211_scan_state *ss = ic->ic_scan;
187 
188 	IEEE80211_LOCK_ASSERT(ic);
189 
190 	if (ic->ic_flags & IEEE80211_F_CSAPENDING) {
191 		IEEE80211_DPRINTF(vap, IEEE80211_MSG_SCAN,
192 		    "%s: scan inhibited by pending channel change\n", __func__);
193 	} else if ((ic->ic_flags & IEEE80211_F_SCAN) == 0) {
194 		IEEE80211_DPRINTF(vap, IEEE80211_MSG_SCAN,
195 		    "%s: %s scan, duration %u mindwell %u maxdwell %u, desired mode %s, %s%s%s%s%s%s\n"
196 		    , __func__
197 		    , flags & IEEE80211_SCAN_ACTIVE ? "active" : "passive"
198 		    , duration, mindwell, maxdwell
199 		    , ieee80211_phymode_name[vap->iv_des_mode]
200 		    , flags & IEEE80211_SCAN_FLUSH ? "flush" : "append"
201 		    , flags & IEEE80211_SCAN_NOPICK ? ", nopick" : ""
202 		    , flags & IEEE80211_SCAN_NOJOIN ? ", nojoin" : ""
203 		    , flags & IEEE80211_SCAN_NOBCAST ? ", nobcast" : ""
204 		    , flags & IEEE80211_SCAN_PICK1ST ? ", pick1st" : ""
205 		    , flags & IEEE80211_SCAN_ONCE ? ", once" : ""
206 		);
207 
208 		ieee80211_scan_update_locked(vap, scan);
209 		if (ss->ss_ops != NULL) {
210 			if ((flags & IEEE80211_SCAN_NOSSID) == 0)
211 				ieee80211_scan_copy_ssid(vap, ss, nssid, ssids);
212 
213 			/* NB: top 4 bits for internal use */
214 			ss->ss_flags = flags & 0xfff;
215 			if (ss->ss_flags & IEEE80211_SCAN_ACTIVE)
216 				vap->iv_stats.is_scan_active++;
217 			else
218 				vap->iv_stats.is_scan_passive++;
219 			if (flags & IEEE80211_SCAN_FLUSH)
220 				ss->ss_ops->scan_flush(ss);
221 			if (flags & IEEE80211_SCAN_BGSCAN)
222 				ic->ic_flags_ext |= IEEE80211_FEXT_BGSCAN;
223 
224 			/* Set duration for this particular scan */
225 			ieee80211_swscan_set_scan_duration(vap, duration);
226 
227 			ss->ss_next = 0;
228 			ss->ss_mindwell = mindwell;
229 			ss->ss_maxdwell = maxdwell;
230 			/* NB: scan_start must be before the scan runtask */
231 			ss->ss_ops->scan_start(ss, vap);
232 #ifdef IEEE80211_DEBUG
233 			if (ieee80211_msg_scan(vap))
234 				ieee80211_scan_dump(ss);
235 #endif /* IEEE80211_DEBUG */
236 			ic->ic_flags |= IEEE80211_F_SCAN;
237 
238 			/* Start scan task */
239 			ieee80211_runtask(ic, &SCAN_PRIVATE(ss)->ss_scan_start);
240 		}
241 		return 1;
242 	} else {
243 		IEEE80211_DPRINTF(vap, IEEE80211_MSG_SCAN,
244 		    "%s: %s scan already in progress\n", __func__,
245 		    ss->ss_flags & IEEE80211_SCAN_ACTIVE ? "active" : "passive");
246 	}
247 	return 0;
248 }
249 
250 
251 /*
252  * Start a scan unless one is already going.
253  *
254  * Called without the comlock held; grab the comlock as appropriate.
255  */
256 static int
257 ieee80211_swscan_start_scan(const struct ieee80211_scanner *scan,
258     struct ieee80211vap *vap, int flags,
259     u_int duration, u_int mindwell, u_int maxdwell,
260     u_int nssid, const struct ieee80211_scan_ssid ssids[])
261 {
262 	struct ieee80211com *ic = vap->iv_ic;
263 	int result;
264 
265 	IEEE80211_UNLOCK_ASSERT(ic);
266 
267 	IEEE80211_LOCK(ic);
268 	result = ieee80211_swscan_start_scan_locked(scan, vap, flags, duration,
269 	    mindwell, maxdwell, nssid, ssids);
270 	IEEE80211_UNLOCK(ic);
271 
272 	return result;
273 }
274 
275 /*
276  * Check the scan cache for an ap/channel to use; if that
277  * fails then kick off a new scan.
278  *
279  * Called with the comlock held.
280  *
281  * XXX TODO: split out!
282  */
283 static int
284 ieee80211_swscan_check_scan(const struct ieee80211_scanner *scan,
285     struct ieee80211vap *vap, int flags,
286     u_int duration, u_int mindwell, u_int maxdwell,
287     u_int nssid, const struct ieee80211_scan_ssid ssids[])
288 {
289 	struct ieee80211com *ic = vap->iv_ic;
290 	struct ieee80211_scan_state *ss = ic->ic_scan;
291 	int result;
292 
293 	IEEE80211_LOCK_ASSERT(ic);
294 
295 	if (ss->ss_ops != NULL) {
296 		/* XXX verify ss_ops matches vap->iv_opmode */
297 		if ((flags & IEEE80211_SCAN_NOSSID) == 0) {
298 			/*
299 			 * Update the ssid list and mark flags so if
300 			 * we call start_scan it doesn't duplicate work.
301 			 */
302 			ieee80211_scan_copy_ssid(vap, ss, nssid, ssids);
303 			flags |= IEEE80211_SCAN_NOSSID;
304 		}
305 		if ((ic->ic_flags & IEEE80211_F_SCAN) == 0 &&
306 		    (flags & IEEE80211_SCAN_FLUSH) == 0 &&
307 		    ieee80211_time_before(ticks, ic->ic_lastscan + vap->iv_scanvalid)) {
308 			/*
309 			 * We're not currently scanning and the cache is
310 			 * deemed hot enough to consult.  Lock out others
311 			 * by marking IEEE80211_F_SCAN while we decide if
312 			 * something is already in the scan cache we can
313 			 * use.  Also discard any frames that might come
314 			 * in while temporarily marked as scanning.
315 			 */
316 			SCAN_PRIVATE(ss)->ss_iflags |= ISCAN_DISCARD;
317 			ic->ic_flags |= IEEE80211_F_SCAN;
318 
319 			/* NB: need to use supplied flags in check */
320 			ss->ss_flags = flags & 0xff;
321 			result = ss->ss_ops->scan_end(ss, vap);
322 
323 			ic->ic_flags &= ~IEEE80211_F_SCAN;
324 			SCAN_PRIVATE(ss)->ss_iflags &= ~ISCAN_DISCARD;
325 			if (result) {
326 				ieee80211_notify_scan_done(vap);
327 				return 1;
328 			}
329 		}
330 	}
331 	result = ieee80211_swscan_start_scan_locked(scan, vap, flags, duration,
332 	    mindwell, maxdwell, nssid, ssids);
333 
334 	return result;
335 }
336 
337 /*
338  * Restart a previous scan.  If the previous scan completed
339  * then we start again using the existing channel list.
340  */
341 static int
342 ieee80211_swscan_bg_scan(const struct ieee80211_scanner *scan,
343     struct ieee80211vap *vap, int flags)
344 {
345 	struct ieee80211com *ic = vap->iv_ic;
346 	struct ieee80211_scan_state *ss = ic->ic_scan;
347 
348 	/* XXX assert unlocked? */
349 	// IEEE80211_UNLOCK_ASSERT(ic);
350 
351 	IEEE80211_LOCK(ic);
352 	if ((ic->ic_flags & IEEE80211_F_SCAN) == 0) {
353 		u_int duration;
354 		/*
355 		 * Go off-channel for a fixed interval that is large
356 		 * enough to catch most ap's but short enough that
357 		 * we can return on-channel before our listen interval
358 		 * expires.
359 		 */
360 		duration = IEEE80211_SCAN_OFFCHANNEL;
361 
362 		IEEE80211_DPRINTF(vap, IEEE80211_MSG_SCAN,
363 		    "%s: %s scan, ticks %u duration %u\n", __func__,
364 		    ss->ss_flags & IEEE80211_SCAN_ACTIVE ? "active" : "passive",
365 		    ticks, duration);
366 
367 		ieee80211_scan_update_locked(vap, scan);
368 		if (ss->ss_ops != NULL) {
369 			ss->ss_vap = vap;
370 			/*
371 			 * A background scan does not select a new sta; it
372 			 * just refreshes the scan cache.  Also, indicate
373 			 * the scan logic should follow the beacon schedule:
374 			 * we go off-channel and scan for a while, then
375 			 * return to the bss channel to receive a beacon,
376 			 * then go off-channel again.  All during this time
377 			 * we notify the ap we're in power save mode.  When
378 			 * the scan is complete we leave power save mode.
379 			 * If any beacon indicates there are frames pending
380 			 * for us then we drop out of power save mode
381 			 * (and background scan) automatically by way of the
382 			 * usual sta power save logic.
383 			 */
384 			ss->ss_flags |= IEEE80211_SCAN_NOPICK
385 				     |  IEEE80211_SCAN_BGSCAN
386 				     |  flags
387 				     ;
388 			/* if previous scan completed, restart */
389 			if (ss->ss_next >= ss->ss_last) {
390 				if (ss->ss_flags & IEEE80211_SCAN_ACTIVE)
391 					vap->iv_stats.is_scan_active++;
392 				else
393 					vap->iv_stats.is_scan_passive++;
394 				/*
395 				 * NB: beware of the scan cache being flushed;
396 				 *     if the channel list is empty use the
397 				 *     scan_start method to populate it.
398 				 */
399 				ss->ss_next = 0;
400 				if (ss->ss_last != 0)
401 					ss->ss_ops->scan_restart(ss, vap);
402 				else {
403 					ss->ss_ops->scan_start(ss, vap);
404 #ifdef IEEE80211_DEBUG
405 					if (ieee80211_msg_scan(vap))
406 						ieee80211_scan_dump(ss);
407 #endif /* IEEE80211_DEBUG */
408 				}
409 			}
410 			ieee80211_swscan_set_scan_duration(vap, duration);
411 			ss->ss_maxdwell = duration;
412 			ic->ic_flags |= IEEE80211_F_SCAN;
413 			ic->ic_flags_ext |= IEEE80211_FEXT_BGSCAN;
414 			ieee80211_runtask(ic,
415 			    &SCAN_PRIVATE(ss)->ss_scan_start);
416 		} else {
417 			/* XXX msg+stat */
418 		}
419 	} else {
420 		IEEE80211_DPRINTF(vap, IEEE80211_MSG_SCAN,
421 		    "%s: %s scan already in progress\n", __func__,
422 		    ss->ss_flags & IEEE80211_SCAN_ACTIVE ? "active" : "passive");
423 	}
424 	IEEE80211_UNLOCK(ic);
425 
426 	/* NB: racey, does it matter? */
427 	return (ic->ic_flags & IEEE80211_F_SCAN);
428 }
429 
430 static void
431 cancel_scan(struct ieee80211vap *vap, int any, const char *func)
432 {
433 	struct ieee80211com *ic = vap->iv_ic;
434 	struct ieee80211_scan_state *ss = ic->ic_scan;
435 
436 	IEEE80211_LOCK(ic);
437 	if ((ic->ic_flags & IEEE80211_F_SCAN) &&
438 	    (any || ss->ss_vap == vap) &&
439 	    (SCAN_PRIVATE(ss)->ss_iflags & ISCAN_CANCEL) == 0) {
440 		IEEE80211_DPRINTF(vap, IEEE80211_MSG_SCAN,
441 		    "%s: cancel %s scan\n", func,
442 		    ss->ss_flags & IEEE80211_SCAN_ACTIVE ?
443 			"active" : "passive");
444 
445 		/* clear bg scan NOPICK */
446 		ss->ss_flags &= ~IEEE80211_SCAN_NOPICK;
447 		/* mark cancel request and wake up the scan task */
448 		scan_signal_locked(ss, ISCAN_CANCEL);
449 	} else {
450 		IEEE80211_DPRINTF(vap, IEEE80211_MSG_SCAN,
451 		    "%s: called; F_SCAN=%d, vap=%s, CANCEL=%d\n",
452 			func,
453 			!! (ic->ic_flags & IEEE80211_F_SCAN),
454 			(ss->ss_vap == vap ? "match" : "nomatch"),
455 			!! (SCAN_PRIVATE(ss)->ss_iflags & ISCAN_CANCEL));
456 	}
457 	IEEE80211_UNLOCK(ic);
458 }
459 
460 /*
461  * Cancel any scan currently going on for the specified vap.
462  */
463 static void
464 ieee80211_swscan_cancel_scan(struct ieee80211vap *vap)
465 {
466 	cancel_scan(vap, 0, __func__);
467 }
468 
469 /*
470  * Cancel any scan currently going on.
471  */
472 static void
473 ieee80211_swscan_cancel_anyscan(struct ieee80211vap *vap)
474 {
475 	cancel_scan(vap, 1, __func__);
476 }
477 
478 /*
479  * Manually switch to the next channel in the channel list.
480  * Provided for drivers that manage scanning themselves
481  * (e.g. for firmware-based devices).
482  */
483 static void
484 ieee80211_swscan_scan_next(struct ieee80211vap *vap)
485 {
486 	struct ieee80211_scan_state *ss = vap->iv_ic->ic_scan;
487 
488 	IEEE80211_DPRINTF(vap, IEEE80211_MSG_SCAN, "%s: called\n", __func__);
489 
490 	/* wake up the scan task */
491 	scan_signal(ss, 0);
492 }
493 
494 /*
495  * Manually stop a scan that is currently running.
496  * Provided for drivers that are not able to scan single channels
497  * (e.g. for firmware-based devices).
498  */
499 static void
500 ieee80211_swscan_scan_done(struct ieee80211vap *vap)
501 {
502 	struct ieee80211com *ic = vap->iv_ic;
503 	struct ieee80211_scan_state *ss = ic->ic_scan;
504 
505 	IEEE80211_LOCK_ASSERT(ic);
506 
507 	scan_signal_locked(ss, 0);
508 }
509 
510 /*
511  * Probe the current channel, if allowed, while scanning.
512  * If the channel is not marked passive-only then send
513  * a probe request immediately.  Otherwise mark state and
514  * listen for beacons on the channel; if we receive something
515  * then we'll transmit a probe request.
516  */
517 static void
518 ieee80211_swscan_probe_curchan(struct ieee80211vap *vap, int force)
519 {
520 	struct ieee80211com *ic = vap->iv_ic;
521 	struct ieee80211_scan_state *ss = ic->ic_scan;
522 	struct ifnet *ifp = vap->iv_ifp;
523 	int i;
524 
525 	/*
526 	 * Send directed probe requests followed by any
527 	 * broadcast probe request.
528 	 * XXX remove dependence on ic/vap->iv_bss
529 	 */
530 	for (i = 0; i < ss->ss_nssid; i++)
531 		ieee80211_send_probereq(vap->iv_bss,
532 			vap->iv_myaddr, ifp->if_broadcastaddr,
533 			ifp->if_broadcastaddr,
534 			ss->ss_ssid[i].ssid, ss->ss_ssid[i].len);
535 	if ((ss->ss_flags & IEEE80211_SCAN_NOBCAST) == 0)
536 		ieee80211_send_probereq(vap->iv_bss,
537 			vap->iv_myaddr, ifp->if_broadcastaddr,
538 			ifp->if_broadcastaddr,
539 			"", 0);
540 }
541 
542 /*
543  * Scan curchan.  If this is an active scan and the channel
544  * is not marked passive then send probe request frame(s).
545  * Arrange for the channel change after maxdwell ticks.
546  */
547 static void
548 scan_curchan(struct ieee80211_scan_state *ss, unsigned long maxdwell)
549 {
550 	struct ieee80211vap *vap  = ss->ss_vap;
551 	struct ieee80211com *ic = ss->ss_ic;
552 
553 	IEEE80211_DPRINTF(vap, IEEE80211_MSG_SCAN,
554 	    "%s: calling; maxdwell=%lu\n",
555 	    __func__,
556 	    maxdwell);
557 	IEEE80211_LOCK(ic);
558 	if (ss->ss_flags & IEEE80211_SCAN_ACTIVE)
559 		ieee80211_probe_curchan(vap, 0);
560 	taskqueue_enqueue_timeout(ic->ic_tq,
561 	    &SCAN_PRIVATE(ss)->ss_scan_curchan, maxdwell);
562 	IEEE80211_UNLOCK(ic);
563 }
564 
565 static void
566 scan_signal(struct ieee80211_scan_state *ss, int iflags)
567 {
568 	struct ieee80211com *ic = ss->ss_ic;
569 
570 	IEEE80211_UNLOCK_ASSERT(ic);
571 
572 	IEEE80211_LOCK(ic);
573 	scan_signal_locked(ss, iflags);
574 	IEEE80211_UNLOCK(ic);
575 }
576 
577 static void
578 scan_signal_locked(struct ieee80211_scan_state *ss, int iflags)
579 {
580 	struct scan_state *ss_priv = SCAN_PRIVATE(ss);
581 	struct timeout_task *scan_task = &ss_priv->ss_scan_curchan;
582 	struct ieee80211com *ic = ss->ss_ic;
583 
584 	IEEE80211_LOCK_ASSERT(ic);
585 
586 	ss_priv->ss_iflags |= iflags;
587 	if (ss_priv->ss_iflags & ISCAN_RUNNING) {
588 		if (taskqueue_cancel_timeout(ic->ic_tq, scan_task, NULL) == 0)
589 			taskqueue_enqueue_timeout(ic->ic_tq, scan_task, 0);
590 	}
591 }
592 
593 /*
594  * Handle mindwell requirements completed; initiate a channel
595  * change to the next channel asap.
596  */
597 static void
598 scan_mindwell(struct ieee80211_scan_state *ss)
599 {
600 
601 	IEEE80211_DPRINTF(ss->ss_vap, IEEE80211_MSG_SCAN, "%s: called\n",
602 	    __func__);
603 
604 	scan_signal(ss, 0);
605 }
606 
607 static void
608 scan_start(void *arg, int pending)
609 {
610 #define	ISCAN_REP	(ISCAN_MINDWELL | ISCAN_DISCARD)
611 	struct ieee80211_scan_state *ss = (struct ieee80211_scan_state *) arg;
612 	struct scan_state *ss_priv = SCAN_PRIVATE(ss);
613 	struct ieee80211vap *vap = ss->ss_vap;
614 	struct ieee80211com *ic = ss->ss_ic;
615 
616 	IEEE80211_LOCK(ic);
617 	if (vap == NULL || (ic->ic_flags & IEEE80211_F_SCAN) == 0 ||
618 	    (ss_priv->ss_iflags & ISCAN_ABORT)) {
619 		/* Cancelled before we started */
620 		scan_done(ss, 0);
621 		return;
622 	}
623 
624 	if (ss->ss_next == ss->ss_last) {
625 		IEEE80211_DPRINTF(vap, IEEE80211_MSG_SCAN,
626 			"%s: no channels to scan\n", __func__);
627 		scan_done(ss, 1);
628 		return;
629 	}
630 
631 	if (vap->iv_opmode == IEEE80211_M_STA &&
632 	    vap->iv_state == IEEE80211_S_RUN) {
633 		if ((vap->iv_bss->ni_flags & IEEE80211_NODE_PWR_MGT) == 0) {
634 			/* Enable station power save mode */
635 			vap->iv_sta_ps(vap, 1);
636 			/* Wait until null data frame will be ACK'ed */
637 #if defined(__DragonFly__)
638 			lksleep(vap, IEEE80211_LOCK_OBJ(ic), PCATCH,
639 			    "sta_ps", msecs_to_ticks(10));
640 #else
641 			mtx_sleep(vap, IEEE80211_LOCK_OBJ(ic), PCATCH,
642 			    "sta_ps", msecs_to_ticks(10));
643 #endif
644 			if (ss_priv->ss_iflags & ISCAN_ABORT) {
645 				scan_done(ss, 0);
646 				return;
647 			}
648 		}
649 	}
650 
651 	ss_priv->ss_scanend = ticks + ss_priv->ss_duration;
652 
653 	/* XXX scan state can change! Re-validate scan state! */
654 
655 	IEEE80211_UNLOCK(ic);
656 
657 	ic->ic_scan_start(ic);		/* notify driver */
658 
659 	scan_curchan_task(ss, 0);
660 }
661 
662 static void
663 scan_curchan_task(void *arg, int pending)
664 {
665 	struct ieee80211_scan_state *ss = arg;
666 	struct scan_state *ss_priv = SCAN_PRIVATE(ss);
667 	struct ieee80211com *ic = ss->ss_ic;
668 	struct ieee80211_channel *chan;
669 	unsigned long maxdwell;
670 	int scandone;
671 
672 	IEEE80211_LOCK(ic);
673 end:
674 	scandone = (ss->ss_next >= ss->ss_last) ||
675 	    (ss_priv->ss_iflags & ISCAN_CANCEL) != 0;
676 
677 	IEEE80211_DPRINTF(ss->ss_vap, IEEE80211_MSG_SCAN,
678 	    "%s: loop start; scandone=%d\n",
679 	    __func__,
680 	    scandone);
681 
682 	if (scandone || (ss->ss_flags & IEEE80211_SCAN_GOTPICK) ||
683 	    (ss_priv->ss_iflags & ISCAN_ABORT) ||
684 	     ieee80211_time_after(ticks + ss->ss_mindwell, ss_priv->ss_scanend)) {
685 		ss_priv->ss_iflags &= ~ISCAN_RUNNING;
686 		scan_end(ss, scandone);
687 		return;
688 	} else
689 		ss_priv->ss_iflags |= ISCAN_RUNNING;
690 
691 	chan = ss->ss_chans[ss->ss_next++];
692 
693 	/*
694 	 * Watch for truncation due to the scan end time.
695 	 */
696 	if (ieee80211_time_after(ticks + ss->ss_maxdwell, ss_priv->ss_scanend))
697 		maxdwell = ss_priv->ss_scanend - ticks;
698 	else
699 		maxdwell = ss->ss_maxdwell;
700 
701 	IEEE80211_DPRINTF(ss->ss_vap, IEEE80211_MSG_SCAN,
702 	    "%s: chan %3d%c -> %3d%c [%s, dwell min %lums max %lums]\n",
703 	    __func__,
704 	    ieee80211_chan2ieee(ic, ic->ic_curchan),
705 	    ieee80211_channel_type_char(ic->ic_curchan),
706 	    ieee80211_chan2ieee(ic, chan),
707 	    ieee80211_channel_type_char(chan),
708 	    (ss->ss_flags & IEEE80211_SCAN_ACTIVE) &&
709 		(chan->ic_flags & IEEE80211_CHAN_PASSIVE) == 0 ?
710 		"active" : "passive",
711 	    ticks_to_msecs(ss->ss_mindwell), ticks_to_msecs(maxdwell));
712 
713 	/*
714 	 * Potentially change channel and phy mode.
715 	 */
716 	ic->ic_curchan = chan;
717 	ic->ic_rt = ieee80211_get_ratetable(chan);
718 	IEEE80211_UNLOCK(ic);
719 	/*
720 	 * Perform the channel change and scan unlocked so the driver
721 	 * may sleep. Once set_channel returns the hardware has
722 	 * completed the channel change.
723 	 */
724 	ic->ic_set_channel(ic);
725 	ieee80211_radiotap_chan_change(ic);
726 
727 	/*
728 	 * Scan curchan.  Drivers for "intelligent hardware"
729 	 * override ic_scan_curchan to tell the device to do
730 	 * the work.  Otherwise we manage the work ourselves;
731 	 * sending a probe request (as needed), and arming the
732 	 * timeout to switch channels after maxdwell ticks.
733 	 *
734 	 * scan_curchan should only pause for the time required to
735 	 * prepare/initiate the hardware for the scan (if at all).
736 	 */
737 	ic->ic_scan_curchan(ss, maxdwell);
738 	IEEE80211_LOCK(ic);
739 
740 	/* XXX scan state can change! Re-validate scan state! */
741 
742 	ss_priv->ss_chanmindwell = ticks + ss->ss_mindwell;
743 	/* clear mindwell lock and initial channel change flush */
744 	ss_priv->ss_iflags &= ~ISCAN_REP;
745 
746 	if (ss_priv->ss_iflags & (ISCAN_CANCEL|ISCAN_ABORT)) {
747 		taskqueue_cancel_timeout(ic->ic_tq, &ss_priv->ss_scan_curchan,
748 		    NULL);
749 		goto end;
750 	}
751 
752 	IEEE80211_DPRINTF(ss->ss_vap, IEEE80211_MSG_SCAN, "%s: waiting\n",
753 	    __func__);
754 	IEEE80211_UNLOCK(ic);
755 }
756 
757 static void
758 scan_end(struct ieee80211_scan_state *ss, int scandone)
759 {
760 	struct scan_state *ss_priv = SCAN_PRIVATE(ss);
761 	struct ieee80211vap *vap = ss->ss_vap;
762 	struct ieee80211com *ic = ss->ss_ic;
763 
764 	IEEE80211_LOCK_ASSERT(ic);
765 
766 	IEEE80211_DPRINTF(vap, IEEE80211_MSG_SCAN, "%s: out\n", __func__);
767 
768 	if (ss_priv->ss_iflags & ISCAN_ABORT) {
769 		scan_done(ss, scandone);
770 		return;
771 	}
772 
773 	IEEE80211_UNLOCK(ic);
774 	ic->ic_scan_end(ic);		/* notify driver */
775 	IEEE80211_LOCK(ic);
776 	/* XXX scan state can change! Re-validate scan state! */
777 
778 	/*
779 	 * Since a cancellation may have occurred during one of the
780 	 * driver calls (whilst unlocked), update scandone.
781 	 */
782 	if (scandone == 0 && (ss_priv->ss_iflags & ISCAN_CANCEL) != 0) {
783 		/* XXX printf? */
784 		if_printf(vap->iv_ifp,
785 		    "%s: OOPS! scan cancelled during driver call (1)!\n",
786 		    __func__);
787 		scandone = 1;
788 	}
789 
790 	/*
791 	 * Record scan complete time.  Note that we also do
792 	 * this when canceled so any background scan will
793 	 * not be restarted for a while.
794 	 */
795 	if (scandone)
796 		ic->ic_lastscan = ticks;
797 	/* return to the bss channel */
798 	if (ic->ic_bsschan != IEEE80211_CHAN_ANYC &&
799 	    ic->ic_curchan != ic->ic_bsschan) {
800 		ieee80211_setupcurchan(ic, ic->ic_bsschan);
801 		IEEE80211_UNLOCK(ic);
802 		ic->ic_set_channel(ic);
803 		ieee80211_radiotap_chan_change(ic);
804 		IEEE80211_LOCK(ic);
805 	}
806 	/* clear internal flags and any indication of a pick */
807 	ss_priv->ss_iflags &= ~ISCAN_REP;
808 	ss->ss_flags &= ~IEEE80211_SCAN_GOTPICK;
809 
810 	/*
811 	 * If not canceled and scan completed, do post-processing.
812 	 * If the callback function returns 0, then it wants to
813 	 * continue/restart scanning.  Unfortunately we needed to
814 	 * notify the driver to end the scan above to avoid having
815 	 * rx frames alter the scan candidate list.
816 	 */
817 	if ((ss_priv->ss_iflags & ISCAN_CANCEL) == 0 &&
818 	    !ss->ss_ops->scan_end(ss, vap) &&
819 	    (ss->ss_flags & IEEE80211_SCAN_ONCE) == 0 &&
820 	    ieee80211_time_before(ticks + ss->ss_mindwell, ss_priv->ss_scanend)) {
821 		IEEE80211_DPRINTF(vap, IEEE80211_MSG_SCAN,
822 		    "%s: done, restart "
823 		    "[ticks %u, dwell min %lu scanend %lu]\n",
824 		    __func__,
825 		    ticks, ss->ss_mindwell, ss_priv->ss_scanend);
826 		ss->ss_next = 0;	/* reset to beginning */
827 		if (ss->ss_flags & IEEE80211_SCAN_ACTIVE)
828 			vap->iv_stats.is_scan_active++;
829 		else
830 			vap->iv_stats.is_scan_passive++;
831 
832 		ss->ss_ops->scan_restart(ss, vap);	/* XXX? */
833 		ieee80211_runtask(ic, &ss_priv->ss_scan_start);
834 		IEEE80211_UNLOCK(ic);
835 		return;
836 	}
837 
838 	/* past here, scandone is ``true'' if not in bg mode */
839 	if ((ss->ss_flags & IEEE80211_SCAN_BGSCAN) == 0)
840 		scandone = 1;
841 
842 	IEEE80211_DPRINTF(vap, IEEE80211_MSG_SCAN,
843 	    "%s: %s, [ticks %u, dwell min %lu scanend %lu]\n",
844 	    __func__, scandone ? "done" : "stopped",
845 	    ticks, ss->ss_mindwell, ss_priv->ss_scanend);
846 
847 	/*
848 	 * Since a cancellation may have occurred during one of the
849 	 * driver calls (whilst unlocked), update scandone.
850 	 */
851 	if (scandone == 0 && (ss_priv->ss_iflags & ISCAN_CANCEL) != 0) {
852 		/* XXX printf? */
853 		if_printf(vap->iv_ifp,
854 		    "%s: OOPS! scan cancelled during driver call (2)!\n",
855 		    __func__);
856 		scandone = 1;
857 	}
858 
859 	scan_done(ss, scandone);
860 }
861 
862 static void
863 scan_done(struct ieee80211_scan_state *ss, int scandone)
864 {
865 	struct scan_state *ss_priv = SCAN_PRIVATE(ss);
866 	struct ieee80211com *ic = ss->ss_ic;
867 	struct ieee80211vap *vap = ss->ss_vap;
868 
869 	IEEE80211_LOCK_ASSERT(ic);
870 
871 	/*
872 	 * Clear the SCAN bit first in case frames are
873 	 * pending on the station power save queue.  If
874 	 * we defer this then the dispatch of the frames
875 	 * may generate a request to cancel scanning.
876 	 */
877 	ic->ic_flags &= ~IEEE80211_F_SCAN;
878 
879 	/*
880 	 * Drop out of power save mode when a scan has
881 	 * completed.  If this scan was prematurely terminated
882 	 * because it is a background scan then don't notify
883 	 * the ap; we'll either return to scanning after we
884 	 * receive the beacon frame or we'll drop out of power
885 	 * save mode because the beacon indicates we have frames
886 	 * waiting for us.
887 	 */
888 	if (scandone) {
889 		vap->iv_sta_ps(vap, 0);
890 		if (ss->ss_next >= ss->ss_last)
891 			ic->ic_flags_ext &= ~IEEE80211_FEXT_BGSCAN;
892 		ieee80211_notify_scan_done(vap);
893 	}
894 	ss_priv->ss_iflags &= ~(ISCAN_CANCEL|ISCAN_ABORT);
895 	ss_priv->ss_scanend = 0;
896 	ss->ss_flags &= ~(IEEE80211_SCAN_ONCE | IEEE80211_SCAN_PICK1ST);
897 	IEEE80211_UNLOCK(ic);
898 #undef ISCAN_REP
899 }
900 
901 /*
902  * Process a beacon or probe response frame.
903  */
904 static void
905 ieee80211_swscan_add_scan(struct ieee80211vap *vap,
906 	struct ieee80211_channel *curchan,
907 	const struct ieee80211_scanparams *sp,
908 	const struct ieee80211_frame *wh,
909 	int subtype, int rssi, int noise)
910 {
911 	struct ieee80211com *ic = vap->iv_ic;
912 	struct ieee80211_scan_state *ss = ic->ic_scan;
913 
914 	/* XXX locking */
915 	/*
916 	 * Frames received during startup are discarded to avoid
917 	 * using scan state setup on the initial entry to the timer
918 	 * callback.  This can occur because the device may enable
919 	 * rx prior to our doing the initial channel change in the
920 	 * timer routine.
921 	 */
922 	if (SCAN_PRIVATE(ss)->ss_iflags & ISCAN_DISCARD)
923 		return;
924 #ifdef IEEE80211_DEBUG
925 	if (ieee80211_msg_scan(vap) && (ic->ic_flags & IEEE80211_F_SCAN))
926 		ieee80211_scan_dump_probe_beacon(subtype, 1, wh->i_addr2, sp, rssi);
927 #endif
928 	if (ss->ss_ops != NULL &&
929 	    ss->ss_ops->scan_add(ss, curchan, sp, wh, subtype, rssi, noise)) {
930 		/*
931 		 * If we've reached the min dwell time terminate
932 		 * the timer so we'll switch to the next channel.
933 		 */
934 		if ((SCAN_PRIVATE(ss)->ss_iflags & ISCAN_MINDWELL) == 0 &&
935 		    ieee80211_time_after_eq(ticks, SCAN_PRIVATE(ss)->ss_chanmindwell)) {
936 			IEEE80211_DPRINTF(vap, IEEE80211_MSG_SCAN,
937 			    "%s: chan %3d%c min dwell met (%u > %lu)\n",
938 			    __func__,
939 			    ieee80211_chan2ieee(ic, ic->ic_curchan),
940 			    ieee80211_channel_type_char(ic->ic_curchan),
941 			    ticks, SCAN_PRIVATE(ss)->ss_chanmindwell);
942 			SCAN_PRIVATE(ss)->ss_iflags |= ISCAN_MINDWELL;
943 			/*
944 			 * NB: trigger at next clock tick or wait for the
945 			 * hardware.
946 			 */
947 			ic->ic_scan_mindwell(ss);
948 		}
949 	}
950 }
951 
952 static struct ieee80211_scan_methods swscan_methods = {
953 	.sc_attach = ieee80211_swscan_attach,
954 	.sc_detach = ieee80211_swscan_detach,
955 	.sc_vattach = ieee80211_swscan_vattach,
956 	.sc_vdetach = ieee80211_swscan_vdetach,
957 	.sc_set_scan_duration = ieee80211_swscan_set_scan_duration,
958 	.sc_start_scan = ieee80211_swscan_start_scan,
959 	.sc_check_scan = ieee80211_swscan_check_scan,
960 	.sc_bg_scan = ieee80211_swscan_bg_scan,
961 	.sc_cancel_scan = ieee80211_swscan_cancel_scan,
962 	.sc_cancel_anyscan = ieee80211_swscan_cancel_anyscan,
963 	.sc_scan_next = ieee80211_swscan_scan_next,
964 	.sc_scan_done = ieee80211_swscan_scan_done,
965 	.sc_scan_probe_curchan = ieee80211_swscan_probe_curchan,
966 	.sc_add_scan = ieee80211_swscan_add_scan
967 };
968 
969 /*
970  * Default scan attach method.
971  */
972 void
973 ieee80211_swscan_attach(struct ieee80211com *ic)
974 {
975 	struct scan_state *ss;
976 
977 	/*
978 	 * Setup the default methods
979 	 */
980 	ic->ic_scan_methods = &swscan_methods;
981 
982 	/* Allocate initial scan state */
983 #if defined(__DragonFly__)
984 	ss = (struct scan_state *) kmalloc(sizeof(struct scan_state),
985 		M_80211_SCAN, M_INTWAIT | M_ZERO);
986 #else
987 	ss = (struct scan_state *) IEEE80211_MALLOC(sizeof(struct scan_state),
988 		M_80211_SCAN, IEEE80211_M_NOWAIT | IEEE80211_M_ZERO);
989 #endif
990 	if (ss == NULL) {
991 		ic->ic_scan = NULL;
992 		return;
993 	}
994 	TASK_INIT(&ss->ss_scan_start, 0, scan_start, ss);
995 	TIMEOUT_TASK_INIT(ic->ic_tq, &ss->ss_scan_curchan, 0,
996 	    scan_curchan_task, ss);
997 
998 	ic->ic_scan = &ss->base;
999 	ss->base.ss_ic = ic;
1000 
1001 	ic->ic_scan_curchan = scan_curchan;
1002 	ic->ic_scan_mindwell = scan_mindwell;
1003 }
1004