Submitted by 李马
您可以任意转载这篇文章,但请在转载时注明原始链接和作者,谢谢。
在一般情况下,动态调用DLL导出函数的方法是:
- 用typedef为目标函数定义函数指针类型。
- 用GetProcAddress获取函数指针。
- 用函数指针进行调用。
但是,如果要调用的函数太多的话,这个方法难免流于繁琐——有太多的typedef、太多的GetProcAddress和太多的函数指针。在本文中将给出一个通用的解决方法,使这些动态调用更加简便。
先看看我们这个函数的声明:
C++代码
BOOL __cdecl DllCall(PCTSTR lpszDll, // 目标函数所在DLL的名称
PCSTR lpszFunc, // 目标函数名称
int argc, // 要调用的参数个数
PVOID pRet, // 函数调用的返回值
...); //函数传参数
以MessageBoxA为例,使用方法为:
C++代码
int ret;
DllCall(_T("user32.dll"), "MessageBoxA", 4, &ret,
NULL, "Hello, World!", "Hello", MB_ICONINFORMATION | MB_YESNO);
换用一个参数的MessageBoxIndirectA,则是:
C++代码
MSGBOXPARAMSA param;
ZeroMemory(¶m, sizeof(MSGBOXPARAMSA));
param.cbSize = sizeof(MSGBOXPARAMSA);
param.dwLanguageId = GetSystemDefaultLangID();
param.dwStyle = MB_ICONINFORMATION;
param.lpszCaption = "Hello";
param.lpszText = "Hello, World";
int ret;
DllCall(_T("user32.dll"), "MessageBoxIndirectA", 1, &ret, ¶m);
实现的原理是动态生成汇编代码,也就是类似这样的一段:
C++代码
__declspec(naked) DWORD __cdecl DllCallProc(void)
{
__asm
{
push argn
...
push arg2
push arg1
call proc
ret
};
}
下面列出DllCall的代码,和所有可变参数函数的实现(如sprintf)都差不多。
C++代码
BOOL __cdecl DllCall(PCTSTR lpszDll,
PCSTR lpszFunc,
int argc,
PVOID pRet,
...)
{
va_list arglist;
int ret;
va_start(arglist, pRet);
ret = vDllCall(lpszDll, lpszFunc, argc, pRet, arglist);
va_end(arglist);
return ret;
}
最为关键的就是vDllCall的代码了,如下:
C++代码
#pragma pack(push, 1)
typedef struct {
BYTE op;
DWORD_PTR dwValue;
} OPCODE, *POPCODE;
#pragma pack(pop)
typedef DWORD (__cdecl * DLLCALL)(void);
BOOL __cdecl CRfidNfc::vDllCall(PCTSTR lpszDll,
PCSTR lpszFunc,
int argc,
PVOID pRet,
va_list arglist)
{
HMODULE hDll = LoadLibrary(lpszDll);
if (NULL == hDll)
return FALSE;
FARPROC proc = GetProcAddress(hDll, lpszFunc);
if (NULL == proc)
return FALSE;
HANDLE hHeap = GetProcessHeap();
POPCODE p = (POPCODE)HeapAlloc(hHeap, 0, sizeof(OPCODE) * (argc + 2));
int i;
for (i = argc - 1; i >= 0; --i)
{
// push arg[i]
p[i].op = 0x68;
p[i].dwValue = va_arg(arglist, DWORD_PTR);
}
// call proc
p[argc].op = 0xe8;
p[argc].dwValue = (INT_PTR)proc - (INT_PTR)&p[argc + 1];
// ret
p[argc + 1].op = 0xc3;
p[argc + 1].dwValue = 0x90909090; // nop nop nop nop
DLLCALL pfn = (DLLCALL)p;
DWORD ret = pfn();
HeapFree(hHeap, 0, p);
FreeLibrary(hDll);
if (NULL != pRet)
*(PDWORD)pRet = ret;
return TRUE;
}
其中的指针p就是我们动态生成的调用代码,最后转换成DLLCALL类型的函数指针进行了调用。
最后,需要补充说明四点:
- DllCall只适用于__stdcall调用约定的目标函数。
- 这份vDllCall的代码只适用于x86的CPU,如果在WinCE的环境下(如arm或mips的CPU)使用,需要酌情重新编写动态调用的汇编代码。
- 其中argc参数是指实际压栈的参数个数,而不是C语言调用的参数个数。如API函数WindowFromPoint,虽然函数声明中只有一个参数,但是实际上是将POINT::x、POINT::y分别压栈的。在这种情况下,需要将argc设置为2。
- DllCall的返回值只获取了eax,如果有的函数会返回一个超过4字节的庞大结构,那么这个返回值将并不是你想要的。