Attachment #3138 for bug #2945

View | Details | Raw Unified | Return to bug 2945
Collapse All | Expand All




#include "ns3/packet.h"
#include <algorithm>
#include "ns3/hwmp-protocol.h"
// All information elements:
#include "ns3/ie-dot11s-beacon-timing.h"
#include "ns3/ie-dot11s-configuration.h"
#include "ns3/ie-dot11s-id.h"

{
}


TypeId
MeshInformationElementVector::GetTypeId ()
{

      i.Prev (2);
      return WifiInformationElementVector::DeserializeSingleIe (i);
    }
  if (GetSize () + length > m_maxSize)
    {
      NS_FATAL_ERROR ("Check max size for information element!");
    }
  newElement->DeserializeInformationField (i, length);
  i.Next (length);
  m_elements.push_back (newElement);
  return i.GetDistanceFrom (start);
}


} // namespace ns3




  return rates;
}

Ptr<DsssParameterSet>
ApWifiMac::GetDsssParameterSet (void) const
{
  NS_LOG_FUNCTION (this);
  Ptr<DsssParameterSet> dsssParameters = Create<DsssParameterSet> ();
  if (GetDsssSupported ())
    {
      dsssParameters->SetDsssSupported (1);
      dsssParameters->SetCurrentChannel (m_phy->GetChannelNumber ());
    }
  return dsssParameters;
}

  return capabilities;
}

Ptr<ErpInformation>
ApWifiMac::GetErpInformation (void) const
{
  NS_LOG_FUNCTION (this);
  Ptr<ErpInformation> information = Create<ErpInformation> ();
  information->SetErpSupported (1);
  if (GetErpSupported ())
    {
      information->SetNonErpPresent (!m_nonErpStations.empty ());
      information->SetUseProtection (GetUseNonErpProtection ());
      if (GetShortPreambleEnabled ())
        {
          information->SetBarkerPreambleMode (0);
        }
      else
        {
          information->SetBarkerPreambleMode (1);
        }
    }
  return information;
}

Ptr<EdcaParameterSet>
ApWifiMac::GetEdcaParameterSet (void) const
{
  NS_LOG_FUNCTION (this);
  Ptr<EdcaParameterSet> edcaParameters = Create<EdcaParameterSet> ();
  if (GetQosSupported ())
    {
      edcaParameters->SetQosSupported (1);
      Ptr<QosTxop> edca;
      Time txopLimit;

      edca = m_edca.find (AC_BE)->second;
      txopLimit = edca->GetTxopLimit ();
      edcaParameters->SetBeAci (0);
      edcaParameters->SetBeCWmin (edca->GetMinCw ());
      edcaParameters->SetBeCWmax (edca->GetMaxCw ());
      edcaParameters->SetBeAifsn (edca->GetAifsn ());
      edcaParameters->SetBeTXOPLimit (static_cast<uint16_t> (txopLimit.GetMicroSeconds () / 32));

      edca = m_edca.find (AC_BK)->second;
      txopLimit = edca->GetTxopLimit ();
      edcaParameters->SetBkAci (1);
      edcaParameters->SetBkCWmin (edca->GetMinCw ());
      edcaParameters->SetBkCWmax (edca->GetMaxCw ());
      edcaParameters->SetBkAifsn (edca->GetAifsn ());
      edcaParameters->SetBkTXOPLimit (static_cast<uint16_t> (txopLimit.GetMicroSeconds () / 32));

      edca = m_edca.find (AC_VI)->second;
      txopLimit = edca->GetTxopLimit ();
      edcaParameters->SetViAci (2);
      edcaParameters->SetViCWmin (edca->GetMinCw ());
      edcaParameters->SetViCWmax (edca->GetMaxCw ());
      edcaParameters->SetViAifsn (edca->GetAifsn ());
      edcaParameters->SetViTXOPLimit (static_cast<uint16_t> (txopLimit.GetMicroSeconds () / 32));

      edca = m_edca.find (AC_VO)->second;
      txopLimit = edca->GetTxopLimit ();
      edcaParameters->SetVoAci (3);
      edcaParameters->SetVoCWmin (edca->GetMinCw ());
      edcaParameters->SetVoCWmax (edca->GetMaxCw ());
      edcaParameters->SetVoAifsn (edca->GetAifsn ());
      edcaParameters->SetVoTXOPLimit (static_cast<uint16_t> (txopLimit.GetMicroSeconds () / 32));

      edcaParameters->SetQosInfo (0);
    }
  return edcaParameters;
}

Ptr<CfParameterSet>
ApWifiMac::GetCfParameterSet (void) const
{
  Ptr<CfParameterSet> cfParameterSet = Create<CfParameterSet> ();
  if (GetPcfSupported ())
    {
      cfParameterSet->SetPcfSupported (1);
      if (!m_cfPollingList.empty ())
        {
          cfParameterSet->SetCFPCount (0);
          cfParameterSet->SetCFPPeriod (1);
          cfParameterSet->SetCFPMaxDurationUs (GetCfpMaxDuration ().GetMicroSeconds ());
          cfParameterSet->SetCFPDurRemainingUs (GetCfpMaxDuration ().GetMicroSeconds ());
        }
    }
  return cfParameterSet;
}

Ptr<HtOperation>
ApWifiMac::GetHtOperation (void) const
{
  NS_LOG_FUNCTION (this);
  Ptr<HtOperation> operation = Create<HtOperation> ();
  if (GetHtSupported () || GetVhtSupported () || GetHeSupported ())
    {
      operation->SetHtSupported (1);
      operation->SetPrimaryChannel (m_phy->GetChannelNumber ());
      operation->SetRifsMode (GetRifsMode ());
      operation->SetNonGfHtStasPresent (IsNonGfHtStasPresent ());
      if (m_phy->GetChannelWidth () > 20)
        {
          operation->SetSecondaryChannelOffset (1);
          operation->SetStaChannelWidth (1);
        }
      if (m_nonHtStations.empty ())
        {
          operation->SetHtProtection (NO_PROTECTION);
        }
      else
        {
          operation->SetHtProtection (MIXED_MODE_PROTECTION);
        }
      uint64_t maxSupportedRate = 0; //in bit/s
      for (uint8_t i = 0; i < m_phy->GetNMcs (); i++)

                }
            }
        }
      operation->SetRxHighestSupportedDataRate (static_cast<uint16_t> (maxSupportedRate / 1e6)); //in Mbit/s
      operation->SetTxMcsSetDefined (nMcs > 0);
      operation->SetTxMaxNSpatialStreams (maxSpatialStream);
      //To be filled in once supported
      operation->SetObssNonHtStasPresent (0);
      operation->SetDualBeacon (0);
      operation->SetDualCtsProtection (0);
      operation->SetStbcBeacon (0);
      operation->SetLSigTxopProtectionFullSupport (0);
      operation->SetPcoActive (0);
      operation->SetPhase (0);
      operation->SetRxMcsBitmask (0);
      operation->SetTxRxMcsSetUnequal (0);
      operation->SetTxUnequalModulation (0);
    }
  return operation;
}

Ptr<VhtOperation>
ApWifiMac::GetVhtOperation (void) const
{
  NS_LOG_FUNCTION (this);
  Ptr<VhtOperation> operation = Create<VhtOperation> ();
  if (GetVhtSupported () || GetHeSupported ())
    {
      operation->SetVhtSupported (1);
      uint16_t channelWidth = GetVhtOperationalChannelWidth ();
      if (channelWidth == 160)
        {
          operation->SetChannelWidth (2);
        }
      else if (channelWidth == 80)
        {
          operation->SetChannelWidth (1);
        }
      else
        {
          operation->SetChannelWidth (0);
        }
      for (uint8_t nss = 1; nss <= 8; nss++)
        {

            {
              maxMcs = 0;
            }
          operation->SetMaxVhtMcsPerNss (nss, maxMcs);
        }
    }
  return operation;
}

Ptr<HeOperation>
ApWifiMac::GetHeOperation (void) const
{
  NS_LOG_FUNCTION (this);
  Ptr<HeOperation> operation = Create<HeOperation> ();
  if (GetHeSupported ())
    {
      operation->SetHeSupported (1);
      for (uint8_t nss = 1; nss <= m_phy->GetMaxSupportedRxSpatialStreams (); nss++)
        {
          operation->SetMaxHeMcsPerNss (nss, 11); //TBD: hardcode to 11 for now since we assume all MCS values are supported
        }
    }
  return operation;

  m_stationManager->SetShortSlotTimeEnabled (GetShortSlotTimeEnabled ());
  if (GetDsssSupported ())
    {
      probe.AddWifiInformationElement (GetDsssParameterSet ());
    }
  if (GetErpSupported ())
    {
      probe.AddWifiInformationElement (GetErpInformation ());
    }
  if (GetQosSupported ())
    {
      probe.AddWifiInformationElement (GetEdcaParameterSet ());
    }
  if (GetHtSupported () || GetVhtSupported () || GetHeSupported ())
    {
      probe.AddWifiInformationElement (GetExtendedCapabilities ());
      probe.AddWifiInformationElement (GetHtCapabilities ());
      probe.AddWifiInformationElement (GetHtOperation ());
    }
  if (GetVhtSupported () || GetHeSupported ())
    {
      probe.AddWifiInformationElement (GetVhtCapabilities ());
      probe.AddWifiInformationElement (GetVhtOperation ());
    }
  if (GetHeSupported ())
    {
      probe.AddWifiInformationElement (GetHeCapabilities ());
      probe.AddWifiInformationElement (GetHeOperation ());
    }
  packet->AddHeader (probe);


  assoc.SetCapabilities (GetCapabilities ());
  if (GetErpSupported ())
    {
      assoc.AddWifiInformationElement (GetErpInformation ());
    }
  if (GetQosSupported ())
    {
      assoc.AddWifiInformationElement (GetEdcaParameterSet ());
    }
  if (GetHtSupported () || GetVhtSupported () || GetHeSupported ())
    {
      assoc.AddWifiInformationElement (GetExtendedCapabilities ());
      assoc.AddWifiInformationElement (GetHtCapabilities ());
      assoc.AddWifiInformationElement (GetHtOperation ());
    }
  if (GetVhtSupported () || GetHeSupported ())
    {
      assoc.AddWifiInformationElement (GetVhtCapabilities ());
      assoc.AddWifiInformationElement (GetVhtOperation ());
    }
  if (GetHeSupported ())
    {
      assoc.AddWifiInformationElement (GetHeCapabilities ());
      assoc.AddWifiInformationElement (GetHeOperation ());
    }
  packet->AddHeader (assoc);


  m_stationManager->SetShortSlotTimeEnabled (GetShortSlotTimeEnabled ());
  if (GetPcfSupported ())
    {
      beacon.AddWifiInformationElement (GetCfParameterSet ());
    }
  if (GetDsssSupported ())
    {
      beacon.AddWifiInformationElement (GetDsssParameterSet ());
    }
  if (GetErpSupported ())
    {
      beacon.AddWifiInformationElement (GetErpInformation ());
    }
  if (GetQosSupported ())
    {
      beacon.AddWifiInformationElement (GetEdcaParameterSet ());
    }
  if (GetHtSupported () || GetVhtSupported ())
    {
      beacon.AddWifiInformationElement (GetExtendedCapabilities ());
      beacon.AddWifiInformationElement (GetHtCapabilities ());
      beacon.AddWifiInformationElement (GetHtOperation ());
    }
  if (GetVhtSupported () || GetHeSupported ())
    {
      beacon.AddWifiInformationElement (GetVhtCapabilities ());
      beacon.AddWifiInformationElement (GetVhtOperation ());
    }
  if (GetHeSupported ())
    {
      beacon.AddWifiInformationElement (GetHeCapabilities ());
      beacon.AddWifiInformationElement (GetHeOperation ());
    }
  packet->AddHeader (beacon);


              if (GetHtSupported ())
                {
                  //check whether the HT STA supports all MCSs in Basic MCS Set
                  Ptr<HtCapabilities> htcapabilities =
                      StaticCast<HtCapabilities> (assocReq.GetInformationElement (IE_HT_CAPABILITIES));
                  if (htcapabilities != 0)
                    {
                      isHtStation = true;
                      for (uint8_t i = 0; i < m_stationManager->GetNBasicMcs (); i++)
                        {
                          WifiMode mcs = m_stationManager->GetBasicMcs (i);
                          if (!htcapabilities->IsSupportedMcs (mcs.GetMcsValue ()))
                            {
                              problem = true;
                              break;

              if (GetVhtSupported ())
                {
                  //check whether the VHT STA supports all MCSs in Basic MCS Set
                  Ptr<VhtCapabilities> vhtcapabilities =
                      StaticCast<VhtCapabilities> (assocReq.GetInformationElement (IE_VHT_CAPABILITIES));
                  if (vhtcapabilities != 0)
                    {
                      for (uint8_t i = 0; i < m_stationManager->GetNBasicMcs (); i++)
                        {
                          WifiMode mcs = m_stationManager->GetBasicMcs (i);
                          if (!vhtcapabilities->IsSupportedTxMcs (mcs.GetMcsValue ()))
                            {
                              problem = true;
                              break;

              if (GetHeSupported ())
                {
                  //check whether the HE STA supports all MCSs in Basic MCS Set
                  Ptr<HeCapabilities> hecapabilities =
                      StaticCast<HeCapabilities> (assocReq.GetInformationElement (IE_HE_CAPABILITIES));
                  if (hecapabilities != 0)
                    {
                      for (uint8_t i = 0; i < m_stationManager->GetNBasicMcs (); i++)
                        {
                          WifiMode mcs = m_stationManager->GetBasicMcs (i);
                          if (!hecapabilities->IsSupportedTxMcs (mcs.GetMcsValue ()))
                            {
                              problem = true;
                              break;

                    }
                  if (GetHtSupported ())
                    {
                      Ptr<HtCapabilities> htCapabilities =
                          StaticCast<HtCapabilities> (assocReq.GetInformationElement (IE_HT_CAPABILITIES));
                      if (htCapabilities != 0)
                        {
                          m_stationManager->AddStationHtCapabilities (from, htCapabilities);
                        }
                    }
                  if (GetVhtSupported ())
                    {
                      Ptr<VhtCapabilities> vhtCapabilities =
                          StaticCast<VhtCapabilities> (assocReq.GetInformationElement (IE_VHT_CAPABILITIES));
                      if (vhtCapabilities != 0)
                        {
                          m_stationManager->AddStationVhtCapabilities (from, vhtCapabilities);
                          for (uint8_t i = 0; i < m_phy->GetNMcs (); i++)
                            {
                              WifiMode mcs = m_phy->GetMcs (i);
                              if (mcs.GetModulationClass () == WIFI_MOD_CLASS_VHT && vhtCapabilities->IsSupportedTxMcs (mcs.GetMcsValue ()))
                                {
                                  m_stationManager->AddSupportedMcs (hdr->GetAddr2 (), mcs);
                                  //here should add a control to add basic MCS when it is implemented

                    }
                  if (GetHtSupported () || GetVhtSupported ())
                    {
                      Ptr<ExtendedCapabilities> extendedCapabilities =
                          StaticCast<ExtendedCapabilities> (assocReq.GetInformationElement (IE_EXTENDED_CAPABILITIES));
                      //TODO: to be completed
                    }
                  if (GetHeSupported ())
                    {
                      Ptr<HeCapabilities> heCapabilities =
                          StaticCast<HeCapabilities> (assocReq.GetInformationElement (IE_HE_CAPABILITIES));
                      if (heCapabilities != 0)
                      for (uint8_t i = 0; i < m_phy->GetNMcs (); i++)
                        {
                          m_stationManager->AddStationHeCapabilities (from, heCapabilities);
                          for (uint8_t i = 0; i < m_phy->GetNMcs (); i++)
                            {
                              WifiMode mcs = m_phy->GetMcs (i);
                              if (mcs.GetModulationClass () == WIFI_MOD_CLASS_HE && heCapabilities->IsSupportedTxMcs (mcs.GetMcsValue ()))
                                {
                                  m_stationManager->AddSupportedMcs (hdr->GetAddr2 (), mcs);
                                  //here should add a control to add basic MCS when it is implemented
                                }
                            }
                        }
                    }

              if (GetHtSupported ())
                {
                  //check whether the HT STA supports all MCSs in Basic MCS Set
                  Ptr<HtCapabilities> htcapabilities =
                      StaticCast<HtCapabilities> (reassocReq.GetInformationElement (IE_HT_CAPABILITIES));
                  if (htcapabilities != 0)
                    {
                      isHtStation = true;
                      for (uint8_t i = 0; i < m_stationManager->GetNBasicMcs (); i++)
                        {
                          WifiMode mcs = m_stationManager->GetBasicMcs (i);
                          if (!htcapabilities->IsSupportedMcs (mcs.GetMcsValue ()))
                            {
                              problem = true;
                              break;

              if (GetVhtSupported ())
                {
                  //check whether the VHT STA supports all MCSs in Basic MCS Set
                  Ptr<VhtCapabilities> vhtcapabilities =
                      StaticCast<VhtCapabilities> (reassocReq.GetInformationElement (IE_VHT_CAPABILITIES));
                  if (vhtcapabilities != 0)
                    {
                      for (uint8_t i = 0; i < m_stationManager->GetNBasicMcs (); i++)
                        {
                          WifiMode mcs = m_stationManager->GetBasicMcs (i);
                          if (!vhtcapabilities->IsSupportedTxMcs (mcs.GetMcsValue ()))
                            {
                              problem = true;
                              break;

              if (GetHeSupported ())
                {
                  //check whether the HE STA supports all MCSs in Basic MCS Set
                  Ptr<HeCapabilities> hecapabilities =
                      StaticCast<HeCapabilities> (reassocReq.GetInformationElement (IE_HE_CAPABILITIES));
                  if (hecapabilities != 0)
                    {
                      for (uint8_t i = 0; i < m_stationManager->GetNBasicMcs (); i++)
                        {
                          WifiMode mcs = m_stationManager->GetBasicMcs (i);
                          if (!hecapabilities->IsSupportedTxMcs (mcs.GetMcsValue ()))
                            {
                              problem = true;
                              break;

                    }
                  if (GetHtSupported ())
                    {
                      Ptr<HtCapabilities> htCapabilities =
                          StaticCast<HtCapabilities> (reassocReq.GetInformationElement (IE_HT_CAPABILITIES));
                      if (htCapabilities != 0)
                        {
                          m_stationManager->AddStationHtCapabilities (from, htCapabilities);
                        }
                    }
                  if (GetVhtSupported ())
                    {
                      Ptr<VhtCapabilities> vhtCapabilities =
                          StaticCast<VhtCapabilities> (reassocReq.GetInformationElement (IE_VHT_CAPABILITIES));
                      //we will always fill in RxHighestSupportedLgiDataRate field at TX, so this can be used to check whether it supports VHT
                      if (vhtCapabilities != 0)
                        {
                          m_stationManager->AddStationVhtCapabilities (from, vhtCapabilities);
                          for (uint8_t i = 0; i < m_phy->GetNMcs (); i++)
                            {
                              WifiMode mcs = m_phy->GetMcs (i);
                              if (mcs.GetModulationClass () == WIFI_MOD_CLASS_VHT && vhtCapabilities->IsSupportedTxMcs (mcs.GetMcsValue ()))
                                {
                                  m_stationManager->AddSupportedMcs (hdr->GetAddr2 (), mcs);
                                  //here should add a control to add basic MCS when it is implemented

                    }
                  if (GetHtSupported () || GetVhtSupported ())
                    {
                      Ptr<ExtendedCapabilities> extendedCapabilities =
                          StaticCast<ExtendedCapabilities> (reassocReq.GetInformationElement (IE_EXTENDED_CAPABILITIES));
                      //TODO: to be completed
                    }
                  if (GetHeSupported ())
                    {
                      Ptr<HeCapabilities> heCapabilities =
                          StaticCast<HeCapabilities> (reassocReq.GetInformationElement (IE_HE_CAPABILITIES));
                      if (heCapabilities != 0)
                      for (uint8_t i = 0; i < m_phy->GetNMcs (); i++)
                        {
                          m_stationManager->AddStationHeCapabilities (from, heCapabilities);
                          for (uint8_t i = 0; i < m_phy->GetNMcs (); i++)
                            {
                              WifiMode mcs = m_phy->GetMcs (i);
                              if (mcs.GetModulationClass () == WIFI_MOD_CLASS_HE && heCapabilities->IsSupportedTxMcs (mcs.GetMcsValue ()))
                                {
                                  m_stationManager->AddSupportedMcs (hdr->GetAddr2 (), mcs);
                                  //here should add a control to add basic MCS when it is implemented
                                }
                            }
                        }
                    }




   *
   * \return the ERP information that we support
   */
  Ptr<ErpInformation> GetErpInformation (void) const;
  /**
   * Return the EDCA Parameter Set of the current AP.
   *
   * \return the EDCA Parameter Set that we support
   */
  Ptr<EdcaParameterSet> GetEdcaParameterSet (void) const;
  /**
   * Return the CF parameter set of the current AP.
   *
   * \return the CF parameter set that we support
   */
  Ptr<CfParameterSet> GetCfParameterSet (void) const;
  /**
   * Return the HT operation of the current AP.
   *
   * \return the HT operation that we support
   */
  Ptr<HtOperation> GetHtOperation (void) const;
  /**
   * Return the VHT operation of the current AP.
   *
   * \return the VHT operation that we support
   */
  Ptr<VhtOperation> GetVhtOperation (void) const;
  /**
   * Return the HE operation of the current AP.
   *
   * \return the HE operation that we support
   */
  Ptr<HeOperation> GetHeOperation (void) const;
  /**
   * Return an instance of SupportedRates that contains all rates that we support
   * including HT rates.

   *
   * \return the DSSS Parameter Set that we support
   */
  Ptr<DsssParameterSet> GetDsssParameterSet (void) const;
  /**
   * Enable or disable beacon generation of the AP.
   *




  m_CFPDurRemaining = durRemaining;
}

void
CfParameterSet::Print (std::ostream& os) const
{
  os <<  +GetCFPCount ()
     << "|" << +GetCFPPeriod ()
     << "|" << GetCFPMaxDurationUs ()
     << "|" << GetCFPDurRemainingUs ();
}

/**
 * output operator
 *
 * \param os output stream
 * \param cfparameterset the CF parameter set
 *
 * \return output stream
 */
std::ostream &
operator << (std::ostream &os, const CfParameterSet &cfparameterset)
{
  cfparameterset.Print (os);
     << "|" << (uint16_t)cfParameterSet.GetCFPPeriod ()
     << "|" << cfParameterSet.GetCFPMaxDurationUs ()
     << "|" << cfParameterSet.GetCFPDurRemainingUs ();

  return os;
}





   */
  Buffer::Iterator Serialize (Buffer::Iterator start) const;
  /**
   * Return the serialized size of this CF Parameter Set.
   *
   * \return the serialized size of this CF Parameter Set
   */
  uint16_t GetSerializedSize () const;

  /**
   * Generate human-readable form of CF Parameter Set
   *
   * \param os output stream
   */
  void Print (std::ostream& os) const;


private:
  uint8_t m_CFPCount;                 ///< CFP Count

  uint8_t m_pcfSupported;
};

std::ostream &operator << (std::ostream &os, const CfParameterSet &cfparameterset);

} //namespace ns3





  return length;
}

void
DsssParameterSet::Print (std::ostream& os) const
{
  os << +m_currentChannel;
}

/**
 * output stream output operator
 *
 * \param os output stream
 * \param dsssparameterset the DSSS parameter set
 *
 * \returns output stream
 */
std::ostream &
operator << (std::ostream &os, const DsssParameterSet &dsssparameterset)
{
  dsssparameterset.Print (os);
  return os;
}

} //namespace ns3

(-)a/src/wifi/model/dsss-parameter-set.h (+9 lines)

Lines 90-95	Link Here

*/

*/

  uint16_t GetSerializedSize () const;

  uint16_t GetSerializedSize () const;

/**

   * Generate human-readable form of DSSS Parameter Set

   * \param os output stream

*/

  void Print (std::ostream& os) const;

100

private:

101

private:

  uint8_t m_currentChannel; ///< current channel number

102

  uint8_t m_currentChannel; ///< current channel number

Lines 98-103	Link Here

  bool m_dsssSupported;

105

  bool m_dsssSupported;

};

106

};

100

107

108

std::ostream &operator << (std::ostream &os, const DsssParameterSet &dsssparameterset);

109

101

} //namespace ns3

110

} //namespace ns3

102

111

103

#endif /* DSSS_PARAMETER_SET_H */

112

#endif /* DSSS_PARAMETER_SET_H */




  return length;
}

void
EdcaParameterSet::Print (std::ostream& os) const
{
  os << +m_qosInfo << "|"
     << m_acBE << "|"
     << m_acBK << "|"
     << m_acVI << "|"
     << m_acVO;
}

/**
 * output stream output operator
 *
 * \param os output stream
 * \param edcaparameterset the EDCA parameter set
 *
 * \returns output stream
 */
std::ostream &
operator << (std::ostream &os, const EdcaParameterSet &edcaparameterset)
{
  edcaparameterset.Print (os);
  return os;
}

} //namespace ns3

(-)a/src/wifi/model/edca-parameter-set.h (-2 / +11 lines)

Lines 311-322	Link Here

311

*/

311

*/

312

  Buffer::Iterator Serialize (Buffer::Iterator start) const;

312

  Buffer::Iterator Serialize (Buffer::Iterator start) const;

313

/**

313

/**

314

   * Return the serialized size of this EDCA Parameter Set.

314

   * Return the serialized size of this EDCA Parameter Set IE.

315

316

   * \return the serialized size of this EDCA Parameter Set

316

   * \return the serialized size of this EDCA Parameter Set IE

317

*/

317

*/

318

  uint16_t GetSerializedSize () const;

318

  uint16_t GetSerializedSize () const;

319

320

/**

321

   * Generate human-readable form of EDCA Parameter Set IE

322

323

   * \param os output stream

324

*/

325

  void Print (std::ostream& os) const;

326

320

327

321

private:

328

private:

322

  uint8_t m_qosInfo;  ///< QOS info

329

  uint8_t m_qosInfo;  ///< QOS info

Lines 330-335	Link Here

330

  bool m_qosSupported;

337

  bool m_qosSupported;

331

};

338

};

332

339

340

std::ostream &operator << (std::ostream &os, const EdcaParameterSet &edcaparameterset);

341

333

} //namespace ns3

342

} //namespace ns3

334

343

335

#endif /* EDCA_PARAMETER_SET_H */

344

#endif /* EDCA_PARAMETER_SET_H */




  return length;
}

void
ErpInformation::Print (std::ostream& os) const
{
  os <<  bool (GetBarkerPreambleMode ())
     << "|" << bool (GetUseProtection ())
     << "|" << bool (GetNonErpPresent ());
}

/**
 * output stream output operator
 *
 * \param os output stream
 * \param erpinformation the ERP information
 *
 * \returns output stream
 */
std::ostream &
operator << (std::ostream &os, const ErpInformation &erpinformation)
{
  erpinformation.Print (os);
     << "|" << bool (erpinformation.GetUseProtection ())
     << "|" << bool (erpinformation.GetNonErpPresent ());

  return os;
}





   */
  Buffer::Iterator Serialize (Buffer::Iterator start) const;
  /**
   * Return the serialized size of this ERP Information IE.
   *
   * \return the serialized size of this ERP Information IE
   */
  uint16_t GetSerializedSize () const;

  /**
   * Generate human-readable form of ERP Information IE
   *
   * \param os output stream
   */
  void Print (std::ostream& os) const;


private:
  uint8_t m_erpInformation; ///< ERP information


  bool m_erpSupported;
};

/**
 * output stream output operator
 *
 * \param os output stream
 * \param erpinformation
 *
 * \returns output stream
 */
std::ostream &operator << (std::ostream &os, const ErpInformation &erpinformation);

} //namespace ns3

(-)a/src/wifi/model/extended-capabilities.cc (-1 / +7 lines)

Lines 393-398	Link Here

393

  return length;

393

  return length;

394

395

396

void

397

ExtendedCapabilities::Print (std::ostream& os) const

398

399

  os << +GetExtendedCapabilitiesByte1 ();

400

401

396

/**

402

/**

397

 * output stream output operator

403

 * output stream output operator

398

404

Lines 404-410	Link Here

404

std::ostream &

410

std::ostream &

405

operator << (std::ostream &os, const ExtendedCapabilities &extendedcapabilities)

411

operator << (std::ostream &os, const ExtendedCapabilities &extendedcapabilities)

406

412

407

  os << +extendedcapabilities.GetExtendedCapabilitiesByte1 ();

413

  extendedcapabilities.Print (os);

408

  return os;

414

  return os;

409

415

410

416

(-)a/src/wifi/model/extended-capabilities.h (+7 lines)

Lines 194-199	Link Here

194

*/

194

*/

195

  uint16_t GetSerializedSize () const;

195

  uint16_t GetSerializedSize () const;

196

197

/**

198

   * Generate human-readable form of extended capabilities information element

199

200

   * \param os output stream

201

*/

202

  void Print (std::ostream& os) const;

203

197

204

198

private:

205

private:

199

  //fields supported by HT and VHT

206

  //fields supported by HT and VHT




  return m_highestNssSupportedM1 + 1;
}

void
HeCapabilities::Print (std::ostream& os) const
{
  os << GetHeMacCapabilitiesInfo1 () << "|"
     << +GetHeMacCapabilitiesInfo2 () << "|"
     << GetHePhyCapabilitiesInfo1 () << "|"
     << +GetHePhyCapabilitiesInfo2 () << "|"
     << GetSupportedMcsAndNss ();
}

/**
 * output stream output operator
 * \param os the output stream
 * \param hecapabilities the HE capabilities
 * \returns the output stream
 */
std::ostream &
operator << (std::ostream &os, const HeCapabilities &hecapabilities)
{
  hecapabilities.Print (os);
     << +HeCapabilities.GetHeMacCapabilitiesInfo2 () << "|"
     << HeCapabilities.GetHePhyCapabilitiesInfo1 () << "|"
     << +HeCapabilities.GetHePhyCapabilitiesInfo2 () << "|"
     << HeCapabilities.GetSupportedMcsAndNss ();
  return os;
}


(-)a/src/wifi/model/he-capabilities.h (-1 / +8 lines)

Lines 210-215	Link Here

210

*/

210

*/

211

  uint16_t GetSerializedSize () const;

211

  uint16_t GetSerializedSize () const;

212

213

/**

214

   * Generate human-readable form of HE capabilities information element

215

216

   * \param os output stream

217

*/

218

  void Print (std::ostream& os) const;

219

213

220

214

private:

221

private:

215

  //MAC Capabilities Info fields

222

  //MAC Capabilities Info fields

Lines 285-291	Link Here

285

  uint8_t m_heSupported;

292

  uint8_t m_heSupported;

286

};

293

};

287

294

288

std::ostream &operator << (std::ostream &os, const HeCapabilities &HeCapabilities);

295

std::ostream &operator << (std::ostream &os, const HeCapabilities &hecapabilities);

289

296

290

} //namespace ns3

297

} //namespace ns3

291

298




  return length;
}

void
HeOperation::Print (std::ostream& os) const
{
  os << GetHeOperationParameters () << "|"
     << GetBasicHeMcsAndNssSet ();
}

/**
 * output stream output operator
 *
 * \param os output stream
 * \param heoperation the HE operation
 *
 * \returns output stream
 */
std::ostream &
operator << (std::ostream &os, const HeOperation &heoperation)
{
  heoperation.Print (os);
     << HeOperation.GetBasicHeMcsAndNssSet ();
  return os;
}


(-)a/src/wifi/model/he-operation.h (+7 lines)

Lines 117-122	Link Here

117

*/

117

*/

118

  uint16_t GetSerializedSize () const;

118

  uint16_t GetSerializedSize () const;

119

120

/**

121

   * Generate human-readable form of HE Operations IE

122

123

   * \param os output stream

124

*/

125

  void Print (std::ostream& os) const;

126

120

127

121

private:

128

private:

122

  //HE Operation Parameters fields

129

  //HE Operation Parameters fields




  return length;
}

void
HtCapabilities::Print (std::ostream& os) const
{
  os << bool (GetLdpc ())
     << "|" << bool (GetSupportedChannelWidth ())
     << "|" << bool (GetGreenfield ())
     << "|" << bool (GetShortGuardInterval20 ()) << "|";
  for (uint8_t i = 0; i < MAX_SUPPORTED_MCS; i++)
    {
      os << IsSupportedMcs (i) << " ";
    }
}

/**
 * output stream output operator
 *

std::ostream &
operator << (std::ostream &os, const HtCapabilities &htcapabilities)
{
  htcapabilities.Print (os);
     << "|" << bool (htcapabilities.GetSupportedChannelWidth ())
     << "|" << bool (htcapabilities.GetGreenfield ())
     << "|" << bool (htcapabilities.GetShortGuardInterval20 ()) << "|";
  for (uint8_t i = 0; i < MAX_SUPPORTED_MCS; i++)
    {
      os << htcapabilities.IsSupportedMcs (i) << " ";
    }
  return os;
}





   */
  Buffer::Iterator Serialize (Buffer::Iterator start) const;
  /**
   * Return the serialized size of this HT capabilities
   * information element.
   *
   * \return the serialized size of this HT capabilities information element
   */
  uint16_t GetSerializedSize () const;

  /**
   * Generate human-readable form of HT capabilities information element
   *
   * \param os output stream
   */
  void Print (std::ostream& os) const;


private:
  //HT Capabilities Info field


} //namespace ns3

#endif /* HT_CAPABILITIES_H */




  return length;
}

void
HtOperation::Print (std::ostream& os) const
{
  os << bool (GetPrimaryChannel ())
     << "|" << +GetSecondaryChannelOffset ()
     << "|" << bool (GetStaChannelWidth ())
     << "|" << bool (GetRifsMode ())
     << "|" << +GetHtProtection ()
     << "|" << bool (GetNonGfHtStasPresent ())
     << "|" << bool (GetObssNonHtStasPresent ())
     << "|" << bool (GetDualBeacon ())
     << "|" << bool (GetDualCtsProtection ())
     << "|" << bool (GetStbcBeacon ())
     << "|" << bool (GetLSigTxopProtectionFullSupport ())
     << "|" << bool (GetPcoActive ())
     << "|" << bool (GetPhase ())
     << "|" << GetRxHighestSupportedDataRate ()
     << "|" << bool (GetTxMcsSetDefined ())
     << "|" << bool (GetTxRxMcsSetUnequal ())
     << "|" << +GetTxMaxNSpatialStreams ()
     << "|" << bool (GetTxUnequalModulation ()) << "|";
  for (uint8_t i = 0; i < MAX_SUPPORTED_MCS; i++)
    {
      os << IsSupportedMcs (i) << " ";
    }
}

/**
 * output stream output operator
 *

std::ostream &
operator << (std::ostream &os, const HtOperation &htOperation)
{
  htOperation.Print (os);
     << "|" << +htOperation.GetSecondaryChannelOffset ()
     << "|" << bool (htOperation.GetStaChannelWidth ())
     << "|" << bool (htOperation.GetRifsMode ())
     << "|" << +htOperation.GetHtProtection ()
     << "|" << bool (htOperation.GetNonGfHtStasPresent ())
     << "|" << bool (htOperation.GetObssNonHtStasPresent ())
     << "|" << bool (htOperation.GetDualBeacon ())
     << "|" << bool (htOperation.GetDualCtsProtection ())
     << "|" << bool (htOperation.GetStbcBeacon ())
     << "|" << bool (htOperation.GetLSigTxopProtectionFullSupport ())
     << "|" << bool (htOperation.GetPcoActive ())
     << "|" << bool (htOperation.GetPhase ())
     << "|" << htOperation.GetRxHighestSupportedDataRate ()
     << "|" << bool (htOperation.GetTxMcsSetDefined ())
     << "|" << bool (htOperation.GetTxRxMcsSetUnequal ())
     << "|" << +htOperation.GetTxMaxNSpatialStreams ()
     << "|" << bool (htOperation.GetTxUnequalModulation ()) << "|";
  for (uint8_t i = 0; i < MAX_SUPPORTED_MCS; i++)
    {
      os << htOperation.IsSupportedMcs (i) << " ";
    }
  return os;
}


(-)a/src/wifi/model/ht-operation.h (+7 lines)

Lines 400-405	Link Here

400

*/

400

*/

401

  uint16_t GetSerializedSize () const;

401

  uint16_t GetSerializedSize () const;

402

403

/**

404

   * Generate human-readable form of HT Operation IE

405

406

   * \param os output stream

407

*/

408

  void Print (std::ostream& os) const;

409

403

410

404

private:

411

private:

405

  uint8_t m_primaryChannel; ///< primary channel

412

  uint8_t m_primaryChannel; ///< primary channel





namespace ns3 {

void
MgtCommonHeader::AddWifiInformationElement (Ptr<WifiInformationElement> element)
{
  m_map[element->ElementId ()] = element;
}

Ptr<WifiInformationElement>
MgtCommonHeader::GetInformationElement (WifiInformationElementId id)
{
  WifiInformationElementMap::const_iterator it = m_map.find (id);
  if (it != m_map.end())
    {
      return it->second;
    }
  else
    {
      return 0;
    }
}

uint32_t
MgtCommonHeader::GetInformationElementsSerializedSize (void) const
{
  Ptr<WifiInformationElement> element;
  uint32_t size = 0;
  for (WifiInformationElementMap::const_iterator elem = m_map.begin (); elem != m_map.end (); elem++)
    {
      element = elem->second;
      size += element->GetSerializedSize ();
    }
  return size;
}

WifiInformationElementMap
MgtCommonHeader::GetListOfInformationElement (void) const
{
  return m_map;
}

void
MgtCommonHeader::PrintInformationElements (std::ostream &os) const
{
  Ptr<WifiInformationElement> element;
  for (WifiInformationElementMap::const_iterator elem = m_map.begin (); elem != m_map.end ();)
    {
      switch (elem->first)
        {
          case IE_HT_CAPABILITIES:
              os << "HT Capabilities=";
              break;
          case IE_VHT_CAPABILITIES:
              os << "VHT Capabilities=";
              break;
          case IE_HE_CAPABILITIES:
              os << "HE Capabilities=";
              break;
          case IE_EXTENDED_CAPABILITIES:
              os << "Extended Capabilities=";
              break;
          case IE_HT_OPERATION:
              os << "HT Operation=";
              break;
          case IE_VHT_OPERATION:
              os << "VHT Operation=";
              break;
          case IE_HE_OPERATION:
              os << "HE Operation=";
              break;
          case IE_ERP_INFORMATION:
              os << "ERP information=";
              break;
          case IE_EDCA_PARAMETER_SET:
              os << "EDCA parameter set=";
              break;
          case IE_DSSS_PARAMETER_SET:
              os << "DSSS parameter set=";
              break;
          case IE_CF_PARAMETER_SET:
              os << "CF parameter set=";
              break;
          default:
              os << "Unknown IE=";
              break;
        }
      elem->second->Print (os);
      ++elem;
      if (elem != m_map.end ())
        {
          os << " , ";
        }
    }
}

Buffer::Iterator
MgtCommonHeader::SerializeInformationElements (Buffer::Iterator start) const
{
  Buffer::Iterator i = start;
  Ptr<WifiInformationElement> element;
  for (WifiInformationElementMap::const_iterator elem = m_map.begin (); elem != m_map.end (); elem++)
    {
      element = elem->second;
      i = element->Serialize (i);
    }
  return i;
}

Buffer::Iterator
MgtCommonHeader::DeserializeInformationElements (Buffer::Iterator start)
{
  Buffer::Iterator i = start;
  Ptr<WifiInformationElement> element;
  uint8_t id, length;
  while ((!i.IsEnd ()) && (i.GetRemainingSize() > 2))
    {
      id = i.ReadU8 ();
      length = i.ReadU8 ();
      switch (id)
        {
          case IE_HT_CAPABILITIES:
              element = Create<HtCapabilities> ();
              break;
          case IE_VHT_CAPABILITIES:
              element = Create<VhtCapabilities> ();
              break;
          case IE_HE_CAPABILITIES:
              element = Create<HeCapabilities> ();
              break;
          case IE_EXTENDED_CAPABILITIES:
              element = Create<ExtendedCapabilities> ();
              break;
          case IE_HT_OPERATION:
              element = Create<HtOperation> ();
              break;
          case IE_VHT_OPERATION:
              element = Create<VhtOperation> ();
              break;
          case IE_HE_OPERATION:
              element = Create<HeOperation> ();
              break;
          case IE_ERP_INFORMATION:
              element = Create<ErpInformation> ();
              break;
          case IE_EDCA_PARAMETER_SET:
              element = Create<EdcaParameterSet> ();
              break;
          case IE_DSSS_PARAMETER_SET:
              element = Create<DsssParameterSet> ();
              break;
          case IE_CF_PARAMETER_SET:
              element = Create<CfParameterSet> ();
              break;
          default:
              // We peeked at the ID and length, so we need to back up the
              // pointer before deferring to our parent.
              i.Prev (2);
              return i;
        }
      i = element->DeserializeElementBody (i, length);
      m_map[id] = element;
    }
  return i;
}

/***********************************************************
 *                    Probe Request
 ***********************************************************/

NS_OBJECT_ENSURE_REGISTERED (MgtProbeRequestHeader);

  m_rates = rates;
}

void
MgtProbeRequestHeader::SetExtendedCapabilities (ExtendedCapabilities extendedcapabilities)
{
  m_extendedCapability = extendedcapabilities;
}

ExtendedCapabilities
MgtProbeRequestHeader::GetExtendedCapabilities (void) const
{
  return m_extendedCapability;
}

void
MgtProbeRequestHeader::SetHtCapabilities (HtCapabilities htcapabilities)
{
  m_htCapability = htcapabilities;
}

HtCapabilities
MgtProbeRequestHeader::GetHtCapabilities (void) const
{
  return m_htCapability;
}

void
MgtProbeRequestHeader::SetVhtCapabilities (VhtCapabilities vhtcapabilities)
{
  m_vhtCapability = vhtcapabilities;
}

VhtCapabilities
MgtProbeRequestHeader::GetVhtCapabilities (void) const
{
  return m_vhtCapability;
}

void
MgtProbeRequestHeader::SetHeCapabilities (HeCapabilities hecapabilities)
{
  m_heCapability = hecapabilities;
}

HeCapabilities
MgtProbeRequestHeader::GetHeCapabilities (void) const
{
  return m_heCapability;
}

SupportedRates
MgtProbeRequestHeader::GetSupportedRates (void) const
{

  size += m_ssid.GetSerializedSize ();
  size += m_rates.GetSerializedSize ();
  size += m_rates.extended.GetSerializedSize ();
  size += GetInformationElementsSerializedSize ();
  size += m_htCapability.GetSerializedSize ();
  size += m_vhtCapability.GetSerializedSize ();
  size += m_heCapability.GetSerializedSize ();
  return size;
}


MgtProbeRequestHeader::Print (std::ostream &os) const
{
  os << "ssid=" << m_ssid << ", "
     << "rates=" << m_rates << ", ";
  PrintInformationElements (os);
     << "HT Capabilities=" << m_htCapability << " , "
     << "VHT Capabilities=" << m_vhtCapability << " , "
     << "HE Capabilities=" << m_heCapability;
}

void

  i = m_ssid.Serialize (i);
  i = m_rates.Serialize (i);
  i = m_rates.extended.Serialize (i);
  i = SerializeInformationElements (i);
  i = m_htCapability.Serialize (i);
  i = m_vhtCapability.Serialize (i);
  i = m_heCapability.Serialize (i);
}

uint32_t

  i = m_ssid.Deserialize (i);
  i = m_rates.Deserialize (i);
  i = m_rates.extended.DeserializeIfPresent (i);
  i = DeserializeInformationElements (i);
  i = m_htCapability.DeserializeIfPresent (i);
  i = m_vhtCapability.DeserializeIfPresent (i);
  i = m_heCapability.DeserializeIfPresent (i);
  return i.GetDistanceFrom (start);
}


/***********************************************************
 *                    Probe Response
 ***********************************************************/

NS_OBJECT_ENSURE_REGISTERED (MgtProbeResponseHeader);

}

void
MgtProbeResponseHeader::SetExtendedCapabilities (ExtendedCapabilities extendedcapabilities)
{
  m_extendedCapability = extendedcapabilities;
}

ExtendedCapabilities
MgtProbeResponseHeader::GetExtendedCapabilities (void) const
{
  return m_extendedCapability;
}

void
MgtProbeResponseHeader::SetHtCapabilities (HtCapabilities htcapabilities)
{
  m_htCapability = htcapabilities;
}

HtCapabilities
MgtProbeResponseHeader::GetHtCapabilities (void) const
{
  return m_htCapability;
}

void
MgtProbeResponseHeader::SetHtOperation (HtOperation htoperation)
{
  m_htOperation = htoperation;
}

HtOperation
MgtProbeResponseHeader::GetHtOperation (void) const
{
  return m_htOperation;
}

void
MgtProbeResponseHeader::SetVhtCapabilities (VhtCapabilities vhtcapabilities)
{
  m_vhtCapability = vhtcapabilities;
}

VhtCapabilities
MgtProbeResponseHeader::GetVhtCapabilities (void) const
{
  return m_vhtCapability;
}

void
MgtProbeResponseHeader::SetVhtOperation (VhtOperation vhtoperation)
{
  m_vhtOperation = vhtoperation;
}

VhtOperation
MgtProbeResponseHeader::GetVhtOperation (void) const
{
  return m_vhtOperation;
}

void
MgtProbeResponseHeader::SetHeCapabilities (HeCapabilities hecapabilities)
{
  m_heCapability = hecapabilities;
}

HeCapabilities
MgtProbeResponseHeader::GetHeCapabilities (void) const
{
  return m_heCapability;
}

void
MgtProbeResponseHeader::SetHeOperation (HeOperation heoperation)
{
  m_heOperation = heoperation;
}

HeOperation
MgtProbeResponseHeader::GetHeOperation (void) const
{
  return m_heOperation;
}

void
MgtProbeResponseHeader::SetCfParameterSet (CfParameterSet cfparameterset)
{
  m_cfParameterSet = cfparameterset;
}

CfParameterSet
MgtProbeResponseHeader::GetCfParameterSet (void) const
{
  return m_cfParameterSet;
}

void
MgtProbeResponseHeader::SetSsid (Ssid ssid)
{
  m_ssid = ssid;

  m_rates = rates;
}

void
MgtProbeResponseHeader::SetDsssParameterSet (DsssParameterSet dsssParameterSet)
{
  m_dsssParameterSet = dsssParameterSet;
}

DsssParameterSet
MgtProbeResponseHeader::GetDsssParameterSet (void) const
{
  return m_dsssParameterSet;
}

void
MgtProbeResponseHeader::SetErpInformation (ErpInformation erpInformation)
{
  m_erpInformation = erpInformation;
}

ErpInformation
MgtProbeResponseHeader::GetErpInformation (void) const
{
  return m_erpInformation;
}

void
MgtProbeResponseHeader::SetEdcaParameterSet (EdcaParameterSet edcaparameters)
{
  m_edcaParameterSet = edcaparameters;
}

EdcaParameterSet
MgtProbeResponseHeader::GetEdcaParameterSet (void) const
{
  return m_edcaParameterSet;
}

TypeId
MgtProbeResponseHeader::GetTypeId (void)
{

  size += m_capability.GetSerializedSize ();
  size += m_ssid.GetSerializedSize ();
  size += m_rates.GetSerializedSize ();
  size += m_cfParameterSet.GetSerializedSize ();
  size += m_dsssParameterSet.GetSerializedSize ();
  size += m_erpInformation.GetSerializedSize ();
  size += m_rates.extended.GetSerializedSize ();
  size += GetInformationElementsSerializedSize ();
  size += m_extendedCapability.GetSerializedSize ();
  size += m_htCapability.GetSerializedSize ();
  size += m_htOperation.GetSerializedSize ();
  size += m_vhtCapability.GetSerializedSize ();
  size += m_vhtOperation.GetSerializedSize ();
  size += m_heCapability.GetSerializedSize ();
  size += m_heOperation.GetSerializedSize ();
  return size;
}

void
MgtProbeResponseHeader::Print (std::ostream &os) const
{
  os << "Timestamp=" << m_timestamp << ", " <<
        "BeaconInterval=" << m_beaconInterval  << ", " <<
        "rates=" << m_rates << ", "
        "ssid=" << m_ssid << ", ";
  PrintInformationElements (os);
     << "HT Operation=" << m_htOperation << " , "
     << "VHT Capabilities=" << m_vhtCapability << " , "
     << "VHT Operation=" << m_vhtOperation << " , "
     << "HE Capabilities=" << m_heCapability << " , "
     << "HE Operation=" << m_heOperation;
}

void

  //capability information
  //ssid
  //supported rates
  //fh parameter set
  //ds parameter set
  //cf parameter set
  //ibss parameter set
  Buffer::Iterator i = start;
  i.WriteHtolsbU64 (Simulator::Now ().GetMicroSeconds ());
  i.WriteHtolsbU16 (static_cast<uint16_t> (m_beaconInterval / 1024));
  i = m_capability.Serialize (i);
  i = m_ssid.Serialize (i);
  i = m_rates.Serialize (i);
  i = m_cfParameterSet.Serialize (i);
  i = m_dsssParameterSet.Serialize (i);
  i = m_erpInformation.Serialize (i);
  i = m_rates.extended.Serialize (i);
  i = SerializeInformationElements (i);
  i = m_extendedCapability.Serialize (i);
  i = m_htCapability.Serialize (i);
  i = m_htOperation.Serialize (i);
  i = m_vhtCapability.Serialize (i);
  i = m_vhtOperation.Serialize (i);
  i = m_heCapability.Serialize (i);
  i = m_heOperation.Serialize (i);
}

uint32_t

  i = m_capability.Deserialize (i);
  i = m_ssid.Deserialize (i);
  i = m_rates.Deserialize (i);
  i = m_cfParameterSet.DeserializeIfPresent (i);
  i = m_dsssParameterSet.DeserializeIfPresent (i);
  i = m_erpInformation.DeserializeIfPresent (i);
  i = m_rates.extended.DeserializeIfPresent (i);
  i = DeserializeInformationElements (i);
  i = m_extendedCapability.DeserializeIfPresent (i);
  i = m_htCapability.DeserializeIfPresent (i);
  i = m_htOperation.DeserializeIfPresent (i);
  i = m_vhtCapability.DeserializeIfPresent (i);
  i = m_vhtOperation.DeserializeIfPresent (i);
  i = m_heCapability.DeserializeIfPresent (i);
  i = m_heOperation.DeserializeIfPresent (i);
  return i.GetDistanceFrom (start);
}




/***********************************************************
 *                    Assoc Request
 ***********************************************************/

NS_OBJECT_ENSURE_REGISTERED (MgtAssocRequestHeader);

  return m_capability;
}

void
MgtAssocRequestHeader::SetExtendedCapabilities (ExtendedCapabilities extendedcapabilities)
{
  m_extendedCapability = extendedcapabilities;
}

ExtendedCapabilities
MgtAssocRequestHeader::GetExtendedCapabilities (void) const
{
  return m_extendedCapability;
}

void
MgtAssocRequestHeader::SetHtCapabilities (HtCapabilities htcapabilities)
{
  m_htCapability = htcapabilities;
}

HtCapabilities
MgtAssocRequestHeader::GetHtCapabilities (void) const
{
  return m_htCapability;
}

void
MgtAssocRequestHeader::SetVhtCapabilities (VhtCapabilities vhtcapabilities)
{
  m_vhtCapability = vhtcapabilities;
}

VhtCapabilities
MgtAssocRequestHeader::GetVhtCapabilities (void) const
{
  return m_vhtCapability;
}

void
MgtAssocRequestHeader::SetHeCapabilities (HeCapabilities hecapabilities)
{
  m_heCapability = hecapabilities;
}

HeCapabilities
MgtAssocRequestHeader::GetHeCapabilities (void) const
{
  return m_heCapability;
}

Ssid
MgtAssocRequestHeader::GetSsid (void) const
{

  size += m_ssid.GetSerializedSize ();
  size += m_rates.GetSerializedSize ();
  size += m_rates.extended.GetSerializedSize ();
  size += GetInformationElementsSerializedSize ();
  size += m_htCapability.GetSerializedSize ();
  size += m_vhtCapability.GetSerializedSize ();
  size += m_heCapability.GetSerializedSize ();
  return size;
}

void
MgtAssocRequestHeader::Print (std::ostream &os) const
{
  os << "ssid=" << m_ssid  << ", "
     << "rates=" << m_rates << ", ";
  PrintInformationElements (os);
     << "HT Capabilities=" << m_htCapability << " , "
     << "VHT Capabilities=" << m_vhtCapability << " , "
     << "HE Capabilities=" << m_heCapability;
}

void

  i = m_ssid.Serialize (i);
  i = m_rates.Serialize (i);
  i = m_rates.extended.Serialize (i);
  i = SerializeInformationElements (i);
  i = m_htCapability.Serialize (i);
  i = m_vhtCapability.Serialize (i);
  i = m_heCapability.Serialize (i);
}

uint32_t

  i = m_ssid.Deserialize (i);
  i = m_rates.Deserialize (i);
  i = m_rates.extended.DeserializeIfPresent (i);
  i = DeserializeInformationElements (i);
  i = m_htCapability.DeserializeIfPresent (i);
  i = m_vhtCapability.DeserializeIfPresent (i);
  i = m_heCapability.DeserializeIfPresent (i);
  return i.GetDistanceFrom (start);
}


/***********************************************************
 *                     Ressoc Request
 ***********************************************************/

NS_OBJECT_ENSURE_REGISTERED (MgtReassocRequestHeader);

  return m_capability;
}

void
MgtReassocRequestHeader::SetExtendedCapabilities (ExtendedCapabilities extendedcapabilities)
{
  m_extendedCapability = extendedcapabilities;
}

ExtendedCapabilities
MgtReassocRequestHeader::GetExtendedCapabilities (void) const
{
  return m_extendedCapability;
}

void
MgtReassocRequestHeader::SetHtCapabilities (HtCapabilities htcapabilities)
{
  m_htCapability = htcapabilities;
}

HtCapabilities
MgtReassocRequestHeader::GetHtCapabilities (void) const
{
  return m_htCapability;
}

void
MgtReassocRequestHeader::SetVhtCapabilities (VhtCapabilities vhtcapabilities)
{
  m_vhtCapability = vhtcapabilities;
}

VhtCapabilities
MgtReassocRequestHeader::GetVhtCapabilities (void) const
{
  return m_vhtCapability;
}

void
MgtReassocRequestHeader::SetHeCapabilities (HeCapabilities hecapabilities)
{
  m_heCapability = hecapabilities;
}

HeCapabilities
MgtReassocRequestHeader::GetHeCapabilities (void) const
{
  return m_heCapability;
}

Ssid
MgtReassocRequestHeader::GetSsid (void) const
{

  size += m_ssid.GetSerializedSize ();
  size += m_rates.GetSerializedSize ();
  size += m_rates.extended.GetSerializedSize ();
  size += GetInformationElementsSerializedSize ();
  size += m_htCapability.GetSerializedSize ();
  size += m_vhtCapability.GetSerializedSize ();
  size += m_heCapability.GetSerializedSize ();
  return size;
}


{
  os << "current AP address=" << m_currentApAddr << ", "
     << "ssid=" << m_ssid << ", "
     << "rates=" << m_rates << ", ";
  PrintInformationElements (os);
     << "HT Capabilities=" << m_htCapability << " , "
     << "VHT Capabilities=" << m_vhtCapability << " , "
     << "HE Capabilities=" << m_heCapability;
}

void

  i = m_ssid.Serialize (i);
  i = m_rates.Serialize (i);
  i = m_rates.extended.Serialize (i);
  i = SerializeInformationElements (i);
  i = m_htCapability.Serialize (i);
  i = m_vhtCapability.Serialize (i);
  i = m_heCapability.Serialize (i);
}

uint32_t

  i = m_ssid.Deserialize (i);
  i = m_rates.Deserialize (i);
  i = m_rates.extended.DeserializeIfPresent (i);
  i = DeserializeInformationElements (i);
  i = m_htCapability.DeserializeIfPresent (i);
  i = m_vhtCapability.DeserializeIfPresent (i);
  i = m_heCapability.DeserializeIfPresent (i);
  return i.GetDistanceFrom (start);
}


/***********************************************************
 *                Assoc/Reassoc Response
 ***********************************************************/

NS_OBJECT_ENSURE_REGISTERED (MgtAssocResponseHeader);

  return m_capability;
}

void
MgtAssocResponseHeader::SetExtendedCapabilities (ExtendedCapabilities extendedcapabilities)
{
  m_extendedCapability = extendedcapabilities;
}

ExtendedCapabilities
MgtAssocResponseHeader::GetExtendedCapabilities (void) const
{
  return m_extendedCapability;
}

void
MgtAssocResponseHeader::SetHtCapabilities (HtCapabilities htcapabilities)
{
  m_htCapability = htcapabilities;
}

HtCapabilities
MgtAssocResponseHeader::GetHtCapabilities (void) const
{
  return m_htCapability;
}

void
MgtAssocResponseHeader::SetHtOperation (HtOperation htoperation)
{
  m_htOperation = htoperation;
}

HtOperation
MgtAssocResponseHeader::GetHtOperation (void) const
{
  return m_htOperation;
}

void
MgtAssocResponseHeader::SetVhtCapabilities (VhtCapabilities vhtcapabilities)
{
  m_vhtCapability = vhtcapabilities;
}

VhtCapabilities
MgtAssocResponseHeader::GetVhtCapabilities (void) const
{
  return m_vhtCapability;
}

void
MgtAssocResponseHeader::SetVhtOperation (VhtOperation vhtoperation)
{
  m_vhtOperation = vhtoperation;
}

VhtOperation
MgtAssocResponseHeader::GetVhtOperation (void) const
{
  return m_vhtOperation;
}

void
MgtAssocResponseHeader::SetHeCapabilities (HeCapabilities hecapabilities)
{
  m_heCapability = hecapabilities;
}

HeCapabilities
MgtAssocResponseHeader::GetHeCapabilities (void) const
{
  return m_heCapability;
}

void
MgtAssocResponseHeader::SetHeOperation (HeOperation heoperation)
{
  m_heOperation = heoperation;
}

HeOperation
MgtAssocResponseHeader::GetHeOperation (void) const
{
  return m_heOperation;
}

void
MgtAssocResponseHeader::SetErpInformation (ErpInformation erpInformation)
{
  m_erpInformation = erpInformation;
}

ErpInformation
MgtAssocResponseHeader::GetErpInformation (void) const
{
  return m_erpInformation;
}

void
MgtAssocResponseHeader::SetEdcaParameterSet (EdcaParameterSet edcaparameters)
{
  m_edcaParameterSet = edcaparameters;
}

EdcaParameterSet
MgtAssocResponseHeader::GetEdcaParameterSet (void) const
{
  return m_edcaParameterSet;
}

TypeId
MgtAssocResponseHeader::GetTypeId (void)
{

  size += m_code.GetSerializedSize ();
  size += 2; //aid
  size += m_rates.GetSerializedSize ();
  size += m_erpInformation.GetSerializedSize ();
  size += m_rates.extended.GetSerializedSize ();
  size += GetInformationElementsSerializedSize ();
  size += m_extendedCapability.GetSerializedSize ();
  size += m_htCapability.GetSerializedSize ();
  size += m_htOperation.GetSerializedSize ();
  size += m_vhtCapability.GetSerializedSize ();
  size += m_vhtOperation.GetSerializedSize ();
  size += m_heCapability.GetSerializedSize ();
  size += m_heOperation.GetSerializedSize ();
  return size;
}


{
  os << "status code=" << m_code << ", "
     << "aid=" << m_aid << ", "
     << "rates=" << m_rates << ", ";
  PrintInformationElements (os);
     << "Extended Capabilities=" << m_extendedCapability << " , "
     << "HT Capabilities=" << m_htCapability << " , "
     << "HT Operation=" << m_htOperation << " , "
     << "VHT Capabilities=" << m_vhtCapability << " , "
     << "VHT Operation=" << m_vhtOperation << " , "
     << "HE Capabilities=" << m_heCapability << " , "
     << "HE Operation=" << m_heOperation;
}

void

  i = m_code.Serialize (i);
  i.WriteHtolsbU16 (m_aid);
  i = m_rates.Serialize (i);
  i = m_erpInformation.Serialize (i);
  i = m_rates.extended.Serialize (i);
  i = SerializeInformationElements (i);
  i = m_extendedCapability.Serialize (i);
  i = m_htCapability.Serialize (i);
  i = m_htOperation.Serialize (i);
  i = m_vhtCapability.Serialize (i);
  i = m_vhtOperation.Serialize (i);
  i = m_heCapability.Serialize (i);
  i = m_heOperation.Serialize (i);
}

uint32_t

  i = m_code.Deserialize (i);
  m_aid = i.ReadLsbtohU16 ();
  i = m_rates.Deserialize (i);
  i = m_erpInformation.DeserializeIfPresent (i);
  i = m_rates.extended.DeserializeIfPresent (i);
  i = DeserializeInformationElements (i);
  i = m_extendedCapability.DeserializeIfPresent (i);
  i = m_htCapability.DeserializeIfPresent (i);
  i = m_htOperation.DeserializeIfPresent (i);
  i = m_vhtCapability.DeserializeIfPresent (i);
  i = m_vhtOperation.DeserializeIfPresent (i);
  i = m_heCapability.DeserializeIfPresent (i);
  i = m_heOperation.DeserializeIfPresent (i);
  return i.GetDistanceFrom (start);
}


/**********************************************************
 *                     ActionFrame
 **********************************************************/
WifiActionHeader::WifiActionHeader ()
{




#ifndef MGT_HEADERS_H
#define MGT_HEADERS_H

#include <map>
#include "ns3/mac48-address.h"
#include "status-code.h"
#include "capability-information.h"


namespace ns3 {

typedef std::vector<Ptr<WifiInformationElement> > WifiInfoElementList;
typedef std::vector<WifiInformationElementId> WifiInformationElementIdList;
typedef std::map<WifiInformationElementId, Ptr<WifiInformationElement> > WifiInformationElementMap;

/**
 * \ingroup wifi
 * Implement the common header for management frames.
 */
class MgtCommonHeader : public Header {
public:
  /**
   * Add new Wifi Information Element that is not vendor specific.
   * \param element the Wifi Information Element to be added to the management frame header.
   */
  void AddWifiInformationElement (Ptr<WifiInformationElement> element);
  /**
   * Get a specific Wifi information element by ID.
   * \param id The ID of the Wifi Information Element.
   * \return
   */
  Ptr<WifiInformationElement> GetInformationElement (WifiInformationElementId id);
  /**
   * Get List of Information Element associated with this frame.
   * \return
   */
  WifiInformationElementMap GetListOfInformationElement (void) const;


protected:
  void PrintInformationElements (std::ostream &os) const;
  uint32_t GetInformationElementsSerializedSize (void) const;
  Buffer::Iterator SerializeInformationElements (Buffer::Iterator start) const;
  Buffer::Iterator DeserializeInformationElements (Buffer::Iterator start);


private:
  WifiInformationElementMap m_map; //!< Map of Wifi Information Element.
};

/**
 * \ingroup wifi
 * Implement the header for management frames of type association request.
 */
class MgtAssocRequestHeader : public MgtCommonHeader
{
public:
  MgtAssocRequestHeader ();

   */
  void SetCapabilities (CapabilityInformation capabilities);
  /**
   * Set the Extended Capabilities.
   *
   * \param extendedcapabilities the Extended Capabilities
   */
  void SetExtendedCapabilities (ExtendedCapabilities extendedcapabilities);
  /**
   * Set the HT capabilities.
   *
   * \param htcapabilities HT capabilities
   */
  void SetHtCapabilities (HtCapabilities htcapabilities);
  /**
   * Set the VHT capabilities.
   *
   * \param vhtcapabilities VHT capabilities
   */
  void SetVhtCapabilities (VhtCapabilities vhtcapabilities);
  /**
   * Set the HE capabilities.
   *
   * \param hecapabilities HE capabilities
   */
  void SetHeCapabilities (HeCapabilities hecapabilities);
  /**
   * Return the Capability information.
   *
   * \return Capability information
   */
  CapabilityInformation GetCapabilities (void) const;
  /**
   * Return the extended capabilities.
   *
   * \return the extended capabilities
   */
  ExtendedCapabilities GetExtendedCapabilities (void) const;
  /**
   * Return the HT capabilities.
   *
   * \return HT capabilities
   */
  HtCapabilities GetHtCapabilities (void) const;
  /**
   * Return the VHT capabilities.
   *
   * \return VHT capabilities
   */
  VhtCapabilities GetVhtCapabilities (void) const;
  /**
   * Return the HE capabilities.
   *
   * \return HE capabilities
   */
  HeCapabilities GetHeCapabilities (void) const;
  /**
   * Return the Service Set Identifier (SSID).
   *
   * \return SSID

  Ssid m_ssid;                        //!< Service Set ID (SSID)
  SupportedRates m_rates;             //!< List of supported rates
  CapabilityInformation m_capability; //!< Capability information
  ExtendedCapabilities m_extendedCapability; //!< Extended capabilities
  HtCapabilities m_htCapability;      //!< HT capabilities
  VhtCapabilities m_vhtCapability;    //!< VHT capabilities
  HeCapabilities m_heCapability;      //!< HE capabilities
  uint16_t m_listenInterval;          //!< listen interval
};


 * \ingroup wifi
 * Implement the header for management frames of type reassociation request.
 */
class MgtReassocRequestHeader : public MgtCommonHeader
{
public:
  MgtReassocRequestHeader ();

   */
  void SetCapabilities (CapabilityInformation capabilities);
  /**
   * Set the Extended Capabilities.
   *
   * \param extendedcapabilities the Extended Capabilities
   */
  void SetExtendedCapabilities (ExtendedCapabilities extendedcapabilities);
  /**
   * Set the HT capabilities.
   *
   * \param htcapabilities HT capabilities
   */
  void SetHtCapabilities (HtCapabilities htcapabilities);
  /**
   * Set the VHT capabilities.
   *
   * \param vhtcapabilities VHT capabilities
   */
  void SetVhtCapabilities (VhtCapabilities vhtcapabilities);
  /**
   * Set the HE capabilities.
   *
   * \param hecapabilities HE capabilities
   */
  void SetHeCapabilities (HeCapabilities hecapabilities);
  /**
   * Return the Capability information.
   *
   * \return Capability information
   */
  CapabilityInformation GetCapabilities (void) const;
  /**
   * Return the extended capabilities.
   *
   * \return the extended capabilities
   */
  ExtendedCapabilities GetExtendedCapabilities (void) const;
  /**
   * Return the HT capabilities.
   *
   * \return HT capabilities
   */
  HtCapabilities GetHtCapabilities (void) const;
  /**
   * Return the VHT capabilities.
   *
   * \return VHT capabilities
   */
  VhtCapabilities GetVhtCapabilities (void) const;
  /**
   * Return the HE capabilities.
   *
   * \return HE capabilities
   */
  HeCapabilities GetHeCapabilities (void) const;
  /**
   * Return the Service Set Identifier (SSID).
   *
   * \return SSID

  Ssid m_ssid;                        //!< Service Set ID (SSID)
  SupportedRates m_rates;             //!< List of supported rates
  CapabilityInformation m_capability; //!< Capability information
  ExtendedCapabilities m_extendedCapability; //!< Extended capabilities
  HtCapabilities m_htCapability;      //!< HT capabilities
  VhtCapabilities m_vhtCapability;    //!< VHT capabilities
  HeCapabilities m_heCapability;      //!< HE capabilities
  uint16_t m_listenInterval;          //!< listen interval
};


 * \ingroup wifi
 * Implement the header for management frames of type association and reassociation response.
 */
class MgtAssocResponseHeader : public MgtCommonHeader
{
public:
  MgtAssocResponseHeader ();

   */
  CapabilityInformation GetCapabilities (void) const;
  /**
   * Return the extended capabilities.
   *
   * \return the extended capabilities
   */
  ExtendedCapabilities GetExtendedCapabilities (void) const;
  /**
   * Return the HT capabilities.
   *
   * \return HT capabilities
   */
  HtCapabilities GetHtCapabilities (void) const;
  /**
   * Return the HT operation.
   *
   * \return HT operation
   */
  HtOperation GetHtOperation (void) const;
  /**
   * Return the VHT capabilities.
   *
   * \return VHT capabilities
   */
  VhtCapabilities GetVhtCapabilities (void) const;
  /**
   * Return the VHT operation.
   *
   * \return VHT operation
   */
  VhtOperation GetVhtOperation (void) const;
  /**
   * Return the HE capabilities.
   *
   * \return HE capabilities
   */
  HeCapabilities GetHeCapabilities (void) const;
  /**
   * Return the HE operation.
   *
   * \return HE operation
   */
  HeOperation GetHeOperation (void) const;
  /**
   * Return the ERP information.
   *
   * \return the ERP information
   */
  ErpInformation GetErpInformation (void) const;
  /**
   * Return the EDCA Parameter Set.
   *
   * \return the EDCA Parameter Set
   */
  EdcaParameterSet GetEdcaParameterSet (void) const;
  /**
   * Set the Capability information.
   *
   * \param capabilities Capability information
   */
  void SetCapabilities (CapabilityInformation capabilities);
  /**
   * Set the extended capabilities.
   *
   * \param extendedcapabilities the extended capabilities
   */
  void SetExtendedCapabilities (ExtendedCapabilities extendedcapabilities);
  /**
   * Set the VHT operation.
   *
   * \param vhtoperation VHT operation
   */
  void SetVhtOperation (VhtOperation vhtoperation);
  /**
   * Set the VHT capabilities.
   *
   * \param vhtcapabilities VHT capabilities
   */
  void SetVhtCapabilities (VhtCapabilities vhtcapabilities);
  /**
   * Set the HT capabilities.
   *
   * \param htcapabilities HT capabilities
   */
  void SetHtCapabilities (HtCapabilities htcapabilities);
  /**
   * Set the HT operation.
   *
   * \param htoperation HT operation
   */
  void SetHtOperation (HtOperation htoperation);
  /**
   * Set the supported rates.
   *
   * \param rates the supported rates

   * \param aid the association ID
   */
  void SetAssociationId (uint16_t aid);
  /**
   * Set the ERP information.
   *
   * \param erpInformation the ERP information
   */
  void SetErpInformation (ErpInformation erpInformation);
  /**
   * Set the EDCA Parameter Set.
   *
   * \param edcaParameterSet the EDCA Parameter Set
   */
  void SetEdcaParameterSet (EdcaParameterSet edcaParameterSet);
  /**
   * Set the HE capabilities.
   *
   * \param hecapabilities HE capabilities
   */
  void SetHeCapabilities (HeCapabilities hecapabilities);
  /**
   * Set the HE operation.
   *
   * \param heoperation HE operation
   */
  void SetHeOperation (HeOperation heoperation);

  /**
   * Register this type.

  CapabilityInformation m_capability; //!< Capability information
  StatusCode m_code; //!< Status code
  uint16_t m_aid; //!< aid
  ExtendedCapabilities m_extendedCapability; //!< extended capabilities
  HtCapabilities m_htCapability; //!< HT capabilities
  HtOperation m_htOperation; //!< HT operation
  VhtCapabilities m_vhtCapability; //!< VHT capabilities
  VhtOperation m_vhtOperation; //!< VHT operation
  ErpInformation m_erpInformation; //!< ERP information
  EdcaParameterSet m_edcaParameterSet; //!< EDCA Parameter Set
  HeCapabilities m_heCapability; //!< HE capabilities
  HeOperation m_heOperation; //!< HE operation
};



 * \ingroup wifi
 * Implement the header for management frames of type probe request.
 */
class MgtProbeRequestHeader : public MgtCommonHeader
{
public:
  ~MgtProbeRequestHeader ();

   */
  void SetSupportedRates (SupportedRates rates);
  /**
   * Set the extended capabilities.
   *
   * \param extendedcapabilities the extended capabilities
   */
  void SetExtendedCapabilities (ExtendedCapabilities extendedcapabilities);
  /**
   * Set the HT capabilities.
   *
   * \param htcapabilities HT capabilities
   */
  void SetHtCapabilities (HtCapabilities htcapabilities);
  /**
   * Set the VHT capabilities.
   *
   * \param vhtcapabilities VHT capabilities
   */
  void SetVhtCapabilities (VhtCapabilities vhtcapabilities);
  /**
   * Set the HE capabilities.
   *
   * \param hecapabilities HE capabilities
   */
  void SetHeCapabilities (HeCapabilities hecapabilities);
  /**
   * Return the Service Set Identifier (SSID).
   *
   * \return SSID

   * \return the supported rates
   */
  SupportedRates GetSupportedRates (void) const;
  /**
   * Return the extended capabilities.
   *
   * \return the extended capabilities
   */
  ExtendedCapabilities GetExtendedCapabilities (void) const;
  /**
   * Return the HT capabilities.
   *
   * \return HT capabilities
   */
  HtCapabilities GetHtCapabilities (void) const;
  /**
   * Return the VHT capabilities.
   *
   * \return VHT capabilities
   */
  VhtCapabilities GetVhtCapabilities (void) const;
  /**
   * Return the HE capabilities.
   *
   * \return HE capabilities
   */
  HeCapabilities GetHeCapabilities (void) const;

  /**
   * Register this type.

private:
  Ssid m_ssid;                     //!< Service Set ID (SSID)
  SupportedRates m_rates;          //!< List of supported rates
  ExtendedCapabilities m_extendedCapability; //!< extended capabilities
  HtCapabilities m_htCapability;   //!< HT capabilities
  VhtCapabilities m_vhtCapability; //!< VHT capabilities
  HeCapabilities m_heCapability; //!< HE capabilities
};



 * \ingroup wifi
 * Implement the header for management frames of type probe response.
 */
class MgtProbeResponseHeader : public MgtCommonHeader
{
public:
  MgtProbeResponseHeader ();

   */
  CapabilityInformation GetCapabilities (void) const;
  /**
   * Return the DSSS Parameter Set.
   *
   * \return the DSSS Parameter Set
   */
  DsssParameterSet GetDsssParameterSet (void) const;
  /**
   * Return the extended capabilities.
   *
   * \return the extended capabilities
   */
  ExtendedCapabilities GetExtendedCapabilities (void) const;
  /**
   * Return the HT capabilities.
   *
   * \return HT capabilities
   */
  HtCapabilities GetHtCapabilities (void) const;
  /**
   * Return the HT operation.
   *
   * \return HT operation
   */
  HtOperation GetHtOperation (void) const;
  /**
   * Return the VHT capabilities.
   *
   * \return VHT capabilities
   */
  VhtCapabilities GetVhtCapabilities (void) const;
  /**
   * Return the VHT operation.
   *
   * \return VHT operation
   */
  VhtOperation GetVhtOperation (void) const;
  /**
   * Return the HE capabilities.
   *
   * \return HE capabilities
   */
  HeCapabilities GetHeCapabilities (void) const;
  /**
   * Return the HE operation.
   *
   * \return HE operation
   */
  HeOperation GetHeOperation (void) const;
  /**
   * Return the ERP information.
   *
   * \return the ERP information
   */
  ErpInformation GetErpInformation (void) const;
  /**
   * Return the EDCA Parameter Set.
   *
   * \return the EDCA Parameter Set
   */
  EdcaParameterSet GetEdcaParameterSet (void) const;
  /**
   * Return the CF parameter set.
   *
   * \return CF parameter set
   */
  CfParameterSet GetCfParameterSet (void) const;
  /**
   * Set the Capability information.
   *
   * \param capabilities Capability information
   */
  void SetCapabilities (CapabilityInformation capabilities);
  /**
   * Set the extended capabilities.
   *
   * \param extendedcapabilities the extended capabilities
   */
  void SetExtendedCapabilities (ExtendedCapabilities extendedcapabilities);
  /**
   * Set the HT capabilities.
   *
   * \param htcapabilities HT capabilities
   */
  void SetHtCapabilities (HtCapabilities htcapabilities);
  /**
   * Set the HT operation.
   *
   * \param htoperation HT operation
   */
  void SetHtOperation (HtOperation htoperation);
  /**
   * Set the VHT capabilities.
   *
   * \param vhtcapabilities VHT capabilities
   */
  void SetVhtCapabilities (VhtCapabilities vhtcapabilities);
  /**
   * Set the VHT operation.
   *
   * \param vhtoperation VHT operation
   */
  void SetVhtOperation (VhtOperation vhtoperation);
  /**
   * Set the HE capabilities.
   *
   * \param hecapabilities HE capabilities
   */
  void SetHeCapabilities (HeCapabilities hecapabilities);
  /**
   * Set the HE operation.
   *
   * \param heoperation HE operation
   */
  void SetHeOperation (HeOperation heoperation);
  /**
   * Set the Service Set Identifier (SSID).
   *
   * \param ssid SSID

   */
  void SetSupportedRates (SupportedRates rates);
  /**
   * Set the DSSS Parameter Set.
   *
   * \param dsssParameterSet the DSSS Parameter Set
   */
  void SetDsssParameterSet (DsssParameterSet dsssParameterSet);
  /**
   * Set the ERP information.
   *
   * \param erpInformation the ERP information
   */
  void SetErpInformation (ErpInformation erpInformation);
  /**
   * Set the EDCA Parameter Set.
   *
   * \param edcaParameterSet the EDCA Parameter Set
   */
  void SetEdcaParameterSet (EdcaParameterSet edcaParameterSet);
  /**
   * Set the CF parameter set.
   *
   * \param cfparameterset CF parameter set
   */
  void SetCfParameterSet (CfParameterSet cfparameterset);
  /**
   * Return the time stamp.
   *
   * \return time stamp

  uint64_t m_beaconInterval;           //!< Beacon interval
  SupportedRates m_rates;              //!< List of supported rates
  CapabilityInformation m_capability;  //!< Capability information
  DsssParameterSet m_dsssParameterSet; //!< DSSS Parameter Set
  ExtendedCapabilities m_extendedCapability; //!< extended capabilities
  HtCapabilities m_htCapability;       //!< HT capabilities
  HtOperation m_htOperation;           //!< HT operation
  VhtCapabilities m_vhtCapability;     //!< VHT capabilities
  VhtOperation m_vhtOperation;         //!< VHT operation
  HeCapabilities m_heCapability;       //!< HE capabilities
  HeOperation m_heOperation;         //!< HE operation
  ErpInformation m_erpInformation;     //!< ERP information
  EdcaParameterSet m_edcaParameterSet; //!< EDCA Parameter Set
  CfParameterSet m_cfParameterSet;     //!< CF parameter set
};






  return m_stationManager;
}

Ptr<ExtendedCapabilities>
RegularWifiMac::GetExtendedCapabilities (void) const
{
  NS_LOG_FUNCTION (this);
  Ptr<ExtendedCapabilities> capabilities = Create<ExtendedCapabilities> ();
  if (m_htSupported || m_vhtSupported)
    {
      if (m_htSupported)
        {
          capabilities->SetHtSupported (1);
        }
      if (m_vhtSupported)
        {
          capabilities->SetVhtSupported (1);
        }
    }
  //TODO: to be completed
  return capabilities;
}

Ptr<HtCapabilities>
RegularWifiMac::GetHtCapabilities (void) const
{
  NS_LOG_FUNCTION (this);
  Ptr<HtCapabilities> capabilities = Create<HtCapabilities> ();
  if (m_htSupported || m_vhtSupported || m_heSupported)
    {
      capabilities->SetHtSupported (1);
      capabilities->SetLdpc (m_phy->GetLdpc ());
      capabilities->SetSupportedChannelWidth (m_phy->GetChannelWidth () >= 40);
      capabilities->SetShortGuardInterval20 (m_phy->GetShortGuardInterval ());
      capabilities->SetShortGuardInterval40 (m_phy->GetChannelWidth () >= 40 && m_phy->GetShortGuardInterval ());
      capabilities->SetGreenfield (m_phy->GetGreenfield ());
      uint32_t maxAmsduLength = std::max (std::max (m_beMaxAmsduSize, m_bkMaxAmsduSize), std::max (m_voMaxAmsduSize, m_viMaxAmsduSize));
      capabilities->SetMaxAmsduLength (maxAmsduLength > 3839); //0 if 3839 and 1 if 7935
      capabilities->SetLSigProtectionSupport (!m_phy->GetGreenfield ());
      double maxAmpduLengthExponent = std::max (std::ceil ((std::log (std::max (std::max (m_beMaxAmpduSize, m_bkMaxAmpduSize), std::max (m_voMaxAmpduSize, m_viMaxAmpduSize))
                                                                      + 1.0)
                                                            / std::log (2.0))
                                                           - 13.0),
                                                0.0);
      NS_ASSERT (maxAmpduLengthExponent >= 0 && maxAmpduLengthExponent <= 255);
      capabilities->SetMaxAmpduLength (std::max<uint8_t> (3, static_cast<uint8_t> (maxAmpduLengthExponent))); //0 to 3 for HT
      uint64_t maxSupportedRate = 0; //in bit/s
      for (uint8_t i = 0; i < m_phy->GetNMcs (); i++)
        {

            {
              continue;
            }
          capabilities->SetRxMcsBitmask (mcs.GetMcsValue ());
          uint8_t nss = (mcs.GetMcsValue () / 8) + 1;
          NS_ASSERT (nss > 0 && nss < 5);
          uint64_t dataRate = mcs.GetDataRate (m_phy->GetChannelWidth (), m_phy->GetShortGuardInterval () ? 400 : 800, nss);

              NS_LOG_DEBUG ("Updating maxSupportedRate to " << maxSupportedRate);
            }
        }
      capabilities->SetRxHighestSupportedDataRate (static_cast<uint16_t> (maxSupportedRate / 1e6)); //in Mbit/s
      capabilities->SetTxMcsSetDefined (m_phy->GetNMcs () > 0);
      capabilities->SetTxMaxNSpatialStreams (m_phy->GetMaxSupportedTxSpatialStreams ());
      //we do not support unequal modulations
      capabilities->SetTxRxMcsSetUnequal (0);
      capabilities->SetTxUnequalModulation (0);
    }
  return capabilities;
}

Ptr<VhtCapabilities>
RegularWifiMac::GetVhtCapabilities (void) const
{
  NS_LOG_FUNCTION (this);
  Ptr<VhtCapabilities> capabilities = Create<VhtCapabilities> ();
  if (m_vhtSupported || m_heSupported)
    {
      capabilities->SetVhtSupported (1);
      if (m_phy->GetChannelWidth () == 160)
        {
          capabilities->SetSupportedChannelWidthSet (1);
        }
      else
        {
          capabilities->SetSupportedChannelWidthSet (0);
        }
      uint32_t maxMpduLength = std::max (std::max (m_beMaxAmsduSize, m_bkMaxAmsduSize), std::max (m_voMaxAmsduSize, m_viMaxAmsduSize)) + 56; //see section 9.11 of 11ac standard
      capabilities->SetMaxMpduLength (uint8_t (maxMpduLength > 3895) + uint8_t (maxMpduLength > 7991)); //0 if 3895, 1 if 7991, 2 for 11454
      capabilities->SetRxLdpc (m_phy->GetLdpc ());
      capabilities->SetShortGuardIntervalFor80Mhz ((m_phy->GetChannelWidth () == 80) && m_phy->GetShortGuardInterval ());
      capabilities->SetShortGuardIntervalFor160Mhz ((m_phy->GetChannelWidth () == 160) && m_phy->GetShortGuardInterval ());
      double maxAmpduLengthExponent = std::max (std::ceil ((std::log (std::max (std::max (m_beMaxAmpduSize, m_bkMaxAmpduSize), std::max (m_voMaxAmpduSize, m_viMaxAmpduSize)) + 1.0) / std::log (2.0)) - 13.0), 0.0);
      NS_ASSERT (maxAmpduLengthExponent >= 0 && maxAmpduLengthExponent <= 255);
      capabilities->SetMaxAmpduLengthExponent (std::max<uint8_t> (7, static_cast<uint8_t> (maxAmpduLengthExponent))); //0 to 7 for VHT
      uint8_t maxMcs = 0;
      for (uint8_t i = 0; i < m_phy->GetNMcs (); i++)
        {

              maxMcs = mcs.GetMcsValue ();
            }
        }
      //This is a workaround, a bug should be opened for this!
      //TODO: HE shall support VHT and HT rates!
      if (m_heSupported)
        {
          maxMcs = 9;
        }
      // Support same MaxMCS for each spatial stream
      for (uint8_t nss = 1; nss <= m_phy->GetMaxSupportedRxSpatialStreams (); nss++)
        {
          capabilities->SetRxMcsMap (maxMcs, nss);
        }
      for (uint8_t nss = 1; nss <= m_phy->GetMaxSupportedTxSpatialStreams (); nss++)
        {
          capabilities->SetTxMcsMap (maxMcs, nss);
        }
      uint64_t maxSupportedRateLGI = 0; //in bit/s
      for (uint8_t i = 0; i < m_phy->GetNMcs (); i++)

              NS_LOG_DEBUG ("Updating maxSupportedRateLGI to " << maxSupportedRateLGI);
            }
        }
      capabilities->SetRxHighestSupportedLgiDataRate (static_cast<uint16_t> (maxSupportedRateLGI / 1e6)); //in Mbit/s
      capabilities->SetTxHighestSupportedLgiDataRate (static_cast<uint16_t> (maxSupportedRateLGI / 1e6)); //in Mbit/s
      //To be filled in once supported
      capabilities->SetRxStbc (0);
      capabilities->SetTxStbc (0);
    }
  return capabilities;
}

Ptr<HeCapabilities>
RegularWifiMac::GetHeCapabilities (void) const
{
  NS_LOG_FUNCTION (this);
  Ptr<HeCapabilities> capabilities = Create<HeCapabilities> ();
  if (m_heSupported)
    {
      capabilities->SetHeSupported (1);
      uint8_t channelWidthSet = 0;
      if (m_phy->GetChannelWidth () >= 40 && Is2_4Ghz (m_phy->GetFrequency ()))
        {

        {
          channelWidthSet |= 0x04;
        }
      capabilities->SetChannelWidthSet (channelWidthSet);
      uint8_t gi = 0;
      if (m_phy->GetGuardInterval () <= NanoSeconds (1600))
        {

        {
          gi |= 0x02;
        }
      capabilities->SetHeLtfAndGiForHePpdus (gi);
      double maxAmpduLengthExponent = std::max (std::ceil ((std::log (std::max (std::max (m_beMaxAmpduSize, m_bkMaxAmpduSize), std::max (m_voMaxAmpduSize, m_viMaxAmpduSize))
                                                                      + 1.0)
                                                            / std::log (2.0))
                                                           - 13.0),
                                                0.0);
      NS_ASSERT (maxAmpduLengthExponent >= 0 && maxAmpduLengthExponent <= 255);
      capabilities->SetMaxAmpduLengthExponent (std::max<uint8_t> (7, static_cast<uint8_t> (maxAmpduLengthExponent))); //assume 0 to 7 for HE
      uint8_t maxMcs = 0;
      for (uint8_t i = 0; i < m_phy->GetNMcs (); i++)
        {

              maxMcs = mcs.GetMcsValue ();
            }
        }
      capabilities->SetHighestMcsSupported (maxMcs);
      capabilities->SetHighestNssSupported (m_phy->GetMaxSupportedTxSpatialStreams ());
    }
  return capabilities;
}




   *
   * \return the extended capabilities that we support
   */
  Ptr<ExtendedCapabilities> GetExtendedCapabilities (void) const;
  /**
   * Return the HT capabilities of the device.
   *
   * \return the HT capabilities that we support
   */
  Ptr<HtCapabilities> GetHtCapabilities (void) const;
  /**
   * Return the VHT capabilities of the device.
   *
   * \return the VHT capabilities that we support
   */
  Ptr<VhtCapabilities> GetVhtCapabilities (void) const;
  /**
   * Return the HE capabilities of the device.
   *
   * \return the HE capabilities that we support
   */
  Ptr<HeCapabilities> GetHeCapabilities (void) const;

  /**
   * This type defines the callback of a higher layer that a




  probe.SetSupportedRates (GetSupportedRates ());
  if (GetHtSupported () || GetVhtSupported () || GetHeSupported ())
    {
      probe.AddWifiInformationElement (GetExtendedCapabilities ());
      probe.AddWifiInformationElement (GetHtCapabilities ());
    }
  if (GetVhtSupported () || GetHeSupported ())
    {
      probe.AddWifiInformationElement (GetVhtCapabilities ());
    }
  if (GetHeSupported ())
    {
      probe.AddWifiInformationElement (GetHeCapabilities ());
    }
  packet->AddHeader (probe);


      assoc.SetListenInterval (0);
      if (GetHtSupported () || GetVhtSupported () || GetHeSupported ())
        {
          assoc.AddWifiInformationElement (GetExtendedCapabilities ());
          assoc.AddWifiInformationElement (GetHtCapabilities ());
        }
      if (GetVhtSupported () || GetHeSupported ())
        {
          assoc.AddWifiInformationElement (GetVhtCapabilities ());
        }
      if (GetHeSupported ())
        {
          assoc.AddWifiInformationElement (GetHeCapabilities ());
        }
      packet->AddHeader (assoc);
    }

      reassoc.SetListenInterval (0);
      if (GetHtSupported () || GetVhtSupported () || GetHeSupported ())
        {
          reassoc.AddWifiInformationElement (GetExtendedCapabilities ());
          reassoc.AddWifiInformationElement (GetHtCapabilities ());
        }
      if (GetVhtSupported () || GetHeSupported ())
        {
          reassoc.AddWifiInformationElement (GetVhtCapabilities ());
        }
      if (GetHeSupported ())
        {
          reassoc.AddWifiInformationElement (GetHeCapabilities ());
        }
      packet->AddHeader (reassoc);
    }

          NS_LOG_LOGIC ("Beacon is for our SSID");
          goodBeacon = true;
        }
      Ptr<CfParameterSet> cfParameterSet = StaticCast<CfParameterSet> (beacon.GetInformationElement (IE_CF_PARAMETER_SET));
      if (cfParameterSet != 0 && cfParameterSet->GetCFPCount () == 0)
        {
          //see section 9.3.2.2 802.11-1999
          if (GetPcfSupported ())
            {
              m_low->DoNavStartNow (MicroSeconds (cfParameterSet->GetCFPMaxDurationUs ()));
            }
          else
            {
              m_low->DoNavStartNow (MicroSeconds (cfParameterSet->GetCFPDurRemainingUs ()));
            }
        }
      SupportedRates rates = beacon.GetSupportedRates ();

  bool isShortPreambleEnabled = capabilities.IsShortPreamble ();
  if (GetErpSupported ())
    {
      Ptr<ErpInformation> erpInformation = StaticCast<ErpInformation> (beacon.GetInformationElement (IE_ERP_INFORMATION));
      if (erpInformation != 0)
      if (erpInformation.GetUseProtection () != 0)
        {
          isShortPreambleEnabled &= !erpInformation->GetBarkerPreambleMode ();
          if (erpInformation->GetUseProtection () != 0)
            {
              m_stationManager->SetUseNonErpProtection (true);
            }
          else
            {
              m_stationManager->SetUseNonErpProtection (false);
            }
        }
      if (capabilities.IsShortSlotTime () == true)
        {

  if (GetQosSupported ())
    {
      bool qosSupported = false;
      Ptr<EdcaParameterSet> edcaParameters = StaticCast<EdcaParameterSet> (beacon.GetInformationElement (IE_EDCA_PARAMETER_SET));
      if (edcaParameters != 0 && edcaParameters->IsQosSupported ())
        {
          qosSupported = true;
          //The value of the TXOP Limit field is specified as an unsigned integer, with the least significant octet transmitted first, in units of 32 μs.
          SetEdcaParameters (AC_BE, edcaParameters->GetBeCWmin (), edcaParameters->GetBeCWmax (), edcaParameters->GetBeAifsn (), 32 * MicroSeconds (edcaParameters->GetBeTXOPLimit ()));
          SetEdcaParameters (AC_BK, edcaParameters->GetBkCWmin (), edcaParameters->GetBkCWmax (), edcaParameters->GetBkAifsn (), 32 * MicroSeconds (edcaParameters->GetBkTXOPLimit ()));
          SetEdcaParameters (AC_VI, edcaParameters->GetViCWmin (), edcaParameters->GetViCWmax (), edcaParameters->GetViAifsn (), 32 * MicroSeconds (edcaParameters->GetViTXOPLimit ()));
          SetEdcaParameters (AC_VO, edcaParameters->GetVoCWmin (), edcaParameters->GetVoCWmax (), edcaParameters->GetVoAifsn (), 32 * MicroSeconds (edcaParameters->GetVoTXOPLimit ()));
        }
      m_stationManager->SetQosSupport (apAddr, qosSupported);
    }
  if (GetHtSupported ())
    {
      Ptr<HtCapabilities> htCapabilities = StaticCast<HtCapabilities> (beacon.GetInformationElement (IE_HT_CAPABILITIES));
      if (htCapabilities == 0)
        {
          m_stationManager->RemoveAllSupportedMcs (apAddr);
        }
      else
        {
          m_stationManager->AddStationHtCapabilities (apAddr, htCapabilities);
          Ptr<HtOperation> htOperation = StaticCast<HtOperation> (beacon.GetInformationElement (IE_HT_OPERATION));
          NS_ASSERT (htOperation != 0);
          if (htOperation->GetNonGfHtStasPresent ())
            {
              m_stationManager->SetUseGreenfieldProtection (true);
            }

            {
              m_stationManager->SetUseGreenfieldProtection (false);
            }
          if (!GetVhtSupported () && GetRifsSupported () && htOperation->GetRifsMode ())
            {
              m_stationManager->SetRifsPermitted (true);
            }

    }
  if (GetVhtSupported ())
    {
      Ptr<VhtCapabilities> vhtCapabilities = StaticCast<VhtCapabilities> (beacon.GetInformationElement (IE_VHT_CAPABILITIES));
      //we will always fill in RxHighestSupportedLgiDataRate field at TX, so this can be used to check whether it supports VHT
      if (vhtCapabilities != 0)
        {
          m_stationManager->AddStationVhtCapabilities (apAddr, vhtCapabilities);
          Ptr<VhtOperation> vhtOperation = StaticCast<VhtOperation> (beacon.GetInformationElement (IE_VHT_OPERATION));
          NS_ASSERT (vhtOperation != 0);
          for (uint8_t i = 0; i < m_phy->GetNMcs (); i++)
            {
              WifiMode mcs = m_phy->GetMcs (i);
              if (mcs.GetModulationClass () == WIFI_MOD_CLASS_VHT && vhtCapabilities->IsSupportedRxMcs (mcs.GetMcsValue ()))
                {
                  m_stationManager->AddSupportedMcs (apAddr, mcs);
                }

    }
  if (GetHtSupported () || GetVhtSupported ())
    {
      Ptr<ExtendedCapabilities> extendedCapabilities = StaticCast<ExtendedCapabilities> (beacon.GetInformationElement (IE_EXTENDED_CAPABILITIES));
      //TODO: to be completed
    }
  if (GetHeSupported ())
    {
      Ptr<HeCapabilities> heCapabilities = StaticCast<HeCapabilities> (beacon.GetInformationElement (IE_HE_CAPABILITIES));
      if (heCapabilities != 0)
      m_stationManager->AddStationHeCapabilities (apAddr, heCapabilities);
      HeOperation heOperation = beacon.GetHeOperation ();
      for (uint8_t i = 0; i < m_phy->GetNMcs (); i++)
        {
          m_stationManager->AddStationHeCapabilities (apAddr, heCapabilities);
          Ptr<HeOperation> heOperation = StaticCast<HeOperation> (beacon.GetInformationElement (IE_HE_OPERATION));
          NS_ASSERT (heOperation != 0);
          for (uint8_t i = 0; i < m_phy->GetNMcs (); i++)
            {
              WifiMode mcs = m_phy->GetMcs (i);
              if (mcs.GetModulationClass () == WIFI_MOD_CLASS_HE && heCapabilities->IsSupportedRxMcs (mcs.GetMcsValue ()))
                {
                  m_stationManager->AddSupportedMcs (apAddr, mcs);
                }
            }
        }
    }

        }
      else
        {
          Ptr<ErpInformation> erpInformation = StaticCast<ErpInformation> (probeResp.GetInformationElement (IE_ERP_INFORMATION));
          if (erpInformation != 0)
            {
              isShortPreambleEnabled &= !erpInformation->GetBarkerPreambleMode ();
            }
          if (m_stationManager->GetShortSlotTimeEnabled ())
            {
              //enable short slot time

        }
      else
        {
          Ptr<ErpInformation> erpInformation = StaticCast<ErpInformation> (assocResp.GetInformationElement (IE_ERP_INFORMATION));
          if (erpInformation != 0)
            {
              isShortPreambleEnabled &= !erpInformation->GetBarkerPreambleMode ();
            }
          if (m_stationManager->GetShortSlotTimeEnabled ())
            {
              //enable short slot time

  if (GetQosSupported ())
    {
      bool qosSupported = false;
      Ptr<EdcaParameterSet> edcaParameters = StaticCast<EdcaParameterSet> (assocResp.GetInformationElement (IE_EDCA_PARAMETER_SET));
      if (edcaParameters != 0 && edcaParameters->IsQosSupported ())
        {
          qosSupported = true;
          //The value of the TXOP Limit field is specified as an unsigned integer, with the least significant octet transmitted first, in units of 32 μs.
          SetEdcaParameters (AC_BE, edcaParameters->GetBeCWmin (), edcaParameters->GetBeCWmax (), edcaParameters->GetBeAifsn (), 32 * MicroSeconds (edcaParameters->GetBeTXOPLimit ()));
          SetEdcaParameters (AC_BK, edcaParameters->GetBkCWmin (), edcaParameters->GetBkCWmax (), edcaParameters->GetBkAifsn (), 32 * MicroSeconds (edcaParameters->GetBkTXOPLimit ()));
          SetEdcaParameters (AC_VI, edcaParameters->GetViCWmin (), edcaParameters->GetViCWmax (), edcaParameters->GetViAifsn (), 32 * MicroSeconds (edcaParameters->GetViTXOPLimit ()));
          SetEdcaParameters (AC_VO, edcaParameters->GetVoCWmin (), edcaParameters->GetVoCWmax (), edcaParameters->GetVoAifsn (), 32 * MicroSeconds (edcaParameters->GetVoTXOPLimit ()));
        }
      m_stationManager->SetQosSupport (apAddr, qosSupported);
    }
  if (GetHtSupported ())
    {
      Ptr<HtCapabilities> htCapabilities = StaticCast<HtCapabilities> (assocResp.GetInformationElement (IE_HT_CAPABILITIES));
      if (htCapabilities == 0)
        {
          m_stationManager->RemoveAllSupportedMcs (apAddr);
        }
      else
        {
          m_stationManager->AddStationHtCapabilities (apAddr, htCapabilities);
          Ptr<HtOperation> htOperation = StaticCast<HtOperation> (assocResp.GetInformationElement (IE_HT_OPERATION));
          NS_ASSERT (htOperation != 0);
          if (htOperation->GetNonGfHtStasPresent ())
            {
              m_stationManager->SetUseGreenfieldProtection (true);
            }

            {
              m_stationManager->SetUseGreenfieldProtection (false);
            }
          if (!GetVhtSupported () && GetRifsSupported () && htOperation->GetRifsMode ())
            {
              m_stationManager->SetRifsPermitted (true);
            }

    }
  if (GetVhtSupported ())
    {
      Ptr<VhtCapabilities> vhtCapabilities = StaticCast<VhtCapabilities> (assocResp.GetInformationElement (IE_VHT_CAPABILITIES));
      //we will always fill in RxHighestSupportedLgiDataRate field at TX, so this can be used to check whether it supports VHT
      if (vhtCapabilities != 0)
        {
          m_stationManager->AddStationVhtCapabilities (apAddr, vhtCapabilities);
          Ptr<VhtOperation> vhtOperation = StaticCast<VhtOperation> (assocResp.GetInformationElement (IE_VHT_OPERATION));
        }
    }
  if (GetHeSupported ())
    {
      Ptr<HeCapabilities> hecapabilities = StaticCast<HeCapabilities> (assocResp.GetInformationElement (IE_HE_CAPABILITIES));
      //todo: once we support non constant rate managers, we should add checks here whether HE is supported by the peer
      m_stationManager->AddStationHeCapabilities (apAddr, hecapabilities);
      Ptr<HeOperation> heOperation = StaticCast<HeOperation> (assocResp.GetInformationElement (IE_HE_OPERATION));
    }
  for (uint8_t i = 0; i < m_phy->GetNModes (); i++)
    {

    }
  if (GetHtSupported ())
    {
      Ptr<HtCapabilities> htCapabilities = StaticCast<HtCapabilities> (assocResp.GetInformationElement (IE_HT_CAPABILITIES));
      if (htCapabilities != 0)
        {
          for (uint8_t i = 0; i < m_phy->GetNMcs (); i++)
          if (mcs.GetModulationClass () == WIFI_MOD_CLASS_HT && htCapabilities.IsSupportedMcs (mcs.GetMcsValue ()))
            {
              WifiMode mcs = m_phy->GetMcs (i);
              if (mcs.GetModulationClass () == WIFI_MOD_CLASS_HT && htCapabilities->IsSupportedMcs (mcs.GetMcsValue ()))
                {
                  m_stationManager->AddSupportedMcs (apAddr, mcs);
                  //here should add a control to add basic MCS when it is implemented
                }
            }
        }
    }
  if (GetVhtSupported ())
    {
      Ptr<VhtCapabilities> vhtcapabilities = StaticCast<VhtCapabilities> (assocResp.GetInformationElement (IE_VHT_CAPABILITIES));
      if (vhtcapabilities != 0)
        {
          for (uint8_t i = 0; i < m_phy->GetNMcs (); i++)
          if (mcs.GetModulationClass () == WIFI_MOD_CLASS_VHT && vhtcapabilities.IsSupportedRxMcs (mcs.GetMcsValue ()))
            {
              WifiMode mcs = m_phy->GetMcs (i);
              if (mcs.GetModulationClass () == WIFI_MOD_CLASS_VHT && vhtcapabilities->IsSupportedRxMcs (mcs.GetMcsValue ()))
                {
                  m_stationManager->AddSupportedMcs (apAddr, mcs);
                  //here should add a control to add basic MCS when it is implemented
                }
            }
        }
    }
  if (GetHtSupported () || GetVhtSupported ())
    {
      Ptr<ExtendedCapabilities> extendedCapabilities = StaticCast<ExtendedCapabilities> (assocResp.GetInformationElement (IE_EXTENDED_CAPABILITIES));
      //TODO: to be completed
    }
  if (GetHeSupported ())
    {
      Ptr<HeCapabilities> heCapabilities = StaticCast<HeCapabilities> (assocResp.GetInformationElement (IE_HE_CAPABILITIES));
      if (heCapabilities != 0)
        {
          for (uint8_t i = 0; i < m_phy->GetNMcs (); i++)
          if (mcs.GetModulationClass () == WIFI_MOD_CLASS_HE && heCapabilities.IsSupportedRxMcs (mcs.GetMcsValue ()))
            {
              WifiMode mcs = m_phy->GetMcs (i);
              if (mcs.GetModulationClass () == WIFI_MOD_CLASS_HE && heCapabilities->IsSupportedRxMcs (mcs.GetMcsValue ()))
                {
                  m_stationManager->AddSupportedMcs (apAddr, mcs);
                  //here should add a control to add basic MCS when it is implemented
                }
            }
        }
    }




  return m_rxHighestSupportedLongGuardIntervalDataRate;
}

void
VhtCapabilities::Print (std::ostream& os) const
{
  os << GetVhtCapabilitiesInfo ()
     << "|" << GetSupportedMcsAndNssSet ();
}

/**
 * output stream output operator
 *

std::ostream &
operator << (std::ostream &os, const VhtCapabilities &VhtCapabilities)
{
  VhtCapabilities.Print (os);

  return os;
}


(-)a/src/wifi/model/vht-capabilities.h (+7 lines)

Lines 240-245	Link Here

240

*/

240

*/

241

  uint16_t GetSerializedSize () const;

241

  uint16_t GetSerializedSize () const;

242

243

/**

244

   * Generate human-readable form of VHT capabilities information element

245

246

   * \param os output stream

247

*/

248

  void Print (std::ostream& os) const;

249

243

250

244

private:

251

private:

245

  //Capabilities Info fields

252

  //Capabilities Info fields




  return length;
}

void
VhtOperation::Print (std::ostream& os) const
{
  os << +GetChannelWidth () << "|"
     << +GetChannelCenterFrequencySegment0 () << "|"
     << +GetChannelCenterFrequencySegment1 () << "|"
     << GetBasicVhtMcsAndNssSet ();
}

/**
 * output stream output operator
 *

std::ostream &
operator << (std::ostream &os, const VhtOperation &VhtOperation)
{
  VhtOperation.Print (os);
     << +VhtOperation.GetChannelCenterFrequencySegment0 () << "|"
     << +VhtOperation.GetChannelCenterFrequencySegment1 () << "|"
     << VhtOperation.GetBasicVhtMcsAndNssSet ();
  return os;
}


(-)a/src/wifi/model/vht-operation.h (+7 lines)

Lines 147-152	Link Here

147

*/

147

*/

148

  uint16_t GetSerializedSize () const;

148

  uint16_t GetSerializedSize () const;

149

150

/**

151

   * Generate human-readable form of VHT Operations IE

152

153

   * \param os output stream

154

*/

155

  void Print (std::ostream& os) const;

156

150

157

151

private:

158

private:

152

  //VHT Operation Information

159

  //VHT Operation Information




NS_OBJECT_ENSURE_REGISTERED (WifiInformationElementVector);

WifiInformationElementVector::WifiInformationElementVector ()
  : m_maxSize (1500)
{
}


      NS_FATAL_ERROR ("Information element " << +id << " is not implemented");
      return 0;
    }
  /*  unreachable:  b/c switch is guaranteed to return from this function
  if (GetSize () + length > m_maxSize)
    {
      NS_FATAL_ERROR ("Check max size for information element!");
    }
  newElement->DeserializeInformationField (i, length);
  i.Next (length);
  m_elements.push_back (newElement);
  return i.GetDistanceFrom (start);
  */
}

void

  return m_elements.end ();
}

void
WifiInformationElementVector::AddInformationElement (Ptr<WifiInformationElement> element)
{
  if (element->GetInformationFieldSize () + 2 + GetSize () > m_maxSize)
    {
      return false;
    }
  m_elements.push_back (element);
  return true;
}

Ptr<WifiInformationElement>




   */
  Iterator End ();
  /**
   * add an IE
   *
   * \param element wifi information element to add
   * \returns true is added
   */
  void AddInformationElement (Ptr<WifiInformationElement> element);
  /**
   * vector of pointers to information elements is the body of IeVector
   *

   */
  uint32_t GetSize () const;
  IE_VECTOR m_elements; //!< Information element vector
  /// Size in bytes (actually, max packet length)
  uint16_t m_maxSize;
};

} //namespace ns3

(-)a/src/wifi/model/wifi-information-element.cc (+8 lines)

Lines 84-89	Link Here

  return i;

  return i;

Buffer::Iterator

WifiInformationElement::DeserializeElementBody (Buffer::Iterator i, uint8_t length)

  DeserializeInformationField (i, length);

  i.Next (length);

  return i;

bool

bool

WifiInformationElement::operator== (WifiInformationElement const & a) const

WifiInformationElement::operator== (WifiInformationElement const & a) const




// 200 to 220 are reserved
#define IE_VENDOR_SPECIFIC                      ((WifiInformationElementId)221)
// 222 to 255 are reserved
#define IE_HE_CAPABILITIES                      ((WifiInformationElementId)254) //todo: not defined yet in the standard!
#define IE_HE_OPERATION                         ((WifiInformationElementId)255) //todo: not defined yet in the standard!



   */
  Buffer::Iterator DeserializeIfPresent (Buffer::Iterator i);
  /**
   * Deserialize IE Body. The iterator passed in must be pointing at
   * the Element Body of an information element.
   *
   * \param i an iterator which points to where the IE body should be read.
   *
   * \return an iterator
   */
  Buffer::Iterator DeserializeElementBody (Buffer::Iterator i, uint8_t length);
  /**
   * Get the size of the serialized IE including Element ID and
   * length fields.
   *




}

void
WifiRemoteStationManager::AddStationHtCapabilities (Mac48Address from, Ptr<HtCapabilities> htCapabilities)
{
  //Used by all stations to record HT capabilities of remote stations
  NS_LOG_FUNCTION (this << from << htCapabilities);
  WifiRemoteStationState *state;
  state = LookupState (from);
  state->m_shortGuardInterval = htCapabilities->GetShortGuardInterval20 ();
  if (htCapabilities->GetSupportedChannelWidth () == 1)
    {
      state->m_channelWidth = 40;
    }

      state->m_channelWidth = 20;
    }
  SetQosSupport (from, true);
  state->m_greenfield = htCapabilities->GetGreenfield ();
  state->m_streams = htCapabilities->GetRxHighestSupportedAntennas ();
  for (uint8_t j = 0; j < m_wifiPhy->GetNMcs (); j++)
    {
      WifiMode mcs = m_wifiPhy->GetMcs (j);
      if (mcs.GetModulationClass () == WIFI_MOD_CLASS_HT && htCapabilities->IsSupportedMcs (mcs.GetMcsValue ()))
        {
          AddSupportedMcs (from, mcs);
        }

}

void
WifiRemoteStationManager::AddStationVhtCapabilities (Mac48Address from, Ptr<VhtCapabilities> vhtCapabilities)
{
  //Used by all stations to record VHT capabilities of remote stations
  NS_LOG_FUNCTION (this << from << vhtCapabilities);
  WifiRemoteStationState *state;
  state = LookupState (from);
  if (vhtCapabilities->GetSupportedChannelWidthSet () == 1)
    {
      state->m_channelWidth = 160;
    }

    {
      state->m_channelWidth = m_wifiPhy->GetChannelWidth ();
    }
  state->m_ldpc = (vhtCapabilities->GetRxLdpc () != 0);
  state->m_stbc = (vhtCapabilities->GetTxStbc () != 0 || vhtCapabilities->GetRxStbc () != 0);
  for (uint8_t i = 1; i <= m_wifiPhy->GetMaxSupportedTxSpatialStreams (); i++)
    {
      for (uint8_t j = 0; j < m_wifiPhy->GetNMcs (); j++)
        {
          WifiMode mcs = m_wifiPhy->GetMcs (j);
          if (mcs.GetModulationClass () == WIFI_MOD_CLASS_VHT && vhtCapabilities->IsSupportedMcs (mcs.GetMcsValue (), i))
            {
              AddSupportedMcs (from, mcs);
            }

}

void
WifiRemoteStationManager::AddStationHeCapabilities (Mac48Address from, Ptr<HeCapabilities> heCapabilities)
{
  //Used by all stations to record HE capabilities of remote stations
  NS_LOG_FUNCTION (this << from << heCapabilities);

  state = LookupState (from);
  if (Is5Ghz (m_wifiPhy->GetFrequency ()))
    {
      if (heCapabilities->GetChannelWidthSet () & 0x04)
        {
          state->m_channelWidth = 160;
        }
      else if (heCapabilities->GetChannelWidthSet () & 0x02)
        {
          state->m_channelWidth = 80;
        }

    }
  else if (Is2_4Ghz (m_wifiPhy->GetFrequency ()))
    {
      if (heCapabilities->GetChannelWidthSet () & 0x01)
        {
          state->m_channelWidth = 40;
        }

          state->m_channelWidth = 20;
        }
    }
  if (heCapabilities->GetHeLtfAndGiForHePpdus () >= 2)
    {
      state->m_guardInterval = 800;
    }
  else if (heCapabilities->GetHeLtfAndGiForHePpdus () == 1)
    {
      state->m_guardInterval = 1600;
    }

        {
          WifiMode mcs = m_wifiPhy->GetMcs (j);
          if (mcs.GetModulationClass () == WIFI_MOD_CLASS_HE
              && heCapabilities->GetHighestNssSupported () >= i
              && heCapabilities->GetHighestMcsSupported () >= j)
            {
              AddSupportedMcs (from, mcs);
            }




   * \param from the address of the station being recorded
   * \param htcapabilities the HT capabilities of the station
   */
  void AddStationHtCapabilities (Mac48Address from, Ptr<HtCapabilities> htcapabilities);
  /**
   * Records VHT capabilities of the remote station.
   *
   * \param from the address of the station being recorded
   * \param vhtcapabilities the VHT capabilities of the station
   */
  void AddStationVhtCapabilities (Mac48Address from, Ptr<VhtCapabilities> vhtcapabilities);
  /**
   * Records HE capabilities of the remote station.
   *
   * \param from the address of the station being recorded
   * \param hecapabilities the HE capabilities of the station
   */
  void AddStationHeCapabilities (Mac48Address from, Ptr<HeCapabilities> hecapabilities);
  /**
   * Enable or disable HT capability support.
   *

(-)a/src/wifi/test/wifi-test.cc (-2 / +2 lines)

Lines 1422-1429	Link Here

1422

1423

      MgtBeaconHeader beacon;

1423

      MgtBeaconHeader beacon;

1424

      packet->RemoveHeader (beacon);

1424

      packet->RemoveHeader (beacon);

1425

      HtOperation htOperation = beacon.GetHtOperation ();

1425

      Ptr<HtOperation> htOperation = StaticCast<HtOperation> (beacon.GetInformationElement (IE_HT_OPERATION));

1426

      if (htOperation.GetStaChannelWidth () > 0)

1426

      if (htOperation->GetStaChannelWidth () > 0)

1427

1428

          m_countOperationalChannelWidth40++;

1428

          m_countOperationalChannelWidth40++;

1429

Return to bug 2945