diff options
| -rw-r--r-- | src/buf/buf_mut.rs | 8 | ||||
| -rw-r--r-- | tests/test_buf_mut.rs | 3 |
2 files changed, 11 insertions, 0 deletions
diff --git a/src/buf/buf_mut.rs b/src/buf/buf_mut.rs index eccd509..844474f 100644 --- a/src/buf/buf_mut.rs +++ b/src/buf/buf_mut.rs @@ -1012,6 +1012,14 @@ unsafe impl BufMut for &mut [u8] { let (_, b) = core::mem::replace(self, &mut []).split_at_mut(cnt); *self = b; } + + #[inline] + fn put_slice(&mut self, src: &[u8]) { + self[..src.len()].copy_from_slice(src); + unsafe { + self.advance_mut(src.len()); + } + } } unsafe impl BufMut for Vec<u8> { diff --git a/tests/test_buf_mut.rs b/tests/test_buf_mut.rs index 8d270e3..b85ab9c 100644 --- a/tests/test_buf_mut.rs +++ b/tests/test_buf_mut.rs @@ -70,6 +70,9 @@ fn test_mut_slice() { let mut v = vec![0, 0, 0, 0]; let mut s = &mut v[..]; s.put_u32(42); + + assert_eq!(s.len(), 0); + assert_eq!(&v, &[0, 0, 0, 42]); } #[test] |