Handle limit == 0

tokio-rs · Jan 13, 2025 · 2310cd3 · 2310cd3
1 parent 6000c62
commit 2310cd3
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/buf/take.rs b/src/buf/take.rs
@@ -158,6 +158,10 @@ impl<T: Buf> Buf for Take<T> {
 
     #[cfg(feature = "std")]
     fn chunks_vectored<'a>(&'a self, dst: &mut [IoSlice<'a>]) -> usize {
+        if self.limit == 0 {
+            return 0;
+        }
+
         let mut slices = [IoSlice::new(&[]); 16];
 
         let cnt = self.inner.chunks_vectored(&mut slices[..dst.len().min(16)]);