nr-epc-gtpu-header.cc

// Copyright (c) 2011 Centre Tecnologic de Telecomunicacions de Catalunya (CTTC)
//
// SPDX-License-Identifier: GPL-2.0-only
//
// Author: Jaume Nin <jnin@cttc.cat>
 
#include "nr-epc-gtpu-header.h"
 
#include "ns3/log.h"
#include "ns3/packet.h"
 
namespace ns3
{
 
NS_LOG_COMPONENT_DEFINE("NrGtpuHeader");
 
/********************************************************
 *        GTP-U-v1 Header
 ********************************************************/
 
NS_OBJECT_ENSURE_REGISTERED(NrGtpuHeader);
 
TypeId
NrGtpuHeader::GetTypeId()
{
    static TypeId tid = TypeId("ns3::NrGtpuHeader")
                            .SetParent<Header>()
                            .SetGroupName("Nr")
                            .AddConstructor<NrGtpuHeader>();
    return tid;
}
 
NrGtpuHeader::NrGtpuHeader()
    : m_version(1),
      m_protocolType(true),
      m_extensionHeaderFlag(false),
      m_sequenceNumberFlag(true),
      m_nPduNumberFlag(true),
      m_messageType(255),
      m_length(0),
      m_teid(0),
      m_sequenceNumber(0),
      m_nPduNumber(0),
      m_nextExtensionType(0)
{
}
 
NrGtpuHeader::~NrGtpuHeader()
{
}
 
TypeId
NrGtpuHeader::GetInstanceTypeId() const
{
    return GetTypeId();
}
 
uint32_t
NrGtpuHeader::GetSerializedSize() const
{
    return 12;
}
 
void
NrGtpuHeader::Serialize(Buffer::Iterator start) const
{
    Buffer::Iterator i = start;
    uint8_t firstByte = m_version << 5 | m_protocolType << 4 | 0x1 << 3;
    firstByte |= m_extensionHeaderFlag << 2 | m_sequenceNumberFlag << 1 | m_nPduNumberFlag;
    i.WriteU8(firstByte);
    i.WriteU8(m_messageType);
    i.WriteHtonU16(m_length);
    i.WriteHtonU32(m_teid);
    i.WriteHtonU16(m_sequenceNumber);
    i.WriteU8(m_nPduNumber);
    i.WriteU8(m_nextExtensionType);
}
 
uint32_t
NrGtpuHeader::Deserialize(Buffer::Iterator start)
{
    Buffer::Iterator i = start;
    uint8_t firstByte = i.ReadU8();
    m_version = firstByte >> 5 & 0x7;
    m_protocolType = firstByte >> 4 & 0x1;
    m_extensionHeaderFlag = firstByte >> 2 & 0x1;
    m_sequenceNumberFlag = firstByte >> 1 & 0x1;
    m_nPduNumberFlag = firstByte & 0x1;
    m_messageType = i.ReadU8();
    m_length = i.ReadNtohU16();
    m_teid = i.ReadNtohU32();
    m_sequenceNumber = i.ReadNtohU16();
    m_nPduNumber = i.ReadU8();
    m_nextExtensionType = i.ReadU8();
    return GetSerializedSize();
}
 
void
NrGtpuHeader::Print(std::ostream& os) const
{
    os << " version=" << (uint32_t)m_version << " [";
    if (m_protocolType)
    {
        os << " PT ";
    }
    if (m_extensionHeaderFlag)
    {
        os << " E ";
    }
    if (m_sequenceNumberFlag)
    {
        os << " S ";
    }
    if (m_nPduNumberFlag)
    {
        os << " PN ";
    }
    os << "], messageType=" << (uint32_t)m_messageType << ", length=" << (uint32_t)m_length;
    os << ", teid=" << (uint32_t)m_teid << ", sequenceNumber=" << (uint32_t)m_sequenceNumber;
    os << ", nPduNumber=" << (uint32_t)m_nPduNumber
       << ", nextExtensionType=" << (uint32_t)m_nextExtensionType;
}
 
bool
NrGtpuHeader::GetExtensionHeaderFlag() const
{
    return m_extensionHeaderFlag;
}
 
uint16_t
NrGtpuHeader::GetLength() const
{
    return m_length;
}
 
uint8_t
NrGtpuHeader::GetMessageType() const
{
    return m_messageType;
}
 
uint8_t
NrGtpuHeader::GetNPduNumber() const
{
    return m_nPduNumber;
}
 
bool
NrGtpuHeader::GetNPduNumberFlag() const
{
    return m_nPduNumberFlag;
}
 
uint8_t
NrGtpuHeader::GetNextExtensionType() const
{
    return m_nextExtensionType;
}
 
bool
NrGtpuHeader::GetProtocolType() const
{
    return m_protocolType;
}
 
uint16_t
NrGtpuHeader::GetSequenceNumber() const
{
    return m_sequenceNumber;
}
 
bool
NrGtpuHeader::GetSequenceNumberFlag() const
{
    return m_sequenceNumberFlag;
}
 
uint32_t
NrGtpuHeader::GetTeid() const
{
    return m_teid;
}
 
uint8_t
NrGtpuHeader::GetVersion() const
{
    return m_version;
}
 
void
NrGtpuHeader::SetExtensionHeaderFlag(bool m_extensionHeaderFlag)
{
    this->m_extensionHeaderFlag = m_extensionHeaderFlag;
}
 
void
NrGtpuHeader::SetLength(uint16_t m_length)
{
    this->m_length = m_length;
}
 
void
NrGtpuHeader::SetMessageType(uint8_t m_messageType)
{
    this->m_messageType = m_messageType;
}
 
void
NrGtpuHeader::SetNPduNumber(uint8_t m_nPduNumber)
{
    this->m_nPduNumber = m_nPduNumber;
}
 
void
NrGtpuHeader::SetNPduNumberFlag(bool m_nPduNumberFlag)
{
    this->m_nPduNumberFlag = m_nPduNumberFlag;
}
 
void
NrGtpuHeader::SetNextExtensionType(uint8_t m_nextExtensionType)
{
    this->m_nextExtensionType = m_nextExtensionType;
}
 
void
NrGtpuHeader::SetProtocolType(bool m_protocolType)
{
    this->m_protocolType = m_protocolType;
}
 
void
NrGtpuHeader::SetSequenceNumber(uint16_t m_sequenceNumber)
{
    this->m_sequenceNumber = m_sequenceNumber;
}
 
void
NrGtpuHeader::SetSequenceNumberFlag(bool m_sequenceNumberFlag)
{
    this->m_sequenceNumberFlag = m_sequenceNumberFlag;
}
 
void
NrGtpuHeader::SetTeid(uint32_t m_teid)
{
    this->m_teid = m_teid;
}
 
void
NrGtpuHeader::SetVersion(uint8_t m_version)
{
    // m_version is a uint3_t
    this->m_version = m_version & 0x7;
}
 
bool
NrGtpuHeader::operator==(const NrGtpuHeader& b) const
{
    return m_version == b.m_version && m_protocolType == b.m_protocolType &&
           m_extensionHeaderFlag == b.m_extensionHeaderFlag &&
           m_sequenceNumberFlag == b.m_sequenceNumberFlag &&
           m_nPduNumberFlag == b.m_nPduNumberFlag && m_messageType == b.m_messageType &&
           m_length == b.m_length && m_teid == b.m_teid && m_sequenceNumber == b.m_sequenceNumber &&
           m_nPduNumber == b.m_nPduNumber && m_nextExtensionType == b.m_nextExtensionType;
}
 
} // namespace ns3