From b446b5969f67008795a13b0d786fe17ab4a74cd7 Mon Sep 17 00:00:00 2001
From: Vladimir Avtsenov <vladimir.lsk.cool@gmail.com>
Date: Mon, 21 Oct 2024 00:18:42 +0300
Subject: [PATCH] change ipset type to hash:net

---
 group.go                  | 2 +-
 netfilter-helper/ipset.go | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/group.go b/group.go
index 324c0c2..ff7fd80 100644
--- a/group.go
+++ b/group.go
@@ -22,7 +22,7 @@ type Group struct {
 
 func (g *Group) AddIPv4(address net.IP, ttl time.Duration) error {
 	ttlSeconds := uint32(ttl.Seconds())
-	return g.ipset.Add(address, &ttlSeconds)
+	return g.ipset.AddIP(address, &ttlSeconds)
 }
 
 func (g *Group) DelIPv4(address net.IP) error {
diff --git a/netfilter-helper/ipset.go b/netfilter-helper/ipset.go
index 01e798d..de470d5 100644
--- a/netfilter-helper/ipset.go
+++ b/netfilter-helper/ipset.go
@@ -11,7 +11,7 @@ type IPSet struct {
 	SetName string
 }
 
-func (r *IPSet) Add(addr net.IP, timeout *uint32) error {
+func (r *IPSet) AddIP(addr net.IP, timeout *uint32) error {
 	err := netlink.IpsetAdd(r.SetName, &netlink.IPSetEntry{
 		IP:      addr,
 		Timeout: timeout,
@@ -63,7 +63,7 @@ func (nh *NetfilterHelper) IPSet(name string) (*IPSet, error) {
 	}
 
 	defaultTimeout := uint32(300)
-	err = netlink.IpsetCreate(ipset.SetName, "hash:ip", netlink.IpsetCreateOptions{
+	err = netlink.IpsetCreate(ipset.SetName, "hash:net", netlink.IpsetCreateOptions{
 		Timeout: &defaultTimeout,
 	})
 	if err != nil {