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
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103

//! Encode/Decode implementations for primitive types

use core::fmt::Debug;

use byteorder::{LittleEndian, ByteOrder};

use crate::{Encode, Decode, Error};

/// Helper trait to implement encode/decode on fixed size types
trait FixedEncDec: Sized {
    const N: usize;

    fn e(&self, buff: &mut [u8]) -> ();
    fn d(buff: &[u8]) -> Self;
}

impl <T: FixedEncDec + Debug + Sized> Decode for T {
    type Error = Error;

    fn decode<'a>(buff: &'a[u8]) -> Result<(Self, usize), Self::Error> {
        if buff.len() < T::N {
            return Err(Error::BufferOverrun);
        }

        let d = T::d(&buff[..T::N]);

        Ok((d, T::N))
    }
}

impl <T: FixedEncDec + Debug + Sized> Encode for T {
    type Error = Error;

    fn encode_len(&self) -> Result<usize, Self::Error> {
        Ok(T::N)
    }

    fn encode(&self, buff: &mut [u8]) -> Result<usize, Self::Error> {
        if buff.len() < T::N {
            return Err(Error::BufferOverrun);
        }

        T::e(self, &mut buff[..T::N]);

        Ok(T::N)
    }
}

impl FixedEncDec for u8 {
    const N: usize = 1;

    fn e(&self, buff: &mut [u8]) -> () { buff[0] = *self; }
    fn d(buff: &[u8]) -> Self { buff[0] }
}

impl FixedEncDec for u16 {
    const N: usize = 2;

    fn e(&self, buff: &mut [u8]) -> () { LittleEndian::write_u16(&mut buff[..Self::N], *self) }
    fn d(buff: &[u8]) -> Self { LittleEndian::read_u16(&buff[..Self::N]) }
}

impl FixedEncDec for u32 {
    const N: usize = 4;

    fn e(&self, buff: &mut [u8]) -> () { LittleEndian::write_u32(&mut buff[..Self::N], *self) }
    fn d(buff: &[u8]) -> Self { LittleEndian::read_u32(&buff[..Self::N]) }
}

impl FixedEncDec for u64 {
    const N: usize = 8;

    fn e(&self, buff: &mut [u8]) -> () { LittleEndian::write_u64(&mut buff[..Self::N], *self) }
    fn d(buff: &[u8]) -> Self { LittleEndian::read_u64(&buff[..Self::N]) }
}

impl FixedEncDec for i8 {
    const N: usize = 1;

    fn e(&self, buff: &mut [u8]) -> () { buff[0] = *self as u8; }
    fn d(buff: &[u8]) -> Self { buff[0] as i8 }
}

impl FixedEncDec for i16 {
    const N: usize = 2;

    fn e(&self, buff: &mut [u8]) -> () { LittleEndian::write_i16(&mut buff[..Self::N], *self) }
    fn d(buff: &[u8]) -> Self { LittleEndian::read_i16(&buff[..Self::N]) }
}

impl FixedEncDec for i32 {
    const N: usize = 4;

    fn e(&self, buff: &mut [u8]) -> () { LittleEndian::write_i32(&mut buff[..Self::N], *self) }
    fn d(buff: &[u8]) -> Self { LittleEndian::read_i32(&buff[..Self::N]) }
}

impl FixedEncDec for i64 {
    const N: usize = 8;

    fn e(&self, buff: &mut [u8]) -> () { LittleEndian::write_i64(&mut buff[..Self::N], *self) }
    fn d(buff: &[u8]) -> Self { LittleEndian::read_i64(&buff[..Self::N]) }
}