p2p 0.6.0

NAT Traversal for P2P communication
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

pub use self::overlay::Overlay;
pub use self::peer::Peer;

use common::event_loop::{spawn_event_loop, CoreMsg};
use common::read_config;
use p2p::{Config, NatMsg, TcpRendezvousServer, UdpRendezvousServer};
use std::io;
use std::sync::mpsc;

mod overlay;
mod peer;

#[derive(Serialize, Deserialize)]
pub struct FullConfig {
    pub server_cfg: ServerConfig,
    pub p2p_cfg: Config,
}

#[derive(Serialize, Deserialize)]
pub struct ServerConfig {
    overlay_port: u16,
}

pub fn entry_point() {
    let cfg: FullConfig = read_config("./server-config");

    let el = spawn_event_loop(cfg.p2p_cfg);
    let server_cfg = cfg.server_cfg;

    {
        let (tx, rx) = mpsc::channel();
        unwrap!(el.nat_tx.send(NatMsg::new(move |ifc, poll| {
            let _token_udp = unwrap!(UdpRendezvousServer::start(ifc, poll));
            let _token_tcp = unwrap!(TcpRendezvousServer::start(ifc, poll));
            unwrap!(tx.send(()));
        })));
        unwrap!(rx.recv());
    }

    println!("Rendezvous servers started successfully.");
    println!("Should this node also be the overlay [y/n] ?");

    let mut answer = String::new();
    unwrap!(io::stdin().read_line(&mut answer));
    answer = answer.trim().to_string().to_lowercase();

    if answer == "y" || answer == "yes" {
        let (tx, rx) = mpsc::channel();
        unwrap!(el.core_tx.send(CoreMsg::new(move |core, poll| {
            Overlay::start(core, poll, server_cfg.overlay_port);
            unwrap!(tx.send(()));
        })));
        unwrap!(rx.recv());
        println!("Overlay started successfully.");
    }

    println!("Everything done. Blocking main thread until user quits.");
    let mut quit = String::new();
    loop {
        println!("Enter 'q' to quit");
        unwrap!(io::stdin().read_line(&mut quit));
        quit = quit.trim().to_string().to_lowercase();
        if quit == "q" || quit == "quit" {
            break;
        }
        quit.clear();
    }
}