1 /*
2  * Copyright (c) 2014 Qualcomm Atheros, Inc.
3  *
4  * Permission to use, copy, modify, and/or distribute this software for any
5  * purpose with or without fee is hereby granted, provided that the above
6  * copyright notice and this permission notice appear in all copies.
7  *
8  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
9  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
10  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
11  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
12  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
13  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
14  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
15  */
16 
17 #include "ath9k.h"
18 
19 /* Set/change channels.  If the channel is really being changed, it's done
20  * by reseting the chip.  To accomplish this we must first cleanup any pending
21  * DMA, then restart stuff.
22  */
23 static int ath_set_channel(struct ath_softc *sc)
24 {
25 	struct ath_hw *ah = sc->sc_ah;
26 	struct ath_common *common = ath9k_hw_common(ah);
27 	struct ieee80211_hw *hw = sc->hw;
28 	struct ath9k_channel *hchan;
29 	struct cfg80211_chan_def *chandef = &sc->cur_chan->chandef;
30 	struct ieee80211_channel *chan = chandef->chan;
31 	int pos = chan->hw_value;
32 	int old_pos = -1;
33 	int r;
34 
35 	if (test_bit(ATH_OP_INVALID, &common->op_flags))
36 		return -EIO;
37 
38 	if (ah->curchan)
39 		old_pos = ah->curchan - &ah->channels[0];
40 
41 	ath_dbg(common, CONFIG, "Set channel: %d MHz width: %d\n",
42 		chan->center_freq, chandef->width);
43 
44 	/* update survey stats for the old channel before switching */
45 	spin_lock_bh(&common->cc_lock);
46 	ath_update_survey_stats(sc);
47 	spin_unlock_bh(&common->cc_lock);
48 
49 	ath9k_cmn_get_channel(hw, ah, chandef);
50 
51 	/* If the operating channel changes, change the survey in-use flags
52 	 * along with it.
53 	 * Reset the survey data for the new channel, unless we're switching
54 	 * back to the operating channel from an off-channel operation.
55 	 */
56 	if (!sc->cur_chan->offchannel && sc->cur_survey != &sc->survey[pos]) {
57 		if (sc->cur_survey)
58 			sc->cur_survey->filled &= ~SURVEY_INFO_IN_USE;
59 
60 		sc->cur_survey = &sc->survey[pos];
61 
62 		memset(sc->cur_survey, 0, sizeof(struct survey_info));
63 		sc->cur_survey->filled |= SURVEY_INFO_IN_USE;
64 	} else if (!(sc->survey[pos].filled & SURVEY_INFO_IN_USE)) {
65 		memset(&sc->survey[pos], 0, sizeof(struct survey_info));
66 	}
67 
68 	hchan = &sc->sc_ah->channels[pos];
69 	r = ath_reset(sc, hchan);
70 	if (r)
71 		return r;
72 
73 	/* The most recent snapshot of channel->noisefloor for the old
74 	 * channel is only available after the hardware reset. Copy it to
75 	 * the survey stats now.
76 	 */
77 	if (old_pos >= 0)
78 		ath_update_survey_nf(sc, old_pos);
79 
80 	/* Enable radar pulse detection if on a DFS channel. Spectral
81 	 * scanning and radar detection can not be used concurrently.
82 	 */
83 	if (hw->conf.radar_enabled) {
84 		u32 rxfilter;
85 
86 		rxfilter = ath9k_hw_getrxfilter(ah);
87 		rxfilter |= ATH9K_RX_FILTER_PHYRADAR |
88 				ATH9K_RX_FILTER_PHYERR;
89 		ath9k_hw_setrxfilter(ah, rxfilter);
90 		ath_dbg(common, DFS, "DFS enabled at freq %d\n",
91 			chan->center_freq);
92 	} else {
93 		/* perform spectral scan if requested. */
94 		if (test_bit(ATH_OP_SCANNING, &common->op_flags) &&
95 			sc->spec_priv.spectral_mode == SPECTRAL_CHANSCAN)
96 			ath9k_cmn_spectral_scan_trigger(common, &sc->spec_priv);
97 	}
98 
99 	return 0;
100 }
101 
102 void ath_chanctx_init(struct ath_softc *sc)
103 {
104 	struct ath_chanctx *ctx;
105 	struct ath_common *common = ath9k_hw_common(sc->sc_ah);
106 	struct ieee80211_supported_band *sband;
107 	struct ieee80211_channel *chan;
108 	int i, j;
109 
110 	sband = &common->sbands[NL80211_BAND_2GHZ];
111 	if (!sband->n_channels)
112 		sband = &common->sbands[NL80211_BAND_5GHZ];
113 
114 	chan = &sband->channels[0];
115 	for (i = 0; i < ATH9K_NUM_CHANCTX; i++) {
116 		ctx = &sc->chanctx[i];
117 		cfg80211_chandef_create(&ctx->chandef, chan, NL80211_CHAN_HT20);
118 		INIT_LIST_HEAD(&ctx->vifs);
119 		ctx->txpower = ATH_TXPOWER_MAX;
120 		ctx->flush_timeout = HZ / 5; /* 200ms */
121 		for (j = 0; j < ARRAY_SIZE(ctx->acq); j++) {
122 			INIT_LIST_HEAD(&ctx->acq[j].acq_new);
123 			INIT_LIST_HEAD(&ctx->acq[j].acq_old);
124 			spin_lock_init(&ctx->acq[j].lock);
125 		}
126 	}
127 }
128 
129 void ath_chanctx_set_channel(struct ath_softc *sc, struct ath_chanctx *ctx,
130 			     struct cfg80211_chan_def *chandef)
131 {
132 	struct ath_common *common = ath9k_hw_common(sc->sc_ah);
133 	bool cur_chan;
134 
135 	spin_lock_bh(&sc->chan_lock);
136 	if (chandef)
137 		memcpy(&ctx->chandef, chandef, sizeof(*chandef));
138 	cur_chan = sc->cur_chan == ctx;
139 	spin_unlock_bh(&sc->chan_lock);
140 
141 	if (!cur_chan) {
142 		ath_dbg(common, CHAN_CTX,
143 			"Current context differs from the new context\n");
144 		return;
145 	}
146 
147 	ath_set_channel(sc);
148 }
149 
150 #ifdef CONFIG_ATH9K_CHANNEL_CONTEXT
151 
152 /*************/
153 /* Utilities */
154 /*************/
155 
156 struct ath_chanctx* ath_is_go_chanctx_present(struct ath_softc *sc)
157 {
158 	struct ath_chanctx *ctx;
159 	struct ath_vif *avp;
160 	struct ieee80211_vif *vif;
161 
162 	spin_lock_bh(&sc->chan_lock);
163 
164 	ath_for_each_chanctx(sc, ctx) {
165 		if (!ctx->active)
166 			continue;
167 
168 		list_for_each_entry(avp, &ctx->vifs, list) {
169 			vif = avp->vif;
170 
171 			if (ieee80211_vif_type_p2p(vif) == NL80211_IFTYPE_P2P_GO) {
172 				spin_unlock_bh(&sc->chan_lock);
173 				return ctx;
174 			}
175 		}
176 	}
177 
178 	spin_unlock_bh(&sc->chan_lock);
179 	return NULL;
180 }
181 
182 /**********************************************************/
183 /* Functions to handle the channel context state machine. */
184 /**********************************************************/
185 
186 static const char *offchannel_state_string(enum ath_offchannel_state state)
187 {
188 	switch (state) {
189 		case_rtn_string(ATH_OFFCHANNEL_IDLE);
190 		case_rtn_string(ATH_OFFCHANNEL_PROBE_SEND);
191 		case_rtn_string(ATH_OFFCHANNEL_PROBE_WAIT);
192 		case_rtn_string(ATH_OFFCHANNEL_SUSPEND);
193 		case_rtn_string(ATH_OFFCHANNEL_ROC_START);
194 		case_rtn_string(ATH_OFFCHANNEL_ROC_WAIT);
195 		case_rtn_string(ATH_OFFCHANNEL_ROC_DONE);
196 	default:
197 		return "unknown";
198 	}
199 }
200 
201 static const char *chanctx_event_string(enum ath_chanctx_event ev)
202 {
203 	switch (ev) {
204 		case_rtn_string(ATH_CHANCTX_EVENT_BEACON_PREPARE);
205 		case_rtn_string(ATH_CHANCTX_EVENT_BEACON_SENT);
206 		case_rtn_string(ATH_CHANCTX_EVENT_TSF_TIMER);
207 		case_rtn_string(ATH_CHANCTX_EVENT_BEACON_RECEIVED);
208 		case_rtn_string(ATH_CHANCTX_EVENT_AUTHORIZED);
209 		case_rtn_string(ATH_CHANCTX_EVENT_SWITCH);
210 		case_rtn_string(ATH_CHANCTX_EVENT_ASSIGN);
211 		case_rtn_string(ATH_CHANCTX_EVENT_UNASSIGN);
212 		case_rtn_string(ATH_CHANCTX_EVENT_CHANGE);
213 		case_rtn_string(ATH_CHANCTX_EVENT_ENABLE_MULTICHANNEL);
214 	default:
215 		return "unknown";
216 	}
217 }
218 
219 static const char *chanctx_state_string(enum ath_chanctx_state state)
220 {
221 	switch (state) {
222 		case_rtn_string(ATH_CHANCTX_STATE_IDLE);
223 		case_rtn_string(ATH_CHANCTX_STATE_WAIT_FOR_BEACON);
224 		case_rtn_string(ATH_CHANCTX_STATE_WAIT_FOR_TIMER);
225 		case_rtn_string(ATH_CHANCTX_STATE_SWITCH);
226 		case_rtn_string(ATH_CHANCTX_STATE_FORCE_ACTIVE);
227 	default:
228 		return "unknown";
229 	}
230 }
231 
232 static u32 chanctx_event_delta(struct ath_softc *sc)
233 {
234 	u64 ms;
235 	struct timespec ts, *old;
236 
237 	getrawmonotonic(&ts);
238 	old = &sc->last_event_time;
239 	ms = ts.tv_sec * 1000 + ts.tv_nsec / 1000000;
240 	ms -= old->tv_sec * 1000 + old->tv_nsec / 1000000;
241 	sc->last_event_time = ts;
242 
243 	return (u32)ms;
244 }
245 
246 void ath_chanctx_check_active(struct ath_softc *sc, struct ath_chanctx *ctx)
247 {
248 	struct ath_common *common = ath9k_hw_common(sc->sc_ah);
249 	struct ath_chanctx *ictx;
250 	struct ath_vif *avp;
251 	bool active = false;
252 	u8 n_active = 0;
253 
254 	if (!ctx)
255 		return;
256 
257 	if (ctx == &sc->offchannel.chan) {
258 		spin_lock_bh(&sc->chan_lock);
259 
260 		if (likely(sc->sched.channel_switch_time))
261 			ctx->flush_timeout =
262 				usecs_to_jiffies(sc->sched.channel_switch_time);
263 		else
264 			ctx->flush_timeout =
265 				msecs_to_jiffies(10);
266 
267 		spin_unlock_bh(&sc->chan_lock);
268 
269 		/*
270 		 * There is no need to iterate over the
271 		 * active/assigned channel contexts if
272 		 * the current context is offchannel.
273 		 */
274 		return;
275 	}
276 
277 	ictx = ctx;
278 
279 	list_for_each_entry(avp, &ctx->vifs, list) {
280 		struct ieee80211_vif *vif = avp->vif;
281 
282 		switch (vif->type) {
283 		case NL80211_IFTYPE_P2P_CLIENT:
284 		case NL80211_IFTYPE_STATION:
285 			if (avp->assoc)
286 				active = true;
287 			break;
288 		default:
289 			active = true;
290 			break;
291 		}
292 	}
293 	ctx->active = active;
294 
295 	ath_for_each_chanctx(sc, ctx) {
296 		if (!ctx->assigned || list_empty(&ctx->vifs))
297 			continue;
298 		n_active++;
299 	}
300 
301 	spin_lock_bh(&sc->chan_lock);
302 
303 	if (n_active <= 1) {
304 		ictx->flush_timeout = HZ / 5;
305 		clear_bit(ATH_OP_MULTI_CHANNEL, &common->op_flags);
306 		spin_unlock_bh(&sc->chan_lock);
307 		return;
308 	}
309 
310 	ictx->flush_timeout = usecs_to_jiffies(sc->sched.channel_switch_time);
311 
312 	if (test_and_set_bit(ATH_OP_MULTI_CHANNEL, &common->op_flags)) {
313 		spin_unlock_bh(&sc->chan_lock);
314 		return;
315 	}
316 
317 	spin_unlock_bh(&sc->chan_lock);
318 
319 	if (ath9k_is_chanctx_enabled()) {
320 		ath_chanctx_event(sc, NULL,
321 				  ATH_CHANCTX_EVENT_ENABLE_MULTICHANNEL);
322 	}
323 }
324 
325 static struct ath_chanctx *
326 ath_chanctx_get_next(struct ath_softc *sc, struct ath_chanctx *ctx)
327 {
328 	int idx = ctx - &sc->chanctx[0];
329 
330 	return &sc->chanctx[!idx];
331 }
332 
333 static void ath_chanctx_adjust_tbtt_delta(struct ath_softc *sc)
334 {
335 	struct ath_chanctx *prev, *cur;
336 	struct timespec ts;
337 	u32 cur_tsf, prev_tsf, beacon_int;
338 	s32 offset;
339 
340 	beacon_int = TU_TO_USEC(sc->cur_chan->beacon.beacon_interval);
341 
342 	cur = sc->cur_chan;
343 	prev = ath_chanctx_get_next(sc, cur);
344 
345 	if (!prev->switch_after_beacon)
346 		return;
347 
348 	getrawmonotonic(&ts);
349 	cur_tsf = (u32) cur->tsf_val +
350 		  ath9k_hw_get_tsf_offset(&cur->tsf_ts, &ts);
351 
352 	prev_tsf = prev->last_beacon - (u32) prev->tsf_val + cur_tsf;
353 	prev_tsf -= ath9k_hw_get_tsf_offset(&prev->tsf_ts, &ts);
354 
355 	/* Adjust the TSF time of the AP chanctx to keep its beacons
356 	 * at half beacon interval offset relative to the STA chanctx.
357 	 */
358 	offset = cur_tsf - prev_tsf;
359 
360 	/* Ignore stale data or spurious timestamps */
361 	if (offset < 0 || offset > 3 * beacon_int)
362 		return;
363 
364 	offset = beacon_int / 2 - (offset % beacon_int);
365 	prev->tsf_val += offset;
366 }
367 
368 /* Configure the TSF based hardware timer for a channel switch.
369  * Also set up backup software timer, in case the gen timer fails.
370  * This could be caused by a hardware reset.
371  */
372 static void ath_chanctx_setup_timer(struct ath_softc *sc, u32 tsf_time)
373 {
374 	struct ath_common *common = ath9k_hw_common(sc->sc_ah);
375 	struct ath_hw *ah = sc->sc_ah;
376 	unsigned long timeout;
377 
378 	ath9k_hw_gen_timer_start(ah, sc->p2p_ps_timer, tsf_time, 1000000);
379 	tsf_time -= ath9k_hw_gettsf32(ah);
380 	timeout = msecs_to_jiffies(tsf_time / 1000) + 1;
381 	mod_timer(&sc->sched.timer, jiffies + timeout);
382 
383 	ath_dbg(common, CHAN_CTX,
384 		"Setup chanctx timer with timeout: %d (%d) ms\n",
385 		tsf_time / 1000, jiffies_to_msecs(timeout));
386 }
387 
388 static void ath_chanctx_handle_bmiss(struct ath_softc *sc,
389 				     struct ath_chanctx *ctx,
390 				     struct ath_vif *avp)
391 {
392 	/*
393 	 * Clear the extend_absence flag if it had been
394 	 * set during the previous beacon transmission,
395 	 * since we need to revert to the normal NoA
396 	 * schedule.
397 	 */
398 	if (ctx->active && sc->sched.extend_absence) {
399 		avp->noa_duration = 0;
400 		sc->sched.extend_absence = false;
401 	}
402 
403 	/* If at least two consecutive beacons were missed on the STA
404 	 * chanctx, stay on the STA channel for one extra beacon period,
405 	 * to resync the timer properly.
406 	 */
407 	if (ctx->active && sc->sched.beacon_miss >= 2) {
408 		avp->noa_duration = 0;
409 		sc->sched.extend_absence = true;
410 	}
411 }
412 
413 static void ath_chanctx_offchannel_noa(struct ath_softc *sc,
414 				       struct ath_chanctx *ctx,
415 				       struct ath_vif *avp,
416 				       u32 tsf_time)
417 {
418 	struct ath_common *common = ath9k_hw_common(sc->sc_ah);
419 
420 	avp->noa_index++;
421 	avp->offchannel_start = tsf_time;
422 	avp->offchannel_duration = sc->sched.offchannel_duration;
423 
424 	ath_dbg(common, CHAN_CTX,
425 		"offchannel noa_duration: %d, noa_start: %u, noa_index: %d\n",
426 		avp->offchannel_duration,
427 		avp->offchannel_start,
428 		avp->noa_index);
429 
430 	/*
431 	 * When multiple contexts are active, the NoA
432 	 * has to be recalculated and advertised after
433 	 * an offchannel operation.
434 	 */
435 	if (ctx->active && avp->noa_duration)
436 		avp->noa_duration = 0;
437 }
438 
439 static void ath_chanctx_set_periodic_noa(struct ath_softc *sc,
440 					 struct ath_vif *avp,
441 					 struct ath_beacon_config *cur_conf,
442 					 u32 tsf_time,
443 					 u32 beacon_int)
444 {
445 	struct ath_common *common = ath9k_hw_common(sc->sc_ah);
446 
447 	avp->noa_index++;
448 	avp->noa_start = tsf_time;
449 
450 	if (sc->sched.extend_absence)
451 		avp->noa_duration = (3 * beacon_int / 2) +
452 			sc->sched.channel_switch_time;
453 	else
454 		avp->noa_duration =
455 			TU_TO_USEC(cur_conf->beacon_interval) / 2 +
456 			sc->sched.channel_switch_time;
457 
458 	if (test_bit(ATH_OP_SCANNING, &common->op_flags) ||
459 	    sc->sched.extend_absence)
460 		avp->periodic_noa = false;
461 	else
462 		avp->periodic_noa = true;
463 
464 	ath_dbg(common, CHAN_CTX,
465 		"noa_duration: %d, noa_start: %u, noa_index: %d, periodic: %d\n",
466 		avp->noa_duration,
467 		avp->noa_start,
468 		avp->noa_index,
469 		avp->periodic_noa);
470 }
471 
472 static void ath_chanctx_set_oneshot_noa(struct ath_softc *sc,
473 					struct ath_vif *avp,
474 					u32 tsf_time,
475 					u32 duration)
476 {
477 	struct ath_common *common = ath9k_hw_common(sc->sc_ah);
478 
479 	avp->noa_index++;
480 	avp->noa_start = tsf_time;
481 	avp->periodic_noa = false;
482 	avp->oneshot_noa = true;
483 	avp->noa_duration = duration + sc->sched.channel_switch_time;
484 
485 	ath_dbg(common, CHAN_CTX,
486 		"oneshot noa_duration: %d, noa_start: %u, noa_index: %d, periodic: %d\n",
487 		avp->noa_duration,
488 		avp->noa_start,
489 		avp->noa_index,
490 		avp->periodic_noa);
491 }
492 
493 void ath_chanctx_event(struct ath_softc *sc, struct ieee80211_vif *vif,
494 		       enum ath_chanctx_event ev)
495 {
496 	struct ath_hw *ah = sc->sc_ah;
497 	struct ath_common *common = ath9k_hw_common(ah);
498 	struct ath_beacon_config *cur_conf;
499 	struct ath_vif *avp = NULL;
500 	struct ath_chanctx *ctx;
501 	u32 tsf_time;
502 	u32 beacon_int;
503 
504 	if (vif)
505 		avp = (struct ath_vif *) vif->drv_priv;
506 
507 	spin_lock_bh(&sc->chan_lock);
508 
509 	ath_dbg(common, CHAN_CTX, "cur_chan: %d MHz, event: %s, state: %s, delta: %u ms\n",
510 		sc->cur_chan->chandef.center_freq1,
511 		chanctx_event_string(ev),
512 		chanctx_state_string(sc->sched.state),
513 		chanctx_event_delta(sc));
514 
515 	switch (ev) {
516 	case ATH_CHANCTX_EVENT_BEACON_PREPARE:
517 		if (avp->offchannel_duration)
518 			avp->offchannel_duration = 0;
519 
520 		if (avp->oneshot_noa) {
521 			avp->noa_duration = 0;
522 			avp->oneshot_noa = false;
523 
524 			ath_dbg(common, CHAN_CTX,
525 				"Clearing oneshot NoA\n");
526 		}
527 
528 		if (avp->chanctx != sc->cur_chan) {
529 			ath_dbg(common, CHAN_CTX,
530 				"Contexts differ, not preparing beacon\n");
531 			break;
532 		}
533 
534 		if (sc->sched.offchannel_pending && !sc->sched.wait_switch) {
535 			sc->sched.offchannel_pending = false;
536 			sc->next_chan = &sc->offchannel.chan;
537 			sc->sched.state = ATH_CHANCTX_STATE_WAIT_FOR_BEACON;
538 			ath_dbg(common, CHAN_CTX,
539 				"Setting offchannel_pending to false\n");
540 		}
541 
542 		ctx = ath_chanctx_get_next(sc, sc->cur_chan);
543 		if (ctx->active && sc->sched.state == ATH_CHANCTX_STATE_IDLE) {
544 			sc->next_chan = ctx;
545 			sc->sched.state = ATH_CHANCTX_STATE_WAIT_FOR_BEACON;
546 			ath_dbg(common, CHAN_CTX,
547 				"Set next context, move chanctx state to WAIT_FOR_BEACON\n");
548 		}
549 
550 		/* if the timer missed its window, use the next interval */
551 		if (sc->sched.state == ATH_CHANCTX_STATE_WAIT_FOR_TIMER) {
552 			sc->sched.state = ATH_CHANCTX_STATE_WAIT_FOR_BEACON;
553 			ath_dbg(common, CHAN_CTX,
554 				"Move chanctx state from WAIT_FOR_TIMER to WAIT_FOR_BEACON\n");
555 		}
556 
557 		if (sc->sched.mgd_prepare_tx)
558 			sc->sched.state = ATH_CHANCTX_STATE_WAIT_FOR_BEACON;
559 
560 		/*
561 		 * When a context becomes inactive, for example,
562 		 * disassociation of a station context, the NoA
563 		 * attribute needs to be removed from subsequent
564 		 * beacons.
565 		 */
566 		if (!ctx->active && avp->noa_duration &&
567 		    sc->sched.state != ATH_CHANCTX_STATE_WAIT_FOR_BEACON) {
568 			avp->noa_duration = 0;
569 			avp->periodic_noa = false;
570 
571 			ath_dbg(common, CHAN_CTX,
572 				"Clearing NoA schedule\n");
573 		}
574 
575 		if (sc->sched.state != ATH_CHANCTX_STATE_WAIT_FOR_BEACON)
576 			break;
577 
578 		ath_dbg(common, CHAN_CTX, "Preparing beacon for vif: %pM\n", vif->addr);
579 
580 		sc->sched.beacon_pending = true;
581 		sc->sched.next_tbtt = REG_READ(ah, AR_NEXT_TBTT_TIMER);
582 
583 		cur_conf = &sc->cur_chan->beacon;
584 		beacon_int = TU_TO_USEC(cur_conf->beacon_interval);
585 
586 		/* defer channel switch by a quarter beacon interval */
587 		tsf_time = sc->sched.next_tbtt + beacon_int / 4;
588 		sc->sched.switch_start_time = tsf_time;
589 		sc->cur_chan->last_beacon = sc->sched.next_tbtt;
590 
591 		/*
592 		 * If an offchannel switch is scheduled to happen after
593 		 * a beacon transmission, update the NoA with one-shot
594 		 * values and increment the index.
595 		 */
596 		if (sc->next_chan == &sc->offchannel.chan) {
597 			ath_chanctx_offchannel_noa(sc, ctx, avp, tsf_time);
598 			break;
599 		}
600 
601 		ath_chanctx_handle_bmiss(sc, ctx, avp);
602 
603 		/*
604 		 * If a mgd_prepare_tx() has been called by mac80211,
605 		 * a one-shot NoA needs to be sent. This can happen
606 		 * with one or more active channel contexts - in both
607 		 * cases, a new NoA schedule has to be advertised.
608 		 */
609 		if (sc->sched.mgd_prepare_tx) {
610 			ath_chanctx_set_oneshot_noa(sc, avp, tsf_time,
611 						    jiffies_to_usecs(HZ / 5));
612 			break;
613 		}
614 
615 		/* Prevent wrap-around issues */
616 		if (avp->noa_duration && tsf_time - avp->noa_start > BIT(30))
617 			avp->noa_duration = 0;
618 
619 		/*
620 		 * If multiple contexts are active, start periodic
621 		 * NoA and increment the index for the first
622 		 * announcement.
623 		 */
624 		if (ctx->active &&
625 		    (!avp->noa_duration || sc->sched.force_noa_update))
626 			ath_chanctx_set_periodic_noa(sc, avp, cur_conf,
627 						     tsf_time, beacon_int);
628 
629 		if (ctx->active && sc->sched.force_noa_update)
630 			sc->sched.force_noa_update = false;
631 
632 		break;
633 	case ATH_CHANCTX_EVENT_BEACON_SENT:
634 		if (!sc->sched.beacon_pending) {
635 			ath_dbg(common, CHAN_CTX,
636 				"No pending beacon\n");
637 			break;
638 		}
639 
640 		sc->sched.beacon_pending = false;
641 
642 		if (sc->sched.mgd_prepare_tx) {
643 			sc->sched.mgd_prepare_tx = false;
644 			complete(&sc->go_beacon);
645 			ath_dbg(common, CHAN_CTX,
646 				"Beacon sent, complete go_beacon\n");
647 			break;
648 		}
649 
650 		if (sc->sched.state != ATH_CHANCTX_STATE_WAIT_FOR_BEACON)
651 			break;
652 
653 		ath_dbg(common, CHAN_CTX,
654 			"Move chanctx state to WAIT_FOR_TIMER\n");
655 
656 		sc->sched.state = ATH_CHANCTX_STATE_WAIT_FOR_TIMER;
657 		ath_chanctx_setup_timer(sc, sc->sched.switch_start_time);
658 		break;
659 	case ATH_CHANCTX_EVENT_TSF_TIMER:
660 		if (sc->sched.state != ATH_CHANCTX_STATE_WAIT_FOR_TIMER)
661 			break;
662 
663 		if (!sc->cur_chan->switch_after_beacon &&
664 		    sc->sched.beacon_pending)
665 			sc->sched.beacon_miss++;
666 
667 		ath_dbg(common, CHAN_CTX,
668 			"Move chanctx state to SWITCH\n");
669 
670 		sc->sched.state = ATH_CHANCTX_STATE_SWITCH;
671 		ieee80211_queue_work(sc->hw, &sc->chanctx_work);
672 		break;
673 	case ATH_CHANCTX_EVENT_BEACON_RECEIVED:
674 		if (!test_bit(ATH_OP_MULTI_CHANNEL, &common->op_flags) ||
675 		    sc->cur_chan == &sc->offchannel.chan)
676 			break;
677 
678 		sc->sched.beacon_pending = false;
679 		sc->sched.beacon_miss = 0;
680 
681 		if (sc->sched.state == ATH_CHANCTX_STATE_FORCE_ACTIVE ||
682 		    !sc->sched.beacon_adjust ||
683 		    !sc->cur_chan->tsf_val)
684 			break;
685 
686 		ath_chanctx_adjust_tbtt_delta(sc);
687 
688 		/* TSF time might have been updated by the incoming beacon,
689 		 * need update the channel switch timer to reflect the change.
690 		 */
691 		tsf_time = sc->sched.switch_start_time;
692 		tsf_time -= (u32) sc->cur_chan->tsf_val +
693 			ath9k_hw_get_tsf_offset(&sc->cur_chan->tsf_ts, NULL);
694 		tsf_time += ath9k_hw_gettsf32(ah);
695 
696 		sc->sched.beacon_adjust = false;
697 		ath_chanctx_setup_timer(sc, tsf_time);
698 		break;
699 	case ATH_CHANCTX_EVENT_AUTHORIZED:
700 		if (sc->sched.state != ATH_CHANCTX_STATE_FORCE_ACTIVE ||
701 		    avp->chanctx != sc->cur_chan)
702 			break;
703 
704 		ath_dbg(common, CHAN_CTX,
705 			"Move chanctx state from FORCE_ACTIVE to IDLE\n");
706 
707 		sc->sched.state = ATH_CHANCTX_STATE_IDLE;
708 		/* fall through */
709 	case ATH_CHANCTX_EVENT_SWITCH:
710 		if (!test_bit(ATH_OP_MULTI_CHANNEL, &common->op_flags) ||
711 		    sc->sched.state == ATH_CHANCTX_STATE_FORCE_ACTIVE ||
712 		    sc->cur_chan->switch_after_beacon ||
713 		    sc->cur_chan == &sc->offchannel.chan)
714 			break;
715 
716 		/* If this is a station chanctx, stay active for a half
717 		 * beacon period (minus channel switch time)
718 		 */
719 		sc->next_chan = ath_chanctx_get_next(sc, sc->cur_chan);
720 		cur_conf = &sc->cur_chan->beacon;
721 
722 		ath_dbg(common, CHAN_CTX,
723 			"Move chanctx state to WAIT_FOR_TIMER (event SWITCH)\n");
724 
725 		sc->sched.state = ATH_CHANCTX_STATE_WAIT_FOR_TIMER;
726 		sc->sched.wait_switch = false;
727 
728 		tsf_time = TU_TO_USEC(cur_conf->beacon_interval) / 2;
729 
730 		if (sc->sched.extend_absence) {
731 			sc->sched.beacon_miss = 0;
732 			tsf_time *= 3;
733 		}
734 
735 		tsf_time -= sc->sched.channel_switch_time;
736 		tsf_time += ath9k_hw_gettsf32(sc->sc_ah);
737 		sc->sched.switch_start_time = tsf_time;
738 
739 		ath_chanctx_setup_timer(sc, tsf_time);
740 		sc->sched.beacon_pending = true;
741 		sc->sched.beacon_adjust = true;
742 		break;
743 	case ATH_CHANCTX_EVENT_ENABLE_MULTICHANNEL:
744 		if (sc->cur_chan == &sc->offchannel.chan ||
745 		    sc->cur_chan->switch_after_beacon)
746 			break;
747 
748 		sc->next_chan = ath_chanctx_get_next(sc, sc->cur_chan);
749 		ieee80211_queue_work(sc->hw, &sc->chanctx_work);
750 		break;
751 	case ATH_CHANCTX_EVENT_UNASSIGN:
752 		if (sc->cur_chan->assigned) {
753 			if (sc->next_chan && !sc->next_chan->assigned &&
754 			    sc->next_chan != &sc->offchannel.chan)
755 				sc->sched.state = ATH_CHANCTX_STATE_IDLE;
756 			break;
757 		}
758 
759 		ctx = ath_chanctx_get_next(sc, sc->cur_chan);
760 		sc->sched.state = ATH_CHANCTX_STATE_IDLE;
761 		if (!ctx->assigned)
762 			break;
763 
764 		sc->next_chan = ctx;
765 		ieee80211_queue_work(sc->hw, &sc->chanctx_work);
766 		break;
767 	case ATH_CHANCTX_EVENT_ASSIGN:
768 		break;
769 	case ATH_CHANCTX_EVENT_CHANGE:
770 		break;
771 	}
772 
773 	spin_unlock_bh(&sc->chan_lock);
774 }
775 
776 void ath_chanctx_beacon_sent_ev(struct ath_softc *sc,
777 				enum ath_chanctx_event ev)
778 {
779 	if (sc->sched.beacon_pending)
780 		ath_chanctx_event(sc, NULL, ev);
781 }
782 
783 void ath_chanctx_beacon_recv_ev(struct ath_softc *sc,
784 				enum ath_chanctx_event ev)
785 {
786 	ath_chanctx_event(sc, NULL, ev);
787 }
788 
789 static int ath_scan_channel_duration(struct ath_softc *sc,
790 				     struct ieee80211_channel *chan)
791 {
792 	struct cfg80211_scan_request *req = sc->offchannel.scan_req;
793 
794 	if (!req->n_ssids || (chan->flags & IEEE80211_CHAN_NO_IR))
795 		return (HZ / 9); /* ~110 ms */
796 
797 	return (HZ / 16); /* ~60 ms */
798 }
799 
800 static void ath_chanctx_switch(struct ath_softc *sc, struct ath_chanctx *ctx,
801 			       struct cfg80211_chan_def *chandef)
802 {
803 	struct ath_common *common = ath9k_hw_common(sc->sc_ah);
804 
805 	spin_lock_bh(&sc->chan_lock);
806 
807 	if (test_bit(ATH_OP_MULTI_CHANNEL, &common->op_flags) &&
808 	    (sc->cur_chan != ctx) && (ctx == &sc->offchannel.chan)) {
809 		if (chandef)
810 			ctx->chandef = *chandef;
811 
812 		sc->sched.offchannel_pending = true;
813 		sc->sched.wait_switch = true;
814 		sc->sched.offchannel_duration =
815 			jiffies_to_usecs(sc->offchannel.duration) +
816 			sc->sched.channel_switch_time;
817 
818 		spin_unlock_bh(&sc->chan_lock);
819 		ath_dbg(common, CHAN_CTX,
820 			"Set offchannel_pending to true\n");
821 		return;
822 	}
823 
824 	sc->next_chan = ctx;
825 	if (chandef) {
826 		ctx->chandef = *chandef;
827 		ath_dbg(common, CHAN_CTX,
828 			"Assigned next_chan to %d MHz\n", chandef->center_freq1);
829 	}
830 
831 	if (sc->next_chan == &sc->offchannel.chan) {
832 		sc->sched.offchannel_duration =
833 			jiffies_to_usecs(sc->offchannel.duration) +
834 			sc->sched.channel_switch_time;
835 
836 		if (chandef) {
837 			ath_dbg(common, CHAN_CTX,
838 				"Offchannel duration for chan %d MHz : %u\n",
839 				chandef->center_freq1,
840 				sc->sched.offchannel_duration);
841 		}
842 	}
843 	spin_unlock_bh(&sc->chan_lock);
844 	ieee80211_queue_work(sc->hw, &sc->chanctx_work);
845 }
846 
847 static void ath_chanctx_offchan_switch(struct ath_softc *sc,
848 				       struct ieee80211_channel *chan)
849 {
850 	struct ath_common *common = ath9k_hw_common(sc->sc_ah);
851 	struct cfg80211_chan_def chandef;
852 
853 	cfg80211_chandef_create(&chandef, chan, NL80211_CHAN_NO_HT);
854 	ath_dbg(common, CHAN_CTX,
855 		"Channel definition created: %d MHz\n", chandef.center_freq1);
856 
857 	ath_chanctx_switch(sc, &sc->offchannel.chan, &chandef);
858 }
859 
860 static struct ath_chanctx *ath_chanctx_get_oper_chan(struct ath_softc *sc,
861 						     bool active)
862 {
863 	struct ath_chanctx *ctx;
864 
865 	ath_for_each_chanctx(sc, ctx) {
866 		if (!ctx->assigned || list_empty(&ctx->vifs))
867 			continue;
868 		if (active && !ctx->active)
869 			continue;
870 
871 		if (ctx->switch_after_beacon)
872 			return ctx;
873 	}
874 
875 	return &sc->chanctx[0];
876 }
877 
878 static void
879 ath_scan_next_channel(struct ath_softc *sc)
880 {
881 	struct ath_common *common = ath9k_hw_common(sc->sc_ah);
882 	struct cfg80211_scan_request *req = sc->offchannel.scan_req;
883 	struct ieee80211_channel *chan;
884 
885 	if (sc->offchannel.scan_idx >= req->n_channels) {
886 		ath_dbg(common, CHAN_CTX,
887 			"Moving offchannel state to ATH_OFFCHANNEL_IDLE, "
888 			"scan_idx: %d, n_channels: %d\n",
889 			sc->offchannel.scan_idx,
890 			req->n_channels);
891 
892 		sc->offchannel.state = ATH_OFFCHANNEL_IDLE;
893 		ath_chanctx_switch(sc, ath_chanctx_get_oper_chan(sc, false),
894 				   NULL);
895 		return;
896 	}
897 
898 	ath_dbg(common, CHAN_CTX,
899 		"Moving offchannel state to ATH_OFFCHANNEL_PROBE_SEND, scan_idx: %d\n",
900 		sc->offchannel.scan_idx);
901 
902 	chan = req->channels[sc->offchannel.scan_idx++];
903 	sc->offchannel.duration = ath_scan_channel_duration(sc, chan);
904 	sc->offchannel.state = ATH_OFFCHANNEL_PROBE_SEND;
905 
906 	ath_chanctx_offchan_switch(sc, chan);
907 }
908 
909 void ath_offchannel_next(struct ath_softc *sc)
910 {
911 	struct ieee80211_vif *vif;
912 
913 	if (sc->offchannel.scan_req) {
914 		vif = sc->offchannel.scan_vif;
915 		sc->offchannel.chan.txpower = vif->bss_conf.txpower;
916 		ath_scan_next_channel(sc);
917 	} else if (sc->offchannel.roc_vif) {
918 		vif = sc->offchannel.roc_vif;
919 		sc->offchannel.chan.txpower = vif->bss_conf.txpower;
920 		sc->offchannel.duration =
921 			msecs_to_jiffies(sc->offchannel.roc_duration);
922 		sc->offchannel.state = ATH_OFFCHANNEL_ROC_START;
923 		ath_chanctx_offchan_switch(sc, sc->offchannel.roc_chan);
924 	} else {
925 		spin_lock_bh(&sc->chan_lock);
926 		sc->sched.offchannel_pending = false;
927 		sc->sched.wait_switch = false;
928 		spin_unlock_bh(&sc->chan_lock);
929 
930 		ath_chanctx_switch(sc, ath_chanctx_get_oper_chan(sc, false),
931 				   NULL);
932 		sc->offchannel.state = ATH_OFFCHANNEL_IDLE;
933 		if (sc->ps_idle)
934 			ath_cancel_work(sc);
935 	}
936 }
937 
938 void ath_roc_complete(struct ath_softc *sc, enum ath_roc_complete_reason reason)
939 {
940 	struct ath_common *common = ath9k_hw_common(sc->sc_ah);
941 
942 	sc->offchannel.roc_vif = NULL;
943 	sc->offchannel.roc_chan = NULL;
944 
945 	switch (reason) {
946 	case ATH_ROC_COMPLETE_ABORT:
947 		ath_dbg(common, CHAN_CTX, "RoC aborted\n");
948 		ieee80211_remain_on_channel_expired(sc->hw);
949 		break;
950 	case ATH_ROC_COMPLETE_EXPIRE:
951 		ath_dbg(common, CHAN_CTX, "RoC expired\n");
952 		ieee80211_remain_on_channel_expired(sc->hw);
953 		break;
954 	case ATH_ROC_COMPLETE_CANCEL:
955 		ath_dbg(common, CHAN_CTX, "RoC canceled\n");
956 		break;
957 	}
958 
959 	ath_offchannel_next(sc);
960 	ath9k_ps_restore(sc);
961 }
962 
963 void ath_scan_complete(struct ath_softc *sc, bool abort)
964 {
965 	struct ath_common *common = ath9k_hw_common(sc->sc_ah);
966 	struct cfg80211_scan_info info = {
967 		.aborted = abort,
968 	};
969 
970 	if (abort)
971 		ath_dbg(common, CHAN_CTX, "HW scan aborted\n");
972 	else
973 		ath_dbg(common, CHAN_CTX, "HW scan complete\n");
974 
975 	sc->offchannel.scan_req = NULL;
976 	sc->offchannel.scan_vif = NULL;
977 	sc->offchannel.state = ATH_OFFCHANNEL_IDLE;
978 	ieee80211_scan_completed(sc->hw, &info);
979 	clear_bit(ATH_OP_SCANNING, &common->op_flags);
980 	spin_lock_bh(&sc->chan_lock);
981 	if (test_bit(ATH_OP_MULTI_CHANNEL, &common->op_flags))
982 		sc->sched.force_noa_update = true;
983 	spin_unlock_bh(&sc->chan_lock);
984 	ath_offchannel_next(sc);
985 	ath9k_ps_restore(sc);
986 }
987 
988 static void ath_scan_send_probe(struct ath_softc *sc,
989 				struct cfg80211_ssid *ssid)
990 {
991 	struct cfg80211_scan_request *req = sc->offchannel.scan_req;
992 	struct ieee80211_vif *vif = sc->offchannel.scan_vif;
993 	struct ath_tx_control txctl = {};
994 	struct sk_buff *skb;
995 	struct ieee80211_tx_info *info;
996 	int band = sc->offchannel.chan.chandef.chan->band;
997 
998 	skb = ieee80211_probereq_get(sc->hw, vif->addr,
999 			ssid->ssid, ssid->ssid_len, req->ie_len);
1000 	if (!skb)
1001 		return;
1002 
1003 	info = IEEE80211_SKB_CB(skb);
1004 	if (req->no_cck)
1005 		info->flags |= IEEE80211_TX_CTL_NO_CCK_RATE;
1006 
1007 	if (req->ie_len)
1008 		skb_put_data(skb, req->ie, req->ie_len);
1009 
1010 	skb_set_queue_mapping(skb, IEEE80211_AC_VO);
1011 
1012 	if (!ieee80211_tx_prepare_skb(sc->hw, vif, skb, band, NULL))
1013 		goto error;
1014 
1015 	txctl.txq = sc->tx.txq_map[IEEE80211_AC_VO];
1016 	if (ath_tx_start(sc->hw, skb, &txctl))
1017 		goto error;
1018 
1019 	return;
1020 
1021 error:
1022 	ieee80211_free_txskb(sc->hw, skb);
1023 }
1024 
1025 static void ath_scan_channel_start(struct ath_softc *sc)
1026 {
1027 	struct ath_common *common = ath9k_hw_common(sc->sc_ah);
1028 	struct cfg80211_scan_request *req = sc->offchannel.scan_req;
1029 	int i;
1030 
1031 	if (!(sc->cur_chan->chandef.chan->flags & IEEE80211_CHAN_NO_IR) &&
1032 	    req->n_ssids) {
1033 		for (i = 0; i < req->n_ssids; i++)
1034 			ath_scan_send_probe(sc, &req->ssids[i]);
1035 
1036 	}
1037 
1038 	ath_dbg(common, CHAN_CTX,
1039 		"Moving offchannel state to ATH_OFFCHANNEL_PROBE_WAIT\n");
1040 
1041 	sc->offchannel.state = ATH_OFFCHANNEL_PROBE_WAIT;
1042 	mod_timer(&sc->offchannel.timer, jiffies + sc->offchannel.duration);
1043 }
1044 
1045 static void ath_chanctx_timer(unsigned long data)
1046 {
1047 	struct ath_softc *sc = (struct ath_softc *) data;
1048 	struct ath_common *common = ath9k_hw_common(sc->sc_ah);
1049 
1050 	ath_dbg(common, CHAN_CTX,
1051 		"Channel context timer invoked\n");
1052 
1053 	ath_chanctx_event(sc, NULL, ATH_CHANCTX_EVENT_TSF_TIMER);
1054 }
1055 
1056 static void ath_offchannel_timer(unsigned long data)
1057 {
1058 	struct ath_softc *sc = (struct ath_softc *)data;
1059 	struct ath_chanctx *ctx;
1060 	struct ath_common *common = ath9k_hw_common(sc->sc_ah);
1061 
1062 	ath_dbg(common, CHAN_CTX, "%s: offchannel state: %s\n",
1063 		__func__, offchannel_state_string(sc->offchannel.state));
1064 
1065 	switch (sc->offchannel.state) {
1066 	case ATH_OFFCHANNEL_PROBE_WAIT:
1067 		if (!sc->offchannel.scan_req)
1068 			return;
1069 
1070 		/* get first active channel context */
1071 		ctx = ath_chanctx_get_oper_chan(sc, true);
1072 		if (ctx->active) {
1073 			ath_dbg(common, CHAN_CTX,
1074 				"Switch to oper/active context, "
1075 				"move offchannel state to ATH_OFFCHANNEL_SUSPEND\n");
1076 
1077 			sc->offchannel.state = ATH_OFFCHANNEL_SUSPEND;
1078 			ath_chanctx_switch(sc, ctx, NULL);
1079 			mod_timer(&sc->offchannel.timer, jiffies + HZ / 10);
1080 			break;
1081 		}
1082 		/* fall through */
1083 	case ATH_OFFCHANNEL_SUSPEND:
1084 		if (!sc->offchannel.scan_req)
1085 			return;
1086 
1087 		ath_scan_next_channel(sc);
1088 		break;
1089 	case ATH_OFFCHANNEL_ROC_START:
1090 	case ATH_OFFCHANNEL_ROC_WAIT:
1091 		sc->offchannel.state = ATH_OFFCHANNEL_ROC_DONE;
1092 		ath_roc_complete(sc, ATH_ROC_COMPLETE_EXPIRE);
1093 		break;
1094 	default:
1095 		break;
1096 	}
1097 }
1098 
1099 static bool
1100 ath_chanctx_send_vif_ps_frame(struct ath_softc *sc, struct ath_vif *avp,
1101 			      bool powersave)
1102 {
1103 	struct ieee80211_vif *vif = avp->vif;
1104 	struct ieee80211_sta *sta = NULL;
1105 	struct ieee80211_hdr_3addr *nullfunc;
1106 	struct ath_tx_control txctl;
1107 	struct sk_buff *skb;
1108 	int band = sc->cur_chan->chandef.chan->band;
1109 
1110 	switch (vif->type) {
1111 	case NL80211_IFTYPE_STATION:
1112 		if (!avp->assoc)
1113 			return false;
1114 
1115 		skb = ieee80211_nullfunc_get(sc->hw, vif);
1116 		if (!skb)
1117 			return false;
1118 
1119 		nullfunc = (struct ieee80211_hdr_3addr *) skb->data;
1120 		if (powersave)
1121 			nullfunc->frame_control |=
1122 				cpu_to_le16(IEEE80211_FCTL_PM);
1123 
1124 		skb->priority = 7;
1125 		skb_set_queue_mapping(skb, IEEE80211_AC_VO);
1126 		if (!ieee80211_tx_prepare_skb(sc->hw, vif, skb, band, &sta)) {
1127 			dev_kfree_skb_any(skb);
1128 			return false;
1129 		}
1130 		break;
1131 	default:
1132 		return false;
1133 	}
1134 
1135 	memset(&txctl, 0, sizeof(txctl));
1136 	txctl.txq = sc->tx.txq_map[IEEE80211_AC_VO];
1137 	txctl.sta = sta;
1138 	if (ath_tx_start(sc->hw, skb, &txctl)) {
1139 		ieee80211_free_txskb(sc->hw, skb);
1140 		return false;
1141 	}
1142 
1143 	return true;
1144 }
1145 
1146 static bool
1147 ath_chanctx_send_ps_frame(struct ath_softc *sc, bool powersave)
1148 {
1149 	struct ath_vif *avp;
1150 	bool sent = false;
1151 
1152 	rcu_read_lock();
1153 	list_for_each_entry(avp, &sc->cur_chan->vifs, list) {
1154 		if (ath_chanctx_send_vif_ps_frame(sc, avp, powersave))
1155 			sent = true;
1156 	}
1157 	rcu_read_unlock();
1158 
1159 	return sent;
1160 }
1161 
1162 static bool ath_chanctx_defer_switch(struct ath_softc *sc)
1163 {
1164 	struct ath_common *common = ath9k_hw_common(sc->sc_ah);
1165 
1166 	if (sc->cur_chan == &sc->offchannel.chan)
1167 		return false;
1168 
1169 	switch (sc->sched.state) {
1170 	case ATH_CHANCTX_STATE_SWITCH:
1171 		return false;
1172 	case ATH_CHANCTX_STATE_IDLE:
1173 		if (!sc->cur_chan->switch_after_beacon)
1174 			return false;
1175 
1176 		ath_dbg(common, CHAN_CTX,
1177 			"Defer switch, set chanctx state to WAIT_FOR_BEACON\n");
1178 
1179 		sc->sched.state = ATH_CHANCTX_STATE_WAIT_FOR_BEACON;
1180 		break;
1181 	default:
1182 		break;
1183 	}
1184 
1185 	return true;
1186 }
1187 
1188 static void ath_offchannel_channel_change(struct ath_softc *sc)
1189 {
1190 	struct ath_common *common = ath9k_hw_common(sc->sc_ah);
1191 
1192 	ath_dbg(common, CHAN_CTX, "%s: offchannel state: %s\n",
1193 		__func__, offchannel_state_string(sc->offchannel.state));
1194 
1195 	switch (sc->offchannel.state) {
1196 	case ATH_OFFCHANNEL_PROBE_SEND:
1197 		if (!sc->offchannel.scan_req)
1198 			return;
1199 
1200 		if (sc->cur_chan->chandef.chan !=
1201 		    sc->offchannel.chan.chandef.chan)
1202 			return;
1203 
1204 		ath_scan_channel_start(sc);
1205 		break;
1206 	case ATH_OFFCHANNEL_IDLE:
1207 		if (!sc->offchannel.scan_req)
1208 			return;
1209 
1210 		ath_scan_complete(sc, false);
1211 		break;
1212 	case ATH_OFFCHANNEL_ROC_START:
1213 		if (sc->cur_chan != &sc->offchannel.chan)
1214 			break;
1215 
1216 		sc->offchannel.state = ATH_OFFCHANNEL_ROC_WAIT;
1217 		mod_timer(&sc->offchannel.timer,
1218 			  jiffies + sc->offchannel.duration);
1219 		ieee80211_ready_on_channel(sc->hw);
1220 		break;
1221 	case ATH_OFFCHANNEL_ROC_DONE:
1222 		break;
1223 	default:
1224 		break;
1225 	}
1226 }
1227 
1228 void ath_chanctx_set_next(struct ath_softc *sc, bool force)
1229 {
1230 	struct ath_common *common = ath9k_hw_common(sc->sc_ah);
1231 	struct ath_chanctx *old_ctx;
1232 	struct timespec ts;
1233 	bool measure_time = false;
1234 	bool send_ps = false;
1235 	bool queues_stopped = false;
1236 
1237 	spin_lock_bh(&sc->chan_lock);
1238 	if (!sc->next_chan) {
1239 		spin_unlock_bh(&sc->chan_lock);
1240 		return;
1241 	}
1242 
1243 	if (!force && ath_chanctx_defer_switch(sc)) {
1244 		spin_unlock_bh(&sc->chan_lock);
1245 		return;
1246 	}
1247 
1248 	ath_dbg(common, CHAN_CTX,
1249 		"%s: current: %d MHz, next: %d MHz\n",
1250 		__func__,
1251 		sc->cur_chan->chandef.center_freq1,
1252 		sc->next_chan->chandef.center_freq1);
1253 
1254 	if (sc->cur_chan != sc->next_chan) {
1255 		ath_dbg(common, CHAN_CTX,
1256 			"Stopping current chanctx: %d\n",
1257 			sc->cur_chan->chandef.center_freq1);
1258 		sc->cur_chan->stopped = true;
1259 		spin_unlock_bh(&sc->chan_lock);
1260 
1261 		if (sc->next_chan == &sc->offchannel.chan) {
1262 			getrawmonotonic(&ts);
1263 			measure_time = true;
1264 		}
1265 
1266 		ath9k_chanctx_stop_queues(sc, sc->cur_chan);
1267 		queues_stopped = true;
1268 
1269 		__ath9k_flush(sc->hw, ~0, true, false, false);
1270 
1271 		if (ath_chanctx_send_ps_frame(sc, true))
1272 			__ath9k_flush(sc->hw, BIT(IEEE80211_AC_VO),
1273 				      false, false, false);
1274 
1275 		send_ps = true;
1276 		spin_lock_bh(&sc->chan_lock);
1277 
1278 		if (sc->cur_chan != &sc->offchannel.chan) {
1279 			getrawmonotonic(&sc->cur_chan->tsf_ts);
1280 			sc->cur_chan->tsf_val = ath9k_hw_gettsf64(sc->sc_ah);
1281 		}
1282 	}
1283 	old_ctx = sc->cur_chan;
1284 	sc->cur_chan = sc->next_chan;
1285 	sc->cur_chan->stopped = false;
1286 	sc->next_chan = NULL;
1287 
1288 	if (!sc->sched.offchannel_pending)
1289 		sc->sched.offchannel_duration = 0;
1290 
1291 	if (sc->sched.state != ATH_CHANCTX_STATE_FORCE_ACTIVE)
1292 		sc->sched.state = ATH_CHANCTX_STATE_IDLE;
1293 
1294 	spin_unlock_bh(&sc->chan_lock);
1295 
1296 	if (sc->sc_ah->chip_fullsleep ||
1297 	    memcmp(&sc->cur_chandef, &sc->cur_chan->chandef,
1298 		   sizeof(sc->cur_chandef))) {
1299 		ath_dbg(common, CHAN_CTX,
1300 			"%s: Set channel %d MHz\n",
1301 			__func__, sc->cur_chan->chandef.center_freq1);
1302 		ath_set_channel(sc);
1303 		if (measure_time)
1304 			sc->sched.channel_switch_time =
1305 				ath9k_hw_get_tsf_offset(&ts, NULL);
1306 		/*
1307 		 * A reset will ensure that all queues are woken up,
1308 		 * so there is no need to awaken them again.
1309 		 */
1310 		goto out;
1311 	}
1312 
1313 	if (queues_stopped)
1314 		ath9k_chanctx_wake_queues(sc, old_ctx);
1315 out:
1316 	if (send_ps)
1317 		ath_chanctx_send_ps_frame(sc, false);
1318 
1319 	ath_offchannel_channel_change(sc);
1320 	ath_chanctx_event(sc, NULL, ATH_CHANCTX_EVENT_SWITCH);
1321 }
1322 
1323 static void ath_chanctx_work(struct work_struct *work)
1324 {
1325 	struct ath_softc *sc = container_of(work, struct ath_softc,
1326 					    chanctx_work);
1327 	mutex_lock(&sc->mutex);
1328 	ath_chanctx_set_next(sc, false);
1329 	mutex_unlock(&sc->mutex);
1330 }
1331 
1332 void ath9k_offchannel_init(struct ath_softc *sc)
1333 {
1334 	struct ath_chanctx *ctx;
1335 	struct ath_common *common = ath9k_hw_common(sc->sc_ah);
1336 	struct ieee80211_supported_band *sband;
1337 	struct ieee80211_channel *chan;
1338 	int i;
1339 
1340 	sband = &common->sbands[NL80211_BAND_2GHZ];
1341 	if (!sband->n_channels)
1342 		sband = &common->sbands[NL80211_BAND_5GHZ];
1343 
1344 	chan = &sband->channels[0];
1345 
1346 	ctx = &sc->offchannel.chan;
1347 	INIT_LIST_HEAD(&ctx->vifs);
1348 	ctx->txpower = ATH_TXPOWER_MAX;
1349 	cfg80211_chandef_create(&ctx->chandef, chan, NL80211_CHAN_HT20);
1350 
1351 	for (i = 0; i < ARRAY_SIZE(ctx->acq); i++) {
1352 		INIT_LIST_HEAD(&ctx->acq[i].acq_new);
1353 		INIT_LIST_HEAD(&ctx->acq[i].acq_old);
1354 		spin_lock_init(&ctx->acq[i].lock);
1355 	}
1356 
1357 	sc->offchannel.chan.offchannel = true;
1358 }
1359 
1360 void ath9k_init_channel_context(struct ath_softc *sc)
1361 {
1362 	INIT_WORK(&sc->chanctx_work, ath_chanctx_work);
1363 
1364 	setup_timer(&sc->offchannel.timer, ath_offchannel_timer,
1365 		    (unsigned long)sc);
1366 	setup_timer(&sc->sched.timer, ath_chanctx_timer,
1367 		    (unsigned long)sc);
1368 
1369 	init_completion(&sc->go_beacon);
1370 }
1371 
1372 void ath9k_deinit_channel_context(struct ath_softc *sc)
1373 {
1374 	cancel_work_sync(&sc->chanctx_work);
1375 }
1376 
1377 bool ath9k_is_chanctx_enabled(void)
1378 {
1379 	return (ath9k_use_chanctx == 1);
1380 }
1381 
1382 /********************/
1383 /* Queue management */
1384 /********************/
1385 
1386 void ath9k_chanctx_stop_queues(struct ath_softc *sc, struct ath_chanctx *ctx)
1387 {
1388 	struct ath_hw *ah = sc->sc_ah;
1389 	int i;
1390 
1391 	if (ctx == &sc->offchannel.chan) {
1392 		ieee80211_stop_queue(sc->hw,
1393 				     sc->hw->offchannel_tx_hw_queue);
1394 	} else {
1395 		for (i = 0; i < IEEE80211_NUM_ACS; i++)
1396 			ieee80211_stop_queue(sc->hw,
1397 					     ctx->hw_queue_base + i);
1398 	}
1399 
1400 	if (ah->opmode == NL80211_IFTYPE_AP)
1401 		ieee80211_stop_queue(sc->hw, sc->hw->queues - 2);
1402 }
1403 
1404 
1405 void ath9k_chanctx_wake_queues(struct ath_softc *sc, struct ath_chanctx *ctx)
1406 {
1407 	struct ath_hw *ah = sc->sc_ah;
1408 	int i;
1409 
1410 	if (ctx == &sc->offchannel.chan) {
1411 		ieee80211_wake_queue(sc->hw,
1412 				     sc->hw->offchannel_tx_hw_queue);
1413 	} else {
1414 		for (i = 0; i < IEEE80211_NUM_ACS; i++)
1415 			ieee80211_wake_queue(sc->hw,
1416 					     ctx->hw_queue_base + i);
1417 	}
1418 
1419 	if (ah->opmode == NL80211_IFTYPE_AP)
1420 		ieee80211_wake_queue(sc->hw, sc->hw->queues - 2);
1421 }
1422 
1423 /*****************/
1424 /* P2P Powersave */
1425 /*****************/
1426 
1427 static void ath9k_update_p2p_ps_timer(struct ath_softc *sc, struct ath_vif *avp)
1428 {
1429 	struct ath_common *common = ath9k_hw_common(sc->sc_ah);
1430 	struct ath_hw *ah = sc->sc_ah;
1431 	u32 tsf, target_tsf;
1432 
1433 	if (!avp || !avp->noa.has_next_tsf)
1434 		return;
1435 
1436 	ath9k_hw_gen_timer_stop(ah, sc->p2p_ps_timer);
1437 
1438 	tsf = ath9k_hw_gettsf32(sc->sc_ah);
1439 
1440 	target_tsf = avp->noa.next_tsf;
1441 	if (!avp->noa.absent)
1442 		target_tsf -= ATH_P2P_PS_STOP_TIME;
1443 	else
1444 		target_tsf += ATH_P2P_PS_STOP_TIME;
1445 
1446 	if (target_tsf - tsf < ATH_P2P_PS_STOP_TIME)
1447 		target_tsf = tsf + ATH_P2P_PS_STOP_TIME;
1448 
1449 	ath_dbg(common, CHAN_CTX, "%s absent %d tsf 0x%08X next_tsf 0x%08X (%dms)\n",
1450 		__func__, avp->noa.absent, tsf, target_tsf,
1451 		(target_tsf - tsf) / 1000);
1452 
1453 	ath9k_hw_gen_timer_start(ah, sc->p2p_ps_timer, target_tsf, 1000000);
1454 }
1455 
1456 static void ath9k_update_p2p_ps(struct ath_softc *sc, struct ieee80211_vif *vif)
1457 {
1458 	struct ath_vif *avp = (void *)vif->drv_priv;
1459 	u32 tsf;
1460 
1461 	if (!sc->p2p_ps_timer)
1462 		return;
1463 
1464 	if (vif->type != NL80211_IFTYPE_STATION)
1465 		return;
1466 
1467 	sc->p2p_ps_vif = avp;
1468 
1469 	if (sc->ps_flags & PS_BEACON_SYNC)
1470 		return;
1471 
1472 	tsf = ath9k_hw_gettsf32(sc->sc_ah);
1473 	ieee80211_parse_p2p_noa(&vif->bss_conf.p2p_noa_attr, &avp->noa, tsf);
1474 	ath9k_update_p2p_ps_timer(sc, avp);
1475 }
1476 
1477 static u8 ath9k_get_ctwin(struct ath_softc *sc, struct ath_vif *avp)
1478 {
1479 	struct ath_beacon_config *cur_conf = &sc->cur_chan->beacon;
1480 	u8 switch_time, ctwin;
1481 
1482 	/*
1483 	 * Channel switch in multi-channel mode is deferred
1484 	 * by a quarter beacon interval when handling
1485 	 * ATH_CHANCTX_EVENT_BEACON_PREPARE, so the P2P-GO
1486 	 * interface is guaranteed to be discoverable
1487 	 * for that duration after a TBTT.
1488 	 */
1489 	switch_time = cur_conf->beacon_interval / 4;
1490 
1491 	ctwin = avp->vif->bss_conf.p2p_noa_attr.oppps_ctwindow;
1492 	if (ctwin && (ctwin < switch_time))
1493 		return ctwin;
1494 
1495 	if (switch_time < P2P_DEFAULT_CTWIN)
1496 		return 0;
1497 
1498 	return P2P_DEFAULT_CTWIN;
1499 }
1500 
1501 void ath9k_beacon_add_noa(struct ath_softc *sc, struct ath_vif *avp,
1502 			  struct sk_buff *skb)
1503 {
1504 	static const u8 noa_ie_hdr[] = {
1505 		WLAN_EID_VENDOR_SPECIFIC,	/* type */
1506 		0,				/* length */
1507 		0x50, 0x6f, 0x9a,		/* WFA OUI */
1508 		0x09,				/* P2P subtype */
1509 		0x0c,				/* Notice of Absence */
1510 		0x00,				/* LSB of little-endian len */
1511 		0x00,				/* MSB of little-endian len */
1512 	};
1513 
1514 	struct ieee80211_p2p_noa_attr *noa;
1515 	int noa_len, noa_desc, i = 0;
1516 	u8 *hdr;
1517 
1518 	if (!avp->offchannel_duration && !avp->noa_duration)
1519 		return;
1520 
1521 	noa_desc = !!avp->offchannel_duration + !!avp->noa_duration;
1522 	noa_len = 2 + sizeof(struct ieee80211_p2p_noa_desc) * noa_desc;
1523 
1524 	hdr = skb_put_data(skb, noa_ie_hdr, sizeof(noa_ie_hdr));
1525 	hdr[1] = sizeof(noa_ie_hdr) + noa_len - 2;
1526 	hdr[7] = noa_len;
1527 
1528 	noa = skb_put_zero(skb, noa_len);
1529 
1530 	noa->index = avp->noa_index;
1531 	noa->oppps_ctwindow = ath9k_get_ctwin(sc, avp);
1532 	if (noa->oppps_ctwindow)
1533 		noa->oppps_ctwindow |= BIT(7);
1534 
1535 	if (avp->noa_duration) {
1536 		if (avp->periodic_noa) {
1537 			u32 interval = TU_TO_USEC(sc->cur_chan->beacon.beacon_interval);
1538 			noa->desc[i].count = 255;
1539 			noa->desc[i].interval = cpu_to_le32(interval);
1540 		} else {
1541 			noa->desc[i].count = 1;
1542 		}
1543 
1544 		noa->desc[i].start_time = cpu_to_le32(avp->noa_start);
1545 		noa->desc[i].duration = cpu_to_le32(avp->noa_duration);
1546 		i++;
1547 	}
1548 
1549 	if (avp->offchannel_duration) {
1550 		noa->desc[i].count = 1;
1551 		noa->desc[i].start_time = cpu_to_le32(avp->offchannel_start);
1552 		noa->desc[i].duration = cpu_to_le32(avp->offchannel_duration);
1553 	}
1554 }
1555 
1556 void ath9k_p2p_ps_timer(void *priv)
1557 {
1558 	struct ath_softc *sc = priv;
1559 	struct ath_vif *avp = sc->p2p_ps_vif;
1560 	struct ieee80211_vif *vif;
1561 	struct ieee80211_sta *sta;
1562 	struct ath_node *an;
1563 	u32 tsf;
1564 
1565 	del_timer_sync(&sc->sched.timer);
1566 	ath9k_hw_gen_timer_stop(sc->sc_ah, sc->p2p_ps_timer);
1567 	ath_chanctx_event(sc, NULL, ATH_CHANCTX_EVENT_TSF_TIMER);
1568 
1569 	if (!avp || avp->chanctx != sc->cur_chan)
1570 		return;
1571 
1572 	tsf = ath9k_hw_gettsf32(sc->sc_ah);
1573 	if (!avp->noa.absent)
1574 		tsf += ATH_P2P_PS_STOP_TIME;
1575 	else
1576 		tsf -= ATH_P2P_PS_STOP_TIME;
1577 
1578 	if (!avp->noa.has_next_tsf ||
1579 	    avp->noa.next_tsf - tsf > BIT(31))
1580 		ieee80211_update_p2p_noa(&avp->noa, tsf);
1581 
1582 	ath9k_update_p2p_ps_timer(sc, avp);
1583 
1584 	rcu_read_lock();
1585 
1586 	vif = avp->vif;
1587 	sta = ieee80211_find_sta(vif, avp->bssid);
1588 	if (!sta)
1589 		goto out;
1590 
1591 	an = (void *) sta->drv_priv;
1592 	if (an->sleeping == !!avp->noa.absent)
1593 		goto out;
1594 
1595 	an->sleeping = avp->noa.absent;
1596 	if (an->sleeping)
1597 		ath_tx_aggr_sleep(sta, sc, an);
1598 	else
1599 		ath_tx_aggr_wakeup(sc, an);
1600 
1601 out:
1602 	rcu_read_unlock();
1603 }
1604 
1605 void ath9k_p2p_bss_info_changed(struct ath_softc *sc,
1606 				struct ieee80211_vif *vif)
1607 {
1608 	unsigned long flags;
1609 
1610 	spin_lock_bh(&sc->sc_pcu_lock);
1611 	spin_lock_irqsave(&sc->sc_pm_lock, flags);
1612 	ath9k_update_p2p_ps(sc, vif);
1613 	spin_unlock_irqrestore(&sc->sc_pm_lock, flags);
1614 	spin_unlock_bh(&sc->sc_pcu_lock);
1615 }
1616 
1617 void ath9k_p2p_beacon_sync(struct ath_softc *sc)
1618 {
1619 	if (sc->p2p_ps_vif)
1620 		ath9k_update_p2p_ps(sc, sc->p2p_ps_vif->vif);
1621 }
1622 
1623 void ath9k_p2p_remove_vif(struct ath_softc *sc,
1624 			  struct ieee80211_vif *vif)
1625 {
1626 	struct ath_vif *avp = (void *)vif->drv_priv;
1627 
1628 	spin_lock_bh(&sc->sc_pcu_lock);
1629 	if (avp == sc->p2p_ps_vif) {
1630 		sc->p2p_ps_vif = NULL;
1631 		ath9k_update_p2p_ps_timer(sc, NULL);
1632 	}
1633 	spin_unlock_bh(&sc->sc_pcu_lock);
1634 }
1635 
1636 int ath9k_init_p2p(struct ath_softc *sc)
1637 {
1638 	sc->p2p_ps_timer = ath_gen_timer_alloc(sc->sc_ah, ath9k_p2p_ps_timer,
1639 					       NULL, sc, AR_FIRST_NDP_TIMER);
1640 	if (!sc->p2p_ps_timer)
1641 		return -ENOMEM;
1642 
1643 	return 0;
1644 }
1645 
1646 void ath9k_deinit_p2p(struct ath_softc *sc)
1647 {
1648 	if (sc->p2p_ps_timer)
1649 		ath_gen_timer_free(sc->sc_ah, sc->p2p_ps_timer);
1650 }
1651 
1652 #endif /* CONFIG_ATH9K_CHANNEL_CONTEXT */
1653