xref: /openbmc/linux/drivers/net/wireless/ath/ath9k/main.c (revision 2a215410)
1 /*
2  * Copyright (c) 2008-2011 Atheros Communications 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 <linux/nl80211.h>
18 #include <linux/delay.h>
19 #include "ath9k.h"
20 #include "btcoex.h"
21 
22 static void ath9k_flush(struct ieee80211_hw *hw, struct ieee80211_vif *vif,
23 			u32 queues, bool drop);
24 
25 u8 ath9k_parse_mpdudensity(u8 mpdudensity)
26 {
27 	/*
28 	 * 802.11n D2.0 defined values for "Minimum MPDU Start Spacing":
29 	 *   0 for no restriction
30 	 *   1 for 1/4 us
31 	 *   2 for 1/2 us
32 	 *   3 for 1 us
33 	 *   4 for 2 us
34 	 *   5 for 4 us
35 	 *   6 for 8 us
36 	 *   7 for 16 us
37 	 */
38 	switch (mpdudensity) {
39 	case 0:
40 		return 0;
41 	case 1:
42 	case 2:
43 	case 3:
44 		/* Our lower layer calculations limit our precision to
45 		   1 microsecond */
46 		return 1;
47 	case 4:
48 		return 2;
49 	case 5:
50 		return 4;
51 	case 6:
52 		return 8;
53 	case 7:
54 		return 16;
55 	default:
56 		return 0;
57 	}
58 }
59 
60 static bool ath9k_has_pending_frames(struct ath_softc *sc, struct ath_txq *txq,
61 				     bool sw_pending)
62 {
63 	bool pending = false;
64 
65 	spin_lock_bh(&txq->axq_lock);
66 
67 	if (txq->axq_depth) {
68 		pending = true;
69 		goto out;
70 	}
71 
72 	if (!sw_pending)
73 		goto out;
74 
75 	if (txq->mac80211_qnum >= 0) {
76 		struct ath_acq *acq;
77 
78 		acq = &sc->cur_chan->acq[txq->mac80211_qnum];
79 		if (!list_empty(&acq->acq_new) || !list_empty(&acq->acq_old))
80 			pending = true;
81 	}
82 out:
83 	spin_unlock_bh(&txq->axq_lock);
84 	return pending;
85 }
86 
87 static bool ath9k_setpower(struct ath_softc *sc, enum ath9k_power_mode mode)
88 {
89 	unsigned long flags;
90 	bool ret;
91 
92 	spin_lock_irqsave(&sc->sc_pm_lock, flags);
93 	ret = ath9k_hw_setpower(sc->sc_ah, mode);
94 	spin_unlock_irqrestore(&sc->sc_pm_lock, flags);
95 
96 	return ret;
97 }
98 
99 void ath_ps_full_sleep(struct timer_list *t)
100 {
101 	struct ath_softc *sc = from_timer(sc, t, sleep_timer);
102 	struct ath_common *common = ath9k_hw_common(sc->sc_ah);
103 	unsigned long flags;
104 	bool reset;
105 
106 	spin_lock_irqsave(&common->cc_lock, flags);
107 	ath_hw_cycle_counters_update(common);
108 	spin_unlock_irqrestore(&common->cc_lock, flags);
109 
110 	ath9k_hw_setrxabort(sc->sc_ah, 1);
111 	ath9k_hw_stopdmarecv(sc->sc_ah, &reset);
112 
113 	ath9k_hw_setpower(sc->sc_ah, ATH9K_PM_FULL_SLEEP);
114 }
115 
116 void ath9k_ps_wakeup(struct ath_softc *sc)
117 {
118 	struct ath_common *common = ath9k_hw_common(sc->sc_ah);
119 	unsigned long flags;
120 	enum ath9k_power_mode power_mode;
121 
122 	spin_lock_irqsave(&sc->sc_pm_lock, flags);
123 	if (++sc->ps_usecount != 1)
124 		goto unlock;
125 
126 	del_timer_sync(&sc->sleep_timer);
127 	power_mode = sc->sc_ah->power_mode;
128 	ath9k_hw_setpower(sc->sc_ah, ATH9K_PM_AWAKE);
129 
130 	/*
131 	 * While the hardware is asleep, the cycle counters contain no
132 	 * useful data. Better clear them now so that they don't mess up
133 	 * survey data results.
134 	 */
135 	if (power_mode != ATH9K_PM_AWAKE) {
136 		spin_lock(&common->cc_lock);
137 		ath_hw_cycle_counters_update(common);
138 		memset(&common->cc, 0, sizeof(common->cc));
139 		spin_unlock(&common->cc_lock);
140 	}
141 
142  unlock:
143 	spin_unlock_irqrestore(&sc->sc_pm_lock, flags);
144 }
145 
146 void ath9k_ps_restore(struct ath_softc *sc)
147 {
148 	struct ath_common *common = ath9k_hw_common(sc->sc_ah);
149 	enum ath9k_power_mode mode;
150 	unsigned long flags;
151 
152 	spin_lock_irqsave(&sc->sc_pm_lock, flags);
153 	if (--sc->ps_usecount != 0)
154 		goto unlock;
155 
156 	if (sc->ps_idle) {
157 		mod_timer(&sc->sleep_timer, jiffies + HZ / 10);
158 		goto unlock;
159 	}
160 
161 	if (sc->ps_enabled &&
162 		   !(sc->ps_flags & (PS_WAIT_FOR_BEACON |
163 				     PS_WAIT_FOR_CAB |
164 				     PS_WAIT_FOR_PSPOLL_DATA |
165 				     PS_WAIT_FOR_TX_ACK |
166 				     PS_WAIT_FOR_ANI))) {
167 		mode = ATH9K_PM_NETWORK_SLEEP;
168 		if (ath9k_hw_btcoex_is_enabled(sc->sc_ah))
169 			ath9k_btcoex_stop_gen_timer(sc);
170 	} else {
171 		goto unlock;
172 	}
173 
174 	spin_lock(&common->cc_lock);
175 	ath_hw_cycle_counters_update(common);
176 	spin_unlock(&common->cc_lock);
177 
178 	ath9k_hw_setpower(sc->sc_ah, mode);
179 
180  unlock:
181 	spin_unlock_irqrestore(&sc->sc_pm_lock, flags);
182 }
183 
184 static void __ath_cancel_work(struct ath_softc *sc)
185 {
186 	cancel_work_sync(&sc->paprd_work);
187 	cancel_delayed_work_sync(&sc->hw_check_work);
188 	cancel_delayed_work_sync(&sc->hw_pll_work);
189 
190 #ifdef CONFIG_ATH9K_BTCOEX_SUPPORT
191 	if (ath9k_hw_mci_is_enabled(sc->sc_ah))
192 		cancel_work_sync(&sc->mci_work);
193 #endif
194 }
195 
196 void ath_cancel_work(struct ath_softc *sc)
197 {
198 	__ath_cancel_work(sc);
199 	cancel_work_sync(&sc->hw_reset_work);
200 }
201 
202 void ath_restart_work(struct ath_softc *sc)
203 {
204 	ieee80211_queue_delayed_work(sc->hw, &sc->hw_check_work,
205 				     msecs_to_jiffies(ATH_HW_CHECK_POLL_INT));
206 
207 	if (AR_SREV_9340(sc->sc_ah) || AR_SREV_9330(sc->sc_ah))
208 		ieee80211_queue_delayed_work(sc->hw, &sc->hw_pll_work,
209 				     msecs_to_jiffies(ATH_PLL_WORK_INTERVAL));
210 
211 	ath_start_ani(sc);
212 }
213 
214 static bool ath_prepare_reset(struct ath_softc *sc)
215 {
216 	struct ath_hw *ah = sc->sc_ah;
217 	bool ret = true;
218 
219 	ieee80211_stop_queues(sc->hw);
220 	ath_stop_ani(sc);
221 	ath9k_hw_disable_interrupts(ah);
222 
223 	if (AR_SREV_9300_20_OR_LATER(ah)) {
224 		ret &= ath_stoprecv(sc);
225 		ret &= ath_drain_all_txq(sc);
226 	} else {
227 		ret &= ath_drain_all_txq(sc);
228 		ret &= ath_stoprecv(sc);
229 	}
230 
231 	return ret;
232 }
233 
234 static bool ath_complete_reset(struct ath_softc *sc, bool start)
235 {
236 	struct ath_hw *ah = sc->sc_ah;
237 	struct ath_common *common = ath9k_hw_common(ah);
238 	unsigned long flags;
239 
240 	ath9k_calculate_summary_state(sc, sc->cur_chan);
241 	ath_startrecv(sc);
242 	ath9k_cmn_update_txpow(ah, sc->cur_chan->cur_txpower,
243 			       sc->cur_chan->txpower,
244 			       &sc->cur_chan->cur_txpower);
245 	clear_bit(ATH_OP_HW_RESET, &common->op_flags);
246 
247 	if (!sc->cur_chan->offchannel && start) {
248 		/* restore per chanctx TSF timer */
249 		if (sc->cur_chan->tsf_val) {
250 			u32 offset;
251 
252 			offset = ath9k_hw_get_tsf_offset(&sc->cur_chan->tsf_ts,
253 							 NULL);
254 			ath9k_hw_settsf64(ah, sc->cur_chan->tsf_val + offset);
255 		}
256 
257 
258 		if (!test_bit(ATH_OP_BEACONS, &common->op_flags))
259 			goto work;
260 
261 		if (ah->opmode == NL80211_IFTYPE_STATION &&
262 		    test_bit(ATH_OP_PRIM_STA_VIF, &common->op_flags)) {
263 			spin_lock_irqsave(&sc->sc_pm_lock, flags);
264 			sc->ps_flags |= PS_BEACON_SYNC | PS_WAIT_FOR_BEACON;
265 			spin_unlock_irqrestore(&sc->sc_pm_lock, flags);
266 		} else {
267 			ath9k_set_beacon(sc);
268 		}
269 	work:
270 		ath_restart_work(sc);
271 		ath_txq_schedule_all(sc);
272 	}
273 
274 	sc->gtt_cnt = 0;
275 
276 	ath9k_hw_set_interrupts(ah);
277 	ath9k_hw_enable_interrupts(ah);
278 	ieee80211_wake_queues(sc->hw);
279 	ath9k_p2p_ps_timer(sc);
280 
281 	return true;
282 }
283 
284 static int ath_reset_internal(struct ath_softc *sc, struct ath9k_channel *hchan)
285 {
286 	struct ath_hw *ah = sc->sc_ah;
287 	struct ath_common *common = ath9k_hw_common(ah);
288 	struct ath9k_hw_cal_data *caldata = NULL;
289 	bool fastcc = true;
290 	int r;
291 
292 	__ath_cancel_work(sc);
293 
294 	disable_irq(sc->irq);
295 	tasklet_disable(&sc->intr_tq);
296 	tasklet_disable(&sc->bcon_tasklet);
297 	spin_lock_bh(&sc->sc_pcu_lock);
298 
299 	if (!sc->cur_chan->offchannel) {
300 		fastcc = false;
301 		caldata = &sc->cur_chan->caldata;
302 	}
303 
304 	if (!hchan) {
305 		fastcc = false;
306 		hchan = ah->curchan;
307 	}
308 
309 	if (!hchan) {
310 		fastcc = false;
311 		hchan = ath9k_cmn_get_channel(sc->hw, ah, &sc->cur_chan->chandef);
312 	}
313 
314 	if (!ath_prepare_reset(sc))
315 		fastcc = false;
316 
317 	if (ath9k_is_chanctx_enabled())
318 		fastcc = false;
319 
320 	spin_lock_bh(&sc->chan_lock);
321 	sc->cur_chandef = sc->cur_chan->chandef;
322 	spin_unlock_bh(&sc->chan_lock);
323 
324 	ath_dbg(common, CONFIG, "Reset to %u MHz, HT40: %d fastcc: %d\n",
325 		hchan->channel, IS_CHAN_HT40(hchan), fastcc);
326 
327 	r = ath9k_hw_reset(ah, hchan, caldata, fastcc);
328 	if (r) {
329 		ath_err(common,
330 			"Unable to reset channel, reset status %d\n", r);
331 
332 		ath9k_hw_enable_interrupts(ah);
333 		ath9k_queue_reset(sc, RESET_TYPE_BB_HANG);
334 
335 		goto out;
336 	}
337 
338 	if (ath9k_hw_mci_is_enabled(sc->sc_ah) &&
339 	    sc->cur_chan->offchannel)
340 		ath9k_mci_set_txpower(sc, true, false);
341 
342 	if (!ath_complete_reset(sc, true))
343 		r = -EIO;
344 
345 out:
346 	enable_irq(sc->irq);
347 	spin_unlock_bh(&sc->sc_pcu_lock);
348 	tasklet_enable(&sc->bcon_tasklet);
349 	tasklet_enable(&sc->intr_tq);
350 
351 	return r;
352 }
353 
354 static void ath_node_attach(struct ath_softc *sc, struct ieee80211_sta *sta,
355 			    struct ieee80211_vif *vif)
356 {
357 	struct ath_node *an;
358 	an = (struct ath_node *)sta->drv_priv;
359 
360 	an->sc = sc;
361 	an->sta = sta;
362 	an->vif = vif;
363 	memset(&an->key_idx, 0, sizeof(an->key_idx));
364 
365 	ath_tx_node_init(sc, an);
366 
367 	ath_dynack_node_init(sc->sc_ah, an);
368 }
369 
370 static void ath_node_detach(struct ath_softc *sc, struct ieee80211_sta *sta)
371 {
372 	struct ath_node *an = (struct ath_node *)sta->drv_priv;
373 	ath_tx_node_cleanup(sc, an);
374 
375 	ath_dynack_node_deinit(sc->sc_ah, an);
376 }
377 
378 void ath9k_tasklet(struct tasklet_struct *t)
379 {
380 	struct ath_softc *sc = from_tasklet(sc, t, intr_tq);
381 	struct ath_hw *ah = sc->sc_ah;
382 	struct ath_common *common = ath9k_hw_common(ah);
383 	enum ath_reset_type type;
384 	unsigned long flags;
385 	u32 status;
386 	u32 rxmask;
387 
388 	spin_lock_irqsave(&sc->intr_lock, flags);
389 	status = sc->intrstatus;
390 	sc->intrstatus = 0;
391 	spin_unlock_irqrestore(&sc->intr_lock, flags);
392 
393 	ath9k_ps_wakeup(sc);
394 	spin_lock(&sc->sc_pcu_lock);
395 
396 	if (status & ATH9K_INT_FATAL) {
397 		type = RESET_TYPE_FATAL_INT;
398 		ath9k_queue_reset(sc, type);
399 		ath_dbg(common, RESET, "FATAL: Skipping interrupts\n");
400 		goto out;
401 	}
402 
403 	if ((ah->config.hw_hang_checks & HW_BB_WATCHDOG) &&
404 	    (status & ATH9K_INT_BB_WATCHDOG)) {
405 		spin_lock_irqsave(&common->cc_lock, flags);
406 		ath_hw_cycle_counters_update(common);
407 		ar9003_hw_bb_watchdog_dbg_info(ah);
408 		spin_unlock_irqrestore(&common->cc_lock, flags);
409 
410 		if (ar9003_hw_bb_watchdog_check(ah)) {
411 			type = RESET_TYPE_BB_WATCHDOG;
412 			ath9k_queue_reset(sc, type);
413 
414 			ath_dbg(common, RESET,
415 				"BB_WATCHDOG: Skipping interrupts\n");
416 			goto out;
417 		}
418 	}
419 
420 	if (status & ATH9K_INT_GTT) {
421 		sc->gtt_cnt++;
422 
423 		if ((sc->gtt_cnt >= MAX_GTT_CNT) && !ath9k_hw_check_alive(ah)) {
424 			type = RESET_TYPE_TX_GTT;
425 			ath9k_queue_reset(sc, type);
426 			ath_dbg(common, RESET,
427 				"GTT: Skipping interrupts\n");
428 			goto out;
429 		}
430 	}
431 
432 	spin_lock_irqsave(&sc->sc_pm_lock, flags);
433 	if ((status & ATH9K_INT_TSFOOR) && sc->ps_enabled) {
434 		/*
435 		 * TSF sync does not look correct; remain awake to sync with
436 		 * the next Beacon.
437 		 */
438 		ath_dbg(common, PS, "TSFOOR - Sync with next Beacon\n");
439 		sc->ps_flags |= PS_WAIT_FOR_BEACON | PS_BEACON_SYNC;
440 	}
441 	spin_unlock_irqrestore(&sc->sc_pm_lock, flags);
442 
443 	if (ah->caps.hw_caps & ATH9K_HW_CAP_EDMA)
444 		rxmask = (ATH9K_INT_RXHP | ATH9K_INT_RXLP | ATH9K_INT_RXEOL |
445 			  ATH9K_INT_RXORN);
446 	else
447 		rxmask = (ATH9K_INT_RX | ATH9K_INT_RXEOL | ATH9K_INT_RXORN);
448 
449 	if (status & rxmask) {
450 		/* Check for high priority Rx first */
451 		if ((ah->caps.hw_caps & ATH9K_HW_CAP_EDMA) &&
452 		    (status & ATH9K_INT_RXHP))
453 			ath_rx_tasklet(sc, 0, true);
454 
455 		ath_rx_tasklet(sc, 0, false);
456 	}
457 
458 	if (status & ATH9K_INT_TX) {
459 		if (ah->caps.hw_caps & ATH9K_HW_CAP_EDMA) {
460 			/*
461 			 * For EDMA chips, TX completion is enabled for the
462 			 * beacon queue, so if a beacon has been transmitted
463 			 * successfully after a GTT interrupt, the GTT counter
464 			 * gets reset to zero here.
465 			 */
466 			sc->gtt_cnt = 0;
467 
468 			ath_tx_edma_tasklet(sc);
469 		} else {
470 			ath_tx_tasklet(sc);
471 		}
472 
473 		wake_up(&sc->tx_wait);
474 	}
475 
476 	if (status & ATH9K_INT_GENTIMER)
477 		ath_gen_timer_isr(sc->sc_ah);
478 
479 	ath9k_btcoex_handle_interrupt(sc, status);
480 
481 	/* re-enable hardware interrupt */
482 	ath9k_hw_resume_interrupts(ah);
483 out:
484 	spin_unlock(&sc->sc_pcu_lock);
485 	ath9k_ps_restore(sc);
486 }
487 
488 irqreturn_t ath_isr(int irq, void *dev)
489 {
490 #define SCHED_INTR (				\
491 		ATH9K_INT_FATAL |		\
492 		ATH9K_INT_BB_WATCHDOG |		\
493 		ATH9K_INT_RXORN |		\
494 		ATH9K_INT_RXEOL |		\
495 		ATH9K_INT_RX |			\
496 		ATH9K_INT_RXLP |		\
497 		ATH9K_INT_RXHP |		\
498 		ATH9K_INT_TX |			\
499 		ATH9K_INT_BMISS |		\
500 		ATH9K_INT_CST |			\
501 		ATH9K_INT_GTT |			\
502 		ATH9K_INT_TSFOOR |		\
503 		ATH9K_INT_GENTIMER |		\
504 		ATH9K_INT_MCI)
505 
506 	struct ath_softc *sc = dev;
507 	struct ath_hw *ah = sc->sc_ah;
508 	struct ath_common *common = ath9k_hw_common(ah);
509 	enum ath9k_int status;
510 	u32 sync_cause = 0;
511 	bool sched = false;
512 
513 	/*
514 	 * The hardware is not ready/present, don't
515 	 * touch anything. Note this can happen early
516 	 * on if the IRQ is shared.
517 	 */
518 	if (!ah || test_bit(ATH_OP_INVALID, &common->op_flags))
519 		return IRQ_NONE;
520 
521 	/* shared irq, not for us */
522 	if (!ath9k_hw_intrpend(ah))
523 		return IRQ_NONE;
524 
525 	/*
526 	 * Figure out the reason(s) for the interrupt.  Note
527 	 * that the hal returns a pseudo-ISR that may include
528 	 * bits we haven't explicitly enabled so we mask the
529 	 * value to insure we only process bits we requested.
530 	 */
531 	ath9k_hw_getisr(ah, &status, &sync_cause); /* NB: clears ISR too */
532 	ath9k_debug_sync_cause(sc, sync_cause);
533 	status &= ah->imask;	/* discard unasked-for bits */
534 
535 	if (test_bit(ATH_OP_HW_RESET, &common->op_flags)) {
536 		ath9k_hw_kill_interrupts(sc->sc_ah);
537 		return IRQ_HANDLED;
538 	}
539 
540 	/*
541 	 * If there are no status bits set, then this interrupt was not
542 	 * for me (should have been caught above).
543 	 */
544 	if (!status)
545 		return IRQ_NONE;
546 
547 	/* Cache the status */
548 	spin_lock(&sc->intr_lock);
549 	sc->intrstatus |= status;
550 	spin_unlock(&sc->intr_lock);
551 
552 	if (status & SCHED_INTR)
553 		sched = true;
554 
555 	/*
556 	 * If a FATAL interrupt is received, we have to reset the chip
557 	 * immediately.
558 	 */
559 	if (status & ATH9K_INT_FATAL)
560 		goto chip_reset;
561 
562 	if ((ah->config.hw_hang_checks & HW_BB_WATCHDOG) &&
563 	    (status & ATH9K_INT_BB_WATCHDOG))
564 		goto chip_reset;
565 
566 	if (status & ATH9K_INT_SWBA)
567 		tasklet_schedule(&sc->bcon_tasklet);
568 
569 	if (status & ATH9K_INT_TXURN)
570 		ath9k_hw_updatetxtriglevel(ah, true);
571 
572 	if (status & ATH9K_INT_RXEOL) {
573 		ah->imask &= ~(ATH9K_INT_RXEOL | ATH9K_INT_RXORN);
574 		ath9k_hw_set_interrupts(ah);
575 	}
576 
577 	if (!(ah->caps.hw_caps & ATH9K_HW_CAP_AUTOSLEEP))
578 		if (status & ATH9K_INT_TIM_TIMER) {
579 			if (ATH_DBG_WARN_ON_ONCE(sc->ps_idle))
580 				goto chip_reset;
581 			/* Clear RxAbort bit so that we can
582 			 * receive frames */
583 			ath9k_setpower(sc, ATH9K_PM_AWAKE);
584 			spin_lock(&sc->sc_pm_lock);
585 			ath9k_hw_setrxabort(sc->sc_ah, 0);
586 			sc->ps_flags |= PS_WAIT_FOR_BEACON;
587 			spin_unlock(&sc->sc_pm_lock);
588 		}
589 
590 chip_reset:
591 
592 	ath_debug_stat_interrupt(sc, status);
593 
594 	if (sched) {
595 		/* turn off every interrupt */
596 		ath9k_hw_kill_interrupts(ah);
597 		tasklet_schedule(&sc->intr_tq);
598 	}
599 
600 	return IRQ_HANDLED;
601 
602 #undef SCHED_INTR
603 }
604 
605 /*
606  * This function is called when a HW reset cannot be deferred
607  * and has to be immediate.
608  */
609 int ath_reset(struct ath_softc *sc, struct ath9k_channel *hchan)
610 {
611 	struct ath_common *common = ath9k_hw_common(sc->sc_ah);
612 	int r;
613 
614 	ath9k_hw_kill_interrupts(sc->sc_ah);
615 	set_bit(ATH_OP_HW_RESET, &common->op_flags);
616 
617 	ath9k_ps_wakeup(sc);
618 	r = ath_reset_internal(sc, hchan);
619 	ath9k_ps_restore(sc);
620 
621 	return r;
622 }
623 
624 /*
625  * When a HW reset can be deferred, it is added to the
626  * hw_reset_work workqueue, but we set ATH_OP_HW_RESET before
627  * queueing.
628  */
629 void ath9k_queue_reset(struct ath_softc *sc, enum ath_reset_type type)
630 {
631 	struct ath_common *common = ath9k_hw_common(sc->sc_ah);
632 #ifdef CONFIG_ATH9K_DEBUGFS
633 	RESET_STAT_INC(sc, type);
634 #endif
635 	ath9k_hw_kill_interrupts(sc->sc_ah);
636 	set_bit(ATH_OP_HW_RESET, &common->op_flags);
637 	ieee80211_queue_work(sc->hw, &sc->hw_reset_work);
638 }
639 
640 void ath_reset_work(struct work_struct *work)
641 {
642 	struct ath_softc *sc = container_of(work, struct ath_softc, hw_reset_work);
643 
644 	ath9k_ps_wakeup(sc);
645 	ath_reset_internal(sc, NULL);
646 	ath9k_ps_restore(sc);
647 }
648 
649 /**********************/
650 /* mac80211 callbacks */
651 /**********************/
652 
653 static int ath9k_start(struct ieee80211_hw *hw)
654 {
655 	struct ath_softc *sc = hw->priv;
656 	struct ath_hw *ah = sc->sc_ah;
657 	struct ath_common *common = ath9k_hw_common(ah);
658 	struct ieee80211_channel *curchan = sc->cur_chan->chandef.chan;
659 	struct ath_chanctx *ctx = sc->cur_chan;
660 	struct ath9k_channel *init_channel;
661 	int r;
662 
663 	ath_dbg(common, CONFIG,
664 		"Starting driver with initial channel: %d MHz\n",
665 		curchan->center_freq);
666 
667 	ath9k_ps_wakeup(sc);
668 	mutex_lock(&sc->mutex);
669 
670 	init_channel = ath9k_cmn_get_channel(hw, ah, &ctx->chandef);
671 	sc->cur_chandef = hw->conf.chandef;
672 
673 	/* Reset SERDES registers */
674 	ath9k_hw_configpcipowersave(ah, false);
675 
676 	/*
677 	 * The basic interface to setting the hardware in a good
678 	 * state is ``reset''.  On return the hardware is known to
679 	 * be powered up and with interrupts disabled.  This must
680 	 * be followed by initialization of the appropriate bits
681 	 * and then setup of the interrupt mask.
682 	 */
683 	spin_lock_bh(&sc->sc_pcu_lock);
684 
685 	atomic_set(&ah->intr_ref_cnt, -1);
686 
687 	r = ath9k_hw_reset(ah, init_channel, ah->caldata, false);
688 	if (r) {
689 		ath_err(common,
690 			"Unable to reset hardware; reset status %d (freq %u MHz)\n",
691 			r, curchan->center_freq);
692 		ah->reset_power_on = false;
693 	}
694 
695 	/* Setup our intr mask. */
696 	ah->imask = ATH9K_INT_TX | ATH9K_INT_RXEOL |
697 		    ATH9K_INT_RXORN | ATH9K_INT_FATAL |
698 		    ATH9K_INT_GLOBAL;
699 
700 	if (ah->caps.hw_caps & ATH9K_HW_CAP_EDMA)
701 		ah->imask |= ATH9K_INT_RXHP |
702 			     ATH9K_INT_RXLP;
703 	else
704 		ah->imask |= ATH9K_INT_RX;
705 
706 	if (ah->config.hw_hang_checks & HW_BB_WATCHDOG)
707 		ah->imask |= ATH9K_INT_BB_WATCHDOG;
708 
709 	/*
710 	 * Enable GTT interrupts only for AR9003/AR9004 chips
711 	 * for now.
712 	 */
713 	if (AR_SREV_9300_20_OR_LATER(ah))
714 		ah->imask |= ATH9K_INT_GTT;
715 
716 	if (ah->caps.hw_caps & ATH9K_HW_CAP_HT)
717 		ah->imask |= ATH9K_INT_CST;
718 
719 	ath_mci_enable(sc);
720 
721 	clear_bit(ATH_OP_INVALID, &common->op_flags);
722 	sc->sc_ah->is_monitoring = false;
723 
724 	if (!ath_complete_reset(sc, false))
725 		ah->reset_power_on = false;
726 
727 	if (ah->led_pin >= 0) {
728 		ath9k_hw_set_gpio(ah, ah->led_pin,
729 				  (ah->config.led_active_high) ? 1 : 0);
730 		ath9k_hw_gpio_request_out(ah, ah->led_pin, NULL,
731 					  AR_GPIO_OUTPUT_MUX_AS_OUTPUT);
732 	}
733 
734 	/*
735 	 * Reset key cache to sane defaults (all entries cleared) instead of
736 	 * semi-random values after suspend/resume.
737 	 */
738 	ath9k_cmn_init_crypto(sc->sc_ah);
739 
740 	ath9k_hw_reset_tsf(ah);
741 
742 	spin_unlock_bh(&sc->sc_pcu_lock);
743 
744 	ath9k_rng_start(sc);
745 
746 	mutex_unlock(&sc->mutex);
747 
748 	ath9k_ps_restore(sc);
749 
750 	return 0;
751 }
752 
753 static void ath9k_tx(struct ieee80211_hw *hw,
754 		     struct ieee80211_tx_control *control,
755 		     struct sk_buff *skb)
756 {
757 	struct ath_softc *sc = hw->priv;
758 	struct ath_common *common = ath9k_hw_common(sc->sc_ah);
759 	struct ath_tx_control txctl;
760 	struct ieee80211_hdr *hdr = (struct ieee80211_hdr *) skb->data;
761 	unsigned long flags;
762 
763 	if (sc->ps_enabled) {
764 		/*
765 		 * mac80211 does not set PM field for normal data frames, so we
766 		 * need to update that based on the current PS mode.
767 		 */
768 		if (ieee80211_is_data(hdr->frame_control) &&
769 		    !ieee80211_is_nullfunc(hdr->frame_control) &&
770 		    !ieee80211_has_pm(hdr->frame_control)) {
771 			ath_dbg(common, PS,
772 				"Add PM=1 for a TX frame while in PS mode\n");
773 			hdr->frame_control |= cpu_to_le16(IEEE80211_FCTL_PM);
774 		}
775 	}
776 
777 	if (unlikely(sc->sc_ah->power_mode == ATH9K_PM_NETWORK_SLEEP)) {
778 		/*
779 		 * We are using PS-Poll and mac80211 can request TX while in
780 		 * power save mode. Need to wake up hardware for the TX to be
781 		 * completed and if needed, also for RX of buffered frames.
782 		 */
783 		ath9k_ps_wakeup(sc);
784 		spin_lock_irqsave(&sc->sc_pm_lock, flags);
785 		if (!(sc->sc_ah->caps.hw_caps & ATH9K_HW_CAP_AUTOSLEEP))
786 			ath9k_hw_setrxabort(sc->sc_ah, 0);
787 		if (ieee80211_is_pspoll(hdr->frame_control)) {
788 			ath_dbg(common, PS,
789 				"Sending PS-Poll to pick a buffered frame\n");
790 			sc->ps_flags |= PS_WAIT_FOR_PSPOLL_DATA;
791 		} else {
792 			ath_dbg(common, PS, "Wake up to complete TX\n");
793 			sc->ps_flags |= PS_WAIT_FOR_TX_ACK;
794 		}
795 		/*
796 		 * The actual restore operation will happen only after
797 		 * the ps_flags bit is cleared. We are just dropping
798 		 * the ps_usecount here.
799 		 */
800 		spin_unlock_irqrestore(&sc->sc_pm_lock, flags);
801 		ath9k_ps_restore(sc);
802 	}
803 
804 	/*
805 	 * Cannot tx while the hardware is in full sleep, it first needs a full
806 	 * chip reset to recover from that
807 	 */
808 	if (unlikely(sc->sc_ah->power_mode == ATH9K_PM_FULL_SLEEP)) {
809 		ath_err(common, "TX while HW is in FULL_SLEEP mode\n");
810 		goto exit;
811 	}
812 
813 	memset(&txctl, 0, sizeof(struct ath_tx_control));
814 	txctl.txq = sc->tx.txq_map[skb_get_queue_mapping(skb)];
815 	txctl.sta = control->sta;
816 
817 	ath_dbg(common, XMIT, "transmitting packet, skb: %p\n", skb);
818 
819 	if (ath_tx_start(hw, skb, &txctl) != 0) {
820 		ath_dbg(common, XMIT, "TX failed\n");
821 		TX_STAT_INC(sc, txctl.txq->axq_qnum, txfailed);
822 		goto exit;
823 	}
824 
825 	return;
826 exit:
827 	ieee80211_free_txskb(hw, skb);
828 }
829 
830 static bool ath9k_txq_list_has_key(struct list_head *txq_list, u32 keyix)
831 {
832 	struct ath_buf *bf;
833 	struct ieee80211_tx_info *txinfo;
834 	struct ath_frame_info *fi;
835 
836 	list_for_each_entry(bf, txq_list, list) {
837 		if (bf->bf_state.stale || !bf->bf_mpdu)
838 			continue;
839 
840 		txinfo = IEEE80211_SKB_CB(bf->bf_mpdu);
841 		fi = (struct ath_frame_info *)&txinfo->status.status_driver_data[0];
842 		if (fi->keyix == keyix)
843 			return true;
844 	}
845 
846 	return false;
847 }
848 
849 static bool ath9k_txq_has_key(struct ath_softc *sc, u32 keyix)
850 {
851 	struct ath_hw *ah = sc->sc_ah;
852 	int i, j;
853 	struct ath_txq *txq;
854 	bool key_in_use = false;
855 
856 	for (i = 0; !key_in_use && i < ATH9K_NUM_TX_QUEUES; i++) {
857 		if (!ATH_TXQ_SETUP(sc, i))
858 			continue;
859 		txq = &sc->tx.txq[i];
860 		if (!txq->axq_depth)
861 			continue;
862 		if (!ath9k_hw_numtxpending(ah, txq->axq_qnum))
863 			continue;
864 
865 		ath_txq_lock(sc, txq);
866 		key_in_use = ath9k_txq_list_has_key(&txq->axq_q, keyix);
867 		if (sc->sc_ah->caps.hw_caps & ATH9K_HW_CAP_EDMA) {
868 			int idx = txq->txq_tailidx;
869 
870 			for (j = 0; !key_in_use &&
871 			     !list_empty(&txq->txq_fifo[idx]) &&
872 			     j < ATH_TXFIFO_DEPTH; j++) {
873 				key_in_use = ath9k_txq_list_has_key(
874 					&txq->txq_fifo[idx], keyix);
875 				INCR(idx, ATH_TXFIFO_DEPTH);
876 			}
877 		}
878 		ath_txq_unlock(sc, txq);
879 	}
880 
881 	return key_in_use;
882 }
883 
884 static void ath9k_pending_key_del(struct ath_softc *sc, u8 keyix)
885 {
886 	struct ath_hw *ah = sc->sc_ah;
887 	struct ath_common *common = ath9k_hw_common(ah);
888 
889 	if (!test_bit(keyix, ah->pending_del_keymap) ||
890 	    ath9k_txq_has_key(sc, keyix))
891 		return;
892 
893 	/* No more TXQ frames point to this key cache entry, so delete it. */
894 	clear_bit(keyix, ah->pending_del_keymap);
895 	ath_key_delete(common, keyix);
896 }
897 
898 static void ath9k_stop(struct ieee80211_hw *hw)
899 {
900 	struct ath_softc *sc = hw->priv;
901 	struct ath_hw *ah = sc->sc_ah;
902 	struct ath_common *common = ath9k_hw_common(ah);
903 	bool prev_idle;
904 	int i;
905 
906 	ath9k_deinit_channel_context(sc);
907 
908 	mutex_lock(&sc->mutex);
909 
910 	ath9k_rng_stop(sc);
911 
912 	ath_cancel_work(sc);
913 
914 	if (test_bit(ATH_OP_INVALID, &common->op_flags)) {
915 		ath_dbg(common, ANY, "Device not present\n");
916 		mutex_unlock(&sc->mutex);
917 		return;
918 	}
919 
920 	/* Ensure HW is awake when we try to shut it down. */
921 	ath9k_ps_wakeup(sc);
922 
923 	spin_lock_bh(&sc->sc_pcu_lock);
924 
925 	/* prevent tasklets to enable interrupts once we disable them */
926 	ah->imask &= ~ATH9K_INT_GLOBAL;
927 
928 	/* make sure h/w will not generate any interrupt
929 	 * before setting the invalid flag. */
930 	ath9k_hw_disable_interrupts(ah);
931 
932 	spin_unlock_bh(&sc->sc_pcu_lock);
933 
934 	/* we can now sync irq and kill any running tasklets, since we already
935 	 * disabled interrupts and not holding a spin lock */
936 	synchronize_irq(sc->irq);
937 	tasklet_kill(&sc->intr_tq);
938 	tasklet_kill(&sc->bcon_tasklet);
939 
940 	prev_idle = sc->ps_idle;
941 	sc->ps_idle = true;
942 
943 	spin_lock_bh(&sc->sc_pcu_lock);
944 
945 	if (ah->led_pin >= 0) {
946 		ath9k_hw_set_gpio(ah, ah->led_pin,
947 				  (ah->config.led_active_high) ? 0 : 1);
948 		ath9k_hw_gpio_request_in(ah, ah->led_pin, NULL);
949 	}
950 
951 	ath_prepare_reset(sc);
952 
953 	if (sc->rx.frag) {
954 		dev_kfree_skb_any(sc->rx.frag);
955 		sc->rx.frag = NULL;
956 	}
957 
958 	if (!ah->curchan)
959 		ah->curchan = ath9k_cmn_get_channel(hw, ah,
960 						    &sc->cur_chan->chandef);
961 
962 	ath9k_hw_reset(ah, ah->curchan, ah->caldata, false);
963 
964 	set_bit(ATH_OP_INVALID, &common->op_flags);
965 
966 	ath9k_hw_phy_disable(ah);
967 
968 	ath9k_hw_configpcipowersave(ah, true);
969 
970 	spin_unlock_bh(&sc->sc_pcu_lock);
971 
972 	for (i = 0; i < ATH_KEYMAX; i++)
973 		ath9k_pending_key_del(sc, i);
974 
975 	/* Clear key cache entries explicitly to get rid of any potentially
976 	 * remaining keys.
977 	 */
978 	ath9k_cmn_init_crypto(sc->sc_ah);
979 
980 	ath9k_ps_restore(sc);
981 
982 	sc->ps_idle = prev_idle;
983 
984 	mutex_unlock(&sc->mutex);
985 
986 	ath_dbg(common, CONFIG, "Driver halt\n");
987 }
988 
989 static bool ath9k_uses_beacons(int type)
990 {
991 	switch (type) {
992 	case NL80211_IFTYPE_AP:
993 	case NL80211_IFTYPE_ADHOC:
994 	case NL80211_IFTYPE_MESH_POINT:
995 		return true;
996 	default:
997 		return false;
998 	}
999 }
1000 
1001 static void ath9k_vif_iter_set_beacon(struct ath9k_vif_iter_data *iter_data,
1002 				      struct ieee80211_vif *vif)
1003 {
1004 	/* Use the first (configured) interface, but prefering AP interfaces. */
1005 	if (!iter_data->primary_beacon_vif) {
1006 		iter_data->primary_beacon_vif = vif;
1007 	} else {
1008 		if (iter_data->primary_beacon_vif->type != NL80211_IFTYPE_AP &&
1009 		    vif->type == NL80211_IFTYPE_AP)
1010 			iter_data->primary_beacon_vif = vif;
1011 	}
1012 
1013 	iter_data->beacons = true;
1014 	iter_data->nbcnvifs += 1;
1015 }
1016 
1017 static void ath9k_vif_iter(struct ath9k_vif_iter_data *iter_data,
1018 			   u8 *mac, struct ieee80211_vif *vif)
1019 {
1020 	struct ath_vif *avp = (struct ath_vif *)vif->drv_priv;
1021 	int i;
1022 
1023 	if (iter_data->has_hw_macaddr) {
1024 		for (i = 0; i < ETH_ALEN; i++)
1025 			iter_data->mask[i] &=
1026 				~(iter_data->hw_macaddr[i] ^ mac[i]);
1027 	} else {
1028 		memcpy(iter_data->hw_macaddr, mac, ETH_ALEN);
1029 		iter_data->has_hw_macaddr = true;
1030 	}
1031 
1032 	if (!vif->bss_conf.use_short_slot)
1033 		iter_data->slottime = 20;
1034 
1035 	switch (vif->type) {
1036 	case NL80211_IFTYPE_AP:
1037 		iter_data->naps++;
1038 		if (vif->bss_conf.enable_beacon)
1039 			ath9k_vif_iter_set_beacon(iter_data, vif);
1040 		break;
1041 	case NL80211_IFTYPE_STATION:
1042 		iter_data->nstations++;
1043 		if (avp->assoc && !iter_data->primary_sta)
1044 			iter_data->primary_sta = vif;
1045 		break;
1046 	case NL80211_IFTYPE_OCB:
1047 		iter_data->nocbs++;
1048 		break;
1049 	case NL80211_IFTYPE_ADHOC:
1050 		iter_data->nadhocs++;
1051 		if (vif->bss_conf.enable_beacon)
1052 			ath9k_vif_iter_set_beacon(iter_data, vif);
1053 		break;
1054 	case NL80211_IFTYPE_MESH_POINT:
1055 		iter_data->nmeshes++;
1056 		if (vif->bss_conf.enable_beacon)
1057 			ath9k_vif_iter_set_beacon(iter_data, vif);
1058 		break;
1059 	default:
1060 		break;
1061 	}
1062 }
1063 
1064 static void ath9k_update_bssid_mask(struct ath_softc *sc,
1065 				    struct ath_chanctx *ctx,
1066 				    struct ath9k_vif_iter_data *iter_data)
1067 {
1068 	struct ath_common *common = ath9k_hw_common(sc->sc_ah);
1069 	struct ath_vif *avp;
1070 	int i;
1071 
1072 	if (!ath9k_is_chanctx_enabled())
1073 		return;
1074 
1075 	list_for_each_entry(avp, &ctx->vifs, list) {
1076 		if (ctx->nvifs_assigned != 1)
1077 			continue;
1078 
1079 		if (!iter_data->has_hw_macaddr)
1080 			continue;
1081 
1082 		ether_addr_copy(common->curbssid, avp->bssid);
1083 
1084 		/* perm_addr will be used as the p2p device address. */
1085 		for (i = 0; i < ETH_ALEN; i++)
1086 			iter_data->mask[i] &=
1087 				~(iter_data->hw_macaddr[i] ^
1088 				  sc->hw->wiphy->perm_addr[i]);
1089 	}
1090 }
1091 
1092 /* Called with sc->mutex held. */
1093 void ath9k_calculate_iter_data(struct ath_softc *sc,
1094 			       struct ath_chanctx *ctx,
1095 			       struct ath9k_vif_iter_data *iter_data)
1096 {
1097 	struct ath_vif *avp;
1098 
1099 	/*
1100 	 * The hardware will use primary station addr together with the
1101 	 * BSSID mask when matching addresses.
1102 	 */
1103 	memset(iter_data, 0, sizeof(*iter_data));
1104 	eth_broadcast_addr(iter_data->mask);
1105 	iter_data->slottime = 9;
1106 
1107 	list_for_each_entry(avp, &ctx->vifs, list)
1108 		ath9k_vif_iter(iter_data, avp->vif->addr, avp->vif);
1109 
1110 	ath9k_update_bssid_mask(sc, ctx, iter_data);
1111 }
1112 
1113 static void ath9k_set_assoc_state(struct ath_softc *sc,
1114 				  struct ieee80211_vif *vif, bool changed)
1115 {
1116 	struct ath_common *common = ath9k_hw_common(sc->sc_ah);
1117 	struct ath_vif *avp = (struct ath_vif *)vif->drv_priv;
1118 	unsigned long flags;
1119 
1120 	set_bit(ATH_OP_PRIM_STA_VIF, &common->op_flags);
1121 
1122 	ether_addr_copy(common->curbssid, avp->bssid);
1123 	common->curaid = avp->aid;
1124 	ath9k_hw_write_associd(sc->sc_ah);
1125 
1126 	if (changed) {
1127 		common->last_rssi = ATH_RSSI_DUMMY_MARKER;
1128 		sc->sc_ah->stats.avgbrssi = ATH_RSSI_DUMMY_MARKER;
1129 
1130 		spin_lock_irqsave(&sc->sc_pm_lock, flags);
1131 		sc->ps_flags |= PS_BEACON_SYNC | PS_WAIT_FOR_BEACON;
1132 		spin_unlock_irqrestore(&sc->sc_pm_lock, flags);
1133 	}
1134 
1135 	if (ath9k_hw_mci_is_enabled(sc->sc_ah))
1136 		ath9k_mci_update_wlan_channels(sc, false);
1137 
1138 	ath_dbg(common, CONFIG,
1139 		"Primary Station interface: %pM, BSSID: %pM\n",
1140 		vif->addr, common->curbssid);
1141 }
1142 
1143 #ifdef CONFIG_ATH9K_CHANNEL_CONTEXT
1144 static void ath9k_set_offchannel_state(struct ath_softc *sc)
1145 {
1146 	struct ath_hw *ah = sc->sc_ah;
1147 	struct ath_common *common = ath9k_hw_common(ah);
1148 	struct ieee80211_vif *vif = NULL;
1149 
1150 	ath9k_ps_wakeup(sc);
1151 
1152 	if (sc->offchannel.state < ATH_OFFCHANNEL_ROC_START)
1153 		vif = sc->offchannel.scan_vif;
1154 	else
1155 		vif = sc->offchannel.roc_vif;
1156 
1157 	if (WARN_ON(!vif))
1158 		goto exit;
1159 
1160 	eth_zero_addr(common->curbssid);
1161 	eth_broadcast_addr(common->bssidmask);
1162 	memcpy(common->macaddr, vif->addr, ETH_ALEN);
1163 	common->curaid = 0;
1164 	ah->opmode = vif->type;
1165 	ah->imask &= ~ATH9K_INT_SWBA;
1166 	ah->imask &= ~ATH9K_INT_TSFOOR;
1167 	ah->slottime = 9;
1168 
1169 	ath_hw_setbssidmask(common);
1170 	ath9k_hw_setopmode(ah);
1171 	ath9k_hw_write_associd(sc->sc_ah);
1172 	ath9k_hw_set_interrupts(ah);
1173 	ath9k_hw_init_global_settings(ah);
1174 
1175 exit:
1176 	ath9k_ps_restore(sc);
1177 }
1178 #endif
1179 
1180 /* Called with sc->mutex held. */
1181 void ath9k_calculate_summary_state(struct ath_softc *sc,
1182 				   struct ath_chanctx *ctx)
1183 {
1184 	struct ath_hw *ah = sc->sc_ah;
1185 	struct ath_common *common = ath9k_hw_common(ah);
1186 	struct ath9k_vif_iter_data iter_data;
1187 
1188 	ath_chanctx_check_active(sc, ctx);
1189 
1190 	if (ctx != sc->cur_chan)
1191 		return;
1192 
1193 #ifdef CONFIG_ATH9K_CHANNEL_CONTEXT
1194 	if (ctx == &sc->offchannel.chan)
1195 		return ath9k_set_offchannel_state(sc);
1196 #endif
1197 
1198 	ath9k_ps_wakeup(sc);
1199 	ath9k_calculate_iter_data(sc, ctx, &iter_data);
1200 
1201 	if (iter_data.has_hw_macaddr)
1202 		memcpy(common->macaddr, iter_data.hw_macaddr, ETH_ALEN);
1203 
1204 	memcpy(common->bssidmask, iter_data.mask, ETH_ALEN);
1205 	ath_hw_setbssidmask(common);
1206 
1207 	if (iter_data.naps > 0) {
1208 		ath9k_hw_set_tsfadjust(ah, true);
1209 		ah->opmode = NL80211_IFTYPE_AP;
1210 	} else {
1211 		ath9k_hw_set_tsfadjust(ah, false);
1212 		if (iter_data.beacons)
1213 			ath9k_beacon_ensure_primary_slot(sc);
1214 
1215 		if (iter_data.nmeshes)
1216 			ah->opmode = NL80211_IFTYPE_MESH_POINT;
1217 		else if (iter_data.nocbs)
1218 			ah->opmode = NL80211_IFTYPE_OCB;
1219 		else if (iter_data.nadhocs)
1220 			ah->opmode = NL80211_IFTYPE_ADHOC;
1221 		else
1222 			ah->opmode = NL80211_IFTYPE_STATION;
1223 	}
1224 
1225 	ath9k_hw_setopmode(ah);
1226 
1227 	ctx->switch_after_beacon = false;
1228 	if ((iter_data.nstations + iter_data.nadhocs + iter_data.nmeshes) > 0)
1229 		ah->imask |= ATH9K_INT_TSFOOR;
1230 	else {
1231 		ah->imask &= ~ATH9K_INT_TSFOOR;
1232 		if (iter_data.naps == 1 && iter_data.beacons)
1233 			ctx->switch_after_beacon = true;
1234 	}
1235 
1236 	if (ah->opmode == NL80211_IFTYPE_STATION) {
1237 		bool changed = (iter_data.primary_sta != ctx->primary_sta);
1238 
1239 		if (iter_data.primary_sta) {
1240 			iter_data.primary_beacon_vif = iter_data.primary_sta;
1241 			iter_data.beacons = true;
1242 			ath9k_set_assoc_state(sc, iter_data.primary_sta,
1243 					      changed);
1244 			ctx->primary_sta = iter_data.primary_sta;
1245 		} else {
1246 			ctx->primary_sta = NULL;
1247 			eth_zero_addr(common->curbssid);
1248 			common->curaid = 0;
1249 			ath9k_hw_write_associd(sc->sc_ah);
1250 			if (ath9k_hw_mci_is_enabled(sc->sc_ah))
1251 				ath9k_mci_update_wlan_channels(sc, true);
1252 		}
1253 	}
1254 	sc->nbcnvifs = iter_data.nbcnvifs;
1255 	ath9k_beacon_config(sc, iter_data.primary_beacon_vif,
1256 			    iter_data.beacons);
1257 	ath9k_hw_set_interrupts(ah);
1258 
1259 	if (ah->slottime != iter_data.slottime) {
1260 		ah->slottime = iter_data.slottime;
1261 		ath9k_hw_init_global_settings(ah);
1262 	}
1263 
1264 	if (iter_data.primary_sta)
1265 		set_bit(ATH_OP_PRIM_STA_VIF, &common->op_flags);
1266 	else
1267 		clear_bit(ATH_OP_PRIM_STA_VIF, &common->op_flags);
1268 
1269 	ath_dbg(common, CONFIG,
1270 		"macaddr: %pM, bssid: %pM, bssidmask: %pM\n",
1271 		common->macaddr, common->curbssid, common->bssidmask);
1272 
1273 	ath9k_ps_restore(sc);
1274 }
1275 
1276 static void ath9k_tpc_vif_iter(void *data, u8 *mac, struct ieee80211_vif *vif)
1277 {
1278 	int *power = data;
1279 
1280 	if (vif->bss_conf.txpower == INT_MIN)
1281 		return;
1282 
1283 	if (*power < vif->bss_conf.txpower)
1284 		*power = vif->bss_conf.txpower;
1285 }
1286 
1287 /* Called with sc->mutex held. */
1288 void ath9k_set_txpower(struct ath_softc *sc, struct ieee80211_vif *vif)
1289 {
1290 	int power;
1291 	struct ath_hw *ah = sc->sc_ah;
1292 	struct ath_regulatory *reg = ath9k_hw_regulatory(ah);
1293 
1294 	ath9k_ps_wakeup(sc);
1295 	if (ah->tpc_enabled) {
1296 		power = (vif) ? vif->bss_conf.txpower : -1;
1297 		ieee80211_iterate_active_interfaces_atomic(
1298 				sc->hw, IEEE80211_IFACE_ITER_RESUME_ALL,
1299 				ath9k_tpc_vif_iter, &power);
1300 		if (power == -1)
1301 			power = sc->hw->conf.power_level;
1302 	} else {
1303 		power = sc->hw->conf.power_level;
1304 	}
1305 	sc->cur_chan->txpower = 2 * power;
1306 	ath9k_hw_set_txpowerlimit(ah, sc->cur_chan->txpower, false);
1307 	sc->cur_chan->cur_txpower = reg->max_power_level;
1308 	ath9k_ps_restore(sc);
1309 }
1310 
1311 static void ath9k_assign_hw_queues(struct ieee80211_hw *hw,
1312 				   struct ieee80211_vif *vif)
1313 {
1314 	int i;
1315 
1316 	if (!ath9k_is_chanctx_enabled())
1317 		return;
1318 
1319 	for (i = 0; i < IEEE80211_NUM_ACS; i++)
1320 		vif->hw_queue[i] = i;
1321 
1322 	if (vif->type == NL80211_IFTYPE_AP ||
1323 	    vif->type == NL80211_IFTYPE_MESH_POINT)
1324 		vif->cab_queue = hw->queues - 2;
1325 	else
1326 		vif->cab_queue = IEEE80211_INVAL_HW_QUEUE;
1327 }
1328 
1329 static int ath9k_add_interface(struct ieee80211_hw *hw,
1330 			       struct ieee80211_vif *vif)
1331 {
1332 	struct ath_softc *sc = hw->priv;
1333 	struct ath_hw *ah = sc->sc_ah;
1334 	struct ath_common *common = ath9k_hw_common(ah);
1335 	struct ath_vif *avp = (void *)vif->drv_priv;
1336 	struct ath_node *an = &avp->mcast_node;
1337 
1338 	mutex_lock(&sc->mutex);
1339 	if (IS_ENABLED(CONFIG_ATH9K_TX99)) {
1340 		if (sc->cur_chan->nvifs >= 1) {
1341 			mutex_unlock(&sc->mutex);
1342 			return -EOPNOTSUPP;
1343 		}
1344 		sc->tx99_vif = vif;
1345 	}
1346 
1347 	ath_dbg(common, CONFIG, "Attach a VIF of type: %d\n", vif->type);
1348 	sc->cur_chan->nvifs++;
1349 
1350 	if (vif->type == NL80211_IFTYPE_STATION && ath9k_is_chanctx_enabled())
1351 		vif->driver_flags |= IEEE80211_VIF_GET_NOA_UPDATE;
1352 
1353 	if (ath9k_uses_beacons(vif->type))
1354 		ath9k_beacon_assign_slot(sc, vif);
1355 
1356 	avp->vif = vif;
1357 	if (!ath9k_is_chanctx_enabled()) {
1358 		avp->chanctx = sc->cur_chan;
1359 		list_add_tail(&avp->list, &avp->chanctx->vifs);
1360 	}
1361 
1362 	ath9k_calculate_summary_state(sc, avp->chanctx);
1363 
1364 	ath9k_assign_hw_queues(hw, vif);
1365 
1366 	ath9k_set_txpower(sc, vif);
1367 
1368 	an->sc = sc;
1369 	an->sta = NULL;
1370 	an->vif = vif;
1371 	an->no_ps_filter = true;
1372 	ath_tx_node_init(sc, an);
1373 
1374 	mutex_unlock(&sc->mutex);
1375 	return 0;
1376 }
1377 
1378 static int ath9k_change_interface(struct ieee80211_hw *hw,
1379 				  struct ieee80211_vif *vif,
1380 				  enum nl80211_iftype new_type,
1381 				  bool p2p)
1382 {
1383 	struct ath_softc *sc = hw->priv;
1384 	struct ath_common *common = ath9k_hw_common(sc->sc_ah);
1385 	struct ath_vif *avp = (void *)vif->drv_priv;
1386 
1387 	mutex_lock(&sc->mutex);
1388 
1389 	if (IS_ENABLED(CONFIG_ATH9K_TX99)) {
1390 		mutex_unlock(&sc->mutex);
1391 		return -EOPNOTSUPP;
1392 	}
1393 
1394 	ath_dbg(common, CONFIG, "Change Interface\n");
1395 
1396 	if (ath9k_uses_beacons(vif->type))
1397 		ath9k_beacon_remove_slot(sc, vif);
1398 
1399 	vif->type = new_type;
1400 	vif->p2p = p2p;
1401 
1402 	if (ath9k_uses_beacons(vif->type))
1403 		ath9k_beacon_assign_slot(sc, vif);
1404 
1405 	ath9k_assign_hw_queues(hw, vif);
1406 	ath9k_calculate_summary_state(sc, avp->chanctx);
1407 
1408 	ath9k_set_txpower(sc, vif);
1409 
1410 	mutex_unlock(&sc->mutex);
1411 	return 0;
1412 }
1413 
1414 static void ath9k_remove_interface(struct ieee80211_hw *hw,
1415 				   struct ieee80211_vif *vif)
1416 {
1417 	struct ath_softc *sc = hw->priv;
1418 	struct ath_common *common = ath9k_hw_common(sc->sc_ah);
1419 	struct ath_vif *avp = (void *)vif->drv_priv;
1420 
1421 	ath_dbg(common, CONFIG, "Detach Interface\n");
1422 
1423 	mutex_lock(&sc->mutex);
1424 
1425 	ath9k_p2p_remove_vif(sc, vif);
1426 
1427 	sc->cur_chan->nvifs--;
1428 	sc->tx99_vif = NULL;
1429 	if (!ath9k_is_chanctx_enabled())
1430 		list_del(&avp->list);
1431 
1432 	if (ath9k_uses_beacons(vif->type))
1433 		ath9k_beacon_remove_slot(sc, vif);
1434 
1435 	ath_tx_node_cleanup(sc, &avp->mcast_node);
1436 
1437 	ath9k_calculate_summary_state(sc, avp->chanctx);
1438 
1439 	ath9k_set_txpower(sc, NULL);
1440 
1441 	mutex_unlock(&sc->mutex);
1442 }
1443 
1444 static void ath9k_enable_ps(struct ath_softc *sc)
1445 {
1446 	struct ath_hw *ah = sc->sc_ah;
1447 	struct ath_common *common = ath9k_hw_common(ah);
1448 
1449 	if (IS_ENABLED(CONFIG_ATH9K_TX99))
1450 		return;
1451 
1452 	sc->ps_enabled = true;
1453 	if (!(ah->caps.hw_caps & ATH9K_HW_CAP_AUTOSLEEP)) {
1454 		if ((ah->imask & ATH9K_INT_TIM_TIMER) == 0) {
1455 			ah->imask |= ATH9K_INT_TIM_TIMER;
1456 			ath9k_hw_set_interrupts(ah);
1457 		}
1458 		ath9k_hw_setrxabort(ah, 1);
1459 	}
1460 	ath_dbg(common, PS, "PowerSave enabled\n");
1461 }
1462 
1463 static void ath9k_disable_ps(struct ath_softc *sc)
1464 {
1465 	struct ath_hw *ah = sc->sc_ah;
1466 	struct ath_common *common = ath9k_hw_common(ah);
1467 
1468 	if (IS_ENABLED(CONFIG_ATH9K_TX99))
1469 		return;
1470 
1471 	sc->ps_enabled = false;
1472 	ath9k_hw_setpower(ah, ATH9K_PM_AWAKE);
1473 	if (!(ah->caps.hw_caps & ATH9K_HW_CAP_AUTOSLEEP)) {
1474 		ath9k_hw_setrxabort(ah, 0);
1475 		sc->ps_flags &= ~(PS_WAIT_FOR_BEACON |
1476 				  PS_WAIT_FOR_CAB |
1477 				  PS_WAIT_FOR_PSPOLL_DATA |
1478 				  PS_WAIT_FOR_TX_ACK);
1479 		if (ah->imask & ATH9K_INT_TIM_TIMER) {
1480 			ah->imask &= ~ATH9K_INT_TIM_TIMER;
1481 			ath9k_hw_set_interrupts(ah);
1482 		}
1483 	}
1484 	ath_dbg(common, PS, "PowerSave disabled\n");
1485 }
1486 
1487 static int ath9k_config(struct ieee80211_hw *hw, u32 changed)
1488 {
1489 	struct ath_softc *sc = hw->priv;
1490 	struct ath_hw *ah = sc->sc_ah;
1491 	struct ath_common *common = ath9k_hw_common(ah);
1492 	struct ieee80211_conf *conf = &hw->conf;
1493 	struct ath_chanctx *ctx = sc->cur_chan;
1494 
1495 	ath9k_ps_wakeup(sc);
1496 	mutex_lock(&sc->mutex);
1497 
1498 	if (changed & IEEE80211_CONF_CHANGE_IDLE) {
1499 		sc->ps_idle = !!(conf->flags & IEEE80211_CONF_IDLE);
1500 		if (sc->ps_idle) {
1501 			ath_cancel_work(sc);
1502 			ath9k_stop_btcoex(sc);
1503 		} else {
1504 			ath9k_start_btcoex(sc);
1505 			/*
1506 			 * The chip needs a reset to properly wake up from
1507 			 * full sleep
1508 			 */
1509 			ath_chanctx_set_channel(sc, ctx, &ctx->chandef);
1510 		}
1511 	}
1512 
1513 	/*
1514 	 * We just prepare to enable PS. We have to wait until our AP has
1515 	 * ACK'd our null data frame to disable RX otherwise we'll ignore
1516 	 * those ACKs and end up retransmitting the same null data frames.
1517 	 * IEEE80211_CONF_CHANGE_PS is only passed by mac80211 for STA mode.
1518 	 */
1519 	if (changed & IEEE80211_CONF_CHANGE_PS) {
1520 		unsigned long flags;
1521 		spin_lock_irqsave(&sc->sc_pm_lock, flags);
1522 		if (conf->flags & IEEE80211_CONF_PS)
1523 			ath9k_enable_ps(sc);
1524 		else
1525 			ath9k_disable_ps(sc);
1526 		spin_unlock_irqrestore(&sc->sc_pm_lock, flags);
1527 	}
1528 
1529 	if (changed & IEEE80211_CONF_CHANGE_MONITOR) {
1530 		if (conf->flags & IEEE80211_CONF_MONITOR) {
1531 			ath_dbg(common, CONFIG, "Monitor mode is enabled\n");
1532 			sc->sc_ah->is_monitoring = true;
1533 		} else {
1534 			ath_dbg(common, CONFIG, "Monitor mode is disabled\n");
1535 			sc->sc_ah->is_monitoring = false;
1536 		}
1537 	}
1538 
1539 	if (!ath9k_is_chanctx_enabled() && (changed & IEEE80211_CONF_CHANGE_CHANNEL)) {
1540 		ctx->offchannel = !!(conf->flags & IEEE80211_CONF_OFFCHANNEL);
1541 		ath_chanctx_set_channel(sc, ctx, &hw->conf.chandef);
1542 	}
1543 
1544 	if (changed & IEEE80211_CONF_CHANGE_POWER)
1545 		ath9k_set_txpower(sc, NULL);
1546 
1547 	mutex_unlock(&sc->mutex);
1548 	ath9k_ps_restore(sc);
1549 
1550 	return 0;
1551 }
1552 
1553 #define SUPPORTED_FILTERS			\
1554 	(FIF_ALLMULTI |				\
1555 	FIF_CONTROL |				\
1556 	FIF_PSPOLL |				\
1557 	FIF_OTHER_BSS |				\
1558 	FIF_BCN_PRBRESP_PROMISC |		\
1559 	FIF_PROBE_REQ |				\
1560 	FIF_MCAST_ACTION |			\
1561 	FIF_FCSFAIL)
1562 
1563 /* FIXME: sc->sc_full_reset ? */
1564 static void ath9k_configure_filter(struct ieee80211_hw *hw,
1565 				   unsigned int changed_flags,
1566 				   unsigned int *total_flags,
1567 				   u64 multicast)
1568 {
1569 	struct ath_softc *sc = hw->priv;
1570 	struct ath_chanctx *ctx;
1571 	u32 rfilt;
1572 
1573 	*total_flags &= SUPPORTED_FILTERS;
1574 
1575 	spin_lock_bh(&sc->chan_lock);
1576 	ath_for_each_chanctx(sc, ctx)
1577 		ctx->rxfilter = *total_flags;
1578 #ifdef CONFIG_ATH9K_CHANNEL_CONTEXT
1579 	sc->offchannel.chan.rxfilter = *total_flags;
1580 #endif
1581 	spin_unlock_bh(&sc->chan_lock);
1582 
1583 	ath9k_ps_wakeup(sc);
1584 	rfilt = ath_calcrxfilter(sc);
1585 	ath9k_hw_setrxfilter(sc->sc_ah, rfilt);
1586 	ath9k_ps_restore(sc);
1587 
1588 	ath_dbg(ath9k_hw_common(sc->sc_ah), CONFIG, "Set HW RX filter: 0x%x\n",
1589 		rfilt);
1590 }
1591 
1592 static int ath9k_sta_add(struct ieee80211_hw *hw,
1593 			 struct ieee80211_vif *vif,
1594 			 struct ieee80211_sta *sta)
1595 {
1596 	struct ath_softc *sc = hw->priv;
1597 	struct ath_common *common = ath9k_hw_common(sc->sc_ah);
1598 	struct ath_node *an = (struct ath_node *) sta->drv_priv;
1599 	struct ieee80211_key_conf ps_key = { };
1600 	int key;
1601 
1602 	ath_node_attach(sc, sta, vif);
1603 
1604 	if (vif->type != NL80211_IFTYPE_AP &&
1605 	    vif->type != NL80211_IFTYPE_AP_VLAN)
1606 		return 0;
1607 
1608 	key = ath_key_config(common, vif, sta, &ps_key);
1609 	if (key > 0) {
1610 		an->ps_key = key;
1611 		an->key_idx[0] = key;
1612 	}
1613 
1614 	return 0;
1615 }
1616 
1617 static void ath9k_del_ps_key(struct ath_softc *sc,
1618 			     struct ieee80211_vif *vif,
1619 			     struct ieee80211_sta *sta)
1620 {
1621 	struct ath_common *common = ath9k_hw_common(sc->sc_ah);
1622 	struct ath_node *an = (struct ath_node *) sta->drv_priv;
1623 
1624 	if (!an->ps_key)
1625 	    return;
1626 
1627 	ath_key_delete(common, an->ps_key);
1628 	an->ps_key = 0;
1629 	an->key_idx[0] = 0;
1630 }
1631 
1632 static int ath9k_sta_remove(struct ieee80211_hw *hw,
1633 			    struct ieee80211_vif *vif,
1634 			    struct ieee80211_sta *sta)
1635 {
1636 	struct ath_softc *sc = hw->priv;
1637 
1638 	ath9k_del_ps_key(sc, vif, sta);
1639 	ath_node_detach(sc, sta);
1640 
1641 	return 0;
1642 }
1643 
1644 static int ath9k_sta_state(struct ieee80211_hw *hw,
1645 			   struct ieee80211_vif *vif,
1646 			   struct ieee80211_sta *sta,
1647 			   enum ieee80211_sta_state old_state,
1648 			   enum ieee80211_sta_state new_state)
1649 {
1650 	struct ath_softc *sc = hw->priv;
1651 	struct ath_common *common = ath9k_hw_common(sc->sc_ah);
1652 	int ret = 0;
1653 
1654 	if (old_state == IEEE80211_STA_NOTEXIST &&
1655 	    new_state == IEEE80211_STA_NONE) {
1656 		ret = ath9k_sta_add(hw, vif, sta);
1657 		ath_dbg(common, CONFIG,
1658 			"Add station: %pM\n", sta->addr);
1659 	} else if (old_state == IEEE80211_STA_NONE &&
1660 		   new_state == IEEE80211_STA_NOTEXIST) {
1661 		ret = ath9k_sta_remove(hw, vif, sta);
1662 		ath_dbg(common, CONFIG,
1663 			"Remove station: %pM\n", sta->addr);
1664 	}
1665 
1666 	if (ath9k_is_chanctx_enabled()) {
1667 		if (vif->type == NL80211_IFTYPE_STATION) {
1668 			if (old_state == IEEE80211_STA_ASSOC &&
1669 			    new_state == IEEE80211_STA_AUTHORIZED)
1670 				ath_chanctx_event(sc, vif,
1671 						  ATH_CHANCTX_EVENT_AUTHORIZED);
1672 		}
1673 	}
1674 
1675 	return ret;
1676 }
1677 
1678 static void ath9k_sta_set_tx_filter(struct ath_hw *ah,
1679 				    struct ath_node *an,
1680 				    bool set)
1681 {
1682 	int i;
1683 
1684 	for (i = 0; i < ARRAY_SIZE(an->key_idx); i++) {
1685 		if (!an->key_idx[i])
1686 			continue;
1687 		ath9k_hw_set_tx_filter(ah, an->key_idx[i], set);
1688 	}
1689 }
1690 
1691 static void ath9k_sta_notify(struct ieee80211_hw *hw,
1692 			 struct ieee80211_vif *vif,
1693 			 enum sta_notify_cmd cmd,
1694 			 struct ieee80211_sta *sta)
1695 {
1696 	struct ath_softc *sc = hw->priv;
1697 	struct ath_node *an = (struct ath_node *) sta->drv_priv;
1698 
1699 	switch (cmd) {
1700 	case STA_NOTIFY_SLEEP:
1701 		an->sleeping = true;
1702 		ath_tx_aggr_sleep(sta, sc, an);
1703 		ath9k_sta_set_tx_filter(sc->sc_ah, an, true);
1704 		break;
1705 	case STA_NOTIFY_AWAKE:
1706 		ath9k_sta_set_tx_filter(sc->sc_ah, an, false);
1707 		an->sleeping = false;
1708 		ath_tx_aggr_wakeup(sc, an);
1709 		break;
1710 	}
1711 }
1712 
1713 static int ath9k_conf_tx(struct ieee80211_hw *hw,
1714 			 struct ieee80211_vif *vif,
1715 			 unsigned int link_id, u16 queue,
1716 			 const struct ieee80211_tx_queue_params *params)
1717 {
1718 	struct ath_softc *sc = hw->priv;
1719 	struct ath_common *common = ath9k_hw_common(sc->sc_ah);
1720 	struct ath_txq *txq;
1721 	struct ath9k_tx_queue_info qi;
1722 	int ret = 0;
1723 
1724 	if (queue >= IEEE80211_NUM_ACS)
1725 		return 0;
1726 
1727 	txq = sc->tx.txq_map[queue];
1728 
1729 	ath9k_ps_wakeup(sc);
1730 	mutex_lock(&sc->mutex);
1731 
1732 	memset(&qi, 0, sizeof(struct ath9k_tx_queue_info));
1733 
1734 	qi.tqi_aifs = params->aifs;
1735 	qi.tqi_cwmin = params->cw_min;
1736 	qi.tqi_cwmax = params->cw_max;
1737 	qi.tqi_burstTime = params->txop * 32;
1738 
1739 	ath_dbg(common, CONFIG,
1740 		"Configure tx [queue/halq] [%d/%d], aifs: %d, cw_min: %d, cw_max: %d, txop: %d\n",
1741 		queue, txq->axq_qnum, params->aifs, params->cw_min,
1742 		params->cw_max, params->txop);
1743 
1744 	ath_update_max_aggr_framelen(sc, queue, qi.tqi_burstTime);
1745 	ret = ath_txq_update(sc, txq->axq_qnum, &qi);
1746 	if (ret)
1747 		ath_err(common, "TXQ Update failed\n");
1748 
1749 	mutex_unlock(&sc->mutex);
1750 	ath9k_ps_restore(sc);
1751 
1752 	return ret;
1753 }
1754 
1755 static int ath9k_set_key(struct ieee80211_hw *hw,
1756 			 enum set_key_cmd cmd,
1757 			 struct ieee80211_vif *vif,
1758 			 struct ieee80211_sta *sta,
1759 			 struct ieee80211_key_conf *key)
1760 {
1761 	struct ath_softc *sc = hw->priv;
1762 	struct ath_common *common = ath9k_hw_common(sc->sc_ah);
1763 	struct ath_node *an = NULL;
1764 	int ret = 0, i;
1765 
1766 	if (ath9k_modparam_nohwcrypt)
1767 		return -ENOSPC;
1768 
1769 	if ((vif->type == NL80211_IFTYPE_ADHOC ||
1770 	     vif->type == NL80211_IFTYPE_MESH_POINT) &&
1771 	    (key->cipher == WLAN_CIPHER_SUITE_TKIP ||
1772 	     key->cipher == WLAN_CIPHER_SUITE_CCMP) &&
1773 	    !(key->flags & IEEE80211_KEY_FLAG_PAIRWISE)) {
1774 		/*
1775 		 * For now, disable hw crypto for the RSN IBSS group keys. This
1776 		 * could be optimized in the future to use a modified key cache
1777 		 * design to support per-STA RX GTK, but until that gets
1778 		 * implemented, use of software crypto for group addressed
1779 		 * frames is a acceptable to allow RSN IBSS to be used.
1780 		 */
1781 		return -EOPNOTSUPP;
1782 	}
1783 
1784 	/* There may be MPDUs queued for the outgoing PTK key. Flush queues to
1785 	 * make sure these are not send unencrypted or with a wrong (new) key
1786 	 */
1787 	if (cmd == DISABLE_KEY && key->flags & IEEE80211_KEY_FLAG_PAIRWISE) {
1788 		ieee80211_stop_queues(hw);
1789 		ath9k_flush(hw, vif, 0, true);
1790 		ieee80211_wake_queues(hw);
1791 	}
1792 
1793 	mutex_lock(&sc->mutex);
1794 	ath9k_ps_wakeup(sc);
1795 	ath_dbg(common, CONFIG, "Set HW Key %d\n", cmd);
1796 	if (sta)
1797 		an = (struct ath_node *)sta->drv_priv;
1798 
1799 	/* Delete pending key cache entries if no more frames are pointing to
1800 	 * them in TXQs.
1801 	 */
1802 	for (i = 0; i < ATH_KEYMAX; i++)
1803 		ath9k_pending_key_del(sc, i);
1804 
1805 	switch (cmd) {
1806 	case SET_KEY:
1807 		if (sta)
1808 			ath9k_del_ps_key(sc, vif, sta);
1809 
1810 		key->hw_key_idx = 0;
1811 		ret = ath_key_config(common, vif, sta, key);
1812 		if (ret >= 0) {
1813 			key->hw_key_idx = ret;
1814 			/* push IV and Michael MIC generation to stack */
1815 			key->flags |= IEEE80211_KEY_FLAG_GENERATE_IV;
1816 			if (key->cipher == WLAN_CIPHER_SUITE_TKIP)
1817 				key->flags |= IEEE80211_KEY_FLAG_GENERATE_MMIC;
1818 			if (sc->sc_ah->sw_mgmt_crypto_tx &&
1819 			    key->cipher == WLAN_CIPHER_SUITE_CCMP)
1820 				key->flags |= IEEE80211_KEY_FLAG_SW_MGMT_TX;
1821 			ret = 0;
1822 		}
1823 		if (an && key->hw_key_idx) {
1824 			for (i = 0; i < ARRAY_SIZE(an->key_idx); i++) {
1825 				if (an->key_idx[i])
1826 					continue;
1827 				an->key_idx[i] = key->hw_key_idx;
1828 				break;
1829 			}
1830 			WARN_ON(i == ARRAY_SIZE(an->key_idx));
1831 		}
1832 		break;
1833 	case DISABLE_KEY:
1834 		if (ath9k_txq_has_key(sc, key->hw_key_idx)) {
1835 			/* Delay key cache entry deletion until there are no
1836 			 * remaining TXQ frames pointing to this entry.
1837 			 */
1838 			set_bit(key->hw_key_idx, sc->sc_ah->pending_del_keymap);
1839 			ath_hw_keysetmac(common, key->hw_key_idx, NULL);
1840 		} else {
1841 			ath_key_delete(common, key->hw_key_idx);
1842 		}
1843 		if (an) {
1844 			for (i = 0; i < ARRAY_SIZE(an->key_idx); i++) {
1845 				if (an->key_idx[i] != key->hw_key_idx)
1846 					continue;
1847 				an->key_idx[i] = 0;
1848 				break;
1849 			}
1850 		}
1851 		key->hw_key_idx = 0;
1852 		break;
1853 	default:
1854 		ret = -EINVAL;
1855 	}
1856 
1857 	ath9k_ps_restore(sc);
1858 	mutex_unlock(&sc->mutex);
1859 
1860 	return ret;
1861 }
1862 
1863 static void ath9k_bss_info_changed(struct ieee80211_hw *hw,
1864 				   struct ieee80211_vif *vif,
1865 				   struct ieee80211_bss_conf *bss_conf,
1866 				   u64 changed)
1867 {
1868 #define CHECK_ANI				\
1869 	(BSS_CHANGED_ASSOC |			\
1870 	 BSS_CHANGED_IBSS |			\
1871 	 BSS_CHANGED_BEACON_ENABLED)
1872 
1873 	struct ath_softc *sc = hw->priv;
1874 	struct ath_hw *ah = sc->sc_ah;
1875 	struct ath_common *common = ath9k_hw_common(ah);
1876 	struct ath_vif *avp = (void *)vif->drv_priv;
1877 	int slottime;
1878 
1879 	ath9k_ps_wakeup(sc);
1880 	mutex_lock(&sc->mutex);
1881 
1882 	if (changed & BSS_CHANGED_ASSOC) {
1883 		ath_dbg(common, CONFIG, "BSSID %pM Changed ASSOC %d\n",
1884 			bss_conf->bssid, vif->cfg.assoc);
1885 
1886 		memcpy(avp->bssid, bss_conf->bssid, ETH_ALEN);
1887 		avp->aid = vif->cfg.aid;
1888 		avp->assoc = vif->cfg.assoc;
1889 
1890 		ath9k_calculate_summary_state(sc, avp->chanctx);
1891 	}
1892 
1893 	if ((changed & BSS_CHANGED_IBSS) ||
1894 	      (changed & BSS_CHANGED_OCB)) {
1895 		memcpy(common->curbssid, bss_conf->bssid, ETH_ALEN);
1896 		common->curaid = vif->cfg.aid;
1897 		ath9k_hw_write_associd(sc->sc_ah);
1898 	}
1899 
1900 	if ((changed & BSS_CHANGED_BEACON_ENABLED) ||
1901 	    (changed & BSS_CHANGED_BEACON_INT) ||
1902 	    (changed & BSS_CHANGED_BEACON_INFO)) {
1903 		ath9k_calculate_summary_state(sc, avp->chanctx);
1904 	}
1905 
1906 	if ((avp->chanctx == sc->cur_chan) &&
1907 	    (changed & BSS_CHANGED_ERP_SLOT)) {
1908 		if (bss_conf->use_short_slot)
1909 			slottime = 9;
1910 		else
1911 			slottime = 20;
1912 
1913 		if (vif->type == NL80211_IFTYPE_AP) {
1914 			/*
1915 			 * Defer update, so that connected stations can adjust
1916 			 * their settings at the same time.
1917 			 * See beacon.c for more details
1918 			 */
1919 			sc->beacon.slottime = slottime;
1920 			sc->beacon.updateslot = UPDATE;
1921 		} else {
1922 			ah->slottime = slottime;
1923 			ath9k_hw_init_global_settings(ah);
1924 		}
1925 	}
1926 
1927 	if (changed & BSS_CHANGED_P2P_PS)
1928 		ath9k_p2p_bss_info_changed(sc, vif);
1929 
1930 	if (changed & CHECK_ANI)
1931 		ath_check_ani(sc);
1932 
1933 	if (changed & BSS_CHANGED_TXPOWER) {
1934 		ath_dbg(common, CONFIG, "vif %pM power %d dbm power_type %d\n",
1935 			vif->addr, bss_conf->txpower, bss_conf->txpower_type);
1936 		ath9k_set_txpower(sc, vif);
1937 	}
1938 
1939 	mutex_unlock(&sc->mutex);
1940 	ath9k_ps_restore(sc);
1941 
1942 #undef CHECK_ANI
1943 }
1944 
1945 static u64 ath9k_get_tsf(struct ieee80211_hw *hw, struct ieee80211_vif *vif)
1946 {
1947 	struct ath_softc *sc = hw->priv;
1948 	struct ath_vif *avp = (void *)vif->drv_priv;
1949 	u64 tsf;
1950 
1951 	mutex_lock(&sc->mutex);
1952 	ath9k_ps_wakeup(sc);
1953 	/* Get current TSF either from HW or kernel time. */
1954 	if (sc->cur_chan == avp->chanctx) {
1955 		tsf = ath9k_hw_gettsf64(sc->sc_ah);
1956 	} else {
1957 		tsf = sc->cur_chan->tsf_val +
1958 		      ath9k_hw_get_tsf_offset(&sc->cur_chan->tsf_ts, NULL);
1959 	}
1960 	tsf += le64_to_cpu(avp->tsf_adjust);
1961 	ath9k_ps_restore(sc);
1962 	mutex_unlock(&sc->mutex);
1963 
1964 	return tsf;
1965 }
1966 
1967 static void ath9k_set_tsf(struct ieee80211_hw *hw,
1968 			  struct ieee80211_vif *vif,
1969 			  u64 tsf)
1970 {
1971 	struct ath_softc *sc = hw->priv;
1972 	struct ath_vif *avp = (void *)vif->drv_priv;
1973 
1974 	mutex_lock(&sc->mutex);
1975 	ath9k_ps_wakeup(sc);
1976 	tsf -= le64_to_cpu(avp->tsf_adjust);
1977 	ktime_get_raw_ts64(&avp->chanctx->tsf_ts);
1978 	if (sc->cur_chan == avp->chanctx)
1979 		ath9k_hw_settsf64(sc->sc_ah, tsf);
1980 	avp->chanctx->tsf_val = tsf;
1981 	ath9k_ps_restore(sc);
1982 	mutex_unlock(&sc->mutex);
1983 }
1984 
1985 static void ath9k_reset_tsf(struct ieee80211_hw *hw, struct ieee80211_vif *vif)
1986 {
1987 	struct ath_softc *sc = hw->priv;
1988 	struct ath_vif *avp = (void *)vif->drv_priv;
1989 
1990 	mutex_lock(&sc->mutex);
1991 
1992 	ath9k_ps_wakeup(sc);
1993 	ktime_get_raw_ts64(&avp->chanctx->tsf_ts);
1994 	if (sc->cur_chan == avp->chanctx)
1995 		ath9k_hw_reset_tsf(sc->sc_ah);
1996 	avp->chanctx->tsf_val = 0;
1997 	ath9k_ps_restore(sc);
1998 
1999 	mutex_unlock(&sc->mutex);
2000 }
2001 
2002 static int ath9k_ampdu_action(struct ieee80211_hw *hw,
2003 			      struct ieee80211_vif *vif,
2004 			      struct ieee80211_ampdu_params *params)
2005 {
2006 	struct ath_softc *sc = hw->priv;
2007 	struct ath_common *common = ath9k_hw_common(sc->sc_ah);
2008 	bool flush = false;
2009 	int ret = 0;
2010 	struct ieee80211_sta *sta = params->sta;
2011 	struct ath_node *an = (struct ath_node *)sta->drv_priv;
2012 	enum ieee80211_ampdu_mlme_action action = params->action;
2013 	u16 tid = params->tid;
2014 	u16 *ssn = &params->ssn;
2015 	struct ath_atx_tid *atid;
2016 
2017 	mutex_lock(&sc->mutex);
2018 
2019 	switch (action) {
2020 	case IEEE80211_AMPDU_RX_START:
2021 		break;
2022 	case IEEE80211_AMPDU_RX_STOP:
2023 		break;
2024 	case IEEE80211_AMPDU_TX_START:
2025 		if (ath9k_is_chanctx_enabled()) {
2026 			if (test_bit(ATH_OP_SCANNING, &common->op_flags)) {
2027 				ret = -EBUSY;
2028 				break;
2029 			}
2030 		}
2031 		ath9k_ps_wakeup(sc);
2032 		ret = ath_tx_aggr_start(sc, sta, tid, ssn);
2033 		if (!ret)
2034 			ret = IEEE80211_AMPDU_TX_START_IMMEDIATE;
2035 		ath9k_ps_restore(sc);
2036 		break;
2037 	case IEEE80211_AMPDU_TX_STOP_FLUSH:
2038 	case IEEE80211_AMPDU_TX_STOP_FLUSH_CONT:
2039 		flush = true;
2040 		fallthrough;
2041 	case IEEE80211_AMPDU_TX_STOP_CONT:
2042 		ath9k_ps_wakeup(sc);
2043 		ath_tx_aggr_stop(sc, sta, tid);
2044 		if (!flush)
2045 			ieee80211_stop_tx_ba_cb_irqsafe(vif, sta->addr, tid);
2046 		ath9k_ps_restore(sc);
2047 		break;
2048 	case IEEE80211_AMPDU_TX_OPERATIONAL:
2049 		atid = ath_node_to_tid(an, tid);
2050 		atid->baw_size = IEEE80211_MIN_AMPDU_BUF <<
2051 					sta->deflink.ht_cap.ampdu_factor;
2052 		break;
2053 	default:
2054 		ath_err(ath9k_hw_common(sc->sc_ah), "Unknown AMPDU action\n");
2055 	}
2056 
2057 	mutex_unlock(&sc->mutex);
2058 
2059 	return ret;
2060 }
2061 
2062 static int ath9k_get_survey(struct ieee80211_hw *hw, int idx,
2063 			     struct survey_info *survey)
2064 {
2065 	struct ath_softc *sc = hw->priv;
2066 	struct ath_common *common = ath9k_hw_common(sc->sc_ah);
2067 	struct ieee80211_supported_band *sband;
2068 	struct ieee80211_channel *chan;
2069 	unsigned long flags;
2070 	int pos;
2071 
2072 	if (IS_ENABLED(CONFIG_ATH9K_TX99))
2073 		return -EOPNOTSUPP;
2074 
2075 	spin_lock_irqsave(&common->cc_lock, flags);
2076 	if (idx == 0)
2077 		ath_update_survey_stats(sc);
2078 
2079 	sband = hw->wiphy->bands[NL80211_BAND_2GHZ];
2080 	if (sband && idx >= sband->n_channels) {
2081 		idx -= sband->n_channels;
2082 		sband = NULL;
2083 	}
2084 
2085 	if (!sband)
2086 		sband = hw->wiphy->bands[NL80211_BAND_5GHZ];
2087 
2088 	if (!sband || idx >= sband->n_channels) {
2089 		spin_unlock_irqrestore(&common->cc_lock, flags);
2090 		return -ENOENT;
2091 	}
2092 
2093 	chan = &sband->channels[idx];
2094 	pos = chan->hw_value;
2095 	memcpy(survey, &sc->survey[pos], sizeof(*survey));
2096 	survey->channel = chan;
2097 	spin_unlock_irqrestore(&common->cc_lock, flags);
2098 
2099 	return 0;
2100 }
2101 
2102 static void ath9k_enable_dynack(struct ath_softc *sc)
2103 {
2104 #ifdef CONFIG_ATH9K_DYNACK
2105 	u32 rfilt;
2106 	struct ath_hw *ah = sc->sc_ah;
2107 
2108 	ath_dynack_reset(ah);
2109 
2110 	ah->dynack.enabled = true;
2111 	rfilt = ath_calcrxfilter(sc);
2112 	ath9k_hw_setrxfilter(ah, rfilt);
2113 #endif
2114 }
2115 
2116 static void ath9k_set_coverage_class(struct ieee80211_hw *hw,
2117 				     s16 coverage_class)
2118 {
2119 	struct ath_softc *sc = hw->priv;
2120 	struct ath_hw *ah = sc->sc_ah;
2121 
2122 	if (IS_ENABLED(CONFIG_ATH9K_TX99))
2123 		return;
2124 
2125 	mutex_lock(&sc->mutex);
2126 
2127 	if (coverage_class >= 0) {
2128 		ah->coverage_class = coverage_class;
2129 		if (ah->dynack.enabled) {
2130 			u32 rfilt;
2131 
2132 			ah->dynack.enabled = false;
2133 			rfilt = ath_calcrxfilter(sc);
2134 			ath9k_hw_setrxfilter(ah, rfilt);
2135 		}
2136 		ath9k_ps_wakeup(sc);
2137 		ath9k_hw_init_global_settings(ah);
2138 		ath9k_ps_restore(sc);
2139 	} else if (!ah->dynack.enabled) {
2140 		ath9k_enable_dynack(sc);
2141 	}
2142 
2143 	mutex_unlock(&sc->mutex);
2144 }
2145 
2146 static bool ath9k_has_tx_pending(struct ath_softc *sc,
2147 				 bool sw_pending)
2148 {
2149 	int i, npend = 0;
2150 
2151 	for (i = 0; i < ATH9K_NUM_TX_QUEUES; i++) {
2152 		if (!ATH_TXQ_SETUP(sc, i))
2153 			continue;
2154 
2155 		npend = ath9k_has_pending_frames(sc, &sc->tx.txq[i],
2156 						 sw_pending);
2157 		if (npend)
2158 			break;
2159 	}
2160 
2161 	return !!npend;
2162 }
2163 
2164 static void ath9k_flush(struct ieee80211_hw *hw, struct ieee80211_vif *vif,
2165 			u32 queues, bool drop)
2166 {
2167 	struct ath_softc *sc = hw->priv;
2168 	struct ath_common *common = ath9k_hw_common(sc->sc_ah);
2169 
2170 	if (ath9k_is_chanctx_enabled()) {
2171 		if (!test_bit(ATH_OP_MULTI_CHANNEL, &common->op_flags))
2172 			goto flush;
2173 
2174 		/*
2175 		 * If MCC is active, extend the flush timeout
2176 		 * and wait for the HW/SW queues to become
2177 		 * empty. This needs to be done outside the
2178 		 * sc->mutex lock to allow the channel scheduler
2179 		 * to switch channel contexts.
2180 		 *
2181 		 * The vif queues have been stopped in mac80211,
2182 		 * so there won't be any incoming frames.
2183 		 */
2184 		__ath9k_flush(hw, queues, drop, true, true);
2185 		return;
2186 	}
2187 flush:
2188 	mutex_lock(&sc->mutex);
2189 	__ath9k_flush(hw, queues, drop, true, false);
2190 	mutex_unlock(&sc->mutex);
2191 }
2192 
2193 void __ath9k_flush(struct ieee80211_hw *hw, u32 queues, bool drop,
2194 		   bool sw_pending, bool timeout_override)
2195 {
2196 	struct ath_softc *sc = hw->priv;
2197 	struct ath_hw *ah = sc->sc_ah;
2198 	struct ath_common *common = ath9k_hw_common(ah);
2199 	int timeout;
2200 	bool drain_txq;
2201 
2202 	cancel_delayed_work_sync(&sc->hw_check_work);
2203 
2204 	if (ah->ah_flags & AH_UNPLUGGED) {
2205 		ath_dbg(common, ANY, "Device has been unplugged!\n");
2206 		return;
2207 	}
2208 
2209 	if (test_bit(ATH_OP_INVALID, &common->op_flags)) {
2210 		ath_dbg(common, ANY, "Device not present\n");
2211 		return;
2212 	}
2213 
2214 	spin_lock_bh(&sc->chan_lock);
2215 	if (timeout_override)
2216 		timeout = HZ / 5;
2217 	else
2218 		timeout = sc->cur_chan->flush_timeout;
2219 	spin_unlock_bh(&sc->chan_lock);
2220 
2221 	ath_dbg(common, CHAN_CTX,
2222 		"Flush timeout: %d\n", jiffies_to_msecs(timeout));
2223 
2224 	if (wait_event_timeout(sc->tx_wait, !ath9k_has_tx_pending(sc, sw_pending),
2225 			       timeout) > 0)
2226 		drop = false;
2227 
2228 	if (drop) {
2229 		ath9k_ps_wakeup(sc);
2230 		spin_lock_bh(&sc->sc_pcu_lock);
2231 		drain_txq = ath_drain_all_txq(sc);
2232 		spin_unlock_bh(&sc->sc_pcu_lock);
2233 
2234 		if (!drain_txq)
2235 			ath_reset(sc, NULL);
2236 
2237 		ath9k_ps_restore(sc);
2238 	}
2239 
2240 	ieee80211_queue_delayed_work(hw, &sc->hw_check_work,
2241 				     msecs_to_jiffies(ATH_HW_CHECK_POLL_INT));
2242 }
2243 
2244 static bool ath9k_tx_frames_pending(struct ieee80211_hw *hw)
2245 {
2246 	struct ath_softc *sc = hw->priv;
2247 
2248 	return ath9k_has_tx_pending(sc, true);
2249 }
2250 
2251 static int ath9k_tx_last_beacon(struct ieee80211_hw *hw)
2252 {
2253 	struct ath_softc *sc = hw->priv;
2254 	struct ath_hw *ah = sc->sc_ah;
2255 	struct ieee80211_vif *vif;
2256 	struct ath_vif *avp;
2257 	struct ath_buf *bf;
2258 	struct ath_tx_status ts;
2259 	bool edma = !!(ah->caps.hw_caps & ATH9K_HW_CAP_EDMA);
2260 	int status;
2261 
2262 	vif = sc->beacon.bslot[0];
2263 	if (!vif)
2264 		return 0;
2265 
2266 	if (!vif->bss_conf.enable_beacon)
2267 		return 0;
2268 
2269 	avp = (void *)vif->drv_priv;
2270 
2271 	if (!sc->beacon.tx_processed && !edma) {
2272 		tasklet_disable(&sc->bcon_tasklet);
2273 
2274 		bf = avp->av_bcbuf;
2275 		if (!bf || !bf->bf_mpdu)
2276 			goto skip;
2277 
2278 		status = ath9k_hw_txprocdesc(ah, bf->bf_desc, &ts);
2279 		if (status == -EINPROGRESS)
2280 			goto skip;
2281 
2282 		sc->beacon.tx_processed = true;
2283 		sc->beacon.tx_last = !(ts.ts_status & ATH9K_TXERR_MASK);
2284 
2285 skip:
2286 		tasklet_enable(&sc->bcon_tasklet);
2287 	}
2288 
2289 	return sc->beacon.tx_last;
2290 }
2291 
2292 static int ath9k_get_stats(struct ieee80211_hw *hw,
2293 			   struct ieee80211_low_level_stats *stats)
2294 {
2295 	struct ath_softc *sc = hw->priv;
2296 	struct ath_hw *ah = sc->sc_ah;
2297 	struct ath9k_mib_stats *mib_stats = &ah->ah_mibStats;
2298 
2299 	stats->dot11ACKFailureCount = mib_stats->ackrcv_bad;
2300 	stats->dot11RTSFailureCount = mib_stats->rts_bad;
2301 	stats->dot11FCSErrorCount = mib_stats->fcs_bad;
2302 	stats->dot11RTSSuccessCount = mib_stats->rts_good;
2303 	return 0;
2304 }
2305 
2306 static u32 fill_chainmask(u32 cap, u32 new)
2307 {
2308 	u32 filled = 0;
2309 	int i;
2310 
2311 	for (i = 0; cap && new; i++, cap >>= 1) {
2312 		if (!(cap & BIT(0)))
2313 			continue;
2314 
2315 		if (new & BIT(0))
2316 			filled |= BIT(i);
2317 
2318 		new >>= 1;
2319 	}
2320 
2321 	return filled;
2322 }
2323 
2324 static bool validate_antenna_mask(struct ath_hw *ah, u32 val)
2325 {
2326 	if (AR_SREV_9300_20_OR_LATER(ah))
2327 		return true;
2328 
2329 	switch (val & 0x7) {
2330 	case 0x1:
2331 	case 0x3:
2332 	case 0x7:
2333 		return true;
2334 	case 0x2:
2335 		return (ah->caps.rx_chainmask == 1);
2336 	default:
2337 		return false;
2338 	}
2339 }
2340 
2341 static int ath9k_set_antenna(struct ieee80211_hw *hw, u32 tx_ant, u32 rx_ant)
2342 {
2343 	struct ath_softc *sc = hw->priv;
2344 	struct ath_hw *ah = sc->sc_ah;
2345 
2346 	if (ah->caps.rx_chainmask != 1)
2347 		rx_ant |= tx_ant;
2348 
2349 	if (!validate_antenna_mask(ah, rx_ant) || !tx_ant)
2350 		return -EINVAL;
2351 
2352 	sc->ant_rx = rx_ant;
2353 	sc->ant_tx = tx_ant;
2354 
2355 	if (ah->caps.rx_chainmask == 1)
2356 		return 0;
2357 
2358 	/* AR9100 runs into calibration issues if not all rx chains are enabled */
2359 	if (AR_SREV_9100(ah))
2360 		ah->rxchainmask = 0x7;
2361 	else
2362 		ah->rxchainmask = fill_chainmask(ah->caps.rx_chainmask, rx_ant);
2363 
2364 	ah->txchainmask = fill_chainmask(ah->caps.tx_chainmask, tx_ant);
2365 	ath9k_cmn_reload_chainmask(ah);
2366 
2367 	return 0;
2368 }
2369 
2370 static int ath9k_get_antenna(struct ieee80211_hw *hw, u32 *tx_ant, u32 *rx_ant)
2371 {
2372 	struct ath_softc *sc = hw->priv;
2373 
2374 	*tx_ant = sc->ant_tx;
2375 	*rx_ant = sc->ant_rx;
2376 	return 0;
2377 }
2378 
2379 static void ath9k_sw_scan_start(struct ieee80211_hw *hw,
2380 				struct ieee80211_vif *vif,
2381 				const u8 *mac_addr)
2382 {
2383 	struct ath_softc *sc = hw->priv;
2384 	struct ath_common *common = ath9k_hw_common(sc->sc_ah);
2385 	set_bit(ATH_OP_SCANNING, &common->op_flags);
2386 }
2387 
2388 static void ath9k_sw_scan_complete(struct ieee80211_hw *hw,
2389 				   struct ieee80211_vif *vif)
2390 {
2391 	struct ath_softc *sc = hw->priv;
2392 	struct ath_common *common = ath9k_hw_common(sc->sc_ah);
2393 	clear_bit(ATH_OP_SCANNING, &common->op_flags);
2394 }
2395 
2396 #ifdef CONFIG_ATH9K_CHANNEL_CONTEXT
2397 
2398 static void ath9k_cancel_pending_offchannel(struct ath_softc *sc)
2399 {
2400 	struct ath_common *common = ath9k_hw_common(sc->sc_ah);
2401 
2402 	if (sc->offchannel.roc_vif) {
2403 		ath_dbg(common, CHAN_CTX,
2404 			"%s: Aborting RoC\n", __func__);
2405 
2406 		del_timer_sync(&sc->offchannel.timer);
2407 		if (sc->offchannel.state >= ATH_OFFCHANNEL_ROC_START)
2408 			ath_roc_complete(sc, ATH_ROC_COMPLETE_ABORT);
2409 	}
2410 
2411 	if (test_bit(ATH_OP_SCANNING, &common->op_flags)) {
2412 		ath_dbg(common, CHAN_CTX,
2413 			"%s: Aborting HW scan\n", __func__);
2414 
2415 		del_timer_sync(&sc->offchannel.timer);
2416 		ath_scan_complete(sc, true);
2417 	}
2418 }
2419 
2420 static int ath9k_hw_scan(struct ieee80211_hw *hw, struct ieee80211_vif *vif,
2421 			 struct ieee80211_scan_request *hw_req)
2422 {
2423 	struct cfg80211_scan_request *req = &hw_req->req;
2424 	struct ath_softc *sc = hw->priv;
2425 	struct ath_common *common = ath9k_hw_common(sc->sc_ah);
2426 	int ret = 0;
2427 
2428 	mutex_lock(&sc->mutex);
2429 
2430 	if (WARN_ON(sc->offchannel.scan_req)) {
2431 		ret = -EBUSY;
2432 		goto out;
2433 	}
2434 
2435 	ath9k_ps_wakeup(sc);
2436 	set_bit(ATH_OP_SCANNING, &common->op_flags);
2437 	sc->offchannel.scan_vif = vif;
2438 	sc->offchannel.scan_req = req;
2439 	sc->offchannel.scan_idx = 0;
2440 
2441 	ath_dbg(common, CHAN_CTX, "HW scan request received on vif: %pM\n",
2442 		vif->addr);
2443 
2444 	if (sc->offchannel.state == ATH_OFFCHANNEL_IDLE) {
2445 		ath_dbg(common, CHAN_CTX, "Starting HW scan\n");
2446 		ath_offchannel_next(sc);
2447 	}
2448 
2449 out:
2450 	mutex_unlock(&sc->mutex);
2451 
2452 	return ret;
2453 }
2454 
2455 static void ath9k_cancel_hw_scan(struct ieee80211_hw *hw,
2456 				 struct ieee80211_vif *vif)
2457 {
2458 	struct ath_softc *sc = hw->priv;
2459 	struct ath_common *common = ath9k_hw_common(sc->sc_ah);
2460 
2461 	ath_dbg(common, CHAN_CTX, "Cancel HW scan on vif: %pM\n", vif->addr);
2462 
2463 	mutex_lock(&sc->mutex);
2464 	del_timer_sync(&sc->offchannel.timer);
2465 	ath_scan_complete(sc, true);
2466 	mutex_unlock(&sc->mutex);
2467 }
2468 
2469 static int ath9k_remain_on_channel(struct ieee80211_hw *hw,
2470 				   struct ieee80211_vif *vif,
2471 				   struct ieee80211_channel *chan, int duration,
2472 				   enum ieee80211_roc_type type)
2473 {
2474 	struct ath_softc *sc = hw->priv;
2475 	struct ath_common *common = ath9k_hw_common(sc->sc_ah);
2476 	int ret = 0;
2477 
2478 	mutex_lock(&sc->mutex);
2479 
2480 	if (WARN_ON(sc->offchannel.roc_vif)) {
2481 		ret = -EBUSY;
2482 		goto out;
2483 	}
2484 
2485 	ath9k_ps_wakeup(sc);
2486 	sc->offchannel.roc_vif = vif;
2487 	sc->offchannel.roc_chan = chan;
2488 	sc->offchannel.roc_duration = duration;
2489 
2490 	ath_dbg(common, CHAN_CTX,
2491 		"RoC request on vif: %pM, type: %d duration: %d\n",
2492 		vif->addr, type, duration);
2493 
2494 	if (sc->offchannel.state == ATH_OFFCHANNEL_IDLE) {
2495 		ath_dbg(common, CHAN_CTX, "Starting RoC period\n");
2496 		ath_offchannel_next(sc);
2497 	}
2498 
2499 out:
2500 	mutex_unlock(&sc->mutex);
2501 
2502 	return ret;
2503 }
2504 
2505 static int ath9k_cancel_remain_on_channel(struct ieee80211_hw *hw,
2506 					  struct ieee80211_vif *vif)
2507 {
2508 	struct ath_softc *sc = hw->priv;
2509 	struct ath_common *common = ath9k_hw_common(sc->sc_ah);
2510 
2511 	mutex_lock(&sc->mutex);
2512 
2513 	ath_dbg(common, CHAN_CTX, "Cancel RoC\n");
2514 	del_timer_sync(&sc->offchannel.timer);
2515 
2516 	if (sc->offchannel.roc_vif) {
2517 		if (sc->offchannel.state >= ATH_OFFCHANNEL_ROC_START)
2518 			ath_roc_complete(sc, ATH_ROC_COMPLETE_CANCEL);
2519 	}
2520 
2521 	mutex_unlock(&sc->mutex);
2522 
2523 	return 0;
2524 }
2525 
2526 static int ath9k_add_chanctx(struct ieee80211_hw *hw,
2527 			     struct ieee80211_chanctx_conf *conf)
2528 {
2529 	struct ath_softc *sc = hw->priv;
2530 	struct ath_common *common = ath9k_hw_common(sc->sc_ah);
2531 	struct ath_chanctx *ctx, **ptr;
2532 	int pos;
2533 
2534 	mutex_lock(&sc->mutex);
2535 
2536 	ath_for_each_chanctx(sc, ctx) {
2537 		if (ctx->assigned)
2538 			continue;
2539 
2540 		ptr = (void *) conf->drv_priv;
2541 		*ptr = ctx;
2542 		ctx->assigned = true;
2543 		pos = ctx - &sc->chanctx[0];
2544 		ctx->hw_queue_base = pos * IEEE80211_NUM_ACS;
2545 
2546 		ath_dbg(common, CHAN_CTX,
2547 			"Add channel context: %d MHz\n",
2548 			conf->def.chan->center_freq);
2549 
2550 		ath_chanctx_set_channel(sc, ctx, &conf->def);
2551 
2552 		mutex_unlock(&sc->mutex);
2553 		return 0;
2554 	}
2555 
2556 	mutex_unlock(&sc->mutex);
2557 	return -ENOSPC;
2558 }
2559 
2560 
2561 static void ath9k_remove_chanctx(struct ieee80211_hw *hw,
2562 				 struct ieee80211_chanctx_conf *conf)
2563 {
2564 	struct ath_softc *sc = hw->priv;
2565 	struct ath_common *common = ath9k_hw_common(sc->sc_ah);
2566 	struct ath_chanctx *ctx = ath_chanctx_get(conf);
2567 
2568 	mutex_lock(&sc->mutex);
2569 
2570 	ath_dbg(common, CHAN_CTX,
2571 		"Remove channel context: %d MHz\n",
2572 		conf->def.chan->center_freq);
2573 
2574 	ctx->assigned = false;
2575 	ctx->hw_queue_base = 0;
2576 	ath_chanctx_event(sc, NULL, ATH_CHANCTX_EVENT_UNASSIGN);
2577 
2578 	mutex_unlock(&sc->mutex);
2579 }
2580 
2581 static void ath9k_change_chanctx(struct ieee80211_hw *hw,
2582 				 struct ieee80211_chanctx_conf *conf,
2583 				 u32 changed)
2584 {
2585 	struct ath_softc *sc = hw->priv;
2586 	struct ath_common *common = ath9k_hw_common(sc->sc_ah);
2587 	struct ath_chanctx *ctx = ath_chanctx_get(conf);
2588 
2589 	mutex_lock(&sc->mutex);
2590 	ath_dbg(common, CHAN_CTX,
2591 		"Change channel context: %d MHz\n",
2592 		conf->def.chan->center_freq);
2593 	ath_chanctx_set_channel(sc, ctx, &conf->def);
2594 	mutex_unlock(&sc->mutex);
2595 }
2596 
2597 static int ath9k_assign_vif_chanctx(struct ieee80211_hw *hw,
2598 				    struct ieee80211_vif *vif,
2599 				    struct ieee80211_bss_conf *link_conf,
2600 				    struct ieee80211_chanctx_conf *conf)
2601 {
2602 	struct ath_softc *sc = hw->priv;
2603 	struct ath_common *common = ath9k_hw_common(sc->sc_ah);
2604 	struct ath_vif *avp = (void *)vif->drv_priv;
2605 	struct ath_chanctx *ctx = ath_chanctx_get(conf);
2606 	int i;
2607 
2608 	ath9k_cancel_pending_offchannel(sc);
2609 
2610 	mutex_lock(&sc->mutex);
2611 
2612 	ath_dbg(common, CHAN_CTX,
2613 		"Assign VIF (addr: %pM, type: %d, p2p: %d) to channel context: %d MHz\n",
2614 		vif->addr, vif->type, vif->p2p,
2615 		conf->def.chan->center_freq);
2616 
2617 	avp->chanctx = ctx;
2618 	ctx->nvifs_assigned++;
2619 	list_add_tail(&avp->list, &ctx->vifs);
2620 	ath9k_calculate_summary_state(sc, ctx);
2621 	for (i = 0; i < IEEE80211_NUM_ACS; i++)
2622 		vif->hw_queue[i] = ctx->hw_queue_base + i;
2623 
2624 	mutex_unlock(&sc->mutex);
2625 
2626 	return 0;
2627 }
2628 
2629 static void ath9k_unassign_vif_chanctx(struct ieee80211_hw *hw,
2630 				       struct ieee80211_vif *vif,
2631 				       struct ieee80211_bss_conf *link_conf,
2632 				       struct ieee80211_chanctx_conf *conf)
2633 {
2634 	struct ath_softc *sc = hw->priv;
2635 	struct ath_common *common = ath9k_hw_common(sc->sc_ah);
2636 	struct ath_vif *avp = (void *)vif->drv_priv;
2637 	struct ath_chanctx *ctx = ath_chanctx_get(conf);
2638 	int ac;
2639 
2640 	ath9k_cancel_pending_offchannel(sc);
2641 
2642 	mutex_lock(&sc->mutex);
2643 
2644 	ath_dbg(common, CHAN_CTX,
2645 		"Remove VIF (addr: %pM, type: %d, p2p: %d) from channel context: %d MHz\n",
2646 		vif->addr, vif->type, vif->p2p,
2647 		conf->def.chan->center_freq);
2648 
2649 	avp->chanctx = NULL;
2650 	ctx->nvifs_assigned--;
2651 	list_del(&avp->list);
2652 	ath9k_calculate_summary_state(sc, ctx);
2653 	for (ac = 0; ac < IEEE80211_NUM_ACS; ac++)
2654 		vif->hw_queue[ac] = IEEE80211_INVAL_HW_QUEUE;
2655 
2656 	mutex_unlock(&sc->mutex);
2657 }
2658 
2659 static void ath9k_mgd_prepare_tx(struct ieee80211_hw *hw,
2660 				 struct ieee80211_vif *vif,
2661 				 struct ieee80211_prep_tx_info *info)
2662 {
2663 	struct ath_softc *sc = hw->priv;
2664 	struct ath_common *common = ath9k_hw_common(sc->sc_ah);
2665 	struct ath_vif *avp = (struct ath_vif *) vif->drv_priv;
2666 	struct ath_beacon_config *cur_conf;
2667 	struct ath_chanctx *go_ctx;
2668 	unsigned long timeout;
2669 	bool changed = false;
2670 	u32 beacon_int;
2671 
2672 	if (!test_bit(ATH_OP_MULTI_CHANNEL, &common->op_flags))
2673 		return;
2674 
2675 	if (!avp->chanctx)
2676 		return;
2677 
2678 	mutex_lock(&sc->mutex);
2679 
2680 	spin_lock_bh(&sc->chan_lock);
2681 	if (sc->next_chan || (sc->cur_chan != avp->chanctx))
2682 		changed = true;
2683 	spin_unlock_bh(&sc->chan_lock);
2684 
2685 	if (!changed)
2686 		goto out;
2687 
2688 	ath9k_cancel_pending_offchannel(sc);
2689 
2690 	go_ctx = ath_is_go_chanctx_present(sc);
2691 
2692 	if (go_ctx) {
2693 		/*
2694 		 * Wait till the GO interface gets a chance
2695 		 * to send out an NoA.
2696 		 */
2697 		spin_lock_bh(&sc->chan_lock);
2698 		sc->sched.mgd_prepare_tx = true;
2699 		cur_conf = &go_ctx->beacon;
2700 		beacon_int = TU_TO_USEC(cur_conf->beacon_interval);
2701 		spin_unlock_bh(&sc->chan_lock);
2702 
2703 		timeout = usecs_to_jiffies(beacon_int * 2);
2704 		init_completion(&sc->go_beacon);
2705 
2706 		mutex_unlock(&sc->mutex);
2707 
2708 		if (wait_for_completion_timeout(&sc->go_beacon,
2709 						timeout) == 0) {
2710 			ath_dbg(common, CHAN_CTX,
2711 				"Failed to send new NoA\n");
2712 
2713 			spin_lock_bh(&sc->chan_lock);
2714 			sc->sched.mgd_prepare_tx = false;
2715 			spin_unlock_bh(&sc->chan_lock);
2716 		}
2717 
2718 		mutex_lock(&sc->mutex);
2719 	}
2720 
2721 	ath_dbg(common, CHAN_CTX,
2722 		"%s: Set chanctx state to FORCE_ACTIVE for vif: %pM\n",
2723 		__func__, vif->addr);
2724 
2725 	spin_lock_bh(&sc->chan_lock);
2726 	sc->next_chan = avp->chanctx;
2727 	sc->sched.state = ATH_CHANCTX_STATE_FORCE_ACTIVE;
2728 	spin_unlock_bh(&sc->chan_lock);
2729 
2730 	ath_chanctx_set_next(sc, true);
2731 out:
2732 	mutex_unlock(&sc->mutex);
2733 }
2734 
2735 void ath9k_fill_chanctx_ops(void)
2736 {
2737 	if (!ath9k_is_chanctx_enabled())
2738 		return;
2739 
2740 	ath9k_ops.hw_scan                  = ath9k_hw_scan;
2741 	ath9k_ops.cancel_hw_scan           = ath9k_cancel_hw_scan;
2742 	ath9k_ops.remain_on_channel        = ath9k_remain_on_channel;
2743 	ath9k_ops.cancel_remain_on_channel = ath9k_cancel_remain_on_channel;
2744 	ath9k_ops.add_chanctx              = ath9k_add_chanctx;
2745 	ath9k_ops.remove_chanctx           = ath9k_remove_chanctx;
2746 	ath9k_ops.change_chanctx           = ath9k_change_chanctx;
2747 	ath9k_ops.assign_vif_chanctx       = ath9k_assign_vif_chanctx;
2748 	ath9k_ops.unassign_vif_chanctx     = ath9k_unassign_vif_chanctx;
2749 	ath9k_ops.mgd_prepare_tx           = ath9k_mgd_prepare_tx;
2750 }
2751 
2752 #endif
2753 
2754 static int ath9k_get_txpower(struct ieee80211_hw *hw, struct ieee80211_vif *vif,
2755 			     int *dbm)
2756 {
2757 	struct ath_softc *sc = hw->priv;
2758 	struct ath_vif *avp = (void *)vif->drv_priv;
2759 
2760 	mutex_lock(&sc->mutex);
2761 	if (avp->chanctx)
2762 		*dbm = avp->chanctx->cur_txpower;
2763 	else
2764 		*dbm = sc->cur_chan->cur_txpower;
2765 	mutex_unlock(&sc->mutex);
2766 
2767 	*dbm /= 2;
2768 
2769 	return 0;
2770 }
2771 
2772 struct ieee80211_ops ath9k_ops = {
2773 	.tx 		    = ath9k_tx,
2774 	.start 		    = ath9k_start,
2775 	.stop 		    = ath9k_stop,
2776 	.add_interface 	    = ath9k_add_interface,
2777 	.change_interface   = ath9k_change_interface,
2778 	.remove_interface   = ath9k_remove_interface,
2779 	.config 	    = ath9k_config,
2780 	.configure_filter   = ath9k_configure_filter,
2781 	.sta_state          = ath9k_sta_state,
2782 	.sta_notify         = ath9k_sta_notify,
2783 	.conf_tx 	    = ath9k_conf_tx,
2784 	.bss_info_changed   = ath9k_bss_info_changed,
2785 	.set_key            = ath9k_set_key,
2786 	.get_tsf 	    = ath9k_get_tsf,
2787 	.set_tsf 	    = ath9k_set_tsf,
2788 	.reset_tsf 	    = ath9k_reset_tsf,
2789 	.ampdu_action       = ath9k_ampdu_action,
2790 	.get_survey	    = ath9k_get_survey,
2791 	.rfkill_poll        = ath9k_rfkill_poll_state,
2792 	.set_coverage_class = ath9k_set_coverage_class,
2793 	.flush		    = ath9k_flush,
2794 	.tx_frames_pending  = ath9k_tx_frames_pending,
2795 	.tx_last_beacon     = ath9k_tx_last_beacon,
2796 	.release_buffered_frames = ath9k_release_buffered_frames,
2797 	.get_stats	    = ath9k_get_stats,
2798 	.set_antenna	    = ath9k_set_antenna,
2799 	.get_antenna	    = ath9k_get_antenna,
2800 
2801 #ifdef CONFIG_ATH9K_WOW
2802 	.suspend	    = ath9k_suspend,
2803 	.resume		    = ath9k_resume,
2804 	.set_wakeup	    = ath9k_set_wakeup,
2805 #endif
2806 
2807 #ifdef CONFIG_ATH9K_DEBUGFS
2808 	.get_et_sset_count  = ath9k_get_et_sset_count,
2809 	.get_et_stats       = ath9k_get_et_stats,
2810 	.get_et_strings     = ath9k_get_et_strings,
2811 #endif
2812 
2813 #if defined(CONFIG_MAC80211_DEBUGFS) && defined(CONFIG_ATH9K_STATION_STATISTICS)
2814 	.sta_add_debugfs    = ath9k_sta_add_debugfs,
2815 #endif
2816 	.sw_scan_start	    = ath9k_sw_scan_start,
2817 	.sw_scan_complete   = ath9k_sw_scan_complete,
2818 	.get_txpower        = ath9k_get_txpower,
2819 	.wake_tx_queue      = ath9k_wake_tx_queue,
2820 };
2821