diff options
Diffstat (limited to 'src/command/tokio.rs')
-rw-r--r-- | src/command/tokio.rs | 36 |
1 files changed, 36 insertions, 0 deletions
diff --git a/src/command/tokio.rs b/src/command/tokio.rs new file mode 100644 index 0000000..f29c247 --- /dev/null +++ b/src/command/tokio.rs @@ -0,0 +1,36 @@ +use crate::error::*; + +use std::os::unix::io::{AsRawFd as _, FromRawFd as _}; + +impl super::Command<tokio::process::Child> for tokio::process::Command { + fn spawn_pty( + &mut self, + size: Option<&crate::pty::Size>, + ) -> Result<super::Child<tokio::process::Child>> { + let (pty, pts, stdin, stdout, stderr) = super::setup_pty(size)?; + + let pt_fd = pty.pt().as_raw_fd(); + let pts_fd = pts.as_raw_fd(); + + // safe because the fds are valid (otherwise pty.pts() or dup() would + // have returned an Err and we would have exited early) and are not + // owned by any other structure (since dup() returns a fresh copy of + // the file descriptor), allowing from_raw_fd to take ownership of it. + self.stdin(unsafe { std::process::Stdio::from_raw_fd(stdin) }) + .stdout(unsafe { std::process::Stdio::from_raw_fd(stdout) }) + .stderr(unsafe { std::process::Stdio::from_raw_fd(stderr) }); + + // safe because setsid() and close() are async-signal-safe functions + // and ioctl() is a raw syscall (which is inherently + // async-signal-safe). + unsafe { + self.pre_exec(move || { + super::pre_exec(pt_fd, pts_fd, stdin, stdout, stderr) + }); + } + + let child = self.spawn().map_err(Error::Spawn)?; + + Ok(super::Child { child, pty }) + } +} |