From ce93cd84160a5a3f6a4deb5bd2d1b83b7fdc2a09 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Tom=C3=A1s=20Senart?= <tsenart@gmail.com>
Date: Sat, 27 Jul 2024 20:09:00 +0200
Subject: [PATCH] lib: use fixed IP selection function

Last release introduced a fixed firstOfEachIPFamily function, but didn't
integrate it.
---
 lib/attack.go | 15 +--------------
 1 file changed, 1 insertion(+), 14 deletions(-)

diff --git a/lib/attack.go b/lib/attack.go
index 8467b943..0e43e91f 100644
--- a/lib/attack.go
+++ b/lib/attack.go
@@ -329,20 +329,7 @@ func DNSCaching(ttl time.Duration) func(*Attacker) {
 
 				rng.Shuffle(len(ips), func(i, j int) { ips[i], ips[j] = ips[j], ips[i] })
 
-				// In place filtering of ips to only include the first IPv4 and IPv6.
-				j := 0
-				for i := 0; i < len(ips) && j < 2; i++ {
-					ip := net.ParseIP(ips[i])
-					switch {
-					case len(ip) == net.IPv4len && (j == 0 || len(ips[j-1]) == net.IPv6len):
-						fallthrough
-					case len(ip) == net.IPv6len && (j == 0 || len(ips[j-1]) == net.IPv4len):
-						ips[j] = ips[i]
-						j++
-					}
-				}
-
-				ips = ips[:j]
+				ips = firstOfEachIPFamily(ips)
 
 				type result struct {
 					conn net.Conn