UdpLayer.h

Go to the documentation of this file.

#pragma once
 
#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)
    static_assert(sizeof(udphdr) == 8, "udphdr size is not 8 bytes");
 
    class UdpLayer : public Layer
    {
    public:
        UdpLayer(uint8_t* data, size_t dataLen, Layer* prevLayer, Packet* packet)
            : Layer(data, dataLen, prevLayer, packet, UDP)
        {}
 
        UdpLayer(uint16_t portSrc, uint16_t portDst);
 
        udphdr* getUdpHeader() const
        {
            return reinterpret_cast<udphdr*>(m_Data);
        }
 
        uint16_t getSrcPort() const;
 
        uint16_t getDstPort() const;
 
        uint16_t calculateChecksum(bool writeResultToPacket);
 
        static inline bool isDataValid(const uint8_t* data, size_t dataLen);
 
        // implement abstract methods
 
        void parseNextLayer() override;
 
        size_t getHeaderLen() const override
        {
            return sizeof(udphdr);
        }
 
        void computeCalculateFields() override;
 
        std::string toString() const override;
 
        OsiModelLayer getOsiModelLayer() const override
        {
            return OsiModelTransportLayer;
        }
    };
 
    bool UdpLayer::isDataValid(const uint8_t* data, size_t dataLen)
    {
        return data && dataLen >= sizeof(udphdr);
    }
}  // namespace pcpp