rlibdht 0.1.0

Rust DHT library
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

use std::net::IpAddr;

pub struct NetMask {
    pub(crate) address: [u8; 16],//Vec<u8>,
    pub(crate) mask: u32
}

impl NetMask {

    pub fn new(address: IpAddr, mask: u32) -> Result<Self, ()> {
        if let IpAddr::V6(v6) = address {
            let octets = v6.octets();

            return Ok(Self {
                address: octets,
                mask
            })
        }

        Err(())
    }

    pub fn contains(&self, other_address: IpAddr) -> bool {
        let other = match other_address {
            IpAddr::V4(ipv4) => ipv4.octets().to_vec(),
            IpAddr::V6(ipv6) => ipv6.octets().to_vec(),
        };

        if self.address.len() != other.len() {
            return false;
        }

        for i in 0..(self.mask / 8) as usize {
            if self.address[i] != other[i] {
                return false;
            }
        }

        if self.mask % 8 == 0 {
            return true;
        }

        let offset = (self.mask / 8) as usize;
        let probe_mask = ((0xff00 >> (self.mask % 8)) & 0xff) as u8;

        (self.address[offset] & probe_mask) == (other[offset] & probe_mask)
    }
}