mirror of
https://github.com/achanda/ipnetwork.git
synced 2025-06-17 01:08:53 +00:00
Move around some stuff
This commit is contained in:
132
src/ipnetwork.rs
Normal file
132
src/ipnetwork.rs
Normal file
@ -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");
|
||||||
|
}
|
||||||
|
}
|
134
src/lib.rs
134
src/lib.rs
@ -1,134 +1,2 @@
|
|||||||
#![feature(ip_addr)]
|
#![feature(ip_addr)]
|
||||||
|
pub mod ipnetwork;
|
||||||
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");
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
Reference in New Issue
Block a user