From 016fdbdc7a95f6764186dacfcaeef94d77e89cd1 Mon Sep 17 00:00:00 2001
From: Stepan Koltsov <stepan.koltsov@gmail.com>
Date: Tue, 19 Aug 2025 11:50:38 +0100
Subject: [PATCH] Reserve capacity in BytesMut::put (#794)

---
 src/bytes_mut.rs | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/src/bytes_mut.rs b/src/bytes_mut.rs
index ad3aa77..d8d04d3 100644
--- a/src/bytes_mut.rs
+++ b/src/bytes_mut.rs
@@ -1210,6 +1210,9 @@ unsafe impl BufMut for BytesMut {
                 Err(bytes) => self.extend_from_slice(&bytes),
             }
         } else {
+            // In case the src isn't contiguous, reserve upfront.
+            self.reserve(src.remaining());
+
             while src.has_remaining() {
                 let s = src.chunk();
                 let l = s.len();