Bumped v0.5.2

Signed-off-by: Vishal Rana <[email protected]>

Author: vishr

Makefile

@@ -1,5 +1,5 @@
 IMAGE = labstack/tunnel
-VERSION = 0.5.1
+VERSION = 0.5.2
 
 publish:
 	git tag v$(VERSION)

cmd/root.go

@@ -4,8 +4,10 @@ import (
   "errors"
   "fmt"
   "github.com/labstack/tunnel-client/daemon"
+  "net"
   "os"
   "path/filepath"
+  "strings"
 
   "github.com/spf13/cobra"
 
@@ -34,9 +36,14 @@ var rootCmd = &cobra.Command{
     defer c.Close()
     rep := new(daemon.ConnectReply)
     s.Start()
+    addr := args[0]
+    _, _, err = net.SplitHostPort(addr)
+    if err != nil && strings.Contains(err.Error(), "missing port") {
+      addr = ":" + addr
+    }
     err = c.Call("Server.Connect", &daemon.ConnectRequest{
       Name:     name,
-      Address:  args[0],
+      Address:  addr,
       Protocol: daemon.Protocol(protocol),
     }, rep)
     if err != nil {

daemon/connection.go

@@ -214,7 +214,8 @@ RECONNECT:
   // Remote listener
   l, err := sc.Listen("tcp", fmt.Sprintf("%s:%d", c.RemoteHost, c.RemotePort))
   if err != nil {
-    log.Fatalf("failed to listen on remote host: %v", err)
+    log.Errorf("failed to listen on remote host: %v", err)
+    return
   }
   // Note: Don't close the listener as it prevents closing the underlying connection