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

use std::{str, net};

#[derive(Debug, PartialEq, Clone, Copy)]
pub enum InternetProtocol {
	Any,
	IpV4,
	IpV6,
}

impl Default for InternetProtocol {
	fn default() -> Self {
		InternetProtocol::Any
	}
}

impl str::FromStr for InternetProtocol {
	type Err = &'static str;

	fn from_str(s: &str) -> Result<Self, Self::Err> {
		match s {
			"ipv4" => Ok(InternetProtocol::IpV4),
			"ipv6" => Ok(InternetProtocol::IpV6),
			_ => Err("Invalid internet protocol"),
		}
	}
}

impl InternetProtocol {
	pub fn is_allowed(&self, addr: &net::SocketAddr) -> bool {
		match *self {
			InternetProtocol::Any => true,
			InternetProtocol::IpV4 => match *addr {
				net::SocketAddr::V4(_) => true,
				_ => false,
			},
			InternetProtocol::IpV6 => match *addr {
				net::SocketAddr::V6(_) => true,
				_ => false,
			}
		}
	}
}

#[cfg(test)]
mod tests {
	use super::InternetProtocol;

	#[test]
	fn test_default_internet_protocol() {
		assert_eq!(InternetProtocol::default(), InternetProtocol::Any);
	}

	#[test]
	fn test_parsing_internet_protocol() {
		assert_eq!(InternetProtocol::IpV4, "ipv4".parse().unwrap());
		assert_eq!(InternetProtocol::IpV6, "ipv6".parse().unwrap());
		assert!("sa".parse::<InternetProtocol>().is_err());
	}
}