quic-p2p 0.7.0

Peer-to-peer networking library using QUIC
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

// Copyright 2019 MaidSafe.net limited.
//
// This SAFE Network Software is licensed to you under the MIT license <LICENSE-MIT
// http://opensource.org/licenses/MIT> or the Modified BSD license <LICENSE-BSD
// https://opensource.org/licenses/BSD-3-Clause>, at your option. This file may not be copied,
// modified, or distributed except according to those terms. Please review the Licences for the
// specific language governing permissions and limitations relating to use of the SAFE Network
// Software.

use crate::{
    communicate,
    connection::{BootstrapGroupRef, Connection, FromPeer, QConn, ToPeer},
    context::ctx_mut,
    event::Event,
    peer::Peer,
    QuicP2pError,
};
use futures::future::{self, TryFutureExt};
use futures::stream::StreamExt;
use log::{debug, info, trace};

/// Start listening
pub fn listen(incoming_connections: quinn::Incoming) {
    let leaf = incoming_connections.for_each(move |connecting| {
        let leaf = connecting
            .map_err(|e| debug!("New connection errored out: {}", e))
            .map_ok(handle_new_conn);
        let _ = tokio::spawn(leaf);
        future::ready(())
    });

    let _ = tokio::spawn(leaf);
}

enum Action {
    HandleDuplicate(QConn),
    HandleAlreadyBootstrapped,
    Continue(Option<BootstrapGroupRef>),
}

fn handle_new_conn(
    quinn::NewConnection {
        connection,
        uni_streams,
        bi_streams,
        ..
    }: quinn::NewConnection,
) {
    let q_conn = QConn::from(connection);

    let peer_addr = q_conn.remote_address();

    trace!("Incoming connection from peer: {}", peer_addr);

    let state = ctx_mut(|c| {
        let event_tx = c.event_tx.clone();
        let conn = c
            .connections
            .entry(peer_addr)
            .or_insert_with(|| Connection::new(peer_addr, event_tx, None));
        if conn.from_peer.is_no_connection() {
            conn.from_peer = FromPeer::Established {
                q_conn,
                pending_reads: Default::default(),
            };

            if let ToPeer::Established { .. } = conn.to_peer {
                // TODO come back to all the connected-to events and see if we are handling all
                // cases
                if let Some(mut bootstrap_group_ref) = conn.bootstrap_group_ref.take() {
                    if bootstrap_group_ref.is_bootstrap_successful_yet() {
                        return Action::HandleAlreadyBootstrapped;
                    }

                    if let Err(e) =
                        bootstrap_group_ref.send(Event::BootstrappedTo { node: peer_addr })
                    {
                        info!("ERROR in informing user about a new peer: {:?} - {}", e, e);
                    }

                    return Action::Continue(Some(bootstrap_group_ref));
                } else if let Err(e) = c.event_tx.send(Event::ConnectedTo {
                    peer: Peer::Node(peer_addr),
                }) {
                    info!("ERROR in informing user about a new peer: {:?} - {}", e, e);
                };
            };

            Action::Continue(None)
        } else {
            Action::HandleDuplicate(q_conn)
        }
    });

    match state {
        Action::HandleDuplicate(_q_conn) => {
            debug!("Not allowing duplicate connection from peer: {}", peer_addr);
        }
        Action::HandleAlreadyBootstrapped => crate::utils::handle_communication_err(
            peer_addr,
            &QuicP2pError::ConnectionCancelled,
            "Connection already bootstrapped",
            None,
        ),
        Action::Continue(bootstrap_group) => {
            if let Some(bootstrap_group_ref) = bootstrap_group {
                bootstrap_group_ref.terminate_group(true);
            }
            communicate::read_from_peer(peer_addr, uni_streams, bi_streams);
        }
    }
}