From 2acc174a05ac442cd4fe204781781a89ce01a31e Mon Sep 17 00:00:00 2001 From: Andy Wang Date: Sun, 29 Dec 2019 19:47:17 +0000 Subject: [PATCH] Update goWeb with new state fields --- cmd/ck-server/ck-server.go | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/cmd/ck-server/ck-server.go b/cmd/ck-server/ck-server.go index 7a64e9d..8014629 100644 --- a/cmd/ck-server/ck-server.go +++ b/cmd/ck-server/ck-server.go @@ -40,8 +40,11 @@ func dispatchConnection(conn net.Conn, sta *server.State) { data := buf[:i] goWeb := func() { - _, remotePort, _ := net.SplitHostPort(conn.LocalAddr().String()) - webConn, err := net.Dial("tcp", net.JoinHostPort(sta.RedirAddr.String(), remotePort)) + redirPort := sta.RedirPort + if redirPort == "" { + _, redirPort, _ = net.SplitHostPort(conn.LocalAddr().String()) + } + webConn, err := net.Dial("tcp", net.JoinHostPort(sta.RedirHost.String(), redirPort)) if err != nil { log.Errorf("Making connection to redirection server: %v", err) return