如何用C++获取应用程序主窗体的IUIAutomationElement？

思路1：基于进程ID/进程名定位 #

你已经能通过进程名拿到进程ID，接下来只需枚举该进程下的顶层窗口，再转换为UIA元素：

实现步骤 #

1. 初始化COM环境（UIA基于COM架构）
2. 枚举所有顶层窗口，筛选出属于目标进程的窗口
3. 对筛选出的窗口，判断是否为主窗体（排除子窗口、弹窗）
4. 将符合条件的窗口句柄转换为IUIAutomationElement

代码示例 #

#include <Windows.h>
#include <UIAutomation.h>
#include <string>

#pragma comment(lib, "UIAutomation.lib")

// 全局变量传递进程ID和结果窗口句柄
DWORD g_targetPid = 0;
HWND g_mainHwnd = NULL;

// 窗口枚举回调函数
BOOL CALLBACK EnumWindowsProc(HWND hwnd, LPARAM lParam) {
    DWORD pid;
    GetWindowThreadProcessId(hwnd, &pid);
    if (pid != g_targetPid) {
        return TRUE;
    }

    // 判断主窗体：无所有者且带APPWINDOW扩展风格
    if (GetWindow(hwnd, GW_OWNER) == NULL && 
        (GetWindowLongPtr(hwnd, GWL_EXSTYLE) & WS_EX_APPWINDOW)) {
        g_mainHwnd = hwnd;
        return FALSE; // 找到后终止枚举
    }
    return TRUE;
}

IUIAutomationElement* GetMainElementByPid(DWORD pid) {
    IUIAutomationElement* pElement = NULL;
    IUIAutomation* pUIA = NULL;

    // 初始化COM
    if (CoInitialize(NULL) != S_OK) {
        return NULL;
    }

    if (CoCreateInstance(CLSID_CUIAutomation, NULL, CLSCTX_INPROC_SERVER, 
        IID_IUIAutomation, (void**)&pUIA) != S_OK) {
        CoUninitialize();
        return NULL;
    }

    // 枚举窗口找目标进程主窗体
    g_targetPid = pid;
    g_mainHwnd = NULL;
    EnumWindows(EnumWindowsProc, 0);

    if (g_mainHwnd != NULL) {
        pUIA->ElementFromHandle(g_mainHwnd, &pElement);
    }

    // 释放资源
    if (pUIA) pUIA->Release();
    CoUninitialize();
    return pElement;
}

// 调用示例：假设已获取目标进程ID targetPid
// IUIAutomationElement* pMainElement = GetMainElementByPid(targetPid);
// 使用后记得释放：if(pMainElement) pMainElement->Release();

思路2：基于窗体标题子串筛选 #

通过枚举所有顶层窗口，匹配标题中的目标子串，再转换为UIA元素：

实现步骤 #

1. 初始化COM环境
2. 枚举所有顶层窗口，获取窗口标题
3. 检查标题是否包含目标子串
4. 验证是否为主窗体，转换为IUIAutomationElement

代码示例 #

#include <Windows.h>
#include <UIAutomation.h>
#include <string>
#include <algorithm>

#pragma comment(lib, "UIAutomation.lib")

std::wstring g_titleSubstr;
HWND g_mainHwnd = NULL;

BOOL CALLBACK EnumWindowsProcByTitle(HWND hwnd, LPARAM lParam) {
    // 跳过不可见窗口
    if (!IsWindowVisible(hwnd)) {
        return TRUE;
    }

    // 获取窗口标题
    WCHAR title[256];
    GetWindowTextW(hwnd, title, 256);
    std::wstring windowTitle(title);

    // 不区分大小写匹配子串
    std::wstring lowerTitle = windowTitle;
    std::wstring lowerSubstr = g_titleSubstr;
    std::transform(lowerTitle.begin(), lowerTitle.end(), lowerTitle.begin(), ::towlower);
    std::transform(lowerSubstr.begin(), lowerSubstr.end(), lowerSubstr.begin(), ::towlower);

    if (lowerTitle.find(lowerSubstr) != std::wstring::npos) {
        // 验证主窗体属性
        if (GetWindow(hwnd, GW_OWNER) == NULL && 
            (GetWindowLongPtr(hwnd, GWL_EXSTYLE) & WS_EX_APPWINDOW)) {
            g_mainHwnd = hwnd;
            return FALSE;
        }
    }
    return TRUE;
}

IUIAutomationElement* GetMainElementByTitleSubstr(const std::wstring& substr) {
    IUIAutomationElement* pElement = NULL;
    IUIAutomation* pUIA = NULL;

    if (CoInitialize(NULL) != S_OK) {
        return NULL;
    }

    if (CoCreateInstance(CLSID_CUIAutomation, NULL, CLSCTX_INPROC_SERVER, 
        IID_IUIAutomation, (void**)&pUIA) != S_OK) {
        CoUninitialize();
        return NULL;
    }

    g_titleSubstr = substr;
    g_mainHwnd = NULL;
    EnumWindows(EnumWindowsProcByTitle, 0);

    if (g_mainHwnd != NULL) {
        pUIA->ElementFromHandle(g_mainHwnd, &pElement);
    }

    if (pUIA) pUIA->Release();
    CoUninitialize();
    return pElement;
}

// 调用示例：查找标题包含"Notepad"的主窗体
// IUIAutomationElement* pMainElement = GetMainElementByTitleSubstr(L"Notepad");
// 使用后记得释放：if(pMainElement) pMainElement->Release();

注意事项 #

• 所有COM接口使用后必须调用Release()释放资源，避免内存泄漏
• 通过GW_OWNER和WS_EX_APPWINDOW筛选主窗体，可有效排除弹窗、子窗口等非目标元素
• 若目标进程存在多个主窗口（罕见场景），可额外添加窗口类名等筛选条件

内容的提问来源于stack exchange，提问作者arnold_w