transport Dial now returns an error

1 files changed, 6 insertions, 2 deletions

diff --git a/shapeshifter.go b/shapeshifter.go
index 8fc49dd..d17873c 100644
--- a/shapeshifter.go
+++ b/shapeshifter.go
@@ -54,14 +54,18 @@ func (ss ShapeShifter) clientHandler(conn net.Conn) {
 	defer conn.Close()
 
 	transport := obfs4.NewObfs4Client(ss.Cert, ss.IatMode)
-	remote := transport.Dial(ss.Target)
+	remote, err := transport.Dial(ss.Target)
+	if err != nil {
+		log.Printf("outgoing connection failed %s: %v", ss.Target, err)
+		return
+	}
 	if remote == nil {
 		log.Printf("outgoing connection failed %s", ss.Target)
 		return
 	}
 	defer remote.Close()
 
-	err := copyLoop(conn, remote)
+	err = copyLoop(conn, remote)
 	if err != nil {
 		log.Printf("%s - closed connection: %v", ss.Target, err)
 	} else {