🚧 sketching out the request-response control protocol

main
_ 2021-07-17 06:43:14 +00:00
parent b344e3f8ee
commit ce817eab1a
3 changed files with 134 additions and 75 deletions

View File

@ -13,52 +13,63 @@ async fn main () -> anyhow::Result <()> {
debug! ("Connecting to relay server");
let quinn::NewConnection {
connection,
mut bi_streams,
..
} = endpoint.connect (&server_addr, "localhost")?.await?;
debug! ("Waiting for relay server to forward a bi stream");
let (mut send, mut recv) = connection.open_bi ().await?;
let (mut relay_send, mut relay_recv) = bi_streams.next ().await.ok_or_else (|| anyhow::anyhow! ("Relay server didn't open a bi stream"))??;
let req_buf = [4u8, 0, 0, 0];
send.write_all (&req_buf).await?;
debug! ("Connecting to TCP end server");
let mut resp_buf = [0u8, 0, 0, 0];
recv.read_exact (&mut resp_buf).await?;
let stream = TcpStream::connect ("127.0.0.1:30382").await?;
let (mut local_recv, mut local_send) = stream.into_split ();
if false {
debug! ("Waiting for relay server to forward a bi stream");
debug! ("Relaying bytes...");
let (mut relay_send, mut relay_recv) = bi_streams.next ().await.ok_or_else (|| anyhow::anyhow! ("Relay server didn't open a bi stream"))??;
let uplink_task = tokio::spawn (async move {
// Uplink - local client to relay server
debug! ("Connecting to TCP end server");
let mut buf = vec! [0u8; 65_536];
loop {
let bytes_read = local_recv.read (&mut buf).await?;
let buf_slice = &buf [0..bytes_read];
relay_send.write_all (buf_slice).await?;
}
let stream = TcpStream::connect ("127.0.0.1:30382").await?;
let (mut local_recv, mut local_send) = stream.into_split ();
debug! ("Uplink closed");
debug! ("Relaying bytes...");
Ok::<_, anyhow::Error> (())
});
let uplink_task = tokio::spawn (async move {
// Uplink - local client to relay server
let downlink_task = tokio::spawn (async move {
// Downlink - Relay server to local client
let mut buf = vec! [0u8; 65_536];
loop {
let bytes_read = local_recv.read (&mut buf).await?;
let buf_slice = &buf [0..bytes_read];
relay_send.write_all (buf_slice).await?;
}
let mut buf = vec! [0u8; 65_536];
while let Some (bytes_read) = relay_recv.read (&mut buf).await? {
let buf_slice = &buf [0..bytes_read];
local_send.write_all (buf_slice).await?;
}
debug! ("Uplink closed");
debug! ("Downlink closed");
Ok::<_, anyhow::Error> (())
});
Ok::<_, anyhow::Error> (())
});
let downlink_task = tokio::spawn (async move {
// Downlink - Relay server to local client
uplink_task.await??;
downlink_task.await??;
let mut buf = vec! [0u8; 65_536];
while let Some (bytes_read) = relay_recv.read (&mut buf).await? {
let buf_slice = &buf [0..bytes_read];
local_send.write_all (buf_slice).await?;
}
debug! ("Downlink closed");
Ok::<_, anyhow::Error> (())
});
uplink_task.await??;
downlink_task.await??;
}
Ok (())
}

View File

@ -1,5 +1,3 @@
use tokio::task::JoinHandle;
use quic_demo::prelude::*;
#[tokio::main]
@ -10,52 +8,99 @@ async fn main () -> anyhow::Result <()> {
let (mut incoming, server_cert) = make_server_endpoint (server_addr)?;
tokio::fs::write ("quic_server.crt", &server_cert).await?;
debug! ("Waiting for end server to connect");
let relay_state = RelayState::default ();
let relay_state = Arc::new (Mutex::new (relay_state));
let end_server_conn = incoming.next ().await.ok_or_else (|| anyhow::anyhow! ("No end server connection"))?;
while let Some (conn) = incoming.next ().await {
let relay_state = Arc::clone (&relay_state);
let end_server_conn = end_server_conn.await?;
// Each new connection gets its own task
tokio::spawn (async move {
let quinn::NewConnection {
connection,
mut bi_streams,
..
} = conn.await?;
let quinn::NewConnection {
connection: end_server_conn,
..
} = end_server_conn;
// Everyone who connects must identify themselves with the first
// bi stream
// TODO: Timeout
debug! ("Waiting for client to connect");
let (mut send, mut recv) = bi_streams.next ().await.ok_or_else (|| anyhow::anyhow! ("QUIC client didn't identify itself"))??;
let client_conn = incoming.next ().await.ok_or_else (|| anyhow::anyhow! ("No client connection"))?;
let mut req_buf = [0u8; 4];
recv.read_exact (&mut req_buf).await?;
let client_conn = client_conn.await?;
match req_buf [0] {
4 => debug! ("Server-side proxy (P4) connected"),
2 => debug! ("Client-side proxy (P2) connected"),
_ => bail! ("Unknown QUIC client type"),
}
let quinn::NewConnection {
connection: _client_conn,
bi_streams: mut client_incoming_bi_streams,
..
} = client_conn;
let resp_buf = [20u8, 0, 0, 0];
send.write_all (&resp_buf).await?;
debug! ("Waiting for client to open bi stream");
Ok::<_, anyhow::Error> (())
});
}
let (client_send, client_recv) = client_incoming_bi_streams.next ().await.ok_or_else (|| anyhow::anyhow! ("Client didn't open a bi stream"))??;
if false {
debug! ("Waiting for end server to connect");
debug! ("Opening bi stream to the end server");
let end_server_conn = incoming.next ().await.ok_or_else (|| anyhow::anyhow! ("No end server connection"))?;
let (server_send, server_recv) = end_server_conn.open_bi ().await?;
let end_server_conn = end_server_conn.await?;
debug! ("Relaying bytes...");
let quinn::NewConnection {
connection: end_server_conn,
..
} = end_server_conn;
let ptth_conn = PtthNewConnection {
client_send,
client_recv,
server_send,
server_recv,
}.build ();
debug! ("Waiting for client to connect");
ptth_conn.uplink_task.await??;
ptth_conn.downlink_task.await??;
let client_conn = incoming.next ().await.ok_or_else (|| anyhow::anyhow! ("No client connection"))?;
let client_conn = client_conn.await?;
let quinn::NewConnection {
connection: _client_conn,
bi_streams: mut client_incoming_bi_streams,
..
} = client_conn;
debug! ("Waiting for client to open bi stream");
let (client_send, client_recv) = client_incoming_bi_streams.next ().await.ok_or_else (|| anyhow::anyhow! ("Client didn't open a bi stream"))??;
debug! ("Opening bi stream to the end server");
let (server_send, server_recv) = end_server_conn.open_bi ().await?;
debug! ("Relaying bytes...");
let ptth_conn = PtthNewConnection {
client_send,
client_recv,
server_send,
server_recv,
}.build ();
ptth_conn.uplink_task.await??;
ptth_conn.downlink_task.await??;
}
Ok (())
}
#[derive (Default)]
struct RelayState {
}
impl RelayState {
}
struct PtthNewConnection {
client_send: quinn::SendStream,
client_recv: quinn::RecvStream,

View File

@ -1,4 +1,5 @@
pub use std::{
sync::Arc,
time::Duration,
};
@ -9,6 +10,8 @@ pub use tokio::{
AsyncReadExt,
AsyncWriteExt,
},
sync::Mutex,
task::JoinHandle,
};
pub use tracing::{
debug,