MLO: Get the correct AA and SPA based on MLD operation for RSN authenticator

Signed-off-by: Ilan Peer <ilan.peer@intel.com>

diff --git a/src/ap/wpa_auth.c b/src/ap/wpa_auth.c
index f759ed322c8b807ba5c2c0901a5501a0f7042a3e..9d153da7da1f3abc1c559d30afa58b594fb5bd1d 100644 (file)
--- a/src/ap/wpa_auth.c
+++ b/src/ap/wpa_auth.c
@@ -85,12 +85,20 @@ static const int dot11RSNAConfigSATimeout = 60;
 
 static const u8 * wpa_auth_get_aa(const struct wpa_state_machine *sm)
 {
+#ifdef CONFIG_IEEE80211BE
+       if (sm->mld_assoc_link_id >= 0)
+               return sm->own_mld_addr;
+#endif /* CONFIG_IEEE80211BE */
        return sm->wpa_auth->addr;
 }
 
 
 static const u8 * wpa_auth_get_spa(const struct wpa_state_machine *sm)
 {
+#ifdef CONFIG_IEEE80211BE
+       if (sm->mld_assoc_link_id >= 0)
+               return sm->peer_mld_addr;
+#endif /* CONFIG_IEEE80211BE */
        return sm->addr;
 }