65 lines
1.6 KiB
Rust
65 lines
1.6 KiB
Rust
use tokio::net::TcpStream;
|
|
|
|
use quic_demo::prelude::*;
|
|
|
|
#[tokio::main]
|
|
async fn main () -> anyhow::Result <()> {
|
|
tracing_subscriber::fmt::init ();
|
|
|
|
let server_cert = tokio::fs::read ("quic_server.crt").await?;
|
|
let server_addr = "127.0.0.1:30380".parse ()?;
|
|
let endpoint = make_client_endpoint ("0.0.0.0:0".parse ()?, &[&server_cert])?;
|
|
|
|
debug! ("Connecting to relay server");
|
|
|
|
let quinn::NewConnection {
|
|
mut bi_streams,
|
|
..
|
|
} = endpoint.connect (&server_addr, "localhost")?.await?;
|
|
|
|
debug! ("Waiting for relay server to forward a bi stream");
|
|
|
|
let (mut relay_send, mut relay_recv) = bi_streams.next ().await.ok_or_else (|| anyhow::anyhow! ("Relay server didn't open a bi stream"))??;
|
|
|
|
debug! ("Connecting to TCP end server");
|
|
|
|
let stream = TcpStream::connect ("127.0.0.1:30382").await?;
|
|
let (mut local_recv, mut local_send) = stream.into_split ();
|
|
|
|
debug! ("Relaying bytes...");
|
|
|
|
let uplink_task = tokio::spawn (async move {
|
|
// Uplink - local client to relay 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?;
|
|
}
|
|
|
|
debug! ("Uplink closed");
|
|
|
|
Ok::<_, anyhow::Error> (())
|
|
});
|
|
|
|
let downlink_task = tokio::spawn (async move {
|
|
// Downlink - Relay server to local client
|
|
|
|
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 (())
|
|
}
|