46 lines
1.8 KiB
Rust
46 lines
1.8 KiB
Rust
|
//! General subprocess plumbing shared by the harnesses in this crate.
|
||
|
|
|
||
|
|
use std::sync::Arc;
|
||
|
|
|
||
|
|
/// Pipe all three stdio handles, `kill_on_drop`, spawn, and drain the child's
|
||
|
|
/// stderr into the returned buffer on a background task. The one spawn path
|
||
|
|
/// shared by every subprocess harness in this crate (`GrokStdioClient`,
|
||
|
|
/// `RawStdioClient`, `leader::LeaderStdioClient`); env/args stay with the
|
||
|
|
/// callers, whose hermeticity models differ (sandbox-inherit vs `env_clear`).
|
||
|
|
/// The drain future is `Send`, so this works on and off a `LocalSet`.
|
||
|
|
pub(crate) fn spawn_piped_with_stderr_capture(
|
||
|
|
mut cmd: tokio::process::Command,
|
||
|
|
) -> (tokio::process::Child, Arc<std::sync::Mutex<Vec<u8>>>) {
|
||
|
|
cmd.stdin(std::process::Stdio::piped())
|
||
|
|
.stdout(std::process::Stdio::piped())
|
||
|
|
.stderr(std::process::Stdio::piped())
|
||
|
|
.kill_on_drop(true);
|
||
|
|
|
||
|
|
// Derived from `cmd` itself so the panic can never name a different binary
|
||
|
|
// than the one actually spawned.
|
||
|
|
let program = cmd.as_std().get_program().to_string_lossy().into_owned();
|
||
|
|
let mut child = cmd
|
||
|
|
.spawn()
|
||
|
|
.unwrap_or_else(|e| panic!("failed to spawn grok at {program}: {e}"));
|
||
|
|
|
||
|
|
let stderr = Arc::new(std::sync::Mutex::new(Vec::new()));
|
||
|
|
let stderr_capture = stderr.clone();
|
||
|
|
let mut child_stderr = child.stderr.take().expect("child stderr missing");
|
||
|
|
tokio::spawn(async move {
|
||
|
|
use tokio::io::AsyncReadExt as _;
|
||
|
|
|
||
|
|
let mut buf = [0_u8; 1024];
|
||
|
|
loop {
|
||
|
|
match child_stderr.read(&mut buf).await {
|
||
|
|
Ok(0) => break,
|
||
|
|
Ok(read) => stderr_capture
|
||
|
|
.lock()
|
||
|
|
.unwrap()
|
||
|
|
.extend_from_slice(&buf[..read]),
|
||
|
|
Err(_) => break,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
});
|
||
|
|
|
||
|
|
(child, stderr)
|
||
|
|
}
|