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