fix infinity loop

Jen94 · Aug 31, 2021 · dc8a624 · dc8a624
1 parent 147aa3c
commit dc8a624
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/pkg/utils/util.go b/pkg/utils/util.go
@@ -70,7 +70,12 @@ func ListenUDPInPortRange(portMin, portMax int, laddr *net.UDPAddr) (*net.UDPCon
 	}
 	portStart := rand.Intn(j-i+1) + i
 	portCurrent := portStart
+	tryCount := 10
 	for {
+		tryCount -= 1
+		if tryCount < 0 {
+			return nil, ErrPort
+		}
 		*laddr = net.UDPAddr{IP: laddr.IP, Port: portCurrent}
 		c, e := net.ListenUDP("udp", laddr)
 		if e == nil {