remove port trimming from getLocalAddr

CorentinB · Sep 26, 2024 · 7d0e9b1 · 7d0e9b1
1 parent b09abca
commit 7d0e9b1
Showing 1 changed file with 2 additions and 9 deletions.
diff --git a/random_local_ip.go b/random_local_ip.go
@@ -125,16 +125,9 @@ func getNextIP(availableIPs *availableIPs) net.IP {
 }
 
 func getLocalAddr(network, IP string) any {
-	lastColon := strings.LastIndex(IP, ":")
-	if lastColon == -1 {
-		return nil
-	}
-
-	ip := IP[:lastColon]
-	ip = strings.TrimPrefix(ip, "[")
-	ip = strings.TrimSuffix(ip, "]")
+	IP = strings.Trim(IP, "[]")
 
-	destIP := net.ParseIP(ip)
+	destIP := net.ParseIP(IP)
 	if destIP == nil {
 		return nil
 	}