#include "VxlanLayer.h"

#include "EthLayer.h"

#include "PayloadLayer.h"

#include "EndianPortable.h"

#include <cstring>

namespace pcpp

{

  VxlanLayer::VxlanLayer(uint32_t vni, uint16_t groupPolicyID, bool setGbpFlag, bool setPolicyAppliedFlag,

                         bool setDontLearnFlag)

  {

    allocData(sizeof(vxlan_header));

    m_Protocol = VXLAN;

    if (vni != 0)

      setVNI(vni);

    vxlan_header* vxlanHeader = getVxlanHeader();

    if (groupPolicyID != 0)

      vxlanHeader->groupPolicyID = htobe16(groupPolicyID);

    vxlanHeader->vniPresentFlag = 1;

    if (setGbpFlag)

      vxlanHeader->gbpFlag = 1;

    if (setPolicyAppliedFlag)

      vxlanHeader->policyAppliedFlag = 1;

    if (setDontLearnFlag)

      vxlanHeader->dontLearnFlag = 1;

  }

  uint32_t VxlanLayer::getVNI() const

  {

    return (be32toh(getVxlanHeader()->vni) >> 8);

  }

  void VxlanLayer::setVNI(uint32_t vni)

  {

    getVxlanHeader()->vni = htobe32(vni << 8);

  }

  std::string VxlanLayer::toString() const

  {

    return "VXLAN Layer";

  }

  void VxlanLayer::parseNextLayer()

  {

    if (m_DataLen <= sizeof(vxlan_header))

      return;

    auto payload = m_Data + sizeof(vxlan_header);

    auto payloadLen = m_DataLen - sizeof(vxlan_header);

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

  }

  const FieldDescriptor VxlanLayer::SerializedFields::ValidVNI{ Layer::SerializedFields::MaxID + 1, "validVNI" };

  const FieldDescriptor VxlanLayer::SerializedFields::GroupPolicyID{ Layer::SerializedFields::MaxID + 2,

                                                                   "groupPolicyID" };

  const FieldDescriptor VxlanLayer::SerializedFields::VNI{ Layer::SerializedFields::MaxID + 3, "vni" };

  void VxlanLayer::serializeLayer(ObjectScope& serializer) const

  {

    auto* header = getVxlanHeader();

    serializer.writeField(SerializedFields::ValidVNI, header->vniPresentFlag ? true : false);

    serializer.writeField(SerializedFields::GroupPolicyID, be16toh(header->groupPolicyID));

    serializer.writeField(SerializedFields::VNI, getVNI());

  }

}  // namespace pcpp