dsf_core/options/
helpers.rs

1use core::str;
2
3#[cfg(feature = "alloc")]
4use alloc::prelude::v1::*;
5
6use byteorder::{ByteOrder, NetworkEndian};
7
8use super::OptionsError;
9use crate::base::{Encode, Parse};
10use crate::types::DateTime;
11
12impl Parse for String {
13    type Output = String;
14    type Error = OptionsError;
15
16    fn parse<'a>(data: &[u8]) -> Result<(Self::Output, usize), Self::Error> {
17        let length = NetworkEndian::read_u16(&data[0..2]) as usize;
18        let value = str::from_utf8(&data[2..2 + length]).unwrap().to_owned();
19
20        Ok((value, length + 2))
21    }
22}
23
24impl Encode for String {
25    type Error = OptionsError;
26
27    fn encode(&self, data: &mut [u8]) -> Result<usize, Self::Error> {
28        let value = self.as_bytes();
29
30        NetworkEndian::write_u16(&mut data[0..], value.len() as u16);
31        &mut data[2..value.len() + 2].copy_from_slice(value);
32
33        Ok(value.len() + 2)
34    }
35}
36
37impl Parse for Vec<u8> {
38    type Output = Vec<u8>;
39    type Error = OptionsError;
40
41    fn parse<'a>(data: &[u8]) -> Result<(Self::Output, usize), Self::Error> {
42        let length = NetworkEndian::read_u16(&data) as usize;
43        let value = Vec::from(&data[2..2 + length]);
44
45        Ok((value, length + 2))
46    }
47}
48
49impl Encode for Vec<u8> {
50    type Error = OptionsError;
51
52    fn encode(&self, data: &mut [u8]) -> Result<usize, Self::Error> {
53        NetworkEndian::write_u16(&mut data[..], self.len() as u16);
54        &mut data[2..self.len() + 2].copy_from_slice(self);
55
56        Ok(self.len() + 2)
57    }
58}
59
60impl Parse for DateTime {
61    type Output = DateTime;
62    type Error = OptionsError;
63
64    fn parse<'a>(data: &'a [u8]) -> Result<(Self::Output, usize), Self::Error> {
65        let raw = NetworkEndian::read_u64(&data[0..]);
66        let when = DateTime::from_secs(raw);
67
68        Ok((when, 10))
69    }
70}
71
72impl Encode for DateTime {
73    type Error = OptionsError;
74
75    fn encode(&self, data: &mut [u8]) -> Result<usize, Self::Error> {
76        NetworkEndian::write_u16(&mut data[..], 8);
77        let time_s = self.as_secs();
78
79        NetworkEndian::write_u64(&mut data[2..], time_s);
80
81        Ok(10)
82    }
83}