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