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() = 0;
    };


    class DpdkDeviceList
    {
    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();

        inline bool isInitialized() { return (m_IsInitialized && m_IsDpdkInitialized); }
        bool initDpdkDevices(uint32_t mBufPoolSizePerDevice);
        static bool verifyHugePagesAndDpdkDriver();

        static int dpdkWorkerThreadStart(void *ptr);
    public:

        ~DpdkDeviceList();

        static inline 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);

        DpdkDevice* getDeviceByPciAddress(const PciAddress& pciAddr);

        inline const std::vector<DpdkDevice*>& getDpdkDeviceList() { return m_DpdkDeviceList; }

        SystemCore getDpdkMasterCore();

        void setDpdkLogLevel(LoggerPP::LogLevel logLevel);

        LoggerPP::LogLevel getDpdkLogLevel();

        bool writeDpdkLogToFile(FILE* logFile);

        bool startDpdkWorkerThreads(CoreMask coreMask, std::vector<DpdkWorkerThread*>& workerThreadsVec);

        void stopDpdkWorkerThreads();
    };

} // namespace pcpp

#endif /* PCAPPP_DPDK_DEVICE_LIST */