mirror of https://github.com/Dreamacro/clash.git
58 lines
1.0 KiB
Go
58 lines
1.0 KiB
Go
package rules
|
|
|
|
import (
|
|
"github.com/Dreamacro/clash/component/ipset"
|
|
C "github.com/Dreamacro/clash/constant"
|
|
"github.com/Dreamacro/clash/log"
|
|
)
|
|
|
|
// Implements C.Rule
|
|
var _ C.Rule = (*IPSet)(nil)
|
|
|
|
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
|
|
}
|