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_

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值