compitable previous public ip setting (netsec-ethz#76)

cneukom · Jul 12, 2019 · c46bdf8 · c46bdf8
1 parent 2ddf435
commit c46bdf8
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/conn.go b/conn.go
@@ -58,10 +58,13 @@ func (conn *Conn) PublicIp() string {
 }
 
 func (conn *Conn) passiveListenIP() string {
+	var listenIP string
 	if len(conn.PublicIp()) > 0 {
-		return conn.PublicIp()
+		listenIP = conn.PublicIp()
+	} else {
+		listenIP = conn.conn.LocalAddr().String()
 	}
-	listenIP := conn.conn.LocalAddr().String()
+
 	lastIdx := strings.LastIndex(listenIP, ":")
 	if lastIdx <= 0 {
 		return listenIP