mirror of
https://github.com/MetaCubeX/mihomo.git
synced 2024-09-12 11:52:33 +00:00
# Conflicts: # .github/workflows/codeql-analysis.yml # .github/workflows/docker.yml # .github/workflows/linter.yml # .github/workflows/stale.yml # Makefile # component/dialer/dialer.go # config/config.go # constant/metadata.go # constant/rule.go # rule/common/domain.go # rule/common/domain_keyword.go # rule/common/domain_suffix.go # rule/common/final.go # rule/common/ipcidr.go # rule/geoip.go # rule/parser.go # rule/port.go # rule/process.go
83 lines
1.3 KiB
Go
83 lines
1.3 KiB
Go
package common
|
|
|
|
import (
|
|
"net"
|
|
|
|
C "github.com/Dreamacro/clash/constant"
|
|
)
|
|
|
|
type IPCIDROption func(*IPCIDR)
|
|
|
|
func WithIPCIDRSourceIP(b bool) IPCIDROption {
|
|
return func(i *IPCIDR) {
|
|
i.isSourceIP = b
|
|
}
|
|
}
|
|
|
|
func WithIPCIDRNoResolve(noResolve bool) IPCIDROption {
|
|
return func(i *IPCIDR) {
|
|
i.noResolveIP = noResolve
|
|
}
|
|
}
|
|
|
|
type IPCIDR struct {
|
|
ipnet *net.IPNet
|
|
adapter string
|
|
ruleExtra *C.RuleExtra
|
|
isSourceIP bool
|
|
noResolveIP bool
|
|
}
|
|
|
|
func (i *IPCIDR) RuleType() C.RuleType {
|
|
if i.isSourceIP {
|
|
return C.SrcIPCIDR
|
|
}
|
|
return C.IPCIDR
|
|
}
|
|
|
|
func (i *IPCIDR) Match(metadata *C.Metadata) bool {
|
|
ip := metadata.DstIP
|
|
if i.isSourceIP {
|
|
ip = metadata.SrcIP
|
|
}
|
|
return ip != nil && i.ipnet.Contains(ip)
|
|
}
|
|
|
|
func (i *IPCIDR) Adapter() string {
|
|
return i.adapter
|
|
}
|
|
|
|
func (i *IPCIDR) Payload() string {
|
|
return i.ipnet.String()
|
|
}
|
|
|
|
func (i *IPCIDR) ShouldResolveIP() bool {
|
|
return !i.noResolveIP
|
|
}
|
|
|
|
func (i *IPCIDR) ShouldFindProcess() bool {
|
|
return false
|
|
}
|
|
|
|
func (i *IPCIDR) RuleExtra() *C.RuleExtra {
|
|
return i.ruleExtra
|
|
}
|
|
|
|
func NewIPCIDR(s string, adapter string, ruleExtra *C.RuleExtra, opts ...IPCIDROption) (*IPCIDR, error) {
|
|
_, ipnet, err := net.ParseCIDR(s)
|
|
if err != nil {
|
|
return nil, errPayload
|
|
}
|
|
|
|
ipcidr := &IPCIDR{
|
|
ipnet: ipnet,
|
|
adapter: adapter,
|
|
ruleExtra: ruleExtra,
|
|
}
|
|
|
|
for _, o := range opts {
|
|
o(ipcidr)
|
|
}
|
|
|
|
return ipcidr, nil
|
|
}
|