WTSIsRemoteSession 始终返回 TRUE

Question

我目前正在编写一个 Windows 服务，当用户登录时它也会做一些事情。如果登录来自远程计算机（例如远程桌面），则有什么都不做的想法，并试图找到一种方法来解决这个问题。但以下不起作用 - 它总是返回 true（Windows 10 64 位 V1809） - 我在这里做错了吗？

DWORD SvcHandlerEx(DWORD controlCode, DWORD eventType, ... )
{
    ...

    switch(controlCode)
    {
        case SERVICE_CONTROL_SESSIONCHANGE:
        {
            WTSSESSION_NOTIFICATION *pSessInfo = (WTSSESSION_NOTIFICATION *)pEvtData;
            // invoke SessionChangeHandler(eventId, pSessInfo->dwSessionId)
        }

        ...
    }

    ...
}

...

VOID SessionChangeHandler(DWORD reason, DWORD sessionId)
{
    LPWSTR *pSessionInfo = nullptr;
    DWORD dataLen = 0;
    BOOL isRDP = false;

    if (!WTSQuerySessionInformationW(WTS_CURRENT_SERVER_HANDLE, sessionId, 
                                     WTSIsRemoteSession, &pSessionInfo, &dataLen))
    {
        // Do some error handling...
        return;
    }

    if (dataLen)
    {
        if (dataLen)
        {
            isRDP = (bool)pSessionInfo;    // Always 1 (TRUE) !!!
        }

        WTSFreeMemory(pSessionInfo);
    }

    ...

}

Answer 1

根据documentation 为WTSIsRemoteSession：

WTSIsRemoteSession

确定当前会话是否为远程会话。

WTSQuerySessionInformation 函数返回值TRUE 表示当前会话是远程会话，FALSE 表示当前会话是本地会话。该值只能用于本地机器，所以WTSQuerySessionInformation函数的hServer参数必须包含WTS_CURRENT_SERVER_HANDLE。

Windows Server 2008 和 Windows Vista：不支持此值。

这意味着WTSQuerySessionInformation() 的返回值 包含您要查找的值，并且函数可能分配的任何内存（如果有）都是次要的，在查询WTSIsRemoteSession 时应忽略，例如：

VOID SessionChangeHandler(DWORD reason, DWORD sessionId)
{
    LPWSTR *pSessionInfo = nullptr;
    DWORD dataLen = 0;

    bool isRDP = WTSQuerySessionInformationW(WTS_CURRENT_SERVER_HANDLE, sessionId, WTSIsRemoteSession, &pSessionInfo, &dataLen);
    if ((!isRDP) && (GetLastError() != 0))
    {
        // Do some error handling...
        return;
    }

    if (pSessionInfo)
        WTSFreeMemory(pSessionInfo);

    // use isRDP as needed...
    ...
}

但是，如果您发现isRDP 在这种情况下始终为真，那么文档具有误导性，您应该检查pSessionInfo 缓冲区的内容。您正在根据 WTSQuerySessionInformation() 是否分配任何内存来设置您的 isRDP 变量，您并没有查看数据中的实际内容。

例如，假设 dataLen 在输出时被设置为 sizeof(BOOL) 然后将您的 pSessionInfo 指针转换为 BOOL* 指针并取消引用它，例如：

VOID SessionChangeHandler(DWORD reason, DWORD sessionId)
{
    LPWSTR *pSessionInfo = nullptr;
    DWORD dataLen = 0;

    if (!WTSQuerySessionInformationW(WTS_CURRENT_SERVER_HANDLE, sessionId, WTSIsRemoteSession, &pSessionInfo, &dataLen))
    {
        // Do some error handling...
        return;
    }

    bool isRDP = * (BOOL*) pSessionInfo;
    WTSFreeMemory(pSessionInfo);

    // use isRDP as needed...
    ...
}

或者：

VOID SessionChangeHandler(DWORD reason, DWORD sessionId)
{
    BOOL *isRDP = nullptr;
    DWORD dataLen = 0;

    if (!WTSQuerySessionInformationW(WTS_CURRENT_SERVER_HANDLE, sessionId, WTSIsRemoteSession, (LPWSTR*)&isRDP, &dataLen))
    {
        // Do some error handling...
        return;
    }

    // use isRDP as needed...

    WTSFreeMemory(isRDP);
    ...
}