From 4202f3be439e9d453a13c4d3fd5eeafa84989560 Mon Sep 17 00:00:00 2001
From: roblabla <unfiltered@roblab.la>
Date: Mon, 18 Jan 2021 12:58:39 +0000
Subject: [PATCH] Add UninitSlice::split_at_mut

---
 src/buf/uninit_slice.rs | 33 +++++++++++++++++++++++++++++++++
 1 file changed, 33 insertions(+)

diff --git a/src/buf/uninit_slice.rs b/src/buf/uninit_slice.rs
index 73f4e8924..41af16b9a 100644
--- a/src/buf/uninit_slice.rs
+++ b/src/buf/uninit_slice.rs
@@ -136,6 +136,39 @@ impl UninitSlice {
     pub fn len(&self) -> usize {
         self.0.len()
     }
+
+    /// Divides one mutable slice into two at an index.
+    ///
+    /// The first will contain all indices from `[0, mid)` (excluding the index
+    /// `mid` itself) and the second will contain all indices from [mid, len)
+    /// (excluding the index `len` itself).
+    ///
+    /// # Panics
+    ///
+    /// Panics if `mid > len`.
+    ///
+    /// # Examples
+    ///
+    /// ```
+    /// use bytes::BufMut;
+    ///
+    /// let mut data = [0, 1, 2];
+    /// let mut slice = &mut data[..];
+    /// let chunk = BufMut::chunk_mut(&mut slice);
+    ///
+    /// let (left, right) = chunk.split_at_mut(2);
+    /// left.copy_from_slice(&[3, 4]);
+    ///
+    /// assert_eq!(data, [3, 4, 2]);
+    /// ```
+    pub fn split_at_mut(&mut self, mid: usize) -> (&mut UninitSlice, &mut UninitSlice) {
+        let (left, right) = self.0.split_at_mut(mid);
+        unsafe {
+            let left = UninitSlice::from_raw_parts_mut(left.as_mut_ptr() as *mut u8, left.len());
+            let right = UninitSlice::from_raw_parts_mut(right.as_mut_ptr() as *mut u8, right.len());
+            (left, right)
+        }
+    }
 }
 
 impl fmt::Debug for UninitSlice {