Rust 使用createprocessA 启动cmd窗口实现ssh登录

extern crate winapi;
use winapi::um::synchapi::Sleep;
use winapi::um::winuser::{INPUT, INPUT_KEYBOARD, KEYBDINPUT, SendInput, KEYEVENTF_KEYUP, KEYEVENTF_SCANCODE, KEYEVENTF_UNICODE, VK_SHIFT};
use std::collections::HashMap;
use std::env;
use std::ffi::CString;
use std::mem::{size_of, zeroed};
use std::ptr::null_mut;
use std::time::Duration;
use winapi::um::processthreadsapi::{CreateProcessA, PROCESS_INFORMATION, STARTUPINFOA};
use winapi::um::winnt::HANDLE;
use std::thread;

// 定义按下键的函数
fn press_key(vk: u16) {
    let mut input = INPUT {
        type_: INPUT_KEYBOARD,
        u: unsafe { std::mem::zeroed() },
    };

    unsafe {
        *input.u.ki_mut() = KEYBDINPUT {
            wVk: vk,
            wScan: 0,
            dwFlags: 0,
            time: 0,
            dwExtraInfo: 0usize,
        };
        SendInput(1, &mut input, std::mem::size_of::<INPUT>() as i32);
    }
}

// 定义释放键的函数
fn release_key(vk: u16) {
    let mut input = INPUT {
        type_: INPUT_KEYBOARD,
        u: unsafe { std::mem::zeroed() },
    };

    unsafe {
        *input.u.ki_mut() = KEYBDINPUT {
            wVk: vk,
            wScan: 0,
            dwFlags: KEYEVENTF_KEYUP,
            time: 0,
            dwExtraInfo: 0usize,
        };
        SendInput(1, &mut input, std::mem::size_of::<INPUT>() as i32);
    }
}

// 定义按下并释放按键的函数
fn send_key(vk: u16) {
    press_key(vk);
    release_key(vk);
}

// 定义发送按键序列的函数
fn send_keys(sequence: &str) {
    let mut key_map = HashMap::new();
    key_map.insert('e', 0x45); // 'E' 键
    key_map.insert('l', 0x4C); // 'L' 键
    key_map.insert('i', 0x49); // 'I' 键
    key_map.insert('n', 0x4E); // 'N' 键
    key_map.insert('k', 0x4B); // 'K' 键
    key_map.insert('r', 0x52); // 'R' 键
    key_map.insert('1', 0x31); // '1' 键
    key_map.insert('2', 0x32); // '2' 键
    key_map.insert('3', 0x33); // '3' 键
    key_map.insert('s', 0x53); // 'S' 键
    key_map.insert('y', 0x59); // 'Y' 键
    key_map.insert('H', 0x48); // 大写 'H' 键
    key_map.insert('h', 0x48); // 小写 'h' 键
    key_map.insert('a', 0x41); // 'A' 键
    key_map.insert('o', 0x4F); // 'O' 键
    key_map.insert('w', 0x57); // 'W' 键
    key_map.insert('m', 0x4D); // 'M' 键
    key_map.insert('x', 0x58); // 'X' 键
    key_map.insert('9', 0x39); // '9' 键
    key_map.insert('-', 0xBD); // '9' 键
    key_map.insert('\n', 0x0D); // 回车键

    for ch in sequence.chars() {
        // 如果是大写字母，按下 Shift 键
        if ch.is_uppercase() {
            press_key(VK_SHIFT as u16);
        }

        if let Some(&vk) = key_map.get(&ch.to_ascii_lowercase()) {
            send_key(vk);
        }

        // 如果是大写字母，释放 Shift 键
        if ch.is_uppercase() {
            release_key(VK_SHIFT as u16);
        }
    }
}

fn main() {
    let args: Vec<String> = env::args().collect();
    if args.len() != 2 {
        eprintln!("Usage: {} <your_pro_ip_address>", args[0]);
        return;
    }

    let ip = &args[1]; // 直接获取传递的IP地址

    let ssh_command = format!("ssh elinker@{}", ip); // 将 'username' 替换为实际的用户名
    let cmd_line = format!("cmd.exe /K {}", ssh_command); // 使用 /K 保持命令行窗口打开
    
    // 将命令行转换为 CString
    let command_line = CString::new(cmd_line).unwrap();
    
    let mut startup_info: STARTUPINFOA = unsafe { zeroed() };
    startup_info.cb = size_of::<STARTUPINFOA>() as u32;
    let mut process_info: PROCESS_INFORMATION = unsafe { zeroed() };
    let result = unsafe {
        CreateProcessA(
            null_mut(),
            command_line.as_ptr() as *mut _,
            null_mut(),
            null_mut(),
            0,
            0,
            null_mut(),
            null_mut(),
            &mut startup_info, 
            &mut process_info,
        )
    };

    if result == 0 {
        println!("Failed to create process");
    } else {
        println!("Process created successfully");
    }

    send_keys("\n");
    // 等待命令行窗口打开
    thread::sleep(Duration::from_secs(6)); // 等待命令行窗口加载并等待SSH提示符

    // 输入 "elinker123" 并回车
    send_keys("elinker123\n");

    // 等待 2 秒
    thread::sleep(Duration::from_secs(2));

    // 输入 "sys" 并回车
    send_keys("sys\n");

    // 再次等待 2 秒
    thread::sleep(Duration::from_secs(2));

    // 输入 "Hanshow-imx9" 并回车
    send_keys("Hanshow-imx9\n");

    // 保持命令行窗口打开状态，让用户可以继续使用
    unsafe {
        winapi::um::synchapi::WaitForSingleObject(process_info.hProcess, winapi::um::winbase::INFINITE);
    }
}

[dependencies]
windows-sys="0.59.0"
winapi = { version = "0.3.8", features = ["winuser","libloaderapi","winnt","processthreadsapi","synchapi","winbase"] }
anyhow_ext = "0.2"
 
[dependencies.windows]
version = "0.58.0"
features = [
    "Win32_Foundation",
    "Win32_UI_WindowsAndMessaging",
    "Win32_UI_Input_KeyboardAndMouse",
    "Win32_Security",
    "Win32_System_Diagnostics",
    "Win32_System_Threading",
    "Win32_System_Console",
    "Win32_Graphics_Gdi",
    "Win32_System_LibraryLoader",
    "Win32_UI_WindowsAndMessaging",
    "Win32_System_Diagnostics_Debug",
]