DpdkDeviceList.h

Go to the documentation of this file.

#pragma once
 
// GCOVR_EXCL_START
 
#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 uint16_t m_MBufDataSize;
        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, uint16_t mMbufDataSize);
        static bool verifyHugePagesAndDpdkDriver();
 
        static int dpdkWorkerThreadStart(void* ptr);
 
    public:
        ~DpdkDeviceList();
 
        static DpdkDeviceList& getInstance()
        {
            static DpdkDeviceList instance;
            if (!instance.isInitialized())
                instance.initDpdkDevices(DpdkDeviceList::m_MBufPoolSizePerDevice, DpdkDeviceList::m_MBufDataSize);
 
            return instance;
        }
 
        static bool initDpdk(CoreMask coreMask, uint32_t mBufPoolSizePerDevice, uint16_t mBufDataSize = 0,
                             uint8_t masterCore = 0, uint32_t initDpdkArgc = 0, char** initDpdkArgv = nullptr,
                             const std::string& appName = "pcapplusplusapp", bool verifyHugePagesAndDriver = true);
 
        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(Logger::LogLevel logLevel);
 
        Logger::LogLevel getDpdkLogLevel() const;
 
        bool writeDpdkLogToFile(FILE* logFile);
 
        bool startDpdkWorkerThreads(CoreMask coreMask, std::vector<DpdkWorkerThread*>& workerThreadsVec);
 
        void stopDpdkWorkerThreads();
    };
 
}  // namespace pcpp
 
// GCOVR_EXCL_STOP