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