#define LOG_MODULE PacketLogModulePayloadLayer

#include "PayloadLayer.h"

#include "GeneralUtils.h"

#include <string.h>

#include <sstream>

namespace pcpp

{

PayloadLayer::PayloadLayer(const uint8_t* data, size_t dataLen, bool) : Layer()

{

  m_Data = new uint8_t[dataLen];

  memcpy(m_Data, data, dataLen);

  m_DataLen = dataLen;

  m_Protocol = GenericPayload;

}

PayloadLayer::PayloadLayer(const std::string& payloadAsHexStream)

{

  m_DataLen = payloadAsHexStream.length() / 2;

  m_Data = new uint8_t[m_DataLen];

  m_Protocol = GenericPayload;

  if (hexStringToByteArray(payloadAsHexStream, m_Data, m_DataLen) == 0)

  {

    delete [] m_Data;

    m_Data = nullptr;

    m_DataLen = 0;

  }

}

void PayloadLayer::setPayload(const uint8_t* newPayload, size_t newPayloadLength)

{

  if (newPayloadLength < m_DataLen)

  {

    // shorten payload layer

    shortenLayer(newPayloadLength, m_DataLen - newPayloadLength);

  }

  else if (newPayloadLength > m_DataLen)

  {

    // extend payload layer

    extendLayer(m_DataLen, newPayloadLength - m_DataLen);

  }

  // and copy data to layer

  // this is also executed if the newPayloadLength == m_DataLen

  memcpy(m_Data, newPayload, newPayloadLength);

}

std::string PayloadLayer::toString() const

{

  std::ostringstream dataLenStream;

  dataLenStream << m_DataLen;

  return "Payload Layer, Data length: " + dataLenStream.str() + " [Bytes]";

}

} // namespace pcpp