1
  2
  3
  4
  5
  6
  7
  8
  9
 10
 11
 12
 13
 14
 15
 16
 17
 18
 19
 20
 21
 22
 23
 24
 25
 26
 27
 28
 29
 30
 31
 32
 33
 34
 35
 36
 37
 38
 39
 40
 41
 42
 43
 44
 45
 46
 47
 48
 49
 50
 51
 52
 53
 54
 55
 56
 57
 58
 59
 60
 61
 62
 63
 64
 65
 66
 67
 68
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105

use std::{str, net, io};
use byteorder::{BigEndian, ReadBytesExt, WriteBytesExt};
use ser::{Serializable, Stream, Deserializable, Reader, Error as ReaderError};

#[derive(Debug, PartialEq, Clone, Copy)]
pub struct IpAddress(net::IpAddr);

impl Default for IpAddress {
	fn default() -> Self {
		IpAddress(net::IpAddr::V4(net::Ipv4Addr::new(0, 0, 0, 0)))
	}
}

impl From<net::IpAddr> for IpAddress {
	fn from(ip: net::IpAddr) -> Self {
		IpAddress(ip)
	}
}

impl From<IpAddress> for net::IpAddr {
	fn from(ip: IpAddress) -> Self {
		ip.0
	}
}

impl From<&'static str> for IpAddress {
	fn from(s: &'static str) -> Self {
		s.parse().unwrap()
	}
}

impl str::FromStr for IpAddress {
	type Err = <net::IpAddr as str::FromStr>::Err;

	fn from_str(s: &str) -> Result<Self, Self::Err> {
		s.parse().map(IpAddress)
	}
}

impl Serializable for IpAddress {
	fn serialize(&self, stream: &mut Stream) {
		match self.0 {
			net::IpAddr::V4(address) => {
				stream
					.append_slice(&[0u8; 12])
					.append_slice(&address.octets());
			},
			net::IpAddr::V6(address) => {
				for segment in &address.segments() {
					stream.write_u16::<BigEndian>(*segment).unwrap();
				}
			},
		}
	}
}

impl Deserializable for IpAddress {
	fn deserialize<T>(reader: &mut Reader<T>) -> Result<Self, ReaderError> where T: io::Read {
		let bytes: &mut [u8] = &mut [0u8; 12];
		try!(reader.read_slice(bytes));
		if bytes == &[0u8; 12] {
			let address: &mut [u8] = &mut [0u8; 4];
			try!(reader.read_slice(address));
			let address = net::Ipv4Addr::new(address[0], address[1], address[2], address[3]);
			Ok(IpAddress(net::IpAddr::V4(address)))
		} else {
			// compiler needs some help here...
			let mut b = bytes as &[u8];
			let address = net::Ipv6Addr::new(
				try!(b.read_u16::<BigEndian>()),
				try!(b.read_u16::<BigEndian>()),
				try!(b.read_u16::<BigEndian>()),
				try!(b.read_u16::<BigEndian>()),
				try!(b.read_u16::<BigEndian>()),
				try!(b.read_u16::<BigEndian>()),
				try!(reader.read_u16::<BigEndian>()),
				try!(reader.read_u16::<BigEndian>())
			);
			Ok(IpAddress(net::IpAddr::V6(address)))
		}
	}
}

#[cfg(test)]
mod test {
	use std::net;
	use ser::{serialize, deserialize};
	use super::IpAddress;

	#[test]
	fn test_ip_serialize() {
		let ip = IpAddress(net::IpAddr::V6("::ffff:a00:1".parse().unwrap()));
		assert_eq!(serialize(&ip), vec![0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0xff, 0xff, 0x0a, 0x00, 0x00, 0x01].into());
		let ip = IpAddress(net::IpAddr::V4("10.0.0.1".parse().unwrap()));
		assert_eq!(serialize(&ip), vec![0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x0a, 0x00, 0x00, 0x01].into());
	}

	#[test]
	fn test_ip_deserialize() {
		let ip: IpAddress = deserialize(&[0x00u8, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0xff, 0xff, 0x0a, 0x00, 0x00, 0x01] as &[u8]).unwrap();
		assert_eq!(ip, IpAddress(net::IpAddr::V6("::ffff:a00:1".parse().unwrap())));
		let ip: IpAddress = deserialize(&[0x00u8, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x0a, 0x00, 0x00, 0x01] as &[u8]).unwrap();
		assert_eq!(ip, IpAddress(net::IpAddr::V4("10.0.0.1".parse().unwrap())));
	}
}