Handle remote failure (#19)

Similar to changes in #18 - closes open connections immediately on remote dial failure.
elliotchance · Oct 16, 2023 · d87a21f · d87a21f
1 parent 93cd1e6
commit d87a21f
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/ssh_tunnel.go b/ssh_tunnel.go
@@ -147,6 +147,13 @@ func (tunnel *SSHTunnel) forward(localConn net.Conn) {
 	remoteConn, err := serverConn.Dial("tcp", tunnel.Remote.String())
 	if err != nil {
 		tunnel.logf("remote dial error: %s", err)
+
+		if err := serverConn.Close(); err != nil {
+			tunnel.logf("failed to close server connection: %v", err)
+		}
+		if err := localConn.Close(); err != nil {
+			tunnel.logf("failed to close local connection: %v", err)
+		}
 		return
 	}
 	tunnel.Conns = append(tunnel.Conns, remoteConn)