UdpLayer.h

Go to the documentation of this file.

#ifndef PACKETPP_UDP_LAYER
#define PACKETPP_UDP_LAYER

#include "Layer.h"


namespace pcpp
{

#pragma pack(push,1)
    struct udphdr {
        uint16_t portSrc;
        uint16_t portDst;
        uint16_t length;
        uint16_t headerChecksum;
    };
#pragma pack(pop)


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

        UdpLayer(uint16_t portSrc, uint16_t portDst);

        udphdr* getUdpHeader() const { return (udphdr*)m_Data; }

        uint16_t getSrcPort() const;

        uint16_t getDstPort() const;

        uint16_t calculateChecksum(bool writeResultToPacket);

        // implement abstract methods

        void parseNextLayer();

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

        void computeCalculateFields();

        std::string toString() const;

        OsiModelLayer getOsiModelLayer() const { return OsiModelTransportLayer; }
    };

} // namespace pcpp

#endif /* PACKETPP_UDP_LAYER */