C++11 DllParser

最新推荐文章于 2021-07-20 21:22:55 发布

小蚂蚁_CrkRes

最新推荐文章于 2021-07-20 21:22:55 发布

阅读量394

点赞数

CC 4.0 BY-SA版权

分类专栏： c++

本文链接：https://blog.youkuaiyun.com/zang141588761/article/details/85260396

c++ 专栏收录该内容

68 篇文章

订阅专栏

本文介绍了一个DLL解析器的设计与实现，通过C++代码展示了如何加载和卸载DLL，获取DLL中的函数地址，并安全地调用这些函数。解析器使用了模板和函数对象来提高代码的通用性和灵活性。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >


#include "pch.h"
#include <iostream>
#include <Windows.h>
#include <string>
#include <map>
#include <functional>

using namespace std;
class DllParser
{
public:
	DllParser():m_hMod(nullptr)
	{
			
	}
	~DllParser()
	{
		UnLoad();
	}

	bool Load(const std::string & dllPath)
	{
		m_hMod = LoadLibraryA(dllPath.data());
		if (nullptr == m_hMod)
		{
			cout << "LoadLibrary failed" << endl;
		}
		return true;
	}

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

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

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

int main()
{
	/*
		类似于如下调用
		auto max = ExcecuteFunc<int(int,int)>("Max",5,8);
		auto ret = ExecuteFunc<int(int)><"Get",5);	
	*/
}