通过GetModuleHandleEx 得到函数调用者所在的DLL/EXE(转)

在有些情况下需要得到函数调用者的模块名字。比如你想限制你的某个函数只能被自己某个特定的DLL调用。 或者比如在异常处理中你想了解是那个DLL/EXE抛出了异常。

 

API函数_ReturnAddress 和GetModuleHandleEx 函数可以帮助我们达到这个目的。以下代码演示它们的用法：

1. void  ShowCallerModuleName()  
2. {  
3.     HMODULE  hCallerModule = NULL;  
4.     TCHAR  szModuleName[MAX_PATH] = _T( "" );  
5.     void  *callerAddress = _ReturnAddress();  
6.     if  (GetModuleHandleEx(GET_MODULE_HANDLE_EX_FLAG_FROM_ADDRESS, ( LPCTSTR )callerAddress, &hCallerModule))  
7.     {  
8.         GetModuleFileName(hCallerModule, szModuleName, ARRAYSIZE(szModuleName));  
9.         MessageBox(NULL, szModuleName, _T("Who is calling me?" ), MB_OK);  
10.     }  
11. } 
    

解释：

_ReturnAddress 可以给你所在函数的返回地址，也就是调用者的地址。

GetModuleHandleEx 可以给你某个地址所在的模块Handle 。 得到这个Handle 之后，就可以用GetModuleFileName 来得到模块的文件名了。

 

///

 

扩展：

 用在dll劫持的时候，在判断呼叫本DLL的主程序模块中，获取返回地址，可以在naked函数中用汇编获取返回地址

 

以下是我测试的代码，程序运行正确！11（可以在函数里面判断返回地址，这样可以对抗DLL劫持）

 

void __declspec(naked) test()
{


__asm pop returnaddress
char aa21[1024];
sprintf(aa21,"%08X",returnaddress);
MessageBox(NULL,aa21,NULL,NULL);
__asm jmp returnaddress

}