webrtc_dtls/handshake/
handshake_message_certificate.rs

1use std::io::{Read, Write};
2
3use byteorder::{BigEndian, ReadBytesExt, WriteBytesExt};
4
5use super::*;
6
7#[cfg(test)]
8mod handshake_message_certificate_test;
9
10const HANDSHAKE_MESSAGE_CERTIFICATE_LENGTH_FIELD_SIZE: usize = 3;
11
12#[derive(PartialEq, Eq, Debug, Clone)]
13pub struct HandshakeMessageCertificate {
14    pub(crate) certificate: Vec<Vec<u8>>,
15}
16
17impl HandshakeMessageCertificate {
18    pub fn handshake_type(&self) -> HandshakeType {
19        HandshakeType::Certificate
20    }
21
22    pub fn size(&self) -> usize {
23        let mut len = 3;
24
25        for r in &self.certificate {
26            len += HANDSHAKE_MESSAGE_CERTIFICATE_LENGTH_FIELD_SIZE + r.len();
27        }
28
29        len
30    }
31
32    pub fn marshal<W: Write>(&self, writer: &mut W) -> Result<()> {
33        let mut payload_size = 0;
34        for r in &self.certificate {
35            payload_size += HANDSHAKE_MESSAGE_CERTIFICATE_LENGTH_FIELD_SIZE + r.len();
36        }
37
38        // Total Payload Size
39        writer.write_u24::<BigEndian>(payload_size as u32)?;
40
41        for r in &self.certificate {
42            // Certificate Length
43            writer.write_u24::<BigEndian>(r.len() as u32)?;
44
45            // Certificate body
46            writer.write_all(r)?;
47        }
48
49        Ok(writer.flush()?)
50    }
51
52    pub fn unmarshal<R: Read>(reader: &mut R) -> Result<Self> {
53        let mut certificate: Vec<Vec<u8>> = vec![];
54
55        let payload_size = reader.read_u24::<BigEndian>()? as usize;
56        let mut offset = 0;
57        while offset < payload_size {
58            let certificate_len = reader.read_u24::<BigEndian>()? as usize;
59            offset += HANDSHAKE_MESSAGE_CERTIFICATE_LENGTH_FIELD_SIZE;
60
61            let mut buf = vec![0; certificate_len];
62            reader.read_exact(&mut buf)?;
63            offset += certificate_len;
64
65            certificate.push(buf);
66        }
67
68        Ok(HandshakeMessageCertificate { certificate })
69    }
70}
webrtc_dtls/handshake/handshake_message_certificate.rs

webrtc_dtls/handshake/
handshake_message_certificate.rs
