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

原文地址：http://www.programlife.net/get-main-window-handler-in-dll.html

有的时候难免需要在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;
}

效果截图：

---

本文地址: 程序人生 >> 在DLL中获取主进程窗口句柄
作者：代码疯子（Wins0n） 本站内容如无声明均属原创，转载请保留作者信息与原文链接，谢谢！