From 2b0d4ef3dec1900629745d5ec0cee817abf33106 Mon Sep 17 00:00:00 2001
From: Stefano Avallone <stavallo@unina.it>
Date: Sun, 7 Feb 2021 12:14:45 +0100
Subject: [PATCH] wifi: Add HeConfiguration methods to set/get the BSS color

---
 src/wifi/model/he/he-configuration.cc | 16 +++++++++++++++-
 src/wifi/model/he/he-configuration.h  |  8 ++++++++
 2 files changed, 23 insertions(+), 1 deletion(-)

diff --git a/src/wifi/model/he/he-configuration.cc b/src/wifi/model/he/he-configuration.cc
index 726b36b63..5da10a4c6 100644
--- a/src/wifi/model/he/he-configuration.cc
+++ b/src/wifi/model/he/he-configuration.cc
@@ -49,7 +49,8 @@ HeConfiguration::GetTypeId (void)
     .AddAttribute ("BssColor",
                    "The BSS color",
                    UintegerValue (0),
-                   MakeUintegerAccessor (&HeConfiguration::m_bssColor),
+                   MakeUintegerAccessor (&HeConfiguration::GetBssColor,
+                                         &HeConfiguration::SetBssColor),
                    MakeUintegerChecker<uint8_t> ())
     .AddAttribute ("MpduBufferSize",
                    "The MPDU buffer size for receiving A-MPDUs",
@@ -75,6 +76,19 @@ HeConfiguration::GetGuardInterval (void) const
   return m_guardInterval;
 }
 
+void
+HeConfiguration::SetBssColor (uint8_t bssColor)
+{
+  NS_LOG_FUNCTION (this << +bssColor);
+  m_bssColor = bssColor;
+}
+
+uint8_t
+HeConfiguration::GetBssColor (void) const
+{
+  return m_bssColor;
+}
+
 void
 HeConfiguration::SetMpduBufferSize (uint16_t size)
 {
diff --git a/src/wifi/model/he/he-configuration.h b/src/wifi/model/he/he-configuration.h
index 69f84821e..3f1374b7f 100644
--- a/src/wifi/model/he/he-configuration.h
+++ b/src/wifi/model/he/he-configuration.h
@@ -52,6 +52,14 @@ public:
    * \return the supported HE guard interval
    */
   Time GetGuardInterval (void) const;
+  /**
+   * \param bssColor the BSS color
+   */
+  void SetBssColor (uint8_t bssColor);
+  /**
+   * \return the BSS color
+   */
+  uint8_t GetBssColor (void) const;
   /**
    * \param size the MPDU buffer size to receive A-MPDUs
    */