C++在System上下文枚举窗口的实战指南：从原理到避坑

1次阅读

没有评论

共计 4382 个字符，预计需要花费 11 分钟才能阅读完成。

在 Windows 开发中，枚举窗口是一个常见的需求，但在 System 账户下执行这个操作会遇到一些特殊问题。System 账户是 Windows 系统中权限最高的账户，但正因为其高权限，反而会遇到一些普通用户不会遇到的问题。

Session 隔离：Windows 系统采用 Session 隔离机制，每个登录用户都有自己的 Session。System 账户运行在 Session 0 中，而用户界面通常在其他 Session 中。这意味着直接调用 EnumWindows 可能无法枚举到用户 Session 中的窗口。
UIPI 限制 ：用户界面特权隔离(User Interface Privilege Isolation) 机制会阻止低完整性级别的进程向高完整性级别的进程发送消息或获取信息。System 账户虽然是高权限，但仍可能受到这个机制的影响。
跨进程访问：在 System 上下文下，直接访问其他进程的窗口可能会遇到访问拒绝的问题，即使你拥有管理员权限。

Windows 提供了多种枚举窗口的 API，各有适用场景：

EnumWindows (Windows 2000+)：
最常用的窗口枚举函数
只能枚举当前 Session 的窗口
不适用于 System 账户枚举用户 Session 窗口
EnumDesktopWindows (Windows 2000+)：
可以指定桌面进行枚举
需要先获取目标桌面的句柄
适合 System 账户跨 Session 枚举
GetGUIThreadInfo (Windows 2000+)：
获取特定线程的 GUI 信息
适合获取特定线程创建的窗口
不适合批量枚举

要解决 Session 隔离问题，我们需要使用 WTSEnumerateSessions API 获取所有 Session 信息，然后针对每个 Session 进行处理：

#include <WtsApi32.h>
#pragma comment(lib, "Wtsapi32.lib")

void EnumerateSessions() {
    WTS_SESSION_INFO* pSessionInfo = nullptr;
    DWORD sessionCount = 0;

    if (WTSEnumerateSessions(WTS_CURRENT_SERVER_HANDLE, 0, 1, &pSessionInfo, &sessionCount)) {for (DWORD i = 0; i < sessionCount; ++i) {if (pSessionInfo[i].State == WTSActive) {ProcessSession(pSessionInfo[i].SessionId);
            }
        }
        WTSFreeMemory(pSessionInfo);
    }
}

下面是组合使用 OpenInputDesktop 和 EnumDesktopWindows 的完整示例：

BOOL CALLBACK EnumWindowProc(HWND hwnd, LPARAM lParam) {
    // 处理每个窗口
    DWORD pid;
    GetWindowThreadProcessId(hwnd, &pid);

    std::wstring title(GetWindowTextLength(hwnd) + 1, L'\0');
    GetWindowText(hwnd, &title[0], static_cast<int>(title.size()));

    // 添加到结果列表
    auto& windows = *reinterpret_cast<std::vector<WindowInfo>*>(lParam);
    windows.emplace_back(hwnd, pid, title);

    return TRUE;
}

void EnumerateWindowsInSession(DWORD sessionId) {
    HDESK hDesktop = nullptr;
    try {
        // 切换到目标 Session
        if (!WTSConnectSession(sessionId, WTS_CURRENT_SESSION, L"", FALSE)) {throw std::runtime_error("WTSConnectSession failed");
        }

        // 打开输入桌面
        hDesktop = OpenInputDesktop(0, FALSE, DESKTOP_READOBJECTS);
        if (!hDesktop) {throw std::runtime_error("OpenInputDesktop failed");
        }

        // 设置当前线程的桌面
        if (!SetThreadDesktop(hDesktop)) {throw std::runtime_error("SetThreadDesktop failed");
        }

        // 枚举窗口
        std::vector<WindowInfo> windows;
        if (!EnumDesktopWindows(hDesktop, EnumWindowProc, reinterpret_cast<LPARAM>(&windows))) {throw std::runtime_error("EnumDesktopWindows failed");
        }

        // 处理窗口信息...

    } catch (const std::exception& e) {
        // 错误处理
        std::cerr << "Error:" << e.what() << std::endl;}

    // 清理资源
    if (hDesktop) {CloseDesktop(hDesktop);
    }
}

ERROR_ACCESS_DENIED：
通常是由于权限不足或 UIPI 限制
解决方法：提升进程权限或调整目标窗口的完整性级别
ERROR_INVALID_WINDOW_HANDLE：
窗口句柄已失效
解决方法：检查窗口是否仍然存在
ERROR_BAD_DESKTOP：
桌面句柄无效
解决方法：重新获取桌面句柄

句柄有效性：窗口句柄可能会随时失效，特别是在 System 账户下枚举用户 Session 窗口时。
引用计数：不要长期持有窗口句柄，应该在需要使用的时候重新获取。
跨进程使用：如果需要在不同进程间传递窗口句柄，确保目标进程有足够的权限访问该窗口。

32 位进程枚举 64 位窗口：
使用 Wow64 禁用文件系统重定向
可能需要特殊的 API 调用方式
64 位进程枚举 32 位窗口：
通常没有问题
但要注意指针大小的差异

缓存策略：
对不常变化的窗口信息进行缓存
设置合理的过期时间
异步回调：
避免在主线程中执行耗时枚举
使用工作线程或异步模式

// 异步枚举示例
std::future<std::vector<WindowInfo>> AsyncEnumerateWindows() {return std::async(std::launch::async, [] {
        std::vector<WindowInfo> result;
        // 执行枚举操作...
        return result;
    });
}

使用智能指针管理资源：

struct DesktopHandle {
    HDESK handle;

    DesktopHandle(HDESK h) : handle(h) {}
    ~DesktopHandle() { if (handle) CloseDesktop(handle); }

    operator HDESK() const { return handle;}
};

// 使用示例
void Example() {DesktopHandle hDesktop(OpenInputDesktop(0, FALSE, DESKTOP_READOBJECTS));
    if (!hDesktop) {throw std::runtime_error("OpenInputDesktop failed");
    }
    // 使用 hDesktop...
}

微软的 WIL(Windows Implementation Libraries)提供了方便的 COM 对象管理：

#include <wil/com.h>
#include <wil/resource.h>

void ExampleWithWIL() {wil::unique_hdesk hDesktop(OpenInputDesktop(0, FALSE, DESKTOP_READOBJECTS));
    if (!hDesktop) {THROW_LAST_ERROR();
    }
    // 使用 hDesktop...
}

如何检测窗口所属的完整性级别(IL)？这里提供一个思路：

获取窗口所属进程 ID
打开进程句柄
查询进程的完整性级别

DWORD GetWindowIntegrityLevel(HWND hwnd) {
    DWORD pid;
    GetWindowThreadProcessId(hwnd, &pid);

    wil::unique_process_handle hProcess(OpenProcess(PROCESS_QUERY_LIMITED_INFORMATION, FALSE, pid));
    if (!hProcess) {THROW_LAST_ERROR();
    }

    wil::unique_handle hToken;
    if (!OpenProcessToken(hProcess.get(), TOKEN_QUERY, &hToken)) {THROW_LAST_ERROR();
    }

    DWORD lengthNeeded = 0;
    GetTokenInformation(hToken.get(), TokenIntegrityLevel, nullptr, 0, &lengthNeeded);

    auto buffer = std::make_unique<BYTE[]>(lengthNeeded);
    if (!GetTokenInformation(hToken.get(), TokenIntegrityLevel, buffer.get(), lengthNeeded, &lengthNeeded)) {THROW_LAST_ERROR();
    }

    auto integrityLevel = reinterpret_cast<TOKEN_MANDATORY_LABEL*>(buffer.get())->Label;
    return *GetSidSubAuthority(integrityLevel, *GetSidSubAuthorityCount(integrityLevel) - 1);
}

在 System 账户下枚举窗口确实比普通用户上下文更复杂，但通过正确使用 Windows API 组合，我们可以可靠地实现这一功能。关键点包括：