From ecffca222c299361df4cec1b23ae9df87133d0d4 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?S=C3=A9bastien=20Deronne?= <sebastien.deronne@gmail.com>
Date: Sun, 11 Sep 2022 12:23:46 +0200
Subject: [PATCH] wifi: 802.11be STAs always support 40 MHz in 5 GHz and 6 GHz
 bands

---
 src/wifi/model/wifi-mac.cc | 8 +-------
 1 file changed, 1 insertion(+), 7 deletions(-)

diff --git a/src/wifi/model/wifi-mac.cc b/src/wifi/model/wifi-mac.cc
index fdc6de3e8..9bd95b3a2 100644
--- a/src/wifi/model/wifi-mac.cc
+++ b/src/wifi/model/wifi-mac.cc
@@ -1531,7 +1531,7 @@ WifiMac::GetHeCapabilities(uint8_t linkId) const
     {
         channelWidthSet |= 0x01;
     }
-    if ((phy->GetChannelWidth() >= 80) &&
+    if (((phy->GetChannelWidth() >= 80) || GetEhtSupported()) &&
         ((phy->GetPhyBand() == WIFI_PHY_BAND_5GHZ) || (phy->GetPhyBand() == WIFI_PHY_BAND_6GHZ)))
     {
         channelWidthSet |= 0x02;
@@ -1634,12 +1634,6 @@ WifiMac::GetEhtCapabilities(uint8_t linkId) const
     }
     else
     {
-        if (phy->GetPhyBand() != WIFI_PHY_BAND_2_4GHZ)
-        {
-            NS_ABORT_MSG_IF(phy->GetChannelWidth() == 40,
-                            "A 802.11be STA cannot support 40 MHz without supporting 80 MHz except "
-                            "in 2.4 GHz band");
-        }
         for (auto maxMcs : {9, 11, 13})
         {
             capabilities.SetSupportedRxEhtMcsAndNss(