From fa91d46b2cfa4600132237539664bdfec6512392 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?S=C3=A9bastien=20Deronne?= <sebastien.deronne@gmail.com>
Date: Sat, 2 Sep 2023 10:03:41 +0200
Subject: [PATCH] wifi: Update GCR STAs membership upon (re)assoc/deassoc

---
 src/wifi/model/ap-wifi-mac.cc | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/src/wifi/model/ap-wifi-mac.cc b/src/wifi/model/ap-wifi-mac.cc
index bf31fbb4d..f13da1531 100644
--- a/src/wifi/model/ap-wifi-mac.cc
+++ b/src/wifi/model/ap-wifi-mac.cc
@@ -1412,6 +1412,15 @@ ApWifiMac::SetAid(MgtAssocResponseHeader& assoc, const LinkIdStaAddrMap& linkIdS
                             "AID " << it->first << " already assigned to " << staAddr
                                    << ", could not assign " << aid);
         }
+
+        if (auto extendedCapabilities =
+                GetWifiRemoteStationManager(linkId)->GetStationExtendedCapabilities(staAddr);
+            m_gcrManager)
+        {
+            const auto isGcrCapable =
+                extendedCapabilities && extendedCapabilities->m_robustAvStreaming;
+            m_gcrManager->NotifyStaAssociated(staAddr, isGcrCapable);
+        }
     }
 
     // set the AID in all the Association Responses. NOTE that the Association
@@ -2060,6 +2069,10 @@ ApWifiMac::Receive(Ptr<const WifiMpdu> mpdu, uint8_t linkId)
                         UpdateShortSlotTimeEnabled(linkId);
                         UpdateShortPreambleEnabled(linkId);
                         StaSwitchingToActiveModeOrDeassociated(from, linkId);
+                        if (m_gcrManager)
+                        {
+                            m_gcrManager->NotifyStaDeassociated(from);
+                        }
                         break;
                     }
                 }