MBufRawPacket.h

#ifndef PCAPPP_MBUF_RAW_PACKET
#define PCAPPP_MBUF_RAW_PACKET

#include <time.h>
#include "Packet.h"
#include "PointerVector.h"

struct rte_mbuf;
struct rte_mempool;

namespace pcpp
{

    class DpdkDevice;
    class KniDevice;

    #define MBUFRAWPACKET_OBJECT_TYPE 1

    class MBufRawPacket : public RawPacket
    {
        friend class DpdkDevice;
        friend class KniDevice;
        static const int MBUF_DATA_SIZE;

    protected:
        struct rte_mbuf* m_MBuf;
        struct rte_mempool* m_Mempool;
        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(NULL), m_Mempool(NULL), m_FreeMbuf(true) { m_DeleteRawDataAtDestructor = false; }

        virtual ~MBufRawPacket();

        MBufRawPacket(const MBufRawPacket& other);

        bool init(DpdkDevice* device);
        bool init(KniDevice* device);

        bool initFromRawPacket(const RawPacket* rawPacket, DpdkDevice* device);
        bool initFromRawPacket(const RawPacket* rawPacket, KniDevice* device);

        inline rte_mbuf* getMBuf() { return m_MBuf; }

        // overridden methods

        virtual inline uint8_t getObjectType() const { return MBUFRAWPACKET_OBJECT_TYPE; }

        MBufRawPacket& operator=(const MBufRawPacket& other);

        bool setRawData(const uint8_t* pRawData, int rawDataLen, timespec timestamp, LinkLayerType layerType = LINKTYPE_ETHERNET, int frameLength = -1);

        void clear();

        void appendData(const uint8_t* dataToAppend, size_t dataToAppendLen);

        void insertData(int atIndex, const uint8_t* dataToInsert, size_t dataToInsertLen);

        bool removeData(int atIndex, size_t numOfBytesToRemove);

        bool reallocateData(size_t newBufferLength);

        inline void setFreeMbuf(bool val = true) { m_FreeMbuf = val; }
    };

    typedef PointerVector<MBufRawPacket> MBufRawPacketVector;

} // namespace pcpp

#endif /* PCAPPP_MBUF_RAW_PACKET */