fix: Tunnel doesn't stop on connection errors

limanmys · Nov 2, 2023 · 4149f45 · 4149f45
1 parent f0b43db
commit 4149f45
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/internal/bridge/ssh_tunnel.go b/internal/bridge/ssh_tunnel.go
@@ -268,6 +268,7 @@ func (t *Tunnel) dialTunnel(ctx context.Context, wg *sync.WaitGroup, client *ssh
 		defer cancel()
 		if _, err := io.Copy(cn1, cn2); err != nil {
 			once.Do(func() {
+				t.Stop()
 				t.errHandler()
 				t.log.Errorw("connection error", "details", fmt.Sprintf("%v, %v", t, err))
 			})
@@ -278,6 +279,7 @@ func (t *Tunnel) dialTunnel(ctx context.Context, wg *sync.WaitGroup, client *ssh
 		defer cancel()
 		if _, err := io.Copy(cn2, cn1); err != nil {
 			once.Do(func() {
+				t.Stop()
 				t.errHandler()
 				t.log.Errorw("connection error", "details", fmt.Sprintf("%v, %v", t, err))
 			})