Move around some stuff

src/ipnetwork.rs

@@ -0,0 +1,132 @@
+use std::fmt;
+use std::net::{IpAddr, Ipv4Addr, Ipv6Addr};
+
+// A network
+#[derive(Debug)]
+pub enum IpNetwork {
+    V4(Ipv4Network),
+    V6(Ipv6Network),
+}
+
+pub struct Ipv4Network {
+    addr: Ipv4Addr,
+    prefix: u8,
+}
+
+pub struct Ipv6Network {
+    addr: Ipv6Addr,
+    prefix: u8,
+}
+
+impl Ipv4Network {
+    fn new(addr: Ipv4Addr, prefix: u8) -> Ipv4Network {
+        Ipv4Network { addr: addr, prefix: prefix }
+    }
+
+    fn ip(&self) -> &Ipv4Addr {
+        &(self.addr)
+    }
+
+    fn prefix(&self) -> u8 {
+        self.prefix
+    }
+
+    fn mask(&self) -> u32 {
+        let prefix = self.prefix;
+        !(0xffffffff >> prefix)
+    }
+
+    fn mask_to_string(&self) -> String {
+        let mask = self.mask();
+        Ipv4Network::int_to_ip(mask).to_string()
+    }
+
+    fn ip_to_int(addr: Ipv4Addr) -> u32 {
+        let ip = addr.octets();
+        (ip[0] as u32) << 24 + (ip[1] as u32) << 16 + (ip[2] as u32) << 8 + (ip[3] as u32)
+    }
+
+    fn int_to_ip(ip: u32) -> Ipv4Addr {
+        Ipv4Addr::new(((ip >> 24) & 0xff) as u8, ((ip >> 16) & 0xff) as u8,
+                     ((ip >> 8) & 0xff) as u8, (ip & 0xff) as u8)
+    }
+}
+
+impl Ipv6Network {
+    fn new(addr: Ipv6Addr, prefix: u8) -> Ipv6Network {
+        Ipv6Network { addr: addr, prefix: prefix }
+    }
+
+    fn ip(&self) -> &Ipv6Addr {
+        &(self.addr)
+    }
+
+    fn prefix(&self) -> u8 {
+        self.prefix
+    }
+}
+
+impl IpNetwork {
+    pub fn new(ip: IpAddr, prefix: u8) -> IpNetwork {
+        match ip {
+            IpAddr::V4(a) => IpNetwork::V4(Ipv4Network::new(a, prefix)),
+            IpAddr::V6(a) => IpNetwork::V6(Ipv6Network::new(a, prefix)),
+        }
+    }
+
+    pub fn ip(&self) -> IpAddr {
+        match *self {
+            IpNetwork::V4(ref a) => IpAddr::V4(*a.ip()),
+            IpNetwork::V6(ref a) => IpAddr::V6(*a.ip()),
+        }
+    }
+
+    pub fn prefix(&self) -> u8 {
+        match *self {
+            IpNetwork::V4(ref a) => a.prefix(),
+            IpNetwork::V6(ref a) => a.prefix(),
+        }
+    }
+}
+
+impl fmt::Debug for Ipv4Network {
+    fn fmt(&self, fmt: &mut fmt::Formatter) -> fmt::Result {
+        write!(fmt, "{}/{}", self.ip(), self.prefix())
+    }
+}
+
+impl fmt::Debug for Ipv6Network {
+    fn fmt(&self, fmt: &mut fmt::Formatter) -> fmt::Result {
+        write!(fmt, "{}/{}", self.ip(), self.prefix())
+    }
+}
+
+#[cfg(test)]
+mod test {
+    use std::net::{Ipv4Addr, Ipv6Addr};
+    use super::*;
+
+    #[test]
+    fn create_v4() {
+        let cidr = Ipv4Network::new(Ipv4Addr::new(77, 88, 21, 11), 24);
+        assert_eq!(cidr.prefix(), 24);
+    }
+
+    #[test]
+    fn create_v6() {
+        let cidr = Ipv6Network::new(Ipv6Addr::new(0, 0, 0, 0, 0, 0, 0, 1), 24);
+        assert_eq!(cidr.prefix(), 24);
+    }
+
+    #[test]
+    fn mask_v4() {
+        let cidr = Ipv4Network::new(Ipv4Addr::new(74, 125, 227, 0), 29);
+        assert_eq!(cidr.mask(), 4294967288);
+    }
+
+    #[test]
+    fn mask_string_v4() {
+        let cidr = Ipv4Network::new(Ipv4Addr::new(74, 125, 227, 0), 29);
+        assert_eq!(cidr.mask_to_string(), "255.255.255.248");
+    }
+}

src/lib.rs

@@ -1,134 +1,2 @@
 #![feature(ip_addr)]
-
-use std::fmt;
-use std::net::{IpAddr, Ipv4Addr, Ipv6Addr};
-
-// A network
-#[derive(Debug)]
-pub enum IpNetwork {
-    V4(Ipv4Network),
-    V6(Ipv6Network),
-}
-
-pub struct Ipv4Network {
-    addr: Ipv4Addr,
-    prefix: u8,
-}
-
-pub struct Ipv6Network {
-    addr: Ipv6Addr,
-    prefix: u8,
-}
-
-impl Ipv4Network {
-    fn new(addr: Ipv4Addr, prefix: u8) -> Ipv4Network {
-        Ipv4Network { addr: addr, prefix: prefix }
-    }
-
-    fn ip(&self) -> &Ipv4Addr {
-        &(self.addr)
-    }
-
-    fn prefix(&self) -> u8 {
-        self.prefix
-    }
-
-    fn mask(&self) -> u32 {
-        let prefix = self.prefix;
-        !(0xffffffff >> prefix)
-    }
-
-    fn mask_to_string(&self) -> String {
-        let mask = self.mask();
-        Ipv4Network::int_to_ip(mask).to_string()
-    }
-
-    fn ip_to_int(addr: Ipv4Addr) -> u32 {
-        let ip = addr.octets();
-        (ip[0] as u32) << 24 + (ip[1] as u32) << 16 + (ip[2] as u32) << 8 + (ip[3] as u32)
-    }
-
-    fn int_to_ip(ip: u32) -> Ipv4Addr {
-        Ipv4Addr::new(((ip >> 24) & 0xff) as u8, ((ip >> 16) & 0xff) as u8,
-                     ((ip >> 8) & 0xff) as u8, (ip & 0xff) as u8)
-    }
-}
-
-impl Ipv6Network {
-    fn new(addr: Ipv6Addr, prefix: u8) -> Ipv6Network {
-        Ipv6Network { addr: addr, prefix: prefix }
-    }
-
-    fn ip(&self) -> &Ipv6Addr {
-        &(self.addr)
-    }
-
-    fn prefix(&self) -> u8 {
-        self.prefix
-    }
-}
-
-impl IpNetwork {
-    pub fn new(ip: IpAddr, prefix: u8) -> IpNetwork {
-        match ip {
-            IpAddr::V4(a) => IpNetwork::V4(Ipv4Network::new(a, prefix)),
-            IpAddr::V6(a) => IpNetwork::V6(Ipv6Network::new(a, prefix)),
-        }
-    }
-
-    pub fn ip(&self) -> IpAddr {
-        match *self {
-            IpNetwork::V4(ref a) => IpAddr::V4(*a.ip()),
-            IpNetwork::V6(ref a) => IpAddr::V6(*a.ip()),
-        }
-    }
-
-    pub fn prefix(&self) -> u8 {
-        match *self {
-            IpNetwork::V4(ref a) => a.prefix(),
-            IpNetwork::V6(ref a) => a.prefix(),
-        }
-    }
-}
-
-impl fmt::Debug for Ipv4Network {
-    fn fmt(&self, fmt: &mut fmt::Formatter) -> fmt::Result {
-        write!(fmt, "{}/{}", self.ip(), self.prefix())
-    }
-}
-
-impl fmt::Debug for Ipv6Network {
-    fn fmt(&self, fmt: &mut fmt::Formatter) -> fmt::Result {
-        write!(fmt, "{}/{}", self.ip(), self.prefix())
-    }
-}
-
-#[cfg(test)]
-mod test {
-    use std::net::{Ipv4Addr, Ipv6Addr};
-    use super::*;
-
-    #[test]
-    fn create_v4() {
-        let cidr = Ipv4Network::new(Ipv4Addr::new(77, 88, 21, 11), 24);
-        assert_eq!(cidr.prefix(), 24);
-    }
-
-    #[test]
-    fn create_v6() {
-        let cidr = Ipv6Network::new(Ipv6Addr::new(0, 0, 0, 0, 0, 0, 0, 1), 24);
-        assert_eq!(cidr.prefix(), 24);
-    }
-
-    #[test]
-    fn mask_v4() {
-        let cidr = Ipv4Network::new(Ipv4Addr::new(74, 125, 227, 0), 29);
-        assert_eq!(cidr.mask(), 4294967288);
-    }
-
-    #[test]
-    fn mask_string_v4() {
-        let cidr = Ipv4Network::new(Ipv4Addr::new(74, 125, 227, 0), 29);
-        assert_eq!(cidr.mask_to_string(), "255.255.255.248");
-    }
-}
+pub mod ipnetwork;