#[cfg(test)]
mod peeraddr_test;
use std::fmt;
use std::net::{IpAddr, Ipv4Addr};
use shared::error::Result;
use stun::attributes::*;
use stun::message::*;
use stun::xoraddr::*;
#[derive(PartialEq, Eq, Debug)]
pub struct PeerAddress {
pub ip: IpAddr,
pub port: u16,
}
impl Default for PeerAddress {
fn default() -> Self {
PeerAddress {
ip: IpAddr::V4(Ipv4Addr::from(0)),
port: 0,
}
}
}
impl fmt::Display for PeerAddress {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
match self.ip {
IpAddr::V4(_) => write!(f, "{}:{}", self.ip, self.port),
IpAddr::V6(_) => write!(f, "[{}]:{}", self.ip, self.port),
}
}
}
impl Setter for PeerAddress {
fn add_to(&self, m: &mut Message) -> Result<()> {
let a = XorMappedAddress {
ip: self.ip,
port: self.port,
};
a.add_to_as(m, ATTR_XOR_PEER_ADDRESS)
}
}
impl Getter for PeerAddress {
fn get_from(&mut self, m: &Message) -> Result<()> {
let mut a = XorMappedAddress::default();
a.get_from_as(m, ATTR_XOR_PEER_ADDRESS)?;
self.ip = a.ip;
self.port = a.port;
Ok(())
}
}
pub type XorPeerAddress = PeerAddress;