Fixed infinite loop

Signed-off-by: Vishal Rana <vr@labstack.com>

Author: vishr

.gitignore

@@ -1 +1,2 @@
 dist
+.idea

Makefile

@@ -1,4 +1,4 @@
-VERSION = 0.2.5
+VERSION = 0.2.6
 
 publish:
 	git tag $(VERSION)

cmd/version.go

@@ -12,7 +12,7 @@ var (
 		Short: "Print the version of Tunnel",
 		// Long:  ``,
 		Run: func(cmd *cobra.Command, args []string) {
-			fmt.Println("0.2.4")
+			fmt.Println("0.2.6")
 		},
 	}
 )

tunnel.go

@@ -114,23 +114,26 @@ func Create(c *Config) {
 	br := bufio.NewReader(r)
 	go func() {
 		for {
-			line, _, _ := br.ReadLine()
+			line, _, err := br.ReadLine()
+			if err != nil {
+				log.Fatalf("Failed to read: %v", err)
+			}
 			fmt.Printf("%s\n", line)
 		}
 	}()
 
 	// Remote listener
 	ln, err := client.Listen("tcp", fmt.Sprintf("%s:%d", c.RemoteHost, c.RemotePort))
 	if err != nil {
-		log.Fatalf("Failed to listen on remote host %v", err)
+		log.Fatalf("Failed to listen on remote host: %v", err)
 	}
 	defer ln.Close()
 
 	for {
 		// Handle inbound connection
 		in, err := ln.Accept()
 		if err != nil {
-			log.Printf("Failed to accept connection %v", err)
+			log.Printf("Failed to accept connection: %v", err)
 			return
 		}
 
@@ -140,7 +143,7 @@ func Create(c *Config) {
 			// Target connection
 			out, err := net.Dial("tcp", fmt.Sprintf("%s:%d", c.TargetHost, c.TargetPort))
 			if err != nil {
-				log.Printf("Failed to connect to target %v", err)
+				log.Printf("Failed to connect to target: %v", err)
 				return
 			}
 			defer out.Close()
@@ -157,7 +160,7 @@ func Create(c *Config) {
 			// Handle error
 			err = <-errCh
 			if err != nil && err != io.EOF {
-				log.Printf("Failed to copy %v", err)
+				log.Printf("Failed to copy: %v", err)
 			}
 		}(in)
 	}