如何用Rust检测当前运行环境是CMD还是PowerShell？

问题核心 #

你直接运行Stack Overflow上的跨Shell检测命令报错，是因为该命令需要通过对应Shell解析执行，不能直接作为独立程序调用。得先准确识别当前Shell类型，再针对性调用Shell执行命令。

可靠检测方法 #

方法1：通过环境变量判断 #

Windows系统中，Shell类型可通过环境变量快速识别：

• PowerShell：环境变量PSModulePath存在且包含PowerShell模块路径；或者ComSpec指向cmd.exe但当前进程父进程是powershell.exe/pwsh.exe
• CMD：环境变量ComSpec指向cmd.exe，且父进程不是PowerShell相关程序

Rust实现代码：

use std::env;

fn detect_shell() -> Option<&'static str> {
    // 优先判断PowerShell
    if env::var("PSModulePath").is_ok() {
        return Some("POWERSHELL");
    }
    // 再判断CMD
    if let Ok(comspec) = env::var("ComSpec") {
        if comspec.ends_with("cmd.exe") {
            return Some("CMD");
        }
    }
    None
}

方法2：通过父进程判断 #

读取当前进程的父进程名称，判断是否为cmd.exe、powershell.exe或pwsh.exe（PowerShell Core）。可借助sysinfo crate实现：

use sysinfo::{System, SystemExt, ProcessExt};

fn detect_shell_by_parent() -> Option<&'static str> {
    let mut sys = System::new_all();
    sys.refresh_processes();
    
    let current_pid = std::process::id();
    let current_process = sys.process(current_pid)?;
    let parent_pid = current_process.parent()?;
    let parent_process = sys.process(parent_pid)?;
    let parent_name = parent_process.name().to_lowercase();

    match parent_name.as_str() {
        "cmd.exe" => Some("CMD"),
        "powershell.exe" | "pwsh.exe" => Some("POWERSHELL"),
        _ => None,
    }
}

需要在Cargo.toml中添加依赖：

[dependencies]
sysinfo = "0.29.0"

执行对应命令的正确方式 #

检测到Shell类型后，需调用对应Shell执行命令：

执行CMD命令 #

use std::process::Command;

fn run_cmd_command(cmd: &str) -> std::io::Result<()> {
    Command::new("cmd.exe")
        .args(["/C", cmd])
        .status()?;
    Ok(())
}

执行PowerShell命令 #

use std::process::Command;

fn run_powershell_command(cmd: &str) -> std::io::Result<()> {
    // 优先使用pwsh（PowerShell Core），不存在则回退到powershell.exe
    let shell = if Command::new("pwsh").spawn().is_ok() {
        "pwsh"
    } else {
        "powershell.exe"
    };

    Command::new(shell)
        .args(["-Command", cmd])
        .status()?;
    Ok(())
}

适配环境变量与命令参数 #

针对不同Shell的参数语法差异，可做如下处理：

• 环境变量设置：CMD用set VAR=value，PowerShell用$Env:VAR = "value"
• 参数转义：CMD中特殊字符用^转义，PowerShell用`转义；包含空格的参数，两者都支持用双引号包裹

示例：临时设置环境变量并执行用户命令

fn run_with_env(shell_type: &str, env_var: (&str, &str), user_cmd: &str) -> std::io::Result<()> {
    let (var_name, var_value) = env_var;
    let full_cmd = match shell_type {
        "CMD" => format!("set {}={} && {}", var_name, var_value, user_cmd),
        "POWERSHELL" => format!("$Env:{} = \"{}\"; {}", var_name, var_value, user_cmd),
        _ => return Err(std::io::Error::new(std::io::ErrorKind::Other, "Unknown shell type")),
    };

    match shell_type {
        "CMD" => run_cmd_command(&full_cmd),
        "POWERSHELL" => run_powershell_command(&full_cmd),
        _ => Err(std::io::Error::new(std::io::ErrorKind::Other, "Unknown shell type")),
    }
}

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