3 files changed, 4 insertions, 6 deletions

diff --git a/net/mac80211/ieee80211_sta.c b/net/mac80211/ieee80211_sta.c
index f98b2e1cedf3..1b4ebe81191c 100644
--- a/net/mac80211/ieee80211_sta.c
+++ b/net/mac80211/ieee80211_sta.c
@@ -1279,8 +1279,7 @@ static void ieee80211_rx_mgmt_assoc_resp(struct net_device *dev,
         }
 
         sta->dev = dev;
-        sta->flags |= WLAN_STA_AUTH | WLAN_STA_ASSOC;
-        sta->assoc_ap = 1;
+        sta->flags |= WLAN_STA_AUTH | WLAN_STA_ASSOC | WLAN_STA_ASSOC_AP;
 
         rates = 0;
         mode = local->oper_hw_mode;
diff --git a/net/mac80211/rx.c b/net/mac80211/rx.c
index 75a1e204b9b6..4fb8c7026f11 100644
--- a/net/mac80211/rx.c
+++ b/net/mac80211/rx.c
@@ -1398,7 +1398,7 @@ void __ieee80211_rx(struct ieee80211_hw *hw, struct sk_buff *skb,
         skb = rx.skb;
 
         skb_push(skb, radiotap_len);
-        if (sta && !sta->assoc_ap && !(sta->flags & WLAN_STA_WDS) &&
+        if (sta && !(sta->flags & (WLAN_STA_WDS | WLAN_STA_ASSOC_AP)) &&
             !local->iff_promiscs && !is_multicast_ether_addr(hdr->addr1)) {
                 rx.flags |= IEEE80211_TXRXD_RXRA_MATCH;
                 ieee80211_invoke_rx_handlers(local, local->rx_handlers, &rx,
diff --git a/net/mac80211/sta_info.h b/net/mac80211/sta_info.h
index 3b0fcb255717..e3ae0f5cc228 100644
--- a/net/mac80211/sta_info.h
+++ b/net/mac80211/sta_info.h
@@ -26,6 +26,8 @@
                                     * send and receive non-IEEE 802.1X frames
                                     */
 #define WLAN_STA_SHORT_PREAMBLE BIT(7)
+/* whether this is an AP that we are associated with as a client */
+#define WLAN_STA_ASSOC_AP BIT(8)
 #define WLAN_STA_WME BIT(9)
 #define WLAN_STA_WDS BIT(27)
 
@@ -90,9 +92,6 @@ struct sta_info {
         int channel_use;
         int channel_use_raw;
 
-        int assoc_ap; /* whether this is an AP that we are
-                       * associated with as a client */
-
 #ifdef CONFIG_MAC80211_DEBUG_COUNTERS
         unsigned int wme_rx_queue[NUM_RX_DATA_QUEUES];
         unsigned int wme_tx_queue[NUM_RX_DATA_QUEUES];