mac80211: add new IEEE80211_VIF_GET_NOA_UPDATE flag

Add new VIF flag, that will allow get NOA update
notification when driver will request this, even
this is not pure P2P vif (eg. STA vif).

Signed-off-by: Janusz Dziedzic <janusz.dziedzic@tieto.com>
Signed-off-by: Johannes Berg <johannes.berg@intel.com>

1 files changed, 4 insertions, 2 deletions

diff --git a/net/mac80211/mlme.c b/net/mac80211/mlme.c
index b140cc6651f4..123b26d177e8 100644
--- a/net/mac80211/mlme.c
+++ b/net/mac80211/mlme.c
@@ -1930,7 +1930,8 @@ static void ieee80211_set_associated(struct ieee80211_sub_if_data *sdata,
 
         sdata->u.mgd.flags |= IEEE80211_STA_RESET_SIGNAL_AVE;
 
-        if (sdata->vif.p2p) {
+        if (sdata->vif.p2p ||
+            sdata->vif.driver_flags & IEEE80211_VIF_GET_NOA_UPDATE) {
                 const struct cfg80211_bss_ies *ies;
 
                 rcu_read_lock();
@@ -3458,7 +3459,8 @@ static void ieee80211_rx_mgmt_beacon(struct ieee80211_sub_if_data *sdata,
                 }
         }
 
-        if (sdata->vif.p2p) {
+        if (sdata->vif.p2p ||
+            sdata->vif.driver_flags & IEEE80211_VIF_GET_NOA_UPDATE) {
                 struct ieee80211_p2p_noa_attr noa = {};
                 int ret;