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 */