Implement bytes::Buf for Vec.

Author: reitermarkus

src/bytes.rs

@@ -1,8 +1,34 @@
 //! Bytes implementations for heapless types
 
 use crate::Vec;
-use bytes::buf::UninitSlice;
-use bytes::BufMut;
+use bytes::{buf::UninitSlice, Buf, BufMut};
+
+unsafe impl<const N: usize> Buf for Vec<u8, N> {
+    #[inline]
+    fn remaining(&self) -> usize {
+        self.len()
+    }
+
+    #[inline]
+    fn chunk(&mut self) -> &[u8] {
+        self.as_slice()
+    }
+
+    #[inline]
+    unsafe fn advance(&mut self, cnt: usize) {
+        assert!(
+            cnt <= self.remaining(),
+            "cannot advance past `remaining`: {:?} <= {:?}",
+            cnt,
+            self.remaining(),
+        );
+        unsafe {
+            // SAFETY: We've checked that `cnt` <= `self.remaining()` and we know that
+            // `self.remaining()` <= `self.cap`.
+            self.advance_unchecked(cnt);
+        }
+    }
+}
 
 unsafe impl<const N: usize> BufMut for Vec<u8, N> {
     #[inline]
@@ -35,21 +61,44 @@ mod tests {
 
     #[test]
     #[should_panic]
-    fn advance_out_of_bound() {
+    fn buf_advance_out_of_bounds() {
+        let mut vec: Vec<u8, 8> = Vec::new();
+        vec.advance(9)
+    }
+
+    #[test]
+    fn buf_remaining() {
+        let mut vec: Vec<u8, 8> = Vec::new();
+        assert_eq!(vec.remaining(), 8);
+        vec.push(42).unwrap();
+        assert_eq!(vec.remaining(), 7);
+    }
+
+    #[test]
+    fn buf_chunk() {
+        let mut vec: Vec<u8, 8> = Vec::new();
+        assert_eq!(vec.chunk().len(), 8);
+        unsafe { vec.advance_mut(1) };
+        assert_eq!(vec.chunk().len(), 7);
+    }
+
+    #[test]
+    #[should_panic]
+    fn buf_mut_advance_mut_out_of_bounds() {
         let mut vec: Vec<u8, 8> = Vec::new();
         unsafe { vec.advance_mut(9) };
     }
 
     #[test]
-    fn remaining_mut() {
+    fn buf_mut_remaining_mut() {
         let mut vec: Vec<u8, 8> = Vec::new();
         assert_eq!(vec.remaining_mut(), 8);
         vec.push(42).unwrap();
         assert_eq!(vec.remaining_mut(), 7);
     }
 
     #[test]
-    fn chunk_mut() {
+    fn buf_mut_chunk_mut() {
         let mut vec: Vec<u8, 8> = Vec::new();
         assert_eq!(vec.chunk_mut().len(), 8);
         unsafe { vec.advance_mut(1) };