mirror of https://github.com/Dreamacro/clash.git
55 lines
986 B
Go
55 lines
986 B
Go
|
package rules
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/Dreamacro/clash/component/ipset"
|
||
|
|
C "github.com/Dreamacro/clash/constant"
|
||
|
|
"github.com/Dreamacro/clash/log"
|
||
|
|
)
|
||
|
|
|
||
|
|
type IPSet struct {
|
||
|
|
name string
|
||
|
|
adapter string
|
||
|
|
noResolveIP bool
|
||
|
|
}
|
||
|
|
|
||
|
|
func (f *IPSet) RuleType() C.RuleType {
|
||
|
|
return C.IPSet
|
||
|
|
}
|
||
|
|
|
||
|
|
func (f *IPSet) Match(metadata *C.Metadata) bool {
|
||
|
|
exist, err := ipset.Test(f.name, metadata.DstIP)
|
||
|
|
if err != nil {
|
||
|
|
log.Warnln("check ipset '%s' failed: %s", f.name, err.Error())
|
||
|
|
return false
|
||
|
|
}
|
||
|
|
return exist
|
||
|
|
}
|
||
|
|
|
||
|
|
func (f *IPSet) Adapter() string {
|
||
|
|
return f.adapter
|
||
|
|
}
|
||
|
|
|
||
|
|
func (f *IPSet) Payload() string {
|
||
|
|
return f.name
|
||
|
|
}
|
||
|
|
|
||
|
|
func (f *IPSet) ShouldResolveIP() bool {
|
||
|
|
return !f.noResolveIP
|
||
|
|
}
|
||
|
|
|
||
|
|
func (f *IPSet) ShouldFindProcess() bool {
|
||
|
|
return false
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewIPSet(name string, adapter string, noResolveIP bool) (*IPSet, error) {
|
||
|
|
if err := ipset.Verify(name); err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
return &IPSet{
|
||
|
|
name: name,
|
||
|
|
adapter: adapter,
|
||
|
|
noResolveIP: noResolveIP,
|
||
|
|
}, nil
|
||
|
|
}
|