MBufRawPacket.h

#pragma once
 
// GCOVR_EXCL_START
 
#include <time.h>
#include "Packet.h"
#include "PointerVector.h"
 
struct rte_mbuf;
struct rte_mempool;
 
namespace pcpp
{
 
    class DpdkDevice;
#ifdef USE_DPDK_KNI
    class KniDevice;
#endif
 
#define MBUFRAWPACKET_OBJECT_TYPE 1
 
    class MBufRawPacket : public RawPacket
    {
        friend class DpdkDevice;
#ifdef USE_DPDK_KNI
        friend class KniDevice;
#endif
 
    protected:
        struct rte_mbuf* m_MBuf;
        struct rte_mempool* m_Mempool;
        uint16_t m_MbufDataSize;
        bool m_FreeMbuf;
 
        void setMBuf(struct rte_mbuf* mBuf, timespec timestamp);
        bool init(struct rte_mempool* mempool);
        bool initFromRawPacket(const RawPacket* rawPacket, struct rte_mempool* mempool);
 
    public:
        MBufRawPacket() : RawPacket(), m_MBuf(nullptr), m_Mempool(nullptr), m_MbufDataSize(0), m_FreeMbuf(true)
        {
            m_OwnsRawData = false;
        }
 
        virtual ~MBufRawPacket();
 
        MBufRawPacket(const MBufRawPacket& other);
 
        bool init(DpdkDevice* device);
 
#ifdef USE_DPDK_KNI
        bool init(KniDevice* device);
#endif
 
        bool initFromRawPacket(const RawPacket* rawPacket, DpdkDevice* device);
 
#ifdef USE_DPDK_KNI
        bool initFromRawPacket(const RawPacket* rawPacket, KniDevice* device);
#endif
 
        inline rte_mbuf* getMBuf()
        {
            return m_MBuf;
        }
 
        // overridden methods
 
        inline uint8_t getObjectType() const override
        {
            return MBUFRAWPACKET_OBJECT_TYPE;
        }
 
        MBufRawPacket& operator=(const MBufRawPacket& other);
 
        MBufRawPacket* clone() const override;
 
        void clear() override;
 
        void appendData(const uint8_t* dataToAppend, size_t dataToAppendLen) override;
 
        void insertData(int atIndex, const uint8_t* dataToInsert, size_t dataToInsertLen) override;
 
        bool removeData(int atIndex, size_t numOfBytesToRemove) override;
 
        bool reallocateData(size_t newBufferLength) override;
 
        inline void setFreeMbuf(bool val = true)
        {
            m_FreeMbuf = val;
        }
 
    protected:
        bool doSetRawData(const uint8_t* pRawData, int rawDataLen, bool takeOwnership, timespec timestamp,
                          LinkLayerType layerType, int frameLength) override;
    };
 
    using MBufRawPacketVector = PointerVector<MBufRawPacket>;
 
}  // namespace pcpp
 
// GCOVR_EXCL_STOP