[C++] GetCurrentModule - Obtain a Module Handle HMODULE to the current DLL

hmodule.cpp

const HMODULE GetCurrentModuleLegacy()
{
	// Query Against Myself
	MEMORY_BASIC_INFORMATION mbi = {};
	if (0 == VirtualQuery(GetCurrentModuleLegacy, &mbi, sizeof(mbi)))
	{
		// See GetLastError()
		return NULL;
	}
	return (HMODULE)mbi.AllocationBase;
}


const HMODULE GetCurrentModule()
{
	HMODULE hModule = NULL;
	// hModule is NULL if GetModuleHandleEx fails.
	GetModuleHandleEx(GET_MODULE_HANDLE_EX_FLAG_FROM_ADDRESS
						| GET_MODULE_HANDLE_EX_FLAG_UNCHANGED_REFCOUNT,
						(LPCTSTR)GetCurrentModule, &hModule);
	return hModule;
}


const HMODULE GetModuleHandle()
{
	// g_hModule set by DllMain
	if (NULL == g_hModule)
	{
		return GetCurrentModule();
	}
	return g_hModule;
}