Merge branch 'bug/resolve-addr' @Release

sower-proxy · Feb 16, 2019 · e8c6d84 · e8c6d84
2 parents 4883e24 + 092e3cf
commit e8c6d84
Showing 1 changed file with 9 additions and 5 deletions.
diff --git a/proxy/client.go b/proxy/client.go
@@ -31,17 +31,21 @@ func NewClient(netType string) Client {
 func StartClient(netType, server, cipher, password, listenIP string) {
 	connCh := listenLocal(listenIP, []string{"80", "443"})
 	client := NewClient(netType)
-	if addr, err := net.ResolveTCPAddr("tcp", server); err != nil {
-		glog.Fatalln(err)
-	} else {
-		server = addr.String()
-	}
+	resolved := false
 
 	glog.Infoln("Client started.")
 	for {
 		conn := <-connCh
 		glog.V(1).Infof("new conn from (%s) to (%s)", conn.RemoteAddr(), server)
 
+		if !resolved {
+			if addr, err := net.ResolveTCPAddr("tcp", server); err != nil {
+				glog.Fatalln(err)
+			} else {
+				server = addr.String()
+			}
+		}
+
 		rc, err := client.Dial(server)
 		if err != nil {
 			conn.Close()