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
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81

macro_rules! impl_read_num {
    ($ty:ident, le: $read_le:ident, be: $read_be:ident) => {
        impl_read_num!($ty, stringify!($ty), le: $read_le, be: $read_be);
    };
    ($ty:ident, $ty_str:expr, le: $read_le:ident, be: $read_be:ident) => {
        #[doc = "Read a little-endian encoded `"]
        #[doc = $ty_str]
        #[doc = "` from the reader."]
        ///
        /// # Errors
        ///
        /// Returns an error if there is not sufficient input left to read.
        pub fn $read_le(&mut self) -> Result<$ty, E>
        where
            E: From<ExpectedLength<'i>>,
        {
            read_num!(self, E, $ty, concat!("little-endian ", $ty_str), from_le_bytes)
        }

        #[doc = "Read a big-endian encoded `"]
        #[doc = $ty_str]
        #[doc = "` from the reader."]
        ///
        /// # Errors
        ///
        /// Returns an error if there is not sufficient input left to read.
        pub fn $read_be(&mut self) -> Result<$ty, E>
        where
            E: From<ExpectedLength<'i>>,
        {
            read_num!(self, E, $ty, concat!("big-endian ", $ty_str), from_be_bytes)
        }
    };
}

macro_rules! read_num {
    ($reader:expr, $err_ty:ident, $num_ty:ident, $expected:expr, $from_xx_bytes:ident) => {{
        let (arr, tail) = split_arr!($reader.input, $num_ty, concat!("read ", $expected))?;
        $reader.input = tail;
        Ok(<$num_ty>::$from_xx_bytes(arr))
    }};
}

macro_rules! split_arr {
    ($input:expr, u8, $expected:expr) => {
        $input.split_arr_1($expected)
    };
    ($input:expr, u16, $expected:expr) => {
        $input.split_arr_2($expected)
    };
    ($input:expr, u32, $expected:expr) => {
        $input.split_arr_4($expected)
    };
    ($input:expr, u64, $expected:expr) => {
        $input.split_arr_8($expected)
    };
    ($input:expr, u128, $expected:expr) => {
        $input.split_arr_16($expected)
    };
    ($input:expr, i8, $expected:expr) => {
        $input.split_arr_1($expected)
    };
    ($input:expr, i16, $expected:expr) => {
        $input.split_arr_2($expected)
    };
    ($input:expr, i32, $expected:expr) => {
        $input.split_arr_4($expected)
    };
    ($input:expr, i64, $expected:expr) => {
        $input.split_arr_8($expected)
    };
    ($input:expr, i128, $expected:expr) => {
        $input.split_arr_16($expected)
    };
    ($input:expr, f32, $expected:expr) => {
        $input.split_arr_4($expected)
    };
    ($input:expr, f64, $expected:expr) => {
        $input.split_arr_8($expected)
    };
}