2021-07-17 23:58:11 +00:00
|
|
|
use structopt::StructOpt;
|
|
|
|
use tokio::net::TcpListener;
|
2021-07-17 02:39:08 +00:00
|
|
|
|
2021-07-17 01:10:30 +00:00
|
|
|
use quic_demo::prelude::*;
|
2021-07-18 20:32:49 +00:00
|
|
|
use protocol::PeerId;
|
2021-07-17 01:10:30 +00:00
|
|
|
|
2021-07-17 23:58:11 +00:00
|
|
|
#[derive (Debug, StructOpt)]
|
|
|
|
struct Opt {
|
|
|
|
#[structopt (long)]
|
|
|
|
relay_addr: Option <String>,
|
|
|
|
#[structopt (long)]
|
2021-07-18 22:22:48 +00:00
|
|
|
client_id: Option <PeerId>,
|
2021-07-17 23:58:11 +00:00
|
|
|
#[structopt (long)]
|
2021-07-18 20:32:49 +00:00
|
|
|
server_id: Option <PeerId>,
|
2021-07-18 22:55:32 +00:00
|
|
|
#[structopt (long)]
|
2021-07-19 00:01:46 +00:00
|
|
|
client_tcp_port: Option <u16>,
|
|
|
|
#[structopt (long)]
|
2021-07-18 22:55:32 +00:00
|
|
|
server_tcp_port: Option <u16>,
|
2021-07-17 23:58:11 +00:00
|
|
|
}
|
|
|
|
|
2021-07-17 01:10:30 +00:00
|
|
|
#[tokio::main]
|
|
|
|
async fn main () -> anyhow::Result <()> {
|
|
|
|
tracing_subscriber::fmt::init ();
|
|
|
|
|
2021-07-17 23:58:11 +00:00
|
|
|
let opt = Opt::from_args ();
|
|
|
|
|
2021-07-17 01:10:30 +00:00
|
|
|
let server_cert = tokio::fs::read ("quic_server.crt").await?;
|
2021-07-17 23:58:11 +00:00
|
|
|
let relay_addr = opt.relay_addr.unwrap_or_else (|| String::from ("127.0.0.1:30380")).parse ()?;
|
2021-07-17 01:10:30 +00:00
|
|
|
let endpoint = make_client_endpoint ("0.0.0.0:0".parse ()?, &[&server_cert])?;
|
|
|
|
|
2021-07-21 23:15:57 +00:00
|
|
|
debug! ("Connecting to relay server");
|
2021-07-17 01:10:30 +00:00
|
|
|
|
2021-07-18 21:18:49 +00:00
|
|
|
let client_id = opt.client_id.unwrap_or_else (|| "bogus_client".to_string ());
|
2021-07-18 18:26:19 +00:00
|
|
|
|
2021-07-17 01:10:30 +00:00
|
|
|
let quinn::NewConnection {
|
|
|
|
connection,
|
|
|
|
..
|
2021-07-18 21:18:49 +00:00
|
|
|
} = protocol::p2_connect_to_p3 (&endpoint, &relay_addr, &client_id).await?;
|
2021-07-17 02:39:08 +00:00
|
|
|
|
2021-07-19 00:01:46 +00:00
|
|
|
// End of per-client stuff
|
|
|
|
// Beginning of per-port stuff
|
|
|
|
|
2021-07-18 22:22:48 +00:00
|
|
|
let server_id = opt.server_id.unwrap_or_else (|| "bogus_server".to_string ());
|
2021-07-17 23:14:50 +00:00
|
|
|
|
2021-07-19 00:01:46 +00:00
|
|
|
let client_tcp_port = opt.client_tcp_port.unwrap_or (30381);
|
|
|
|
let server_tcp_port = opt.server_tcp_port.unwrap_or (30382);
|
2021-07-18 22:55:32 +00:00
|
|
|
let listener = TcpListener::bind (("127.0.0.1", client_tcp_port)).await?;
|
2021-07-17 20:02:16 +00:00
|
|
|
|
2021-07-19 00:01:46 +00:00
|
|
|
// End of per-port stuff
|
|
|
|
// Beginning of per-connection stuff
|
|
|
|
|
2021-09-10 20:36:03 +00:00
|
|
|
let task_tcp_server = tokio::spawn (async move {
|
|
|
|
loop {
|
|
|
|
let (tcp_socket, _) = listener.accept ().await?;
|
|
|
|
let connection = connection.clone ();
|
|
|
|
let server_id = server_id.clone ();
|
2021-07-17 20:07:26 +00:00
|
|
|
|
2021-09-10 20:36:03 +00:00
|
|
|
tokio::spawn (async move {
|
|
|
|
let (local_recv, local_send) = tcp_socket.into_split ();
|
|
|
|
|
|
|
|
debug! ("Starting PTTH connection");
|
|
|
|
|
|
|
|
let (relay_send, relay_recv) = protocol::p2_connect_to_p5 (&connection, &server_id, server_tcp_port).await?;
|
|
|
|
|
|
|
|
trace! ("Relaying bytes...");
|
|
|
|
|
|
|
|
let ptth_conn = quic_demo::connection::NewConnection {
|
|
|
|
local_send,
|
|
|
|
local_recv,
|
|
|
|
relay_send,
|
|
|
|
relay_recv,
|
|
|
|
}.build ();
|
|
|
|
|
|
|
|
ptth_conn.wait_for_close ().await?;
|
|
|
|
|
|
|
|
debug! ("Ended PTTH connection");
|
|
|
|
|
|
|
|
Ok::<_, anyhow::Error> (())
|
|
|
|
});
|
|
|
|
}
|
|
|
|
|
|
|
|
Ok::<_, anyhow::Error> (())
|
|
|
|
});
|
|
|
|
|
|
|
|
debug! ("Accepting local TCP connections from P1");
|
|
|
|
|
|
|
|
task_tcp_server.await??;
|
|
|
|
|
|
|
|
Ok (())
|
2021-07-17 20:02:16 +00:00
|
|
|
}
|