diff --git a/src/server.rs b/src/server.rs index 05744a7..f20c17e 100755 --- a/src/server.rs +++ b/src/server.rs @@ -119,8 +119,9 @@ impl FlowgateServer { debug!("{} open connection", addr); let mut stream = StreamOwned::new(connection, stream); - let Ok(_) = stream.conn.complete_io(&mut stream.sock) else { debug!("{} close connection", addr); return }; - + while stream.conn.is_handshaking() { + let Ok(_) = stream.conn.complete_io(&mut stream.sock) else { debug!("{} close connection", addr);return }; + } local_self.accept_stream( &mut stream, addr,