1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56

use super::*;

pub trait Bytes {
    /// # Panics
    /// This function may panic, if the data is slice `&mut [u8]`, and has not enough capacity.
    ///
    /// But If the data is vector `Vec<u8>`, then it may reserve extra capacity if necessary.
    fn write_slice_at(&mut self, offset: usize, slice: impl AsRef<[u8]>) -> usize;

    /// # Warning
    /// #### This function is very very very unsafe!!! Should be used with care.
    #[doc(hidden)]
    unsafe fn new_len(&mut self, total_len: usize, count: usize);

    fn as_ref(&mut self) -> &[u8];
    fn as_mut(&mut self) -> &mut [u8];
}

macro_rules! impls {
    [$($ty:ty),* : $func:item] => {$(
        impl Bytes for $ty {
            #[inline]
            fn write_slice_at(&mut self, offset: usize, slice: impl AsRef<[u8]>) -> usize {
                let src = slice.as_ref();
                let count = src.len();
                unsafe {
                    let total_len = offset + count;
                    self.new_len(total_len, count);
                    ptr::copy_nonoverlapping(src.as_ptr(), self.as_mut_ptr().add(offset), count);
                    total_len
                }
            }
            #[inline] fn as_ref(&mut self) -> &[u8] { self }
            #[inline] fn as_mut(&mut self) -> &mut [u8] { self }
            
            #[inline]
            #[doc(hidden)]
            $func
        }
    )*};
}

impls!(&mut Vec<u8>, Vec<u8>:
    unsafe fn new_len(&mut self, total_len: usize, count: usize) {
        if total_len > self.len() {
            self.reserve(count);
            #[allow(clippy::uninit_vec)]
            self.set_len(total_len);
        }
    }
);
impls!(&mut [u8]:
    unsafe fn new_len(&mut self, total_len: usize, _: usize) {
        assert!(total_len <= self.len(), "total len: {total_len} <= buffer len: {}", self.len());
    }
);