feat(fs,unix): add impl TryAsRawFd for stdio

Author: Berrysoft

compio-fs/src/stdio/mod.rs

@@ -11,11 +11,12 @@ cfg_if::cfg_if! {
 /// Constructs a new handle to the standard input of the current process.
 ///
 /// ## Platform specific
-/// * Windows: This handle is best used for non-interactive uses, such as when a file
-/// is piped into the application. For technical reasons, if `stdin` is a
-/// console handle, the read method is implemented by using an ordinary blocking
-/// read on a separate thread, and it is impossible to cancel that read. This
-/// can make shutdown of the runtime hang until the user presses enter.
+/// * Windows: This handle is best used for non-interactive uses, such as when a
+///   file is piped into the application. For technical reasons, if `stdin` is a
+///   console handle, the read method is implemented by using an ordinary
+///   blocking read on a separate thread, and it is impossible to cancel that
+///   read. This can make shutdown of the runtime hang until the user presses
+///   enter.
 ///
 /// [`AsyncRead`]: compio_io::AsyncRead
 pub fn stdin() -> Stdin {

compio-fs/src/stdio/unix.rs

@@ -1,10 +1,11 @@
-use std::mem::ManuallyDrop;
+use std::{io, mem::ManuallyDrop};
 
-use crate::pipe::{Receiver, Sender};
 use compio_buf::{BufResult, IoBuf, IoBufMut};
-use compio_driver::AsRawFd;
+use compio_driver::{AsRawFd, FromRawFd, RawFd};
 use compio_io::{AsyncRead, AsyncWrite};
-use compio_runtime::FromRawFd;
+use compio_runtime::TryAsRawFd;
+
+use crate::pipe::{Receiver, Sender};
 
 /// A handle to the standard input stream of a process.
 ///
@@ -14,7 +15,7 @@ pub struct Stdin(ManuallyDrop<Receiver>);
 impl Stdin {
     pub(crate) fn new() -> Self {
         Self(ManuallyDrop::new(unsafe {
-            Receiver::from_raw_fd(std::io::stdin().as_raw_fd())
+            Receiver::from_raw_fd(io::stdin().as_raw_fd())
         }))
     }
 }
@@ -25,6 +26,16 @@ impl AsyncRead for Stdin {
     }
 }
 
+impl TryAsRawFd for Stdin {
+    fn try_as_raw_fd(&self) -> io::Result<RawFd> {
+        self.0.try_as_raw_fd()
+    }
+
+    unsafe fn as_raw_fd_unchecked(&self) -> RawFd {
+        self.0.as_raw_fd_unchecked()
+    }
+}
+
 /// A handle to the standard output stream of a process.
 ///
 /// See [`stdout`].
@@ -33,7 +44,7 @@ pub struct Stdout(ManuallyDrop<Sender>);
 impl Stdout {
     pub(crate) fn new() -> Self {
         Self(ManuallyDrop::new(unsafe {
-            Sender::from_raw_fd(std::io::stdout().as_raw_fd())
+            Sender::from_raw_fd(io::stdout().as_raw_fd())
         }))
     }
 }
@@ -43,15 +54,25 @@ impl AsyncWrite for Stdout {
         self.0.write(buf).await
     }
 
-    async fn flush(&mut self) -> std::io::Result<()> {
+    async fn flush(&mut self) -> io::Result<()> {
         self.0.flush().await
     }
 
-    async fn shutdown(&mut self) -> std::io::Result<()> {
+    async fn shutdown(&mut self) -> io::Result<()> {
         self.0.shutdown().await
     }
 }
 
+impl TryAsRawFd for Stdout {
+    fn try_as_raw_fd(&self) -> io::Result<RawFd> {
+        self.0.try_as_raw_fd()
+    }
+
+    unsafe fn as_raw_fd_unchecked(&self) -> RawFd {
+        self.0.as_raw_fd_unchecked()
+    }
+}
+
 /// A handle to the standard output stream of a process.
 ///
 /// See [`stderr`].
@@ -60,7 +81,7 @@ pub struct Stderr(ManuallyDrop<Sender>);
 impl Stderr {
     pub(crate) fn new() -> Self {
         Self(ManuallyDrop::new(unsafe {
-            Sender::from_raw_fd(std::io::stderr().as_raw_fd())
+            Sender::from_raw_fd(io::stderr().as_raw_fd())
         }))
     }
 }
@@ -70,11 +91,21 @@ impl AsyncWrite for Stderr {
         self.0.write(buf).await
     }
 
-    async fn flush(&mut self) -> std::io::Result<()> {
+    async fn flush(&mut self) -> io::Result<()> {
         self.0.flush().await
     }
 
-    async fn shutdown(&mut self) -> std::io::Result<()> {
+    async fn shutdown(&mut self) -> io::Result<()> {
         self.0.shutdown().await
     }
 }
+
+impl TryAsRawFd for Stderr {
+    fn try_as_raw_fd(&self) -> io::Result<RawFd> {
+        self.0.try_as_raw_fd()
+    }
+
+    unsafe fn as_raw_fd_unchecked(&self) -> RawFd {
+        self.0.as_raw_fd_unchecked()
+    }
+}