]> Pileus Git - ~andy/linux/blob - drivers/net/wireless/ath/ath9k/main.c
ath9k: use a timer to put hardware into full sleep
[~andy/linux] / drivers / net / wireless / ath / ath9k / main.c
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_set_assoc_state(struct ath_softc *sc,
23                                   struct ieee80211_vif *vif);
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 {
62         bool pending = false;
63
64         spin_lock_bh(&txq->axq_lock);
65
66         if (txq->axq_depth || !list_empty(&txq->axq_acq))
67                 pending = true;
68
69         spin_unlock_bh(&txq->axq_lock);
70         return pending;
71 }
72
73 static bool ath9k_setpower(struct ath_softc *sc, enum ath9k_power_mode mode)
74 {
75         unsigned long flags;
76         bool ret;
77
78         spin_lock_irqsave(&sc->sc_pm_lock, flags);
79         ret = ath9k_hw_setpower(sc->sc_ah, mode);
80         spin_unlock_irqrestore(&sc->sc_pm_lock, flags);
81
82         return ret;
83 }
84
85 void ath_ps_full_sleep(unsigned long data)
86 {
87         struct ath_softc *sc = (struct ath_softc *) data;
88         struct ath_common *common = ath9k_hw_common(sc->sc_ah);
89         bool reset;
90
91         spin_lock(&common->cc_lock);
92         ath_hw_cycle_counters_update(common);
93         spin_unlock(&common->cc_lock);
94
95         ath9k_hw_setrxabort(sc->sc_ah, 1);
96         ath9k_hw_stopdmarecv(sc->sc_ah, &reset);
97
98         ath9k_hw_setpower(sc->sc_ah, ATH9K_PM_FULL_SLEEP);
99 }
100
101 void ath9k_ps_wakeup(struct ath_softc *sc)
102 {
103         struct ath_common *common = ath9k_hw_common(sc->sc_ah);
104         unsigned long flags;
105         enum ath9k_power_mode power_mode;
106
107         spin_lock_irqsave(&sc->sc_pm_lock, flags);
108         if (++sc->ps_usecount != 1)
109                 goto unlock;
110
111         del_timer_sync(&sc->sleep_timer);
112         power_mode = sc->sc_ah->power_mode;
113         ath9k_hw_setpower(sc->sc_ah, ATH9K_PM_AWAKE);
114
115         /*
116          * While the hardware is asleep, the cycle counters contain no
117          * useful data. Better clear them now so that they don't mess up
118          * survey data results.
119          */
120         if (power_mode != ATH9K_PM_AWAKE) {
121                 spin_lock(&common->cc_lock);
122                 ath_hw_cycle_counters_update(common);
123                 memset(&common->cc_survey, 0, sizeof(common->cc_survey));
124                 memset(&common->cc_ani, 0, sizeof(common->cc_ani));
125                 spin_unlock(&common->cc_lock);
126         }
127
128  unlock:
129         spin_unlock_irqrestore(&sc->sc_pm_lock, flags);
130 }
131
132 void ath9k_ps_restore(struct ath_softc *sc)
133 {
134         struct ath_common *common = ath9k_hw_common(sc->sc_ah);
135         enum ath9k_power_mode mode;
136         unsigned long flags;
137
138         spin_lock_irqsave(&sc->sc_pm_lock, flags);
139         if (--sc->ps_usecount != 0)
140                 goto unlock;
141
142         if (sc->ps_idle) {
143                 mod_timer(&sc->sleep_timer, jiffies + HZ / 10);
144                 goto unlock;
145         }
146
147         if (sc->ps_enabled &&
148                    !(sc->ps_flags & (PS_WAIT_FOR_BEACON |
149                                      PS_WAIT_FOR_CAB |
150                                      PS_WAIT_FOR_PSPOLL_DATA |
151                                      PS_WAIT_FOR_TX_ACK |
152                                      PS_WAIT_FOR_ANI))) {
153                 mode = ATH9K_PM_NETWORK_SLEEP;
154                 if (ath9k_hw_btcoex_is_enabled(sc->sc_ah))
155                         ath9k_btcoex_stop_gen_timer(sc);
156         } else {
157                 goto unlock;
158         }
159
160         spin_lock(&common->cc_lock);
161         ath_hw_cycle_counters_update(common);
162         spin_unlock(&common->cc_lock);
163
164         ath9k_hw_setpower(sc->sc_ah, mode);
165
166  unlock:
167         spin_unlock_irqrestore(&sc->sc_pm_lock, flags);
168 }
169
170 static void __ath_cancel_work(struct ath_softc *sc)
171 {
172         cancel_work_sync(&sc->paprd_work);
173         cancel_work_sync(&sc->hw_check_work);
174         cancel_delayed_work_sync(&sc->tx_complete_work);
175         cancel_delayed_work_sync(&sc->hw_pll_work);
176
177 #ifdef CONFIG_ATH9K_BTCOEX_SUPPORT
178         if (ath9k_hw_mci_is_enabled(sc->sc_ah))
179                 cancel_work_sync(&sc->mci_work);
180 #endif
181 }
182
183 void ath_cancel_work(struct ath_softc *sc)
184 {
185         __ath_cancel_work(sc);
186         cancel_work_sync(&sc->hw_reset_work);
187 }
188
189 void ath_restart_work(struct ath_softc *sc)
190 {
191         ieee80211_queue_delayed_work(sc->hw, &sc->tx_complete_work, 0);
192
193         if (AR_SREV_9340(sc->sc_ah) || AR_SREV_9330(sc->sc_ah))
194                 ieee80211_queue_delayed_work(sc->hw, &sc->hw_pll_work,
195                                      msecs_to_jiffies(ATH_PLL_WORK_INTERVAL));
196
197         ath_start_rx_poll(sc, 3);
198         ath_start_ani(sc);
199 }
200
201 static bool ath_prepare_reset(struct ath_softc *sc)
202 {
203         struct ath_hw *ah = sc->sc_ah;
204         bool ret = true;
205
206         ieee80211_stop_queues(sc->hw);
207
208         sc->hw_busy_count = 0;
209         ath_stop_ani(sc);
210         del_timer_sync(&sc->rx_poll_timer);
211
212         ath9k_hw_disable_interrupts(ah);
213
214         if (!ath_drain_all_txq(sc))
215                 ret = false;
216
217         if (!ath_stoprecv(sc))
218                 ret = false;
219
220         return ret;
221 }
222
223 static bool ath_complete_reset(struct ath_softc *sc, bool start)
224 {
225         struct ath_hw *ah = sc->sc_ah;
226         struct ath_common *common = ath9k_hw_common(ah);
227         unsigned long flags;
228         int i;
229
230         if (ath_startrecv(sc) != 0) {
231                 ath_err(common, "Unable to restart recv logic\n");
232                 return false;
233         }
234
235         ath9k_cmn_update_txpow(ah, sc->curtxpow,
236                                sc->config.txpowlimit, &sc->curtxpow);
237
238         clear_bit(SC_OP_HW_RESET, &sc->sc_flags);
239         ath9k_hw_set_interrupts(ah);
240         ath9k_hw_enable_interrupts(ah);
241
242         if (!(sc->hw->conf.flags & IEEE80211_CONF_OFFCHANNEL) && start) {
243                 if (!test_bit(SC_OP_BEACONS, &sc->sc_flags))
244                         goto work;
245
246                 if (ah->opmode == NL80211_IFTYPE_STATION &&
247                     test_bit(SC_OP_PRIM_STA_VIF, &sc->sc_flags)) {
248                         spin_lock_irqsave(&sc->sc_pm_lock, flags);
249                         sc->ps_flags |= PS_BEACON_SYNC | PS_WAIT_FOR_BEACON;
250                         spin_unlock_irqrestore(&sc->sc_pm_lock, flags);
251                 } else {
252                         ath9k_set_beacon(sc);
253                 }
254         work:
255                 ath_restart_work(sc);
256
257                 for (i = 0; i < ATH9K_NUM_TX_QUEUES; i++) {
258                         if (!ATH_TXQ_SETUP(sc, i))
259                                 continue;
260
261                         spin_lock_bh(&sc->tx.txq[i].axq_lock);
262                         ath_txq_schedule(sc, &sc->tx.txq[i]);
263                         spin_unlock_bh(&sc->tx.txq[i].axq_lock);
264                 }
265         }
266
267         ieee80211_wake_queues(sc->hw);
268
269         return true;
270 }
271
272 static int ath_reset_internal(struct ath_softc *sc, struct ath9k_channel *hchan)
273 {
274         struct ath_hw *ah = sc->sc_ah;
275         struct ath_common *common = ath9k_hw_common(ah);
276         struct ath9k_hw_cal_data *caldata = NULL;
277         bool fastcc = true;
278         int r;
279
280         __ath_cancel_work(sc);
281
282         tasklet_disable(&sc->intr_tq);
283         spin_lock_bh(&sc->sc_pcu_lock);
284
285         if (!(sc->hw->conf.flags & IEEE80211_CONF_OFFCHANNEL)) {
286                 fastcc = false;
287                 caldata = &sc->caldata;
288         }
289
290         if (!hchan) {
291                 fastcc = false;
292                 hchan = ah->curchan;
293         }
294
295         if (!ath_prepare_reset(sc))
296                 fastcc = false;
297
298         ath_dbg(common, CONFIG, "Reset to %u MHz, HT40: %d fastcc: %d\n",
299                 hchan->channel, IS_CHAN_HT40(hchan), fastcc);
300
301         r = ath9k_hw_reset(ah, hchan, caldata, fastcc);
302         if (r) {
303                 ath_err(common,
304                         "Unable to reset channel, reset status %d\n", r);
305
306                 ath9k_hw_enable_interrupts(ah);
307                 ath9k_queue_reset(sc, RESET_TYPE_BB_HANG);
308
309                 goto out;
310         }
311
312         if (ath9k_hw_mci_is_enabled(sc->sc_ah) &&
313             (sc->hw->conf.flags & IEEE80211_CONF_OFFCHANNEL))
314                 ath9k_mci_set_txpower(sc, true, false);
315
316         if (!ath_complete_reset(sc, true))
317                 r = -EIO;
318
319 out:
320         spin_unlock_bh(&sc->sc_pcu_lock);
321         tasklet_enable(&sc->intr_tq);
322
323         return r;
324 }
325
326
327 /*
328  * Set/change channels.  If the channel is really being changed, it's done
329  * by reseting the chip.  To accomplish this we must first cleanup any pending
330  * DMA, then restart stuff.
331 */
332 static int ath_set_channel(struct ath_softc *sc, struct cfg80211_chan_def *chandef)
333 {
334         struct ath_hw *ah = sc->sc_ah;
335         struct ath_common *common = ath9k_hw_common(ah);
336         struct ieee80211_hw *hw = sc->hw;
337         struct ath9k_channel *hchan;
338         struct ieee80211_channel *chan = chandef->chan;
339         unsigned long flags;
340         bool offchannel;
341         int pos = chan->hw_value;
342         int old_pos = -1;
343         int r;
344
345         if (test_bit(SC_OP_INVALID, &sc->sc_flags))
346                 return -EIO;
347
348         offchannel = !!(hw->conf.flags & IEEE80211_CONF_OFFCHANNEL);
349
350         if (ah->curchan)
351                 old_pos = ah->curchan - &ah->channels[0];
352
353         ath_dbg(common, CONFIG, "Set channel: %d MHz width: %d\n",
354                 chan->center_freq, chandef->width);
355
356         /* update survey stats for the old channel before switching */
357         spin_lock_irqsave(&common->cc_lock, flags);
358         ath_update_survey_stats(sc);
359         spin_unlock_irqrestore(&common->cc_lock, flags);
360
361         ath9k_cmn_get_channel(hw, ah, chandef);
362
363         /*
364          * If the operating channel changes, change the survey in-use flags
365          * along with it.
366          * Reset the survey data for the new channel, unless we're switching
367          * back to the operating channel from an off-channel operation.
368          */
369         if (!offchannel && sc->cur_survey != &sc->survey[pos]) {
370                 if (sc->cur_survey)
371                         sc->cur_survey->filled &= ~SURVEY_INFO_IN_USE;
372
373                 sc->cur_survey = &sc->survey[pos];
374
375                 memset(sc->cur_survey, 0, sizeof(struct survey_info));
376                 sc->cur_survey->filled |= SURVEY_INFO_IN_USE;
377         } else if (!(sc->survey[pos].filled & SURVEY_INFO_IN_USE)) {
378                 memset(&sc->survey[pos], 0, sizeof(struct survey_info));
379         }
380
381         hchan = &sc->sc_ah->channels[pos];
382         r = ath_reset_internal(sc, hchan);
383         if (r)
384                 return r;
385
386         /*
387          * The most recent snapshot of channel->noisefloor for the old
388          * channel is only available after the hardware reset. Copy it to
389          * the survey stats now.
390          */
391         if (old_pos >= 0)
392                 ath_update_survey_nf(sc, old_pos);
393
394         /*
395          * Enable radar pulse detection if on a DFS channel. Spectral
396          * scanning and radar detection can not be used concurrently.
397          */
398         if (hw->conf.radar_enabled) {
399                 u32 rxfilter;
400
401                 /* set HW specific DFS configuration */
402                 ath9k_hw_set_radar_params(ah);
403                 rxfilter = ath9k_hw_getrxfilter(ah);
404                 rxfilter |= ATH9K_RX_FILTER_PHYRADAR |
405                                 ATH9K_RX_FILTER_PHYERR;
406                 ath9k_hw_setrxfilter(ah, rxfilter);
407                 ath_dbg(common, DFS, "DFS enabled at freq %d\n",
408                         chan->center_freq);
409         } else {
410                 /* perform spectral scan if requested. */
411                 if (test_bit(SC_OP_SCANNING, &sc->sc_flags) &&
412                         sc->spectral_mode == SPECTRAL_CHANSCAN)
413                         ath9k_spectral_scan_trigger(hw);
414         }
415
416         return 0;
417 }
418
419 static void ath_node_attach(struct ath_softc *sc, struct ieee80211_sta *sta,
420                             struct ieee80211_vif *vif)
421 {
422         struct ath_node *an;
423         an = (struct ath_node *)sta->drv_priv;
424
425         an->sc = sc;
426         an->sta = sta;
427         an->vif = vif;
428
429         ath_tx_node_init(sc, an);
430
431         if (sta->ht_cap.ht_supported) {
432                 an->maxampdu = 1 << (IEEE80211_HT_MAX_AMPDU_FACTOR +
433                                      sta->ht_cap.ampdu_factor);
434                 an->mpdudensity = ath9k_parse_mpdudensity(sta->ht_cap.ampdu_density);
435         }
436 }
437
438 static void ath_node_detach(struct ath_softc *sc, struct ieee80211_sta *sta)
439 {
440         struct ath_node *an = (struct ath_node *)sta->drv_priv;
441         ath_tx_node_cleanup(sc, an);
442 }
443
444 void ath9k_tasklet(unsigned long data)
445 {
446         struct ath_softc *sc = (struct ath_softc *)data;
447         struct ath_hw *ah = sc->sc_ah;
448         struct ath_common *common = ath9k_hw_common(ah);
449         enum ath_reset_type type;
450         unsigned long flags;
451         u32 status = sc->intrstatus;
452         u32 rxmask;
453
454         ath9k_ps_wakeup(sc);
455         spin_lock(&sc->sc_pcu_lock);
456
457         if ((status & ATH9K_INT_FATAL) ||
458             (status & ATH9K_INT_BB_WATCHDOG)) {
459
460                 if (status & ATH9K_INT_FATAL)
461                         type = RESET_TYPE_FATAL_INT;
462                 else
463                         type = RESET_TYPE_BB_WATCHDOG;
464
465                 ath9k_queue_reset(sc, type);
466
467                 /*
468                  * Increment the ref. counter here so that
469                  * interrupts are enabled in the reset routine.
470                  */
471                 atomic_inc(&ah->intr_ref_cnt);
472                 ath_dbg(common, ANY, "FATAL: Skipping interrupts\n");
473                 goto out;
474         }
475
476         spin_lock_irqsave(&sc->sc_pm_lock, flags);
477         if ((status & ATH9K_INT_TSFOOR) && sc->ps_enabled) {
478                 /*
479                  * TSF sync does not look correct; remain awake to sync with
480                  * the next Beacon.
481                  */
482                 ath_dbg(common, PS, "TSFOOR - Sync with next Beacon\n");
483                 sc->ps_flags |= PS_WAIT_FOR_BEACON | PS_BEACON_SYNC;
484         }
485         spin_unlock_irqrestore(&sc->sc_pm_lock, flags);
486
487         if (ah->caps.hw_caps & ATH9K_HW_CAP_EDMA)
488                 rxmask = (ATH9K_INT_RXHP | ATH9K_INT_RXLP | ATH9K_INT_RXEOL |
489                           ATH9K_INT_RXORN);
490         else
491                 rxmask = (ATH9K_INT_RX | ATH9K_INT_RXEOL | ATH9K_INT_RXORN);
492
493         if (status & rxmask) {
494                 /* Check for high priority Rx first */
495                 if ((ah->caps.hw_caps & ATH9K_HW_CAP_EDMA) &&
496                     (status & ATH9K_INT_RXHP))
497                         ath_rx_tasklet(sc, 0, true);
498
499                 ath_rx_tasklet(sc, 0, false);
500         }
501
502         if (status & ATH9K_INT_TX) {
503                 if (ah->caps.hw_caps & ATH9K_HW_CAP_EDMA)
504                         ath_tx_edma_tasklet(sc);
505                 else
506                         ath_tx_tasklet(sc);
507         }
508
509         ath9k_btcoex_handle_interrupt(sc, status);
510
511         /* re-enable hardware interrupt */
512         ath9k_hw_enable_interrupts(ah);
513 out:
514         spin_unlock(&sc->sc_pcu_lock);
515         ath9k_ps_restore(sc);
516 }
517
518 irqreturn_t ath_isr(int irq, void *dev)
519 {
520 #define SCHED_INTR (                            \
521                 ATH9K_INT_FATAL |               \
522                 ATH9K_INT_BB_WATCHDOG |         \
523                 ATH9K_INT_RXORN |               \
524                 ATH9K_INT_RXEOL |               \
525                 ATH9K_INT_RX |                  \
526                 ATH9K_INT_RXLP |                \
527                 ATH9K_INT_RXHP |                \
528                 ATH9K_INT_TX |                  \
529                 ATH9K_INT_BMISS |               \
530                 ATH9K_INT_CST |                 \
531                 ATH9K_INT_TSFOOR |              \
532                 ATH9K_INT_GENTIMER |            \
533                 ATH9K_INT_MCI)
534
535         struct ath_softc *sc = dev;
536         struct ath_hw *ah = sc->sc_ah;
537         struct ath_common *common = ath9k_hw_common(ah);
538         enum ath9k_int status;
539         bool sched = false;
540
541         /*
542          * The hardware is not ready/present, don't
543          * touch anything. Note this can happen early
544          * on if the IRQ is shared.
545          */
546         if (test_bit(SC_OP_INVALID, &sc->sc_flags))
547                 return IRQ_NONE;
548
549         /* shared irq, not for us */
550
551         if (!ath9k_hw_intrpend(ah))
552                 return IRQ_NONE;
553
554         if (test_bit(SC_OP_HW_RESET, &sc->sc_flags)) {
555                 ath9k_hw_kill_interrupts(ah);
556                 return IRQ_HANDLED;
557         }
558
559         /*
560          * Figure out the reason(s) for the interrupt.  Note
561          * that the hal returns a pseudo-ISR that may include
562          * bits we haven't explicitly enabled so we mask the
563          * value to insure we only process bits we requested.
564          */
565         ath9k_hw_getisr(ah, &status);   /* NB: clears ISR too */
566         status &= ah->imask;    /* discard unasked-for bits */
567
568         /*
569          * If there are no status bits set, then this interrupt was not
570          * for me (should have been caught above).
571          */
572         if (!status)
573                 return IRQ_NONE;
574
575         /* Cache the status */
576         sc->intrstatus = status;
577
578         if (status & SCHED_INTR)
579                 sched = true;
580
581         /*
582          * If a FATAL or RXORN interrupt is received, we have to reset the
583          * chip immediately.
584          */
585         if ((status & ATH9K_INT_FATAL) || ((status & ATH9K_INT_RXORN) &&
586             !(ah->caps.hw_caps & ATH9K_HW_CAP_EDMA)))
587                 goto chip_reset;
588
589         if ((ah->caps.hw_caps & ATH9K_HW_CAP_EDMA) &&
590             (status & ATH9K_INT_BB_WATCHDOG)) {
591
592                 spin_lock(&common->cc_lock);
593                 ath_hw_cycle_counters_update(common);
594                 ar9003_hw_bb_watchdog_dbg_info(ah);
595                 spin_unlock(&common->cc_lock);
596
597                 goto chip_reset;
598         }
599
600 #ifdef CONFIG_ATH9K_WOW
601         if (status & ATH9K_INT_BMISS) {
602                 if (atomic_read(&sc->wow_sleep_proc_intr) == 0) {
603                         ath_dbg(common, ANY, "during WoW we got a BMISS\n");
604                         atomic_inc(&sc->wow_got_bmiss_intr);
605                         atomic_dec(&sc->wow_sleep_proc_intr);
606                 }
607         }
608 #endif
609
610
611         if (status & ATH9K_INT_SWBA)
612                 tasklet_schedule(&sc->bcon_tasklet);
613
614         if (status & ATH9K_INT_TXURN)
615                 ath9k_hw_updatetxtriglevel(ah, true);
616
617         if (status & ATH9K_INT_RXEOL) {
618                 ah->imask &= ~(ATH9K_INT_RXEOL | ATH9K_INT_RXORN);
619                 ath9k_hw_set_interrupts(ah);
620         }
621
622         if (!(ah->caps.hw_caps & ATH9K_HW_CAP_AUTOSLEEP))
623                 if (status & ATH9K_INT_TIM_TIMER) {
624                         if (ATH_DBG_WARN_ON_ONCE(sc->ps_idle))
625                                 goto chip_reset;
626                         /* Clear RxAbort bit so that we can
627                          * receive frames */
628                         ath9k_setpower(sc, ATH9K_PM_AWAKE);
629                         spin_lock(&sc->sc_pm_lock);
630                         ath9k_hw_setrxabort(sc->sc_ah, 0);
631                         sc->ps_flags |= PS_WAIT_FOR_BEACON;
632                         spin_unlock(&sc->sc_pm_lock);
633                 }
634
635 chip_reset:
636
637         ath_debug_stat_interrupt(sc, status);
638
639         if (sched) {
640                 /* turn off every interrupt */
641                 ath9k_hw_disable_interrupts(ah);
642                 tasklet_schedule(&sc->intr_tq);
643         }
644
645         return IRQ_HANDLED;
646
647 #undef SCHED_INTR
648 }
649
650 int ath_reset(struct ath_softc *sc)
651 {
652         int r;
653
654         ath9k_ps_wakeup(sc);
655         r = ath_reset_internal(sc, NULL);
656         ath9k_ps_restore(sc);
657
658         return r;
659 }
660
661 void ath9k_queue_reset(struct ath_softc *sc, enum ath_reset_type type)
662 {
663 #ifdef CONFIG_ATH9K_DEBUGFS
664         RESET_STAT_INC(sc, type);
665 #endif
666         set_bit(SC_OP_HW_RESET, &sc->sc_flags);
667         ieee80211_queue_work(sc->hw, &sc->hw_reset_work);
668 }
669
670 void ath_reset_work(struct work_struct *work)
671 {
672         struct ath_softc *sc = container_of(work, struct ath_softc, hw_reset_work);
673
674         ath_reset(sc);
675 }
676
677 /**********************/
678 /* mac80211 callbacks */
679 /**********************/
680
681 static int ath9k_start(struct ieee80211_hw *hw)
682 {
683         struct ath_softc *sc = hw->priv;
684         struct ath_hw *ah = sc->sc_ah;
685         struct ath_common *common = ath9k_hw_common(ah);
686         struct ieee80211_channel *curchan = hw->conf.chandef.chan;
687         struct ath9k_channel *init_channel;
688         int r;
689
690         ath_dbg(common, CONFIG,
691                 "Starting driver with initial channel: %d MHz\n",
692                 curchan->center_freq);
693
694         ath9k_ps_wakeup(sc);
695         mutex_lock(&sc->mutex);
696
697         init_channel = ath9k_cmn_get_channel(hw, ah, &hw->conf.chandef);
698
699         /* Reset SERDES registers */
700         ath9k_hw_configpcipowersave(ah, false);
701
702         /*
703          * The basic interface to setting the hardware in a good
704          * state is ``reset''.  On return the hardware is known to
705          * be powered up and with interrupts disabled.  This must
706          * be followed by initialization of the appropriate bits
707          * and then setup of the interrupt mask.
708          */
709         spin_lock_bh(&sc->sc_pcu_lock);
710
711         atomic_set(&ah->intr_ref_cnt, -1);
712
713         r = ath9k_hw_reset(ah, init_channel, ah->caldata, false);
714         if (r) {
715                 ath_err(common,
716                         "Unable to reset hardware; reset status %d (freq %u MHz)\n",
717                         r, curchan->center_freq);
718                 ah->reset_power_on = false;
719         }
720
721         /* Setup our intr mask. */
722         ah->imask = ATH9K_INT_TX | ATH9K_INT_RXEOL |
723                     ATH9K_INT_RXORN | ATH9K_INT_FATAL |
724                     ATH9K_INT_GLOBAL;
725
726         if (ah->caps.hw_caps & ATH9K_HW_CAP_EDMA)
727                 ah->imask |= ATH9K_INT_RXHP |
728                              ATH9K_INT_RXLP |
729                              ATH9K_INT_BB_WATCHDOG;
730         else
731                 ah->imask |= ATH9K_INT_RX;
732
733         ah->imask |= ATH9K_INT_GTT;
734
735         if (ah->caps.hw_caps & ATH9K_HW_CAP_HT)
736                 ah->imask |= ATH9K_INT_CST;
737
738         ath_mci_enable(sc);
739
740         clear_bit(SC_OP_INVALID, &sc->sc_flags);
741         sc->sc_ah->is_monitoring = false;
742
743         if (!ath_complete_reset(sc, false))
744                 ah->reset_power_on = false;
745
746         if (ah->led_pin >= 0) {
747                 ath9k_hw_cfg_output(ah, ah->led_pin,
748                                     AR_GPIO_OUTPUT_MUX_AS_OUTPUT);
749                 ath9k_hw_set_gpio(ah, ah->led_pin, 0);
750         }
751
752         /*
753          * Reset key cache to sane defaults (all entries cleared) instead of
754          * semi-random values after suspend/resume.
755          */
756         ath9k_cmn_init_crypto(sc->sc_ah);
757
758         spin_unlock_bh(&sc->sc_pcu_lock);
759
760         mutex_unlock(&sc->mutex);
761
762         ath9k_ps_restore(sc);
763
764         return 0;
765 }
766
767 static void ath9k_tx(struct ieee80211_hw *hw,
768                      struct ieee80211_tx_control *control,
769                      struct sk_buff *skb)
770 {
771         struct ath_softc *sc = hw->priv;
772         struct ath_common *common = ath9k_hw_common(sc->sc_ah);
773         struct ath_tx_control txctl;
774         struct ieee80211_hdr *hdr = (struct ieee80211_hdr *) skb->data;
775         unsigned long flags;
776
777         if (sc->ps_enabled) {
778                 /*
779                  * mac80211 does not set PM field for normal data frames, so we
780                  * need to update that based on the current PS mode.
781                  */
782                 if (ieee80211_is_data(hdr->frame_control) &&
783                     !ieee80211_is_nullfunc(hdr->frame_control) &&
784                     !ieee80211_has_pm(hdr->frame_control)) {
785                         ath_dbg(common, PS,
786                                 "Add PM=1 for a TX frame while in PS mode\n");
787                         hdr->frame_control |= cpu_to_le16(IEEE80211_FCTL_PM);
788                 }
789         }
790
791         if (unlikely(sc->sc_ah->power_mode == ATH9K_PM_NETWORK_SLEEP)) {
792                 /*
793                  * We are using PS-Poll and mac80211 can request TX while in
794                  * power save mode. Need to wake up hardware for the TX to be
795                  * completed and if needed, also for RX of buffered frames.
796                  */
797                 ath9k_ps_wakeup(sc);
798                 spin_lock_irqsave(&sc->sc_pm_lock, flags);
799                 if (!(sc->sc_ah->caps.hw_caps & ATH9K_HW_CAP_AUTOSLEEP))
800                         ath9k_hw_setrxabort(sc->sc_ah, 0);
801                 if (ieee80211_is_pspoll(hdr->frame_control)) {
802                         ath_dbg(common, PS,
803                                 "Sending PS-Poll to pick a buffered frame\n");
804                         sc->ps_flags |= PS_WAIT_FOR_PSPOLL_DATA;
805                 } else {
806                         ath_dbg(common, PS, "Wake up to complete TX\n");
807                         sc->ps_flags |= PS_WAIT_FOR_TX_ACK;
808                 }
809                 /*
810                  * The actual restore operation will happen only after
811                  * the ps_flags bit is cleared. We are just dropping
812                  * the ps_usecount here.
813                  */
814                 spin_unlock_irqrestore(&sc->sc_pm_lock, flags);
815                 ath9k_ps_restore(sc);
816         }
817
818         /*
819          * Cannot tx while the hardware is in full sleep, it first needs a full
820          * chip reset to recover from that
821          */
822         if (unlikely(sc->sc_ah->power_mode == ATH9K_PM_FULL_SLEEP)) {
823                 ath_err(common, "TX while HW is in FULL_SLEEP mode\n");
824                 goto exit;
825         }
826
827         memset(&txctl, 0, sizeof(struct ath_tx_control));
828         txctl.txq = sc->tx.txq_map[skb_get_queue_mapping(skb)];
829         txctl.sta = control->sta;
830
831         ath_dbg(common, XMIT, "transmitting packet, skb: %p\n", skb);
832
833         if (ath_tx_start(hw, skb, &txctl) != 0) {
834                 ath_dbg(common, XMIT, "TX failed\n");
835                 TX_STAT_INC(txctl.txq->axq_qnum, txfailed);
836                 goto exit;
837         }
838
839         return;
840 exit:
841         ieee80211_free_txskb(hw, skb);
842 }
843
844 static void ath9k_stop(struct ieee80211_hw *hw)
845 {
846         struct ath_softc *sc = hw->priv;
847         struct ath_hw *ah = sc->sc_ah;
848         struct ath_common *common = ath9k_hw_common(ah);
849         bool prev_idle;
850
851         mutex_lock(&sc->mutex);
852
853         ath_cancel_work(sc);
854         del_timer_sync(&sc->rx_poll_timer);
855
856         if (test_bit(SC_OP_INVALID, &sc->sc_flags)) {
857                 ath_dbg(common, ANY, "Device not present\n");
858                 mutex_unlock(&sc->mutex);
859                 return;
860         }
861
862         /* Ensure HW is awake when we try to shut it down. */
863         ath9k_ps_wakeup(sc);
864
865         spin_lock_bh(&sc->sc_pcu_lock);
866
867         /* prevent tasklets to enable interrupts once we disable them */
868         ah->imask &= ~ATH9K_INT_GLOBAL;
869
870         /* make sure h/w will not generate any interrupt
871          * before setting the invalid flag. */
872         ath9k_hw_disable_interrupts(ah);
873
874         spin_unlock_bh(&sc->sc_pcu_lock);
875
876         /* we can now sync irq and kill any running tasklets, since we already
877          * disabled interrupts and not holding a spin lock */
878         synchronize_irq(sc->irq);
879         tasklet_kill(&sc->intr_tq);
880         tasklet_kill(&sc->bcon_tasklet);
881
882         prev_idle = sc->ps_idle;
883         sc->ps_idle = true;
884
885         spin_lock_bh(&sc->sc_pcu_lock);
886
887         if (ah->led_pin >= 0) {
888                 ath9k_hw_set_gpio(ah, ah->led_pin, 1);
889                 ath9k_hw_cfg_gpio_input(ah, ah->led_pin);
890         }
891
892         ath_prepare_reset(sc);
893
894         if (sc->rx.frag) {
895                 dev_kfree_skb_any(sc->rx.frag);
896                 sc->rx.frag = NULL;
897         }
898
899         if (!ah->curchan)
900                 ah->curchan = ath9k_cmn_get_channel(hw, ah, &hw->conf.chandef);
901
902         ath9k_hw_reset(ah, ah->curchan, ah->caldata, false);
903         ath9k_hw_phy_disable(ah);
904
905         ath9k_hw_configpcipowersave(ah, true);
906
907         spin_unlock_bh(&sc->sc_pcu_lock);
908
909         ath9k_ps_restore(sc);
910
911         set_bit(SC_OP_INVALID, &sc->sc_flags);
912         sc->ps_idle = prev_idle;
913
914         mutex_unlock(&sc->mutex);
915
916         ath_dbg(common, CONFIG, "Driver halt\n");
917 }
918
919 static bool ath9k_uses_beacons(int type)
920 {
921         switch (type) {
922         case NL80211_IFTYPE_AP:
923         case NL80211_IFTYPE_ADHOC:
924         case NL80211_IFTYPE_MESH_POINT:
925                 return true;
926         default:
927                 return false;
928         }
929 }
930
931 static void ath9k_vif_iter(void *data, u8 *mac, struct ieee80211_vif *vif)
932 {
933         struct ath9k_vif_iter_data *iter_data = data;
934         int i;
935
936         if (iter_data->has_hw_macaddr) {
937                 for (i = 0; i < ETH_ALEN; i++)
938                         iter_data->mask[i] &=
939                                 ~(iter_data->hw_macaddr[i] ^ mac[i]);
940         } else {
941                 memcpy(iter_data->hw_macaddr, mac, ETH_ALEN);
942                 iter_data->has_hw_macaddr = true;
943         }
944
945         switch (vif->type) {
946         case NL80211_IFTYPE_AP:
947                 iter_data->naps++;
948                 break;
949         case NL80211_IFTYPE_STATION:
950                 iter_data->nstations++;
951                 break;
952         case NL80211_IFTYPE_ADHOC:
953                 iter_data->nadhocs++;
954                 break;
955         case NL80211_IFTYPE_MESH_POINT:
956                 iter_data->nmeshes++;
957                 break;
958         case NL80211_IFTYPE_WDS:
959                 iter_data->nwds++;
960                 break;
961         default:
962                 break;
963         }
964 }
965
966 static void ath9k_sta_vif_iter(void *data, u8 *mac, struct ieee80211_vif *vif)
967 {
968         struct ath_softc *sc = data;
969         struct ath_vif *avp = (void *)vif->drv_priv;
970
971         if (vif->type != NL80211_IFTYPE_STATION)
972                 return;
973
974         if (avp->primary_sta_vif)
975                 ath9k_set_assoc_state(sc, vif);
976 }
977
978 /* Called with sc->mutex held. */
979 void ath9k_calculate_iter_data(struct ieee80211_hw *hw,
980                                struct ieee80211_vif *vif,
981                                struct ath9k_vif_iter_data *iter_data)
982 {
983         struct ath_softc *sc = hw->priv;
984         struct ath_hw *ah = sc->sc_ah;
985         struct ath_common *common = ath9k_hw_common(ah);
986
987         /*
988          * Use the hardware MAC address as reference, the hardware uses it
989          * together with the BSSID mask when matching addresses.
990          */
991         memset(iter_data, 0, sizeof(*iter_data));
992         memset(&iter_data->mask, 0xff, ETH_ALEN);
993
994         if (vif)
995                 ath9k_vif_iter(iter_data, vif->addr, vif);
996
997         /* Get list of all active MAC addresses */
998         ieee80211_iterate_active_interfaces_atomic(
999                 sc->hw, IEEE80211_IFACE_ITER_RESUME_ALL,
1000                 ath9k_vif_iter, iter_data);
1001
1002         memcpy(common->macaddr, iter_data->hw_macaddr, ETH_ALEN);
1003 }
1004
1005 /* Called with sc->mutex held. */
1006 static void ath9k_calculate_summary_state(struct ieee80211_hw *hw,
1007                                           struct ieee80211_vif *vif)
1008 {
1009         struct ath_softc *sc = hw->priv;
1010         struct ath_hw *ah = sc->sc_ah;
1011         struct ath_common *common = ath9k_hw_common(ah);
1012         struct ath9k_vif_iter_data iter_data;
1013         enum nl80211_iftype old_opmode = ah->opmode;
1014
1015         ath9k_calculate_iter_data(hw, vif, &iter_data);
1016
1017         memcpy(common->bssidmask, iter_data.mask, ETH_ALEN);
1018         ath_hw_setbssidmask(common);
1019
1020         if (iter_data.naps > 0) {
1021                 ath9k_hw_set_tsfadjust(ah, true);
1022                 ah->opmode = NL80211_IFTYPE_AP;
1023         } else {
1024                 ath9k_hw_set_tsfadjust(ah, false);
1025
1026                 if (iter_data.nmeshes)
1027                         ah->opmode = NL80211_IFTYPE_MESH_POINT;
1028                 else if (iter_data.nwds)
1029                         ah->opmode = NL80211_IFTYPE_AP;
1030                 else if (iter_data.nadhocs)
1031                         ah->opmode = NL80211_IFTYPE_ADHOC;
1032                 else
1033                         ah->opmode = NL80211_IFTYPE_STATION;
1034         }
1035
1036         ath9k_hw_setopmode(ah);
1037
1038         if ((iter_data.nstations + iter_data.nadhocs + iter_data.nmeshes) > 0)
1039                 ah->imask |= ATH9K_INT_TSFOOR;
1040         else
1041                 ah->imask &= ~ATH9K_INT_TSFOOR;
1042
1043         ath9k_hw_set_interrupts(ah);
1044
1045         /*
1046          * If we are changing the opmode to STATION,
1047          * a beacon sync needs to be done.
1048          */
1049         if (ah->opmode == NL80211_IFTYPE_STATION &&
1050             old_opmode == NL80211_IFTYPE_AP &&
1051             test_bit(SC_OP_PRIM_STA_VIF, &sc->sc_flags)) {
1052                 ieee80211_iterate_active_interfaces_atomic(
1053                         sc->hw, IEEE80211_IFACE_ITER_RESUME_ALL,
1054                         ath9k_sta_vif_iter, sc);
1055         }
1056 }
1057
1058 static int ath9k_add_interface(struct ieee80211_hw *hw,
1059                                struct ieee80211_vif *vif)
1060 {
1061         struct ath_softc *sc = hw->priv;
1062         struct ath_hw *ah = sc->sc_ah;
1063         struct ath_common *common = ath9k_hw_common(ah);
1064         struct ath_vif *avp = (void *)vif->drv_priv;
1065         struct ath_node *an = &avp->mcast_node;
1066
1067         mutex_lock(&sc->mutex);
1068
1069         if (config_enabled(CONFIG_ATH9K_TX99)) {
1070                 if (sc->nvifs >= 1) {
1071                         mutex_unlock(&sc->mutex);
1072                         return -EOPNOTSUPP;
1073                 }
1074                 sc->tx99_vif = vif;
1075         }
1076
1077         ath_dbg(common, CONFIG, "Attach a VIF of type: %d\n", vif->type);
1078         sc->nvifs++;
1079
1080         ath9k_ps_wakeup(sc);
1081         ath9k_calculate_summary_state(hw, vif);
1082         ath9k_ps_restore(sc);
1083
1084         if (ath9k_uses_beacons(vif->type))
1085                 ath9k_beacon_assign_slot(sc, vif);
1086
1087         an->sc = sc;
1088         an->sta = NULL;
1089         an->vif = vif;
1090         an->no_ps_filter = true;
1091         ath_tx_node_init(sc, an);
1092
1093         mutex_unlock(&sc->mutex);
1094         return 0;
1095 }
1096
1097 static int ath9k_change_interface(struct ieee80211_hw *hw,
1098                                   struct ieee80211_vif *vif,
1099                                   enum nl80211_iftype new_type,
1100                                   bool p2p)
1101 {
1102         struct ath_softc *sc = hw->priv;
1103         struct ath_common *common = ath9k_hw_common(sc->sc_ah);
1104
1105         mutex_lock(&sc->mutex);
1106
1107         if (config_enabled(CONFIG_ATH9K_TX99)) {
1108                 mutex_unlock(&sc->mutex);
1109                 return -EOPNOTSUPP;
1110         }
1111
1112         ath_dbg(common, CONFIG, "Change Interface\n");
1113
1114         if (ath9k_uses_beacons(vif->type))
1115                 ath9k_beacon_remove_slot(sc, vif);
1116
1117         vif->type = new_type;
1118         vif->p2p = p2p;
1119
1120         ath9k_ps_wakeup(sc);
1121         ath9k_calculate_summary_state(hw, vif);
1122         ath9k_ps_restore(sc);
1123
1124         if (ath9k_uses_beacons(vif->type))
1125                 ath9k_beacon_assign_slot(sc, vif);
1126
1127         mutex_unlock(&sc->mutex);
1128         return 0;
1129 }
1130
1131 static void ath9k_remove_interface(struct ieee80211_hw *hw,
1132                                    struct ieee80211_vif *vif)
1133 {
1134         struct ath_softc *sc = hw->priv;
1135         struct ath_common *common = ath9k_hw_common(sc->sc_ah);
1136         struct ath_vif *avp = (void *)vif->drv_priv;
1137
1138         ath_dbg(common, CONFIG, "Detach Interface\n");
1139
1140         mutex_lock(&sc->mutex);
1141
1142         sc->nvifs--;
1143         sc->tx99_vif = NULL;
1144
1145         if (ath9k_uses_beacons(vif->type))
1146                 ath9k_beacon_remove_slot(sc, vif);
1147
1148         if (sc->csa_vif == vif)
1149                 sc->csa_vif = NULL;
1150
1151         ath9k_ps_wakeup(sc);
1152         ath9k_calculate_summary_state(hw, NULL);
1153         ath9k_ps_restore(sc);
1154
1155         ath_tx_node_cleanup(sc, &avp->mcast_node);
1156
1157         mutex_unlock(&sc->mutex);
1158 }
1159
1160 static void ath9k_enable_ps(struct ath_softc *sc)
1161 {
1162         struct ath_hw *ah = sc->sc_ah;
1163         struct ath_common *common = ath9k_hw_common(ah);
1164
1165         if (config_enabled(CONFIG_ATH9K_TX99))
1166                 return;
1167
1168         sc->ps_enabled = true;
1169         if (!(ah->caps.hw_caps & ATH9K_HW_CAP_AUTOSLEEP)) {
1170                 if ((ah->imask & ATH9K_INT_TIM_TIMER) == 0) {
1171                         ah->imask |= ATH9K_INT_TIM_TIMER;
1172                         ath9k_hw_set_interrupts(ah);
1173                 }
1174                 ath9k_hw_setrxabort(ah, 1);
1175         }
1176         ath_dbg(common, PS, "PowerSave enabled\n");
1177 }
1178
1179 static void ath9k_disable_ps(struct ath_softc *sc)
1180 {
1181         struct ath_hw *ah = sc->sc_ah;
1182         struct ath_common *common = ath9k_hw_common(ah);
1183
1184         if (config_enabled(CONFIG_ATH9K_TX99))
1185                 return;
1186
1187         sc->ps_enabled = false;
1188         ath9k_hw_setpower(ah, ATH9K_PM_AWAKE);
1189         if (!(ah->caps.hw_caps & ATH9K_HW_CAP_AUTOSLEEP)) {
1190                 ath9k_hw_setrxabort(ah, 0);
1191                 sc->ps_flags &= ~(PS_WAIT_FOR_BEACON |
1192                                   PS_WAIT_FOR_CAB |
1193                                   PS_WAIT_FOR_PSPOLL_DATA |
1194                                   PS_WAIT_FOR_TX_ACK);
1195                 if (ah->imask & ATH9K_INT_TIM_TIMER) {
1196                         ah->imask &= ~ATH9K_INT_TIM_TIMER;
1197                         ath9k_hw_set_interrupts(ah);
1198                 }
1199         }
1200         ath_dbg(common, PS, "PowerSave disabled\n");
1201 }
1202
1203 void ath9k_spectral_scan_trigger(struct ieee80211_hw *hw)
1204 {
1205         struct ath_softc *sc = hw->priv;
1206         struct ath_hw *ah = sc->sc_ah;
1207         struct ath_common *common = ath9k_hw_common(ah);
1208         u32 rxfilter;
1209
1210         if (config_enabled(CONFIG_ATH9K_TX99))
1211                 return;
1212
1213         if (!ath9k_hw_ops(ah)->spectral_scan_trigger) {
1214                 ath_err(common, "spectrum analyzer not implemented on this hardware\n");
1215                 return;
1216         }
1217
1218         ath9k_ps_wakeup(sc);
1219         rxfilter = ath9k_hw_getrxfilter(ah);
1220         ath9k_hw_setrxfilter(ah, rxfilter |
1221                                  ATH9K_RX_FILTER_PHYRADAR |
1222                                  ATH9K_RX_FILTER_PHYERR);
1223
1224         /* TODO: usually this should not be neccesary, but for some reason
1225          * (or in some mode?) the trigger must be called after the
1226          * configuration, otherwise the register will have its values reset
1227          * (on my ar9220 to value 0x01002310)
1228          */
1229         ath9k_spectral_scan_config(hw, sc->spectral_mode);
1230         ath9k_hw_ops(ah)->spectral_scan_trigger(ah);
1231         ath9k_ps_restore(sc);
1232 }
1233
1234 int ath9k_spectral_scan_config(struct ieee80211_hw *hw,
1235                                enum spectral_mode spectral_mode)
1236 {
1237         struct ath_softc *sc = hw->priv;
1238         struct ath_hw *ah = sc->sc_ah;
1239         struct ath_common *common = ath9k_hw_common(ah);
1240
1241         if (!ath9k_hw_ops(ah)->spectral_scan_trigger) {
1242                 ath_err(common, "spectrum analyzer not implemented on this hardware\n");
1243                 return -1;
1244         }
1245
1246         switch (spectral_mode) {
1247         case SPECTRAL_DISABLED:
1248                 sc->spec_config.enabled = 0;
1249                 break;
1250         case SPECTRAL_BACKGROUND:
1251                 /* send endless samples.
1252                  * TODO: is this really useful for "background"?
1253                  */
1254                 sc->spec_config.endless = 1;
1255                 sc->spec_config.enabled = 1;
1256                 break;
1257         case SPECTRAL_CHANSCAN:
1258         case SPECTRAL_MANUAL:
1259                 sc->spec_config.endless = 0;
1260                 sc->spec_config.enabled = 1;
1261                 break;
1262         default:
1263                 return -1;
1264         }
1265
1266         ath9k_ps_wakeup(sc);
1267         ath9k_hw_ops(ah)->spectral_scan_config(ah, &sc->spec_config);
1268         ath9k_ps_restore(sc);
1269
1270         sc->spectral_mode = spectral_mode;
1271
1272         return 0;
1273 }
1274
1275 static int ath9k_config(struct ieee80211_hw *hw, u32 changed)
1276 {
1277         struct ath_softc *sc = hw->priv;
1278         struct ath_hw *ah = sc->sc_ah;
1279         struct ath_common *common = ath9k_hw_common(ah);
1280         struct ieee80211_conf *conf = &hw->conf;
1281         bool reset_channel = false;
1282
1283         ath9k_ps_wakeup(sc);
1284         mutex_lock(&sc->mutex);
1285
1286         if (changed & IEEE80211_CONF_CHANGE_IDLE) {
1287                 sc->ps_idle = !!(conf->flags & IEEE80211_CONF_IDLE);
1288                 if (sc->ps_idle) {
1289                         ath_cancel_work(sc);
1290                         ath9k_stop_btcoex(sc);
1291                 } else {
1292                         ath9k_start_btcoex(sc);
1293                         /*
1294                          * The chip needs a reset to properly wake up from
1295                          * full sleep
1296                          */
1297                         reset_channel = ah->chip_fullsleep;
1298                 }
1299         }
1300
1301         /*
1302          * We just prepare to enable PS. We have to wait until our AP has
1303          * ACK'd our null data frame to disable RX otherwise we'll ignore
1304          * those ACKs and end up retransmitting the same null data frames.
1305          * IEEE80211_CONF_CHANGE_PS is only passed by mac80211 for STA mode.
1306          */
1307         if (changed & IEEE80211_CONF_CHANGE_PS) {
1308                 unsigned long flags;
1309                 spin_lock_irqsave(&sc->sc_pm_lock, flags);
1310                 if (conf->flags & IEEE80211_CONF_PS)
1311                         ath9k_enable_ps(sc);
1312                 else
1313                         ath9k_disable_ps(sc);
1314                 spin_unlock_irqrestore(&sc->sc_pm_lock, flags);
1315         }
1316
1317         if (changed & IEEE80211_CONF_CHANGE_MONITOR) {
1318                 if (conf->flags & IEEE80211_CONF_MONITOR) {
1319                         ath_dbg(common, CONFIG, "Monitor mode is enabled\n");
1320                         sc->sc_ah->is_monitoring = true;
1321                 } else {
1322                         ath_dbg(common, CONFIG, "Monitor mode is disabled\n");
1323                         sc->sc_ah->is_monitoring = false;
1324                 }
1325         }
1326
1327         if ((changed & IEEE80211_CONF_CHANGE_CHANNEL) || reset_channel) {
1328                 if (ath_set_channel(sc, &hw->conf.chandef) < 0) {
1329                         ath_err(common, "Unable to set channel\n");
1330                         mutex_unlock(&sc->mutex);
1331                         ath9k_ps_restore(sc);
1332                         return -EINVAL;
1333                 }
1334         }
1335
1336         if (changed & IEEE80211_CONF_CHANGE_POWER) {
1337                 ath_dbg(common, CONFIG, "Set power: %d\n", conf->power_level);
1338                 sc->config.txpowlimit = 2 * conf->power_level;
1339                 ath9k_cmn_update_txpow(ah, sc->curtxpow,
1340                                        sc->config.txpowlimit, &sc->curtxpow);
1341         }
1342
1343         mutex_unlock(&sc->mutex);
1344         ath9k_ps_restore(sc);
1345
1346         return 0;
1347 }
1348
1349 #define SUPPORTED_FILTERS                       \
1350         (FIF_PROMISC_IN_BSS |                   \
1351         FIF_ALLMULTI |                          \
1352         FIF_CONTROL |                           \
1353         FIF_PSPOLL |                            \
1354         FIF_OTHER_BSS |                         \
1355         FIF_BCN_PRBRESP_PROMISC |               \
1356         FIF_PROBE_REQ |                         \
1357         FIF_FCSFAIL)
1358
1359 /* FIXME: sc->sc_full_reset ? */
1360 static void ath9k_configure_filter(struct ieee80211_hw *hw,
1361                                    unsigned int changed_flags,
1362                                    unsigned int *total_flags,
1363                                    u64 multicast)
1364 {
1365         struct ath_softc *sc = hw->priv;
1366         u32 rfilt;
1367
1368         changed_flags &= SUPPORTED_FILTERS;
1369         *total_flags &= SUPPORTED_FILTERS;
1370
1371         sc->rx.rxfilter = *total_flags;
1372         ath9k_ps_wakeup(sc);
1373         rfilt = ath_calcrxfilter(sc);
1374         ath9k_hw_setrxfilter(sc->sc_ah, rfilt);
1375         ath9k_ps_restore(sc);
1376
1377         ath_dbg(ath9k_hw_common(sc->sc_ah), CONFIG, "Set HW RX filter: 0x%x\n",
1378                 rfilt);
1379 }
1380
1381 static int ath9k_sta_add(struct ieee80211_hw *hw,
1382                          struct ieee80211_vif *vif,
1383                          struct ieee80211_sta *sta)
1384 {
1385         struct ath_softc *sc = hw->priv;
1386         struct ath_common *common = ath9k_hw_common(sc->sc_ah);
1387         struct ath_node *an = (struct ath_node *) sta->drv_priv;
1388         struct ieee80211_key_conf ps_key = { };
1389         int key;
1390
1391         ath_node_attach(sc, sta, vif);
1392
1393         if (vif->type != NL80211_IFTYPE_AP &&
1394             vif->type != NL80211_IFTYPE_AP_VLAN)
1395                 return 0;
1396
1397         key = ath_key_config(common, vif, sta, &ps_key);
1398         if (key > 0)
1399                 an->ps_key = key;
1400
1401         return 0;
1402 }
1403
1404 static void ath9k_del_ps_key(struct ath_softc *sc,
1405                              struct ieee80211_vif *vif,
1406                              struct ieee80211_sta *sta)
1407 {
1408         struct ath_common *common = ath9k_hw_common(sc->sc_ah);
1409         struct ath_node *an = (struct ath_node *) sta->drv_priv;
1410         struct ieee80211_key_conf ps_key = { .hw_key_idx = an->ps_key };
1411
1412         if (!an->ps_key)
1413             return;
1414
1415         ath_key_delete(common, &ps_key);
1416         an->ps_key = 0;
1417 }
1418
1419 static int ath9k_sta_remove(struct ieee80211_hw *hw,
1420                             struct ieee80211_vif *vif,
1421                             struct ieee80211_sta *sta)
1422 {
1423         struct ath_softc *sc = hw->priv;
1424
1425         ath9k_del_ps_key(sc, vif, sta);
1426         ath_node_detach(sc, sta);
1427
1428         return 0;
1429 }
1430
1431 static void ath9k_sta_notify(struct ieee80211_hw *hw,
1432                          struct ieee80211_vif *vif,
1433                          enum sta_notify_cmd cmd,
1434                          struct ieee80211_sta *sta)
1435 {
1436         struct ath_softc *sc = hw->priv;
1437         struct ath_node *an = (struct ath_node *) sta->drv_priv;
1438
1439         switch (cmd) {
1440         case STA_NOTIFY_SLEEP:
1441                 an->sleeping = true;
1442                 ath_tx_aggr_sleep(sta, sc, an);
1443                 break;
1444         case STA_NOTIFY_AWAKE:
1445                 an->sleeping = false;
1446                 ath_tx_aggr_wakeup(sc, an);
1447                 break;
1448         }
1449 }
1450
1451 static int ath9k_conf_tx(struct ieee80211_hw *hw,
1452                          struct ieee80211_vif *vif, u16 queue,
1453                          const struct ieee80211_tx_queue_params *params)
1454 {
1455         struct ath_softc *sc = hw->priv;
1456         struct ath_common *common = ath9k_hw_common(sc->sc_ah);
1457         struct ath_txq *txq;
1458         struct ath9k_tx_queue_info qi;
1459         int ret = 0;
1460
1461         if (queue >= IEEE80211_NUM_ACS)
1462                 return 0;
1463
1464         txq = sc->tx.txq_map[queue];
1465
1466         ath9k_ps_wakeup(sc);
1467         mutex_lock(&sc->mutex);
1468
1469         memset(&qi, 0, sizeof(struct ath9k_tx_queue_info));
1470
1471         qi.tqi_aifs = params->aifs;
1472         qi.tqi_cwmin = params->cw_min;
1473         qi.tqi_cwmax = params->cw_max;
1474         qi.tqi_burstTime = params->txop * 32;
1475
1476         ath_dbg(common, CONFIG,
1477                 "Configure tx [queue/halq] [%d/%d], aifs: %d, cw_min: %d, cw_max: %d, txop: %d\n",
1478                 queue, txq->axq_qnum, params->aifs, params->cw_min,
1479                 params->cw_max, params->txop);
1480
1481         ath_update_max_aggr_framelen(sc, queue, qi.tqi_burstTime);
1482         ret = ath_txq_update(sc, txq->axq_qnum, &qi);
1483         if (ret)
1484                 ath_err(common, "TXQ Update failed\n");
1485
1486         mutex_unlock(&sc->mutex);
1487         ath9k_ps_restore(sc);
1488
1489         return ret;
1490 }
1491
1492 static int ath9k_set_key(struct ieee80211_hw *hw,
1493                          enum set_key_cmd cmd,
1494                          struct ieee80211_vif *vif,
1495                          struct ieee80211_sta *sta,
1496                          struct ieee80211_key_conf *key)
1497 {
1498         struct ath_softc *sc = hw->priv;
1499         struct ath_common *common = ath9k_hw_common(sc->sc_ah);
1500         int ret = 0;
1501
1502         if (ath9k_modparam_nohwcrypt)
1503                 return -ENOSPC;
1504
1505         if ((vif->type == NL80211_IFTYPE_ADHOC ||
1506              vif->type == NL80211_IFTYPE_MESH_POINT) &&
1507             (key->cipher == WLAN_CIPHER_SUITE_TKIP ||
1508              key->cipher == WLAN_CIPHER_SUITE_CCMP) &&
1509             !(key->flags & IEEE80211_KEY_FLAG_PAIRWISE)) {
1510                 /*
1511                  * For now, disable hw crypto for the RSN IBSS group keys. This
1512                  * could be optimized in the future to use a modified key cache
1513                  * design to support per-STA RX GTK, but until that gets
1514                  * implemented, use of software crypto for group addressed
1515                  * frames is a acceptable to allow RSN IBSS to be used.
1516                  */
1517                 return -EOPNOTSUPP;
1518         }
1519
1520         mutex_lock(&sc->mutex);
1521         ath9k_ps_wakeup(sc);
1522         ath_dbg(common, CONFIG, "Set HW Key\n");
1523
1524         switch (cmd) {
1525         case SET_KEY:
1526                 if (sta)
1527                         ath9k_del_ps_key(sc, vif, sta);
1528
1529                 ret = ath_key_config(common, vif, sta, key);
1530                 if (ret >= 0) {
1531                         key->hw_key_idx = ret;
1532                         /* push IV and Michael MIC generation to stack */
1533                         key->flags |= IEEE80211_KEY_FLAG_GENERATE_IV;
1534                         if (key->cipher == WLAN_CIPHER_SUITE_TKIP)
1535                                 key->flags |= IEEE80211_KEY_FLAG_GENERATE_MMIC;
1536                         if (sc->sc_ah->sw_mgmt_crypto &&
1537                             key->cipher == WLAN_CIPHER_SUITE_CCMP)
1538                                 key->flags |= IEEE80211_KEY_FLAG_SW_MGMT_TX;
1539                         ret = 0;
1540                 }
1541                 break;
1542         case DISABLE_KEY:
1543                 ath_key_delete(common, key);
1544                 break;
1545         default:
1546                 ret = -EINVAL;
1547         }
1548
1549         ath9k_ps_restore(sc);
1550         mutex_unlock(&sc->mutex);
1551
1552         return ret;
1553 }
1554
1555 static void ath9k_set_assoc_state(struct ath_softc *sc,
1556                                   struct ieee80211_vif *vif)
1557 {
1558         struct ath_common *common = ath9k_hw_common(sc->sc_ah);
1559         struct ath_vif *avp = (void *)vif->drv_priv;
1560         struct ieee80211_bss_conf *bss_conf = &vif->bss_conf;
1561         unsigned long flags;
1562
1563         set_bit(SC_OP_PRIM_STA_VIF, &sc->sc_flags);
1564         avp->primary_sta_vif = true;
1565
1566         /*
1567          * Set the AID, BSSID and do beacon-sync only when
1568          * the HW opmode is STATION.
1569          *
1570          * But the primary bit is set above in any case.
1571          */
1572         if (sc->sc_ah->opmode != NL80211_IFTYPE_STATION)
1573                 return;
1574
1575         memcpy(common->curbssid, bss_conf->bssid, ETH_ALEN);
1576         common->curaid = bss_conf->aid;
1577         ath9k_hw_write_associd(sc->sc_ah);
1578
1579         sc->last_rssi = ATH_RSSI_DUMMY_MARKER;
1580         sc->sc_ah->stats.avgbrssi = ATH_RSSI_DUMMY_MARKER;
1581
1582         spin_lock_irqsave(&sc->sc_pm_lock, flags);
1583         sc->ps_flags |= PS_BEACON_SYNC | PS_WAIT_FOR_BEACON;
1584         spin_unlock_irqrestore(&sc->sc_pm_lock, flags);
1585
1586         if (ath9k_hw_mci_is_enabled(sc->sc_ah))
1587                 ath9k_mci_update_wlan_channels(sc, false);
1588
1589         ath_dbg(common, CONFIG,
1590                 "Primary Station interface: %pM, BSSID: %pM\n",
1591                 vif->addr, common->curbssid);
1592 }
1593
1594 static void ath9k_bss_assoc_iter(void *data, u8 *mac, struct ieee80211_vif *vif)
1595 {
1596         struct ath_softc *sc = data;
1597         struct ieee80211_bss_conf *bss_conf = &vif->bss_conf;
1598
1599         if (test_bit(SC_OP_PRIM_STA_VIF, &sc->sc_flags))
1600                 return;
1601
1602         if (bss_conf->assoc)
1603                 ath9k_set_assoc_state(sc, vif);
1604 }
1605
1606 static void ath9k_bss_info_changed(struct ieee80211_hw *hw,
1607                                    struct ieee80211_vif *vif,
1608                                    struct ieee80211_bss_conf *bss_conf,
1609                                    u32 changed)
1610 {
1611 #define CHECK_ANI                               \
1612         (BSS_CHANGED_ASSOC |                    \
1613          BSS_CHANGED_IBSS |                     \
1614          BSS_CHANGED_BEACON_ENABLED)
1615
1616         struct ath_softc *sc = hw->priv;
1617         struct ath_hw *ah = sc->sc_ah;
1618         struct ath_common *common = ath9k_hw_common(ah);
1619         struct ath_vif *avp = (void *)vif->drv_priv;
1620         int slottime;
1621
1622         ath9k_ps_wakeup(sc);
1623         mutex_lock(&sc->mutex);
1624
1625         if (changed & BSS_CHANGED_ASSOC) {
1626                 ath_dbg(common, CONFIG, "BSSID %pM Changed ASSOC %d\n",
1627                         bss_conf->bssid, bss_conf->assoc);
1628
1629                 if (avp->primary_sta_vif && !bss_conf->assoc) {
1630                         clear_bit(SC_OP_PRIM_STA_VIF, &sc->sc_flags);
1631                         avp->primary_sta_vif = false;
1632
1633                         if (ah->opmode == NL80211_IFTYPE_STATION)
1634                                 clear_bit(SC_OP_BEACONS, &sc->sc_flags);
1635                 }
1636
1637                 ieee80211_iterate_active_interfaces_atomic(
1638                         sc->hw, IEEE80211_IFACE_ITER_RESUME_ALL,
1639                         ath9k_bss_assoc_iter, sc);
1640
1641                 if (!test_bit(SC_OP_PRIM_STA_VIF, &sc->sc_flags) &&
1642                     ah->opmode == NL80211_IFTYPE_STATION) {
1643                         memset(common->curbssid, 0, ETH_ALEN);
1644                         common->curaid = 0;
1645                         ath9k_hw_write_associd(sc->sc_ah);
1646                         if (ath9k_hw_mci_is_enabled(sc->sc_ah))
1647                                 ath9k_mci_update_wlan_channels(sc, true);
1648                 }
1649         }
1650
1651         if (changed & BSS_CHANGED_IBSS) {
1652                 memcpy(common->curbssid, bss_conf->bssid, ETH_ALEN);
1653                 common->curaid = bss_conf->aid;
1654                 ath9k_hw_write_associd(sc->sc_ah);
1655         }
1656
1657         if ((changed & BSS_CHANGED_BEACON_ENABLED) ||
1658             (changed & BSS_CHANGED_BEACON_INT)) {
1659                 if (ah->opmode == NL80211_IFTYPE_AP &&
1660                     bss_conf->enable_beacon)
1661                         ath9k_set_tsfadjust(sc, vif);
1662                 if (ath9k_allow_beacon_config(sc, vif))
1663                         ath9k_beacon_config(sc, vif, changed);
1664         }
1665
1666         if (changed & BSS_CHANGED_ERP_SLOT) {
1667                 if (bss_conf->use_short_slot)
1668                         slottime = 9;
1669                 else
1670                         slottime = 20;
1671                 if (vif->type == NL80211_IFTYPE_AP) {
1672                         /*
1673                          * Defer update, so that connected stations can adjust
1674                          * their settings at the same time.
1675                          * See beacon.c for more details
1676                          */
1677                         sc->beacon.slottime = slottime;
1678                         sc->beacon.updateslot = UPDATE;
1679                 } else {
1680                         ah->slottime = slottime;
1681                         ath9k_hw_init_global_settings(ah);
1682                 }
1683         }
1684
1685         if (changed & CHECK_ANI)
1686                 ath_check_ani(sc);
1687
1688         mutex_unlock(&sc->mutex);
1689         ath9k_ps_restore(sc);
1690
1691 #undef CHECK_ANI
1692 }
1693
1694 static u64 ath9k_get_tsf(struct ieee80211_hw *hw, struct ieee80211_vif *vif)
1695 {
1696         struct ath_softc *sc = hw->priv;
1697         u64 tsf;
1698
1699         mutex_lock(&sc->mutex);
1700         ath9k_ps_wakeup(sc);
1701         tsf = ath9k_hw_gettsf64(sc->sc_ah);
1702         ath9k_ps_restore(sc);
1703         mutex_unlock(&sc->mutex);
1704
1705         return tsf;
1706 }
1707
1708 static void ath9k_set_tsf(struct ieee80211_hw *hw,
1709                           struct ieee80211_vif *vif,
1710                           u64 tsf)
1711 {
1712         struct ath_softc *sc = hw->priv;
1713
1714         mutex_lock(&sc->mutex);
1715         ath9k_ps_wakeup(sc);
1716         ath9k_hw_settsf64(sc->sc_ah, tsf);
1717         ath9k_ps_restore(sc);
1718         mutex_unlock(&sc->mutex);
1719 }
1720
1721 static void ath9k_reset_tsf(struct ieee80211_hw *hw, struct ieee80211_vif *vif)
1722 {
1723         struct ath_softc *sc = hw->priv;
1724
1725         mutex_lock(&sc->mutex);
1726
1727         ath9k_ps_wakeup(sc);
1728         ath9k_hw_reset_tsf(sc->sc_ah);
1729         ath9k_ps_restore(sc);
1730
1731         mutex_unlock(&sc->mutex);
1732 }
1733
1734 static int ath9k_ampdu_action(struct ieee80211_hw *hw,
1735                               struct ieee80211_vif *vif,
1736                               enum ieee80211_ampdu_mlme_action action,
1737                               struct ieee80211_sta *sta,
1738                               u16 tid, u16 *ssn, u8 buf_size)
1739 {
1740         struct ath_softc *sc = hw->priv;
1741         bool flush = false;
1742         int ret = 0;
1743
1744         mutex_lock(&sc->mutex);
1745
1746         switch (action) {
1747         case IEEE80211_AMPDU_RX_START:
1748                 break;
1749         case IEEE80211_AMPDU_RX_STOP:
1750                 break;
1751         case IEEE80211_AMPDU_TX_START:
1752                 ath9k_ps_wakeup(sc);
1753                 ret = ath_tx_aggr_start(sc, sta, tid, ssn);
1754                 if (!ret)
1755                         ieee80211_start_tx_ba_cb_irqsafe(vif, sta->addr, tid);
1756                 ath9k_ps_restore(sc);
1757                 break;
1758         case IEEE80211_AMPDU_TX_STOP_FLUSH:
1759         case IEEE80211_AMPDU_TX_STOP_FLUSH_CONT:
1760                 flush = true;
1761         case IEEE80211_AMPDU_TX_STOP_CONT:
1762                 ath9k_ps_wakeup(sc);
1763                 ath_tx_aggr_stop(sc, sta, tid);
1764                 if (!flush)
1765                         ieee80211_stop_tx_ba_cb_irqsafe(vif, sta->addr, tid);
1766                 ath9k_ps_restore(sc);
1767                 break;
1768         case IEEE80211_AMPDU_TX_OPERATIONAL:
1769                 ath9k_ps_wakeup(sc);
1770                 ath_tx_aggr_resume(sc, sta, tid);
1771                 ath9k_ps_restore(sc);
1772                 break;
1773         default:
1774                 ath_err(ath9k_hw_common(sc->sc_ah), "Unknown AMPDU action\n");
1775         }
1776
1777         mutex_unlock(&sc->mutex);
1778
1779         return ret;
1780 }
1781
1782 static int ath9k_get_survey(struct ieee80211_hw *hw, int idx,
1783                              struct survey_info *survey)
1784 {
1785         struct ath_softc *sc = hw->priv;
1786         struct ath_common *common = ath9k_hw_common(sc->sc_ah);
1787         struct ieee80211_supported_band *sband;
1788         struct ieee80211_channel *chan;
1789         unsigned long flags;
1790         int pos;
1791
1792         if (config_enabled(CONFIG_ATH9K_TX99))
1793                 return -EOPNOTSUPP;
1794
1795         spin_lock_irqsave(&common->cc_lock, flags);
1796         if (idx == 0)
1797                 ath_update_survey_stats(sc);
1798
1799         sband = hw->wiphy->bands[IEEE80211_BAND_2GHZ];
1800         if (sband && idx >= sband->n_channels) {
1801                 idx -= sband->n_channels;
1802                 sband = NULL;
1803         }
1804
1805         if (!sband)
1806                 sband = hw->wiphy->bands[IEEE80211_BAND_5GHZ];
1807
1808         if (!sband || idx >= sband->n_channels) {
1809                 spin_unlock_irqrestore(&common->cc_lock, flags);
1810                 return -ENOENT;
1811         }
1812
1813         chan = &sband->channels[idx];
1814         pos = chan->hw_value;
1815         memcpy(survey, &sc->survey[pos], sizeof(*survey));
1816         survey->channel = chan;
1817         spin_unlock_irqrestore(&common->cc_lock, flags);
1818
1819         return 0;
1820 }
1821
1822 static void ath9k_set_coverage_class(struct ieee80211_hw *hw, u8 coverage_class)
1823 {
1824         struct ath_softc *sc = hw->priv;
1825         struct ath_hw *ah = sc->sc_ah;
1826
1827         if (config_enabled(CONFIG_ATH9K_TX99))
1828                 return;
1829
1830         mutex_lock(&sc->mutex);
1831         ah->coverage_class = coverage_class;
1832
1833         ath9k_ps_wakeup(sc);
1834         ath9k_hw_init_global_settings(ah);
1835         ath9k_ps_restore(sc);
1836
1837         mutex_unlock(&sc->mutex);
1838 }
1839
1840 static void ath9k_flush(struct ieee80211_hw *hw, u32 queues, bool drop)
1841 {
1842         struct ath_softc *sc = hw->priv;
1843         struct ath_hw *ah = sc->sc_ah;
1844         struct ath_common *common = ath9k_hw_common(ah);
1845         int timeout = 200; /* ms */
1846         int i, j;
1847         bool drain_txq;
1848
1849         mutex_lock(&sc->mutex);
1850         cancel_delayed_work_sync(&sc->tx_complete_work);
1851
1852         if (ah->ah_flags & AH_UNPLUGGED) {
1853                 ath_dbg(common, ANY, "Device has been unplugged!\n");
1854                 mutex_unlock(&sc->mutex);
1855                 return;
1856         }
1857
1858         if (test_bit(SC_OP_INVALID, &sc->sc_flags)) {
1859                 ath_dbg(common, ANY, "Device not present\n");
1860                 mutex_unlock(&sc->mutex);
1861                 return;
1862         }
1863
1864         for (j = 0; j < timeout; j++) {
1865                 bool npend = false;
1866
1867                 if (j)
1868                         usleep_range(1000, 2000);
1869
1870                 for (i = 0; i < ATH9K_NUM_TX_QUEUES; i++) {
1871                         if (!ATH_TXQ_SETUP(sc, i))
1872                                 continue;
1873
1874                         npend = ath9k_has_pending_frames(sc, &sc->tx.txq[i]);
1875
1876                         if (npend)
1877                                 break;
1878                 }
1879
1880                 if (!npend)
1881                     break;
1882         }
1883
1884         if (drop) {
1885                 ath9k_ps_wakeup(sc);
1886                 spin_lock_bh(&sc->sc_pcu_lock);
1887                 drain_txq = ath_drain_all_txq(sc);
1888                 spin_unlock_bh(&sc->sc_pcu_lock);
1889
1890                 if (!drain_txq)
1891                         ath_reset(sc);
1892
1893                 ath9k_ps_restore(sc);
1894                 ieee80211_wake_queues(hw);
1895         }
1896
1897         ieee80211_queue_delayed_work(hw, &sc->tx_complete_work, 0);
1898         mutex_unlock(&sc->mutex);
1899 }
1900
1901 static bool ath9k_tx_frames_pending(struct ieee80211_hw *hw)
1902 {
1903         struct ath_softc *sc = hw->priv;
1904         int i;
1905
1906         for (i = 0; i < ATH9K_NUM_TX_QUEUES; i++) {
1907                 if (!ATH_TXQ_SETUP(sc, i))
1908                         continue;
1909
1910                 if (ath9k_has_pending_frames(sc, &sc->tx.txq[i]))
1911                         return true;
1912         }
1913         return false;
1914 }
1915
1916 static int ath9k_tx_last_beacon(struct ieee80211_hw *hw)
1917 {
1918         struct ath_softc *sc = hw->priv;
1919         struct ath_hw *ah = sc->sc_ah;
1920         struct ieee80211_vif *vif;
1921         struct ath_vif *avp;
1922         struct ath_buf *bf;
1923         struct ath_tx_status ts;
1924         bool edma = !!(ah->caps.hw_caps & ATH9K_HW_CAP_EDMA);
1925         int status;
1926
1927         vif = sc->beacon.bslot[0];
1928         if (!vif)
1929                 return 0;
1930
1931         if (!vif->bss_conf.enable_beacon)
1932                 return 0;
1933
1934         avp = (void *)vif->drv_priv;
1935
1936         if (!sc->beacon.tx_processed && !edma) {
1937                 tasklet_disable(&sc->bcon_tasklet);
1938
1939                 bf = avp->av_bcbuf;
1940                 if (!bf || !bf->bf_mpdu)
1941                         goto skip;
1942
1943                 status = ath9k_hw_txprocdesc(ah, bf->bf_desc, &ts);
1944                 if (status == -EINPROGRESS)
1945                         goto skip;
1946
1947                 sc->beacon.tx_processed = true;
1948                 sc->beacon.tx_last = !(ts.ts_status & ATH9K_TXERR_MASK);
1949
1950 skip:
1951                 tasklet_enable(&sc->bcon_tasklet);
1952         }
1953
1954         return sc->beacon.tx_last;
1955 }
1956
1957 static int ath9k_get_stats(struct ieee80211_hw *hw,
1958                            struct ieee80211_low_level_stats *stats)
1959 {
1960         struct ath_softc *sc = hw->priv;
1961         struct ath_hw *ah = sc->sc_ah;
1962         struct ath9k_mib_stats *mib_stats = &ah->ah_mibStats;
1963
1964         stats->dot11ACKFailureCount = mib_stats->ackrcv_bad;
1965         stats->dot11RTSFailureCount = mib_stats->rts_bad;
1966         stats->dot11FCSErrorCount = mib_stats->fcs_bad;
1967         stats->dot11RTSSuccessCount = mib_stats->rts_good;
1968         return 0;
1969 }
1970
1971 static u32 fill_chainmask(u32 cap, u32 new)
1972 {
1973         u32 filled = 0;
1974         int i;
1975
1976         for (i = 0; cap && new; i++, cap >>= 1) {
1977                 if (!(cap & BIT(0)))
1978                         continue;
1979
1980                 if (new & BIT(0))
1981                         filled |= BIT(i);
1982
1983                 new >>= 1;
1984         }
1985
1986         return filled;
1987 }
1988
1989 static bool validate_antenna_mask(struct ath_hw *ah, u32 val)
1990 {
1991         if (AR_SREV_9300_20_OR_LATER(ah))
1992                 return true;
1993
1994         switch (val & 0x7) {
1995         case 0x1:
1996         case 0x3:
1997         case 0x7:
1998                 return true;
1999         case 0x2:
2000                 return (ah->caps.rx_chainmask == 1);
2001         default:
2002                 return false;
2003         }
2004 }
2005
2006 static int ath9k_set_antenna(struct ieee80211_hw *hw, u32 tx_ant, u32 rx_ant)
2007 {
2008         struct ath_softc *sc = hw->priv;
2009         struct ath_hw *ah = sc->sc_ah;
2010
2011         if (ah->caps.rx_chainmask != 1)
2012                 rx_ant |= tx_ant;
2013
2014         if (!validate_antenna_mask(ah, rx_ant) || !tx_ant)
2015                 return -EINVAL;
2016
2017         sc->ant_rx = rx_ant;
2018         sc->ant_tx = tx_ant;
2019
2020         if (ah->caps.rx_chainmask == 1)
2021                 return 0;
2022
2023         /* AR9100 runs into calibration issues if not all rx chains are enabled */
2024         if (AR_SREV_9100(ah))
2025                 ah->rxchainmask = 0x7;
2026         else
2027                 ah->rxchainmask = fill_chainmask(ah->caps.rx_chainmask, rx_ant);
2028
2029         ah->txchainmask = fill_chainmask(ah->caps.tx_chainmask, tx_ant);
2030         ath9k_reload_chainmask_settings(sc);
2031
2032         return 0;
2033 }
2034
2035 static int ath9k_get_antenna(struct ieee80211_hw *hw, u32 *tx_ant, u32 *rx_ant)
2036 {
2037         struct ath_softc *sc = hw->priv;
2038
2039         *tx_ant = sc->ant_tx;
2040         *rx_ant = sc->ant_rx;
2041         return 0;
2042 }
2043
2044 static void ath9k_sw_scan_start(struct ieee80211_hw *hw)
2045 {
2046         struct ath_softc *sc = hw->priv;
2047         set_bit(SC_OP_SCANNING, &sc->sc_flags);
2048 }
2049
2050 static void ath9k_sw_scan_complete(struct ieee80211_hw *hw)
2051 {
2052         struct ath_softc *sc = hw->priv;
2053         clear_bit(SC_OP_SCANNING, &sc->sc_flags);
2054 }
2055
2056 static void ath9k_channel_switch_beacon(struct ieee80211_hw *hw,
2057                                         struct ieee80211_vif *vif,
2058                                         struct cfg80211_chan_def *chandef)
2059 {
2060         struct ath_softc *sc = hw->priv;
2061
2062         /* mac80211 does not support CSA in multi-if cases (yet) */
2063         if (WARN_ON(sc->csa_vif))
2064                 return;
2065
2066         sc->csa_vif = vif;
2067 }
2068
2069 struct ieee80211_ops ath9k_ops = {
2070         .tx                 = ath9k_tx,
2071         .start              = ath9k_start,
2072         .stop               = ath9k_stop,
2073         .add_interface      = ath9k_add_interface,
2074         .change_interface   = ath9k_change_interface,
2075         .remove_interface   = ath9k_remove_interface,
2076         .config             = ath9k_config,
2077         .configure_filter   = ath9k_configure_filter,
2078         .sta_add            = ath9k_sta_add,
2079         .sta_remove         = ath9k_sta_remove,
2080         .sta_notify         = ath9k_sta_notify,
2081         .conf_tx            = ath9k_conf_tx,
2082         .bss_info_changed   = ath9k_bss_info_changed,
2083         .set_key            = ath9k_set_key,
2084         .get_tsf            = ath9k_get_tsf,
2085         .set_tsf            = ath9k_set_tsf,
2086         .reset_tsf          = ath9k_reset_tsf,
2087         .ampdu_action       = ath9k_ampdu_action,
2088         .get_survey         = ath9k_get_survey,
2089         .rfkill_poll        = ath9k_rfkill_poll_state,
2090         .set_coverage_class = ath9k_set_coverage_class,
2091         .flush              = ath9k_flush,
2092         .tx_frames_pending  = ath9k_tx_frames_pending,
2093         .tx_last_beacon     = ath9k_tx_last_beacon,
2094         .release_buffered_frames = ath9k_release_buffered_frames,
2095         .get_stats          = ath9k_get_stats,
2096         .set_antenna        = ath9k_set_antenna,
2097         .get_antenna        = ath9k_get_antenna,
2098
2099 #ifdef CONFIG_ATH9K_WOW
2100         .suspend            = ath9k_suspend,
2101         .resume             = ath9k_resume,
2102         .set_wakeup         = ath9k_set_wakeup,
2103 #endif
2104
2105 #ifdef CONFIG_ATH9K_DEBUGFS
2106         .get_et_sset_count  = ath9k_get_et_sset_count,
2107         .get_et_stats       = ath9k_get_et_stats,
2108         .get_et_strings     = ath9k_get_et_strings,
2109 #endif
2110
2111 #if defined(CONFIG_MAC80211_DEBUGFS) && defined(CONFIG_ATH9K_DEBUGFS)
2112         .sta_add_debugfs    = ath9k_sta_add_debugfs,
2113 #endif
2114         .sw_scan_start      = ath9k_sw_scan_start,
2115         .sw_scan_complete   = ath9k_sw_scan_complete,
2116         .channel_switch_beacon     = ath9k_channel_switch_beacon,
2117 };