fix index out of range err (#286)

* fix index out of range err * minor * fix tests
projectdiscovery · Nov 23, 2023 · 26cc775 · 26cc775
1 parent 3ae3e72
commit 26cc775
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/ip.go b/ip.go
@@ -285,8 +285,11 @@ func ipNetToRange(ipNet net.IPNet) netWithRange {
 	lastIPMask := make(net.IPMask, len(ipNet.Mask))
 	copy(lastIPMask, ipNet.Mask)
 	for i := range lastIPMask {
-		lastIPMask[len(lastIPMask)-i-1] = ^lastIPMask[len(lastIPMask)-i-1]
-		lastIP[net.IPv6len-i-1] |= lastIPMask[len(lastIPMask)-i-1]
+		index := len(lastIPMask) - 1 - i
+		lastIPMask[index] = ^lastIPMask[index]
+		if net.IPv6len-i-1 < len(lastIP) {
+			lastIP[net.IPv6len-i-1] |= lastIPMask[index]
+		}
 	}
 
 	return netWithRange{First: &firstIP, Last: &lastIP, Network: &ipNet}