openc2 0.2.0

Rust types for OpenC2 commands and responses
Documentation 
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
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153

use std::{
    fmt,
    net::{Ipv4Addr, Ipv6Addr},
    str::FromStr,
};

use serde_with::{DeserializeFromStr, SerializeDisplay};

use crate::error::ValidationError;

#[derive(
    Clone, Copy, PartialEq, Eq, Hash, PartialOrd, Ord, SerializeDisplay, DeserializeFromStr,
)]
pub struct Ipv4Net {
    address: Ipv4Addr,
    /// The prefix length - this should never be 32.
    prefix_len: Option<u8>,
}

impl Ipv4Net {
    pub fn new(address: Ipv4Addr, prefix_len: Option<u8>) -> Result<Self, ValidationError> {
        Ok(Self {
            address,
            prefix_len: match prefix_len {
                Some(pf) if pf > 32 => {
                    return Err(ValidationError::new(
                        "Prefix length must be between 0 and 32",
                    ));
                }
                Some(32) | None => None,
                Some(other) => Some(other),
            },
        })
    }
}

impl fmt::Debug for Ipv4Net {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        write!(f, "{self}")
    }
}

impl fmt::Display for Ipv4Net {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        if let Some(prefix_len) = self.prefix_len {
            write!(f, "{}/{prefix_len}", self.address)
        } else {
            self.address.fmt(f)
        }
    }
}

impl FromStr for Ipv4Net {
    type Err = ValidationError;
    fn from_str(s: &str) -> Result<Self, Self::Err> {
        let (addr, prefix) = s
            .split_once('/')
            .map(|(s, t)| (s, Some(t)))
            .unwrap_or_else(|| (s, None));
        let address = addr
            .parse()
            .map_err(|e| ValidationError::new(format!("Invalid IPv4 address: {e}")))?;
        let prefix_len: Option<u8> = prefix
            .map(|prefix| {
                prefix
                    .parse()
                    .map_err(|e| ValidationError::new(format!("Invalid prefix length: {e}")))
            })
            .transpose()?;
        Self::new(address, prefix_len)
    }
}

impl From<Ipv4Addr> for Ipv4Net {
    fn from(address: Ipv4Addr) -> Self {
        Self {
            address,
            prefix_len: None,
        }
    }
}

#[derive(
    Clone, Copy, PartialEq, Eq, Hash, PartialOrd, Ord, SerializeDisplay, DeserializeFromStr,
)]
pub struct Ipv6Net {
    address: Ipv6Addr,
    /// The prefix length - this should never be 128.
    prefix_len: Option<u8>,
}

impl Ipv6Net {
    pub fn new(address: Ipv6Addr, prefix_len: Option<u8>) -> Result<Self, ValidationError> {
        Ok(Self {
            address,
            prefix_len: match prefix_len {
                Some(pf) if pf > 128 => {
                    return Err(ValidationError::new(
                        "Prefix length must be between 0 and 128",
                    ));
                }
                Some(128) | None => None,
                Some(other) => Some(other),
            },
        })
    }
}

impl fmt::Debug for Ipv6Net {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        write!(f, "{self}")
    }
}

impl fmt::Display for Ipv6Net {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        if let Some(prefix_len) = self.prefix_len {
            write!(f, "{}/{}", self.address, prefix_len)
        } else {
            self.address.fmt(f)
        }
    }
}

impl FromStr for Ipv6Net {
    type Err = ValidationError;
    fn from_str(s: &str) -> Result<Self, Self::Err> {
        let (addr, prefix) = s
            .split_once('/')
            .map(|(s, t)| (s, Some(t)))
            .unwrap_or_else(|| (s, None));
        let address = addr
            .parse()
            .map_err(|e| ValidationError::new(format!("Invalid IPv6 address: {e}")))?;
        let prefix_len: Option<u8> = prefix
            .map(|prefix| {
                prefix
                    .parse()
                    .map_err(|e| ValidationError::new(format!("Invalid prefix length: {e}")))
            })
            .transpose()?;
        Self::new(address, prefix_len)
    }
}

impl From<Ipv6Addr> for Ipv6Net {
    fn from(address: Ipv6Addr) -> Self {
        Self {
            address,
            prefix_len: None,
        }
    }
}