mac80211: merge some pending fixes
[openwrt.git] / package / mac80211 / patches / 300-pending_work.patch
index 0900596..7319079 100644 (file)
+--- a/drivers/net/wireless/ath/ath5k/base.c
++++ b/drivers/net/wireless/ath/ath5k/base.c
+@@ -2417,6 +2417,22 @@ ath5k_tx_complete_poll_work(struct work_
+ * Initialization routines *
+ \*************************/
++static const struct ieee80211_iface_limit if_limits[] = {
++      { .max = 2048,  .types = BIT(NL80211_IFTYPE_STATION) },
++      { .max = 4,     .types =
++#ifdef CONFIG_MAC80211_MESH
++                               BIT(NL80211_IFTYPE_MESH_POINT) |
++#endif
++                               BIT(NL80211_IFTYPE_AP) },
++};
++
++static const struct ieee80211_iface_combination if_comb = {
++      .limits = if_limits,
++      .n_limits = ARRAY_SIZE(if_limits),
++      .max_interfaces = 2048,
++      .num_different_channels = 1,
++};
++
+ int __devinit
+ ath5k_init_ah(struct ath5k_hw *ah, const struct ath_bus_ops *bus_ops)
+ {
+@@ -2438,6 +2454,9 @@ ath5k_init_ah(struct ath5k_hw *ah, const
+               BIT(NL80211_IFTYPE_ADHOC) |
+               BIT(NL80211_IFTYPE_MESH_POINT);
++      hw->wiphy->iface_combinations = &if_comb;
++      hw->wiphy->n_iface_combinations = 1;
++
+       /* SW support for IBSS_RSN is provided by mac80211 */
+       hw->wiphy->flags |= WIPHY_FLAG_IBSS_RSN;
 --- a/net/mac80211/agg-rx.c
 +++ b/net/mac80211/agg-rx.c
-@@ -176,6 +176,8 @@ static void ieee80211_send_addba_resp(st
-               memcpy(mgmt->bssid, sdata->vif.addr, ETH_ALEN);
-       else if (sdata->vif.type == NL80211_IFTYPE_STATION)
+@@ -145,15 +145,20 @@ static void sta_rx_agg_session_timer_exp
+       struct tid_ampdu_rx *tid_rx;
+       unsigned long timeout;
++      rcu_read_lock();
+       tid_rx = rcu_dereference(sta->ampdu_mlme.tid_rx[*ptid]);
+-      if (!tid_rx)
++      if (!tid_rx) {
++              rcu_read_unlock();
+               return;
++      }
+       timeout = tid_rx->last_rx + TU_TO_JIFFIES(tid_rx->timeout);
+       if (time_is_after_jiffies(timeout)) {
+               mod_timer(&tid_rx->session_timer, timeout);
++              rcu_read_unlock();
+               return;
+       }
++      rcu_read_unlock();
+ #ifdef CONFIG_MAC80211_HT_DEBUG
+       printk(KERN_DEBUG "rx session timer expired on tid %d\n", (u16)*ptid);
+@@ -200,6 +205,8 @@ static void ieee80211_send_addba_resp(st
                memcpy(mgmt->bssid, sdata->u.mgd.bssid, ETH_ALEN);
+       else if (sdata->vif.type == NL80211_IFTYPE_ADHOC)
+               memcpy(mgmt->bssid, sdata->u.ibss.bssid, ETH_ALEN);
 +      else if (sdata->vif.type == NL80211_IFTYPE_WDS)
 +              memcpy(mgmt->bssid, da, ETH_ALEN);
  
        mgmt->frame_control = cpu_to_le16(IEEE80211_FTYPE_MGMT |
                                          IEEE80211_STYPE_ACTION);
-@@ -262,7 +264,11 @@ void ieee80211_process_addba_request(str
-                               "%pM on tid %u\n",
-                               mgmt->sa, tid);
- #endif /* CONFIG_MAC80211_HT_DEBUG */
--              goto end;
-+
-+              /* delete existing Rx BA session on the same tid */
-+              ___ieee80211_stop_rx_ba_session(sta, tid, WLAN_BACK_RECIPIENT,
-+                                              WLAN_STATUS_UNSPECIFIED_QOS,
-+                                              false);
-       }
-       /* prepare A-MPDU MLME for Rx aggregation */
 --- a/net/mac80211/agg-tx.c
 +++ b/net/mac80211/agg-tx.c
-@@ -79,7 +79,8 @@ static void ieee80211_send_addba_request
-       memcpy(mgmt->da, da, ETH_ALEN);
+@@ -81,7 +81,8 @@ static void ieee80211_send_addba_request
        memcpy(mgmt->sa, sdata->vif.addr, ETH_ALEN);
        if (sdata->vif.type == NL80211_IFTYPE_AP ||
--          sdata->vif.type == NL80211_IFTYPE_AP_VLAN)
-+          sdata->vif.type == NL80211_IFTYPE_AP_VLAN ||
+           sdata->vif.type == NL80211_IFTYPE_AP_VLAN ||
+-          sdata->vif.type == NL80211_IFTYPE_MESH_POINT)
++          sdata->vif.type == NL80211_IFTYPE_MESH_POINT ||
 +          sdata->vif.type == NL80211_IFTYPE_WDS)
                memcpy(mgmt->bssid, sdata->vif.addr, ETH_ALEN);
        else if (sdata->vif.type == NL80211_IFTYPE_STATION)
                memcpy(mgmt->bssid, sdata->u.mgd.bssid, ETH_ALEN);
-@@ -398,7 +399,8 @@ int ieee80211_start_tx_ba_session(struct
-        */
-       if (sdata->vif.type != NL80211_IFTYPE_STATION &&
+@@ -490,6 +491,7 @@ int ieee80211_start_tx_ba_session(struct
+           sdata->vif.type != NL80211_IFTYPE_MESH_POINT &&
            sdata->vif.type != NL80211_IFTYPE_AP_VLAN &&
--          sdata->vif.type != NL80211_IFTYPE_AP)
-+          sdata->vif.type != NL80211_IFTYPE_AP &&
-+          sdata->vif.type != NL80211_IFTYPE_WDS)
+           sdata->vif.type != NL80211_IFTYPE_AP &&
++          sdata->vif.type != NL80211_IFTYPE_WDS &&
+           sdata->vif.type != NL80211_IFTYPE_ADHOC)
                return -EINVAL;
  
-       if (test_sta_flags(sta, WLAN_STA_BLOCK_BA)) {
 --- a/net/mac80211/debugfs_sta.c
 +++ b/net/mac80211/debugfs_sta.c
-@@ -59,7 +59,7 @@ static ssize_t sta_flags_read(struct fil
-       char buf[100];
-       struct sta_info *sta = file->private_data;
-       u32 staflags = get_sta_flags(sta);
--      int res = scnprintf(buf, sizeof(buf), "%s%s%s%s%s%s%s%s%s",
-+      int res = scnprintf(buf, sizeof(buf), "%s%s%s%s%s%s%s%s",
-               staflags & WLAN_STA_AUTH ? "AUTH\n" : "",
-               staflags & WLAN_STA_ASSOC ? "ASSOC\n" : "",
-               staflags & WLAN_STA_PS_STA ? "PS (sta)\n" : "",
-@@ -67,7 +67,6 @@ static ssize_t sta_flags_read(struct fil
-               staflags & WLAN_STA_AUTHORIZED ? "AUTHORIZED\n" : "",
-               staflags & WLAN_STA_SHORT_PREAMBLE ? "SHORT PREAMBLE\n" : "",
-               staflags & WLAN_STA_WME ? "WME\n" : "",
--              staflags & WLAN_STA_WDS ? "WDS\n" : "",
-               staflags & WLAN_STA_MFP ? "MFP\n" : "");
-       return simple_read_from_buffer(userbuf, count, ppos, buf, res);
- }
+@@ -63,11 +63,11 @@ static ssize_t sta_flags_read(struct fil
+       test_sta_flag(sta, WLAN_STA_##flg) ? #flg "\n" : ""
+       int res = scnprintf(buf, sizeof(buf),
+-                          "%s%s%s%s%s%s%s%s%s%s%s%s%s%s%s%s%s%s%s%s",
++                          "%s%s%s%s%s%s%s%s%s%s%s%s%s%s%s%s%s%s%s",
+                           TEST(AUTH), TEST(ASSOC), TEST(PS_STA),
+                           TEST(PS_DRIVER), TEST(AUTHORIZED),
+                           TEST(SHORT_PREAMBLE),
+-                          TEST(WME), TEST(WDS), TEST(CLEAR_PS_FILT),
++                          TEST(WME), TEST(CLEAR_PS_FILT),
+                           TEST(MFP), TEST(BLOCK_BA), TEST(PSPOLL),
+                           TEST(UAPSD), TEST(SP), TEST(TDLS_PEER),
+                           TEST(TDLS_PEER_AUTH), TEST(4ADDR_EVENT),
 --- a/net/mac80211/iface.c
 +++ b/net/mac80211/iface.c
-@@ -178,7 +178,6 @@ static int ieee80211_do_open(struct net_
+@@ -284,7 +284,6 @@ static int ieee80211_do_open(struct net_
  {
        struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
        struct ieee80211_local *local = sdata->local;
        u32 changed = 0;
        int res;
        u32 hw_reconf_flags = 0;
-@@ -290,27 +289,6 @@ static int ieee80211_do_open(struct net_
+@@ -430,28 +429,6 @@ static int ieee80211_do_open(struct net_
  
        set_bit(SDATA_STATE_RUNNING, &sdata->state);
  
 -                      goto err_del_interface;
 -              }
 -
--              /* no locking required since STA is not live yet */
--              sta->flags |= WLAN_STA_AUTHORIZED;
+-              sta_info_pre_move_state(sta, IEEE80211_STA_AUTH);
+-              sta_info_pre_move_state(sta, IEEE80211_STA_ASSOC);
+-              sta_info_pre_move_state(sta, IEEE80211_STA_AUTHORIZED);
 -
 -              res = sta_info_insert(sta);
 -              if (res) {
        /*
         * set_multicast_list will be invoked by the networking core
         * which will check whether any increments here were done in
-@@ -344,8 +322,7 @@ static int ieee80211_do_open(struct net_
-       netif_tx_start_all_queues(dev);
-       return 0;
-- err_del_interface:
--      drv_remove_interface(local, &sdata->vif);
-+
-  err_stop:
-       if (!local->open_count)
-               drv_stop(local);
-@@ -718,6 +695,70 @@ static void ieee80211_if_setup(struct ne
+@@ -642,6 +619,8 @@ static void ieee80211_do_stop(struct iee
+               ieee80211_configure_filter(local);
+               break;
+       default:
++              flush_work(&local->hw_roc_start);
++              flush_work(&local->hw_roc_done);
+               flush_work(&sdata->work);
+               /*
+                * When we get here, the interface is marked down.
+@@ -848,6 +827,72 @@ static void ieee80211_if_setup(struct ne
        dev->destructor = free_netdev;
  }
  
 +      ieee802_11_parse_elems(mgmt->u.probe_resp.variable,
 +                             skb->len - baselen, &elems);
 +
-+      rates = ieee80211_sta_get_rates(local, &elems, band);
++      rates = ieee80211_sta_get_rates(local, &elems, band, NULL);
 +
 +      rcu_read_lock();
 +
 +      sta->sta.supp_rates[local->hw.conf.channel->band] = rates;
 +
 +      if (elems.ht_cap_elem)
-+              ieee80211_ht_cap_ie_to_sta_ht_cap(sband,
++              ieee80211_ht_cap_ie_to_sta_ht_cap(sdata, sband,
 +                              elems.ht_cap_elem, &sta->sta.ht_cap);
 +
 +      if (elems.wmm_param)
-+              set_sta_flags(sta, WLAN_STA_WME);
++              set_sta_flag(sta, WLAN_STA_WME);
 +
 +      if (new) {
-+              sta->flags = WLAN_STA_AUTHORIZED;
++              sta_info_pre_move_state(sta, IEEE80211_STA_AUTH);
++              sta_info_pre_move_state(sta, IEEE80211_STA_ASSOC);
++              sta_info_pre_move_state(sta, IEEE80211_STA_AUTHORIZED);
 +              rate_control_rate_init(sta);
 +              sta_info_insert_rcu(sta);
 +      }
  static void ieee80211_iface_work(struct work_struct *work)
  {
        struct ieee80211_sub_if_data *sdata =
-@@ -822,6 +863,9 @@ static void ieee80211_iface_work(struct 
+@@ -952,6 +997,9 @@ static void ieee80211_iface_work(struct 
                                break;
                        ieee80211_mesh_rx_queued_mgmt(sdata, skb);
                        break;
                        break;
 --- a/net/mac80211/rx.c
 +++ b/net/mac80211/rx.c
-@@ -2137,7 +2137,8 @@ ieee80211_rx_h_action(struct ieee80211_r
-                */
-               if (sdata->vif.type != NL80211_IFTYPE_STATION &&
+@@ -2284,6 +2284,7 @@ ieee80211_rx_h_action(struct ieee80211_r
+                   sdata->vif.type != NL80211_IFTYPE_MESH_POINT &&
                    sdata->vif.type != NL80211_IFTYPE_AP_VLAN &&
--                  sdata->vif.type != NL80211_IFTYPE_AP)
-+                  sdata->vif.type != NL80211_IFTYPE_AP &&
-+                  sdata->vif.type != NL80211_IFTYPE_WDS)
+                   sdata->vif.type != NL80211_IFTYPE_AP &&
++                  sdata->vif.type != NL80211_IFTYPE_WDS &&
+                   sdata->vif.type != NL80211_IFTYPE_ADHOC)
                        break;
  
-               /* verify action_code is present */
-@@ -2335,13 +2336,14 @@ ieee80211_rx_h_mgmt(struct ieee80211_rx_
+@@ -2498,14 +2499,15 @@ ieee80211_rx_h_mgmt(struct ieee80211_rx_
  
        if (!ieee80211_vif_is_mesh(&sdata->vif) &&
            sdata->vif.type != NL80211_IFTYPE_ADHOC &&
                return RX_DROP_MONITOR;
  
        switch (stype) {
+       case cpu_to_le16(IEEE80211_STYPE_AUTH):
        case cpu_to_le16(IEEE80211_STYPE_BEACON):
        case cpu_to_le16(IEEE80211_STYPE_PROBE_RESP):
 -              /* process for all: mesh, mlme, ibss */
 +              /* process for all: mesh, mlme, ibss, wds */
                break;
-       case cpu_to_le16(IEEE80211_STYPE_DEAUTH):
-       case cpu_to_le16(IEEE80211_STYPE_DISASSOC):
-@@ -2680,10 +2682,16 @@ static int prepare_for_handlers(struct i
+       case cpu_to_le16(IEEE80211_STYPE_ASSOC_RESP):
+       case cpu_to_le16(IEEE80211_STYPE_REASSOC_RESP):
+@@ -2839,10 +2841,16 @@ static int prepare_for_handlers(struct i
                }
                break;
        case NL80211_IFTYPE_WDS:
                /* should never get here */
 --- a/net/mac80211/sta_info.h
 +++ b/net/mac80211/sta_info.h
-@@ -31,7 +31,6 @@
+@@ -32,7 +32,6 @@
+  * @WLAN_STA_SHORT_PREAMBLE: Station is capable of receiving short-preamble
   *    frames.
-  * @WLAN_STA_ASSOC_AP: We're associated to that station, it is an AP.
   * @WLAN_STA_WME: Station is a QoS-STA.
 - * @WLAN_STA_WDS: Station is one of our WDS peers.
   * @WLAN_STA_CLEAR_PS_FILT: Clear PS filter in hardware (using the
   *    IEEE80211_TX_CTL_CLEAR_PS_FILT control flag) when the next
   *    frame to this station is transmitted.
-@@ -54,7 +53,6 @@ enum ieee80211_sta_info_flags {
-       WLAN_STA_SHORT_PREAMBLE = 1<<4,
-       WLAN_STA_ASSOC_AP       = 1<<5,
-       WLAN_STA_WME            = 1<<6,
--      WLAN_STA_WDS            = 1<<7,
-       WLAN_STA_CLEAR_PS_FILT  = 1<<9,
-       WLAN_STA_MFP            = 1<<10,
-       WLAN_STA_BLOCK_BA       = 1<<11,
---- a/drivers/net/wireless/ath/ath9k/beacon.c
-+++ b/drivers/net/wireless/ath/ath9k/beacon.c
-@@ -384,7 +384,9 @@ void ath_beacon_tasklet(unsigned long da
-                       ath_dbg(common, ATH_DBG_BSTUCK,
-                               "beacon is officially stuck\n");
-                       sc->sc_flags |= SC_OP_TSF_RESET;
-+                      spin_lock(&sc->sc_pcu_lock);
-                       ath_reset(sc, true);
-+                      spin_unlock(&sc->sc_pcu_lock);
+@@ -64,7 +63,6 @@ enum ieee80211_sta_info_flags {
+       WLAN_STA_AUTHORIZED,
+       WLAN_STA_SHORT_PREAMBLE,
+       WLAN_STA_WME,
+-      WLAN_STA_WDS,
+       WLAN_STA_CLEAR_PS_FILT,
+       WLAN_STA_MFP,
+       WLAN_STA_BLOCK_BA,
+--- a/net/mac80211/util.c
++++ b/net/mac80211/util.c
+@@ -804,7 +804,7 @@ void ieee80211_set_wmm_default(struct ie
+       struct ieee80211_local *local = sdata->local;
+       struct ieee80211_tx_queue_params qparam;
+       int ac;
+-      bool use_11b;
++      bool use_11b, enable_qos;
+       int aCWmin, aCWmax;
+       if (!local->ops->conf_tx)
+@@ -818,6 +818,13 @@ void ieee80211_set_wmm_default(struct ie
+       use_11b = (local->hw.conf.channel->band == IEEE80211_BAND_2GHZ) &&
+                !(sdata->flags & IEEE80211_SDATA_OPERATING_GMODE);
++      /*
++       * By default disable QoS in STA mode for old access points, which do
++       * not support 802.11e. New APs will provide proper queue parameters,
++       * that we will configure later.
++       */
++      enable_qos = (sdata->vif.type != NL80211_IFTYPE_STATION);
++
+       for (ac = 0; ac < IEEE80211_NUM_ACS; ac++) {
+               /* Set defaults according to 802.11-2007 Table 7-37 */
+               aCWmax = 1023;
+@@ -826,38 +833,47 @@ void ieee80211_set_wmm_default(struct ie
+               else
+                       aCWmin = 15;
+-              switch (ac) {
+-              case IEEE80211_AC_BK:
+-                      qparam.cw_max = aCWmax;
+-                      qparam.cw_min = aCWmin;
+-                      qparam.txop = 0;
+-                      qparam.aifs = 7;
+-                      break;
+-              default: /* never happens but let's not leave undefined */
+-              case IEEE80211_AC_BE:
++              if (enable_qos) {
++                      switch (ac) {
++                      case IEEE80211_AC_BK:
++                              qparam.cw_max = aCWmax;
++                              qparam.cw_min = aCWmin;
++                              qparam.txop = 0;
++                              qparam.aifs = 7;
++                              break;
++                      /* never happens but let's not leave undefined */
++                      default:
++                      case IEEE80211_AC_BE:
++                              qparam.cw_max = aCWmax;
++                              qparam.cw_min = aCWmin;
++                              qparam.txop = 0;
++                              qparam.aifs = 3;
++                              break;
++                      case IEEE80211_AC_VI:
++                              qparam.cw_max = aCWmin;
++                              qparam.cw_min = (aCWmin + 1) / 2 - 1;
++                              if (use_11b)
++                                      qparam.txop = 6016/32;
++                              else
++                                      qparam.txop = 3008/32;
++                              qparam.aifs = 2;
++                              break;
++                      case IEEE80211_AC_VO:
++                              qparam.cw_max = (aCWmin + 1) / 2 - 1;
++                              qparam.cw_min = (aCWmin + 1) / 4 - 1;
++                              if (use_11b)
++                                      qparam.txop = 3264/32;
++                              else
++                                      qparam.txop = 1504/32;
++                              qparam.aifs = 2;
++                              break;
++                      }
++              } else {
++                      /* Confiure old 802.11b/g medium access rules. */
+                       qparam.cw_max = aCWmax;
+                       qparam.cw_min = aCWmin;
+                       qparam.txop = 0;
+-                      qparam.aifs = 3;
+-                      break;
+-              case IEEE80211_AC_VI:
+-                      qparam.cw_max = aCWmin;
+-                      qparam.cw_min = (aCWmin + 1) / 2 - 1;
+-                      if (use_11b)
+-                              qparam.txop = 6016/32;
+-                      else
+-                              qparam.txop = 3008/32;
+                       qparam.aifs = 2;
+-                      break;
+-              case IEEE80211_AC_VO:
+-                      qparam.cw_max = (aCWmin + 1) / 2 - 1;
+-                      qparam.cw_min = (aCWmin + 1) / 4 - 1;
+-                      if (use_11b)
+-                              qparam.txop = 3264/32;
+-                      else
+-                              qparam.txop = 1504/32;
+-                      qparam.aifs = 2;
+-                      break;
                }
  
-               return;
---- a/drivers/net/wireless/ath/ath9k/main.c
-+++ b/drivers/net/wireless/ath/ath9k/main.c
-@@ -617,8 +617,11 @@ void ath_hw_check(struct work_struct *wo
-       ath_dbg(common, ATH_DBG_RESET, "Possible baseband hang, "
-               "busy=%d (try %d)\n", busy, sc->hw_busy_count + 1);
-       if (busy >= 99) {
--              if (++sc->hw_busy_count >= 3)
-+              if (++sc->hw_busy_count >= 3) {
-+                      spin_lock_bh(&sc->sc_pcu_lock);
-                       ath_reset(sc, true);
-+                      spin_unlock_bh(&sc->sc_pcu_lock);
-+              }
-       } else if (busy >= 0)
-               sc->hw_busy_count = 0;
-@@ -637,7 +640,9 @@ static void ath_hw_pll_rx_hang_check(str
-                       /* Rx is hung for more than 500ms. Reset it */
-                       ath_dbg(common, ATH_DBG_RESET,
-                               "Possible RX hang, resetting");
-+                      spin_lock_bh(&sc->sc_pcu_lock);
-                       ath_reset(sc, true);
-+                      spin_unlock_bh(&sc->sc_pcu_lock);
-                       count = 0;
-               }
-       } else
-@@ -674,7 +679,9 @@ void ath9k_tasklet(unsigned long data)
-       if ((status & ATH9K_INT_FATAL) ||
-           (status & ATH9K_INT_BB_WATCHDOG)) {
-+              spin_lock(&sc->sc_pcu_lock);
-               ath_reset(sc, true);
-+              spin_unlock(&sc->sc_pcu_lock);
-               return;
+               qparam.uapsd = false;
+@@ -866,12 +882,8 @@ void ieee80211_set_wmm_default(struct ie
+               drv_conf_tx(local, sdata, ac, &qparam);
        }
  
-@@ -980,7 +987,6 @@ int ath_reset(struct ath_softc *sc, bool
-       del_timer_sync(&common->ani.timer);
-       ath9k_ps_wakeup(sc);
--      spin_lock_bh(&sc->sc_pcu_lock);
-       ieee80211_stop_queues(hw);
-@@ -1023,7 +1029,6 @@ int ath_reset(struct ath_softc *sc, bool
+-      /* after reinitialize QoS TX queues setting to default,
+-       * disable QoS at all */
+-
+       if (sdata->vif.type != NL80211_IFTYPE_MONITOR) {
+-              sdata->vif.bss_conf.qos =
+-                      sdata->vif.type != NL80211_IFTYPE_STATION;
++              sdata->vif.bss_conf.qos = enable_qos;
+               if (bss_notify)
+                       ieee80211_bss_info_change_notify(sdata,
+                                                        BSS_CHANGED_QOS);
+--- a/drivers/net/wireless/ath/ath9k/beacon.c
++++ b/drivers/net/wireless/ath/ath9k/beacon.c
+@@ -48,7 +48,10 @@ int ath_beaconq_config(struct ath_softc 
+               txq = sc->tx.txq_map[WME_AC_BE];
+               ath9k_hw_get_txq_props(ah, txq->axq_qnum, &qi_be);
+               qi.tqi_aifs = qi_be.tqi_aifs;
+-              qi.tqi_cwmin = 4*qi_be.tqi_cwmin;
++              if (ah->slottime == ATH9K_SLOT_TIME_20)
++                      qi.tqi_cwmin = 2*qi_be.tqi_cwmin;
++              else
++                      qi.tqi_cwmin = 4*qi_be.tqi_cwmin;
+               qi.tqi_cwmax = qi_be.tqi_cwmax;
        }
  
-       ieee80211_wake_queues(hw);
--      spin_unlock_bh(&sc->sc_pcu_lock);
-       /* Start ANI */
-       if (!common->disable_ani)
-@@ -2326,9 +2331,9 @@ static void ath9k_flush(struct ieee80211
-       ath9k_ps_wakeup(sc);
-       spin_lock_bh(&sc->sc_pcu_lock);
-       drain_txq = ath_drain_all_txq(sc, false);
--      spin_unlock_bh(&sc->sc_pcu_lock);
-       if (!drain_txq)
-               ath_reset(sc, false);
-+      spin_unlock_bh(&sc->sc_pcu_lock);
-       ath9k_ps_restore(sc);
-       ieee80211_wake_queues(hw);
---- a/drivers/net/wireless/ath/ath9k/xmit.c
-+++ b/drivers/net/wireless/ath/ath9k/xmit.c
-@@ -565,11 +565,8 @@ static void ath_tx_complete_aggr(struct 
-       rcu_read_unlock();
--      if (needreset) {
--              spin_unlock_bh(&sc->sc_pcu_lock);
-+      if (needreset)
-               ath_reset(sc, false);
--              spin_lock_bh(&sc->sc_pcu_lock);
--      }
+--- a/net/mac80211/mlme.c
++++ b/net/mac80211/mlme.c
+@@ -1220,6 +1220,22 @@ static void ieee80211_sta_wmm_params(str
+       sdata->vif.bss_conf.qos = true;
  }
  
- static u32 ath_lookup_rate(struct ath_softc *sc, struct ath_buf *bf,
-@@ -664,7 +661,8 @@ static int ath_compute_num_delims(struct
-        * TODO - this could be improved to be dependent on the rate.
-        *      The hardware can keep up at lower rates, but not higher rates
-        */
--      if (fi->keyix != ATH9K_TXKEYIX_INVALID)
-+      if ((fi->keyix != ATH9K_TXKEYIX_INVALID) &&
-+          !(sc->sc_ah->caps.hw_caps & ATH9K_HW_CAP_EDMA))
-               ndelim += ATH_AGGR_ENCRYPTDELIM;
++static void __ieee80211_stop_poll(struct ieee80211_sub_if_data *sdata)
++{
++      lockdep_assert_held(&sdata->local->mtx);
++
++      sdata->u.mgd.flags &= ~(IEEE80211_STA_CONNECTION_POLL |
++                              IEEE80211_STA_BEACON_POLL);
++      ieee80211_run_deferred_scan(sdata->local);
++}
++
++static void ieee80211_stop_poll(struct ieee80211_sub_if_data *sdata)
++{
++      mutex_lock(&sdata->local->mtx);
++      __ieee80211_stop_poll(sdata);
++      mutex_unlock(&sdata->local->mtx);
++}
++
+ static u32 ieee80211_handle_bss_capability(struct ieee80211_sub_if_data *sdata,
+                                          u16 capab, bool erp_valid, u8 erp)
+ {
+@@ -1285,8 +1301,7 @@ static void ieee80211_set_associated(str
+       sdata->u.mgd.flags |= IEEE80211_STA_RESET_SIGNAL_AVE;
+       /* just to be sure */
+-      sdata->u.mgd.flags &= ~(IEEE80211_STA_CONNECTION_POLL |
+-                              IEEE80211_STA_BEACON_POLL);
++      ieee80211_stop_poll(sdata);
  
-       /*
-@@ -2169,7 +2167,9 @@ static void ath_tx_complete_poll_work(st
-       if (needreset) {
-               ath_dbg(ath9k_hw_common(sc->sc_ah), ATH_DBG_RESET,
-                       "tx hung, resetting the chip\n");
-+              spin_lock_bh(&sc->sc_pcu_lock);
-               ath_reset(sc, true);
-+              spin_unlock_bh(&sc->sc_pcu_lock);
-       }
+       ieee80211_led_assoc(local, 1);
  
-       ieee80211_queue_delayed_work(sc->hw, &sc->tx_complete_work,
---- a/drivers/net/wireless/ath/ath9k/ar9003_paprd.c
-+++ b/drivers/net/wireless/ath/ath9k/ar9003_paprd.c
-@@ -236,7 +236,7 @@ static void ar9003_paprd_get_gain_table(
-       memset(entry, 0, sizeof(ah->paprd_gain_table_entries));
-       memset(index, 0, sizeof(ah->paprd_gain_table_index));
--      for (i = 0; i < 32; i++) {
-+      for (i = 0; i < PAPRD_GAIN_TABLE_ENTRIES; i++) {
-               entry[i] = REG_READ(ah, reg);
-               index[i] = (entry[i] >> 24) & 0xff;
-               reg += 4;
-@@ -246,13 +246,13 @@ static void ar9003_paprd_get_gain_table(
- static unsigned int ar9003_get_desired_gain(struct ath_hw *ah, int chain,
-                                           int target_power)
- {
--      int olpc_gain_delta = 0;
-+      int olpc_gain_delta = 0, cl_gain_mod;
-       int alpha_therm, alpha_volt;
-       int therm_cal_value, volt_cal_value;
-       int therm_value, volt_value;
-       int thermal_gain_corr, voltage_gain_corr;
-       int desired_scale, desired_gain = 0;
--      u32 reg;
-+      u32 reg_olpc  = 0, reg_cl_gain  = 0;
-       REG_CLR_BIT(ah, AR_PHY_PAPRD_TRAINER_STAT1,
-                   AR_PHY_PAPRD_TRAINER_STAT1_PAPRD_TRAIN_DONE);
-@@ -271,15 +271,29 @@ static unsigned int ar9003_get_desired_g
-       volt_value = REG_READ_FIELD(ah, AR_PHY_BB_THERM_ADC_4,
-                                   AR_PHY_BB_THERM_ADC_4_LATEST_VOLT_VALUE);
--      if (chain == 0)
--              reg = AR_PHY_TPC_11_B0;
--      else if (chain == 1)
--              reg = AR_PHY_TPC_11_B1;
--      else
--              reg = AR_PHY_TPC_11_B2;
-+      switch (chain) {
-+      case 0:
-+              reg_olpc = AR_PHY_TPC_11_B0;
-+              reg_cl_gain = AR_PHY_CL_TAB_0;
-+              break;
-+      case 1:
-+              reg_olpc = AR_PHY_TPC_11_B1;
-+              reg_cl_gain = AR_PHY_CL_TAB_1;
-+              break;
-+      case 2:
-+              reg_olpc = AR_PHY_TPC_11_B2;
-+              reg_cl_gain = AR_PHY_CL_TAB_2;
-+              break;
-+      default:
-+              ath_dbg(ath9k_hw_common(ah), ATH_DBG_CALIBRATE,
-+              "Invalid chainmask: %d\n", chain);
-+              break;
-+      }
+@@ -1456,8 +1471,7 @@ static void ieee80211_reset_ap_probe(str
+               return;
+       }
  
--      olpc_gain_delta = REG_READ_FIELD(ah, reg,
-+      olpc_gain_delta = REG_READ_FIELD(ah, reg_olpc,
-                                        AR_PHY_TPC_11_OLPC_GAIN_DELTA);
-+      cl_gain_mod = REG_READ_FIELD(ah, reg_cl_gain,
-+                                       AR_PHY_CL_TAB_CL_GAIN_MOD);
-       if (olpc_gain_delta >= 128)
-               olpc_gain_delta = olpc_gain_delta - 256;
-@@ -289,7 +303,7 @@ static unsigned int ar9003_get_desired_g
-       voltage_gain_corr = (alpha_volt * (volt_value - volt_cal_value) +
-                            (128 / 2)) / 128;
-       desired_gain = target_power - olpc_gain_delta - thermal_gain_corr -
--          voltage_gain_corr + desired_scale;
-+          voltage_gain_corr + desired_scale + cl_gain_mod;
-       return desired_gain;
+-      ifmgd->flags &= ~(IEEE80211_STA_CONNECTION_POLL |
+-                        IEEE80211_STA_BEACON_POLL);
++      __ieee80211_stop_poll(sdata);
+       mutex_lock(&local->iflist_mtx);
+       ieee80211_recalc_ps(local, -1);
+@@ -1477,7 +1491,6 @@ static void ieee80211_reset_ap_probe(str
+                 round_jiffies_up(jiffies +
+                                  IEEE80211_CONNECTION_IDLE_TIME));
+ out:
+-      ieee80211_run_deferred_scan(local);
+       mutex_unlock(&local->mtx);
  }
-@@ -727,7 +741,7 @@ int ar9003_paprd_setup_gain_table(struct
-       desired_gain = ar9003_get_desired_gain(ah, chain, train_power);
  
-       gain_index = 0;
--      for (i = 0; i < 32; i++) {
-+      for (i = 0; i < PAPRD_GAIN_TABLE_ENTRIES; i++) {
-               if (ah->paprd_gain_table_index[i] >= desired_gain)
-                       break;
-               gain_index++;
---- a/drivers/net/wireless/ath/ath9k/ar9003_phy.h
-+++ b/drivers/net/wireless/ath/ath9k/ar9003_phy.h
-@@ -1121,6 +1121,9 @@
- #define AR_PHY_POWERTX_RATE8_POWERTXHT40_5    0x3F00
- #define AR_PHY_POWERTX_RATE8_POWERTXHT40_5_S  8
-+#define AR_PHY_CL_TAB_CL_GAIN_MOD             0x1f
-+#define AR_PHY_CL_TAB_CL_GAIN_MOD_S           0
+@@ -2413,7 +2426,11 @@ static void ieee80211_rx_mgmt_beacon(str
+                              "to a received beacon\n", sdata->name);
+               }
+ #endif
++              mutex_lock(&local->mtx);
+               ifmgd->flags &= ~IEEE80211_STA_BEACON_POLL;
++              ieee80211_run_deferred_scan(local);
++              mutex_unlock(&local->mtx);
++
+               mutex_lock(&local->iflist_mtx);
+               ieee80211_recalc_ps(local, -1);
+               mutex_unlock(&local->iflist_mtx);
+@@ -2600,8 +2617,7 @@ static void ieee80211_sta_connection_los
+       struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
+       u8 frame_buf[DEAUTH_DISASSOC_LEN];
+-      ifmgd->flags &= ~(IEEE80211_STA_CONNECTION_POLL |
+-                        IEEE80211_STA_BEACON_POLL);
++      ieee80211_stop_poll(sdata);
+       ieee80211_set_disassoc(sdata, IEEE80211_STYPE_DEAUTH, reason,
+                              false, frame_buf);
+@@ -2879,8 +2895,7 @@ static void ieee80211_restart_sta_timer(
+       u32 flags;
+       if (sdata->vif.type == NL80211_IFTYPE_STATION) {
+-              sdata->u.mgd.flags &= ~(IEEE80211_STA_BEACON_POLL |
+-                                      IEEE80211_STA_CONNECTION_POLL);
++              __ieee80211_stop_poll(sdata);
+               /* let's probe the connection once */
+               flags = sdata->local->hw.flags;
+@@ -2949,7 +2964,10 @@ void ieee80211_sta_restart(struct ieee80
+       if (test_and_clear_bit(TMR_RUNNING_CHANSW, &ifmgd->timers_running))
+               add_timer(&ifmgd->chswitch_timer);
+       ieee80211_sta_reset_beacon_monitor(sdata);
 +
- void ar9003_hw_set_chain_masks(struct ath_hw *ah, u8 rx, u8 tx);
++      mutex_lock(&sdata->local->mtx);
+       ieee80211_restart_sta_timer(sdata);
++      mutex_unlock(&sdata->local->mtx);
+ }
+ #endif
  
- #endif  /* AR9003_PHY_H */