浮生猎趣

ls *.fun >> 罗列一切乐趣

在程序运行时获取被调用的DLL的本地文件路径

| Comments

很多时候,DLL都是被别的进程所调用,所以在运行时,下面方法获取到的实际上是EXE所在的文件路径:

1
  QCoreApplication::applicationFilePath();//Qt

但是我们实际上想要获得的是DLL本身的路径,因为运行的EXE与被调用的DLL不一定在同一个目录,比如,当把abc.dll注入到notepad.exe进程,很明显abc.dll是不在notepad所在的目录的。在dll代码中调用上述语句,返回的实际上是notepad的EXE路径。

那么,我们要获得abc.dll真实路径,有两个方法。

方法一:

在DLL的入口函数附近,声明以下变量,并定义函数:

1
2
3
4
5
6
7
8
  EXTERN_C IMAGE_DOS_HEADER __ImageBase;

  VOID CALLBACK abcAPCProc(ULONG_PTR dwParam)
  {
      WCHAR path[MAX_PATH + 1] = {0};
      ::GetModuleFileName((HINSTANCE)&__ImageBase, path, MAX_PATH);
      //path就是当前dll的文件路径
  }

然后,需要在导出函数里调用一下上面的函数:

1
2
3
4
5
6
7
8
9
10
11
12
13
  extern "C" DWORD __declspec(dllexport) __stdcall showABC(void *)
  {
      HANDLE hThread = NULL;
      DWORD dwRes = 0;
      hThread = ::OpenThread(THREAD_SET_CONTEXT,
          FALSE, getMainThreadId(0));
      DW_ASSERT(NULL != hThread);
      dwRes = ::QueueUserAPC(abcAPCProc,
          hThread, 0);
      ::CloseHandle(hThread);
      hThread = NULL;
      return dwRes;
  }

方法二:

定义以下函数:

1
2
3
4
5
6
7
8
9
10
  HMODULE GetCurrentModule()
  { // NB: XP+ solution!
      HMODULE hModule = NULL;
      GetModuleHandleEx(
          GET_MODULE_HANDLE_EX_FLAG_FROM_ADDRESS,
          (LPCTSTR)GetCurrentModule,
          &hModule);
  
      return hModule;
  }

调用上面函数,获取当前局部,然后就可以使用GetModuleFileName方法获取DLL路径:

1
2
3
  WCHAR path[MAX_PATH + 1] = {0};
  HMODULE hm = GetCurrentModule();
  ::GetModuleFileName(hm, path, MAX_PATH);
原文作者: lslin
原文链接: http://blog.lessfun.com/blog/2014/03/18/get-dll-path-at-runtime/
版权声明:自由转载-非商用-非衍生-保持署名 | Creative Commons BY-NC-ND 3.0

发表评论