webrtc_turn/proto/
reqtrans.rs

1#[cfg(test)]
2mod reqtrans_test;
3
4use super::*;
5
6use stun::attributes::*;
7use stun::checks::*;
8use stun::message::*;
9
10use util::Error;
11
12use std::fmt;
13
14// RequestedTransport represents REQUESTED-TRANSPORT attribute.
15//
16// This attribute is used by the client to request a specific transport
17// protocol for the allocated transport address. RFC 5766 only allows the use of
18// codepoint 17 (User Datagram protocol).
19//
20// RFC 5766 Section 14.7
21#[derive(Default, Debug, PartialEq)]
22pub struct RequestedTransport {
23    pub protocol: Protocol,
24}
25
26impl fmt::Display for RequestedTransport {
27    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
28        write!(f, "protocol: {}", self.protocol)
29    }
30}
31
32const REQUESTED_TRANSPORT_SIZE: usize = 4;
33
34impl Setter for RequestedTransport {
35    // AddTo adds REQUESTED-TRANSPORT to message.
36    fn add_to(&self, m: &mut Message) -> Result<(), Error> {
37        let mut v = vec![0; REQUESTED_TRANSPORT_SIZE];
38        v[0] = self.protocol.0;
39        // b[1:4] is RFFU = 0.
40        // The RFFU field MUST be set to zero on transmission and MUST be
41        // ignored on reception. It is reserved for future uses.
42        m.add(ATTR_REQUESTED_TRANSPORT, &v);
43        Ok(())
44    }
45}
46
47impl Getter for RequestedTransport {
48    // GetFrom decodes REQUESTED-TRANSPORT from message.
49    fn get_from(&mut self, m: &Message) -> Result<(), Error> {
50        let v = m.get(ATTR_REQUESTED_TRANSPORT)?;
51
52        check_size(ATTR_REQUESTED_TRANSPORT, v.len(), REQUESTED_TRANSPORT_SIZE)?;
53        self.protocol = Protocol(v[0]);
54        Ok(())
55    }
56}
webrtc_turn/proto/reqtrans.rs

webrtc_turn/proto/
reqtrans.rs
