IPSecLayer.h

Go to the documentation of this file.

#ifndef PACKETPP_IPSEC_LAYER
#define PACKETPP_IPSEC_LAYER


#include "Layer.h"

namespace pcpp
{
#pragma pack(push, 1)
    struct ipsec_authentication_header
    {
        uint8_t nextHeader;
        uint8_t payloadLen;
        uint16_t reserved;
        uint32_t spi;
        uint32_t sequenceNumber;
    };
#pragma pack(pop)

#pragma pack(push, 1)
    struct ipsec_esp
    {
        uint32_t spi;
        uint32_t sequenceNumber;
    };
#pragma pack(pop)

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

        ipsec_authentication_header* getAHHeader() const { return (ipsec_authentication_header*)m_Data; }

        uint32_t getSPI() const;

        uint32_t getSequenceNumber() const;

        size_t getICVLength() const;

        uint8_t* getICVBytes() const;

        std::string getICVHexStream() const;

        static inline bool isDataValid(const uint8_t* data, size_t dataLen);

        // implement abstract methods

        size_t getHeaderLen() const { return 4*(getAHHeader()->payloadLen + 2); }

        void parseNextLayer();

        void computeCalculateFields() {}

        std::string toString() const;

        OsiModelLayer getOsiModelLayer() const { return OsiModelNetworkLayer; }

    private:
        // this layer supports parsing only
        AuthenticationHeaderLayer() {}
    };



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

        ipsec_esp* getESPHeader() const { return (ipsec_esp*)m_Data; }

        uint32_t getSPI() const;

        uint32_t getSequenceNumber() const;

        static inline bool isDataValid(const uint8_t* data, size_t dataLen);

        // implement abstract methods

        size_t getHeaderLen() const { return sizeof(ipsec_esp); }

        void parseNextLayer();

        void computeCalculateFields() {}

        std::string toString() const;

        OsiModelLayer getOsiModelLayer() const { return OsiModelTransportLayer; }

    private:
        // this layer supports parsing only
        ESPLayer() {}
    };


    // implementation of inline methods

    bool AuthenticationHeaderLayer::isDataValid(const uint8_t* data, size_t dataLen)
    {
        if (dataLen < sizeof(ipsec_authentication_header))
            return false;

        size_t payloadLen = 4 * (data[1] + 2);
        if (payloadLen < sizeof(ipsec_authentication_header) || payloadLen > dataLen)
            return false;

        return true;
    }

    bool ESPLayer::isDataValid(const uint8_t* data, size_t dataLen)
    {
        return dataLen >= sizeof(ipsec_esp);
    }
}

#endif // PACKETPP_IPSEC_LAYER