From: Jouni Malinen <quic_jouni@quicinc.com>
Date: Thu, 24 Aug 2023 13:57:40 +0000 (+0300)
Subject: wlantest: Search SPA using MLO aware find for FT Request/Response frame
X-Git-Tag: hostap_2_11~1027
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=37c87efecfe39e28780950f742eb4b77f803e34d;p=thirdparty%2Fhostap.git

wlantest: Search SPA using MLO aware find for FT Request/Response frame

This is needed to be able to find a previously added STA entry when
roaming using FT over-the-DS back to an AP MLD that was used previously.

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

diff --git a/wlantest/rx_mgmt.c b/wlantest/rx_mgmt.c
index f0b661076..d37440a96 100644
--- a/wlantest/rx_mgmt.c
+++ b/wlantest/rx_mgmt.c
@@ -2108,7 +2108,9 @@ static void rx_mgmt_action_ft_request(struct wlantest *wt,
 		return;
 	}
 
-	sta = sta_get(bss, spa);
+	sta = sta_find_mlo(wt, bss, spa);
+	if (!sta)
+		sta = sta_get(bss, spa);
 	if (!sta)
 		return;
 
@@ -2177,7 +2179,9 @@ static void rx_mgmt_action_ft_response(struct wlantest *wt,
 		return;
 	sta->pmk_r1_len = sta->pmk_r0_len;
 
-	new_sta = sta_get(bss, spa);
+	new_sta = sta_find_mlo(wt, bss, spa);
+	if (!new_sta)
+		new_sta = sta_get(bss, spa);
 	if (!new_sta)
 		return;
 	os_memcpy(new_sta->pmk_r0, sta->pmk_r0, sta->pmk_r0_len);