summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--light-client/src/network.rs9
1 files changed, 7 insertions, 2 deletions
diff --git a/light-client/src/network.rs b/light-client/src/network.rs
index f2c42518..3eb3f50a 100644
--- a/light-client/src/network.rs
+++ b/light-client/src/network.rs
@@ -59,9 +59,14 @@ impl Network {
});
let stream = TcpStream::connect((host, port))?;
stream.set_nodelay(true).unwrap();
- // stream.set_nonblocking(true).unwrap();
- let (sock, _) = client_tls_with_config(request, stream, None, None).unwrap();
+ let (mut sock, _) = client_tls_with_config(request, stream, None, None).unwrap();
+
+ match sock.get_mut() {
+ MaybeTlsStream::Plain(s) => s.set_nonblocking(true).unwrap(),
+ MaybeTlsStream::Rustls(s) => s.sock.set_nonblocking(true).unwrap(),
+ _ => todo!(),
+ };
Ok(Self {
sock,