fix(host): 跨进程启动串行化可继承管道窗口

This commit is contained in:
2026-09-09 05:51:47 +08:00
parent 57acbc0e32
commit b3b4feb2db
6 changed files with 84 additions and 9 deletions
+5 -1
View File
@@ -325,7 +325,11 @@ impl CoreSupervisor {
use std::os::windows::process::CommandExt;
command.creation_flags(0x08000000); // CREATE_NO_WINDOW
}
let child = command.group_spawn().map_err(|_| "CORE_SPAWN_FAILED")?;
let child = {
#[cfg(windows)]
let _creation = crate::process_creation::lock()?;
command.group_spawn().map_err(|_| "CORE_SPAWN_FAILED")?
};
let mut session = Session {
child,
lifetime: Arc::new(Mutex::new(None)),
@@ -148,6 +148,9 @@ impl<'a> Suspended<'a> {
startup.lpAttributeList = attributes.buffer.as_mut_ptr().cast();
// Keep both the handle array and the owning pipe ends alive across
// CreateProcessW. No arbitrary inheritable Host handle is admitted.
let io = io
.map(crate::extension_stdio::ChildIo::inherit)
.transpose()?;
let inherited = io.as_ref().map(|value| value.handles());
if let Some(handles) = &inherited {
if unsafe {
+55 -8
View File
@@ -46,13 +46,6 @@ impl ChildIo {
output,
error,
};
for handle in child.handles() {
if unsafe { SetHandleInformation(handle, HANDLE_FLAG_INHERIT, HANDLE_FLAG_INHERIT) }
== 0
{
return Err(HostError::new("EXTENSION_PIPE_CREATE_FAILED"));
}
}
Ok((
child,
HostIo {
@@ -62,6 +55,23 @@ impl ChildIo {
},
))
}
/// Own both the pipe ends and the launch lock. Field drop order closes all
/// inheritable ends before allowing a competing Host launch to proceed.
pub(crate) fn inherit(self) -> Result<InheritedIo> {
let lock = crate::process_creation::lock().map_err(HostError::new)?;
let guarded = InheritedIo {
child: self,
_creation: lock,
};
for handle in guarded.handles() {
if unsafe { SetHandleInformation(handle, HANDLE_FLAG_INHERIT, HANDLE_FLAG_INHERIT) }
== 0
{
return Err(HostError::new("EXTENSION_PIPE_CREATE_FAILED"));
}
}
Ok(guarded)
}
pub(crate) fn handles(&self) -> [HANDLE; 3] {
[
self.input.as_raw_handle(),
@@ -71,6 +81,16 @@ impl ChildIo {
}
}
pub(crate) struct InheritedIo {
child: ChildIo,
_creation: std::sync::MutexGuard<'static, ()>,
}
impl InheritedIo {
pub(crate) fn handles(&self) -> [HANDLE; 3] {
self.child.handles()
}
}
/// NDJSON maximum excludes the line terminator. A protocol/IO error poisons
/// the decoder; the runtime must terminate the instance and close its pipes.
/// This synchronous decoder needs a separate IO cancellation/deadline owner.
@@ -148,7 +168,7 @@ pub fn write_frame(writer: &mut impl std::io::Write, frame: &[u8]) -> Result<()>
mod tests {
use super::*;
#[test]
fn only_child_ends_are_inheritable_and_all_streams_are_distinct() {
fn pipes_are_private_until_the_serialized_creation_window() {
let (child, host) = ChildIo::create().unwrap();
let ends = child
.handles()
@@ -159,12 +179,39 @@ mod tests {
host.error.as_raw_handle(),
])
.collect::<Vec<_>>();
for handle in &ends {
let mut flags = 0;
assert_ne!(unsafe { GetHandleInformation(*handle, &mut flags) }, 0);
assert_eq!(flags & HANDLE_FLAG_INHERIT, 0);
}
let child = child.inherit().unwrap();
for (index, handle) in ends.iter().enumerate() {
let mut flags = 0;
assert_ne!(unsafe { GetHandleInformation(*handle, &mut flags) }, 0);
assert_eq!(flags & HANDLE_FLAG_INHERIT != 0, index < 3);
assert!(!ends[..index].contains(handle));
}
let (started_tx, started_rx) = std::sync::mpsc::channel();
let (acquired_tx, acquired_rx) = std::sync::mpsc::channel();
let worker = std::thread::spawn(move || {
started_tx.send(()).unwrap();
let _creation = crate::process_creation::lock().unwrap();
acquired_tx.send(()).unwrap();
});
started_rx.recv().unwrap();
assert!(acquired_rx
.recv_timeout(std::time::Duration::from_millis(30))
.is_err());
drop(child);
acquired_rx
.recv_timeout(std::time::Duration::from_secs(5))
.unwrap();
worker.join().unwrap();
// The last writer was closed before the lock was released; no child
// process was launched in this ownership test, so Host sees EOF.
use std::io::Read;
let mut output = host.output;
assert_eq!(output.read(&mut [0; 1]).unwrap(), 0);
}
#[test]
fn frames_are_bounded_across_fragmentation_and_poison_after_errors() {
+3
View File
@@ -96,3 +96,6 @@ pub mod extension_call_authorization;
#[cfg(all(windows, feature = "desktop"))]
pub mod extension_instance;
#[cfg(windows)]
mod process_creation;
@@ -0,0 +1,8 @@
//! Coordinate Host-controlled Windows launches while inheritable handles exist.
//! This does not serialize foreign libraries that bypass this Host boundary.
use std::sync::{Mutex, MutexGuard};
static CREATION: Mutex<()> = Mutex::new(());
pub(crate) fn lock() -> Result<MutexGuard<'static, ()>, &'static str> {
CREATION.lock().map_err(|_| "PROCESS_CREATION_LOCK_FAILED")
}