From 94785d98d3706331a70430e20273fd993bf3d0d9 Mon Sep 17 00:00:00 2001
From: Stefano Avallone <stavallo@unina.it>
Date: Wed, 12 Oct 2022 22:42:29 +0200
Subject: [PATCH] wifi: Enable checking addresses of ADDBA_REQUEST frames in
 wifi-mlo test

---
 src/wifi/test/wifi-mlo-test.cc | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/wifi/test/wifi-mlo-test.cc b/src/wifi/test/wifi-mlo-test.cc
index b4ced72fa..14201c481 100644
--- a/src/wifi/test/wifi-mlo-test.cc
+++ b/src/wifi/test/wifi-mlo-test.cc
@@ -1044,7 +1044,8 @@ MultiLinkTxNoBaTest::Transmit(uint8_t linkId,
     switch (psdu->GetHeader(0).GetType())
     {
     case WIFI_MAC_MGT_ACTION:
-        // CheckAddresses (psdu, true); TODO uncomment when ADDBA_REQUEST contains proper addresses
+        // a management frame is a DL frame if TA equals BSSID
+        CheckAddresses(psdu, psdu->GetHeader(0).GetAddr2() == psdu->GetHeader(0).GetAddr3());
         // corrupt all management action frames (ADDBA Request frames) to prevent
         // the establishment of a BA agreement
         m_uidList.push_front(uid);