1 use async_tls::TlsConnector;
5 use async_std::net::{SocketAddr, TcpStream, ToSocketAddrs};
8 use futures::AsyncReadExt;
10 pub enum TlsConnectionMetadata {
12 local_addr: SocketAddr,
13 peer_addr: SocketAddr,
14 stream: client::TlsStream<TcpStream>,
17 local_addr: SocketAddr,
18 peer_addr: SocketAddr,
19 stream: server::TlsStream<TcpStream>,
24 pub async fn tls_client<A: ToSocketAddrs + std::fmt::Display>(
27 connector: TlsConnector,
28 ) -> anyhow::Result<Self> {
29 let stream = TcpStream::connect(&ip_addrs).await?;
30 info!("Established client TCP connection to {}", ip_addrs);
31 stream.set_nodelay(true)?;
33 let local_addr = stream.peer_addr()?;
34 let peer_addr = stream.peer_addr()?;
36 let encrypted_stream: client::TlsStream<TcpStream> = connector.connect(domain, stream).await?;
37 info!("Completed TLS handshake with {}", peer_addr);
39 Ok(Self::from(TlsConnectionMetadata::Client {
42 stream: encrypted_stream,
47 impl From<TlsConnectionMetadata> for Connection {
48 fn from(metadata: TlsConnectionMetadata) -> Self {
50 TlsConnectionMetadata::Client {
55 let (read_stream, write_stream) = stream.split();
60 Box::pin(read_stream),
61 Box::pin(write_stream),
65 TlsConnectionMetadata::Server {
70 let (read_stream, write_stream) = stream.split();
75 Box::pin(read_stream),
76 Box::pin(write_stream),