encdec_base/encode/
prefixed.rs

1use core::fmt::Debug;
2
3use num_traits::FromPrimitive;
4
5use super::Encode;
6use crate::Error;
7
8/// Encode for fields with length prefixes
9pub trait EncodePrefixed<P: Encode> {
10    /// Error type returned on parse error
11    type Error: From<Error> + Debug;
12
13    /// Parse method consumes a slice and returns an object
14    fn encode_prefixed(&self, buff: &mut [u8]) -> Result<usize, Self::Error>;
15}
16
17impl<'a, T, P> EncodePrefixed<P> for T
18where
19    T: Encode,
20    P: Encode<Error = Error> + FromPrimitive,
21    <T as Encode>::Error: From<Error>,
22{
23    type Error = <T as Encode>::Error;
24
25    fn encode_prefixed(&self, buff: &mut [u8]) -> Result<usize, Self::Error> {
26        let mut index = 0;
27
28        // Compute encoded length and write prefix
29        let len = P::from_usize(self.encode_len()?).unwrap();
30        index += len.encode(buff)?;
31
32        // Encode object
33        index += self.encode(&mut buff[index..])?;
34
35        Ok(index)
36    }
37}