#[cfg(test)]
mod lifetime_test;
use std::fmt;
use std::time::Duration;
use shared::error::Result;
use stun::attributes::*;
use stun::checks::*;
use stun::message::*;
pub const DEFAULT_LIFETIME: Duration = Duration::from_secs(10 * 60);
#[derive(Default, Debug, PartialEq, Eq)]
pub struct Lifetime(pub Duration);
impl fmt::Display for Lifetime {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
write!(f, "{}s", self.0.as_secs())
}
}
const LIFETIME_SIZE: usize = 4;
impl Setter for Lifetime {
fn add_to(&self, m: &mut Message) -> Result<()> {
let mut v = vec![0; LIFETIME_SIZE];
v.copy_from_slice(&(self.0.as_secs() as u32).to_be_bytes());
m.add(ATTR_LIFETIME, &v);
Ok(())
}
}
impl Getter for Lifetime {
fn get_from(&mut self, m: &Message) -> Result<()> {
let v = m.get(ATTR_LIFETIME)?;
check_size(ATTR_LIFETIME, v.len(), LIFETIME_SIZE)?;
let seconds = u32::from_be_bytes([v[0], v[1], v[2], v[3]]);
self.0 = Duration::from_secs(seconds as u64);
Ok(())
}
}