如何在Rust多线程环境中捕获调用C库产生的输出？

问题背景 #

• 基于Rust开发的项目，通过FFI调用静态链接的C库（通过build.rs脚本完成链接）
• 运行环境为类Unix操作系统
• 项目会创建多个工作线程，每个线程均会调用C库中的函数
• 被调用的C库函数会向STDOUT打印进度及其他信息
• 需要针对每个线程单独捕获对应C库函数的输出

已尝试的方案 #

• 进程级STDOUT重定向：仅在单线程场景下有效，多线程环境中会出现输出混乱或无法区分归属的问题
• Rust nightly版本的set_print：仅能捕获Rust自身println!宏的输出，无法捕获FFI调用产生的C库输出

可行解决方案：线程级文件描述符重定向（类Unix专属） #

在类Unix系统中，STDOUT对应的文件描述符为1，我们可以利用**管道（pipe）**和文件描述符复制（dup2）操作，为每个线程单独重定向STDOUT，具体步骤如下：

1. 创建一对匿名管道，用于中转输出内容
2. 保存当前线程的STDOUT文件描述符副本
3. 将STDOUT重定向到管道的写入端
4. 调用目标C库函数（此时C库的输出会写入管道）
5. 恢复原STDOUT文件描述符
6. 从管道的读取端读取捕获到的输出内容

代码示例 #

首先需在Cargo.toml中引入libc crate以调用Unix系统调用：

[dependencies]
libc = "0.2"

线程内捕获逻辑的实现：

use libc::{c_int, dup, dup2, pipe, read, close, STDOUT_FILENO};
use std::thread;

// 示例：假设的C库外部函数声明
extern "C" {
    fn c_library_function();
}

/// 捕获闭包执行过程中C库输出到STDOUT的内容
fn capture_c_stdout<F: FnOnce()>(func: F) -> String {
    let mut pipe_fds = [0; 2];
    // 创建匿名管道
    unsafe { pipe(pipe_fds.as_mut_ptr()) };
    let read_fd = pipe_fds[0];
    let write_fd = pipe_fds[1];

    // 保存当前STDOUT的文件描述符副本
    let original_stdout = unsafe { dup(STDOUT_FILENO) };

    // 将STDOUT重定向到管道写入端
    unsafe { dup2(write_fd, STDOUT_FILENO) };
    unsafe { close(write_fd) }; // 关闭多余的写入端句柄

    // 执行目标C库函数
    func();

    // 恢复原STDOUT
    unsafe { dup2(original_stdout, STDOUT_FILENO) };
    unsafe { close(original_stdout) };

    // 从管道读取捕获到的输出
    let mut buffer = Vec::new();
    let mut buf = [0u8; 1024];
    loop {
        let bytes_read = unsafe { read(read_fd, buf.as_mut_ptr() as *mut _, buf.len()) };
        if bytes_read <= 0 {
            break;
        }
        buffer.extend_from_slice(&buf[0..bytes_read as usize]);
    }
    unsafe { close(read_fd) };

    String::from_utf8_lossy(&buffer).into_owned()
}

fn main() {
    // 启动多个线程测试捕获逻辑
    let handles: Vec<_> = (0..3).map(|thread_id| {
        thread::spawn(move || {
            let output = capture_c_stdout(|| {
                unsafe { c_library_function() };
            });
            println!("线程 {} 捕获的C库输出:\n{}", thread_id, output);
        })
    }).collect();

    for handle in handles {
        handle.join().unwrap();
    }
}

关键说明 #

• 线程安全：每个线程操作独立的文件描述符副本，dup2仅修改当前线程的STDOUT指向，不会干扰其他线程
• 资源清理：必须确保关闭多余的管道句柄并恢复原STDOUT，避免文件描述符泄漏
• 兼容性：该方案仅针对类Unix系统有效，若需支持Windows需使用CreatePipe和SetStdHandle等Windows API

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