Add a broadcast function for IPv4

This commit is contained in:
Abhishek Chanda
2016-01-31 22:35:36 -08:00
parent d84dc6161a
commit 42adf4ee93

View File

@ -56,6 +56,13 @@ impl Ipv4Network {
(Ipv4Addr::from(ip), ip) (Ipv4Addr::from(ip), ip)
} }
pub fn broadcast(&self) -> (Ipv4Addr, u32) {
let (_, network) = self.network();
let (_, mask) = self.mask();
let broadcast = network | !mask;
(Ipv4Addr::from(broadcast), broadcast)
}
pub fn contains(&self, ip: Ipv4Addr) -> bool { pub fn contains(&self, ip: Ipv4Addr) -> bool {
let (_, net) = self.network(); let (_, net) = self.network();
(u32::from(ip) & net) == net (u32::from(ip) & net) == net
@ -147,6 +154,14 @@ mod test {
assert_eq!(int, 168427520); assert_eq!(int, 168427520);
} }
#[test]
fn broadcast_v4() {
let cidr = Ipv4Network::new(Ipv4Addr::new(10, 10, 1, 97), 23);
let (ip, int) = cidr.broadcast();
assert_eq!(ip, Ipv4Addr::new(10, 10, 1, 255));
assert_eq!(int, 168428031);
}
#[test] #[test]
fn contains_v4() { fn contains_v4() {
let cidr = Ipv4Network::new(Ipv4Addr::new(74, 125, 227, 0), 25); let cidr = Ipv4Network::new(Ipv4Addr::new(74, 125, 227, 0), 25);