blob: 076c3ac757dc22aa66cbbeaa2044d8382c2fe812 (
plain)
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
|
extern crate clap;
extern crate env_logger;
extern crate url;
/// An example of a client-server-agnostic WebSocket that takes input from stdin and sends that
/// input to all other peers.
///
/// For example, to create a network like this:
///
/// 3013 ---- 3012 ---- 3014
/// \ |
/// \ |
/// \ |
/// \ |
/// \ |
/// \ |
/// \ |
/// 3015
///
/// Run these commands in separate processes
/// ./peer2peer
/// ./peer2peer --server localhost:3013 ws://localhost:3012
/// ./peer2peer --server localhost:3014 ws://localhost:3012
/// ./peer2peer --server localhost:3015 ws://localhost:3012 ws://localhost:3013
///
/// Stdin on 3012 will be sent to all other peers
/// Stdin on 3013 will be sent to 3012 and 3015
/// Stdin on 3014 will be sent to 3012 only
/// Stdin on 3015 will be sent to 3012 and 2013
extern crate ws;
#[macro_use]
extern crate log;
use std::io;
use std::io::prelude::*;
use std::thread;
use clap::{App, Arg};
fn main() {
// Setup logging
env_logger::init();
// Parse command line arguments
let matches = App::new("Simple Peer 2 Peer")
.version("1.0")
.author("Jason Housley <housleyjk@gmail.com>")
.about("Connect to other peers and listen for incoming connections.")
.arg(
Arg::with_name("server")
.short("s")
.long("server")
.value_name("SERVER")
.help("Set the address to listen for new connections."),
)
.arg(
Arg::with_name("PEER")
.help("A WebSocket URL to attempt to connect to at start.")
.multiple(true),
)
.get_matches();
// Get address of this peer
let my_addr = matches.value_of("server").unwrap_or("localhost:3012");
// Create simple websocket that just prints out messages
let mut me = ws::WebSocket::new(|_| {
move |msg| {
info!("Peer {} got message: {}", my_addr, msg);
Ok(())
}
}).unwrap();
// Get a sender for ALL connections to the websocket
let broacaster = me.broadcaster();
// Setup thread for listening to stdin and sending messages to connections
let input = thread::spawn(move || {
let stdin = io::stdin();
for line in stdin.lock().lines() {
// Send a message to all connections regardless of
// how those connections were established
broacaster.send(line.unwrap()).unwrap();
}
});
// Connect to any existing peers specified on the cli
if let Some(peers) = matches.values_of("PEER") {
for peer in peers {
me.connect(url::Url::parse(peer).unwrap()).unwrap();
}
}
// Run the websocket
me.listen(my_addr).unwrap();
input.join().unwrap();
}
|
