ipset_linux.go

package ipset

import "github.com/nadoo/ipset/internal/netlink"

var nl *netlink.NetLink

// Option is used to set parameters of ipset operations.
type Option = netlink.Option

// OptIPv6 sets `family inet6` parameter to operations.
func OptIPv6() Option { return func(opts *netlink.Options) { opts.IPv6 = true } }

// OptTimeout sets `timeout xx` parameter to operations.
func OptTimeout(timeout uint32) Option { return func(opts *netlink.Options) { opts.Timeout = timeout } }

// Init prepares a netlink socket of ipset.
func Init() (err error) {
	nl, err = netlink.New()
	return err
}

// Create creates a new set.
func Create(setName string, opts ...Option) (err error) {
	return nl.CreateSet(setName, opts...)
}

// Destroy destroys a named set.
func Destroy(setName string) (err error) {
	return nl.DestroySet(setName)
}

// Flush flushes a named set.
func Flush(setName string) (err error) {
	return nl.FlushSet(setName)
}

// Add adds an entry to the named set.
// entry could be: "1.1.1.1" or "192.168.1.0/24".
func Add(setName, entry string, opts ...Option) (err error) {
	return nl.AddToSet(setName, entry, opts...)
}

// Del deletes an entry from the named set.
func Del(setName, entry string) (err error) {
	return nl.DelFromSet(setName, entry)
}