From e19f5475296d839b4447d23b856fe94d3919fab3 Mon Sep 17 00:00:00 2001
From: James Phillips <slackpad@gmail.com>
Date: Mon, 25 Sep 2017 17:42:53 -0700
Subject: [PATCH] Makes porter more conservative by trying to connect to ports
 before handing them out.

---
 test/porter/cmd/porter/main.go | 40 ++++++++++++++++++++++++----------
 1 file changed, 28 insertions(+), 12 deletions(-)

diff --git a/test/porter/cmd/porter/main.go b/test/porter/cmd/porter/main.go
index f7d2069d57..1b98fe7d03 100644
--- a/test/porter/cmd/porter/main.go
+++ b/test/porter/cmd/porter/main.go
@@ -6,6 +6,7 @@ import (
 	"fmt"
 	"io/ioutil"
 	"log"
+	"net"
 	"net/http"
 	"os"
 	"os/exec"
@@ -97,7 +98,6 @@ func run(args []string) error {
 	return cmd.Run()
 }
 
-// todo(fs): check which ports are currently bound and exclude them
 func servePort(w http.ResponseWriter, r *http.Request) {
 	var count int
 	n, err := strconv.Atoi(r.RequestURI[1:])
@@ -108,26 +108,42 @@ func servePort(w http.ResponseWriter, r *http.Request) {
 		count = 1
 	}
 
-	mu.Lock()
-	if port < firstPort {
-		port = firstPort
+	// getPort assumes the lock is already held and tries to return a port
+	// that's not in use. It will panic if it has to try too many times.
+	getPort := func() int {
+		for i := 0; i < 10; i++ {
+			port++
+			if port < firstPort {
+				port = firstPort
+			}
+			if port >= lastPort {
+				port = firstPort
+			}
+
+			conn, err := net.Dial("tcp", fmt.Sprintf("127.0.0.1:%d", port))
+			if err != nil {
+				return port
+			}
+			conn.Close()
+			if verbose {
+				log.Printf("porter: skipping port %d, already in use", port)
+			}
+		}
+		panic(fmt.Errorf("could not find a free port"))
 	}
-	if port+count >= lastPort {
-		port = firstPort
-	}
-	from, to := port, port+count
-	port = to
-	mu.Unlock()
 
 	p := make([]int, count)
+	mu.Lock()
 	for i := 0; i < count; i++ {
-		p[i] = from + i
+		p[i] = getPort()
 	}
+	mu.Unlock()
+
 	if err := json.NewEncoder(w).Encode(p); err != nil {
 		// this shouldn't happen so we panic since we can't recover
 		panic(err)
 	}
 	if verbose {
-		log.Printf("porter: allocated ports %d-%d (%d)", from, to, count)
+		log.Printf("porter: allocated ports %v", p)
 	}
 }