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
use super::errors::{BadLength, Error};
pub trait Serializable<const N: usize> {
const SIZE: usize = N;
type Error;
fn from_bytes(buf: &[u8; N]) -> Result<Self, Self::Error>
where
Self: Sized;
fn to_bytes(&self) -> [u8; N];
}
pub trait DeserializableSlice<const N: usize>: Serializable<N> {
fn from_slice(buf: &[u8]) -> Result<Self, Self::Error>
where
Self: Sized,
Self::Error: BadLength,
{
if buf.len() < N {
Err(Self::Error::bad_length(buf.len(), N))
} else {
let mut bytes = [0u8; N];
(&mut bytes[..N]).copy_from_slice(&buf[..N]);
Self::from_bytes(&bytes)
}
}
fn from_reader<R>(buf: &mut R) -> Result<Self, Self::Error>
where
R: Read,
Self: Sized,
Self::Error: BadLength,
{
let mut bytes = [0u8; N];
buf.read(&mut bytes)
.map_err(|_| Self::Error::bad_length(buf.size(), N))?;
Self::from_bytes(&bytes)
}
}
impl<T, const N: usize> DeserializableSlice<N> for T where T: Serializable<N> {}
pub trait Read {
fn size(&self) -> usize;
fn read(&mut self, buf: &mut [u8]) -> Result<usize, Error>;
}
impl Read for &[u8] {
#[inline]
fn size(&self) -> usize {
self.len()
}
#[inline]
fn read(&mut self, buf: &mut [u8]) -> Result<usize, Error> {
if buf.len() > self.len() {
return Err(Error::bad_length(self.len(), buf.len()));
}
let amt = buf.len();
let (a, b) = self.split_at(amt);
if amt == 1 {
buf[0] = a[0];
} else {
buf[..amt].copy_from_slice(a);
}
*self = b;
Ok(amt)
}
}