[PATCH 1/3] mt76: add driver callback for when a sta is associated

[Felix Fietkau]

MT7603 needs this to update the HT/VHT capabilities

Signed-off-by: Felix Fietkau <n...@nbd.name>
---
 drivers/net/wireless/mediatek/mt76/mac80211.c | 5 +++++
 drivers/net/wireless/mediatek/mt76/mt76.h     | 3 +++
 2 files changed, 8 insertions(+)

diff --git a/drivers/net/wireless/mediatek/mt76/mac80211.c 
b/drivers/net/wireless/mediatek/mt76/mac80211.c
index ee3b65a14870..489953a47646 100644
--- a/drivers/net/wireless/mediatek/mt76/mac80211.c
+++ b/drivers/net/wireless/mediatek/mt76/mac80211.c
@@ -709,6 +709,11 @@ int mt76_sta_state(struct ieee80211_hw *hw, struct 
ieee80211_vif *vif,
            new_state == IEEE80211_STA_NONE)
                return mt76_sta_add(dev, vif, sta);
 
+       if (old_state == IEEE80211_STA_AUTH &&
+           new_state == IEEE80211_STA_ASSOC &&
+           dev->drv->sta_assoc)
+               dev->drv->sta_assoc(dev, vif, sta);
+
        if (old_state == IEEE80211_STA_NONE &&
                 new_state == IEEE80211_STA_NOTEXIST)
                mt76_sta_remove(dev, vif, sta);
diff --git a/drivers/net/wireless/mediatek/mt76/mt76.h 
b/drivers/net/wireless/mediatek/mt76/mt76.h
index 2bb9db4ed80a..71708b111120 100644
--- a/drivers/net/wireless/mediatek/mt76/mt76.h
+++ b/drivers/net/wireless/mediatek/mt76/mt76.h
@@ -300,6 +300,9 @@ struct mt76_driver_ops {
        int (*sta_add)(struct mt76_dev *dev, struct ieee80211_vif *vif,
                       struct ieee80211_sta *sta);
 
+       void (*sta_assoc)(struct mt76_dev *dev, struct ieee80211_vif *vif,
+                         struct ieee80211_sta *sta);
+
        void (*sta_remove)(struct mt76_dev *dev, struct ieee80211_vif *vif,
                           struct ieee80211_sta *sta);
 };
-- 
2.17.0