From: Jouni Malinen <quic_jouni@quicinc.com>
Date: Mon, 27 Mar 2023 14:26:41 +0000 (+0300)
Subject: wlantest: MLO aware STA entry search for unprotected Data frames
X-Git-Tag: hostap_2_11~1220
X-Git-Url: http://git.ipfire.org/?a=commitdiff_plain;h=d423baa42aaf4e9d06e93e35eff97bbe1f392ec9;p=thirdparty%2Fhostap.git

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 0f88b9d37..2a03f5c9a 100644
--- 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) {