From 4afb0ecc510bad6196ec0621a2805fcaf0ff3159 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Kuba=20Podg=C3=B3rski?= <kuba--@users.noreply.github.com>
Date: Wed, 13 Jan 2021 13:53:27 +0100
Subject: [PATCH] Close TCP listeneer on port mapper

---
 test/port/port_mapper.go | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/test/port/port_mapper.go b/test/port/port_mapper.go
index 3901ff983..94ac32978 100644
--- a/test/port/port_mapper.go
+++ b/test/port/port_mapper.go
@@ -80,6 +80,7 @@ func (pm *globalPortMapper) allocate() {
 		})
 		if err != nil {
 			fmt.Printf("UDP port %v was available on tcp but not udp: %v\n", port, err)
+			lsn.Close()
 		} else {
 			_ = udpConn.Close()
 			if lsn == nil {
@@ -96,10 +97,9 @@ func (pm *globalPortMapper) allocateAtTop() {
 	println("888888 allocateAtTop ports called")
 	pm.availPorts = make([]net.Listener, pm.numPorts)
 	i := 0
-	next := 65000
-	for i < pm.numPorts {
+
+	for next := 65000; i < pm.numPorts && next > 0; next-- {
 		lsn, err := net.Listen("tcp", fmt.Sprintf(":%d", next))
-		next--
 		if err != nil {
 			//fmt.Printf("next=%v, err = %v\n", next+1, err)
 			continue
@@ -115,6 +115,7 @@ func (pm *globalPortMapper) allocateAtTop() {
 		})
 		if err != nil {
 			fmt.Printf("UDP port %v was available on tcp but not udp: %v\n", port, err)
+			lsn.Close()
 		} else {
 			_ = udpConn.Close()
 			if lsn == nil {