From ee040b94c97f27e7b632a52637e9bad16b74f069 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Erik=20Hedenstr=C3=B6m?= <erik@hedenstroem.com>
Date: Fri, 4 Feb 2022 01:06:14 +0100
Subject: [PATCH] Made sure program exits if ssh fails
---
ssh/tunnel.go | 9 +++++----
1 file changed, 5 insertions(+), 4 deletions(-)
diff --git a/ssh/tunnel.go b/ssh/tunnel.go
index f5f363f..6f2e02e 100644
--- a/ssh/tunnel.go
+++ b/ssh/tunnel.go
@@ -4,6 +4,7 @@ import (
"io"
"log"
"net"
+ "os"
"golang.org/x/crypto/ssh"
)
@@ -44,14 +45,14 @@ func (tunnel *SSHTunnel) Start() error {
func (tunnel *SSHTunnel) forward(localConn net.Conn) {
serverConn, err := ssh.Dial("tcp", tunnel.Server.String(), tunnel.Config)
if err != nil {
- tunnel.logf("server dial error: %s", err)
- return
+ tunnel.logf("server dial %s error: %s", tunnel.Server, err)
+ os.Exit(-1)
}
tunnel.logf("connected to %s (1 of 2)\n", tunnel.Server.String())
remoteConn, err := serverConn.Dial("tcp", tunnel.Remote.String())
if err != nil {
- tunnel.logf("remote dial error: %s", err)
- return
+ tunnel.logf("remote dial %s error: %s", tunnel.Remote, err)
+ os.Exit(-1)
}
tunnel.logf("connected to %s (2 of 2)\n", tunnel.Remote.String())
copyConn := func(writer, reader net.Conn) {
--
GitLab