mirror of https://github.com/v2ray/v2ray-core
120 lines
2.2 KiB
Go
120 lines
2.2 KiB
Go
package router
|
|
|
|
import (
|
|
"context"
|
|
)
|
|
|
|
// CIDRList is an alias of []*CIDR to provide sort.Interface.
|
|
type CIDRList []*CIDR
|
|
|
|
// Len implements sort.Interface.
|
|
func (l *CIDRList) Len() int {
|
|
return len(*l)
|
|
}
|
|
|
|
// Less implements sort.Interface.
|
|
func (l *CIDRList) Less(i int, j int) bool {
|
|
ci := (*l)[i]
|
|
cj := (*l)[j]
|
|
|
|
if len(ci.Ip) < len(cj.Ip) {
|
|
return true
|
|
}
|
|
|
|
if len(ci.Ip) > len(cj.Ip) {
|
|
return false
|
|
}
|
|
|
|
for k := 0; k < len(ci.Ip); k++ {
|
|
if ci.Ip[k] < cj.Ip[k] {
|
|
return true
|
|
}
|
|
|
|
if ci.Ip[k] > cj.Ip[k] {
|
|
return false
|
|
}
|
|
}
|
|
|
|
return ci.Prefix < cj.Prefix
|
|
}
|
|
|
|
// Swap implements sort.Interface.
|
|
func (l *CIDRList) Swap(i int, j int) {
|
|
(*l)[i], (*l)[j] = (*l)[j], (*l)[i]
|
|
}
|
|
|
|
type Rule struct {
|
|
Tag string
|
|
Condition Condition
|
|
}
|
|
|
|
func (r *Rule) Apply(ctx context.Context) bool {
|
|
return r.Condition.Apply(ctx)
|
|
}
|
|
|
|
func (rr *RoutingRule) BuildCondition() (Condition, error) {
|
|
conds := NewConditionChan()
|
|
|
|
if len(rr.Domain) > 0 {
|
|
matcher, err := NewDomainMatcher(rr.Domain)
|
|
if err != nil {
|
|
return nil, newError("failed to build domain condition").Base(err)
|
|
}
|
|
conds.Add(matcher)
|
|
}
|
|
|
|
if len(rr.UserEmail) > 0 {
|
|
conds.Add(NewUserMatcher(rr.UserEmail))
|
|
}
|
|
|
|
if len(rr.InboundTag) > 0 {
|
|
conds.Add(NewInboundTagMatcher(rr.InboundTag))
|
|
}
|
|
|
|
if rr.PortRange != nil {
|
|
conds.Add(NewPortMatcher(*rr.PortRange))
|
|
}
|
|
|
|
if rr.NetworkList != nil {
|
|
conds.Add(NewNetworkMatcher(rr.NetworkList))
|
|
}
|
|
|
|
if len(rr.Geoip) > 0 {
|
|
cond, err := NewMultiGeoIPMatcher(rr.Geoip, false)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
conds.Add(cond)
|
|
} else if len(rr.Cidr) > 0 {
|
|
cond, err := NewMultiGeoIPMatcher([]*GeoIP{{Cidr: rr.Cidr}}, false)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
conds.Add(cond)
|
|
}
|
|
|
|
if len(rr.SourceGeoip) > 0 {
|
|
cond, err := NewMultiGeoIPMatcher(rr.SourceGeoip, true)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
conds.Add(cond)
|
|
} else if len(rr.SourceCidr) > 0 {
|
|
cond, err := NewMultiGeoIPMatcher([]*GeoIP{{Cidr: rr.SourceCidr}}, true)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
conds.Add(cond)
|
|
}
|
|
|
|
if len(rr.Protocol) > 0 {
|
|
conds.Add(NewProtocolMatcher(rr.Protocol))
|
|
}
|
|
|
|
if conds.Len() == 0 {
|
|
return nil, newError("this rule has no effective fields").AtWarning()
|
|
}
|
|
|
|
return conds, nil
|
|
}
|