From 4e8e515f92b9da0989b07552e933589422168b35 Mon Sep 17 00:00:00 2001 From: Jouni Malinen Date: Thu, 10 Aug 2023 11:31:26 +0300 Subject: [PATCH] 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 --- 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 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;