From 588e650d80fd5e7ab15d9b1a0bb035c6213a2823 Mon Sep 17 00:00:00 2001
From: Zack Scholl <zack.scholl@gmail.com>
Date: Wed, 26 Aug 2020 17:46:35 -0700
Subject: [PATCH] don't split host if localhost

---
 src/croc/croc.go | 13 +++++++++----
 1 file changed, 9 insertions(+), 4 deletions(-)

diff --git a/src/croc/croc.go b/src/croc/croc.go
index 38790b27..992168bd 100644
--- a/src/croc/croc.go
+++ b/src/croc/croc.go
@@ -728,10 +728,15 @@ func (c *Client) procesMessagePake(m message.Message) (err error) {
 			log.Debugf("port: [%s]", c.Options.RelayPorts[i])
 			go func(j int) {
 				defer wg.Done()
-				host, _, err := net.SplitHostPort(c.Options.RelayAddress)
-				if err != nil {
-					log.Errorf("bad relay address %s", c.Options.RelayAddress)
-					return
+				var host string
+				if c.Options.RelayAddress == "localhost" {
+					host = c.Options.RelayAddress
+				} else {
+					host, _, err = net.SplitHostPort(c.Options.RelayAddress)
+					if err != nil {
+						log.Errorf("bad relay address %s", c.Options.RelayAddress)
+						return
+					}
 				}
 				server := net.JoinHostPort(host, c.Options.RelayPorts[j])
 				log.Debugf("connecting to %s", server)