pea-p2p 0.1.1

Peacash P2P
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

use libp2p::multiaddr::Protocol;
use libp2p::Multiaddr;
use std::net::IpAddr;
pub fn ip(multiaddr: &Multiaddr) -> Option<Multiaddr> {
    let components = multiaddr.iter().collect::<Vec<_>>();
    let mut multiaddr: Multiaddr = "".parse().unwrap();
    match components.get(0) {
        Some(Protocol::Ip4(ip)) => multiaddr.push(Protocol::Ip4(*ip)),
        Some(Protocol::Ip6(ip)) => multiaddr.push(Protocol::Ip6(*ip)),
        _ => return None,
    };
    Some(multiaddr)
}
pub fn ip_port(multiaddr: &Multiaddr) -> Option<Multiaddr> {
    let components = multiaddr.iter().collect::<Vec<_>>();
    let mut multiaddr: Multiaddr = "".parse().unwrap();
    match components.get(0) {
        Some(Protocol::Ip4(ip)) => multiaddr.push(Protocol::Ip4(*ip)),
        Some(Protocol::Ip6(ip)) => multiaddr.push(Protocol::Ip6(*ip)),
        _ => return None,
    };
    match components.get(1) {
        Some(Protocol::Tcp(port)) => {
            if port == &9333_u16 {
                return Some(multiaddr);
            }
            multiaddr.push(Protocol::Tcp(*port))
        }
        _ => return Some(multiaddr),
    };
    Some(multiaddr)
}
pub fn has_port(multiaddr: &Multiaddr) -> bool {
    let components = multiaddr.iter().collect::<Vec<_>>();
    matches!(components.get(1), Some(Protocol::Tcp(_)))
}
pub fn ip_addr(multiaddr: &Multiaddr) -> Option<IpAddr> {
    match multiaddr.iter().collect::<Vec<_>>().first() {
        Some(Protocol::Ip4(ip)) => Some(IpAddr::V4(*ip)),
        Some(Protocol::Ip6(ip)) => Some(IpAddr::V6(*ip)),
        _ => None,
    }
}
#[cfg(test)]
mod tests {
    use super::*;
    #[test]
    fn test_filter_ip() {
        assert_eq!(ip(&"".parse::<Multiaddr>().unwrap()), None);
        assert_eq!(ip(&"/tcp/9333".parse::<Multiaddr>().unwrap()), None);
        assert_eq!(
            ip(&"/ip4/0.0.0.0/tcp/9333".parse::<Multiaddr>().unwrap()).unwrap(),
            "/ip4/0.0.0.0".parse::<Multiaddr>().unwrap()
        );
    }
    #[test]
    fn test_filter_ip_port() {
        assert_eq!(ip_port(&"".parse::<Multiaddr>().unwrap()), None);
        assert_eq!(ip_port(&"/tcp/9333".parse::<Multiaddr>().unwrap()), None);
        assert_eq!(
            ip_port(&"/ip4/0.0.0.0".parse::<Multiaddr>().unwrap()).unwrap(),
            "/ip4/0.0.0.0".parse::<Multiaddr>().unwrap()
        );
        assert_eq!(
            ip_port(&"/ip4/0.0.0.0/tcp/9333".parse::<Multiaddr>().unwrap()).unwrap(),
            "/ip4/0.0.0.0".parse::<Multiaddr>().unwrap()
        );
        assert_eq!(
            ip_port(&"/ip4/0.0.0.0/tcp/9334".parse::<Multiaddr>().unwrap()).unwrap(),
            "/ip4/0.0.0.0/tcp/9334".parse::<Multiaddr>().unwrap()
        );
    }
    #[test]
    fn test_has_port() {
        assert_eq!(has_port(&"".parse::<Multiaddr>().unwrap()), false);
        assert_eq!(has_port(&"/ip4/0.0.0.0".parse::<Multiaddr>().unwrap()), false);
        assert_eq!(has_port(&"/tcp/9333".parse::<Multiaddr>().unwrap()), false);
        assert_eq!(has_port(&"/ip4/0.0.0.0/tcp/9333".parse::<Multiaddr>().unwrap()), true);
    }
}