1 use crate::tls::TlsConnectionMetadata;
4 use async_std::pin::Pin;
5 use async_std::prelude::*;
7 use async_tls::TlsAcceptor;
8 use futures::task::{Context, Poll};
12 pub struct TlsServer {
13 local_addrs: SocketAddr,
14 listener: TcpListener,
15 acceptor: TlsAcceptor,
19 pub fn new<A: ToSocketAddrs + std::fmt::Display>(
21 acceptor: TlsAcceptor,
22 ) -> anyhow::Result<Self> {
23 let listener = task::block_on(TcpListener::bind(ip_addrs))?;
24 info!("Started TLS server at {}", listener.local_addr()?);
27 local_addrs: listener.local_addr()?,
34 impl Stream for TlsServer {
35 type Item = Connection;
37 fn poll_next(self: Pin<&mut Self>, _cx: &mut Context<'_>) -> Poll<Option<Self::Item>> {
38 if let Some(Ok(tcp_stream)) = futures::executor::block_on(self.listener.incoming().next()) {
39 let local_addr = tcp_stream
41 .expect("Local address could not be retrieved");
43 let peer_addr = tcp_stream
45 .expect("Peer address could not be retrieved");
46 debug!("Received connection attempt from {}", peer_addr);
48 if let Ok(tls_stream) = futures::executor::block_on(self.acceptor.accept(tcp_stream)) {
49 debug!("Completed TLS handshake with {}", peer_addr);
50 Poll::Ready(Some(Connection::from(TlsConnectionMetadata::Server {
56 debug!("Could not encrypt connection with TLS from {}", peer_addr);
60 debug!("Shutting TLS server down at {}", self.local_addrs);