DLL獲取自己的模組控制代碼的方法
這幾天看了一下window核心程式設計,第22章有一個例子使用遠端呼叫注入Dll的。其中注入Dll的時候載入dll的程序呼叫VirtualQuery查詢程序虛擬空間得到程序載入的所有模組路徑。
但是,查詢程式碼很奇怪,於是翻看文件,VirtualQuery說明中沒有說到過任何與dll有關的話題,但作者又如何肯定可以讀出dll的資訊呢?看來值得深入研究一下。
首先看一下程式碼:
- #include <Windows.h>
- #include <fstream>
- #include <tchar.h>
- int WINAPI DllMain(HINSTANCE hDllHandle,
- {
- if (dwReason == DLL_PROCESS_ATTACH)
- {
- std::wfstream file;
- file.open(_T("D://SeeIn.txt"), std::ios_base::out);
- if (!file)
- {
- return FALSE;
- }
- MEMORY_BASIC_INFORMATION memory;
- memset(&memory, 0, sizeof(memory));
- DWORD
- while(VirtualQuery((LPVOID)pFind, &memory, sizeof(memory)) == sizeof(memory))
- {
- if (memory.State == MEM_FREE)
- {
- memory.AllocationBase = memory.BaseAddress;
- }
- if (memory.BaseAddress == hDllHandle ||
- memory.AllocationBase != memory.BaseAddress ||
- memory.AllocationBase == 0)
- {
- pFind += memory.RegionSize;
- memset(&memory, 0, sizeof(memory));
- continue;
- }
- else
- {
- TCHAR szName[MAX_PATH] = {0};
- if (0 != GetModuleFileName((HMODULE)memory.AllocationBase, szName, sizeof(szName) / sizeof(TCHAR)))
- {
- file.write(szName, sizeof(szName));
- //file.write(_T("/r/n"), sizeof(_T("/r/n")));
- file.clear();
- }
- }
- pFind += memory.RegionSize;
- memset(&memory, 0, sizeof(memory));
- }
- file.close();
- }
- return TRUE;
- }
其中可以看出,AllocationBase == BaseAddress的時候,這個頁是一個dll模組載入進來的時候分配的,檢視文件,對於AllocationBase 是這麼說的:
A pointer to the base address of a range of pages allocated by the VirtualAlloc function. The page pointed to by the BaseAddress member is contained within this allocation range.
翻譯:一個指向一系列範圍內的頁的基地址的指標,這些頁是通過VirtualAlloc 函式分配的。被BaseAddress 成員指向的頁被包含在這個分配範圍內。
通過程式碼和文件,看一看出,AllocationBase 是頁的基地址,BaseAddress 也是頁的基地址,唯一不同的是,BaseAddress 是VirtualAlloc 分配記憶體後第一頁的地址,AllocationBase 是每一頁的地址。
比如假設windows為每一塊由VirtualAlloc 分配的記憶體準備了若干頁:p1,p2,p3,p4,……那麼,BaseAddress 就是p1的基地址,而AllocationBase 就分別是p1,p2,p3,p4,……的基地址。
當然,要想獲得每一個頁的AllocationBase ,那就必須不斷的呼叫VirtualQuery遍歷這些頁。
還有一個重要的地方,從這個程式看出,程序載入dll記憶體空間是由VirtualAlloc 分配的。
綜上,當AllocationBase == BaseAddress的時候,AllocationBase 就是模組在程序記憶體中的起始地址,即HMODULE。
至此,VirtualQuery應該是比較明瞭了。
通俗說VirtualQuery的功能就是:以頁為單位,遍歷程序虛擬空間。
一點理解,高手路過請指正!