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