forked from rust-lang/rust
-
Notifications
You must be signed in to change notification settings - Fork 7
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Initial implementation of anonymous_pipe
Co-authored-by: Alphyr <47725341+a1phyr@users.noreply.github.com> Co-authored-by: Jubilee <46493976+workingjubilee@users.noreply.github.com> Signed-off-by: Jiahao XU <Jiahao_XU@outlook.com>
- Loading branch information
3 people
committed
Jul 21, 2024
1 parent
52f3c71
commit 44071a0
Showing
15 changed files
with
522 additions
and
16 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,146 @@ | ||
| //! Module for anonymous pipe | ||
| //! | ||
| //! ``` | ||
| //! #![feature(anonymous_pipe)] | ||
| //! | ||
| //! # #[cfg(miri)] fn main() {} | ||
| //! # #[cfg(not(miri))] | ||
| //! # fn main() -> std::io::Result<()> { | ||
| //! let (reader, writer) = std::pipe::pipe()?; | ||
| //! # Ok(()) | ||
| //! # } | ||
| //! ``` | ||
|
|
||
| cfg_if::cfg_if! { | ||
| if #[cfg(unix)] { | ||
| mod unix; | ||
| use unix::{AnonPipe, pipe as pipe_inner}; | ||
|
|
||
| #[cfg(all(test, not(miri)))] | ||
| mod tests; | ||
| } else if #[cfg(windows)] { | ||
| mod windows; | ||
| use windows::{AnonPipe, pipe as pipe_inner}; | ||
|
|
||
| #[cfg(all(test, not(miri)))] | ||
| mod tests; | ||
| } else { | ||
| mod unsupported; | ||
| use unsupported::{AnonPipe, pipe as pipe_inner}; | ||
| } | ||
| } | ||
|
|
||
| use crate::io; | ||
|
|
||
| /// Create anonymous pipe that is close-on-exec and blocking. | ||
| #[unstable(feature = "anonymous_pipe", issue = "127154")] | ||
| #[inline] | ||
| pub fn pipe() -> io::Result<(PipeReader, PipeWriter)> { | ||
| pipe_inner().map(|(reader, writer)| (PipeReader(reader), PipeWriter(writer))) | ||
| } | ||
|
|
||
| /// Read end of the anonymous pipe. | ||
| #[unstable(feature = "anonymous_pipe", issue = "127154")] | ||
| #[derive(Debug)] | ||
| pub struct PipeReader(AnonPipe); | ||
|
|
||
| /// Write end of the anonymous pipe. | ||
| #[unstable(feature = "anonymous_pipe", issue = "127154")] | ||
| #[derive(Debug)] | ||
| pub struct PipeWriter(AnonPipe); | ||
|
|
||
| impl PipeReader { | ||
| /// Create a new [`PipeReader`] instance that shares the same underlying file description. | ||
| #[unstable(feature = "anonymous_pipe", issue = "127154")] | ||
| pub fn try_clone(&self) -> io::Result<Self> { | ||
| self.0.try_clone().map(Self) | ||
| } | ||
| } | ||
|
|
||
| impl PipeWriter { | ||
| /// Create a new [`PipeWriter`] instance that shares the same underlying file description. | ||
| #[unstable(feature = "anonymous_pipe", issue = "127154")] | ||
| pub fn try_clone(&self) -> io::Result<Self> { | ||
| self.0.try_clone().map(Self) | ||
| } | ||
| } | ||
|
|
||
| #[unstable(feature = "anonymous_pipe", issue = "127154")] | ||
| impl io::Read for &PipeReader { | ||
| fn read(&mut self, buf: &mut [u8]) -> io::Result<usize> { | ||
| self.0.read(buf) | ||
| } | ||
| fn read_vectored(&mut self, bufs: &mut [io::IoSliceMut<'_>]) -> io::Result<usize> { | ||
| self.0.read_vectored(bufs) | ||
| } | ||
| #[inline] | ||
| fn is_read_vectored(&self) -> bool { | ||
| self.0.is_read_vectored() | ||
| } | ||
| fn read_to_end(&mut self, buf: &mut Vec<u8>) -> io::Result<usize> { | ||
| self.0.read_to_end(buf) | ||
| } | ||
| fn read_buf(&mut self, buf: io::BorrowedCursor<'_>) -> io::Result<()> { | ||
| self.0.read_buf(buf) | ||
| } | ||
| } | ||
|
|
||
| #[unstable(feature = "anonymous_pipe", issue = "127154")] | ||
| impl io::Read for PipeReader { | ||
| fn read(&mut self, buf: &mut [u8]) -> io::Result<usize> { | ||
| self.0.read(buf) | ||
| } | ||
| fn read_vectored(&mut self, bufs: &mut [io::IoSliceMut<'_>]) -> io::Result<usize> { | ||
| self.0.read_vectored(bufs) | ||
| } | ||
| #[inline] | ||
| fn is_read_vectored(&self) -> bool { | ||
| self.0.is_read_vectored() | ||
| } | ||
| fn read_to_end(&mut self, buf: &mut Vec<u8>) -> io::Result<usize> { | ||
| self.0.read_to_end(buf) | ||
| } | ||
| fn read_buf(&mut self, buf: io::BorrowedCursor<'_>) -> io::Result<()> { | ||
| self.0.read_buf(buf) | ||
| } | ||
| } | ||
|
|
||
| #[unstable(feature = "anonymous_pipe", issue = "127154")] | ||
| impl io::Write for &PipeWriter { | ||
| fn write(&mut self, buf: &[u8]) -> io::Result<usize> { | ||
| self.0.write(buf) | ||
| } | ||
| #[inline] | ||
| fn flush(&mut self) -> io::Result<()> { | ||
| Ok(()) | ||
| } | ||
|
|
||
| fn write_vectored(&mut self, bufs: &[io::IoSlice<'_>]) -> io::Result<usize> { | ||
| self.0.write_vectored(bufs) | ||
| } | ||
|
|
||
| #[inline] | ||
| fn is_write_vectored(&self) -> bool { | ||
| self.0.is_write_vectored() | ||
| } | ||
| } | ||
|
|
||
| #[unstable(feature = "anonymous_pipe", issue = "127154")] | ||
| impl io::Write for PipeWriter { | ||
| fn write(&mut self, buf: &[u8]) -> io::Result<usize> { | ||
| self.0.write(buf) | ||
| } | ||
| #[inline] | ||
| fn flush(&mut self) -> io::Result<()> { | ||
| Ok(()) | ||
| } | ||
|
|
||
| fn write_vectored(&mut self, bufs: &[io::IoSlice<'_>]) -> io::Result<usize> { | ||
| self.0.write_vectored(bufs) | ||
| } | ||
|
|
||
| #[inline] | ||
| fn is_write_vectored(&self) -> bool { | ||
| self.0.is_write_vectored() | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,18 @@ | ||
| use super::*; | ||
| use crate::io::{Read, Write}; | ||
|
|
||
| #[test] | ||
| fn pipe_creation_clone_and_rw() { | ||
| let (rx, tx) = pipe().unwrap(); | ||
|
|
||
| tx.try_clone().unwrap().write_all(b"12345").unwrap(); | ||
| drop(tx); | ||
|
|
||
| let mut rx2 = rx.try_clone().unwrap(); | ||
| drop(rx); | ||
|
|
||
| let mut s = String::new(); | ||
| rx2.read_to_string(&mut s).unwrap(); | ||
| drop(rx2); | ||
| assert_eq!(s, "12345"); | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,103 @@ | ||
| use super::*; | ||
|
|
||
| use crate::{ | ||
| os::fd::{AsFd, AsRawFd, BorrowedFd, FromRawFd, IntoRawFd, OwnedFd, RawFd}, | ||
| process::Stdio, | ||
| sys::{fd::FileDesc, pipe::anon_pipe}, | ||
| sys_common::{FromInner, IntoInner}, | ||
| }; | ||
|
|
||
| pub(super) type AnonPipe = FileDesc; | ||
|
|
||
| #[inline] | ||
| pub(super) fn pipe() -> io::Result<(AnonPipe, AnonPipe)> { | ||
| anon_pipe().map(|(rx, wx)| (rx.into_inner(), wx.into_inner())) | ||
| } | ||
|
|
||
| #[unstable(feature = "anonymous_pipe", issue = "127154")] | ||
| impl AsFd for PipeReader { | ||
| fn as_fd(&self) -> BorrowedFd<'_> { | ||
| self.0.as_fd() | ||
| } | ||
| } | ||
| #[unstable(feature = "anonymous_pipe", issue = "127154")] | ||
| impl AsRawFd for PipeReader { | ||
| fn as_raw_fd(&self) -> RawFd { | ||
| self.0.as_raw_fd() | ||
| } | ||
| } | ||
| #[unstable(feature = "anonymous_pipe", issue = "127154")] | ||
| impl From<PipeReader> for OwnedFd { | ||
| fn from(pipe: PipeReader) -> Self { | ||
| FileDesc::into_inner(pipe.0) | ||
| } | ||
| } | ||
| #[unstable(feature = "anonymous_pipe", issue = "127154")] | ||
| impl FromRawFd for PipeReader { | ||
| unsafe fn from_raw_fd(raw_fd: RawFd) -> Self { | ||
| Self(FileDesc::from_raw_fd(raw_fd)) | ||
| } | ||
| } | ||
| #[unstable(feature = "anonymous_pipe", issue = "127154")] | ||
| impl IntoRawFd for PipeReader { | ||
| fn into_raw_fd(self) -> RawFd { | ||
| self.0.into_raw_fd() | ||
| } | ||
| } | ||
| #[unstable(feature = "anonymous_pipe", issue = "127154")] | ||
| impl From<PipeReader> for Stdio { | ||
| fn from(pipe: PipeReader) -> Self { | ||
| Self::from(OwnedFd::from(pipe)) | ||
| } | ||
| } | ||
|
|
||
| #[unstable(feature = "anonymous_pipe", issue = "127154")] | ||
| impl AsFd for PipeWriter { | ||
| fn as_fd(&self) -> BorrowedFd<'_> { | ||
| self.0.as_fd() | ||
| } | ||
| } | ||
| #[unstable(feature = "anonymous_pipe", issue = "127154")] | ||
| impl AsRawFd for PipeWriter { | ||
| fn as_raw_fd(&self) -> RawFd { | ||
| self.0.as_raw_fd() | ||
| } | ||
| } | ||
| #[unstable(feature = "anonymous_pipe", issue = "127154")] | ||
| impl From<PipeWriter> for OwnedFd { | ||
| fn from(pipe: PipeWriter) -> Self { | ||
| FileDesc::into_inner(pipe.0) | ||
| } | ||
| } | ||
| #[unstable(feature = "anonymous_pipe", issue = "127154")] | ||
| impl FromRawFd for PipeWriter { | ||
| unsafe fn from_raw_fd(raw_fd: RawFd) -> Self { | ||
| Self(FileDesc::from_raw_fd(raw_fd)) | ||
| } | ||
| } | ||
| #[unstable(feature = "anonymous_pipe", issue = "127154")] | ||
| impl IntoRawFd for PipeWriter { | ||
| fn into_raw_fd(self) -> RawFd { | ||
| self.0.into_raw_fd() | ||
| } | ||
| } | ||
| #[unstable(feature = "anonymous_pipe", issue = "127154")] | ||
| impl From<PipeWriter> for Stdio { | ||
| fn from(pipe: PipeWriter) -> Self { | ||
| Self::from(OwnedFd::from(pipe)) | ||
| } | ||
| } | ||
|
|
||
| #[unstable(feature = "anonymous_pipe", issue = "127154")] | ||
| impl From<OwnedFd> for PipeReader { | ||
| fn from(owned_fd: OwnedFd) -> Self { | ||
| Self(FileDesc::from_inner(owned_fd)) | ||
| } | ||
| } | ||
|
|
||
| #[unstable(feature = "anonymous_pipe", issue = "127154")] | ||
| impl From<OwnedFd> for PipeWriter { | ||
| fn from(owned_fd: OwnedFd) -> Self { | ||
| Self(FileDesc::from_inner(owned_fd)) | ||
| } | ||
| } |
Oops, something went wrong.