在DLL中获取主进程窗口句柄

最新推荐文章于 2023-07-16 11:30:28 发布

Entre_Lee

最新推荐文章于 2023-07-16 11:30:28 发布

阅读量5.7k

点赞数

分类专栏： C++ 文章标签： C++ dll 句柄

C++ 专栏收录该内容

13 篇文章 0 订阅

订阅专栏

有的时候难免需要在DLL中获取主进程的窗口句柄，比如在DLL注入的时候等等。那么如何在DLL中获取主进程的窗口句柄呢？可以通过EnumWindows来实现。先通过GetCurrentProcessId获取进程的PID，然后在EnumWindows中调用GetWindowThreadProcessId获得与窗口句柄关联的进程PID，然后对比PID，看是否相等，并判断是不是主窗口即可。
以上方法参考自网络，不一定很完善，但是通常情况下已经够用了。附上测试代码：

// Author: 代码疯子
// Blog: http://www.programlife.net/
#include <windows.h>
 
BOOL CALLBACK EnumWindowsProc(HWND hwnd, LPARAM lParam) 
{ 
	DWORD dwCurProcessId = *((DWORD*)lParam); 
	DWORD dwProcessId = 0; 
 
	GetWindowThreadProcessId(hwnd, &dwProcessId); 
	if(dwProcessId == dwCurProcessId && GetParent(hwnd) == NULL)
	{ 
		*((HWND *)lParam) = hwnd;
		return FALSE; 
	} 
	return TRUE; 
} 
 
 
HWND GetMainWindow() 
{ 
	DWORD dwCurrentProcessId = GetCurrentProcessId();
	if(!EnumWindows(EnumWindowsProc, (LPARAM)&dwCurrentProcessId)) 
	{     
		return (HWND)dwCurrentProcessId; 
	} 
	return NULL; 
} 
 
BOOLEAN WINAPI DllMain(
		IN HINSTANCE hDllHandle, 
		IN DWORD     nReason,    
		IN LPVOID    Reserved)
{
	BOOLEAN bSuccess = TRUE;
 
	switch ( nReason )
	{
	case DLL_PROCESS_ATTACH:
		MessageBox(GetMainWindow(), TEXT("OMG~ You are Attacked!"), TEXT("Warning"), MB_ICONWARNING);
		break;
 
	case DLL_PROCESS_DETACH:
		break;
	}
 
	return bSuccess;
}