Reuse remote connection

yosebyte · Dec 3, 2024 · 61cc49c · 61cc49c
1 parent e481073
commit 61cc49c
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 3 deletions.
diff --git a/internal/tunnel/client.go b/internal/tunnel/client.go
@@ -44,8 +44,15 @@ func Client(parsedURL *url.URL) error {
 				}
 				defer targetConn.Close()
 				targetConn.SetNoDelay(true)
-				log.Info("Target connection established, starting data exchange: [%v] <-> [%v]", linkAddr, targetAddr)
-				util.HandleConn(linkConn, targetConn)
+				remoteConn, err := net.DialTCP("tcp", nil, linkAddr)
+				if err != nil {
+					log.Error("Unable to dial target address: [%v], %v", linkAddr, err)
+					return
+				}
+				defer remoteConn.Close()
+				remoteConn.SetNoDelay(true)
+				log.Info("Target connection established, starting data exchange: [%v] <-> [%v]", remoteConn, targetAddr)
+				util.HandleConn(remoteConn, targetConn)
 				log.Info("Connection closed successfully")
 			}()
 		}

diff --git a/internal/tunnel/server.go b/internal/tunnel/server.go
@@ -72,8 +72,14 @@ func Server(parsedURL *url.URL, whiteList *sync.Map) error {
 				targetConn.Close()
 				return
 			}
+			remoteConn, err := linkListen.AcceptTCP()
+			if err != nil {
+				log.Error("Unable to accept connections form link address: [%v] %v", linkAddr, err)
+				return
+			}
+			defer remoteConn.Close()
 			log.Info("Starting data exchange: [%v] <-> [%v]", clientAddr, targetAddr)
-			util.HandleConn(linkConn, targetConn)
+			util.HandleConn(remoteConn, targetConn)
 			log.Info("Connection closed successfully")
 		}(targetConn)
 	}