fix TunnelIn

neevek · Nov 20, 2023 · 21ee685 · 21ee685
1 parent cd473f8
commit 21ee685
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/src/access_server.rs b/src/access_server.rs
@@ -107,6 +107,7 @@ impl AccessServer {
         drop(tcp_receiver);
 
         // initiate a new connection to wake up the accept() loop
+        self.set_drop_conn(false);
         TcpStream::connect(self.addr).await?;
         Ok(())
     }

diff --git a/src/server.rs b/src/server.rs
@@ -339,6 +339,7 @@ impl Server {
             }
         });
 
+        access_server.set_drop_conn(false);
         let mut tcp_receiver = access_server.take_tcp_receiver();
         while let Some(Some(ChannelMessage::Request(tcp_stream))) = tcp_receiver.recv().await {
             match client_conn.open_bi().await {
@@ -363,7 +364,7 @@ impl Server {
 
         access_server.shutdown(tcp_receiver).await.ok();
 
-        info!("will quit access server: {}", addr);
+        info!("access server quit: {}", addr);
 
         Ok(())
     }