rtc_dtls/handshake/
handshake_header.rs

1use super::*;
2
3use byteorder::{BigEndian, ReadBytesExt, WriteBytesExt};
4use std::io::{Read, Write};
5
6// msg_len for Handshake messages assumes an extra 12 bytes for
7// sequence, Fragment and version information
8pub(crate) const HANDSHAKE_HEADER_LENGTH: usize = 12;
9
10#[derive(Copy, Clone, PartialEq, Eq, Debug, Default)]
11pub struct HandshakeHeader {
12    pub(crate) handshake_type: HandshakeType,
13    pub(crate) length: u32, // uint24 in spec
14    pub(crate) message_sequence: u16,
15    pub(crate) fragment_offset: u32, // uint24 in spec
16    pub(crate) fragment_length: u32, // uint24 in spec
17}
18
19impl HandshakeHeader {
20    pub fn size(&self) -> usize {
21        1 + 3 + 2 + 3 + 3
22    }
23
24    pub fn marshal<W: Write>(&self, writer: &mut W) -> Result<()> {
25        writer.write_u8(self.handshake_type as u8)?;
26        writer.write_u24::<BigEndian>(self.length)?;
27        writer.write_u16::<BigEndian>(self.message_sequence)?;
28        writer.write_u24::<BigEndian>(self.fragment_offset)?;
29        writer.write_u24::<BigEndian>(self.fragment_length)?;
30
31        Ok(writer.flush()?)
32    }
33
34    pub fn unmarshal<R: Read>(reader: &mut R) -> Result<Self> {
35        let handshake_type = reader.read_u8()?.into();
36        let length = reader.read_u24::<BigEndian>()?;
37        let message_sequence = reader.read_u16::<BigEndian>()?;
38        let fragment_offset = reader.read_u24::<BigEndian>()?;
39        let fragment_length = reader.read_u24::<BigEndian>()?;
40
41        Ok(HandshakeHeader {
42            handshake_type,
43            length,
44            message_sequence,
45            fragment_offset,
46            fragment_length,
47        })
48    }
49}
rtc_dtls/handshake/handshake_header.rs

rtc_dtls/handshake/
handshake_header.rs
