48 lines
778 B
Go
48 lines
778 B
Go
package rules
|
|
|
|
import (
|
|
"net"
|
|
|
|
C "github.com/Dreamacro/clash/constant"
|
|
)
|
|
|
|
type IPCIDR struct {
|
|
ipnet *net.IPNet
|
|
adapter string
|
|
isSourceIP bool
|
|
}
|
|
|
|
func (i *IPCIDR) RuleType() C.RuleType {
|
|
if i.isSourceIP {
|
|
return C.SrcIPCIDR
|
|
}
|
|
return C.IPCIDR
|
|
}
|
|
|
|
func (i *IPCIDR) IsMatch(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 NewIPCIDR(s string, adapter string, isSourceIP bool) *IPCIDR {
|
|
_, ipnet, err := net.ParseCIDR(s)
|
|
if err != nil {
|
|
return nil
|
|
}
|
|
return &IPCIDR{
|
|
ipnet: ipnet,
|
|
adapter: adapter,
|
|
isSourceIP: isSourceIP,
|
|
}
|
|
}
|