#include "TpktLayer.h"

#include "EndianPortable.h"

#include "CotpLayer.h"

#include "PayloadLayer.h"

#include <cstring>

#include <sstream>

namespace pcpp

{

  TpktLayer::TpktLayer(uint8_t version, uint16_t length)

  {

    m_DataLen = sizeof(tpkthdr);

    m_Data = new uint8_t[m_DataLen];

    memset(m_Data, 0, m_DataLen);

    tpkthdr* tpktHdr = getTpktHeader();

    tpktHdr->version = version;

    tpktHdr->reserved = 0;

    tpktHdr->length = htobe16(length);

    m_Protocol = TPKT;

  }

  uint8_t TpktLayer::getReserved() const

  {

    return getTpktHeader()->reserved;

  }

  uint8_t TpktLayer::getVersion() const

  {

    return getTpktHeader()->version;

  }

  uint16_t TpktLayer::getLength() const

  {

    return htobe16(getTpktHeader()->length);

  }

  void TpktLayer::setLength(uint16_t length) const

  {

    getTpktHeader()->length = htobe16(length);

  }

  void TpktLayer::setVersion(uint8_t version) const

  {

    getTpktHeader()->version = version;

  }

  std::string TpktLayer::toString() const

  {

    std::ostringstream versionStream;

    versionStream << std::to_string(getVersion());

    std::ostringstream lengthStream;

    lengthStream << std::to_string(getLength());

    return "TPKT Layer, version: " + versionStream.str() + ", length: " + lengthStream.str();

  }

  void TpktLayer::parseNextLayer()

  {

    size_t headerLen = getHeaderLen();

    if (m_DataLen <= headerLen)

      return;

    uint8_t* payload = m_Data + headerLen;

    size_t payloadLen = m_DataLen - headerLen;

    tryConstructNextLayerWithFallback<CotpLayer, PayloadLayer>(payload, payloadLen);

  }

}  // namespace pcpp