首页 > 代码库 > 通过GetModuleHandleEx 得到函数调用者所在的DLL/EXE(转)

通过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

}

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