#[cfg(test)]
mod reqtrans_test;
use std::fmt;
use stun::attributes::*;
use stun::checks::*;
use stun::message::*;
use super::*;
use shared::error::Result;
#[derive(Default, Debug, PartialEq, Eq)]
pub struct RequestedTransport {
pub protocol: Protocol,
}
impl fmt::Display for RequestedTransport {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
write!(f, "protocol: {}", self.protocol)
}
}
const REQUESTED_TRANSPORT_SIZE: usize = 4;
impl Setter for RequestedTransport {
fn add_to(&self, m: &mut Message) -> Result<()> {
let mut v = vec![0; REQUESTED_TRANSPORT_SIZE];
v[0] = self.protocol.0;
m.add(ATTR_REQUESTED_TRANSPORT, &v);
Ok(())
}
}
impl Getter for RequestedTransport {
fn get_from(&mut self, m: &Message) -> Result<()> {
let v = m.get(ATTR_REQUESTED_TRANSPORT)?;
check_size(ATTR_REQUESTED_TRANSPORT, v.len(), REQUESTED_TRANSPORT_SIZE)?;
self.protocol = Protocol(v[0]);
Ok(())
}
}