iroiro147
145 lines
4.4 KiB
Rust
145 lines
4.4 KiB
Rust
use rmcp::transport::TokioChildProcess;
|
|
use std::io;
|
|
#[cfg(target_os = "linux")]
|
|
use std::sync::{mpsc, OnceLock};
|
|
use tokio::process::ChildStderr;
|
|
use tokio::process::Command;
|
|
|
|
#[cfg(windows)]
|
|
const CREATE_NO_WINDOW_FLAG: u32 = 0x08000000;
|
|
|
|
#[cfg(target_os = "linux")]
|
|
fn configure_parent_death_signal(command: &mut Command) {
|
|
let parent_pid = unsafe { libc::getpid() };
|
|
|
|
unsafe {
|
|
command.pre_exec(move || {
|
|
if libc::prctl(libc::PR_SET_PDEATHSIG, libc::SIGTERM) != 0 {
|
|
return Err(std::io::Error::last_os_error());
|
|
}
|
|
|
|
if libc::getppid() != parent_pid {
|
|
return Err(std::io::Error::from_raw_os_error(libc::ESRCH));
|
|
}
|
|
|
|
Ok(())
|
|
});
|
|
}
|
|
}
|
|
|
|
pub trait SubprocessExt {
|
|
fn set_no_window(&mut self) -> &mut Self;
|
|
}
|
|
|
|
/// Creates a Git command that rejects implicit bare repositories and cannot run a
|
|
/// repository-configured fsmonitor hook.
|
|
pub fn git_command() -> std::process::Command {
|
|
let mut command = std::process::Command::new("git");
|
|
command.args([
|
|
"-c",
|
|
"safe.bareRepository=explicit",
|
|
"-c",
|
|
"core.fsmonitor=false",
|
|
]);
|
|
command
|
|
}
|
|
|
|
impl SubprocessExt for Command {
|
|
fn set_no_window(&mut self) -> &mut Self {
|
|
#[cfg(windows)]
|
|
{
|
|
self.creation_flags(CREATE_NO_WINDOW_FLAG);
|
|
}
|
|
self
|
|
}
|
|
}
|
|
|
|
impl SubprocessExt for std::process::Command {
|
|
fn set_no_window(&mut self) -> &mut Self {
|
|
#[cfg(windows)]
|
|
{
|
|
use std::os::windows::process::CommandExt;
|
|
self.creation_flags(CREATE_NO_WINDOW_FLAG);
|
|
}
|
|
self
|
|
}
|
|
}
|
|
|
|
fn configure_common_subprocess(command: &mut Command) {
|
|
// Isolate subprocess into its own process group so it does not receive
|
|
// SIGINT when the user presses Ctrl+C in the terminal.
|
|
#[cfg(unix)]
|
|
command.process_group(0);
|
|
command.set_no_window();
|
|
}
|
|
|
|
#[allow(unused_variables)]
|
|
pub fn configure_subprocess(command: &mut Command) {
|
|
configure_common_subprocess(command);
|
|
#[cfg(target_os = "linux")]
|
|
configure_parent_death_signal(command);
|
|
}
|
|
|
|
#[cfg(target_os = "linux")]
|
|
struct LongLivedSpawnRequest {
|
|
command: Command,
|
|
runtime: tokio::runtime::Handle,
|
|
response: tokio::sync::oneshot::Sender<io::Result<(TokioChildProcess, Option<ChildStderr>)>>,
|
|
}
|
|
|
|
#[cfg(target_os = "linux")]
|
|
fn long_lived_spawn_sender() -> io::Result<mpsc::Sender<LongLivedSpawnRequest>> {
|
|
static SENDER: OnceLock<io::Result<mpsc::Sender<LongLivedSpawnRequest>>> = OnceLock::new();
|
|
|
|
match SENDER.get_or_init(|| {
|
|
let (sender, receiver) = mpsc::channel::<LongLivedSpawnRequest>();
|
|
std::thread::Builder::new()
|
|
.name("goose-extension-spawner".to_owned())
|
|
.spawn(move || {
|
|
while let Ok(mut request) = receiver.recv() {
|
|
let _runtime_guard = request.runtime.enter();
|
|
configure_subprocess(&mut request.command);
|
|
let result = TokioChildProcess::builder(request.command)
|
|
.stderr(std::process::Stdio::piped())
|
|
.spawn();
|
|
let _ = request.response.send(result);
|
|
}
|
|
})
|
|
.map(|_| sender)
|
|
}) {
|
|
Ok(sender) => Ok(sender.clone()),
|
|
Err(error) => Err(io::Error::new(error.kind(), error.to_string())),
|
|
}
|
|
}
|
|
|
|
/// Spawn a long-lived MCP subprocess without tying Linux parent-death cleanup
|
|
/// to the Tokio worker that happened to request it.
|
|
pub async fn spawn_long_lived_mcp_subprocess(
|
|
command: Command,
|
|
) -> io::Result<(TokioChildProcess, Option<ChildStderr>)> {
|
|
#[cfg(target_os = "linux")]
|
|
{
|
|
let runtime = tokio::runtime::Handle::try_current().map_err(io::Error::other)?;
|
|
let (response_tx, response_rx) = tokio::sync::oneshot::channel();
|
|
long_lived_spawn_sender()?
|
|
.send(LongLivedSpawnRequest {
|
|
command,
|
|
runtime,
|
|
response: response_tx,
|
|
})
|
|
.map_err(|_| io::Error::new(io::ErrorKind::BrokenPipe, "extension spawner exited"))?;
|
|
response_rx
|
|
.await
|
|
.map_err(|_| io::Error::new(io::ErrorKind::BrokenPipe, "extension spawner exited"))?
|
|
}
|
|
|
|
#[cfg(not(target_os = "linux"))]
|
|
{
|
|
let mut command = command;
|
|
configure_subprocess(&mut command);
|
|
TokioChildProcess::builder(command)
|
|
.stderr(std::process::Stdio::piped())
|
|
.spawn()
|
|
}
|
|
}
|