VlanLayer.h

Go to the documentation of this file.

#ifndef PACKETPP_VLAN_LAYER
#define PACKETPP_VLAN_LAYER

#include "Layer.h"
#include "EthLayer.h"
#if defined(WIN32) || defined(WINx64)
#include <winsock2.h>
#endif


namespace pcpp
{

#pragma pack(push, 1)
    struct vlan_header {
#if (BYTE_ORDER == LITTLE_ENDIAN)

        uint16_t priority:3,
                 cfi:1,
                 vlanID:12;
#else
        uint16_t vlanID:12,
                 cfi:1,
                 priority:3;
#endif

        uint16_t etherType;
    };
#pragma pack(pop)


    class VlanLayer : public Layer
    {
    public:
        VlanLayer(uint8_t* data, size_t dataLen, Layer* prevLayer, Packet* packet) : Layer(data, dataLen, prevLayer, packet) { m_Protocol = VLAN; }

        VlanLayer(const uint16_t vlanID, bool cfi, uint8_t priority, uint16_t etherType);

        ~VlanLayer() {}

        inline vlan_header* getVlanHeader() { return (vlan_header*)m_Data; }

        inline uint16_t getVlanID() { return getVlanHeader()->vlanID >> 4;  }

        inline uint8_t getCFI() { return getVlanHeader()->vlanID & 0x1; }

        inline uint8_t getPriority() { return (getVlanHeader()->vlanID & 0xF) >> 1; }

        inline void setVlanID(uint16_t id) { getVlanHeader()->vlanID = ((id << 4) & 0xff0) | (id >> 8); }

        inline void setCFI(bool cfi) { getVlanHeader()->vlanID |= cfi; }

        inline void setPriority(uint8_t priority) { getVlanHeader()->vlanID |= ((priority & 0x7) << 1); }

        // implement abstract methods

        void parseNextLayer();

        inline size_t getHeaderLen() { return sizeof(vlan_header); }

        void computeCalculateFields() {}

        std::string toString();

        OsiModelLayer getOsiModelLayer() { return OsiModelDataLinkLayer; }
    };

} // namespace pcpp

#endif /* PACKETPP_VLAN_LAYER */