mirror of
https://github.com/achanda/ipnetwork.git
synced 2025-06-17 01:08:53 +00:00
Compare commits
103 Commits
Author | SHA1 | Date | |
---|---|---|---|
1b6904bfc7 | |||
ab6f1012b3 | |||
5146125c35 | |||
86dbbd997a | |||
372f9bdbb7 | |||
c0e04d534a | |||
7c9ff1bc30 | |||
6d4e4ba47a | |||
e842e0edc0 | |||
61550eda45 | |||
3503e4c094 | |||
91528476a9 | |||
501e5533d7 | |||
7120b93837 | |||
a57afd9e1a | |||
8ad43541cd | |||
9651b3b721 | |||
5764406000 | |||
4cbd1f2424 | |||
772db6c569 | |||
c9b2bc5519 | |||
fe9f1ba8d5 | |||
9921661ebe | |||
14108098d6 | |||
1f3e10f73d | |||
c26c836b39 | |||
8fcff82c7b | |||
46cc64d424 | |||
e15e8cb812 | |||
0b98eecf6a | |||
6c6bdb333d | |||
eded5f13a9 | |||
1d4665dfe6 | |||
e80eab0c80 | |||
8f1c6b59bb | |||
a9f4547c82 | |||
fa128680b5 | |||
620ec4d42c | |||
b6397ca2bc | |||
c0257fc59d | |||
45fbf458b3 | |||
ca1d332e71 | |||
121820add8 | |||
65bd6724c2 | |||
41e0f8a8a3 | |||
f23ea4d7d6 | |||
fc958f16dc | |||
23b95c3b85 | |||
38460c0598 | |||
ca2da8ae30 | |||
156ae55da0 | |||
4dfdf142d0 | |||
6851a0b272 | |||
00256cf885 | |||
80c6794829 | |||
18677e3b69 | |||
986ee0ba1c | |||
f3ac9ba3fe | |||
bc1064e5f0 | |||
ddc9ef2f7b | |||
87ea7b24fa | |||
08126f5103 | |||
376081d253 | |||
fa85cdb500 | |||
fa5deed40b | |||
5b09d82d43 | |||
dbf366cfdf | |||
6cbb011597 | |||
f7d9524f70 | |||
e1f8566301 | |||
a3197a8411 | |||
0479ca45a9 | |||
11025d3b1a | |||
43a38d4117 | |||
152f1a8393 | |||
31fc25f9ce | |||
bdaa8a7843 | |||
d54f463f86 | |||
fb782086ce | |||
26706b3790 | |||
2eed2de227 | |||
7395af2658 | |||
fefd674721 | |||
20ed033826 | |||
7af6e16f98 | |||
9e42aaeb3e | |||
2b34ba991a | |||
2dff6a0863 | |||
a45e76c92f | |||
a6dfadc21d | |||
61eeb5843f | |||
d8f3cc1992 | |||
c2d4b9aeae | |||
3479b6f272 | |||
10d12cd945 | |||
47a539f2f1 | |||
e7c85b5b81 | |||
439a0deeb7 | |||
b2c458e6f1 | |||
1b06452e88 | |||
2b9936fa8e | |||
553fa0ba1c | |||
a19f24a9c6 |
7
.github/dependabot.yml
vendored
Normal file
7
.github/dependabot.yml
vendored
Normal file
@ -0,0 +1,7 @@
|
|||||||
|
version: 2
|
||||||
|
updates:
|
||||||
|
- package-ecosystem: cargo
|
||||||
|
directory: "/"
|
||||||
|
schedule:
|
||||||
|
interval: daily
|
||||||
|
open-pull-requests-limit: 10
|
19
.github/workflows/publish.yml
vendored
Normal file
19
.github/workflows/publish.yml
vendored
Normal file
@ -0,0 +1,19 @@
|
|||||||
|
name: Publish to crates.io
|
||||||
|
on:
|
||||||
|
workflow_dispatch:
|
||||||
|
push:
|
||||||
|
tags:
|
||||||
|
- 'v.*.*'
|
||||||
|
|
||||||
|
jobs:
|
||||||
|
publish:
|
||||||
|
runs-on: ubuntu-latest
|
||||||
|
steps:
|
||||||
|
- uses: actions/checkout@v2
|
||||||
|
- uses: actions-rs/toolchain@v1
|
||||||
|
with:
|
||||||
|
toolchain: stable
|
||||||
|
override: true
|
||||||
|
- uses: katyo/publish-crates@v1
|
||||||
|
with:
|
||||||
|
registry-token: ${{ secrets.CARGO_REGISTRY_TOKEN }}
|
28
.github/workflows/rust.yml
vendored
Normal file
28
.github/workflows/rust.yml
vendored
Normal file
@ -0,0 +1,28 @@
|
|||||||
|
name: Rust
|
||||||
|
|
||||||
|
on:
|
||||||
|
pull_request:
|
||||||
|
branches:
|
||||||
|
- master
|
||||||
|
|
||||||
|
jobs:
|
||||||
|
build:
|
||||||
|
runs-on: ubuntu-latest
|
||||||
|
strategy:
|
||||||
|
matrix:
|
||||||
|
rust: [stable, beta, nightly]
|
||||||
|
steps:
|
||||||
|
- uses: hecrj/setup-rust-action@v1
|
||||||
|
with:
|
||||||
|
rust-version: ${{ matrix.rust }}
|
||||||
|
- uses: actions/checkout@master
|
||||||
|
with:
|
||||||
|
ref: ${{ github.ref }}
|
||||||
|
- name: Show current git branch
|
||||||
|
run: git branch
|
||||||
|
- name: Build
|
||||||
|
run: cargo build --verbose
|
||||||
|
- name: Run tests
|
||||||
|
run: cargo test --verbose --all-features
|
||||||
|
- name: Build docs
|
||||||
|
run: cargo doc --verbose
|
@ -9,6 +9,8 @@ env:
|
|||||||
- FEATURES: default
|
- FEATURES: default
|
||||||
|
|
||||||
script:
|
script:
|
||||||
|
- cargo build --verbose --no-default-features
|
||||||
|
- cargo test --verbose --no-default-features
|
||||||
- cargo build --verbose
|
- cargo build --verbose
|
||||||
- cargo test --verbose
|
- cargo test --verbose
|
||||||
- cargo build --release --verbose
|
- cargo build --release --verbose
|
||||||
|
27
Cargo.toml
27
Cargo.toml
@ -1,26 +1,31 @@
|
|||||||
[package]
|
[package]
|
||||||
name = "ipnetwork"
|
name = "ipnetwork"
|
||||||
version = "0.14.0" # When updating version, also modify html_root_url in the lib.rs
|
version = "0.20.0"
|
||||||
authors = ["Abhishek Chanda <abhishek.becs@gmail.com>", "Linus Färnstrand <faern@faern.net>"]
|
authors = ["Abhishek Chanda <abhishek.becs@gmail.com>", "Linus Färnstrand <faern@faern.net>"]
|
||||||
description = "A library to work with IP CIDRs in Rust, heavily WIP"
|
description = "A library to work with IP CIDRs in Rust"
|
||||||
license = "Apache-2.0"
|
license = "MIT OR Apache-2.0"
|
||||||
repository = "https://github.com/achanda/ipnetwork"
|
repository = "https://github.com/achanda/ipnetwork"
|
||||||
keywords = ["network", "ip", "address"]
|
keywords = ["network", "ip", "address", "cidr"]
|
||||||
readme = "README.md"
|
readme = "README.md"
|
||||||
documentation = "https://docs.rs/ipnetwork/"
|
categories = ["network-programming", "parser-implementations"]
|
||||||
categories = ["network-programming", "os"]
|
edition = "2021"
|
||||||
|
|
||||||
[dependencies]
|
[dependencies]
|
||||||
clippy = {version = "0.0.302", optional = true}
|
serde = { version = "1", optional = true }
|
||||||
serde = ">=0.8.0, <2.0"
|
schemars = { version = "0.8.10", optional = true }
|
||||||
|
|
||||||
[dev-dependencies]
|
[dev-dependencies]
|
||||||
serde_json = "1.0"
|
serde_json = "1.0"
|
||||||
serde_derive = ">=0.8.0, <2.0"
|
criterion = {version = "0.3.4", features= ["html_reports"]}
|
||||||
|
does-it-json = "0.0.3"
|
||||||
|
|
||||||
[badges]
|
[badges]
|
||||||
travis-ci = { repository = "achanda/ipnetwork" }
|
travis-ci = { repository = "achanda/ipnetwork" }
|
||||||
|
maintenance = { status = "passively-maintained" }
|
||||||
|
|
||||||
[features]
|
[features]
|
||||||
default = []
|
default = ["serde"]
|
||||||
dev = ["clippy"]
|
|
||||||
|
[[bench]]
|
||||||
|
name = "parse_bench"
|
||||||
|
harness = false
|
||||||
|
25
LICENSE-MIT.md
Normal file
25
LICENSE-MIT.md
Normal file
@ -0,0 +1,25 @@
|
|||||||
|
Copyright 2020 Developers of the ipnetwork project
|
||||||
|
|
||||||
|
Permission is hereby granted, free of charge, to any
|
||||||
|
person obtaining a copy of this software and associated
|
||||||
|
documentation files (the "Software"), to deal in the
|
||||||
|
Software without restriction, including without
|
||||||
|
limitation the rights to use, copy, modify, merge,
|
||||||
|
publish, distribute, sublicense, and/or sell copies of
|
||||||
|
the Software, and to permit persons to whom the Software
|
||||||
|
is furnished to do so, subject to the following
|
||||||
|
conditions:
|
||||||
|
|
||||||
|
The above copyright notice and this permission notice
|
||||||
|
shall be included in all copies or substantial portions
|
||||||
|
of the Software.
|
||||||
|
|
||||||
|
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF
|
||||||
|
ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED
|
||||||
|
TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A
|
||||||
|
PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT
|
||||||
|
SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY
|
||||||
|
CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
|
||||||
|
OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR
|
||||||
|
IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
|
||||||
|
DEALINGS IN THE SOFTWARE.
|
@ -1,14 +1,14 @@
|
|||||||
ipnetwork
|
ipnetwork
|
||||||
===
|
===
|
||||||
This is a library to work with IPv4 and v6 CIDRs in rust
|
This is a library to work with IPv4 and IPv6 CIDRs in Rust
|
||||||
The IPv4 implementation is stable, IPv6 implementation is not done yet.
|
|
||||||
|
|
||||||
[](https://travis-ci.org/achanda/ipnetwork)
|
[](https://travis-ci.org/achanda/ipnetwork)
|
||||||
[](https://crates.io/crates/ipnetwork)
|
[](https://crates.io/crates/ipnetwork)
|
||||||
|
|
||||||
Run Clippy by doing
|
Run Clippy by doing
|
||||||
```
|
```
|
||||||
cargo test --features "dev"
|
rustup component add clippy
|
||||||
|
cargo clippy
|
||||||
```
|
```
|
||||||
|
|
||||||
### Installation
|
### Installation
|
||||||
@ -20,7 +20,7 @@ cargo test
|
|||||||
```
|
```
|
||||||
|
|
||||||
You can also add `ipnetwork` as a dependency to your project's `Cargo.toml`:
|
You can also add `ipnetwork` as a dependency to your project's `Cargo.toml`:
|
||||||
```
|
```toml
|
||||||
[dependencies]
|
[dependencies]
|
||||||
ipnetwork = "*"
|
ipnetwork = "*"
|
||||||
```
|
```
|
||||||
|
49
benches/parse_bench.rs
Normal file
49
benches/parse_bench.rs
Normal file
@ -0,0 +1,49 @@
|
|||||||
|
use criterion::{criterion_group, criterion_main, Criterion};
|
||||||
|
use ipnetwork::{Ipv4Network, Ipv6Network};
|
||||||
|
use std::net::{Ipv4Addr, Ipv6Addr};
|
||||||
|
|
||||||
|
fn parse_ipv4_prefix_benchmark(c: &mut Criterion) {
|
||||||
|
c.bench_function("parse ipv4 prefix", |b| {
|
||||||
|
b.iter(|| "127.1.0.0/24".parse::<Ipv4Network>().unwrap())
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
fn parse_ipv6_benchmark(c: &mut Criterion) {
|
||||||
|
c.bench_function("parse ipv6", |b| {
|
||||||
|
b.iter(|| "FF01:0:0:17:0:0:0:2/64".parse::<Ipv6Network>().unwrap())
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
fn parse_ipv4_netmask_benchmark(c: &mut Criterion) {
|
||||||
|
c.bench_function("parse ipv4 netmask", |b| {
|
||||||
|
b.iter(|| "127.1.0.0/255.255.255.0".parse::<Ipv4Network>().unwrap())
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
fn contains_ipv4_benchmark(c: &mut Criterion) {
|
||||||
|
let cidr = "74.125.227.0/25".parse::<Ipv4Network>().unwrap();
|
||||||
|
c.bench_function("contains ipv4", |b| {
|
||||||
|
b.iter(|| {
|
||||||
|
cidr.contains(Ipv4Addr::new(74, 125, 227, 4))
|
||||||
|
})
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
fn contains_ipv6_benchmark(c: &mut Criterion) {
|
||||||
|
let cidr = "FF01:0:0:17:0:0:0:2/65".parse::<Ipv6Network>().unwrap();
|
||||||
|
c.bench_function("contains ipv6", |b| {
|
||||||
|
b.iter(|| {
|
||||||
|
cidr.contains(Ipv6Addr::new(0xff01, 0, 0, 0x17, 0x7fff, 0, 0, 0x2))
|
||||||
|
})
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
criterion_group!(
|
||||||
|
benches,
|
||||||
|
parse_ipv4_prefix_benchmark,
|
||||||
|
parse_ipv6_benchmark,
|
||||||
|
parse_ipv4_netmask_benchmark,
|
||||||
|
contains_ipv4_benchmark,
|
||||||
|
contains_ipv6_benchmark
|
||||||
|
);
|
||||||
|
criterion_main!(benches);
|
@ -1,5 +1,4 @@
|
|||||||
use std::error::Error;
|
use std::{error::Error, fmt};
|
||||||
use std::fmt;
|
|
||||||
|
|
||||||
/// Represents a bunch of errors that can occur while working with a `IpNetwork`
|
/// Represents a bunch of errors that can occur while working with a `IpNetwork`
|
||||||
#[derive(Debug, Clone, PartialEq, Eq)]
|
#[derive(Debug, Clone, PartialEq, Eq)]
|
||||||
@ -10,8 +9,8 @@ pub enum IpNetworkError {
|
|||||||
}
|
}
|
||||||
|
|
||||||
impl fmt::Display for IpNetworkError {
|
impl fmt::Display for IpNetworkError {
|
||||||
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
|
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
|
||||||
use IpNetworkError::*;
|
use crate::IpNetworkError::*;
|
||||||
match *self {
|
match *self {
|
||||||
InvalidAddr(ref s) => write!(f, "invalid address: {}", s),
|
InvalidAddr(ref s) => write!(f, "invalid address: {}", s),
|
||||||
InvalidPrefix => write!(f, "invalid prefix"),
|
InvalidPrefix => write!(f, "invalid prefix"),
|
||||||
@ -22,7 +21,7 @@ impl fmt::Display for IpNetworkError {
|
|||||||
|
|
||||||
impl Error for IpNetworkError {
|
impl Error for IpNetworkError {
|
||||||
fn description(&self) -> &str {
|
fn description(&self) -> &str {
|
||||||
use IpNetworkError::*;
|
use crate::IpNetworkError::*;
|
||||||
match *self {
|
match *self {
|
||||||
InvalidAddr(_) => "address is invalid",
|
InvalidAddr(_) => "address is invalid",
|
||||||
InvalidPrefix => "prefix is invalid",
|
InvalidPrefix => "prefix is invalid",
|
||||||
@ -32,16 +31,22 @@ impl Error for IpNetworkError {
|
|||||||
}
|
}
|
||||||
|
|
||||||
pub fn cidr_parts(cidr: &str) -> Result<(&str, Option<&str>), IpNetworkError> {
|
pub fn cidr_parts(cidr: &str) -> Result<(&str, Option<&str>), IpNetworkError> {
|
||||||
let parts = cidr.split('/').collect::<Vec<&str>>();
|
// Try to find a single slash
|
||||||
if parts.len() == 1 {
|
if let Some(sep) = cidr.find('/') {
|
||||||
Ok((parts[0], None))
|
let (ip, prefix) = cidr.split_at(sep);
|
||||||
} else if parts.len() == 2 {
|
// Error if cidr has multiple slashes
|
||||||
Ok((parts[0], Some(parts[1])))
|
if prefix[1..].find('/').is_some() {
|
||||||
|
Err(IpNetworkError::InvalidCidrFormat(format!(
|
||||||
|
"CIDR must contain a single '/': {}",
|
||||||
|
cidr
|
||||||
|
)))
|
||||||
|
} else {
|
||||||
|
// Handle the case when cidr has exactly one slash
|
||||||
|
Ok((ip, Some(&prefix[1..])))
|
||||||
|
}
|
||||||
} else {
|
} else {
|
||||||
Err(IpNetworkError::InvalidCidrFormat(format!(
|
// Handle the case when cidr does not have a slash
|
||||||
"CIDR must contain a single '/': {}",
|
Ok((cidr, None))
|
||||||
cidr
|
|
||||||
)))
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
319
src/ipv4.rs
319
src/ipv4.rs
@ -1,10 +1,5 @@
|
|||||||
use std::fmt;
|
use crate::common::{cidr_parts, parse_prefix, IpNetworkError};
|
||||||
use std::net::Ipv4Addr;
|
use std::{convert::TryFrom, fmt, net::Ipv4Addr, str::FromStr};
|
||||||
use std::str::FromStr;
|
|
||||||
|
|
||||||
use serde::{de, Deserialize, Deserializer, Serialize, Serializer};
|
|
||||||
|
|
||||||
use common::{cidr_parts, parse_prefix, IpNetworkError};
|
|
||||||
|
|
||||||
const IPV4_BITS: u8 = 32;
|
const IPV4_BITS: u8 = 32;
|
||||||
|
|
||||||
@ -15,29 +10,62 @@ pub struct Ipv4Network {
|
|||||||
prefix: u8,
|
prefix: u8,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl<'de> Deserialize<'de> for Ipv4Network {
|
#[cfg(feature = "serde")]
|
||||||
|
impl<'de> serde::Deserialize<'de> for Ipv4Network {
|
||||||
fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
|
fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
|
||||||
where
|
where
|
||||||
D: Deserializer<'de>,
|
D: serde::Deserializer<'de>,
|
||||||
{
|
{
|
||||||
let s = <&str>::deserialize(deserializer)?;
|
let s = <String>::deserialize(deserializer)?;
|
||||||
Ipv4Network::from_str(s).map_err(de::Error::custom)
|
Ipv4Network::from_str(&s).map_err(serde::de::Error::custom)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Serialize for Ipv4Network {
|
#[cfg(feature = "serde")]
|
||||||
|
impl serde::Serialize for Ipv4Network {
|
||||||
fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
|
fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
|
||||||
where
|
where
|
||||||
S: Serializer,
|
S: serde::Serializer,
|
||||||
{
|
{
|
||||||
serializer.serialize_str(&self.to_string())
|
serializer.collect_str(self)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[cfg(feature = "schemars")]
|
||||||
|
impl schemars::JsonSchema for Ipv4Network {
|
||||||
|
fn schema_name() -> String {
|
||||||
|
"Ipv4Network".to_string()
|
||||||
|
}
|
||||||
|
|
||||||
|
fn json_schema(_: &mut schemars::gen::SchemaGenerator) -> schemars::schema::Schema {
|
||||||
|
schemars::schema::SchemaObject {
|
||||||
|
instance_type: Some(schemars::schema::InstanceType::String.into()),
|
||||||
|
string: Some(Box::new(schemars::schema::StringValidation {
|
||||||
|
pattern: Some(
|
||||||
|
concat!(
|
||||||
|
r#"^((25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)\.){3}"#,
|
||||||
|
r#"(25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)"#,
|
||||||
|
r#"\/(3[0-2]|[0-2]?[0-9])$"#,
|
||||||
|
)
|
||||||
|
.to_string(),
|
||||||
|
),
|
||||||
|
..Default::default()
|
||||||
|
})),
|
||||||
|
extensions: [("x-rust-type".to_string(), "ipnetwork::Ipv4Network".into())]
|
||||||
|
.iter()
|
||||||
|
.cloned()
|
||||||
|
.collect(),
|
||||||
|
..Default::default()
|
||||||
|
}
|
||||||
|
.into()
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Ipv4Network {
|
impl Ipv4Network {
|
||||||
/// Constructs a new `Ipv4Network` from any `Ipv4Addr` and a prefix denoting the network size.
|
/// Constructs a new `Ipv4Network` from any `Ipv4Addr` and a prefix denoting the network size.
|
||||||
|
///
|
||||||
/// If the prefix is larger than 32 this will return an `IpNetworkError::InvalidPrefix`.
|
/// If the prefix is larger than 32 this will return an `IpNetworkError::InvalidPrefix`.
|
||||||
pub fn new(addr: Ipv4Addr, prefix: u8) -> Result<Ipv4Network, IpNetworkError> {
|
pub const fn new(addr: Ipv4Addr, prefix: u8) -> Result<Ipv4Network, IpNetworkError> {
|
||||||
if prefix > IPV4_BITS {
|
if prefix > IPV4_BITS {
|
||||||
Err(IpNetworkError::InvalidPrefix)
|
Err(IpNetworkError::InvalidPrefix)
|
||||||
} else {
|
} else {
|
||||||
@ -45,23 +73,58 @@ impl Ipv4Network {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// Constructs a new `Ipv4Network` from a network address and a network mask.
|
||||||
|
///
|
||||||
|
/// If the netmask is not valid this will return an `IpNetworkError::InvalidPrefix`.
|
||||||
|
pub fn with_netmask(
|
||||||
|
netaddr: Ipv4Addr,
|
||||||
|
netmask: Ipv4Addr,
|
||||||
|
) -> Result<Ipv4Network, IpNetworkError> {
|
||||||
|
let prefix = ipv4_mask_to_prefix(netmask)?;
|
||||||
|
let net = Self {
|
||||||
|
addr: netaddr,
|
||||||
|
prefix,
|
||||||
|
};
|
||||||
|
Ok(net)
|
||||||
|
}
|
||||||
|
|
||||||
/// Returns an iterator over `Ipv4Network`. Each call to `next` will return the next
|
/// Returns an iterator over `Ipv4Network`. Each call to `next` will return the next
|
||||||
/// `Ipv4Addr` in the given network. `None` will be returned when there are no more
|
/// `Ipv4Addr` in the given network. `None` will be returned when there are no more
|
||||||
/// addresses.
|
/// addresses.
|
||||||
pub fn iter(&self) -> Ipv4NetworkIterator {
|
pub fn iter(self) -> Ipv4NetworkIterator {
|
||||||
let start = u32::from(self.network());
|
let start = u32::from(self.network());
|
||||||
let end = start + self.size();
|
let end = start + (self.size() - 1);
|
||||||
Ipv4NetworkIterator { next: start, end }
|
Ipv4NetworkIterator {
|
||||||
|
next: Some(start),
|
||||||
|
end,
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn ip(&self) -> Ipv4Addr {
|
pub fn ip(self) -> Ipv4Addr {
|
||||||
self.addr
|
self.addr
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn prefix(&self) -> u8 {
|
pub fn prefix(self) -> u8 {
|
||||||
self.prefix
|
self.prefix
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// Checks if the given `Ipv4Network` is a subnet of the other.
|
||||||
|
pub fn is_subnet_of(self, other: Ipv4Network) -> bool {
|
||||||
|
other.ip() <= self.ip() && other.broadcast() >= self.broadcast()
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Checks if the given `Ipv4Network` is a supernet of the other.
|
||||||
|
pub fn is_supernet_of(self, other: Ipv4Network) -> bool {
|
||||||
|
other.is_subnet_of(self)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Checks if the given `Ipv4Network` is partly contained in other.
|
||||||
|
pub fn overlaps(self, other: Ipv4Network) -> bool {
|
||||||
|
other.contains(self.ip())
|
||||||
|
|| (other.contains(self.broadcast())
|
||||||
|
|| (self.contains(other.ip()) || (self.contains(other.broadcast()))))
|
||||||
|
}
|
||||||
|
|
||||||
/// Returns the mask for this `Ipv4Network`.
|
/// Returns the mask for this `Ipv4Network`.
|
||||||
/// That means the `prefix` most significant bits will be 1 and the rest 0
|
/// That means the `prefix` most significant bits will be 1 and the rest 0
|
||||||
///
|
///
|
||||||
@ -76,7 +139,7 @@ impl Ipv4Network {
|
|||||||
/// let net: Ipv4Network = "127.0.0.0/16".parse().unwrap();
|
/// let net: Ipv4Network = "127.0.0.0/16".parse().unwrap();
|
||||||
/// assert_eq!(net.mask(), Ipv4Addr::new(255, 255, 0, 0));
|
/// assert_eq!(net.mask(), Ipv4Addr::new(255, 255, 0, 0));
|
||||||
/// ```
|
/// ```
|
||||||
pub fn mask(&self) -> Ipv4Addr {
|
pub fn mask(self) -> Ipv4Addr {
|
||||||
let prefix = self.prefix;
|
let prefix = self.prefix;
|
||||||
let mask = !(0xffff_ffff as u64 >> prefix) as u32;
|
let mask = !(0xffff_ffff as u64 >> prefix) as u32;
|
||||||
Ipv4Addr::from(mask)
|
Ipv4Addr::from(mask)
|
||||||
@ -94,7 +157,7 @@ impl Ipv4Network {
|
|||||||
/// let net: Ipv4Network = "10.1.9.32/16".parse().unwrap();
|
/// let net: Ipv4Network = "10.1.9.32/16".parse().unwrap();
|
||||||
/// assert_eq!(net.network(), Ipv4Addr::new(10, 1, 0, 0));
|
/// assert_eq!(net.network(), Ipv4Addr::new(10, 1, 0, 0));
|
||||||
/// ```
|
/// ```
|
||||||
pub fn network(&self) -> Ipv4Addr {
|
pub fn network(self) -> Ipv4Addr {
|
||||||
let mask = u32::from(self.mask());
|
let mask = u32::from(self.mask());
|
||||||
let ip = u32::from(self.addr) & mask;
|
let ip = u32::from(self.addr) & mask;
|
||||||
Ipv4Addr::from(ip)
|
Ipv4Addr::from(ip)
|
||||||
@ -112,7 +175,7 @@ impl Ipv4Network {
|
|||||||
/// let net: Ipv4Network = "10.9.0.32/16".parse().unwrap();
|
/// let net: Ipv4Network = "10.9.0.32/16".parse().unwrap();
|
||||||
/// assert_eq!(net.broadcast(), Ipv4Addr::new(10, 9, 255, 255));
|
/// assert_eq!(net.broadcast(), Ipv4Addr::new(10, 9, 255, 255));
|
||||||
/// ```
|
/// ```
|
||||||
pub fn broadcast(&self) -> Ipv4Addr {
|
pub fn broadcast(self) -> Ipv4Addr {
|
||||||
let mask = u32::from(self.mask());
|
let mask = u32::from(self.mask());
|
||||||
let broadcast = u32::from(self.addr) | !mask;
|
let broadcast = u32::from(self.addr) | !mask;
|
||||||
Ipv4Addr::from(broadcast)
|
Ipv4Addr::from(broadcast)
|
||||||
@ -130,9 +193,10 @@ impl Ipv4Network {
|
|||||||
/// assert!(net.contains(Ipv4Addr::new(127, 0, 0, 70)));
|
/// assert!(net.contains(Ipv4Addr::new(127, 0, 0, 70)));
|
||||||
/// assert!(!net.contains(Ipv4Addr::new(127, 0, 1, 70)));
|
/// assert!(!net.contains(Ipv4Addr::new(127, 0, 1, 70)));
|
||||||
/// ```
|
/// ```
|
||||||
pub fn contains(&self, ip: Ipv4Addr) -> bool {
|
#[inline]
|
||||||
let net = u32::from(self.network());
|
pub fn contains(self, ip: Ipv4Addr) -> bool {
|
||||||
let mask = u32::from(self.mask());
|
let mask = !(0xffff_ffff as u64 >> self.prefix) as u32;
|
||||||
|
let net = u32::from(self.addr) & mask;
|
||||||
(u32::from(ip) & mask) == net
|
(u32::from(ip) & mask) == net
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -150,7 +214,7 @@ impl Ipv4Network {
|
|||||||
/// let tinynet: Ipv4Network = "0.0.0.0/32".parse().unwrap();
|
/// let tinynet: Ipv4Network = "0.0.0.0/32".parse().unwrap();
|
||||||
/// assert_eq!(tinynet.size(), 1);
|
/// assert_eq!(tinynet.size(), 1);
|
||||||
/// ```
|
/// ```
|
||||||
pub fn size(&self) -> u32 {
|
pub fn size(self) -> u32 {
|
||||||
let host_bits = u32::from(IPV4_BITS - self.prefix);
|
let host_bits = u32::from(IPV4_BITS - self.prefix);
|
||||||
(2 as u32).pow(host_bits)
|
(2 as u32).pow(host_bits)
|
||||||
}
|
}
|
||||||
@ -172,7 +236,7 @@ impl Ipv4Network {
|
|||||||
/// let net2: Ipv4Network = "10.0.0.0/16".parse().unwrap();
|
/// let net2: Ipv4Network = "10.0.0.0/16".parse().unwrap();
|
||||||
/// assert_eq!(net2.nth(256).unwrap(), Ipv4Addr::new(10, 0, 1, 0));
|
/// assert_eq!(net2.nth(256).unwrap(), Ipv4Addr::new(10, 0, 1, 0));
|
||||||
/// ```
|
/// ```
|
||||||
pub fn nth(&self, n: u32) -> Option<Ipv4Addr> {
|
pub fn nth(self, n: u32) -> Option<Ipv4Addr> {
|
||||||
if n < self.size() {
|
if n < self.size() {
|
||||||
let net = u32::from(self.network());
|
let net = u32::from(self.network());
|
||||||
Some(Ipv4Addr::from(net + n))
|
Some(Ipv4Addr::from(net + n))
|
||||||
@ -183,7 +247,7 @@ impl Ipv4Network {
|
|||||||
}
|
}
|
||||||
|
|
||||||
impl fmt::Display for Ipv4Network {
|
impl fmt::Display for Ipv4Network {
|
||||||
fn fmt(&self, fmt: &mut fmt::Formatter) -> fmt::Result {
|
fn fmt(&self, fmt: &mut fmt::Formatter<'_>) -> fmt::Result {
|
||||||
write!(fmt, "{}/{}", self.ip(), self.prefix())
|
write!(fmt, "{}/{}", self.ip(), self.prefix())
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -203,18 +267,32 @@ impl fmt::Display for Ipv4Network {
|
|||||||
/// ```
|
/// ```
|
||||||
impl FromStr for Ipv4Network {
|
impl FromStr for Ipv4Network {
|
||||||
type Err = IpNetworkError;
|
type Err = IpNetworkError;
|
||||||
fn from_str(s: &str) -> Result<Ipv4Network, IpNetworkError> {
|
fn from_str(s: &str) -> Result<Self, Self::Err> {
|
||||||
let (addr_str, prefix_str) = cidr_parts(s)?;
|
let (addr_str, prefix_str) = cidr_parts(s)?;
|
||||||
let addr = Ipv4Addr::from_str(addr_str)
|
let addr = Ipv4Addr::from_str(addr_str)
|
||||||
.map_err(|_| IpNetworkError::InvalidAddr(addr_str.to_string()))?;
|
.map_err(|_| IpNetworkError::InvalidAddr(addr_str.to_string()))?;
|
||||||
let prefix = match prefix_str {
|
let prefix = match prefix_str {
|
||||||
Some(v) => parse_prefix(v, IPV4_BITS)?,
|
Some(v) => {
|
||||||
|
if let Ok(netmask) = Ipv4Addr::from_str(v) {
|
||||||
|
ipv4_mask_to_prefix(netmask)?
|
||||||
|
} else {
|
||||||
|
parse_prefix(v, IPV4_BITS)?
|
||||||
|
}
|
||||||
|
}
|
||||||
None => IPV4_BITS,
|
None => IPV4_BITS,
|
||||||
};
|
};
|
||||||
Ipv4Network::new(addr, prefix)
|
Ipv4Network::new(addr, prefix)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
impl TryFrom<&str> for Ipv4Network {
|
||||||
|
type Error = IpNetworkError;
|
||||||
|
|
||||||
|
fn try_from(s: &str) -> Result<Self, Self::Error> {
|
||||||
|
Ipv4Network::from_str(s)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
impl From<Ipv4Addr> for Ipv4Network {
|
impl From<Ipv4Addr> for Ipv4Network {
|
||||||
fn from(a: Ipv4Addr) -> Ipv4Network {
|
fn from(a: Ipv4Addr) -> Ipv4Network {
|
||||||
Ipv4Network {
|
Ipv4Network {
|
||||||
@ -224,8 +302,9 @@ impl From<Ipv4Addr> for Ipv4Network {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[derive(Clone, Debug)]
|
||||||
pub struct Ipv4NetworkIterator {
|
pub struct Ipv4NetworkIterator {
|
||||||
next: u32,
|
next: Option<u32>,
|
||||||
end: u32,
|
end: u32,
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -233,17 +312,26 @@ impl Iterator for Ipv4NetworkIterator {
|
|||||||
type Item = Ipv4Addr;
|
type Item = Ipv4Addr;
|
||||||
|
|
||||||
fn next(&mut self) -> Option<Ipv4Addr> {
|
fn next(&mut self) -> Option<Ipv4Addr> {
|
||||||
if self.next < self.end {
|
let next = self.next?;
|
||||||
let next = Ipv4Addr::from(self.next as u32);
|
self.next = if next == self.end {
|
||||||
self.next += 1;
|
|
||||||
Some(next)
|
|
||||||
} else {
|
|
||||||
None
|
None
|
||||||
}
|
} else {
|
||||||
|
Some(next + 1)
|
||||||
|
};
|
||||||
|
Some(next.into())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl IntoIterator for &'_ Ipv4Network {
|
||||||
|
type IntoIter = Ipv4NetworkIterator;
|
||||||
|
type Item = Ipv4Addr;
|
||||||
|
fn into_iter(self) -> Ipv4NetworkIterator {
|
||||||
|
self.iter()
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Converts a `Ipv4Addr` network mask into a prefix.
|
/// Converts a `Ipv4Addr` network mask into a prefix.
|
||||||
|
///
|
||||||
/// If the mask is invalid this will return an `IpNetworkError::InvalidPrefix`.
|
/// If the mask is invalid this will return an `IpNetworkError::InvalidPrefix`.
|
||||||
pub fn ipv4_mask_to_prefix(mask: Ipv4Addr) -> Result<u8, IpNetworkError> {
|
pub fn ipv4_mask_to_prefix(mask: Ipv4Addr) -> Result<u8, IpNetworkError> {
|
||||||
let mask = u32::from(mask);
|
let mask = u32::from(mask);
|
||||||
@ -429,6 +517,22 @@ mod test {
|
|||||||
assert_eq!(prefix, 25);
|
assert_eq!(prefix, 25);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// Parse netmask as well as prefix
|
||||||
|
#[test]
|
||||||
|
fn parse_netmask() {
|
||||||
|
let from_netmask: Ipv4Network = "192.168.1.0/255.255.255.0".parse().unwrap();
|
||||||
|
let from_prefix: Ipv4Network = "192.168.1.0/24".parse().unwrap();
|
||||||
|
assert_eq!(from_netmask, from_prefix);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn parse_netmask_broken_v4() {
|
||||||
|
assert_eq!(
|
||||||
|
"192.168.1.0/255.0.255.0".parse::<Ipv4Network>(),
|
||||||
|
Err(IpNetworkError::InvalidPrefix)
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn invalid_v4_mask_to_prefix() {
|
fn invalid_v4_mask_to_prefix() {
|
||||||
let mask = Ipv4Addr::new(255, 0, 255, 0);
|
let mask = Ipv4Addr::new(255, 0, 255, 0);
|
||||||
@ -436,6 +540,24 @@ mod test {
|
|||||||
assert!(prefix.is_err());
|
assert!(prefix.is_err());
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn ipv4network_with_netmask() {
|
||||||
|
{
|
||||||
|
// Positive test-case.
|
||||||
|
let addr = Ipv4Addr::new(127, 0, 0, 1);
|
||||||
|
let mask = Ipv4Addr::new(255, 0, 0, 0);
|
||||||
|
let net = Ipv4Network::with_netmask(addr, mask).unwrap();
|
||||||
|
let expected = Ipv4Network::new(Ipv4Addr::new(127, 0, 0, 1), 8).unwrap();
|
||||||
|
assert_eq!(net, expected);
|
||||||
|
}
|
||||||
|
{
|
||||||
|
// Negative test-case.
|
||||||
|
let addr = Ipv4Addr::new(127, 0, 0, 1);
|
||||||
|
let mask = Ipv4Addr::new(255, 0, 255, 0);
|
||||||
|
Ipv4Network::with_netmask(addr, mask).unwrap_err();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn ipv4network_from_ipv4addr() {
|
fn ipv4network_from_ipv4addr() {
|
||||||
let net = Ipv4Network::from(Ipv4Addr::new(127, 0, 0, 1));
|
let net = Ipv4Network::from(Ipv4Addr::new(127, 0, 0, 1));
|
||||||
@ -454,4 +576,125 @@ mod test {
|
|||||||
fn assert_sync<T: Sync>() {}
|
fn assert_sync<T: Sync>() {}
|
||||||
assert_sync::<Ipv4Network>();
|
assert_sync::<Ipv4Network>();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Tests from cpython https://github.com/python/cpython/blob/e9bc4172d18db9c182d8e04dd7b033097a994c06/Lib/test/test_ipaddress.py
|
||||||
|
#[test]
|
||||||
|
fn test_is_subnet_of() {
|
||||||
|
let mut test_cases: HashMap<(Ipv4Network, Ipv4Network), bool> = HashMap::new();
|
||||||
|
|
||||||
|
test_cases.insert(
|
||||||
|
(
|
||||||
|
"10.0.0.0/30".parse().unwrap(),
|
||||||
|
"10.0.1.0/24".parse().unwrap(),
|
||||||
|
),
|
||||||
|
false,
|
||||||
|
);
|
||||||
|
test_cases.insert(
|
||||||
|
(
|
||||||
|
"10.0.0.0/30".parse().unwrap(),
|
||||||
|
"10.0.0.0/24".parse().unwrap(),
|
||||||
|
),
|
||||||
|
true,
|
||||||
|
);
|
||||||
|
test_cases.insert(
|
||||||
|
(
|
||||||
|
"10.0.0.0/30".parse().unwrap(),
|
||||||
|
"10.0.1.0/24".parse().unwrap(),
|
||||||
|
),
|
||||||
|
false,
|
||||||
|
);
|
||||||
|
test_cases.insert(
|
||||||
|
(
|
||||||
|
"10.0.1.0/24".parse().unwrap(),
|
||||||
|
"10.0.0.0/30".parse().unwrap(),
|
||||||
|
),
|
||||||
|
false,
|
||||||
|
);
|
||||||
|
|
||||||
|
for (key, val) in test_cases.iter() {
|
||||||
|
let (src, dest) = (key.0, key.1);
|
||||||
|
assert_eq!(
|
||||||
|
src.is_subnet_of(dest),
|
||||||
|
*val,
|
||||||
|
"testing with {} and {}",
|
||||||
|
src,
|
||||||
|
dest
|
||||||
|
);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_is_supernet_of() {
|
||||||
|
let mut test_cases: HashMap<(Ipv4Network, Ipv4Network), bool> = HashMap::new();
|
||||||
|
|
||||||
|
test_cases.insert(
|
||||||
|
(
|
||||||
|
"10.0.0.0/30".parse().unwrap(),
|
||||||
|
"10.0.1.0/24".parse().unwrap(),
|
||||||
|
),
|
||||||
|
false,
|
||||||
|
);
|
||||||
|
test_cases.insert(
|
||||||
|
(
|
||||||
|
"10.0.0.0/30".parse().unwrap(),
|
||||||
|
"10.0.0.0/24".parse().unwrap(),
|
||||||
|
),
|
||||||
|
false,
|
||||||
|
);
|
||||||
|
test_cases.insert(
|
||||||
|
(
|
||||||
|
"10.0.0.0/30".parse().unwrap(),
|
||||||
|
"10.0.1.0/24".parse().unwrap(),
|
||||||
|
),
|
||||||
|
false,
|
||||||
|
);
|
||||||
|
test_cases.insert(
|
||||||
|
(
|
||||||
|
"10.0.0.0/24".parse().unwrap(),
|
||||||
|
"10.0.0.0/30".parse().unwrap(),
|
||||||
|
),
|
||||||
|
true,
|
||||||
|
);
|
||||||
|
|
||||||
|
for (key, val) in test_cases.iter() {
|
||||||
|
let (src, dest) = (key.0, key.1);
|
||||||
|
assert_eq!(
|
||||||
|
src.is_supernet_of(dest),
|
||||||
|
*val,
|
||||||
|
"testing with {} and {}",
|
||||||
|
src,
|
||||||
|
dest
|
||||||
|
);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_overlaps() {
|
||||||
|
let other: Ipv4Network = "1.2.3.0/30".parse().unwrap();
|
||||||
|
let other2: Ipv4Network = "1.2.2.0/24".parse().unwrap();
|
||||||
|
let other3: Ipv4Network = "1.2.2.64/26".parse().unwrap();
|
||||||
|
|
||||||
|
let skynet: Ipv4Network = "1.2.3.0/24".parse().unwrap();
|
||||||
|
assert_eq!(skynet.overlaps(other), true);
|
||||||
|
assert_eq!(skynet.overlaps(other2), false);
|
||||||
|
assert_eq!(other2.overlaps(other3), true);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn edges() {
|
||||||
|
let low: Ipv4Network = "0.0.0.0/24".parse().unwrap();
|
||||||
|
let low_addrs: Vec<Ipv4Addr> = low.iter().collect();
|
||||||
|
assert_eq!(256, low_addrs.len());
|
||||||
|
assert_eq!("0.0.0.0".parse::<Ipv4Addr>().unwrap(), low_addrs[0]);
|
||||||
|
assert_eq!("0.0.0.255".parse::<Ipv4Addr>().unwrap(), low_addrs[255]);
|
||||||
|
|
||||||
|
let high: Ipv4Network = "255.255.255.0/24".parse().unwrap();
|
||||||
|
let high_addrs: Vec<Ipv4Addr> = high.iter().collect();
|
||||||
|
assert_eq!(256, high_addrs.len());
|
||||||
|
assert_eq!("255.255.255.0".parse::<Ipv4Addr>().unwrap(), high_addrs[0]);
|
||||||
|
assert_eq!(
|
||||||
|
"255.255.255.255".parse::<Ipv4Addr>().unwrap(),
|
||||||
|
high_addrs[255]
|
||||||
|
);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
293
src/ipv6.rs
293
src/ipv6.rs
@ -1,11 +1,5 @@
|
|||||||
use std::cmp;
|
use crate::common::{cidr_parts, parse_prefix, IpNetworkError};
|
||||||
use std::fmt;
|
use std::{cmp, convert::TryFrom, fmt, net::Ipv6Addr, str::FromStr};
|
||||||
use std::net::Ipv6Addr;
|
|
||||||
use std::str::FromStr;
|
|
||||||
|
|
||||||
use serde::{de, Deserialize, Deserializer, Serialize, Serializer};
|
|
||||||
|
|
||||||
use common::{cidr_parts, parse_prefix, IpNetworkError};
|
|
||||||
|
|
||||||
const IPV6_BITS: u8 = 128;
|
const IPV6_BITS: u8 = 128;
|
||||||
const IPV6_SEGMENT_BITS: u8 = 16;
|
const IPV6_SEGMENT_BITS: u8 = 16;
|
||||||
@ -17,29 +11,72 @@ pub struct Ipv6Network {
|
|||||||
prefix: u8,
|
prefix: u8,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl<'de> Deserialize<'de> for Ipv6Network {
|
#[cfg(feature = "serde")]
|
||||||
|
impl<'de> serde::Deserialize<'de> for Ipv6Network {
|
||||||
fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
|
fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
|
||||||
where
|
where
|
||||||
D: Deserializer<'de>,
|
D: serde::Deserializer<'de>,
|
||||||
{
|
{
|
||||||
let s = <&str>::deserialize(deserializer)?;
|
let s = <String>::deserialize(deserializer)?;
|
||||||
Ipv6Network::from_str(s).map_err(de::Error::custom)
|
Ipv6Network::from_str(&s).map_err(serde::de::Error::custom)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Serialize for Ipv6Network {
|
#[cfg(feature = "serde")]
|
||||||
|
impl serde::Serialize for Ipv6Network {
|
||||||
fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
|
fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
|
||||||
where
|
where
|
||||||
S: Serializer,
|
S: serde::Serializer,
|
||||||
{
|
{
|
||||||
serializer.serialize_str(&self.to_string())
|
serializer.collect_str(self)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[cfg(feature = "schemars")]
|
||||||
|
impl schemars::JsonSchema for Ipv6Network {
|
||||||
|
fn schema_name() -> String {
|
||||||
|
"Ipv6Network".to_string()
|
||||||
|
}
|
||||||
|
|
||||||
|
fn json_schema(_: &mut schemars::gen::SchemaGenerator) -> schemars::schema::Schema {
|
||||||
|
schemars::schema::SchemaObject {
|
||||||
|
instance_type: Some(schemars::schema::InstanceType::String.into()),
|
||||||
|
string: Some(Box::new(schemars::schema::StringValidation {
|
||||||
|
pattern: Some(
|
||||||
|
concat!(
|
||||||
|
r#"^("#,
|
||||||
|
r#"([0-9a-fA-F]{1,4}:){7,7}[0-9a-fA-F]{1,4}"#,
|
||||||
|
r#"|([0-9a-fA-F]{1,4}:){1,7}:"#,
|
||||||
|
r#"|([0-9a-fA-F]{1,4}:){1,6}:[0-9a-fA-F]{1,4}"#,
|
||||||
|
r#"|([0-9a-fA-F]{1,4}:){1,5}(:[0-9a-fA-F]{1,4}){1,2}"#,
|
||||||
|
r#"|([0-9a-fA-F]{1,4}:){1,4}(:[0-9a-fA-F]{1,4}){1,3}"#,
|
||||||
|
r#"|([0-9a-fA-F]{1,4}:){1,3}(:[0-9a-fA-F]{1,4}){1,4}"#,
|
||||||
|
r#"|([0-9a-fA-F]{1,4}:){1,2}(:[0-9a-fA-F]{1,4}){1,5}"#,
|
||||||
|
r#"|[0-9a-fA-F]{1,4}:((:[0-9a-fA-F]{1,4}){1,6})"#,
|
||||||
|
r#"|:((:[0-9a-fA-F]{1,4}){1,7}|:)"#,
|
||||||
|
r#"|fe80:(:[0-9a-fA-F]{0,4}){0,4}%[0-9a-zA-Z]{1,}"#,
|
||||||
|
r#"|::(ffff(:0{1,4}){0,1}:){0,1}((25[0-5]|(2[0-4]|1{0,1}[0-9]){0,1}[0-9])\.){3,3}(25[0-5]|(2[0-4]|1{0,1}[0-9]){0,1}[0-9])"#,
|
||||||
|
r#"|([0-9a-fA-F]{1,4}:){1,4}:((25[0-5]|(2[0-4]|1{0,1}[0-9]){0,1}[0-9])\.){3,3}(25[0-5]|(2[0-4]|1{0,1}[0-9]){0,1}[0-9])"#,
|
||||||
|
r#"")[/](12[0-8]|1[0-1][0-9]|[0-9]?[0-9])$"#,
|
||||||
|
).to_string(),
|
||||||
|
),
|
||||||
|
..Default::default()
|
||||||
|
})),
|
||||||
|
extensions: [("x-rust-type".to_string(), "ipnetwork::Ipv6Network".into())]
|
||||||
|
.iter()
|
||||||
|
.cloned()
|
||||||
|
.collect(),
|
||||||
|
..Default::default()
|
||||||
|
}
|
||||||
|
.into()
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Ipv6Network {
|
impl Ipv6Network {
|
||||||
/// Constructs a new `Ipv6Network` from any `Ipv6Addr` and a prefix denoting the network size.
|
/// Constructs a new `Ipv6Network` from any `Ipv6Addr` and a prefix denoting the network size.
|
||||||
|
///
|
||||||
/// If the prefix is larger than 128 this will return an `IpNetworkError::InvalidPrefix`.
|
/// If the prefix is larger than 128 this will return an `IpNetworkError::InvalidPrefix`.
|
||||||
pub fn new(addr: Ipv6Addr, prefix: u8) -> Result<Ipv6Network, IpNetworkError> {
|
pub const fn new(addr: Ipv6Addr, prefix: u8) -> Result<Ipv6Network, IpNetworkError> {
|
||||||
if prefix > IPV6_BITS {
|
if prefix > IPV6_BITS {
|
||||||
Err(IpNetworkError::InvalidPrefix)
|
Err(IpNetworkError::InvalidPrefix)
|
||||||
} else {
|
} else {
|
||||||
@ -47,6 +84,18 @@ impl Ipv6Network {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// Constructs a new `Ipv6Network` from a network address and a network mask.
|
||||||
|
///
|
||||||
|
/// If the netmask is not valid this will return an `IpNetworkError::InvalidPrefix`.
|
||||||
|
pub fn with_netmask(netaddr: Ipv6Addr, netmask: Ipv6Addr) -> Result<Self, IpNetworkError> {
|
||||||
|
let prefix = ipv6_mask_to_prefix(netmask)?;
|
||||||
|
let net = Self {
|
||||||
|
addr: netaddr,
|
||||||
|
prefix,
|
||||||
|
};
|
||||||
|
Ok(net)
|
||||||
|
}
|
||||||
|
|
||||||
/// Returns an iterator over `Ipv6Network`. Each call to `next` will return the next
|
/// Returns an iterator over `Ipv6Network`. Each call to `next` will return the next
|
||||||
/// `Ipv6Addr` in the given network. `None` will be returned when there are no more
|
/// `Ipv6Addr` in the given network. `None` will be returned when there are no more
|
||||||
/// addresses.
|
/// addresses.
|
||||||
@ -62,8 +111,8 @@ impl Ipv6Network {
|
|||||||
let end: u128 = dec | mask;
|
let end: u128 = dec | mask;
|
||||||
|
|
||||||
Ipv6NetworkIterator {
|
Ipv6NetworkIterator {
|
||||||
next: start,
|
next: Some(start),
|
||||||
end: end,
|
end,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -111,6 +160,23 @@ impl Ipv6Network {
|
|||||||
self.prefix
|
self.prefix
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// Checks if the given `Ipv6Network` is a subnet of the other.
|
||||||
|
pub fn is_subnet_of(self, other: Ipv6Network) -> bool {
|
||||||
|
other.ip() <= self.ip() && other.broadcast() >= self.broadcast()
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Checks if the given `Ipv6Network` is a supernet of the other.
|
||||||
|
pub fn is_supernet_of(self, other: Ipv6Network) -> bool {
|
||||||
|
other.is_subnet_of(self)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Checks if the given `Ipv6Network` is partly contained in other.
|
||||||
|
pub fn overlaps(self, other: Ipv6Network) -> bool {
|
||||||
|
other.contains(self.ip())
|
||||||
|
|| (other.contains(self.broadcast())
|
||||||
|
|| (self.contains(other.ip()) || (self.contains(other.broadcast()))))
|
||||||
|
}
|
||||||
|
|
||||||
/// Returns the mask for this `Ipv6Network`.
|
/// Returns the mask for this `Ipv6Network`.
|
||||||
/// That means the `prefix` most significant bits will be 1 and the rest 0
|
/// That means the `prefix` most significant bits will be 1 and the rest 0
|
||||||
///
|
///
|
||||||
@ -148,6 +214,7 @@ impl Ipv6Network {
|
|||||||
/// assert!(net.contains(Ipv6Addr::new(0xff01, 0, 0, 0, 0, 0, 0, 0x1)));
|
/// assert!(net.contains(Ipv6Addr::new(0xff01, 0, 0, 0, 0, 0, 0, 0x1)));
|
||||||
/// assert!(!net.contains(Ipv6Addr::new(0xffff, 0, 0, 0, 0, 0, 0, 0x1)));
|
/// assert!(!net.contains(Ipv6Addr::new(0xffff, 0, 0, 0, 0, 0, 0, 0x1)));
|
||||||
/// ```
|
/// ```
|
||||||
|
#[inline]
|
||||||
pub fn contains(&self, ip: Ipv6Addr) -> bool {
|
pub fn contains(&self, ip: Ipv6Addr) -> bool {
|
||||||
let a = self.addr.segments();
|
let a = self.addr.segments();
|
||||||
let b = ip.segments();
|
let b = ip.segments();
|
||||||
@ -179,9 +246,22 @@ impl Ipv6Network {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// Creates an `Ipv6Network` from parsing a string in CIDR notation.
|
||||||
|
///
|
||||||
|
/// # Examples
|
||||||
|
///
|
||||||
|
/// ```
|
||||||
|
/// use std::net::Ipv6Addr;
|
||||||
|
/// use ipnetwork::Ipv6Network;
|
||||||
|
///
|
||||||
|
/// let new = Ipv6Network::new(Ipv6Addr::new(0xff01, 0, 0, 0x17, 0, 0, 0, 0x2), 65).unwrap();
|
||||||
|
/// let from_cidr: Ipv6Network = "FF01:0:0:17:0:0:0:2/65".parse().unwrap();
|
||||||
|
/// assert_eq!(new.ip(), from_cidr.ip());
|
||||||
|
/// assert_eq!(new.prefix(), from_cidr.prefix());
|
||||||
|
/// ```
|
||||||
impl FromStr for Ipv6Network {
|
impl FromStr for Ipv6Network {
|
||||||
type Err = IpNetworkError;
|
type Err = IpNetworkError;
|
||||||
fn from_str(s: &str) -> Result<Ipv6Network, IpNetworkError> {
|
fn from_str(s: &str) -> Result<Self, Self::Err> {
|
||||||
let (addr_str, prefix_str) = cidr_parts(s)?;
|
let (addr_str, prefix_str) = cidr_parts(s)?;
|
||||||
let addr = Ipv6Addr::from_str(addr_str)
|
let addr = Ipv6Addr::from_str(addr_str)
|
||||||
.map_err(|_| IpNetworkError::InvalidAddr(addr_str.to_string()))?;
|
.map_err(|_| IpNetworkError::InvalidAddr(addr_str.to_string()))?;
|
||||||
@ -193,6 +273,14 @@ impl FromStr for Ipv6Network {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
impl TryFrom<&str> for Ipv6Network {
|
||||||
|
type Error = IpNetworkError;
|
||||||
|
|
||||||
|
fn try_from(s: &str) -> Result<Self, Self::Error> {
|
||||||
|
Ipv6Network::from_str(s)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
impl From<Ipv6Addr> for Ipv6Network {
|
impl From<Ipv6Addr> for Ipv6Network {
|
||||||
fn from(a: Ipv6Addr) -> Ipv6Network {
|
fn from(a: Ipv6Addr) -> Ipv6Network {
|
||||||
Ipv6Network {
|
Ipv6Network {
|
||||||
@ -202,8 +290,9 @@ impl From<Ipv6Addr> for Ipv6Network {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[derive(Clone, Debug)]
|
||||||
pub struct Ipv6NetworkIterator {
|
pub struct Ipv6NetworkIterator {
|
||||||
next: u128,
|
next: Option<u128>,
|
||||||
end: u128,
|
end: u128,
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -211,18 +300,26 @@ impl Iterator for Ipv6NetworkIterator {
|
|||||||
type Item = Ipv6Addr;
|
type Item = Ipv6Addr;
|
||||||
|
|
||||||
fn next(&mut self) -> Option<Ipv6Addr> {
|
fn next(&mut self) -> Option<Ipv6Addr> {
|
||||||
if self.next <= self.end {
|
let next = self.next?;
|
||||||
let next = Ipv6Addr::from(self.next);
|
self.next = if next == self.end {
|
||||||
self.next += 1;
|
|
||||||
Some(next)
|
|
||||||
} else {
|
|
||||||
None
|
None
|
||||||
}
|
} else {
|
||||||
|
Some(next + 1)
|
||||||
|
};
|
||||||
|
Some(next.into())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl IntoIterator for &'_ Ipv6Network {
|
||||||
|
type IntoIter = Ipv6NetworkIterator;
|
||||||
|
type Item = Ipv6Addr;
|
||||||
|
fn into_iter(self) -> Ipv6NetworkIterator {
|
||||||
|
self.iter()
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl fmt::Display for Ipv6Network {
|
impl fmt::Display for Ipv6Network {
|
||||||
fn fmt(&self, fmt: &mut fmt::Formatter) -> fmt::Result {
|
fn fmt(&self, fmt: &mut fmt::Formatter<'_>) -> fmt::Result {
|
||||||
write!(fmt, "{}/{}", self.ip(), self.prefix())
|
write!(fmt, "{}/{}", self.ip(), self.prefix())
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -231,7 +328,7 @@ impl fmt::Display for Ipv6Network {
|
|||||||
/// If the mask is invalid this will return an `IpNetworkError::InvalidPrefix`.
|
/// If the mask is invalid this will return an `IpNetworkError::InvalidPrefix`.
|
||||||
pub fn ipv6_mask_to_prefix(mask: Ipv6Addr) -> Result<u8, IpNetworkError> {
|
pub fn ipv6_mask_to_prefix(mask: Ipv6Addr) -> Result<u8, IpNetworkError> {
|
||||||
let mask = mask.segments();
|
let mask = mask.segments();
|
||||||
let mut mask_iter = mask.into_iter();
|
let mut mask_iter = mask.iter();
|
||||||
|
|
||||||
// Count the number of set bits from the start of the address
|
// Count the number of set bits from the start of the address
|
||||||
let mut prefix = 0;
|
let mut prefix = 0;
|
||||||
@ -265,6 +362,7 @@ pub fn ipv6_mask_to_prefix(mask: Ipv6Addr) -> Result<u8, IpNetworkError> {
|
|||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
mod test {
|
mod test {
|
||||||
use super::*;
|
use super::*;
|
||||||
|
use std::collections::HashMap;
|
||||||
use std::net::Ipv6Addr;
|
use std::net::Ipv6Addr;
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
@ -273,6 +371,14 @@ mod test {
|
|||||||
assert_eq!(cidr.prefix(), 24);
|
assert_eq!(cidr.prefix(), 24);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn parse_netmask_broken_v6() {
|
||||||
|
assert_eq!(
|
||||||
|
"FF01:0:0:17:0:0:0:2/255.255.255.0".parse::<Ipv6Network>(),
|
||||||
|
Err(IpNetworkError::InvalidPrefix)
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn create_v6_invalid_prefix() {
|
fn create_v6_invalid_prefix() {
|
||||||
let cidr = Ipv6Network::new(Ipv6Addr::new(0, 0, 0, 0, 0, 0, 0, 1), 129);
|
let cidr = Ipv6Network::new(Ipv6Addr::new(0, 0, 0, 0, 0, 0, 0, 1), 129);
|
||||||
@ -353,6 +459,25 @@ mod test {
|
|||||||
assert!(prefix.is_err());
|
assert!(prefix.is_err());
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn ipv6network_with_netmask() {
|
||||||
|
{
|
||||||
|
// Positive test-case.
|
||||||
|
let addr = Ipv6Addr::new(0xff01, 0, 0, 0x17, 0, 0, 0, 0x2);
|
||||||
|
let mask = Ipv6Addr::new(0xffff, 0xffff, 0xffff, 0, 0, 0, 0, 0);
|
||||||
|
let net = Ipv6Network::with_netmask(addr, mask).unwrap();
|
||||||
|
let expected =
|
||||||
|
Ipv6Network::new(Ipv6Addr::new(0xff01, 0, 0, 0x17, 0, 0, 0, 0x2), 48).unwrap();
|
||||||
|
assert_eq!(net, expected);
|
||||||
|
}
|
||||||
|
{
|
||||||
|
// Negative test-case.
|
||||||
|
let addr = Ipv6Addr::new(0xff01, 0, 0, 0x17, 0, 0, 0, 0x2);
|
||||||
|
let mask = Ipv6Addr::new(0, 0, 0xffff, 0xffff, 0, 0, 0, 0);
|
||||||
|
Ipv6Network::with_netmask(addr, mask).unwrap_err();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn iterator_v6() {
|
fn iterator_v6() {
|
||||||
let cidr: Ipv6Network = "2001:db8::/126".parse().unwrap();
|
let cidr: Ipv6Network = "2001:db8::/126".parse().unwrap();
|
||||||
@ -436,4 +561,116 @@ mod test {
|
|||||||
fn assert_sync<T: Sync>() {}
|
fn assert_sync<T: Sync>() {}
|
||||||
assert_sync::<Ipv6Network>();
|
assert_sync::<Ipv6Network>();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Tests from cpython https://github.com/python/cpython/blob/e9bc4172d18db9c182d8e04dd7b033097a994c06/Lib/test/test_ipaddress.py
|
||||||
|
#[test]
|
||||||
|
fn test_is_subnet_of() {
|
||||||
|
let mut test_cases: HashMap<(Ipv6Network, Ipv6Network), bool> = HashMap::new();
|
||||||
|
|
||||||
|
test_cases.insert(
|
||||||
|
(
|
||||||
|
"2000:999::/56".parse().unwrap(),
|
||||||
|
"2000:aaa::/48".parse().unwrap(),
|
||||||
|
),
|
||||||
|
false,
|
||||||
|
);
|
||||||
|
test_cases.insert(
|
||||||
|
(
|
||||||
|
"2000:aaa::/56".parse().unwrap(),
|
||||||
|
"2000:aaa::/48".parse().unwrap(),
|
||||||
|
),
|
||||||
|
true,
|
||||||
|
);
|
||||||
|
test_cases.insert(
|
||||||
|
(
|
||||||
|
"2000:bbb::/56".parse().unwrap(),
|
||||||
|
"2000:aaa::/48".parse().unwrap(),
|
||||||
|
),
|
||||||
|
false,
|
||||||
|
);
|
||||||
|
test_cases.insert(
|
||||||
|
(
|
||||||
|
"2000:aaa::/48".parse().unwrap(),
|
||||||
|
"2000:aaa::/56".parse().unwrap(),
|
||||||
|
),
|
||||||
|
false,
|
||||||
|
);
|
||||||
|
|
||||||
|
for (key, val) in test_cases.iter() {
|
||||||
|
let (src, dest) = (key.0, key.1);
|
||||||
|
assert_eq!(
|
||||||
|
src.is_subnet_of(dest),
|
||||||
|
*val,
|
||||||
|
"testing with {} and {}",
|
||||||
|
src,
|
||||||
|
dest
|
||||||
|
);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_is_supernet_of() {
|
||||||
|
let mut test_cases: HashMap<(Ipv6Network, Ipv6Network), bool> = HashMap::new();
|
||||||
|
|
||||||
|
test_cases.insert(
|
||||||
|
(
|
||||||
|
"2000:999::/56".parse().unwrap(),
|
||||||
|
"2000:aaa::/48".parse().unwrap(),
|
||||||
|
),
|
||||||
|
false,
|
||||||
|
);
|
||||||
|
test_cases.insert(
|
||||||
|
(
|
||||||
|
"2000:aaa::/56".parse().unwrap(),
|
||||||
|
"2000:aaa::/48".parse().unwrap(),
|
||||||
|
),
|
||||||
|
false,
|
||||||
|
);
|
||||||
|
test_cases.insert(
|
||||||
|
(
|
||||||
|
"2000:bbb::/56".parse().unwrap(),
|
||||||
|
"2000:aaa::/48".parse().unwrap(),
|
||||||
|
),
|
||||||
|
false,
|
||||||
|
);
|
||||||
|
test_cases.insert(
|
||||||
|
(
|
||||||
|
"2000:aaa::/48".parse().unwrap(),
|
||||||
|
"2000:aaa::/56".parse().unwrap(),
|
||||||
|
),
|
||||||
|
true,
|
||||||
|
);
|
||||||
|
|
||||||
|
for (key, val) in test_cases.iter() {
|
||||||
|
let (src, dest) = (key.0, key.1);
|
||||||
|
assert_eq!(
|
||||||
|
src.is_supernet_of(dest),
|
||||||
|
*val,
|
||||||
|
"testing with {} and {}",
|
||||||
|
src,
|
||||||
|
dest
|
||||||
|
);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_overlaps() {
|
||||||
|
let other: Ipv6Network = "2001:DB8:ACAD::1/64".parse().unwrap();
|
||||||
|
let other2: Ipv6Network = "2001:DB8:ACAD::20:2/64".parse().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(other2.overlaps(other), true);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn edges() {
|
||||||
|
let low: Ipv6Network = "::0/120".parse().unwrap();
|
||||||
|
let low_addrs: Vec<Ipv6Addr> = low.iter().collect();
|
||||||
|
assert_eq!(256, low_addrs.len());
|
||||||
|
|
||||||
|
let high: Ipv6Network = "ffff:ffff:ffff:ffff:ffff:ffff:ffff:ff00/120"
|
||||||
|
.parse()
|
||||||
|
.unwrap();
|
||||||
|
let high_addrs: Vec<Ipv6Addr> = high.iter().collect();
|
||||||
|
assert_eq!(256, high_addrs.len());
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
184
src/lib.rs
184
src/lib.rs
@ -1,27 +1,24 @@
|
|||||||
//! The `ipnetwork` crate provides a set of APIs to work with IP CIDRs in
|
//! The `ipnetwork` crate provides a set of APIs to work with IP CIDRs in
|
||||||
//! Rust. Implementation for IPv4 is more or less stable, IPv6 implementation
|
//! Rust.
|
||||||
//! is still WIP.
|
|
||||||
#![cfg_attr(feature = "dev", feature(plugin))]
|
|
||||||
#![cfg_attr(feature = "dev", plugin(clippy))]
|
|
||||||
#![crate_type = "lib"]
|
#![crate_type = "lib"]
|
||||||
#![doc(html_root_url = "https://docs.rs/ipnetwork/0.14.0")]
|
#![deny(
|
||||||
|
missing_debug_implementations,
|
||||||
|
unsafe_code,
|
||||||
|
unused_extern_crates,
|
||||||
|
unused_import_braces
|
||||||
|
)]
|
||||||
|
|
||||||
extern crate serde;
|
use std::{convert::TryFrom, fmt, net::IpAddr, str::FromStr};
|
||||||
|
|
||||||
use std::fmt;
|
|
||||||
use std::net::IpAddr;
|
|
||||||
|
|
||||||
mod common;
|
mod common;
|
||||||
mod ipv4;
|
mod ipv4;
|
||||||
mod ipv6;
|
mod ipv6;
|
||||||
|
|
||||||
use std::str::FromStr;
|
pub use crate::common::IpNetworkError;
|
||||||
|
pub use crate::ipv4::Ipv4NetworkIterator;
|
||||||
use serde::{de, Deserialize, Deserializer, Serialize, Serializer};
|
pub use crate::ipv4::{ipv4_mask_to_prefix, Ipv4Network};
|
||||||
|
pub use crate::ipv6::Ipv6NetworkIterator;
|
||||||
pub use common::IpNetworkError;
|
pub use crate::ipv6::{ipv6_mask_to_prefix, Ipv6Network};
|
||||||
pub use ipv4::{ipv4_mask_to_prefix, Ipv4Network};
|
|
||||||
pub use ipv6::{ipv6_mask_to_prefix, Ipv6Network};
|
|
||||||
|
|
||||||
/// Represents a generic network range. This type can have two variants:
|
/// Represents a generic network range. This type can have two variants:
|
||||||
/// the v4 and the v6 case.
|
/// the v4 and the v6 case.
|
||||||
@ -31,22 +28,92 @@ pub enum IpNetwork {
|
|||||||
V6(Ipv6Network),
|
V6(Ipv6Network),
|
||||||
}
|
}
|
||||||
|
|
||||||
impl<'de> Deserialize<'de> for IpNetwork {
|
#[cfg(feature = "serde")]
|
||||||
|
impl<'de> serde::Deserialize<'de> for IpNetwork {
|
||||||
fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
|
fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
|
||||||
where
|
where
|
||||||
D: Deserializer<'de>,
|
D: serde::Deserializer<'de>,
|
||||||
{
|
{
|
||||||
let s = <String>::deserialize(deserializer)?;
|
let s = <String>::deserialize(deserializer)?;
|
||||||
IpNetwork::from_str(&s).map_err(de::Error::custom)
|
IpNetwork::from_str(&s).map_err(serde::de::Error::custom)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Serialize for IpNetwork {
|
#[cfg(feature = "serde")]
|
||||||
|
impl serde::Serialize for IpNetwork {
|
||||||
fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
|
fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
|
||||||
where
|
where
|
||||||
S: Serializer,
|
S: serde::Serializer,
|
||||||
{
|
{
|
||||||
serializer.serialize_str(&self.to_string())
|
serializer.collect_str(self)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[cfg(feature = "schemars")]
|
||||||
|
impl schemars::JsonSchema for IpNetwork {
|
||||||
|
fn schema_name() -> String {
|
||||||
|
"IpNetwork".to_string()
|
||||||
|
}
|
||||||
|
|
||||||
|
fn json_schema(gen: &mut schemars::gen::SchemaGenerator) -> schemars::schema::Schema {
|
||||||
|
schemars::schema::SchemaObject {
|
||||||
|
metadata: Some(
|
||||||
|
schemars::schema::Metadata {
|
||||||
|
..Default::default()
|
||||||
|
}
|
||||||
|
.into(),
|
||||||
|
),
|
||||||
|
subschemas: Some(
|
||||||
|
schemars::schema::SubschemaValidation {
|
||||||
|
one_of: Some(vec![
|
||||||
|
schemars::schema::SchemaObject {
|
||||||
|
metadata: Some(
|
||||||
|
schemars::schema::Metadata {
|
||||||
|
title: Some("v4".to_string()),
|
||||||
|
..Default::default()
|
||||||
|
}
|
||||||
|
.into(),
|
||||||
|
),
|
||||||
|
subschemas: Some(
|
||||||
|
schemars::schema::SubschemaValidation {
|
||||||
|
all_of: Some(vec![gen.subschema_for::<Ipv4Network>()]),
|
||||||
|
..Default::default()
|
||||||
|
}
|
||||||
|
.into(),
|
||||||
|
),
|
||||||
|
..Default::default()
|
||||||
|
}
|
||||||
|
.into(),
|
||||||
|
schemars::schema::SchemaObject {
|
||||||
|
metadata: Some(
|
||||||
|
schemars::schema::Metadata {
|
||||||
|
title: Some("v6".to_string()),
|
||||||
|
..Default::default()
|
||||||
|
}
|
||||||
|
.into(),
|
||||||
|
),
|
||||||
|
subschemas: Some(
|
||||||
|
schemars::schema::SubschemaValidation {
|
||||||
|
all_of: Some(vec![gen.subschema_for::<Ipv6Network>()]),
|
||||||
|
..Default::default()
|
||||||
|
}
|
||||||
|
.into(),
|
||||||
|
),
|
||||||
|
..Default::default()
|
||||||
|
}
|
||||||
|
.into(),
|
||||||
|
]),
|
||||||
|
..Default::default()
|
||||||
|
}
|
||||||
|
.into(),
|
||||||
|
),
|
||||||
|
extensions: [("x-rust-type".to_string(), "ipnetwork::IpNetwork".into())]
|
||||||
|
.iter()
|
||||||
|
.cloned()
|
||||||
|
.collect(),
|
||||||
|
..Default::default()
|
||||||
|
}
|
||||||
|
.into()
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -69,6 +136,14 @@ impl IpNetwork {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// Constructs a new `IpNetwork` from a network address and a network mask.
|
||||||
|
///
|
||||||
|
/// If the netmask is not valid this will return an `IpNetworkError::InvalidPrefix`.
|
||||||
|
pub fn with_netmask(netaddr: IpAddr, netmask: IpAddr) -> Result<Self, IpNetworkError> {
|
||||||
|
let prefix = ip_mask_to_prefix(netmask)?;
|
||||||
|
Self::new(netaddr, prefix)
|
||||||
|
}
|
||||||
|
|
||||||
/// Returns the IP part of a given `IpNetwork`
|
/// Returns the IP part of a given `IpNetwork`
|
||||||
pub fn ip(&self) -> IpAddr {
|
pub fn ip(&self) -> IpAddr {
|
||||||
match *self {
|
match *self {
|
||||||
@ -200,6 +275,10 @@ impl IpNetwork {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// TODO(abhishek) when TryFrom is stable, implement it for IpNetwork to
|
||||||
|
// variant conversions. Then use that to implement a generic is_subnet_of
|
||||||
|
// is_supernet_of, overlaps
|
||||||
|
|
||||||
/// Checks if a given `IpAddr` is in this `IpNetwork`
|
/// Checks if a given `IpAddr` is in this `IpNetwork`
|
||||||
///
|
///
|
||||||
/// # Examples
|
/// # Examples
|
||||||
@ -216,6 +295,7 @@ impl IpNetwork {
|
|||||||
/// assert!(!net.contains(ip2));
|
/// assert!(!net.contains(ip2));
|
||||||
/// assert!(!net.contains(ip4));
|
/// assert!(!net.contains(ip4));
|
||||||
/// ```
|
/// ```
|
||||||
|
#[inline]
|
||||||
pub fn contains(&self, ip: IpAddr) -> bool {
|
pub fn contains(&self, ip: IpAddr) -> bool {
|
||||||
match (*self, ip) {
|
match (*self, ip) {
|
||||||
(IpNetwork::V4(net), IpAddr::V4(ip)) => net.contains(ip),
|
(IpNetwork::V4(net), IpAddr::V4(ip)) => net.contains(ip),
|
||||||
@ -241,6 +321,17 @@ impl IpNetwork {
|
|||||||
IpNetwork::V6(ref ip) => NetworkSize::V6(ip.size()),
|
IpNetwork::V6(ref ip) => NetworkSize::V6(ip.size()),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// Returns an iterator over the addresses contained in the network.
|
||||||
|
///
|
||||||
|
/// This lists all the addresses in the network range, in ascending order.
|
||||||
|
pub fn iter(&self) -> IpNetworkIterator {
|
||||||
|
let inner = match self {
|
||||||
|
IpNetwork::V4(ip) => IpNetworkIteratorInner::V4(ip.iter()),
|
||||||
|
IpNetwork::V6(ip) => IpNetworkIteratorInner::V6(ip.iter()),
|
||||||
|
};
|
||||||
|
IpNetworkIterator { inner }
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Tries to parse the given string into a `IpNetwork`. Will first try to parse
|
/// Tries to parse the given string into a `IpNetwork`. Will first try to parse
|
||||||
@ -259,7 +350,7 @@ impl IpNetwork {
|
|||||||
/// ```
|
/// ```
|
||||||
impl FromStr for IpNetwork {
|
impl FromStr for IpNetwork {
|
||||||
type Err = IpNetworkError;
|
type Err = IpNetworkError;
|
||||||
fn from_str(s: &str) -> Result<IpNetwork, IpNetworkError> {
|
fn from_str(s: &str) -> Result<Self, Self::Err> {
|
||||||
if let Ok(net) = Ipv4Network::from_str(s) {
|
if let Ok(net) = Ipv4Network::from_str(s) {
|
||||||
Ok(IpNetwork::V4(net))
|
Ok(IpNetwork::V4(net))
|
||||||
} else if let Ok(net) = Ipv6Network::from_str(s) {
|
} else if let Ok(net) = Ipv6Network::from_str(s) {
|
||||||
@ -270,6 +361,14 @@ impl FromStr for IpNetwork {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
impl TryFrom<&str> for IpNetwork {
|
||||||
|
type Error = IpNetworkError;
|
||||||
|
|
||||||
|
fn try_from(s: &str) -> Result<Self, Self::Error> {
|
||||||
|
IpNetwork::from_str(s)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
impl From<Ipv4Network> for IpNetwork {
|
impl From<Ipv4Network> for IpNetwork {
|
||||||
fn from(v4: Ipv4Network) -> IpNetwork {
|
fn from(v4: Ipv4Network) -> IpNetwork {
|
||||||
IpNetwork::V4(v4)
|
IpNetwork::V4(v4)
|
||||||
@ -292,7 +391,7 @@ impl From<IpAddr> for IpNetwork {
|
|||||||
}
|
}
|
||||||
|
|
||||||
impl fmt::Display for IpNetwork {
|
impl fmt::Display for IpNetwork {
|
||||||
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
|
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
|
||||||
match *self {
|
match *self {
|
||||||
IpNetwork::V4(net) => net.fmt(f),
|
IpNetwork::V4(net) => net.fmt(f),
|
||||||
IpNetwork::V6(net) => net.fmt(f),
|
IpNetwork::V6(net) => net.fmt(f),
|
||||||
@ -300,6 +399,35 @@ impl fmt::Display for IpNetwork {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[derive(Clone, Debug)]
|
||||||
|
enum IpNetworkIteratorInner {
|
||||||
|
V4(Ipv4NetworkIterator),
|
||||||
|
V6(Ipv6NetworkIterator),
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Clone, Debug)]
|
||||||
|
pub struct IpNetworkIterator {
|
||||||
|
inner: IpNetworkIteratorInner,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Iterator for IpNetworkIterator {
|
||||||
|
type Item = IpAddr;
|
||||||
|
fn next(&mut self) -> Option<IpAddr> {
|
||||||
|
match &mut self.inner {
|
||||||
|
IpNetworkIteratorInner::V4(iter) => iter.next().map(IpAddr::V4),
|
||||||
|
IpNetworkIteratorInner::V6(iter) => iter.next().map(IpAddr::V6),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl IntoIterator for &'_ IpNetwork {
|
||||||
|
type IntoIter = IpNetworkIterator;
|
||||||
|
type Item = IpAddr;
|
||||||
|
fn into_iter(self) -> IpNetworkIterator {
|
||||||
|
self.iter()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
/// Converts a `IpAddr` network mask into a prefix.
|
/// Converts a `IpAddr` network mask into a prefix.
|
||||||
/// If the mask is invalid this will return an `IpNetworkError::InvalidPrefix`.
|
/// If the mask is invalid this will return an `IpNetworkError::InvalidPrefix`.
|
||||||
pub fn ip_mask_to_prefix(mask: IpAddr) -> Result<u8, IpNetworkError> {
|
pub fn ip_mask_to_prefix(mask: IpAddr) -> Result<u8, IpNetworkError> {
|
||||||
@ -312,11 +440,13 @@ pub fn ip_mask_to_prefix(mask: IpAddr) -> Result<u8, IpNetworkError> {
|
|||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
mod test {
|
mod test {
|
||||||
#[test]
|
#[test]
|
||||||
|
#[cfg(feature = "serde")]
|
||||||
fn deserialize_from_serde_json_value() {
|
fn deserialize_from_serde_json_value() {
|
||||||
use super::*;
|
use super::*;
|
||||||
let network = IpNetwork::from_str("0.0.0.0/0").unwrap();
|
let network = IpNetwork::from_str("0.0.0.0/0").unwrap();
|
||||||
let val: serde_json::value::Value = serde_json::from_str(&serde_json::to_string(&network).unwrap()).unwrap();
|
let val: serde_json::value::Value =
|
||||||
let _deser: IpNetwork =
|
serde_json::from_str(&serde_json::to_string(&network).unwrap()).unwrap();
|
||||||
serde_json::from_value(val).expect("Fails to deserialize from json_value::value::Value");
|
let _deser: IpNetwork = serde_json::from_value(val)
|
||||||
|
.expect("Fails to deserialize from json_value::value::Value");
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -1,16 +1,9 @@
|
|||||||
extern crate serde;
|
#![cfg(feature = "serde")]
|
||||||
|
|
||||||
extern crate serde_json;
|
|
||||||
|
|
||||||
#[macro_use]
|
|
||||||
extern crate serde_derive;
|
|
||||||
|
|
||||||
extern crate ipnetwork;
|
|
||||||
|
|
||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
mod tests {
|
mod tests {
|
||||||
|
|
||||||
use ipnetwork::{IpNetwork, Ipv4Network, Ipv6Network};
|
use ipnetwork::{IpNetwork, Ipv4Network, Ipv6Network};
|
||||||
|
use serde::{Deserialize, Serialize};
|
||||||
use std::net::{Ipv4Addr, Ipv6Addr};
|
use std::net::{Ipv4Addr, Ipv6Addr};
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
@ -18,6 +11,7 @@ mod tests {
|
|||||||
let json_string = r#"{"ipnetwork":"127.1.0.0/24"}"#;
|
let json_string = r#"{"ipnetwork":"127.1.0.0/24"}"#;
|
||||||
|
|
||||||
#[derive(Serialize, Deserialize)]
|
#[derive(Serialize, Deserialize)]
|
||||||
|
#[cfg_attr(feature = "schemars", derive(schemars::JsonSchema))]
|
||||||
struct MyStruct {
|
struct MyStruct {
|
||||||
ipnetwork: Ipv4Network,
|
ipnetwork: Ipv4Network,
|
||||||
}
|
}
|
||||||
@ -28,6 +22,11 @@ mod tests {
|
|||||||
assert_eq!(mystruct.ipnetwork.prefix(), 24);
|
assert_eq!(mystruct.ipnetwork.prefix(), 24);
|
||||||
|
|
||||||
assert_eq!(::serde_json::to_string(&mystruct).unwrap(), json_string);
|
assert_eq!(::serde_json::to_string(&mystruct).unwrap(), json_string);
|
||||||
|
|
||||||
|
#[cfg(feature = "schemars")]
|
||||||
|
if let Err(s) = does_it_json::validate_with_output(&mystruct) {
|
||||||
|
panic!("{}", s);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
@ -35,6 +34,7 @@ mod tests {
|
|||||||
let json_string = r#"{"ipnetwork":"::1/0"}"#;
|
let json_string = r#"{"ipnetwork":"::1/0"}"#;
|
||||||
|
|
||||||
#[derive(Serialize, Deserialize)]
|
#[derive(Serialize, Deserialize)]
|
||||||
|
#[cfg_attr(feature = "schemars", derive(schemars::JsonSchema))]
|
||||||
struct MyStruct {
|
struct MyStruct {
|
||||||
ipnetwork: Ipv6Network,
|
ipnetwork: Ipv6Network,
|
||||||
}
|
}
|
||||||
@ -48,6 +48,11 @@ mod tests {
|
|||||||
assert_eq!(mystruct.ipnetwork.prefix(), 0);
|
assert_eq!(mystruct.ipnetwork.prefix(), 0);
|
||||||
|
|
||||||
assert_eq!(::serde_json::to_string(&mystruct).unwrap(), json_string);
|
assert_eq!(::serde_json::to_string(&mystruct).unwrap(), json_string);
|
||||||
|
|
||||||
|
#[cfg(feature = "schemars")]
|
||||||
|
if let Err(s) = does_it_json::validate_with_output(&mystruct) {
|
||||||
|
panic!("{}", s);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
@ -55,6 +60,7 @@ mod tests {
|
|||||||
let json_string = r#"{"ipnetwork":["127.1.0.0/24","::1/0"]}"#;
|
let json_string = r#"{"ipnetwork":["127.1.0.0/24","::1/0"]}"#;
|
||||||
|
|
||||||
#[derive(Serialize, Deserialize)]
|
#[derive(Serialize, Deserialize)]
|
||||||
|
#[cfg_attr(feature = "schemars", derive(schemars::JsonSchema))]
|
||||||
struct MyStruct {
|
struct MyStruct {
|
||||||
ipnetwork: Vec<IpNetwork>,
|
ipnetwork: Vec<IpNetwork>,
|
||||||
}
|
}
|
||||||
@ -70,5 +76,10 @@ mod tests {
|
|||||||
assert_eq!(mystruct.ipnetwork[1].prefix(), 0);
|
assert_eq!(mystruct.ipnetwork[1].prefix(), 0);
|
||||||
|
|
||||||
assert_eq!(::serde_json::to_string(&mystruct).unwrap(), json_string);
|
assert_eq!(::serde_json::to_string(&mystruct).unwrap(), json_string);
|
||||||
|
|
||||||
|
#[cfg(feature = "schemars")]
|
||||||
|
if let Err(s) = does_it_json::validate_with_output(&mystruct) {
|
||||||
|
panic!("{}", s);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
Reference in New Issue
Block a user