mirror of
https://github.com/MetaCubeX/mihomo.git
synced 2024-11-16 11:42:43 +08:00
101 lines
2.0 KiB
Go
101 lines
2.0 KiB
Go
package provider
|
|
|
|
import (
|
|
"errors"
|
|
"io"
|
|
"net/netip"
|
|
|
|
"github.com/metacubex/mihomo/component/cidr"
|
|
C "github.com/metacubex/mihomo/constant"
|
|
P "github.com/metacubex/mihomo/constant/provider"
|
|
"github.com/metacubex/mihomo/log"
|
|
|
|
"go4.org/netipx"
|
|
)
|
|
|
|
type ipcidrStrategy struct {
|
|
count int
|
|
shouldResolveIP bool
|
|
cidrSet *cidr.IpCidrSet
|
|
//trie *trie.IpCidrTrie
|
|
}
|
|
|
|
func (i *ipcidrStrategy) Behavior() P.RuleBehavior {
|
|
return P.IPCIDR
|
|
}
|
|
|
|
func (i *ipcidrStrategy) ShouldFindProcess() bool {
|
|
return false
|
|
}
|
|
|
|
func (i *ipcidrStrategy) Match(metadata *C.Metadata) bool {
|
|
// return i.trie != nil && i.trie.IsContain(metadata.DstIP.AsSlice())
|
|
return i.cidrSet != nil && i.cidrSet.IsContain(metadata.DstIP)
|
|
}
|
|
|
|
func (i *ipcidrStrategy) Count() int {
|
|
return i.count
|
|
}
|
|
|
|
func (i *ipcidrStrategy) ShouldResolveIP() bool {
|
|
return i.shouldResolveIP
|
|
}
|
|
|
|
func (i *ipcidrStrategy) Reset() {
|
|
// i.trie = trie.NewIpCidrTrie()
|
|
i.cidrSet = cidr.NewIpCidrSet()
|
|
i.count = 0
|
|
i.shouldResolveIP = false
|
|
}
|
|
|
|
func (i *ipcidrStrategy) Insert(rule string) {
|
|
//err := i.trie.AddIpCidrForString(rule)
|
|
err := i.cidrSet.AddIpCidrForString(rule)
|
|
if err != nil {
|
|
log.Warnln("invalid Ipcidr:[%s]", rule)
|
|
} else {
|
|
i.shouldResolveIP = true
|
|
i.count++
|
|
}
|
|
}
|
|
|
|
func (i *ipcidrStrategy) FinishInsert() {
|
|
i.cidrSet.Merge()
|
|
}
|
|
|
|
func (i *ipcidrStrategy) FromMrs(r io.Reader, count int) error {
|
|
cidrSet, err := cidr.ReadIpCidrSet(r)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
i.count = count
|
|
i.cidrSet = cidrSet
|
|
if i.count > 0 {
|
|
i.shouldResolveIP = true
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (i *ipcidrStrategy) WriteMrs(w io.Writer) error {
|
|
if i.cidrSet == nil {
|
|
return errors.New("nil cidrSet")
|
|
}
|
|
return i.cidrSet.WriteBin(w)
|
|
}
|
|
|
|
func (i *ipcidrStrategy) DumpMrs(f func(key string) bool) {
|
|
if i.cidrSet != nil {
|
|
i.cidrSet.Foreach(func(prefix netip.Prefix) bool {
|
|
return f(prefix.String())
|
|
})
|
|
}
|
|
}
|
|
|
|
func (i *ipcidrStrategy) ToIpCidr() *netipx.IPSet {
|
|
return i.cidrSet.ToIPSet()
|
|
}
|
|
|
|
func NewIPCidrStrategy() *ipcidrStrategy {
|
|
return &ipcidrStrategy{}
|
|
}
|