From 37c87efecfe39e28780950f742eb4b77f803e34d Mon Sep 17 00:00:00 2001 From: Jouni Malinen Date: Thu, 24 Aug 2023 16:57:40 +0300 Subject: [PATCH] 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 --- wlantest/rx_mgmt.c | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) 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);