wlantest: MLO aware STA entry search for unprotected Data frames

Find a STA entry based on MLO affiliated link addresses for the case
where an unprotected Data frame is being processed. This extends the
changes in commit 228420e2d9a8 ("wlantest: Find a STA entry based on MLO
affiliated link addresses") to cover the unencrypted case. This is
needed in particular for the Null frames used for managing the power
save state to avoid generating duplicate STA entries that can mess up
key information for the following frames.

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

diff --git a/wlantest/rx_data.c b/wlantest/rx_data.c
index 0f88b9d37e07a54c5fed5a374fd1296cedaf6210..2a03f5c9a37ca334b62dda141681e336274e1360 100644 (file)
--- a/wlantest/rx_data.c
+++ b/wlantest/rx_data.c
@@ -812,7 +812,11 @@ static void rx_data_bss(struct wlantest *wt, const struct ieee80211_hdr *hdr,
 
                bss = bss_get(wt, bssid);
                if (bss) {
-                       struct wlantest_sta *sta = sta_get(bss, sta_addr);
+                       struct wlantest_sta *sta;
+
+                       sta = sta_find_mlo(wt, bss, sta_addr);
+                       if (!sta)
+                               sta = sta_get(bss, sta_addr);
 
                        if (sta) {
                                if (qos) {