如何在Rust中fork/exec转发所有参数的进程？并通过Command API实现指定Shell逻辑

我来帮你搞定这两个Rust进程处理的问题，直接上代码和实用解释：

其实Rust标准库的std::process::Command已经封装了fork+exec的底层逻辑，不需要手动调用系统级的fork（当然如果有特殊场景必须手动实现，也可以用nix库，不过标准库方案更简洁安全）。

要转发所有参数，核心是获取当前程序的命令行参数（跳过第一个自身可执行文件路径），再传递给新进程。示例代码如下：

use std::process::Command;
use std::env;

fn main() {
    // 提取当前程序的命令行参数，跳过第一个（自身路径）
    let args: Vec<_> = env::args_os().skip(1).collect();

    // 构建要执行的目标命令，这里以`ls`为例，可替换为任意程序
    let mut cmd = Command::new("ls");
    // 追加所有转发的参数
    cmd.args(&args);

    // 执行命令并处理结果
    match cmd.status() {
        Ok(status) => println!("子进程退出状态: {}", status),
        Err(e) => eprintln!("启动进程失败: {}", e),
    }
}

如果确实需要手动实现fork+exec（比如精细控制进程生命周期），可以用nix库（需在Cargo.toml添加nix = "0.27"），示例：

use nix::unistd::{fork, execvp, ForkResult};
use std::env;
use std::ffi::CString;

fn main() {
    // 转换参数为C兼容格式
    let args: Vec<CString> = env::args_os()
        .skip(1)
        .map(|os_str| CString::new(os_str).unwrap())
        .collect();

    let program = CString::new("ls").unwrap();

    match fork() {
        Ok(ForkResult::Parent { child }) => {
            println!("子进程PID: {}", child);
            // 等待子进程结束
            let _ = nix::sys::wait::waitpid(child, None);
        }
        Ok(ForkResult::Child) => {
            // 在子进程中执行execvp替换进程镜像
            if let Err(e) = execvp(&program, &args) {
                eprintln!("execvp失败: {}", e);
                std::process::exit(1);
            }
        }
        Err(e) => eprintln!("fork失败: {}", e),
    }
}

一般场景下，标准库的Command已经足够满足需求，无需手动操作底层系统调用。

你给出的Shell逻辑是根据DEPLOY_ENV环境变量是否为空，选择执行old_program或new_program并传递所有参数，用Rust可以这样实现：

use std::process::Command;
use std::env;

fn main() {
    // 获取环境变量DEPLOY_ENV，兼容"不存在"和"为空字符串"两种情况
    let deploy_env = env::var_os("DEPLOY_ENV");
    let target_program = if deploy_env.as_ref().map_or(true, |s| s.is_empty()) {
        "old_program"
    } else {
        "new_program"
    };

    // 提取要转发的命令行参数
    let args: Vec<_> = env::args_os().skip(1).collect();

    // 执行目标命令并返回其退出码
    let status = Command::new(target_program)
        .args(&args)
        .status()
        .expect("无法启动目标进程");

    std::process::exit(status.code().unwrap_or(1));
}

几个关键细节：

• 用env::var_os()而非env::var()，避免非UTF-8环境变量导致的错误；
• 判断逻辑同时覆盖了环境变量不存在和存在但为空的情况；
• 最后用std::process::exit()返回子进程的退出码，和Shell脚本行为一致。

如果需要捕获子进程的输出（而非仅等待结束），可以把status()换成output()：

let output = Command::new(target_program)
    .args(&args)
    .output()
    .expect("无法启动目标进程");

println!("子进程标准输出:\n{}", String::from_utf8_lossy(&output.stdout));
eprintln!("子进程错误输出:\n{}", String::from_utf8_lossy(&output.stderr));

std::process::exit(output.status.code().unwrap_or(1));

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