Làm cách nào để lấy mã HMODULE cho mã hiện đang thực thi?

Question

Tôi có một thư viện tĩnh có thể được liên kết với một trong số .exe hoặc .dll. Khi chạy, tôi muốn ony của các hàm thư viện của mình nhận được HMODULE cho bất kỳ điều gì mà mã thư viện tĩnh đã được liên kết.

Tôi hiện đang sử dụng các thủ thuật sau đây (lấy cảm hứng từ this forum):

const HMODULE GetCurrentModule() 
{ 
    MEMORY_BASIC_INFORMATION mbi = {0}; 
    ::VirtualQuery(GetCurrentModule, &mbi, sizeof(mbi)); 

    return reinterpret_cast<HMODULE>(mbi.AllocationBase); 
} 

Có cách nào tốt hơn để làm điều này mà không trông rất hacky?

(Lưu ý: Mục đích của việc này là để tải một số tài nguyên Win32 mà tôi biết người dùng của tôi sẽ được liên kết trong cùng lúc như thư viện tĩnh của tôi.)

Answer 1

HMODULE GetCurrentModule() 
{ // NB: XP+ solution! 
    HMODULE hModule = NULL; 
    GetModuleHandleEx(
    GET_MODULE_HANDLE_EX_FLAG_FROM_ADDRESS, 
    (LPCTSTR)GetCurrentModule, 
    &hModule); 

    return hModule; 
} 

Answer 2

Tôi muốn nhìn vào GetModuleHandleEx() sử dụng cờ GET_MODULE_HANDLE_EX_FLAG_FROM_ADDRESS. Có vẻ như bạn có thể thay đổi số GetCurrentModule() để gọi thường lệ này thay vì VirtualQuery() và chuyển địa chỉ của số GetCurrentModule() làm đối số lpModuleName.

ETA:

const HMODULE GetCurrentModule() 
{ 
    DWORD flags = GET_MODULE_HANDLE_EX_FLAG_FROM_ADDRESS; 
    HMODULE hm = 0; 
    ::GetModuleHandleEx(flags, reinterpret_cast<LPCTSTR>(GetCurrentModule), &hm); 
    return hm; 
} 

Tôi không thử nó, nhưng tôi nghĩ rằng tôi sẽ làm những gì bạn muốn.

Answer 3

HMODULE là HINSTANCE là địa chỉ cơ sở của mô-đun. Vì vậy, tôi muốn xem nó hoạt động như thế nào. Nhưng nếu tất cả những gì bạn muốn là HMODULE của file thực thi, thì tại sao không liệt kê tất cả HMODULE trong quá trình (EnumProcessModules). Một trong số họ sẽ có .lib của bạn được liên kết.

Giới hạn mà tôi thấy là bạn không biết DLL hoặc EXE của bạn .lib đến từ đâu. Bạn có thể muốn so sánh các HMODULE (các địa chỉ cơ sở) với _ReturnAddress bạn nhận được từ .lib của bạn. lib của bạn sẽ thuộc về HMODLUE cao nhất nhỏ hơn _ReturnAddress bạn

Answer 4

__ImageBase là một mối liên kết tạo ra biểu tượng đó là tiêu đề DOS của module (MSVC chỉ). Từ đó, bạn có thể truyền địa chỉ của nó tới số HINSTANCE hoặc HMODULE. Vì vậy, nó thuận tiện hơn là đi qua một API.

Vì vậy, bạn chỉ cần làm điều này:

EXTERN_C IMAGE_DOS_HEADER __ImageBase; 
#define HINST_THISCOMPONENT ((HINSTANCE)&__ImageBase) 

Từ http://blogs.msdn.com/oldnewthing/archive/2004/10/25/247180.aspx