diff --git a/src/transport/noise.rs b/src/transport/noise.rs index 4825b44..5eb053b 100644 --- a/src/transport/noise.rs +++ b/src/transport/noise.rs @@ -72,18 +72,27 @@ impl Transport for NoiseTransport { } async fn accept(&self, a: &Self::Acceptor) -> Result<(Self::Stream, SocketAddr)> { - let (conn, addr) = a.accept().await?; + let (conn, addr) = a + .accept() + .await + .with_context(|| "Failed to accept TCP connection")?; set_tcp_keepalive(&conn); - let conn = NoiseStream::handshake(conn, self.builder().build_responder()?).await?; + let conn = NoiseStream::handshake(conn, self.builder().build_responder()?) + .await + .with_context(|| "Failed to do noise handshake")?; Ok((conn, addr)) } async fn connect(&self, addr: &str) -> Result { - let conn = TcpStream::connect(addr).await?; + let conn = TcpStream::connect(addr) + .await + .with_context(|| "Failed to connect TCP socket")?; set_tcp_keepalive(&conn); - let conn = NoiseStream::handshake(conn, self.builder().build_initiator()?).await?; + let conn = NoiseStream::handshake(conn, self.builder().build_initiator()?) + .await + .with_context(|| "Failed to do noise handshake")?; return Ok(conn); } }