[!] add support for IPv6 (#225)

zhongyibill · pashagolub · web-flow · commit 4869dca373e4 · 2024-05-10T12:59:23.000+02:00
Co-authored-by: Pavlo Golub &lt;pavlo.golub@gmail.com&gt;
diff --git a/main.go b/main.go
@@ -24,11 +24,14 @@ var (
 )
 
 func getMask(vip netip.Addr, mask int) net.IPMask {
-	if mask > 0 || mask < 33 {
-		return net.CIDRMask(mask, 32)
+	if vip.Is4() { //IPv4
+		if mask > 0 && mask < 33 {
+			return net.CIDRMask(mask, 32)
+		}
+		var ip net.IP = vip.AsSlice()
+		return ip.DefaultMask()
 	}
-	var ip net.IP = vip.AsSlice()
-	return ip.DefaultMask()
+	return net.CIDRMask(mask, 128) //IPv6
 }
 
 func getNetIface(iface string) *net.Interface {

Original file line number	Diff line number	Diff line change
`@@ -24,11 +24,14 @@ var (`
`24`	`24`	`)`
`25`	`25`
`26`	`26`	`func getMask(vip netip.Addr, mask int) net.IPMask {`
`27`		`- if mask > 0 \|\| mask < 33 {`
`28`		`- return net.CIDRMask(mask, 32)`
	`27`	`+ if vip.Is4() { //IPv4`
	`28`	`+ if mask > 0 && mask < 33 {`
	`29`	`+ return net.CIDRMask(mask, 32)`
	`30`	`+ }`
	`31`	`+ var ip net.IP = vip.AsSlice()`
	`32`	`+ return ip.DefaultMask()`
`29`	`33`	`}`
`30`		`- var ip net.IP = vip.AsSlice()`
`31`		`- return ip.DefaultMask()`
	`34`	`+ return net.CIDRMask(mask, 128) //IPv6`
`32`	`35`	`}`
`33`	`36`
`34`	`37`	`func getNetIface(iface string) *net.Interface {`