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, uint32_t initDpdkArgc = 0, char **initDpdkArgv = NULL);

        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

#endif /* PCAPPP_DPDK_DEVICE_LIST */