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 1093ca09d7943369e8240bf039e6c920f2f8eb08..3a0755ccf6d5877a4d934fb9e87160484a14fff8 100644 (file)
--- 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;