webrtc_dtls/extension/
extension_server_name.rs

1#[cfg(test)]
2mod extension_server_name_test;
3
4use std::io::{Read, Write};
5
6use byteorder::{BigEndian, ReadBytesExt, WriteBytesExt};
7
8use super::*;
9
10const EXTENSION_SERVER_NAME_TYPE_DNSHOST_NAME: u8 = 0;
11
12#[derive(Clone, Debug, PartialEq, Eq)]
13pub struct ExtensionServerName {
14    pub(crate) server_name: String,
15}
16
17impl ExtensionServerName {
18    pub fn extension_value(&self) -> ExtensionValue {
19        ExtensionValue::ServerName
20    }
21
22    pub fn size(&self) -> usize {
23        //TODO: check how to do cryptobyte?
24        2 + 2 + 1 + 2 + self.server_name.len()
25    }
26
27    pub fn marshal<W: Write>(&self, writer: &mut W) -> Result<()> {
28        //TODO: check how to do cryptobyte?
29        writer.write_u16::<BigEndian>(2 + 1 + 2 + self.server_name.len() as u16)?;
30        writer.write_u16::<BigEndian>(1 + 2 + self.server_name.len() as u16)?;
31        writer.write_u8(EXTENSION_SERVER_NAME_TYPE_DNSHOST_NAME)?;
32        writer.write_u16::<BigEndian>(self.server_name.len() as u16)?;
33        writer.write_all(self.server_name.as_bytes())?;
34
35        Ok(writer.flush()?)
36    }
37
38    pub fn unmarshal<R: Read>(reader: &mut R) -> Result<Self> {
39        //TODO: check how to do cryptobyte?
40        let _ = reader.read_u16::<BigEndian>()? as usize;
41        let _ = reader.read_u16::<BigEndian>()? as usize;
42
43        let name_type = reader.read_u8()?;
44        if name_type != EXTENSION_SERVER_NAME_TYPE_DNSHOST_NAME {
45            return Err(Error::ErrInvalidSniFormat);
46        }
47
48        let buf_len = reader.read_u16::<BigEndian>()? as usize;
49        let mut buf: Vec<u8> = vec![0u8; buf_len];
50        reader.read_exact(&mut buf)?;
51
52        let server_name = String::from_utf8(buf)?;
53
54        Ok(ExtensionServerName { server_name })
55    }
56}
webrtc_dtls/extension/extension_server_name.rs

webrtc_dtls/extension/
extension_server_name.rs
