mirror of https://github.com/stjepang/smol
49 lines
1.4 KiB
Rust
49 lines
1.4 KiB
Rust
//! A TCP client secured by TLS based on `async-native-tls`.
|
|
//!
|
|
//! First start a server:
|
|
//!
|
|
//! ```
|
|
//! cargo run --example tls-server
|
|
//! ```
|
|
//!
|
|
//! Then start a client:
|
|
//!
|
|
//! ```
|
|
//! cargo run --example tls-client
|
|
//! ```
|
|
|
|
use std::net::TcpStream;
|
|
|
|
use anyhow::Result;
|
|
use async_native_tls::{Certificate, TlsConnector};
|
|
use smol::{future, io, Async, Unblock};
|
|
|
|
fn main() -> Result<()> {
|
|
// Initialize TLS with the local certificate.
|
|
let mut builder = native_tls::TlsConnector::builder();
|
|
builder.add_root_certificate(Certificate::from_pem(include_bytes!("certificate.pem"))?);
|
|
let tls = TlsConnector::from(builder);
|
|
|
|
smol::block_on(async {
|
|
// Create async stdin and stdout handles.
|
|
let stdin = Unblock::new(std::io::stdin());
|
|
let mut stdout = Unblock::new(std::io::stdout());
|
|
|
|
// Connect to the server.
|
|
let stream = Async::<TcpStream>::connect(([127, 0, 0, 1], 7001)).await?;
|
|
let stream = tls.connect("127.0.0.1", stream).await?;
|
|
println!("Connected to {}", stream.get_ref().get_ref().peer_addr()?);
|
|
println!("Type a message and hit enter!\n");
|
|
|
|
// Pipe messages from stdin to the server and pipe messages from the server to stdout.
|
|
let stream = async_dup::Mutex::new(stream);
|
|
future::try_zip(
|
|
io::copy(stdin, &mut &stream),
|
|
io::copy(&stream, &mut stdout),
|
|
)
|
|
.await?;
|
|
|
|
Ok(())
|
|
})
|
|
}
|