From: Jouni Malinen <quic_jouni@quicinc.com>
Date: Thu, 10 Aug 2023 08:31:26 +0000 (+0300)
Subject: wlantest: Use MLO search for the STA in reassociation
X-Git-Tag: hostap_2_11~1066
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=4e8e515f92b9da0989b07552e933589422168b35;p=thirdparty%2Fhostap.git

wlantest: Use MLO search for the STA in reassociation

FT over-the-DS might have created the new STA entry on another
affiliated BSS during the FT Request/Response exchange, so use a wider
search to locate the correct STA entry when processing the Reassociation
Request/Response frames.

Signed-off-by: Jouni Malinen <quic_jouni@quicinc.com>
---

diff --git a/wlantest/rx_mgmt.c b/wlantest/rx_mgmt.c
index 1093ca09d..3a0755ccf 100644
--- a/wlantest/rx_mgmt.c
+++ b/wlantest/rx_mgmt.c
@@ -1156,7 +1156,9 @@ static void rx_mgmt_reassoc_req(struct wlantest *wt, const u8 *data,
 	bss = bss_get(wt, mgmt->bssid);
 	if (bss == NULL)
 		return;
-	sta = sta_get(bss, mgmt->sa);
+	sta = sta_find_mlo(wt, bss, mgmt->sa);
+	if (!sta)
+		sta = sta_get(bss, mgmt->sa);
 	if (sta == NULL)
 		return;
 
@@ -1641,7 +1643,9 @@ static void rx_mgmt_reassoc_resp(struct wlantest *wt, const u8 *data,
 	bss = bss_get(wt, mgmt->bssid);
 	if (bss == NULL)
 		return;
-	sta = sta_get(bss, mgmt->da);
+	sta = sta_find_mlo(wt, bss, mgmt->da);
+	if (!sta)
+		sta = sta_get(bss, mgmt->da);
 	if (sta == NULL)
 		return;