io: add `ReadBuf::take` (#2817)

Signed-off-by: Zahari Dichev <zaharidichev@gmail.com>

1 files changed, 1 insertions, 4 deletions

diff --git a/tokio/src/io/util/take.rs b/tokio/src/io/util/take.rs
index 2abc7693..4e424f6c 100644
--- a/tokio/src/io/util/take.rs
+++ b/tokio/src/io/util/take.rs
@@ -85,10 +85,7 @@ impl<R: AsyncRead> AsyncRead for Take<R> {
         }
 
         let me = self.project();
-        let max = std::cmp::min(buf.remaining() as u64, *me.limit_) as usize;
-        // Make a ReadBuf of the unfulled section up to max
-        // Saftey: We don't set any of the `unfilled_mut` with `MaybeUninit::uninit`.
-        let mut b = unsafe { ReadBuf::uninit(&mut buf.unfilled_mut()[..max]) };
+        let mut b = buf.take(*me.limit_ as usize);
         ready!(me.inner.poll_read(cx, &mut b))?;
         let n = b.filled().len();