unsafe impl<'a> BufMut for ReadBuf<'a>

tokio/src/io/read_buf.rs

@@ -1,3 +1,6 @@
+#[cfg(feature = "bytes")]
+use bytes::{buf::UninitSlice, BufMut};
+
 use std::fmt;
 use std::mem::MaybeUninit;
 
@@ -270,6 +273,23 @@ impl<'a> ReadBuf<'a> {
     }
 }
 
+#[cfg(feature = "bytes")]
+unsafe impl<'a> BufMut for ReadBuf<'a> {
+    fn remaining_mut(&self) -> usize {
+        self.remaining()
+    }
+
+    unsafe fn advance_mut(&mut self, cnt: usize) {
+        self.advance(cnt);
+    }
+
+    fn chunk_mut(&mut self) -> &mut UninitSlice {
+        let len = self.filled().len();
+        let ptr = unsafe { self.unfilled_mut().as_mut_ptr() };
+        unsafe { UninitSlice::from_raw_parts_mut(ptr as *mut u8, len) }
+    }
+}
+
 impl fmt::Debug for ReadBuf<'_> {
     fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
         f.debug_struct("ReadBuf")