mgt-headers.h

Go to the documentation of this file.

/*
 * Copyright (c) 2006 INRIA
 * Copyright (c) 2009 MIRKO BANCHI
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License version 2 as
 * published by the Free Software Foundation;
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 *
 * Authors: Mathieu Lacage <mathieu.lacage@sophia.inria.fr>
 *          Mirko Banchi <mk.banchi@gmail.com>
 */
 
#ifndef MGT_HEADERS_H
#define MGT_HEADERS_H
 
#include "capability-information.h"
#include "edca-parameter-set.h"
#include "extended-capabilities.h"
#include "reduced-neighbor-report.h"
#include "ssid.h"
#include "status-code.h"
#include "supported-rates.h"
 
#include "ns3/dsss-parameter-set.h"
#include "ns3/eht-capabilities.h"
#include "ns3/erp-information.h"
#include "ns3/he-capabilities.h"
#include "ns3/he-operation.h"
#include "ns3/ht-capabilities.h"
#include "ns3/ht-operation.h"
#include "ns3/mac48-address.h"
#include "ns3/mu-edca-parameter-set.h"
#include "ns3/multi-link-element.h"
#include "ns3/vht-capabilities.h"
#include "ns3/vht-operation.h"
 
namespace ns3
{
 
class MgtAssocRequestHeader : public Header
{
  public:
    MgtAssocRequestHeader();
    ~MgtAssocRequestHeader() override;
 
    void SetSsid(const Ssid& ssid);
 
    void SetSsid(Ssid&& ssid);
 
    void SetSupportedRates(const SupportedRates& rates);
 
    void SetSupportedRates(SupportedRates&& rates);
 
    void SetListenInterval(uint16_t interval);
 
    void SetCapabilities(const CapabilityInformation& capabilities);
 
    void SetCapabilities(CapabilityInformation&& capabilities);
 
    void SetExtendedCapabilities(const ExtendedCapabilities& extendedCapabilities);
 
    void SetExtendedCapabilities(ExtendedCapabilities&& extendedCapabilities);
 
    void SetHtCapabilities(const HtCapabilities& htCapabilities);
 
    void SetHtCapabilities(HtCapabilities&& htCapabilities);
 
    void SetVhtCapabilities(const VhtCapabilities& vhtCapabilities);
 
    void SetVhtCapabilities(VhtCapabilities&& vhtCapabilities);
 
    void SetHeCapabilities(const HeCapabilities& heCapabilities);
 
    void SetHeCapabilities(HeCapabilities&& heCapabilities);
 
    void SetEhtCapabilities(const EhtCapabilities& ehtCapabilities);
 
    void SetEhtCapabilities(EhtCapabilities&& ehtCapabilities);
 
    void SetMultiLinkElement(const MultiLinkElement& multiLinkElement);
 
    void SetMultiLinkElement(MultiLinkElement&& multiLinkElement);
 
    const CapabilityInformation& GetCapabilities() const;
    const std::optional<ExtendedCapabilities>& GetExtendedCapabilities() const;
    const std::optional<HtCapabilities>& GetHtCapabilities() const;
    const std::optional<VhtCapabilities>& GetVhtCapabilities() const;
    const std::optional<HeCapabilities>& GetHeCapabilities() const;
    const std::optional<EhtCapabilities>& GetEhtCapabilities() const;
    const Ssid& GetSsid() const;
    const SupportedRates& GetSupportedRates() const;
    uint16_t GetListenInterval() const;
    const std::optional<MultiLinkElement>& GetMultiLinkElement() const;
 
    static TypeId GetTypeId();
 
    TypeId GetInstanceTypeId() const override;
    void Print(std::ostream& os) const override;
    uint32_t GetSerializedSize() const override;
    void Serialize(Buffer::Iterator start) const override;
    uint32_t Deserialize(Buffer::Iterator start) override;
 
  private:
    Ssid m_ssid;                                              
    SupportedRates m_rates;                                   
    CapabilityInformation m_capability;                       
    std::optional<ExtendedCapabilities> m_extendedCapability; 
    std::optional<HtCapabilities> m_htCapability;             
    std::optional<VhtCapabilities> m_vhtCapability;           
    std::optional<HeCapabilities> m_heCapability;             
    uint16_t m_listenInterval;                                
    std::optional<EhtCapabilities> m_ehtCapability;           
    std::optional<MultiLinkElement> m_multiLinkElement;       
};
 
class MgtReassocRequestHeader : public Header
{
  public:
    MgtReassocRequestHeader();
    ~MgtReassocRequestHeader() override;
 
    void SetSsid(const Ssid& ssid);
 
    void SetSsid(Ssid&& ssid);
 
    void SetSupportedRates(const SupportedRates& rates);
 
    void SetSupportedRates(SupportedRates&& rates);
 
    void SetListenInterval(uint16_t interval);
 
    void SetCapabilities(const CapabilityInformation& capabilities);
 
    void SetCapabilities(CapabilityInformation&& capabilities);
 
    void SetExtendedCapabilities(const ExtendedCapabilities& extendedCapabilities);
 
    void SetExtendedCapabilities(ExtendedCapabilities&& extendedCapabilities);
 
    void SetHtCapabilities(const HtCapabilities& htCapabilities);
 
    void SetHtCapabilities(HtCapabilities&& htCapabilities);
 
    void SetVhtCapabilities(const VhtCapabilities& vhtCapabilities);
 
    void SetVhtCapabilities(VhtCapabilities&& vhtCapabilities);
 
    void SetHeCapabilities(const HeCapabilities& heCapabilities);
 
    void SetHeCapabilities(HeCapabilities&& heCapabilities);
 
    void SetEhtCapabilities(const EhtCapabilities& ehtCapabilities);
 
    void SetEhtCapabilities(EhtCapabilities&& ehtCapabilities);
 
    void SetMultiLinkElement(const MultiLinkElement& multiLinkElement);
 
    void SetMultiLinkElement(MultiLinkElement&& multiLinkElement);
 
    const CapabilityInformation& GetCapabilities() const;
    const std::optional<ExtendedCapabilities>& GetExtendedCapabilities() const;
    const std::optional<HtCapabilities>& GetHtCapabilities() const;
    const std::optional<VhtCapabilities>& GetVhtCapabilities() const;
    const std::optional<HeCapabilities>& GetHeCapabilities() const;
    const std::optional<EhtCapabilities>& GetEhtCapabilities() const;
    const Ssid& GetSsid() const;
    const SupportedRates& GetSupportedRates() const;
    const std::optional<MultiLinkElement>& GetMultiLinkElement() const;
    uint16_t GetListenInterval() const;
    void SetCurrentApAddress(Mac48Address currentApAddr);
 
    static TypeId GetTypeId();
    TypeId GetInstanceTypeId() const override;
    void Print(std::ostream& os) const override;
    uint32_t GetSerializedSize() const override;
    void Serialize(Buffer::Iterator start) const override;
    uint32_t Deserialize(Buffer::Iterator start) override;
 
  private:
    Mac48Address m_currentApAddr;       
    Ssid m_ssid;                        
    SupportedRates m_rates;             
    CapabilityInformation m_capability; 
    std::optional<ExtendedCapabilities> m_extendedCapability; 
    std::optional<HtCapabilities> m_htCapability;             
    std::optional<VhtCapabilities> m_vhtCapability;           
    std::optional<HeCapabilities> m_heCapability;             
    uint16_t m_listenInterval;                                
    std::optional<EhtCapabilities> m_ehtCapability;           
    std::optional<MultiLinkElement> m_multiLinkElement;       
};
 
class MgtAssocResponseHeader : public Header
{
  public:
    MgtAssocResponseHeader();
    ~MgtAssocResponseHeader() override;
 
    StatusCode GetStatusCode();
    const SupportedRates& GetSupportedRates() const;
    const CapabilityInformation& GetCapabilities() const;
    const std::optional<ExtendedCapabilities>& GetExtendedCapabilities() const;
    const std::optional<HtCapabilities>& GetHtCapabilities() const;
    const std::optional<HtOperation>& GetHtOperation() const;
    const std::optional<VhtCapabilities>& GetVhtCapabilities() const;
    const std::optional<VhtOperation>& GetVhtOperation() const;
    const std::optional<HeCapabilities>& GetHeCapabilities() const;
    const std::optional<HeOperation>& GetHeOperation() const;
    const std::optional<EhtCapabilities>& GetEhtCapabilities() const;
    const std::optional<MultiLinkElement>& GetMultiLinkElement() const;
    uint16_t GetAssociationId() const;
    const std::optional<ErpInformation>& GetErpInformation() const;
    const std::optional<EdcaParameterSet>& GetEdcaParameterSet() const;
    const std::optional<MuEdcaParameterSet>& GetMuEdcaParameterSet() const;
    void SetCapabilities(const CapabilityInformation& capabilities);
 
    void SetCapabilities(CapabilityInformation&& capabilities);
 
    void SetExtendedCapabilities(const ExtendedCapabilities& extendedCapabilities);
 
    void SetExtendedCapabilities(ExtendedCapabilities&& extendedCapabilities);
 
    void SetVhtOperation(const VhtOperation& vhtOperation);
 
    void SetVhtOperation(VhtOperation&& vhtOperation);
 
    void SetVhtCapabilities(const VhtCapabilities& vhtCapabilities);
 
    void SetVhtCapabilities(VhtCapabilities&& vhtCapabilities);
 
    void SetHtCapabilities(const HtCapabilities& htCapabilities);
 
    void SetHtCapabilities(HtCapabilities&& htCapabilities);
 
    void SetHtOperation(const HtOperation& htOperation);
 
    void SetHtOperation(HtOperation&& htOperation);
 
    void SetSupportedRates(const SupportedRates& rates);
 
    void SetSupportedRates(SupportedRates&& rates);
 
    void SetStatusCode(StatusCode code);
 
    void SetAssociationId(uint16_t aid);
 
    void SetErpInformation(const ErpInformation& erpInformation);
 
    void SetErpInformation(ErpInformation&& erpInformation);
 
    void SetEdcaParameterSet(const EdcaParameterSet& edcaParameterSet);
 
    void SetEdcaParameterSet(EdcaParameterSet&& edcaParameterSet);
 
    void SetMuEdcaParameterSet(const MuEdcaParameterSet& muEdcaParameterSet);
 
    void SetMuEdcaParameterSet(MuEdcaParameterSet&& muEdcaParameterSet);
 
    void SetHeCapabilities(const HeCapabilities& heCapabilities);
 
    void SetHeCapabilities(HeCapabilities&& heCapabilities);
 
    void SetHeOperation(const HeOperation& heOperation);
 
    void SetHeOperation(HeOperation&& heOperation);
 
    void SetEhtCapabilities(const EhtCapabilities& ehtCapabilities);
 
    void SetEhtCapabilities(EhtCapabilities&& ehtCapabilities);
 
    void SetMultiLinkElement(const MultiLinkElement& multiLinkElement);
 
    void SetMultiLinkElement(MultiLinkElement&& multiLinkElement);
 
    static TypeId GetTypeId();
    TypeId GetInstanceTypeId() const override;
    void Print(std::ostream& os) const override;
    uint32_t GetSerializedSize() const override;
    void Serialize(Buffer::Iterator start) const override;
    uint32_t Deserialize(Buffer::Iterator start) override;
 
  private:
    SupportedRates m_rates;                                   
    CapabilityInformation m_capability;                       
    StatusCode m_code;                                        
    uint16_t m_aid;                                           
    std::optional<ExtendedCapabilities> m_extendedCapability; 
    std::optional<HtCapabilities> m_htCapability;             
    std::optional<HtOperation> m_htOperation;                 
    std::optional<VhtCapabilities> m_vhtCapability;           
    std::optional<VhtOperation> m_vhtOperation;               
    std::optional<ErpInformation> m_erpInformation;           
    std::optional<EdcaParameterSet> m_edcaParameterSet;       
    std::optional<HeCapabilities> m_heCapability;             
    std::optional<HeOperation> m_heOperation;                 
    std::optional<MuEdcaParameterSet> m_muEdcaParameterSet;   
    std::optional<EhtCapabilities> m_ehtCapability;           
    std::optional<MultiLinkElement> m_multiLinkElement;       
};
 
class MgtProbeRequestHeader : public Header
{
  public:
    ~MgtProbeRequestHeader() override;
 
    void SetSsid(const Ssid& ssid);
 
    void SetSsid(Ssid&& ssid);
 
    void SetSupportedRates(const SupportedRates& rates);
 
    void SetSupportedRates(SupportedRates&& rates);
 
    void SetExtendedCapabilities(const ExtendedCapabilities& extendedCapabilities);
 
    void SetExtendedCapabilities(ExtendedCapabilities&& extendedCapabilities);
 
    void SetHtCapabilities(const HtCapabilities& htCapabilities);
 
    void SetHtCapabilities(HtCapabilities&& htCapabilities);
 
    void SetVhtCapabilities(const VhtCapabilities& vhtCapabilities);
 
    void SetVhtCapabilities(VhtCapabilities&& vhtCapabilities);
 
    void SetHeCapabilities(const HeCapabilities& heCapabilities);
 
    void SetHeCapabilities(HeCapabilities&& heCapabilities);
 
    void SetEhtCapabilities(const EhtCapabilities& ehtCapabilities);
 
    void SetEhtCapabilities(EhtCapabilities&& ehtCapabilities);
 
    const Ssid& GetSsid() const;
 
    const SupportedRates& GetSupportedRates() const;
 
    const std::optional<ExtendedCapabilities>& GetExtendedCapabilities() const;
 
    const std::optional<HtCapabilities>& GetHtCapabilities() const;
 
    const std::optional<VhtCapabilities>& GetVhtCapabilities() const;
 
    const std::optional<HeCapabilities>& GetHeCapabilities() const;
 
    const std::optional<EhtCapabilities>& GetEhtCapabilities() const;
 
    static TypeId GetTypeId();
    TypeId GetInstanceTypeId() const override;
    void Print(std::ostream& os) const override;
    uint32_t GetSerializedSize() const override;
    void Serialize(Buffer::Iterator start) const override;
    uint32_t Deserialize(Buffer::Iterator start) override;
 
  private:
    Ssid m_ssid;                                              
    SupportedRates m_rates;                                   
    std::optional<ExtendedCapabilities> m_extendedCapability; 
    std::optional<HtCapabilities> m_htCapability;             
    std::optional<VhtCapabilities> m_vhtCapability;           
    std::optional<HeCapabilities> m_heCapability;             
    std::optional<EhtCapabilities> m_ehtCapability;           
};
 
class MgtProbeResponseHeader : public Header
{
  public:
    MgtProbeResponseHeader();
    ~MgtProbeResponseHeader() override;
 
    const Ssid& GetSsid() const;
    uint64_t GetBeaconIntervalUs() const;
    const SupportedRates& GetSupportedRates() const;
    const CapabilityInformation& GetCapabilities() const;
    const std::optional<DsssParameterSet>& GetDsssParameterSet() const;
    const std::optional<ExtendedCapabilities>& GetExtendedCapabilities() const;
    const std::optional<HtCapabilities>& GetHtCapabilities() const;
    const std::optional<HtOperation>& GetHtOperation() const;
    const std::optional<VhtCapabilities>& GetVhtCapabilities() const;
    const std::optional<VhtOperation>& GetVhtOperation() const;
    const std::optional<HeCapabilities>& GetHeCapabilities() const;
    const std::optional<HeOperation>& GetHeOperation() const;
    const std::optional<EhtCapabilities>& GetEhtCapabilities() const;
    const std::optional<ErpInformation>& GetErpInformation() const;
    const std::optional<EdcaParameterSet>& GetEdcaParameterSet() const;
    const std::optional<MuEdcaParameterSet>& GetMuEdcaParameterSet() const;
 
    const std::optional<ReducedNeighborReport>& GetReducedNeighborReport() const;
    const std::optional<MultiLinkElement>& GetMultiLinkElement() const;
    void SetCapabilities(const CapabilityInformation& capabilities);
 
    void SetCapabilities(CapabilityInformation&& capabilities);
 
    void SetExtendedCapabilities(const ExtendedCapabilities& extendedCapabilities);
 
    void SetExtendedCapabilities(ExtendedCapabilities&& extendedCapabilities);
 
    void SetHtCapabilities(const HtCapabilities& htCapabilities);
 
    void SetHtCapabilities(HtCapabilities&& htCapabilities);
 
    void SetHtOperation(const HtOperation& htOperation);
 
    void SetHtOperation(HtOperation&& htOperation);
 
    void SetVhtCapabilities(const VhtCapabilities& vhtCapabilities);
 
    void SetVhtCapabilities(VhtCapabilities&& vhtCapabilities);
 
    void SetVhtOperation(const VhtOperation& vhtOperation);
 
    void SetVhtOperation(VhtOperation&& vhtOperation);
 
    void SetHeCapabilities(const HeCapabilities& heCapabilities);
 
    void SetHeCapabilities(HeCapabilities&& heCapabilities);
 
    void SetHeOperation(const HeOperation& heOperation);
 
    void SetHeOperation(HeOperation&& heOperation);
 
    void SetEhtCapabilities(const EhtCapabilities& ehtCapabilities);
 
    void SetEhtCapabilities(EhtCapabilities&& ehtCapabilities);
 
    void SetSsid(const Ssid& ssid);
 
    void SetSsid(Ssid&& ssid);
 
    void SetBeaconIntervalUs(uint64_t us);
 
    void SetSupportedRates(const SupportedRates& rates);
 
    void SetSupportedRates(SupportedRates&& rates);
 
    void SetDsssParameterSet(const DsssParameterSet& dsssParameterSet);
 
    void SetDsssParameterSet(DsssParameterSet&& dsssParameterSet);
 
    void SetErpInformation(const ErpInformation& erpInformation);
 
    void SetErpInformation(ErpInformation&& erpInformation);
 
    void SetEdcaParameterSet(const EdcaParameterSet& edcaParameterSet);
 
    void SetEdcaParameterSet(EdcaParameterSet&& edcaParameterSet);
 
    void SetMuEdcaParameterSet(const MuEdcaParameterSet& muEdcaParameterSet);
 
    void SetMuEdcaParameterSet(MuEdcaParameterSet&& muEdcaParameterSet);
 
    void SetReducedNeighborReport(const ReducedNeighborReport& reducedNeighborReport);
 
    void SetReducedNeighborReport(ReducedNeighborReport&& reducedNeighborReport);
 
    void SetMultiLinkElement(const MultiLinkElement& multiLinkElement);
 
    void SetMultiLinkElement(MultiLinkElement&& multiLinkElement);
 
    uint64_t GetTimestamp();
 
    static TypeId GetTypeId();
    TypeId GetInstanceTypeId() const override;
    void Print(std::ostream& os) const override;
    uint32_t GetSerializedSize() const override;
    void Serialize(Buffer::Iterator start) const override;
    uint32_t Deserialize(Buffer::Iterator start) override;
 
  private:
    uint64_t m_timestamp;                                     
    Ssid m_ssid;                                              
    uint64_t m_beaconInterval;                                
    SupportedRates m_rates;                                   
    CapabilityInformation m_capability;                       
    std::optional<DsssParameterSet> m_dsssParameterSet;       
    std::optional<ExtendedCapabilities> m_extendedCapability; 
    std::optional<HtCapabilities> m_htCapability;             
    std::optional<HtOperation> m_htOperation;                 
    std::optional<VhtCapabilities> m_vhtCapability;           
    std::optional<VhtOperation> m_vhtOperation;               
    std::optional<HeCapabilities> m_heCapability;             
    std::optional<HeOperation> m_heOperation;                 
    std::optional<ErpInformation> m_erpInformation;           
    std::optional<EdcaParameterSet> m_edcaParameterSet;       
    std::optional<MuEdcaParameterSet> m_muEdcaParameterSet;   
    std::optional<EhtCapabilities> m_ehtCapability;           
    std::optional<ReducedNeighborReport>
        m_reducedNeighborReport;                        
    std::optional<MultiLinkElement> m_multiLinkElement; 
};
 
class MgtBeaconHeader : public MgtProbeResponseHeader
{
  public:
    static TypeId GetTypeId();
};
 
/****************************
 *     Action frames
 *****************************/
 
class WifiActionHeader : public Header
{
  public:
    WifiActionHeader();
    ~WifiActionHeader() override;
 
    /*
     * Compatible with table 8-38 IEEE 802.11, Part11, (Year 2012)
     * Category values - see 802.11-2012 Table 8-38
     */
 
    enum CategoryValue // table 8-38 staring from IEEE 802.11, Part11, (Year 2012)
    {
        QOS = 1,
        BLOCK_ACK = 3,
        PUBLIC = 4,
        RADIO_MEASUREMENT = 5, // Category: Radio Measurement
        MESH = 13,             // Category: Mesh
        MULTIHOP = 14,         // not used so far
        SELF_PROTECTED = 15,   // Category: Self Protected
        DMG = 16,              // Category: DMG
        FST = 18,              // Category: Fast Session Transfer
        UNPROTECTED_DMG = 20,  // Category: Unprotected DMG
        // Since vendor specific action has no stationary Action value,the parse process is not
        // here. Refer to vendor-specific-action in wave module.
        VENDOR_SPECIFIC_ACTION = 127,
    };
 
    enum QosActionValue
    {
        ADDTS_REQUEST = 0,
        ADDTS_RESPONSE = 1,
        DELTS = 2,
        SCHEDULE = 3,
        QOS_MAP_CONFIGURE = 4,
    };
 
    enum BlockAckActionValue
    {
        BLOCK_ACK_ADDBA_REQUEST = 0,
        BLOCK_ACK_ADDBA_RESPONSE = 1,
        BLOCK_ACK_DELBA = 2
    };
 
    enum PublicActionValue
    {
        QAB_REQUEST = 16,
        QAB_RESPONSE = 17,
    };
 
    enum RadioMeasurementActionValue
    {
        RADIO_MEASUREMENT_REQUEST = 0,
        RADIO_MEASUREMENT_REPORT = 1,
        LINK_MEASUREMENT_REQUEST = 2,
        LINK_MEASUREMENT_REPORT = 3,
        NEIGHBOR_REPORT_REQUEST = 4,
        NEIGHBOR_REPORT_RESPONSE = 5
    };
 
    enum MeshActionValue
    {
        LINK_METRIC_REPORT = 0,              // Action Value:0 in Category 13: Mesh
        PATH_SELECTION = 1,                  // Action Value:1 in Category 13: Mesh
        PORTAL_ANNOUNCEMENT = 2,             // Action Value:2 in Category 13: Mesh
        CONGESTION_CONTROL_NOTIFICATION = 3, // Action Value:3 in Category 13: Mesh
        MDA_SETUP_REQUEST =
            4, // Action Value:4 in Category 13: Mesh MCCA-Setup-Request (not used so far)
        MDA_SETUP_REPLY =
            5, // Action Value:5 in Category 13: Mesh MCCA-Setup-Reply (not used so far)
        MDAOP_ADVERTISMENT_REQUEST =
            6, // Action Value:6 in Category 13: Mesh MCCA-Advertisement-Request (not used so far)
        MDAOP_ADVERTISMENTS = 7,       // Action Value:7 in Category 13: Mesh (not used so far)
        MDAOP_SET_TEARDOWN = 8,        // Action Value:8 in Category 13: Mesh (not used so far)
        TBTT_ADJUSTMENT_REQUEST = 9,   // Action Value:9 in Category 13: Mesh (not used so far)
        TBTT_ADJUSTMENT_RESPONSE = 10, // Action Value:10 in Category 13: Mesh (not used so far)
    };
 
    enum MultihopActionValue
    {
        PROXY_UPDATE = 0,              // not used so far
        PROXY_UPDATE_CONFIRMATION = 1, // not used so far
    };
 
    enum SelfProtectedActionValue // Category: 15 (Self Protected)
    {
        PEER_LINK_OPEN = 1,    // Mesh Peering Open
        PEER_LINK_CONFIRM = 2, // Mesh Peering Confirm
        PEER_LINK_CLOSE = 3,   // Mesh Peering Close
        GROUP_KEY_INFORM = 4,  // Mesh Group Key Inform
        GROUP_KEY_ACK = 5,     // Mesh Group Key Acknowledge
    };
 
    enum DmgActionValue
    {
        DMG_POWER_SAVE_CONFIGURATION_REQUEST = 0,
        DMG_POWER_SAVE_CONFIGURATION_RESPONSE = 1,
        DMG_INFORMATION_REQUEST = 2,
        DMG_INFORMATION_RESPONSE = 3,
        DMG_HANDOVER_REQUEST = 4,
        DMG_HANDOVER_RESPONSE = 5,
        DMG_DTP_REQUEST = 6,
        DMG_DTP_RESPONSE = 7,
        DMG_RELAY_SEARCH_REQUEST = 8,
        DMG_RELAY_SEARCH_RESPONSE = 9,
        DMG_MULTI_RELAY_CHANNEL_MEASUREMENT_REQUEST = 10,
        DMG_MULTI_RELAY_CHANNEL_MEASUREMENT_REPORT = 11,
        DMG_RLS_REQUEST = 12,
        DMG_RLS_RESPONSE = 13,
        DMG_RLS_ANNOUNCEMENT = 14,
        DMG_RLS_TEARDOWN = 15,
        DMG_RELAY_ACK_REQUEST = 16,
        DMG_RELAY_ACK_RESPONSE = 17,
        DMG_TPA_REQUEST = 18,
        DMG_TPA_RESPONSE = 19,
        DMG_TPA_REPORT = 20,
        DMG_ROC_REQUEST = 21,
        DMG_ROC_RESPONSE = 22
    };
 
    enum FstActionValue
    {
        FST_SETUP_REQUEST = 0,
        FST_SETUP_RESPONSE = 1,
        FST_TEAR_DOWN = 2,
        FST_ACK_REQUEST = 3,
        FST_ACK_RESPONSE = 4,
        ON_CHANNEL_TUNNEL_REQUEST = 5
    };
 
    enum UnprotectedDmgActionValue
    {
        UNPROTECTED_DMG_ANNOUNCE = 0,
        UNPROTECTED_DMG_BRP = 1,
        UNPROTECTED_MIMO_BF_SETUP = 2,
        UNPROTECTED_MIMO_BF_POLL = 3,
        UNPROTECTED_MIMO_BF_FEEDBACK = 4,
        UNPROTECTED_MIMO_BF_SELECTION = 5,
    };
 
    typedef union {
        QosActionValue qos;                                 
        BlockAckActionValue blockAck;                       
        RadioMeasurementActionValue radioMeasurementAction; 
        PublicActionValue publicAction;                     
        SelfProtectedActionValue selfProtectedAction;       
        MultihopActionValue multihopAction;                 
        MeshActionValue meshAction;                         
        DmgActionValue dmgAction;                           
        FstActionValue fstAction;                           
        UnprotectedDmgActionValue unprotectedDmgAction;     
    } ActionValue;                                          
 
    void SetAction(CategoryValue type, ActionValue action);
 
    CategoryValue GetCategory();
    ActionValue GetAction();
 
    static TypeId GetTypeId();
    TypeId GetInstanceTypeId() const override;
    void Print(std::ostream& os) const override;
    uint32_t GetSerializedSize() const override;
    void Serialize(Buffer::Iterator start) const override;
    uint32_t Deserialize(Buffer::Iterator start) override;
 
  private:
    std::string CategoryValueToString(CategoryValue value) const;
    std::string SelfProtectedActionValueToString(SelfProtectedActionValue value) const;
    uint8_t m_category;    
    uint8_t m_actionValue; 
};
 
class MgtAddBaRequestHeader : public Header
{
  public:
    MgtAddBaRequestHeader();
 
    static TypeId GetTypeId();
    TypeId GetInstanceTypeId() const override;
    void Print(std::ostream& os) const override;
    uint32_t GetSerializedSize() const override;
    void Serialize(Buffer::Iterator start) const override;
    uint32_t Deserialize(Buffer::Iterator start) override;
 
    void SetDelayedBlockAck();
    void SetImmediateBlockAck();
    void SetTid(uint8_t tid);
    void SetTimeout(uint16_t timeout);
    void SetBufferSize(uint16_t size);
    void SetStartingSequence(uint16_t seq);
    void SetAmsduSupport(bool supported);
 
    uint16_t GetStartingSequence() const;
    uint8_t GetTid() const;
    bool IsImmediateBlockAck() const;
    uint16_t GetTimeout() const;
    uint16_t GetBufferSize() const;
    bool IsAmsduSupported() const;
 
  private:
    uint16_t GetParameterSet() const;
    void SetParameterSet(uint16_t params);
    uint16_t GetStartingSequenceControl() const;
    void SetStartingSequenceControl(uint16_t seqControl);
 
    uint8_t m_dialogToken;   
    uint8_t m_amsduSupport;  
    uint8_t m_policy;        
    uint8_t m_tid;           
    uint16_t m_bufferSize;   
    uint16_t m_timeoutValue; 
    uint16_t m_startingSeq;  
};
 
class MgtAddBaResponseHeader : public Header
{
  public:
    MgtAddBaResponseHeader();
 
    static TypeId GetTypeId();
    TypeId GetInstanceTypeId() const override;
    void Print(std::ostream& os) const override;
    uint32_t GetSerializedSize() const override;
    void Serialize(Buffer::Iterator start) const override;
    uint32_t Deserialize(Buffer::Iterator start) override;
 
    void SetDelayedBlockAck();
    void SetImmediateBlockAck();
    void SetTid(uint8_t tid);
    void SetTimeout(uint16_t timeout);
    void SetBufferSize(uint16_t size);
    void SetStatusCode(StatusCode code);
    void SetAmsduSupport(bool supported);
 
    StatusCode GetStatusCode() const;
    uint8_t GetTid() const;
    bool IsImmediateBlockAck() const;
    uint16_t GetTimeout() const;
    uint16_t GetBufferSize() const;
    bool IsAmsduSupported() const;
 
  private:
    uint16_t GetParameterSet() const;
    void SetParameterSet(uint16_t params);
 
    uint8_t m_dialogToken;   
    StatusCode m_code;       
    uint8_t m_amsduSupport;  
    uint8_t m_policy;        
    uint8_t m_tid;           
    uint16_t m_bufferSize;   
    uint16_t m_timeoutValue; 
};
 
class MgtDelBaHeader : public Header
{
  public:
    MgtDelBaHeader();
 
    static TypeId GetTypeId();
 
    TypeId GetInstanceTypeId() const override;
    void Print(std::ostream& os) const override;
    uint32_t GetSerializedSize() const override;
    void Serialize(Buffer::Iterator start) const override;
    uint32_t Deserialize(Buffer::Iterator start) override;
 
    bool IsByOriginator() const;
    uint8_t GetTid() const;
    void SetTid(uint8_t tid);
    void SetByOriginator();
    void SetByRecipient();
 
  private:
    uint16_t GetParameterSet() const;
    void SetParameterSet(uint16_t params);
 
    uint16_t m_initiator;  
    uint16_t m_tid;        
    uint16_t m_reasonCode; 
};
 
} // namespace ns3
 
#endif /* MGT_HEADERS_H */