通过进程ID获得该进程主窗口的句柄

最新推荐文章于 2024-11-06 14:08:30 发布

凭谁问

最新推荐文章于 2024-11-06 14:08:30 发布

阅读量3.3k

点赞数

分类专栏： C++ WINDOWS VS

C++ 同时被 2 个专栏收录

48 篇文章 0 订阅

订阅专栏

WINDOWS VS

40 篇文章 0 订阅

订阅专栏

本文介绍了一种改进的方法来获取指定进程的主窗口，包括通过EnumWindows函数遍历所有窗口并使用WNDINFO结构体来存储进程ID和窗口句柄，通过增加DLL窗口句柄的检查逻辑来确保正确识别主窗口。实验结果显示，修改后的函数能够准确地返回宿主窗口句柄，即使在DLL注入场景下。

摘要由CSDN通过智能技术生成

一个进程可以拥有很多主窗口，也可以不拥有主窗口，所以这样的函数是不存在的，所幸的是，相反的函数是有的。所以我们可以调用EnumWindows来判断所有的窗口是否属于这个进程。
typedef struct tagWNDINFO
...{
DWORD dwProcessId;
HWND hWnd;
} WNDINFO, *LPWNDINFO;

BOOL　CALLBACK YourEnumProc(HWND hWnd,LPARAM lParam)
...{
DWORD dwProcessId;
GetWindowThreadProcessId(hWnd, &dwProcessId);
LPWNDINFO pInfo = (LPWNDINFO)lParam;
if(dwProcessId == pInfo->dwProcessId)
...{
pInfo->hWnd = hWnd;
return FALSE;
}
return TRUE;
}
HWND GetProcessMainWnd(DWORD dwProcessId)
...{
WNDINFO wi;
wi.dwProcessId = dwProcessId;
wi.hWnd = NULL;
EnumWindows(YourEnumProc,(LPARAM)&wi);
return wi.hWnd;
}
如果这个进程没有窗口，函数返回NULL
经试验，在宿主窗口中调用GetProcessMainWnd( GetCurrentProcessId() )后得到的窗口句柄有时确实是该窗口的句柄，但有时得到的却是DLL的窗口的句柄，而在DLL的窗口过程中调用GetProcessMainWnd( GetCurrentProcessId() )只得到DLL窗口的句柄，并不能得到宿主窗口的句柄。【将dll直接注入执行注入代码的exe中进行测试】

将该代码做如下修改后，能得到正确的宿主窗口句柄：

typedef struct tagWNDINFO
...{
    DWORD dwProcessId;
    HWND hWnd;
    HWND Dll_hwnd;
} WNDINFO, *LPWNDINFO;
BOOL CALLBACK YourEnumProc(HWND hWnd, LPARAM lParam)
...{
    DWORD dwProcessId;
    GetWindowThreadProcessId(hWnd, &dwProcessId);
    LPWNDINFO pInfo = (LPWNDINFO)lParam;
    if( (dwProcessId == pInfo->dwProcessId) && (hWnd != pInfo->Dll_hwnd) )
    //确认找到的句柄不是DLL窗口的句柄
    //如果找到的是DLL窗口的句柄则返回FALSE以便继续查找宿主的窗口句柄
    ...{
        pInfo->hWnd = hWnd;
        return FALSE;
    }
    return TRUE;
}
HWND GetProcessMainWnd(DWORD dwProcessId, HWND Dll_hwnd)
//将DLL窗口的句柄传入，以便识别找到的句柄不是DLL窗口的句柄
...{
    WNDINFO wi;
    wi.dwProcessId = dwProcessId;
    wi.hWnd = NULL;
    wi.Dll_hwnd = Dll_hwnd;
    EnumWindows(YourEnumProc,(LPARAM)&wi);
    HWND re_hwnd = wi.hWnd;
    while( GetParent(re_hwnd) != NULL )
    //循环查找父窗口，以便保证返回的句柄是最顶层的窗口句柄
    ...{
        re_hwnd = GetParent( re_hwnd );
    }
    return re_hwnd;
}