#[cfg(test)]
mod channnum_test;
use std::fmt;
use stun::attributes::*;
use stun::checks::*;
use stun::message::*;
use shared::error::Result;
const CHANNEL_NUMBER_SIZE: usize = 4;
pub const MIN_CHANNEL_NUMBER: u16 = 0x4000;
pub const MAX_CHANNEL_NUMBER: u16 = 0x7FFF;
#[derive(Default, Eq, PartialEq, Debug, Copy, Clone, Hash)]
pub struct ChannelNumber(pub u16);
impl fmt::Display for ChannelNumber {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
write!(f, "{}", self.0)
}
}
impl Setter for ChannelNumber {
fn add_to(&self, m: &mut Message) -> Result<()> {
let mut v = vec![0; CHANNEL_NUMBER_SIZE];
v[..2].copy_from_slice(&self.0.to_be_bytes());
m.add(ATTR_CHANNEL_NUMBER, &v);
Ok(())
}
}
impl Getter for ChannelNumber {
fn get_from(&mut self, m: &Message) -> Result<()> {
let v = m.get(ATTR_CHANNEL_NUMBER)?;
check_size(ATTR_CHANNEL_NUMBER, v.len(), CHANNEL_NUMBER_SIZE)?;
self.0 = u16::from_be_bytes([v[0], v[1]]);
Ok(())
}
}
impl ChannelNumber {
fn is_channel_number_valid(&self) -> bool {
self.0 >= MIN_CHANNEL_NUMBER && self.0 <= MAX_CHANNEL_NUMBER
}
pub fn valid(&self) -> bool {
self.is_channel_number_valid()
}
}