From 94be0bc4889c3348b7cf73431c3bf0f1ab86ac44 Mon Sep 17 00:00:00 2001
From: Sharan Naribole <sharan.naribole@gmail.com>
Date: Mon, 2 Sep 2024 22:32:15 +0200
Subject: [PATCH] wifi: Fix missing DSSS Param Set in Probe Request sent over
 2.4 GHz links

---
 src/wifi/model/mgt-headers.h   | 1 +
 src/wifi/model/sta-wifi-mac.cc | 6 ++++++
 2 files changed, 7 insertions(+)

diff --git a/src/wifi/model/mgt-headers.h b/src/wifi/model/mgt-headers.h
index 7954814d7..670063ce8 100644
--- a/src/wifi/model/mgt-headers.h
+++ b/src/wifi/model/mgt-headers.h
@@ -82,6 +82,7 @@ struct CanBeInPerStaProfile<Ssid> : std::false_type
 using ProbeRequestElems = std::tuple<Ssid,
                                      SupportedRates,
                                      std::optional<ExtendedSupportedRatesIE>,
+                                     std::optional<DsssParameterSet>,
                                      std::optional<HtCapabilities>,
                                      std::optional<ExtendedCapabilities>,
                                      std::optional<VhtCapabilities>,
diff --git a/src/wifi/model/sta-wifi-mac.cc b/src/wifi/model/sta-wifi-mac.cc
index bda2c263f..c83ec3c3c 100644
--- a/src/wifi/model/sta-wifi-mac.cc
+++ b/src/wifi/model/sta-wifi-mac.cc
@@ -356,6 +356,12 @@ StaWifiMac::GetProbeRequest(uint8_t linkId) const
     auto supportedRates = GetSupportedRates(linkId);
     probe.Get<SupportedRates>() = supportedRates.rates;
     probe.Get<ExtendedSupportedRatesIE>() = supportedRates.extendedRates;
+    if (GetWifiPhy(linkId)->GetPhyBand() == WIFI_PHY_BAND_2_4GHZ)
+    {
+        DsssParameterSet params;
+        params.SetCurrentChannel(GetWifiPhy(linkId)->GetChannelNumber());
+        probe.Get<DsssParameterSet>() = params;
+    }
     if (GetHtSupported(linkId))
     {
         probe.Get<ExtendedCapabilities>() = GetExtendedCapabilities();