c++ Windows服务中实现WTSQueryUserToken获取Local System权限

在Windows服务中获取Local System权限，可以通过以下步骤实现：

1. 使用管理员权限运行服务。在服务安装后，可以在服务属性中设置“登录”选项卡来指定服务登录的账户。选择“本地系统账户”或者其他具有Local System权限的账户。

2. 在服务中获取当前会话的会话ID。使用WTSGetActiveConsoleSessionId函数获取当前会话ID。

3. 使用会话ID调用WTSQueryUserToken函数获取当前会话的用户令牌。代码示例：

DWORD sessionId = WTSGetActiveConsoleSessionId();
HANDLE hToken = NULL;
if (WTSQueryUserToken(sessionId, &hToken))
{
    // 成功获取用户令牌
    // TODO：使用用户令牌执行需要Local System权限的操作
}
else
{
    // 获取用户令牌失败
    // TODO：处理错误
}

4. 使用用户令牌执行需要Local System权限的操作。例如，可以使用CreateProcessAsUser函数来启动另一个进程，并指定Local System权限。

希望这些步骤能够帮助您实现在Windows服务中获取Local System权限。

相关问题

我需要你按照如下需求在 Windows 服务进程中，你可以将上述代码放入你的服务逻辑中进行调用，来获取 Num、Caps 和其他 LED 灯的值。请注意，由于服务进程通常是以 SYSTEM 用户身份运行的，所以它可能无法访问当前用户的键盘状态。为了解决这个问题，你可以使用 WTSGetActiveConsoleSessionId 函数来获取当前活动控制台会话的 ID，并使用 WTSQueryUserToken 函数来获取该会话的用户令牌。然后，使用该用户令牌来打开用户键盘状态的设备对象，以获取键盘状态。;使用rust 编码实现

好的，根据您的需求，我可以提供以下代码示例：

use std::ptr::null_mut;
use winapi::um::winuser::{GetKeyboardState, MapVirtualKeyW, MapVirtualKeyExW, VK_CAPITAL, VK_NUMLOCK, VK_SCROLL};
use winapi::um::wtsapi32::{WTSGetActiveConsoleSessionId, WTSQueryUserToken};
use winapi::um::winnt::{HANDLE, PHANDLE, TOKEN_ALL_ACCESS, TOKEN_QUERY};
use winapi::um::processthreadsapi::GetCurrentProcess;
use winapi::um::securitybaseapi::DuplicateTokenEx;
use winapi::um::handleapi::CloseHandle;
use winapi::shared::minwindef::{BOOL, BYTE, DWORD, FALSE};
use winapi::shared::winerror::ERROR_SUCCESS;

fn get_keyboard_state() -> Result<(bool, bool, bool), String> {
    let mut state: [BYTE; 256] = [0; 256];
    unsafe {
        if GetKeyboardState(state.as_mut_ptr()) == 0 {
            return Err(format!("Error getting keyboard state: {}", std::io::Error::last_os_error()));
        }
    }

    let numlock = state[VK_NUMLOCK as usize] & 1 != 0;
    let capslock = state[VK_CAPITAL as usize] & 1 != 0;
    let scrolllock = state[VK_SCROLL as usize] & 1 != 0;

    Ok((numlock, capslock, scrolllock))
}

fn get_keyboard_leds(session_id: DWORD) -> Result<(bool, bool, bool), String> {
    let mut token_handle: HANDLE = null_mut();
    let mut primary_token_handle: HANDLE = null_mut();
    let mut user_sid: winapi::um::winnt::PSID = null_mut();
    let mut user_token_handle: HANDLE = null_mut();

    unsafe {
        if WTSQueryUserToken(session_id, &mut token_handle) == 0 {
            return Err(format!("Error getting user token: {}", std::io::Error::last_os_error()));
        }

        if DuplicateTokenEx(token_handle, TOKEN_ALL_ACCESS, null_mut(), winapi::um::winnt::SecurityImpersonation, winapi::um::winnt::TokenPrimary, &mut primary_token_handle) == 0 {
            return Err(format!("Error duplicating token: {}", std::io::Error::last_os_error()));
        }

        CloseHandle(token_handle);

        let mut token_info_size: DWORD = 0;
        if GetTokenInformation(primary_token_handle, TokenUser, null_mut(), 0, &mut token_info_size) != FALSE {
            return Err("Unexpected success in GetTokenInformation".to_string());
        }

        let mut token_info: Vec<u8> = vec![0; token_info_size as usize];
        if GetTokenInformation(primary_token_handle, TokenUser, token_info.as_mut_ptr() as *mut _, token_info_size, &mut token_info_size) == FALSE {
            return Err(format!("Error getting token information: {}", std::io::Error::last_os_error()));
        }

        let user_sid = (*(token_info.as_ptr() as *const winapi::um::winnt::TOKEN_USER)).User.Sid;

        if OpenProcessToken(GetCurrentProcess(), TOKEN_QUERY, &mut user_token_handle) == 0 {
            return Err(format!("Error opening process token: {}", std::io::Error::last_os_error()));
        }

        if ImpersonateLoggedOnUser(primary_token_handle) == 0 {
            return Err(format!("Error impersonating user: {}", std::io::Error::last_os_error()));
        }

        let (numlock, capslock, scrolllock) = get_keyboard_state()?;

        if RevertToSelf() == 0 {
            return Err(format!("Error reverting to self: {}", std::io::Error::last_os_error()));
        }

        CloseHandle(primary_token_handle);
        CloseHandle(user_token_handle);

        Ok((numlock, capslock, scrolllock))
    }
}

fn main() {
    let session_id = unsafe { WTSGetActiveConsoleSessionId() };
    let (numlock, capslock, scrolllock) = match get_keyboard_leds(session_id) {
        Ok((numlock, capslock, scrolllock)) => (numlock, capslock, scrolllock),
        Err(e) => {
            println!("Error getting keyboard LEDs: {}", e);
            return;
        }
    };

    println!("Num Lock: {}", numlock);
    println!("Caps Lock: {}", capslock);
    println!("Scroll Lock: {}", scrolllock);
}

这段代码可以获取当前活动控制台会话的 ID，并使用该会话的用户令牌来打开用户键盘状态的设备对象，以获取 Num、Caps 和其他 LED 灯的值。如果无法获取用户令牌，则会返回错误信息。

WTSQueryUserToken failed

WTSQueryUserToken是一个Windows API函数，用于获取指定用户的访问令牌。该函数可以在Windows服务中使用，以便在指定用户的上下文中执行操作。

如果WTSQueryUserToken函数失败，可能有以下几个原因：

1. 权限不足：在服务中调用WTSQueryUserToken函数时，需要足够的权限才能获取指定用户的访问令牌。如果服务没有足够的权限，则会导致该函数失败。您可以尝试使用其他方式获取用户访问令牌，例如使用LogonUser函数进行验证，并使用CreateProcessAsUser函数在用户上下文中启动进程。

2. 用户不存在：如果指定的用户不存在，则WTSQueryUserToken函数会失败。

3. 会话不存在：如果指定的会话不存在，则WTSQueryUserToken函数会失败。

4. 函数调用顺序不正确：在调用WTSQueryUserToken函数之前，需要先调用WTSQuerySessionInformation函数，以获取有关会话和用户的信息。如果没有正确调用WTSQuerySessionInformation函数，则WTSQueryUserToken函数可能会失败。

5. 其他问题：除上述原因外，WTSQueryUserToken函数还可能由于其他问题而失败，例如内存不足、参数错误等。

如果您遇到了WTSQueryUserToken函数失败的问题，建议您检查上述原因，并尝试解决它们。如果问题仍然存在，您可以参考Windows错误日志，以获取更多有关失败原因的信息。