TpktLayer.h

Go to the documentation of this file.

#ifndef PACKETPP_TPKT_LAYER
#define PACKETPP_TPKT_LAYER

#include "EthLayer.h"
#include "Layer.h"


namespace pcpp
{

#pragma pack(push, 1)
    struct tpkthdr
    {
        uint8_t version;
        uint8_t reserved;
        uint16_t length;
    };
#pragma pack(pop)

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

        TpktLayer(uint8_t version, uint16_t length);

        virtual ~TpktLayer() {}

        uint8_t getReserved() const;

        uint8_t getVersion() const;

        uint16_t getLength() const;

        void setVersion(uint8_t version) const;

        void setLength(uint16_t length) const;

        size_t getHeaderLen() const override { return sizeof(tpkthdr); }

        void computeCalculateFields() override {}

        void parseNextLayer() override;

        static bool isTpktPort(uint16_t portSrc, uint16_t portDst) { return portSrc == 102 || portDst == 102; }

        static bool isDataValid(const uint8_t *data, size_t dataSize) { return data && dataSize >= sizeof(tpkthdr); }

        std::string toString() const override;

        OsiModelLayer getOsiModelLayer() const override { return OsiModelTransportLayer; }

      private:
        tpkthdr *getTpktHeader() const { return (tpkthdr *)m_Data; }
    };

} // namespace pcpp
#endif // PACKETPP_TPKT_LAYER