[PATCH v2 37/44] WPA_AUTH: MLO: Get the correct AA and SPA based on MLD operation

[Andrei Otcheretianski]

From: Ilan Peer <ilan.peer at intel.com>

Signed-off-by: Ilan Peer <ilan.peer at intel.com>
---
 src/ap/wpa_auth.c | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/src/ap/wpa_auth.c b/src/ap/wpa_auth.c
index 5f17b5dcab..2ad5cf8a72 100644
--- a/src/ap/wpa_auth.c
+++ b/src/ap/wpa_auth.c
@@ -35,6 +35,21 @@
 #include "wpa_auth_ie.h"
 #include "sta_info.h"
 
+
+static const u8 *wpa_auth_get_aa(struct wpa_state_machine *sm)
+{
+	return (sm && sm->mld_assoc_link_id >= 0) ?
+		sm->own_mld_addr : sm->wpa_auth->addr;
+}
+
+
+static const u8 *wpa_auth_get_spa(struct wpa_state_machine *sm)
+{
+	return sm->mld_assoc_link_id >= 0 ?
+		sm->peer_mld_addr : sm->addr;
+}
+
+
 #define STATE_MACHINE_DATA struct wpa_state_machine
 #define STATE_MACHINE_DEBUG_PREFIX "WPA"
 #define STATE_MACHINE_ADDR wpa_auth_get_spa(sm)
-- 
2.38.1