UdpLayer.h

Go to the documentation of this file.

#pragma once
 
#include "Layer.h"
#include <algorithm>
#include <iterator>
 
 
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;
        }
 
        struct SerializedFields : Layer::SerializedFields
        {
            static std::vector<FieldDescriptor> all()
            {
                auto result = Layer::SerializedFields::all();
                std::initializer_list<FieldDescriptor> extra{ SrcPort, DstPort, Checksum };
                std::copy(extra.begin(), extra.end(), std::back_inserter(result));
                return result;
            }
 
            static const FieldDescriptor SrcPort;
 
            static const FieldDescriptor DstPort;
 
            static const FieldDescriptor Checksum;
        };
 
    protected:
        void serializeLayer(ObjectScope& serializer) const override;
    };
 
    bool UdpLayer::isDataValid(const uint8_t* data, size_t dataLen)
    {
        return data && dataLen >= sizeof(udphdr);
    }
}  // namespace pcpp