ctrl-headers.cc

Go to the documentation of this file.

/* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */
/*
 * Copyright (c) 2009 MIRKO BANCHI
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License version 2 as
 * published by the Free Software Foundation;
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 *
 * Author: Mirko Banchi <mk.banchi@gmail.com>
 */
 
#include "ctrl-headers.h"
#include "wifi-tx-vector.h"
#include "ns3/he-phy.h"
#include "wifi-utils.h"
#include "ns3/address-utils.h"
#include <algorithm>
 
namespace ns3 {
 
 
/***********************************
 *       Block ack request
 ***********************************/
 
NS_OBJECT_ENSURE_REGISTERED (CtrlBAckRequestHeader);
 
CtrlBAckRequestHeader::CtrlBAckRequestHeader ()
  : m_barAckPolicy (false),
    m_barType (BlockAckReqType::BASIC)
{
}
 
CtrlBAckRequestHeader::~CtrlBAckRequestHeader ()
{
}
 
TypeId
CtrlBAckRequestHeader::GetTypeId (void)
{
  static TypeId tid = TypeId ("ns3::CtrlBAckRequestHeader")
    .SetParent<Header> ()
    .SetGroupName ("Wifi")
    .AddConstructor<CtrlBAckRequestHeader> ()
  ;
  return tid;
}
 
TypeId
CtrlBAckRequestHeader::GetInstanceTypeId (void) const
{
  return GetTypeId ();
}
 
void
CtrlBAckRequestHeader::Print (std::ostream &os) const
{
  os << "TID_INFO=" << m_tidInfo << ", StartingSeq=" << std::hex << m_startingSeq << std::dec;
}
 
uint32_t
CtrlBAckRequestHeader::GetSerializedSize () const
{
  uint32_t size = 0;
  size += 2; //Bar control
  switch (m_barType.m_variant)
    {
      case BlockAckReqType::BASIC:
      case BlockAckReqType::COMPRESSED:
      case BlockAckReqType::EXTENDED_COMPRESSED:
        size += 2;
        break;
      case BlockAckReqType::MULTI_TID:
        size += (2 + 2) * (m_tidInfo + 1);
        break;
      default:
        NS_FATAL_ERROR ("Invalid BA type");
        break;
    }
  return size;
}
 
void
CtrlBAckRequestHeader::Serialize (Buffer::Iterator start) const
{
  Buffer::Iterator i = start;
  i.WriteHtolsbU16 (GetBarControl ());
  switch (m_barType.m_variant)
    {
      case BlockAckReqType::BASIC:
      case BlockAckReqType::COMPRESSED:
      case BlockAckReqType::EXTENDED_COMPRESSED:
        i.WriteHtolsbU16 (GetStartingSequenceControl ());
        break;
      case BlockAckReqType::MULTI_TID:
        NS_FATAL_ERROR ("Multi-tid block ack is not supported.");
        break;
      default:
        NS_FATAL_ERROR ("Invalid BA type");
        break;
    }
}
 
uint32_t
CtrlBAckRequestHeader::Deserialize (Buffer::Iterator start)
{
  Buffer::Iterator i = start;
  SetBarControl (i.ReadLsbtohU16 ());
  switch (m_barType.m_variant)
    {
      case BlockAckReqType::BASIC:
      case BlockAckReqType::COMPRESSED:
      case BlockAckReqType::EXTENDED_COMPRESSED:
        SetStartingSequenceControl (i.ReadLsbtohU16 ());
        break;
      case BlockAckReqType::MULTI_TID:
        NS_FATAL_ERROR ("Multi-tid block ack is not supported.");
        break;
      default:
        NS_FATAL_ERROR ("Invalid BA type");
        break;
    }
  return i.GetDistanceFrom (start);
}
 
uint16_t
CtrlBAckRequestHeader::GetBarControl (void) const
{
  uint16_t res = 0;
  switch (m_barType.m_variant)
    {
      case BlockAckReqType::BASIC:
        break;
      case BlockAckReqType::COMPRESSED:
        res |= (0x02 << 1);
        break;
      case BlockAckReqType::EXTENDED_COMPRESSED:
        res |= (0x01 << 1);
        break;
      case BlockAckReqType::MULTI_TID:
        res |= (0x03 << 1);
        break;
      default:
        NS_FATAL_ERROR ("Invalid BA type");
        break;
    }
  res |= (m_tidInfo << 12) & (0xf << 12);
  return res;
}
 
void
CtrlBAckRequestHeader::SetBarControl (uint16_t bar)
{
  m_barAckPolicy = ((bar & 0x01) == 1) ? true : false;
  if (((bar >> 1) & 0x0f) == 0x03)
    {
      m_barType.m_variant = BlockAckReqType::MULTI_TID;
    }
  else if (((bar >> 1) & 0x0f) == 0x01)
    {
      m_barType.m_variant = BlockAckReqType::EXTENDED_COMPRESSED;
    }
  else if (((bar >> 1) & 0x0f) == 0x02)
    {
      m_barType.m_variant = BlockAckReqType::COMPRESSED;
    }
  else
    {
      m_barType.m_variant = BlockAckReqType::BASIC;
    }
  m_tidInfo = (bar >> 12) & 0x0f;
}
 
uint16_t
CtrlBAckRequestHeader::GetStartingSequenceControl (void) const
{
  return (m_startingSeq << 4) & 0xfff0;
}
 
void
CtrlBAckRequestHeader::SetStartingSequenceControl (uint16_t seqControl)
{
  m_startingSeq = (seqControl >> 4) & 0x0fff;
}
 
void
CtrlBAckRequestHeader::SetHtImmediateAck (bool immediateAck)
{
  m_barAckPolicy = immediateAck;
}
 
void
CtrlBAckRequestHeader::SetType (BlockAckReqType type)
{
  m_barType = type;
}
 
BlockAckReqType
CtrlBAckRequestHeader::GetType (void) const
{
  return m_barType;
}
 
void
CtrlBAckRequestHeader::SetTidInfo (uint8_t tid)
{
  m_tidInfo = static_cast<uint16_t> (tid);
}
 
void
CtrlBAckRequestHeader::SetStartingSequence (uint16_t seq)
{
  m_startingSeq = seq;
}
 
bool
CtrlBAckRequestHeader::MustSendHtImmediateAck (void) const
{
  return m_barAckPolicy;
}
 
uint8_t
CtrlBAckRequestHeader::GetTidInfo (void) const
{
  uint8_t tid = static_cast<uint8_t> (m_tidInfo);
  return tid;
}
 
uint16_t
CtrlBAckRequestHeader::GetStartingSequence (void) const
{
  return m_startingSeq;
}
 
bool
CtrlBAckRequestHeader::IsBasic (void) const
{
  return (m_barType.m_variant == BlockAckReqType::BASIC) ? true : false;
}
 
bool
CtrlBAckRequestHeader::IsCompressed (void) const
{
  return (m_barType.m_variant == BlockAckReqType::COMPRESSED) ? true : false;
}
 
bool
CtrlBAckRequestHeader::IsExtendedCompressed (void) const
{
  return (m_barType.m_variant == BlockAckReqType::EXTENDED_COMPRESSED) ? true : false;
}
 
bool
CtrlBAckRequestHeader::IsMultiTid (void) const
{
  return (m_barType.m_variant == BlockAckReqType::MULTI_TID) ? true : false;
}
 
 
/***********************************
 *       Block ack response
 ***********************************/
 
NS_OBJECT_ENSURE_REGISTERED (CtrlBAckResponseHeader);
 
CtrlBAckResponseHeader::CtrlBAckResponseHeader ()
  : m_baAckPolicy (false),
    m_tidInfo (0)
{
  SetType (BlockAckType::BASIC);
}
 
CtrlBAckResponseHeader::~CtrlBAckResponseHeader ()
{
}
 
TypeId
CtrlBAckResponseHeader::GetTypeId (void)
{
  static TypeId tid = TypeId ("ns3::CtrlBAckResponseHeader")
    .SetParent<Header> ()
    .SetGroupName ("Wifi")
    .AddConstructor<CtrlBAckResponseHeader> ()
  ;
  return tid;
}
 
TypeId
CtrlBAckResponseHeader::GetInstanceTypeId (void) const
{
  return GetTypeId ();
}
 
void
CtrlBAckResponseHeader::Print (std::ostream &os) const
{
  if (m_baType.m_variant != BlockAckType::MULTI_STA)
    {
      os << "TID_INFO=" << m_tidInfo << ", StartingSeq=0x" << std::hex << m_baInfo[0].m_startingSeq << std::dec;
    }
  else
    {
      for (std::size_t i = 0; i < m_baInfo.size (); i++)
        {
          os << "{AID=" << GetAid11 (i)
             << ", TID=" << GetTidInfo (i)
             << ", StartingSeq=0x" << std::hex << m_baInfo[i].m_startingSeq << std::dec << "}";
        }
    }
}
 
uint32_t
CtrlBAckResponseHeader::GetSerializedSize (void) const
{
  // This method only makes use of the configured BA type, so that functions like
  // GetBlockAckSize () can easily return the size of a Block Ack of a given type
  uint32_t size = 0;
  size += 2; //BA control
  switch (m_baType.m_variant)
    {
      case BlockAckType::BASIC:
      case BlockAckType::COMPRESSED:
      case BlockAckType::EXTENDED_COMPRESSED:
        size += (2 + m_baType.m_bitmapLen[0]);
        break;
      case BlockAckType::MULTI_TID:
        size += (2 + 2 + 8) * (m_tidInfo + 1); //Multi-TID block ack
        break;
      case BlockAckType::MULTI_STA:
        for (auto& bitmapLen : m_baType.m_bitmapLen)
          {
            size += 2 /* AID TID Info */ + (bitmapLen > 0 ? 2 : 0) /* BA SSC */ + bitmapLen;
          }
        break;
      default:
        NS_FATAL_ERROR ("Invalid BA type");
        break;
    }
  return size;
}
 
void
CtrlBAckResponseHeader::Serialize (Buffer::Iterator start) const
{
  Buffer::Iterator i = start;
  i.WriteHtolsbU16 (GetBaControl ());
  switch (m_baType.m_variant)
    {
      case BlockAckType::BASIC:
      case BlockAckType::COMPRESSED:
      case BlockAckType::EXTENDED_COMPRESSED:
        i.WriteHtolsbU16 (GetStartingSequenceControl ());
        i = SerializeBitmap (i);
        break;
      case BlockAckType::MULTI_STA:
        for (std::size_t index = 0; index < m_baInfo.size (); index++)
          {
            i.WriteHtolsbU16 (m_baInfo[index].m_aidTidInfo);
            if (GetAid11 (index) != 2045)
              {
                if (m_baInfo[index].m_bitmap.size () > 0)
                  {
                    i.WriteHtolsbU16 (GetStartingSequenceControl (index));
                    i = SerializeBitmap (i, index);
                  }
              }
            else
              {
                uint32_t reserved = 0;
                i.WriteHtolsbU32 (reserved);
                WriteTo (i, m_baInfo[index].m_ra);
              }
          }
        break;
      case BlockAckType::MULTI_TID:
        NS_FATAL_ERROR ("Multi-tid block ack is not supported.");
        break;
      default:
        NS_FATAL_ERROR ("Invalid BA type");
        break;
    }
}
 
uint32_t
CtrlBAckResponseHeader::Deserialize (Buffer::Iterator start)
{
  Buffer::Iterator i = start;
  SetBaControl (i.ReadLsbtohU16 ());
  switch (m_baType.m_variant)
    {
      case BlockAckType::BASIC:
      case BlockAckType::COMPRESSED:
      case BlockAckType::EXTENDED_COMPRESSED:
        SetStartingSequenceControl (i.ReadLsbtohU16 ());
        i = DeserializeBitmap (i);
        break;
      case BlockAckType::MULTI_STA:
        {
          std::size_t index = 0;
          while (i.GetRemainingSize () > 0)
            {
              m_baInfo.push_back (BaInfoInstance ());
              m_baType.m_bitmapLen.push_back (0);  // updated by next call to SetStartingSequenceControl
 
              m_baInfo.back ().m_aidTidInfo = i.ReadLsbtohU16 ();
 
              if (GetAid11 (index) != 2045)
                {
                  // the Block Ack Starting Sequence Control and Block Ack Bitmap subfields
                  // are only present in Block acknowledgement context, i.e., if the Ack Type
                  // subfield is set to 0 and the TID subfield is set to a value from 0 to 7.
                  if (!GetAckType (index) && GetTidInfo (index) < 8)
                    {
                      SetStartingSequenceControl (i.ReadLsbtohU16 (), index);
                      i = DeserializeBitmap (i, index);
                    }
                }
              else
                {
                  i.ReadLsbtohU32 ();  // next 4 bytes are reserved
                  ReadFrom (i, m_baInfo.back ().m_ra);
                  // the length of this Per AID TID Info subfield is 12, so set
                  // the bitmap length to 8 to simulate the correct size
                  m_baType.m_bitmapLen.back () = 8;
                }
              index++;
            }
        }
        break;
      case BlockAckType::MULTI_TID:
        NS_FATAL_ERROR ("Multi-tid block ack is not supported.");
        break;
      default:
        NS_FATAL_ERROR ("Invalid BA type");
        break;
    }
  return i.GetDistanceFrom (start);
}
 
void
CtrlBAckResponseHeader::SetHtImmediateAck (bool immediateAck)
{
  m_baAckPolicy = immediateAck;
}
 
void
CtrlBAckResponseHeader::SetType (BlockAckType type)
{
  m_baType = type;
  m_baInfo.clear ();
 
  for (auto& bitmapLen : m_baType.m_bitmapLen)
    {
      m_baInfo.push_back ({.m_aidTidInfo = 0,
                           .m_startingSeq = 0,
                           .m_bitmap = std::vector<uint8_t> (bitmapLen, 0),
                           .m_ra = Mac48Address ()});
    }
}
 
BlockAckType
CtrlBAckResponseHeader::GetType (void) const
{
  return m_baType;
}
 
void
CtrlBAckResponseHeader::SetTidInfo (uint8_t tid, std::size_t index)
{
  NS_ASSERT_MSG (m_baType.m_variant == BlockAckType::MULTI_STA || index == 0,
                 "index can only be non null for Multi-STA Block Ack");
  NS_ASSERT (index < m_baInfo.size ());
 
  if (m_baType.m_variant != BlockAckType::MULTI_STA)
    {
      m_tidInfo = static_cast<uint16_t> (tid);
    }
  else
    {
      m_baInfo[index].m_aidTidInfo |= ((static_cast<uint16_t> (tid) & 0x000f) << 12);
    }
}
 
void
CtrlBAckResponseHeader::SetStartingSequence (uint16_t seq, std::size_t index)
{
  NS_ASSERT_MSG (m_baType.m_variant == BlockAckType::MULTI_STA || index == 0,
                 "index can only be non null for Multi-STA Block Ack");
  NS_ASSERT (index < m_baInfo.size ());
 
  m_baInfo[index].m_startingSeq = seq;
}
 
bool
CtrlBAckResponseHeader::MustSendHtImmediateAck (void) const
{
  return (m_baAckPolicy) ? true : false;
}
 
uint8_t
CtrlBAckResponseHeader::GetTidInfo (std::size_t index) const
{
  NS_ASSERT_MSG (m_baType.m_variant == BlockAckType::MULTI_STA || index == 0,
                 "index can only be non null for Multi-STA Block Ack");
  NS_ASSERT (index < m_baInfo.size ());
 
  uint8_t tid = 0;
 
  if (m_baType.m_variant != BlockAckType::MULTI_STA)
    {
      tid = static_cast<uint8_t> (m_tidInfo);
    }
  else
    {
      tid = static_cast<uint8_t> ((m_baInfo[index].m_aidTidInfo >> 12) & 0x000f);
    }
  return tid;
}
 
uint16_t
CtrlBAckResponseHeader::GetStartingSequence (std::size_t index) const
{
  NS_ASSERT_MSG (m_baType.m_variant == BlockAckType::MULTI_STA || index == 0,
                 "index can only be non null for Multi-STA Block Ack");
  NS_ASSERT (index < m_baInfo.size ());
 
  return m_baInfo[index].m_startingSeq;
}
 
bool
CtrlBAckResponseHeader::IsBasic (void) const
{
  return (m_baType.m_variant == BlockAckType::BASIC) ? true : false;
}
 
bool
CtrlBAckResponseHeader::IsCompressed (void) const
{
  return (m_baType.m_variant == BlockAckType::COMPRESSED) ? true : false;
}
 
bool
CtrlBAckResponseHeader::IsExtendedCompressed (void) const
{
  return (m_baType.m_variant == BlockAckType::EXTENDED_COMPRESSED) ? true : false;
}
 
bool
CtrlBAckResponseHeader::IsMultiTid (void) const
{
  return (m_baType.m_variant == BlockAckType::MULTI_TID) ? true : false;
}
 
bool
CtrlBAckResponseHeader::IsMultiSta (void) const
{
  return (m_baType.m_variant == BlockAckType::MULTI_STA) ? true : false;
}
 
void
CtrlBAckResponseHeader::SetAid11 (uint16_t aid, std::size_t index)
{
  NS_ASSERT (m_baType.m_variant == BlockAckType::MULTI_STA && index < m_baInfo.size ());
 
  m_baInfo[index].m_aidTidInfo |= (aid & 0x07ff);
}
 
uint16_t
CtrlBAckResponseHeader::GetAid11 (std::size_t index) const
{
  NS_ASSERT (m_baType.m_variant == BlockAckType::MULTI_STA && index < m_baInfo.size ());
 
  return m_baInfo[index].m_aidTidInfo & 0x07ff;
}
 
void
CtrlBAckResponseHeader::SetAckType (bool type, std::size_t index)
{
  NS_ASSERT (m_baType.m_variant == BlockAckType::MULTI_STA && index < m_baInfo.size ());
 
  if (type)
    {
      m_baInfo[index].m_aidTidInfo |= (1 << 11);
    }
}
 
bool
CtrlBAckResponseHeader::GetAckType (std::size_t index) const
{
  NS_ASSERT (m_baType.m_variant == BlockAckType::MULTI_STA && index < m_baInfo.size ());
 
  return ((m_baInfo[index].m_aidTidInfo >> 11) & 0x0001) != 0;
}
 
void
CtrlBAckResponseHeader::SetUnassociatedStaAddress (const Mac48Address& ra, std::size_t index)
{
  NS_ASSERT (GetAid11 (index) == 2045);
 
  m_baInfo[index].m_ra = ra;
}
 
Mac48Address
CtrlBAckResponseHeader::GetUnassociatedStaAddress (std::size_t index) const
{
  NS_ASSERT (GetAid11 (index) == 2045);
 
  return m_baInfo[index].m_ra;
}
 
std::size_t
CtrlBAckResponseHeader::GetNPerAidTidInfoSubfields (void) const
{
  NS_ASSERT (m_baType.m_variant == BlockAckType::MULTI_STA);
  return m_baInfo.size ();
}
 
std::vector<uint32_t>
CtrlBAckResponseHeader::FindPerAidTidInfoWithAid (uint16_t aid) const
{
  NS_ASSERT (m_baType.m_variant == BlockAckType::MULTI_STA);
 
  std::vector<uint32_t> ret;
  ret.reserve (m_baInfo.size ());
  for (uint32_t i = 0; i < m_baInfo.size (); i++)
    {
      if (GetAid11 (i) == aid)
        {
          ret.push_back (i);
        }
    }
  return ret;
}
 
uint16_t
CtrlBAckResponseHeader::GetBaControl (void) const
{
  uint16_t res = 0;
  if (m_baAckPolicy)
    {
      res |= 0x1;
    }
  switch (m_baType.m_variant)
    {
      case BlockAckType::BASIC:
        break;
      case BlockAckType::COMPRESSED:
        res |= (0x02 << 1);
        break;
      case BlockAckType::EXTENDED_COMPRESSED:
        res |= (0x01 << 1);
        break;
      case BlockAckType::MULTI_TID:
        res |= (0x03 << 1);
        break;
      case BlockAckType::MULTI_STA:
        res |= (0x0b << 1);
        break;
      default:
        NS_FATAL_ERROR ("Invalid BA type");
        break;
    }
  if (m_baType.m_variant != BlockAckType::MULTI_STA)
    {
      res |= (m_tidInfo << 12) & (0xf << 12);
    }
  return res;
}
 
void
CtrlBAckResponseHeader::SetBaControl (uint16_t ba)
{
  m_baAckPolicy = ((ba & 0x01) == 1) ? true : false;
  if (((ba >> 1) & 0x0f) == 0x03)
    {
      SetType (BlockAckType::MULTI_TID);
    }
  else if (((ba >> 1) & 0x0f) == 0x01)
    {
      SetType (BlockAckType::EXTENDED_COMPRESSED);
    }
  else if (((ba >> 1) & 0x0f) == 0x02)
    {
      SetType (BlockAckType::COMPRESSED);
    }
  else if (((ba >> 1) & 0x0f) == 0)
    {
      SetType (BlockAckType::BASIC);
    }
  else if (((ba >> 1) & 0x0f) == 0x0b)
    {
      SetType (BlockAckType::MULTI_STA);
    }
  else
    {
      NS_FATAL_ERROR ("Invalid BA type");
    }
  if (m_baType.m_variant != BlockAckType::MULTI_STA)
    {
      m_tidInfo = (ba >> 12) & 0x0f;
    }
}
 
uint16_t
CtrlBAckResponseHeader::GetStartingSequenceControl (std::size_t index) const
{
  NS_ASSERT_MSG (m_baType.m_variant == BlockAckType::MULTI_STA || index == 0,
                 "index can only be non null for Multi-STA Block Ack");
  NS_ASSERT (index < m_baInfo.size ());
 
  uint16_t ret = (m_baInfo[index].m_startingSeq << 4) & 0xfff0;
 
  // The Fragment Number subfield encodes the length of the bitmap for
  // Compressed and Multi-STA variants (see sections 9.3.1.9.3 and 9.3.1.9.7
  // of 802.11ax Draft 3.0). Note that Fragmentation Level 3 is not supported.
  if (m_baType.m_variant == BlockAckType::COMPRESSED)
    {
      if (m_baType.m_bitmapLen[0] == 32)
        {
          ret |= 0x0004;
        }
    }
  else if (m_baType.m_variant == BlockAckType::MULTI_STA)
    {
      NS_ASSERT (m_baInfo.size () == m_baType.m_bitmapLen.size ());
      NS_ASSERT_MSG (m_baInfo[index].m_bitmap.size () > 0,
                     "This Per AID TID Info subfield has no Starting Sequence Control subfield");
 
      if (m_baType.m_bitmapLen[index] == 16)
        {
          ret |= 0x0002;
        }
      else if (m_baType.m_bitmapLen[index] == 32)
        {
          ret |= 0x0004;
        }
      else if (m_baType.m_bitmapLen[index] == 4)
        {
          ret |= 0x0006;
        }
    }
  return ret;
}
 
void
CtrlBAckResponseHeader::SetStartingSequenceControl (uint16_t seqControl, std::size_t index)
{
  NS_ASSERT_MSG (m_baType.m_variant == BlockAckType::MULTI_STA || index == 0,
                 "index can only be non null for Multi-STA Block Ack");
  NS_ASSERT (index < m_baInfo.size ());
 
  // The Fragment Number subfield encodes the length of the bitmap for
  // Compressed and Multi-STA variants (see sections 9.3.1.9.3 and 9.3.1.9.7
  // of 802.11ax Draft 3.0). Note that Fragmentation Level 3 is not supported.
  if (m_baType.m_variant == BlockAckType::COMPRESSED)
    {
      if ((seqControl & 0x0001) == 1)
        {
          NS_FATAL_ERROR ("Fragmentation Level 3 unsupported");
        }
      if (((seqControl >> 3) & 0x0001) == 0 && ((seqControl >> 1) & 0x0003) == 0)
        {
          SetType ({BlockAckType::COMPRESSED, {8}});
        }
      else if (((seqControl >> 3) & 0x0001) == 0 && ((seqControl >> 1) & 0x0003) == 2)
        {
          SetType ({BlockAckType::COMPRESSED, {32}});
        }
      else
        {
          NS_FATAL_ERROR ("Reserved configurations");
        }
    }
  else if (m_baType.m_variant == BlockAckType::MULTI_STA)
    {
      if ((seqControl & 0x0001) == 1)
        {
          NS_FATAL_ERROR ("Fragmentation Level 3 unsupported");
        }
      uint8_t bitmapLen = 0;
      if (((seqControl >> 3) & 0x0001) == 0 && ((seqControl >> 1) & 0x0003) == 0)
        {
          bitmapLen = 8;
        }
      else if (((seqControl >> 3) & 0x0001) == 0 && ((seqControl >> 1) & 0x0003) == 1)
        {
          bitmapLen = 16;
        }
      else if (((seqControl >> 3) & 0x0001) == 0 && ((seqControl >> 1) & 0x0003) == 2)
        {
          bitmapLen = 32;
        }
      else if (((seqControl >> 3) & 0x0001) == 0 && ((seqControl >> 1) & 0x0003) == 3)
        {
          bitmapLen = 4;
        }
      else
        {
          NS_FATAL_ERROR ("Reserved configurations");
        }
      m_baType.m_bitmapLen[index] = bitmapLen;
      m_baInfo[index].m_bitmap.assign (bitmapLen, 0);
    }
 
  m_baInfo[index].m_startingSeq = (seqControl >> 4) & 0x0fff;
}
 
Buffer::Iterator
CtrlBAckResponseHeader::SerializeBitmap (Buffer::Iterator start, std::size_t index) const
{
  NS_ASSERT_MSG (m_baType.m_variant == BlockAckType::MULTI_STA || index == 0,
                 "index can only be non null for Multi-STA Block Ack");
  NS_ASSERT (index < m_baInfo.size ());
 
  Buffer::Iterator i = start;
  switch (m_baType.m_variant)
    {
      case BlockAckType::BASIC:
      case BlockAckType::COMPRESSED:
      case BlockAckType::EXTENDED_COMPRESSED:
      case BlockAckType::MULTI_STA:
          for (const auto& byte : m_baInfo[index].m_bitmap)
            {
              i.WriteU8 (byte);
            }
          break;
      case BlockAckType::MULTI_TID:
        NS_FATAL_ERROR ("Multi-tid block ack is not supported.");
        break;
      default:
        NS_FATAL_ERROR ("Invalid BA type");
        break;
    }
  return i;
}
 
Buffer::Iterator
CtrlBAckResponseHeader::DeserializeBitmap (Buffer::Iterator start, std::size_t index)
{
  NS_ASSERT_MSG (m_baType.m_variant == BlockAckType::MULTI_STA || index == 0,
                 "index can only be non null for Multi-STA Block Ack");
  NS_ASSERT (index < m_baInfo.size ());
 
  Buffer::Iterator i = start;
  switch (m_baType.m_variant)
    {
      case BlockAckType::BASIC:
      case BlockAckType::COMPRESSED:
      case BlockAckType::EXTENDED_COMPRESSED:
      case BlockAckType::MULTI_STA:
          for (uint8_t j = 0; j < m_baType.m_bitmapLen[index]; j++)
            {
              m_baInfo[index].m_bitmap[j] = i.ReadU8 ();
            }
          break;
      case BlockAckType::MULTI_TID:
        NS_FATAL_ERROR ("Multi-tid block ack is not supported.");
        break;
      default:
        NS_FATAL_ERROR ("Invalid BA type");
        break;
    }
  return i;
}
 
void
CtrlBAckResponseHeader::SetReceivedPacket (uint16_t seq, std::size_t index)
{
  NS_ASSERT_MSG (m_baType.m_variant == BlockAckType::MULTI_STA || index == 0,
                 "index can only be non null for Multi-STA Block Ack");
  NS_ASSERT (index < m_baInfo.size ());
 
  if (!IsInBitmap (seq, index))
    {
      return;
    }
  switch (m_baType.m_variant)
    {
      case BlockAckType::BASIC:
        /* To set correctly basic block ack bitmap we need fragment number too.
            So if it's not specified, we consider packet not fragmented. */
        m_baInfo[index].m_bitmap[IndexInBitmap (seq) * 2] |= 0x01;
        break;
      case BlockAckType::COMPRESSED:
      case BlockAckType::EXTENDED_COMPRESSED:
      case BlockAckType::MULTI_STA:
        {
          uint16_t i = IndexInBitmap (seq, index);
          m_baInfo[index].m_bitmap[i / 8] |= (uint8_t (0x01) << (i % 8));
          break;
        }
      case BlockAckType::MULTI_TID:
        NS_FATAL_ERROR ("Multi-tid block ack is not supported.");
        break;
      default:
        NS_FATAL_ERROR ("Invalid BA type");
        break;
    }
}
 
void
CtrlBAckResponseHeader::SetReceivedFragment (uint16_t seq, uint8_t frag)
{
  NS_ASSERT (frag < 16);
  if (!IsInBitmap (seq))
    {
      return;
    }
  switch (m_baType.m_variant)
    {
      case BlockAckType::BASIC:
        m_baInfo[0].m_bitmap[IndexInBitmap (seq) * 2 + frag / 8] |= (0x01 << (frag % 8));
        break;
      case BlockAckType::COMPRESSED:
      case BlockAckType::EXTENDED_COMPRESSED:
      case BlockAckType::MULTI_STA:
        /* We can ignore this...compressed block ack doesn't support
           acknowledgment of single fragments */
        break;
      case BlockAckType::MULTI_TID:
        NS_FATAL_ERROR ("Multi-tid block ack is not supported.");
        break;
      default:
        NS_FATAL_ERROR ("Invalid BA type");
        break;
    }
}
 
bool
CtrlBAckResponseHeader::IsPacketReceived (uint16_t seq, std::size_t index) const
{
  NS_ASSERT_MSG (m_baType.m_variant == BlockAckType::MULTI_STA || index == 0,
                 "index can only be non null for Multi-STA Block Ack");
  NS_ASSERT (index < m_baInfo.size ());
 
  if (m_baType.m_variant == BlockAckType::MULTI_STA
      && GetAckType (index) && GetTidInfo (index) == 14)
    {
      // All-ack context
      return true;
    }
  if (!IsInBitmap (seq, index))
    {
      return false;
    }
  switch (m_baType.m_variant)
    {
      case BlockAckType::BASIC:
        /*It's impossible to say if an entire packet was correctly received. */
        return false;
      case BlockAckType::COMPRESSED:
      case BlockAckType::EXTENDED_COMPRESSED:
      case BlockAckType::MULTI_STA:
        {
          uint16_t i = IndexInBitmap (seq, index);
          uint8_t mask = uint8_t (0x01) << (i % 8);
          return ((m_baInfo[index].m_bitmap[i / 8] & mask) != 0) ? true : false;
        }
      case BlockAckType::MULTI_TID:
        NS_FATAL_ERROR ("Multi-tid block ack is not supported.");
        break;
      default:
        NS_FATAL_ERROR ("Invalid BA type");
        break;
    }
  return false;
}
 
bool
CtrlBAckResponseHeader::IsFragmentReceived (uint16_t seq, uint8_t frag) const
{
  NS_ASSERT (frag < 16);
  if (!IsInBitmap (seq))
    {
      return false;
    }
  switch (m_baType.m_variant)
    {
      case BlockAckType::BASIC:
        return ((m_baInfo[0].m_bitmap[IndexInBitmap (seq) * 2 + frag / 8] & (0x01 << (frag % 8))) != 0) ? true : false;
      case BlockAckType::COMPRESSED:
      case BlockAckType::EXTENDED_COMPRESSED:
      case BlockAckType::MULTI_STA:
        /* We can ignore this...compressed block ack doesn't support
           acknowledgement of single fragments */
        return false;
      case BlockAckType::MULTI_TID:
        {
          NS_FATAL_ERROR ("Multi-tid block ack is not supported.");
          break;
        }
      default:
        {
          NS_FATAL_ERROR ("Invalid BA type");
          break;
        }
    }
  return false;
}
 
uint16_t
CtrlBAckResponseHeader::IndexInBitmap (uint16_t seq, std::size_t index) const
{
  uint16_t i;
  if (seq >= m_baInfo[index].m_startingSeq)
    {
      i = seq - m_baInfo[index].m_startingSeq;
    }
  else
    {
      i = SEQNO_SPACE_SIZE - m_baInfo[index].m_startingSeq + seq;
    }
 
  uint16_t nAckedMpdus = m_baType.m_bitmapLen[index] * 8;
 
  if (m_baType.m_variant == BlockAckType::BASIC)
    {
      nAckedMpdus = nAckedMpdus / 16;
    }
 
  NS_ASSERT (i < nAckedMpdus);
  return i;
}
 
bool
CtrlBAckResponseHeader::IsInBitmap (uint16_t seq, std::size_t index) const
{
  NS_ASSERT_MSG (m_baType.m_variant == BlockAckType::MULTI_STA || index == 0,
                 "index can only be non null for Multi-STA Block Ack");
  NS_ASSERT (index < m_baType.m_bitmapLen.size ());
 
  uint16_t nAckedMpdus = m_baType.m_bitmapLen[index] * 8;
 
  if (m_baType.m_variant == BlockAckType::BASIC)
    {
      nAckedMpdus = nAckedMpdus / 16;
    }
 
  return (seq - m_baInfo[index].m_startingSeq + SEQNO_SPACE_SIZE) % SEQNO_SPACE_SIZE < nAckedMpdus;
}
 
const std::vector<uint8_t>&
CtrlBAckResponseHeader::GetBitmap (std::size_t index) const
{
  NS_ASSERT_MSG (m_baType.m_variant == BlockAckType::MULTI_STA || index == 0,
                 "index can only be non null for Multi-STA Block Ack");
  NS_ASSERT (index < m_baInfo.size ());
 
  return m_baInfo[index].m_bitmap;
}
 
void
CtrlBAckResponseHeader::ResetBitmap (std::size_t index)
{
  NS_ASSERT_MSG (m_baType.m_variant == BlockAckType::MULTI_STA || index == 0,
                 "index can only be non null for Multi-STA Block Ack");
  NS_ASSERT (index < m_baInfo.size ());
 
  m_baInfo[index].m_bitmap.assign (m_baType.m_bitmapLen[index], 0);
}
 
 
/***********************************
 * Trigger frame - User Info field
 ***********************************/
 
CtrlTriggerUserInfoField::CtrlTriggerUserInfoField (uint8_t triggerType)
  : m_aid12 (0),
    m_ruAllocation (0),
    m_ulFecCodingType (false),
    m_ulMcs (0),
    m_ulDcm (false),
    m_ulTargetRssi (0),
    m_triggerType (triggerType),
    m_basicTriggerDependentUserInfo (0)
{
  memset (&m_bits26To31, 0, sizeof (m_bits26To31));
}
 
CtrlTriggerUserInfoField::~CtrlTriggerUserInfoField ()
{
}
 
CtrlTriggerUserInfoField&
CtrlTriggerUserInfoField::operator= (const CtrlTriggerUserInfoField& userInfo)
{
  NS_ABORT_MSG_IF (m_triggerType != userInfo.m_triggerType, "Trigger Frame type mismatch");
 
  // check for self-assignment
  if (&userInfo == this)
    {
      return *this;
    }
 
  m_aid12 = userInfo.m_aid12;
  m_ruAllocation = userInfo.m_ruAllocation;
  m_ulFecCodingType = userInfo.m_ulFecCodingType;
  m_ulMcs = userInfo.m_ulMcs;
  m_ulDcm = userInfo.m_ulDcm;
  m_bits26To31 = userInfo.m_bits26To31;
  m_ulTargetRssi = userInfo.m_ulTargetRssi;
  m_basicTriggerDependentUserInfo = userInfo.m_basicTriggerDependentUserInfo;
  m_muBarTriggerDependentUserInfo = userInfo.m_muBarTriggerDependentUserInfo;
  return *this;
}
 
void
CtrlTriggerUserInfoField::Print (std::ostream &os) const
{
  os << ", USER_INFO AID=" << m_aid12 << ", RU_Allocation=" << +m_ruAllocation << ", MCS=" << +m_ulMcs;
}
 
uint32_t
CtrlTriggerUserInfoField::GetSerializedSize (void) const
{
  uint32_t size = 0;
  size += 5;   // User Info (excluding Trigger Dependent User Info)
 
  switch (m_triggerType)
    {
      case BASIC_TRIGGER:
      case BFRP_TRIGGER:
        size += 1;
        break;
      case MU_BAR_TRIGGER:
        size += m_muBarTriggerDependentUserInfo.GetSerializedSize (); // BAR Control and BAR Information
        break;
      // The Trigger Dependent User Info subfield is not present in the other variants
    }
 
  return size;
}
 
Buffer::Iterator
CtrlTriggerUserInfoField::Serialize (Buffer::Iterator start) const
{
  NS_ABORT_MSG_IF (m_triggerType == BFRP_TRIGGER, "BFRP Trigger frame is not supported");
  NS_ABORT_MSG_IF (m_triggerType == GCR_MU_BAR_TRIGGER, "GCR-MU-BAR Trigger frame is not supported");
  NS_ABORT_MSG_IF (m_triggerType == NFRP_TRIGGER, "NFRP Trigger frame is not supported");
 
  Buffer::Iterator i = start;
 
  uint32_t userInfo = 0;   // User Info except the MSB
  userInfo |= (m_aid12 & 0x0fff);
  userInfo |= (m_ruAllocation << 12);
  userInfo |= (m_ulFecCodingType ? 1 << 20 : 0);
  userInfo |= (m_ulMcs & 0x0f) << 21;
  userInfo |= (m_ulDcm ? 1 << 25 : 0);
 
  if (m_aid12 != 0 && m_aid12 != 2045)
    {
      userInfo |= (m_bits26To31.ssAllocation.startingSs & 0x07) << 26;
      userInfo |= (m_bits26To31.ssAllocation.nSs & 0x07) << 29;
    }
  else
    {
      userInfo |= (m_bits26To31.raRuInformation.nRaRu & 0x1f) << 26;
      userInfo |= (m_bits26To31.raRuInformation.moreRaRu ? 1 << 31 : 0);
    }
 
  i.WriteHtolsbU32 (userInfo);
  // Here we need to write 8 bits covering the UL Target RSSI (7 bits) and the
  // Reserved bit. Given how m_ulTargetRssi is set, writing m_ulTargetRssi
  // leads to setting the Reserved bit to zero
  i.WriteU8 (m_ulTargetRssi);
 
  if (m_triggerType == BASIC_TRIGGER)
    {
      i.WriteU8 (m_basicTriggerDependentUserInfo);
    }
  else if (m_triggerType == MU_BAR_TRIGGER)
    {
      m_muBarTriggerDependentUserInfo.Serialize (i);
      i.Next (m_muBarTriggerDependentUserInfo.GetSerializedSize ());
    }
 
  return i;
}
 
Buffer::Iterator
CtrlTriggerUserInfoField::Deserialize (Buffer::Iterator start)
{
  NS_ABORT_MSG_IF (m_triggerType == BFRP_TRIGGER, "BFRP Trigger frame is not supported");
  NS_ABORT_MSG_IF (m_triggerType == GCR_MU_BAR_TRIGGER, "GCR-MU-BAR Trigger frame is not supported");
  NS_ABORT_MSG_IF (m_triggerType == NFRP_TRIGGER, "NFRP Trigger frame is not supported");
 
  Buffer::Iterator i = start;
 
  uint32_t userInfo = i.ReadLsbtohU32 ();
 
  m_aid12 = userInfo & 0x0fff;
  NS_ABORT_MSG_IF (m_aid12 == 4095, "Cannot deserialize a Padding field");
  m_ruAllocation = (userInfo >> 12) & 0xff;
  m_ulFecCodingType = (userInfo >> 20) & 0x01;
  m_ulMcs = (userInfo >> 21) & 0x0f;
  m_ulDcm = (userInfo >> 25) & 0x01;
 
  if (m_aid12 != 0 && m_aid12 != 2045)
    {
      m_bits26To31.ssAllocation.startingSs = (userInfo >> 26) & 0x07;
      m_bits26To31.ssAllocation.nSs = (userInfo >> 29) & 0x07;
    }
  else
    {
      m_bits26To31.raRuInformation.nRaRu = (userInfo >> 26) & 0x1f;
      m_bits26To31.raRuInformation.moreRaRu = (userInfo >> 31) & 0x01;
    }
 
  m_ulTargetRssi = i.ReadU8 () & 0x7f;  // B39 is reserved
 
  if (m_triggerType == BASIC_TRIGGER)
    {
      m_basicTriggerDependentUserInfo= i.ReadU8 ();
    }
  else if (m_triggerType == MU_BAR_TRIGGER)
    {
      uint32_t len = m_muBarTriggerDependentUserInfo.Deserialize (i);
      i.Next (len);
    }
 
  return i;
}
 
TriggerFrameType
CtrlTriggerUserInfoField::GetType (void) const
{
  return static_cast<TriggerFrameType> (m_triggerType);
}
 
void
CtrlTriggerUserInfoField::SetAid12 (uint16_t aid)
{
  m_aid12 = aid & 0x0fff;
}
 
uint16_t
CtrlTriggerUserInfoField::GetAid12 (void) const
{
  return m_aid12;
}
 
bool
CtrlTriggerUserInfoField::HasRaRuForAssociatedSta (void) const
{
  return (m_aid12 == 0);
}
 
bool
CtrlTriggerUserInfoField::HasRaRuForUnassociatedSta (void) const
{
  return (m_aid12 == 2045);
}
 
void
CtrlTriggerUserInfoField::SetRuAllocation (HeRu::RuSpec ru)
{
  NS_ABORT_MSG_IF (ru.GetIndex () == 0, "Valid indices start at 1");
 
  switch (ru.GetRuType ())
    {
    case HeRu::RU_26_TONE:
      m_ruAllocation = ru.GetIndex () - 1;
      break;
    case HeRu::RU_52_TONE:
      m_ruAllocation = ru.GetIndex () + 36;
      break;
    case HeRu::RU_106_TONE:
      m_ruAllocation = ru.GetIndex () + 52;
      break;
    case HeRu::RU_242_TONE:
      m_ruAllocation = ru.GetIndex () + 60;
      break;
    case HeRu::RU_484_TONE:
      m_ruAllocation = ru.GetIndex () + 64;
      break;
    case HeRu::RU_996_TONE:
      m_ruAllocation = 67;
      break;
    case HeRu::RU_2x996_TONE:
      m_ruAllocation = 68;
      break;
    default:
      NS_FATAL_ERROR ("RU type unknown.");
      break;
    }
 
  NS_ABORT_MSG_IF (m_ruAllocation > 68, "Reserved value.");
 
  m_ruAllocation <<= 1;
  if (!ru.GetPrimary80MHz ())
    {
      m_ruAllocation++;
    }
}
 
HeRu::RuSpec
CtrlTriggerUserInfoField::GetRuAllocation (void) const
{
  HeRu::RuType ruType;
  std::size_t index;
 
  bool primary80MHz = ((m_ruAllocation & 0x01) == 0);
 
  uint8_t val = m_ruAllocation >> 1;
 
  if (val < 37)
    {
      ruType = HeRu::RU_26_TONE;
      index = val + 1;
    }
  else if (val < 53)
    {
      ruType = HeRu::RU_52_TONE;
      index = val - 36;
    }
  else if (val < 61)
    {
      ruType = HeRu::RU_106_TONE;
      index = val - 52;
    }
  else if (val < 65)
    {
      ruType = HeRu::RU_242_TONE;
      index = val - 60;
    }
  else if (val < 67)
    {
      ruType = HeRu::RU_484_TONE;
      index = val - 64;
    }
  else if (val == 67)
    {
      ruType = HeRu::RU_996_TONE;
      index = 1;
    }
  else if (val == 68)
    {
      ruType = HeRu::RU_2x996_TONE;
      index = 1;
    }
  else
    {
      NS_FATAL_ERROR ("Reserved value.");
    }
 
  return HeRu::RuSpec (ruType, index, primary80MHz);
}
 
void
CtrlTriggerUserInfoField::SetUlFecCodingType (bool ldpc)
{
  m_ulFecCodingType = ldpc;
}
 
bool
CtrlTriggerUserInfoField::GetUlFecCodingType (void) const
{
  return m_ulFecCodingType;
}
 
void
CtrlTriggerUserInfoField::SetUlMcs (uint8_t mcs)
{
  NS_ABORT_MSG_IF (mcs > 11, "Invalid MCS index");
  m_ulMcs = mcs;
}
 
uint8_t
CtrlTriggerUserInfoField::GetUlMcs (void) const
{
  return m_ulMcs;
}
 
void
CtrlTriggerUserInfoField::SetUlDcm (bool dcm)
{
  m_ulDcm = dcm;
}
 
bool
CtrlTriggerUserInfoField::GetUlDcm (void) const
{
  return m_ulDcm;
}
 
void
CtrlTriggerUserInfoField::SetSsAllocation (uint8_t startingSs, uint8_t nSs)
{
  NS_ABORT_MSG_IF (m_aid12 == 0 || m_aid12 == 2045, "SS Allocation subfield not present");
  NS_ABORT_MSG_IF (!startingSs || startingSs > 8, "Starting SS must be from 1 to 8");
  NS_ABORT_MSG_IF (!nSs || nSs > 8, "Number of SS must be from 1 to 8");
 
  m_bits26To31.ssAllocation.startingSs = startingSs - 1;
  m_bits26To31.ssAllocation.nSs = nSs - 1;
}
 
uint8_t
CtrlTriggerUserInfoField::GetStartingSs (void) const
{
  if (m_aid12 == 0 || m_aid12 == 2045)
    {
      return 1;
    }
  return m_bits26To31.ssAllocation.startingSs + 1;
}
 
uint8_t
CtrlTriggerUserInfoField::GetNss (void) const
{
  if (m_aid12 == 0 || m_aid12 == 2045)
    {
      return 1;
    }
  return m_bits26To31.ssAllocation.nSs + 1;
}
 
void
CtrlTriggerUserInfoField::SetRaRuInformation (uint8_t nRaRu, bool moreRaRu)
{
  NS_ABORT_MSG_IF (m_aid12 != 0 && m_aid12 != 2045, "RA-RU Information subfield not present");
  NS_ABORT_MSG_IF (!nRaRu || nRaRu > 32, "Number of contiguous RA-RUs must be from 1 to 32");
 
  m_bits26To31.raRuInformation.nRaRu = nRaRu - 1;
  m_bits26To31.raRuInformation.moreRaRu = moreRaRu;
}
 
uint8_t
CtrlTriggerUserInfoField::GetNRaRus (void) const
{
  NS_ABORT_MSG_IF (m_aid12 != 0 && m_aid12 != 2045, "RA-RU Information subfield not present");
 
  return m_bits26To31.raRuInformation.nRaRu + 1;
}
 
bool
CtrlTriggerUserInfoField::GetMoreRaRu (void) const
{
  NS_ABORT_MSG_IF (m_aid12 != 0 && m_aid12 != 2045, "RA-RU Information subfield not present");
 
  return m_bits26To31.raRuInformation.moreRaRu;
}
 
void
CtrlTriggerUserInfoField::SetUlTargetRssiMaxTxPower (void)
{
  m_ulTargetRssi = 127;  // see Table 9-25i of 802.11ax amendment D3.0
}
 
void
CtrlTriggerUserInfoField::SetUlTargetRssi (int8_t dBm)
{
  NS_ABORT_MSG_IF (dBm < -110 || dBm > -20, "Invalid values for signal power");
 
  m_ulTargetRssi = static_cast<uint8_t> (110 + dBm);
}
 
bool
CtrlTriggerUserInfoField::IsUlTargetRssiMaxTxPower (void) const
{
  return (m_ulTargetRssi == 127);
}
 
int8_t
CtrlTriggerUserInfoField::GetUlTargetRssi (void) const
{
  NS_ABORT_MSG_IF (m_ulTargetRssi == 127, "STA must use its max TX power");
 
  return static_cast<int8_t> (m_ulTargetRssi) - 110;
}
 
void
CtrlTriggerUserInfoField::SetBasicTriggerDepUserInfo (uint8_t spacingFactor, uint8_t tidLimit, AcIndex prefAc)
{
  NS_ABORT_MSG_IF (m_triggerType != BASIC_TRIGGER, "Not a Basic Trigger Frame");
 
  m_basicTriggerDependentUserInfo = (spacingFactor & 0x03)
                                     | (tidLimit & 0x07) << 2
                                     // B5 is reserved
                                     | (prefAc & 0x03) << 6;
}
 
uint8_t
CtrlTriggerUserInfoField::GetMpduMuSpacingFactor (void) const
{
  NS_ABORT_MSG_IF (m_triggerType != BASIC_TRIGGER, "Not a Basic Trigger Frame");
 
  return m_basicTriggerDependentUserInfo & 0x03;
}
 
uint8_t
CtrlTriggerUserInfoField::GetTidAggregationLimit (void) const
{
  NS_ABORT_MSG_IF (m_triggerType != BASIC_TRIGGER, "Not a Basic Trigger Frame");
 
  return (m_basicTriggerDependentUserInfo & 0x1c) >> 2;
}
 
AcIndex
CtrlTriggerUserInfoField::GetPreferredAc (void) const
{
  NS_ABORT_MSG_IF (m_triggerType != BASIC_TRIGGER, "Not a Basic Trigger Frame");
 
  return AcIndex ((m_basicTriggerDependentUserInfo & 0xc0) >> 6);
}
 
void
CtrlTriggerUserInfoField::SetMuBarTriggerDepUserInfo (const CtrlBAckRequestHeader& bar)
{
  NS_ABORT_MSG_IF (m_triggerType != MU_BAR_TRIGGER, "Not a MU-BAR Trigger frame");
  NS_ABORT_MSG_IF (bar.GetType ().m_variant != BlockAckReqType::COMPRESSED
                   && bar.GetType ().m_variant != BlockAckReqType::MULTI_TID,
                   "BAR Control indicates it is neither the Compressed nor the Multi-TID variant");
  m_muBarTriggerDependentUserInfo = bar;
}
 
const CtrlBAckRequestHeader&
CtrlTriggerUserInfoField::GetMuBarTriggerDepUserInfo (void) const
{
  NS_ABORT_MSG_IF (m_triggerType != MU_BAR_TRIGGER, "Not a MU-BAR Trigger frame");
 
  return m_muBarTriggerDependentUserInfo;
}
 
/***********************************
 *       Trigger frame
 ***********************************/
 
NS_OBJECT_ENSURE_REGISTERED (CtrlTriggerHeader);
 
CtrlTriggerHeader::CtrlTriggerHeader ()
  : m_triggerType (0),
    m_ulLength (0),
    m_moreTF (false),
    m_csRequired (false),
    m_ulBandwidth (0),
    m_giAndLtfType (0),
    m_apTxPower (0),
    m_ulSpatialReuse (0)
{
}
 
CtrlTriggerHeader::CtrlTriggerHeader (TriggerFrameType type, const WifiTxVector& txVector)
  : CtrlTriggerHeader ()
{
  m_triggerType = type;
  SetUlBandwidth (txVector.GetChannelWidth ());
  SetUlLength (txVector.GetLength ());
  uint16_t gi = txVector.GetGuardInterval ();
  if (gi == 800 || gi == 1600)
    {
      m_giAndLtfType = 1;
    }
  else
    {
      m_giAndLtfType = 2;
    }
  for (auto& userInfo : txVector.GetHeMuUserInfoMap ())
    {
      CtrlTriggerUserInfoField& ui = AddUserInfoField ();
      ui.SetAid12 (userInfo.first);
      ui.SetRuAllocation (userInfo.second.ru);
      ui.SetUlMcs (userInfo.second.mcs.GetMcsValue ());
      ui.SetSsAllocation (1, userInfo.second.nss); // MU-MIMO is not supported
    }
}
 
CtrlTriggerHeader::~CtrlTriggerHeader ()
{
}
 
CtrlTriggerHeader&
CtrlTriggerHeader::operator= (const CtrlTriggerHeader& trigger)
{
  // check for self-assignment
  if (&trigger == this)
    {
      return *this;
    }
 
  m_triggerType = trigger.m_triggerType;
  m_ulLength = trigger.m_ulLength;
  m_moreTF = trigger.m_moreTF;
  m_csRequired = trigger.m_csRequired;
  m_ulBandwidth = trigger.m_ulBandwidth;
  m_giAndLtfType = trigger.m_giAndLtfType;
  m_apTxPower = trigger.m_apTxPower;
  m_ulSpatialReuse = trigger.m_ulSpatialReuse;
  m_userInfoFields.clear ();
  m_userInfoFields = trigger.m_userInfoFields;
  return *this;
}
 
TypeId
CtrlTriggerHeader::GetTypeId (void)
{
  static TypeId tid = TypeId ("ns3::CtrlTriggerHeader")
    .SetParent<Header> ()
    .SetGroupName ("Wifi")
    .AddConstructor<CtrlTriggerHeader> ()
  ;
  return tid;
}
 
TypeId
CtrlTriggerHeader::GetInstanceTypeId (void) const
{
  return GetTypeId ();
}
 
void
CtrlTriggerHeader::Print (std::ostream &os) const
{
  os << "TriggerType=" << GetTypeString () << ", Bandwidth=" << +GetUlBandwidth ()
     << ", UL Length=" << m_ulLength;
 
  for (auto& ui : m_userInfoFields)
    {
      ui.Print (os);
    }
}
 
uint32_t
CtrlTriggerHeader::GetSerializedSize (void) const
{
  uint32_t size = 0;
  size += 8;  // Common Info (excluding Trigger Dependent Common Info)
 
  // Add the size of the Trigger Dependent Common Info subfield
  if (m_triggerType == GCR_MU_BAR_TRIGGER)
    {
      size += 4;
    }
 
  for (auto& ui : m_userInfoFields)
    {
      size += ui.GetSerializedSize ();
    }
 
  size += 2;  // Padding field
 
  return size;
}
 
void
CtrlTriggerHeader::Serialize (Buffer::Iterator start) const
{
  NS_ABORT_MSG_IF (m_triggerType == BFRP_TRIGGER, "BFRP Trigger frame is not supported");
  NS_ABORT_MSG_IF (m_triggerType == GCR_MU_BAR_TRIGGER, "GCR-MU-BAR Trigger frame is not supported");
  NS_ABORT_MSG_IF (m_triggerType == NFRP_TRIGGER, "NFRP Trigger frame is not supported");
 
  Buffer::Iterator i = start;
 
  uint64_t commonInfo = 0;
  commonInfo |= (m_triggerType & 0x0f);
  commonInfo |= (m_ulLength & 0x0fff) << 4;
  commonInfo |= (m_moreTF ? 1 << 16 : 0);
  commonInfo |= (m_csRequired ? 1 << 17 : 0);
  commonInfo |= (m_ulBandwidth & 0x03) << 18;
  commonInfo |= (m_giAndLtfType & 0x03) << 20;
  commonInfo |= static_cast<uint64_t> (m_apTxPower & 0x3f) << 28;
  commonInfo |= static_cast<uint64_t> (m_ulSpatialReuse) << 37;
 
  i.WriteHtolsbU64 (commonInfo);
 
  for (auto& ui : m_userInfoFields)
    {
      i = ui.Serialize (i);
    }
 
  i.WriteHtolsbU16 (0xffff);   // Padding field, used as delimiter
}
 
uint32_t
CtrlTriggerHeader::Deserialize (Buffer::Iterator start)
{
  Buffer::Iterator i = start;
 
  uint64_t commonInfo = i.ReadLsbtohU64 ();
 
  m_triggerType = (commonInfo & 0x0f);
  m_ulLength = (commonInfo >> 4) & 0x0fff;
  m_moreTF = (commonInfo >> 16) & 0x01;
  m_csRequired = (commonInfo >> 17) & 0x01;
  m_ulBandwidth = (commonInfo >> 18) & 0x03;
  m_giAndLtfType = (commonInfo >> 20) & 0x03;
  m_apTxPower = (commonInfo >> 28) & 0x3f;
  m_ulSpatialReuse = (commonInfo >> 37) & 0xffff;
  m_userInfoFields.clear ();
 
  NS_ABORT_MSG_IF (m_triggerType == BFRP_TRIGGER, "BFRP Trigger frame is not supported");
  NS_ABORT_MSG_IF (m_triggerType == GCR_MU_BAR_TRIGGER, "GCR-MU-BAR Trigger frame is not supported");
  NS_ABORT_MSG_IF (m_triggerType == NFRP_TRIGGER, "NFRP Trigger frame is not supported");
 
  bool isPadding = false;
 
  // We always add a Padding field (of two octets of all 1s) as delimiter
  while (!isPadding)
    {
      // read the first 2 bytes to check if we encountered the Padding field
      if (i.ReadU16 () == 0xffff)
        {
          isPadding = true;
        }
      else
        {
          // go back 2 bytes to deserialize the User Info field from the beginning
          i.Prev (2);
          CtrlTriggerUserInfoField& ui = AddUserInfoField ();
          i = ui.Deserialize (i);
        }
    }
 
  return i.GetDistanceFrom (start);
}
 
void
CtrlTriggerHeader::SetType (TriggerFrameType type)
{
  m_triggerType = type;
}
 
TriggerFrameType
CtrlTriggerHeader::GetType (void) const
{
  return static_cast<TriggerFrameType> (m_triggerType);
}
 
const char *
CtrlTriggerHeader::GetTypeString (void) const
{
  return GetTypeString (GetType ());
}
 
const char *
CtrlTriggerHeader::GetTypeString (TriggerFrameType type)
{
#define FOO(x) \
case x: \
  return # x; \
  break;
 
  switch (type)
    {
      FOO (BASIC_TRIGGER);
      FOO (BFRP_TRIGGER);
      FOO (MU_BAR_TRIGGER);
      FOO (MU_RTS_TRIGGER);
      FOO (BSRP_TRIGGER);
      FOO (GCR_MU_BAR_TRIGGER);
      FOO (BQRP_TRIGGER);
      FOO (NFRP_TRIGGER);
    default:
      return "ERROR";
    }
#undef FOO
}
 
bool
CtrlTriggerHeader::IsBasic (void) const
{
  return (m_triggerType == BASIC_TRIGGER);
}
 
bool
CtrlTriggerHeader::IsBfrp (void) const
{
  return (m_triggerType == BFRP_TRIGGER);
}
 
bool
CtrlTriggerHeader::IsMuBar (void) const
{
  return (m_triggerType == MU_BAR_TRIGGER);
}
 
bool
CtrlTriggerHeader::IsMuRts (void) const
{
  return (m_triggerType == MU_RTS_TRIGGER);
}
 
bool
CtrlTriggerHeader::IsBsrp (void) const
{
  return (m_triggerType == BSRP_TRIGGER);
}
 
bool
CtrlTriggerHeader::IsGcrMuBar (void) const
{
  return (m_triggerType == GCR_MU_BAR_TRIGGER);
}
 
bool
CtrlTriggerHeader::IsBqrp (void) const
{
  return (m_triggerType == BQRP_TRIGGER);
}
 
bool
CtrlTriggerHeader::IsNfrp (void) const
{
  return (m_triggerType == NFRP_TRIGGER);
}
 
void
CtrlTriggerHeader::SetUlLength (uint16_t len)
{
  m_ulLength = (len & 0x0fff);
}
 
uint16_t
CtrlTriggerHeader::GetUlLength (void) const
{
  return m_ulLength;
}
 
WifiTxVector
CtrlTriggerHeader::GetHeTbTxVector (uint16_t staId) const
{
  auto userInfoIt = FindUserInfoWithAid (staId);
  NS_ASSERT (userInfoIt != end ());
 
  WifiTxVector v;
  v.SetPreambleType (WifiPreamble::WIFI_PREAMBLE_HE_TB);
  v.SetChannelWidth (GetUlBandwidth ());
  v.SetGuardInterval (GetGuardInterval ());
  v.SetLength (GetUlLength ());
  v.SetHeMuUserInfo (staId, {userInfoIt->GetRuAllocation (),
                             HePhy::GetHeMcs (userInfoIt->GetUlMcs ()),
                             userInfoIt->GetNss ()});
  return v;
}
 
void
CtrlTriggerHeader::SetMoreTF (bool more)
{
  m_moreTF = more;
}
 
bool
CtrlTriggerHeader::GetMoreTF (void) const
{
  return m_moreTF;
}
 
void
CtrlTriggerHeader::SetCsRequired (bool cs)
{
  m_csRequired = cs;
}
 
bool
CtrlTriggerHeader::GetCsRequired (void) const
{
  return m_csRequired;
}
 
void
CtrlTriggerHeader::SetUlBandwidth (uint16_t bw)
{
  switch (bw)
    {
    case 20:
      m_ulBandwidth = 0;
      break;
    case 40:
      m_ulBandwidth = 1;
      break;
    case 80:
      m_ulBandwidth = 2;
      break;
    case 160:
      m_ulBandwidth = 3;
      break;
    default:
      NS_FATAL_ERROR ("Bandwidth value not allowed.");
      break;
    }
}
 
uint16_t
CtrlTriggerHeader::GetUlBandwidth (void) const
{
  return (1 << m_ulBandwidth) * 20;
}
 
void
CtrlTriggerHeader::SetGiAndLtfType (uint16_t guardInterval, uint8_t ltfType)
{
  if (ltfType == 1 && guardInterval == 1600)
    {
      m_giAndLtfType = 0;
    }
  else if (ltfType == 2 && guardInterval == 1600)
    {
      m_giAndLtfType = 1;
    }
  else if (ltfType == 4 && guardInterval == 3200)
    {
      m_giAndLtfType = 2;
    }
  else
    {
      NS_FATAL_ERROR ("Invalid combination of GI and LTF type");
    }
}
 
uint16_t
CtrlTriggerHeader::GetGuardInterval (void) const
{
  if (m_giAndLtfType == 0 || m_giAndLtfType == 1)
    {
      return 1600;
    }
  else if (m_giAndLtfType == 2)
    {
      return 3200;
    }
  else
    {
      NS_FATAL_ERROR ("Invalid value for GI And LTF Type subfield");
    }
}
 
uint8_t
CtrlTriggerHeader::GetLtfType (void) const
{
  if (m_giAndLtfType == 0)
    {
      return 1;
    }
  else if (m_giAndLtfType == 1)
    {
      return 2;
    }
  else if (m_giAndLtfType == 2)
    {
      return 4;
    }
  else
    {
      NS_FATAL_ERROR ("Invalid value for GI And LTF Type subfield");
    }
}
 
void
CtrlTriggerHeader::SetApTxPower (int8_t power)
{
  // see Table 9-25f "AP Tx Power subfield encoding" of 802.11ax amendment D3.0
  NS_ABORT_MSG_IF (power < -20 || power > 40, "Out of range power values");
 
  m_apTxPower = static_cast<uint8_t> (power + 20);
}
 
int8_t
CtrlTriggerHeader::GetApTxPower (void) const
{
  // see Table 9-25f "AP Tx Power subfield encoding" of 802.11ax amendment D3.0
  return static_cast<int8_t> (m_apTxPower) - 20;
}
 
void
CtrlTriggerHeader::SetUlSpatialReuse (uint16_t sr)
{
  m_ulSpatialReuse = sr;
}
 
uint16_t
CtrlTriggerHeader::GetUlSpatialReuse (void) const
{
  return m_ulSpatialReuse;
}
 
CtrlTriggerHeader
CtrlTriggerHeader::GetCommonInfoField (void) const
{
  // make a copy of this Trigger Frame and remove the User Info fields from the copy
  CtrlTriggerHeader trigger (*this);
  trigger.m_userInfoFields.clear ();
  return trigger;
}
 
CtrlTriggerUserInfoField&
CtrlTriggerHeader::AddUserInfoField (void)
{
  m_userInfoFields.emplace_back (m_triggerType);
  return m_userInfoFields.back ();
}
 
CtrlTriggerUserInfoField&
CtrlTriggerHeader::AddUserInfoField (const CtrlTriggerUserInfoField& userInfo)
{
  NS_ABORT_MSG_IF (userInfo.GetType () != m_triggerType,
                   "Trying to add a User Info field of a type other than the type of the Trigger Frame");
  m_userInfoFields.push_back (userInfo);
  return m_userInfoFields.back ();
}
 
CtrlTriggerHeader::ConstIterator
CtrlTriggerHeader::begin (void) const
{
  return m_userInfoFields.begin ();
}
 
CtrlTriggerHeader::ConstIterator
CtrlTriggerHeader::end (void) const
{
  return m_userInfoFields.end ();
}
 
CtrlTriggerHeader::Iterator
CtrlTriggerHeader::begin (void)
{
  return m_userInfoFields.begin ();
}
 
CtrlTriggerHeader::Iterator
CtrlTriggerHeader::end (void)
{
  return m_userInfoFields.end ();
}
 
std::size_t
CtrlTriggerHeader::GetNUserInfoFields (void) const
{
  return m_userInfoFields.size ();
}
 
CtrlTriggerHeader::ConstIterator
CtrlTriggerHeader::FindUserInfoWithAid (ConstIterator start, uint16_t aid12) const
{
  // the lambda function returns true if a User Info field has the AID12 subfield
  // equal to the given aid12 value
  return std::find_if (start, end (), [aid12] (const CtrlTriggerUserInfoField& ui) -> bool
                                      { return (ui.GetAid12 () == aid12); });
}
 
CtrlTriggerHeader::ConstIterator
CtrlTriggerHeader::FindUserInfoWithAid (uint16_t aid12) const
{
  return FindUserInfoWithAid (m_userInfoFields.begin (), aid12);
}
 
CtrlTriggerHeader::ConstIterator
CtrlTriggerHeader::FindUserInfoWithRaRuAssociated (ConstIterator start) const
{
  return FindUserInfoWithAid (start, 0);
}
 
CtrlTriggerHeader::ConstIterator
CtrlTriggerHeader::FindUserInfoWithRaRuAssociated (void) const
{
  return FindUserInfoWithAid (0);
}
 
CtrlTriggerHeader::ConstIterator
CtrlTriggerHeader::FindUserInfoWithRaRuUnassociated (ConstIterator start) const
{
  return FindUserInfoWithAid (start, 2045);
}
 
CtrlTriggerHeader::ConstIterator
CtrlTriggerHeader::FindUserInfoWithRaRuUnassociated (void) const
{
  return FindUserInfoWithAid (2045);
}
 
bool
CtrlTriggerHeader::IsValid (void) const
{
  // check that allocated RUs do not overlap
  // TODO This is not a problem in case of UL MU-MIMO
  std::vector<HeRu::RuSpec> prevRus;
 
  for (auto& ui : m_userInfoFields)
    {
      if (HeRu::DoesOverlap (GetUlBandwidth (), ui.GetRuAllocation (), prevRus))
        {
          return false;
        }
      prevRus.push_back (ui.GetRuAllocation ());
    }
  return true;
}
 
}  //namespace ns3