io: remove zeroing for AsyncRead implementors (#2525)

6 files changed, 22 insertions, 1 deletions

diff --git a/tokio/src/io/async_read.rs b/tokio/src/io/async_read.rs
index cc9091c9..a42c4ff2 100644
--- a/tokio/src/io/async_read.rs
+++ b/tokio/src/io/async_read.rs
@@ -76,7 +76,7 @@ pub trait AsyncRead {
     /// [`poll_read_buf`]: #method.poll_read_buf
     unsafe fn prepare_uninitialized_buffer(&self, buf: &mut [MaybeUninit<u8>]) -> bool {
         for x in buf {
-            *x.as_mut_ptr() = 0;
+            *x = MaybeUninit::new(0);
         }
 
         true
diff --git a/tokio/src/io/mod.rs b/tokio/src/io/mod.rs
index eddd6122..cb975f37 100644
--- a/tokio/src/io/mod.rs
+++ b/tokio/src/io/mod.rs
@@ -183,6 +183,7 @@ mod async_buf_read;
 pub use self::async_buf_read::AsyncBufRead;
 
 mod async_read;
+
 pub use self::async_read::AsyncRead;
 
 mod async_seek;
diff --git a/tokio/src/io/stdin.rs b/tokio/src/io/stdin.rs
index d986d3ab..325b8757 100644
--- a/tokio/src/io/stdin.rs
+++ b/tokio/src/io/stdin.rs
@@ -63,6 +63,11 @@ impl std::os::windows::io::AsRawHandle for Stdin {
 }
 
 impl AsyncRead for Stdin {
+    unsafe fn prepare_uninitialized_buffer(&self, _buf: &mut [std::mem::MaybeUninit<u8>]) -> bool {
+        // https://github.com/rust-lang/rust/blob/09c817eeb29e764cfc12d0a8d94841e3ffe34023/src/libstd/io/stdio.rs#L97
+        false
+    }
+
     fn poll_read(
         mut self: Pin<&mut Self>,
         cx: &mut Context<'_>,
diff --git a/tokio/src/io/util/chain.rs b/tokio/src/io/util/chain.rs
index bc76af34..8ba9194f 100644
--- a/tokio/src/io/util/chain.rs
+++ b/tokio/src/io/util/chain.rs
@@ -84,6 +84,15 @@ where
     T: AsyncRead,
     U: AsyncRead,
 {
+    unsafe fn prepare_uninitialized_buffer(&self, buf: &mut [std::mem::MaybeUninit<u8>]) -> bool {
+        if self.first.prepare_uninitialized_buffer(buf) {
+            return true;
+        }
+        if self.second.prepare_uninitialized_buffer(buf) {
+            return true;
+        }
+        false
+    }
     fn poll_read(
         self: Pin<&mut Self>,
         cx: &mut Context<'_>,
diff --git a/tokio/src/io/util/empty.rs b/tokio/src/io/util/empty.rs
index 121102c7..576058d5 100644
--- a/tokio/src/io/util/empty.rs
+++ b/tokio/src/io/util/empty.rs
@@ -47,6 +47,9 @@ cfg_io_util! {
 }
 
 impl AsyncRead for Empty {
+    unsafe fn prepare_uninitialized_buffer(&self, _buf: &mut [std::mem::MaybeUninit<u8>]) -> bool {
+        false
+    }
     #[inline]
     fn poll_read(
         self: Pin<&mut Self>,
diff --git a/tokio/src/io/util/repeat.rs b/tokio/src/io/util/repeat.rs
index 6b9067e8..eeef7cc1 100644
--- a/tokio/src/io/util/repeat.rs
+++ b/tokio/src/io/util/repeat.rs
@@ -47,6 +47,9 @@ cfg_io_util! {
 }
 
 impl AsyncRead for Repeat {
+    unsafe fn prepare_uninitialized_buffer(&self, _buf: &mut [std::mem::MaybeUninit<u8>]) -> bool {
+        false
+    }
     #[inline]
     fn poll_read(
         self: Pin<&mut Self>,