DpdkDeviceList.h

Go to the documentation of this file.

#ifndef PCAPPP_DPDK_DEVICE_LIST
#define PCAPPP_DPDK_DEVICE_LIST
 
#include "SystemUtils.h"
#include "DpdkDevice.h"
#include "Logger.h"
#include <vector>
 
namespace pcpp
{
 
    class DpdkWorkerThread
    {
    public:
        virtual ~DpdkWorkerThread() {}
 
        virtual bool run(uint32_t coreId) = 0;
 
        virtual void stop() = 0;
 
        virtual uint32_t getCoreId() const = 0;
    };
 
    class KniDeviceList;
 
    class DpdkDeviceList
    {
        friend class KniDeviceList;
    private:
        bool m_IsInitialized;
        static bool m_IsDpdkInitialized;
        static uint32_t m_MBufPoolSizePerDevice;
        static CoreMask m_CoreMask;
        std::vector<DpdkDevice*> m_DpdkDeviceList;
        std::vector<DpdkWorkerThread*> m_WorkerThreads;
 
        DpdkDeviceList();
 
        bool isInitialized() const { return (m_IsInitialized && m_IsDpdkInitialized); }
        bool initDpdkDevices(uint32_t mBufPoolSizePerDevice);
        static bool verifyHugePagesAndDpdkDriver();
 
        static int dpdkWorkerThreadStart(void* ptr);
    public:
 
        ~DpdkDeviceList();
 
        static DpdkDeviceList& getInstance()
        {
            static DpdkDeviceList instance;
            if (!instance.isInitialized())
                instance.initDpdkDevices(DpdkDeviceList::m_MBufPoolSizePerDevice);
 
            return instance;
        }
 
        static bool initDpdk(CoreMask coreMask, uint32_t mBufPoolSizePerDevice, uint8_t masterCore = 0);
 
        DpdkDevice* getDeviceByPort(int portId) const;
 
        DpdkDevice* getDeviceByPciAddress(const std::string& pciAddr) const;
 
        const std::vector<DpdkDevice*>& getDpdkDeviceList() const { return m_DpdkDeviceList; }
 
        SystemCore getDpdkMasterCore() const;
 
        void setDpdkLogLevel(LoggerPP::LogLevel logLevel);
 
        LoggerPP::LogLevel getDpdkLogLevel() const;
 
        bool writeDpdkLogToFile(FILE* logFile);
 
        bool startDpdkWorkerThreads(CoreMask coreMask, std::vector<DpdkWorkerThread*>& workerThreadsVec);
 
        void stopDpdkWorkerThreads();
    };
 
} // namespace pcpp
 
#endif /* PCAPPP_DPDK_DEVICE_LIST */