routing 0.5.2

A secured storage DHT
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

// Copyright 2015 MaidSafe.net limited.
//
// This SAFE Network Software is licensed to you under (1) the MaidSafe.net Commercial License,
// version 1.0 or later, or (2) The General Public License (GPL), version 3, depending on which
// licence you accepted on initial access to the Software (the "Licences").
//
// By contributing code to the SAFE Network Software, or to this project generally, you agree to be
// bound by the terms of the MaidSafe Contributor Agreement, version 1.0.  This, along with the
// Licenses can be found in the root directory of this project at LICENSE, COPYING and CONTRIBUTOR.
//
// Unless required by applicable law or agreed to in writing, the SAFE Network Software distributed
// under the GPL Licence is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
// KIND, either express or implied.
//
// Please review the Licences for the specific language governing permissions and limitations
// relating to use of the SAFE Network Software.

use crust::{Endpoint, Port};
use std::collections::HashSet;
use std::net::SocketAddr;
use ip::IpAddr;

pub struct Acceptors {
    endpoints: HashSet<::crust::Endpoint>,
    bootstrap_ip: Option<IpAddr>,
    tcp_accepting_port: Option<Port>,
    utp_accepting_port: Option<Port>,
}

impl Acceptors {
    pub fn new() -> Acceptors {
        Acceptors {
            endpoints: HashSet::new(),
            bootstrap_ip: None,
            tcp_accepting_port: None,
            utp_accepting_port: None,
        }
    }

    pub fn set_bootstrap_ip(&mut self, bootstrap_endpoint: Endpoint) {
        self.bootstrap_ip = Some(Self::ip_from_socketaddr(bootstrap_endpoint.get_address()));
    }

    pub fn set_tcp_accepting_port(&mut self, accepting_port: Port) {
        match accepting_port {
            Port::Tcp(port) => {
                self.tcp_accepting_port = Some(Port::Tcp(port));
                if let Some(ref bootstrap_ip) = self.bootstrap_ip {
                    let _ = self.endpoints
                                .insert(Endpoint::new(bootstrap_ip.clone(), Port::Tcp(port)));
                }
            }
            _ => unreachable!(),
        }
    }

    #[allow(unused)]
    pub fn set_utp_accepting_port(&mut self, accepting_port: Port) {
        match accepting_port {
            Port::Utp(port) => self.utp_accepting_port = Some(Port::Utp(port)),
            _ => unreachable!(),
        }
    }

    pub fn add(&mut self, our_endpoint: Endpoint) {
        match our_endpoint {
            Endpoint::Tcp(socket_addr) => {
                if let Some(ref port) = self.tcp_accepting_port {
                    let _ = self.endpoints
                                .insert(Endpoint::new(Self::ip_from_socketaddr(socket_addr),
                                                      port.clone()));
                }
            }
            Endpoint::Utp(socket_addr) => {
                if let Some(ref port) = self.utp_accepting_port {
                    let _ = self.endpoints
                                .insert(Endpoint::new(Self::ip_from_socketaddr(socket_addr),
                                                      port.clone()));
                }
            }
        }
    }

    pub fn endpoints(&self) -> Vec<Endpoint> {
        self.endpoints.iter().cloned().collect()
    }

    fn ip_from_socketaddr(addr: SocketAddr) -> IpAddr {
        match addr {
            SocketAddr::V4(address) => IpAddr::V4(*address.ip()),
            SocketAddr::V6(address) => IpAddr::V6(*address.ip()),
        }
    }
}