thalassa 0.0.1-pre.2

Implementation of the TLS presentation language data format defined in RFC8446
Documentation 
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
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128

use crate::{
    TlsplDeserialize, TlsplSerialize, TlsplSize,
    error::{TlsplReadResult, TlsplWriteResult},
};

macro_rules! impl_int {
    ($repr:ty) => {
        impl TlsplSize for $repr {
            #[inline]
            fn tlspl_serialized_len(&self) -> usize {
                (<$repr>::BITS / 8) as usize
            }
        }

        impl TlsplSerialize for $repr {
            #[inline]
            fn tlspl_serialize_to<W: crate::io::Write>(
                &self,
                writer: &mut W,
            ) -> TlsplWriteResult<usize> {
                writer.write(&self.to_be_bytes()).map_err(Into::into)
            }
        }

        impl<'a> TlsplDeserialize<'a> for $repr {
            #[inline]
            fn tlspl_deserialize_from<R: crate::io::Read<'a>>(
                reader: &mut R,
            ) -> TlsplReadResult<Self>
            where
                Self: Sized + 'a,
            {
                reader
                    .read_array()
                    .map(|b| <$repr>::from_be_bytes(*b))
                    .map_err(Into::into)
            }
        }
    };
}

// This conflicts with the &[u8] specialization since Serialize for &[T] covers &[u8] since u8 impls Serialize
// impl_int!(u8);
impl_int!(u16);
impl_int!(u32);
impl_int!(u64);
impl_int!(u128);
impl_int!(usize);
impl_int!(i8);
impl_int!(i16);
impl_int!(i32);
impl_int!(i64);
impl_int!(i128);
impl_int!(isize);

#[derive(Debug, Copy, Clone, PartialEq, Eq, Hash, PartialOrd, Ord)]
#[repr(transparent)]
#[allow(non_camel_case_types)]
// 24-bit integer, useful for TLS itself
//
// Is internally backed by a u32 but with some guardrails to prevent misuse (eg: upper byte is masked on conversion from u32)
pub struct u24(u32);

impl TlsplSize for u24 {
    #[inline]
    fn tlspl_serialized_len(&self) -> usize {
        3
    }
}

impl TlsplSerialize for u24 {
    #[inline]
    fn tlspl_serialize_to<W: parsio::Write>(&self, writer: &mut W) -> TlsplWriteResult<usize> {
        // First byte is always 0x00 for a u24
        writer.write(&self.0.to_be_bytes()[1..]).map_err(Into::into)
    }
}

impl<'a> TlsplDeserialize<'a> for u24 {
    fn tlspl_deserialize_from<R: parsio::Read<'a>>(reader: &mut R) -> TlsplReadResult<Self>
    where
        Self: Sized + 'a,
    {
        let array: [u8; 3] = *reader.read_array()?;
        Ok(Self(u32::from_be_bytes([
            0x00, array[0], array[1], array[2],
        ])))
    }
}

impl From<u32> for u24 {
    fn from(value: u32) -> Self {
        // Mask the upper byte
        Self(value & 0x00_FF_FF_FF)
    }
}

impl std::ops::Deref for u24 {
    type Target = u32;

    fn deref(&self) -> &Self::Target {
        &self.0
    }
}

impl TlsplSize for bool {
    #[inline]
    fn tlspl_serialized_len(&self) -> usize {
        1
    }
}

impl TlsplSerialize for bool {
    #[inline]
    fn tlspl_serialize_to<W: crate::io::Write>(&self, writer: &mut W) -> TlsplWriteResult<usize> {
        writer.write(&[*self as u8]).map_err(Into::into)
    }
}

impl<'a> TlsplDeserialize<'a> for bool {
    #[inline]
    fn tlspl_deserialize_from<R: crate::io::Read<'a>>(reader: &mut R) -> TlsplReadResult<Self>
    where
        Self: Sized + 'a,
    {
        reader.read_byte().map(|b| b > 0).map_err(Into::into)
    }
}