default to port 9009 if port is lost during discovery

fixes #222
schollz · Jul 15, 2020 · 7fd8fda · 7fd8fda
1 parent 1255a79
commit 7fd8fda
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/croc/croc.go b/src/croc/croc.go
@@ -434,9 +434,13 @@ func (c *Client) Receive() (err error) {
 					continue
 				}
 				log.Debug("switching to local")
+				portToUse := string(bytes.TrimPrefix(discoveries[0].Payload, []byte("croc")))
+				if portToUse == "" {
+					portToUse = "9009"
+				}
 				c.Options.RelayAddress = fmt.Sprintf("%s:%s",
 					discoveries[0].Address,
-					bytes.TrimPrefix(discoveries[0].Payload, []byte("croc")),
+					portToUse,
 				)
 				c.ExternalIPConnected = c.Options.RelayAddress
 				usingLocal = true