package router import ( "context" "net" v2net "v2ray.com/core/common/net" ) type Rule struct { Tag string Condition Condition } func (r *Rule) Apply(ctx context.Context) bool { return r.Condition.Apply(ctx) } func cidrToCondition(cidr []*CIDR, source bool) (Condition, error) { ipv4Net := v2net.NewIPNet() ipv6Cond := NewAnyCondition() hasIpv6 := false for _, ip := range cidr { switch len(ip.Ip) { case net.IPv4len: ipv4Net.AddIP(ip.Ip, byte(ip.Prefix)) case net.IPv6len: hasIpv6 = true matcher, err := NewCIDRMatcher(ip.Ip, ip.Prefix, source) if err != nil { return nil, err } ipv6Cond.Add(matcher) default: return nil, newError("invalid IP length").AtError() } } if !ipv4Net.IsEmpty() && hasIpv6 { cond := NewAnyCondition() cond.Add(NewIPv4Matcher(ipv4Net, source)) cond.Add(ipv6Cond) return cond, nil } else if !ipv4Net.IsEmpty() { return NewIPv4Matcher(ipv4Net, source), nil } else { return ipv6Cond, nil } } func (rr *RoutingRule) BuildCondition() (Condition, error) { conds := NewConditionChan() if len(rr.Domain) > 0 { anyCond := NewAnyCondition() for _, domain := range rr.Domain { switch domain.Type { case Domain_Plain: anyCond.Add(NewPlainDomainMatcher(domain.Value)) case Domain_Regex: matcher, err := NewRegexpDomainMatcher(domain.Value) if err != nil { return nil, err } anyCond.Add(matcher) case Domain_Domain: anyCond.Add(NewSubDomainMatcher(domain.Value)) default: panic("Unknown domain type.") } } conds.Add(anyCond) } if len(rr.Cidr) > 0 { cond, err := cidrToCondition(rr.Cidr, false) if err != nil { return nil, err } conds.Add(cond) } if rr.PortRange != nil { conds.Add(NewPortMatcher(*rr.PortRange)) } if rr.NetworkList != nil { conds.Add(NewNetworkMatcher(rr.NetworkList)) } if len(rr.SourceCidr) > 0 { cond, err := cidrToCondition(rr.SourceCidr, true) if err != nil { return nil, err } conds.Add(cond) } if len(rr.UserEmail) > 0 { conds.Add(NewUserMatcher(rr.UserEmail)) } if len(rr.InboundTag) > 0 { conds.Add(NewInboundTagMatcher(rr.InboundTag)) } if conds.Len() == 0 { return nil, newError("this rule has no effective fields").AtError() } return conds, nil }