io: fix integer overflow in take (#6080)

tokio-rs · Oct 15, 2023 · 654a3d5 · 654a3d5
1 parent a08ad92
commit 654a3d5
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tokio/src/io/util/take.rs b/tokio/src/io/util/take.rs
@@ -1,6 +1,7 @@
 use crate::io::{AsyncBufRead, AsyncRead, ReadBuf};
 
 use pin_project_lite::pin_project;
+use std::convert::TryFrom;
 use std::pin::Pin;
 use std::task::{Context, Poll};
 use std::{cmp, io};
@@ -85,7 +86,7 @@ impl<R: AsyncRead> AsyncRead for Take<R> {
         }
 
         let me = self.project();
-        let mut b = buf.take(*me.limit_ as usize);
+        let mut b = buf.take(usize::try_from(*me.limit_).unwrap_or(usize::MAX));
 
         let buf_ptr = b.filled().as_ptr();
         ready!(me.inner.poll_read(cx, &mut b))?;