dll parser

#pragma once
#ifndef _DLL_PARSER_H_
#define  _DLL_PARSER_H_
#include <string>
#include <map>
#include <functional>
#include <Windows.h>

class DLLParser
{
public:
    DLLParser() : m_hMod(nullptr)
    {
    }

    ~DLLParser() 
    {
        UnLoad();
    }

    bool Load(const std::string &dll_path)
    {
        m_hMod = LoadLibraryA(dll_path.data());
        if (nullptr == m_hMod) {
            printf("LoadLibrary fail\n");
            return false;
        }
        return true;
    }

    bool UnLoad()
    {
        if (m_hMod == nullptr) 
        {
            return true;
        }
        auto b = FreeLibrary(m_hMod);
        if (!b)return false;
        m_hMod = nullptr;
        return true;
    }

    template<typename T>
    std::function<T> GetFunction(const std::string &func_name)
    {
        auto it = m_map.find(func_name);
        if (it == m_map.end())
        {
            auto addr = GetProcAddress(m_hMod, func_name.c_str());
            if (!addr) return false;
            m_map.insert(std::make_pair(func_name, addr));
            it = m_map.find(func_name);
        }
        return std::function<T>((T*)(it->second));
    }

    template<typename T, typename... Args>
    typename std::result_of<std::function<T>(Args...)>::type ExceuteFunc
    (const std::string &func_name, Args&&... args)
    {
        auto f = GetFunction<T>(func_name);
        if (f == nullptr)
        {
            std::string exception_str = "can not find this function " + func_name;
            throw std::exception(exception_str.c_str());
        }
        return f(std::forward<Args>(args)...);
    }
private:
    HMODULE m_hMod;
    std::map<std::string, FARPROC> m_map;
};

#endif // _DLL_PARSER_H_

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章