feat: s2n quic test
This commit is contained in:
36
__network/s2n_quic/src/bin/client.rs
Normal file
36
__network/s2n_quic/src/bin/client.rs
Normal file
@@ -0,0 +1,36 @@
|
||||
use s2n_quic::{Client, client::Connect};
|
||||
use std::{error::Error, net::SocketAddr};
|
||||
|
||||
/// NOTE: this certificate is to be used for demonstration purposes only!
|
||||
pub static CERT_PEM: &str = include_str!("../../cert.pem");
|
||||
|
||||
#[tokio::main]
|
||||
async fn main() -> Result<(), Box<dyn Error>> {
|
||||
let client = Client::builder()
|
||||
.with_tls(CERT_PEM)?
|
||||
.with_io("0.0.0.0:0")?
|
||||
.start()?;
|
||||
|
||||
let addr: SocketAddr = "127.0.0.1:4433".parse()?;
|
||||
let connect = Connect::new(addr).with_server_name("localhost");
|
||||
let mut connection = client.connect(connect).await?;
|
||||
|
||||
// ensure the connection doesn't time out with inactivity
|
||||
connection.keep_alive(true)?;
|
||||
|
||||
// open a new stream and split the receiving and sending sides
|
||||
let stream = connection.open_bidirectional_stream().await?;
|
||||
let (mut receive_stream, mut send_stream) = stream.split();
|
||||
|
||||
// spawn a task that copies responses from the server to stdout
|
||||
tokio::spawn(async move {
|
||||
let mut stdout = tokio::io::stdout();
|
||||
let _ = tokio::io::copy(&mut receive_stream, &mut stdout).await;
|
||||
});
|
||||
|
||||
// copy data from stdin and send it to the server
|
||||
let mut stdin = tokio::io::stdin();
|
||||
tokio::io::copy(&mut stdin, &mut send_stream).await?;
|
||||
|
||||
Ok(())
|
||||
}
|
||||
39
__network/s2n_quic/src/bin/server.rs
Normal file
39
__network/s2n_quic/src/bin/server.rs
Normal file
@@ -0,0 +1,39 @@
|
||||
use s2n_quic::Server;
|
||||
use std::error::Error;
|
||||
|
||||
/// NOTE: this certificate is to be used for demonstration purposes only!
|
||||
pub static CERT_PEM: &str = include_str!("../../cert.pem");
|
||||
/// NOTE: this certificate is to be used for demonstration purposes only!
|
||||
pub static KEY_PEM: &str = include_str!("../../key.pem");
|
||||
|
||||
#[tokio::main]
|
||||
async fn main() -> Result<(), Box<dyn Error>> {
|
||||
let mut server = Server::builder()
|
||||
.with_tls((CERT_PEM, KEY_PEM))?
|
||||
.with_io("127.0.0.1:4433")?
|
||||
.start()?;
|
||||
|
||||
while let Some(mut connection) = server.accept().await {
|
||||
// spawn a new task for the connection
|
||||
tokio::spawn(async move {
|
||||
eprintln!("Connection accepted from {:?}", connection.remote_addr());
|
||||
|
||||
while let Ok(Some(mut stream)) = connection.accept_bidirectional_stream().await {
|
||||
// spawn a new task for the stream
|
||||
tokio::spawn(async move {
|
||||
eprintln!("Stream opened from {:?}", stream.connection().remote_addr());
|
||||
|
||||
// echo any data back to the stream
|
||||
while let Ok(Some(data)) = stream.receive().await {
|
||||
stream.send(data).await.expect("stream should be open");
|
||||
}
|
||||
|
||||
println!("Stream closed from {:?}", stream.connection().remote_addr());
|
||||
});
|
||||
}
|
||||
println!("Connection closed from {:?}", connection.remote_addr());
|
||||
});
|
||||
}
|
||||
|
||||
Ok(())
|
||||
}
|
||||
Reference in New Issue
Block a user