diff mbox series

wlantest: ICMP: use sta_find_mlo()

Message ID 20240508221542.a6cadcfd35d4.I3950b93f79ebea781ef6737c0c0d3d64a31a3792@changeid
State New
Headers show
Series wlantest: ICMP: use sta_find_mlo() | expand

Commit Message

Johannes Berg May 8, 2024, 8:15 p.m. UTC
From: Johannes Berg <johannes.berg@intel.com>

Using just sta_find() won't work for any link addresses
but the assoc link, use sta_find_mlo() instead.

Signed-off-by: Johannes Berg <johannes.berg@intel.com>
---
 wlantest/rx_ip.c | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)
diff mbox series

Patch

diff --git a/wlantest/rx_ip.c b/wlantest/rx_ip.c
index b0fdd2068c96..5dc186c45577 100644
--- a/wlantest/rx_ip.c
+++ b/wlantest/rx_ip.c
@@ -85,7 +85,7 @@  static void rx_data_icmp(struct wlantest *wt, const u8 *bssid,
 	if (sta_addr == NULL)
 		return; /* FromDS broadcast ping */
 
-	sta = sta_find(bss, sta_addr);
+	sta = sta_find_mlo(wt, bss, sta_addr);
 	if (sta == NULL) {
 		add_note(wt, MSG_INFO, "No STA " MACSTR
 			 " known for ICMP packet", MAC2STR(sta_addr));
@@ -93,7 +93,7 @@  static void rx_data_icmp(struct wlantest *wt, const u8 *bssid,
 	}
 
 	ping_update(wt, sta, hdr->type == ICMP_ECHO, src, dst, id, seq);
-	if (peer_addr && (sta = sta_find(bss, peer_addr)))
+	if (peer_addr && (sta = sta_find_mlo(wt, bss, peer_addr)))
 		ping_update(wt, sta, hdr->type == ICMP_ECHO, src, dst, id, seq);
 }