sing-box-extended-mirror/route/rule/rule_item_rule_set.go

package rule

import (
	"strings"

	"github.com/sagernet/sing-box/adapter"
	"github.com/sagernet/sing/common"
	E "github.com/sagernet/sing/common/exceptions"
	F "github.com/sagernet/sing/common/format"
)

var _ RuleItem = (*RuleSetItem)(nil)

type RuleSetItem struct {
	router            adapter.Router
	tagList           []string
	setList           []adapter.RuleSet
	ipCidrMatchSource bool
	ipCidrAcceptEmpty bool
}

func NewRuleSetItem(router adapter.Router, tagList []string, ipCIDRMatchSource bool, ipCidrAcceptEmpty bool) *RuleSetItem {
	return &RuleSetItem{
		router:            router,
		tagList:           tagList,
		ipCidrMatchSource: ipCIDRMatchSource,
		ipCidrAcceptEmpty: ipCidrAcceptEmpty,
	}
}

func (r *RuleSetItem) Start() error {
	_ = r.Close()
	for _, tag := range r.tagList {
		ruleSet, loaded := r.router.RuleSet(tag)
		if !loaded {
			_ = r.Close()
			return E.New("rule-set not found: ", tag)
		}
		ruleSet.IncRef()
		r.setList = append(r.setList, ruleSet)
	}
	return nil
}

func (r *RuleSetItem) Close() error {
	for _, ruleSet := range r.setList {
		ruleSet.DecRef()
	}
	clear(r.setList)
	r.setList = nil
	return nil
}

func (r *RuleSetItem) Match(metadata *adapter.InboundContext) bool {
	snapshot := snapshotRuleMatch(metadata)
	for _, ruleSet := range r.setList {
		r.prepareNestedMatch(metadata)
		if ruleSet.Match(metadata) {
			snapshot.restore(metadata)
			return true
		}
	}
	snapshot.restore(metadata)
	return false
}

func (r *RuleSetItem) matchWithOuterGroups(metadata *adapter.InboundContext, outerGroups ruleGroupMatch) bool {
	outerDone := outerGroups.done()
	var (
		matched        bool
		deferredGroups uint8
	)
	snapshot := snapshotRuleMatch(metadata)
	for _, ruleSet := range r.setList {
		r.prepareNestedMatch(metadata)
		if provider, isProvider := ruleSet.(mergeableRuleProvider); isProvider {
			branch := provider.mergeableRule()
			if branch != nil {
				branchGroups, branchMatched := branch.evaluateForMerge(metadata)
				if branchMatched {
					merged := outerGroups.mergeWith(branchGroups)
					if merged.done() {
						branchDeferredGroups := metadata.DeferredIPCIDRMatchGroups &^ uint8(merged.satisfied)
						if branchDeferredGroups == 0 {
							snapshot.restore(metadata)
							metadata.DeferredIPCIDRMatchGroups &^= uint8(merged.satisfied)
							return true
						}
						matched = true
						deferredGroups |= branchDeferredGroups
					}
				}
				continue
			}
		}
		if outerDone && ruleSet.Match(metadata) {
			if metadata.DeferredIPCIDRMatchGroups == 0 {
				snapshot.restore(metadata)
				return true
			}
			matched = true
			deferredGroups |= metadata.DeferredIPCIDRMatchGroups
		}
	}
	snapshot.restore(metadata)
	if matched {
		metadata.DeferredIPCIDRMatchGroups |= deferredGroups
	}
	return matched
}

func (r *RuleSetItem) prepareNestedMatch(metadata *adapter.InboundContext) {
	metadata.ResetRuleMatchCache()
	metadata.IPCIDRMatchSource = r.ipCidrMatchSource
	metadata.IPCIDRAcceptEmpty = r.ipCidrAcceptEmpty
}

type mergeableRuleProvider interface {
	mergeableRule() *DefaultHeadlessRule
}

func mergeableRuleIn(rules []adapter.HeadlessRule) *DefaultHeadlessRule {
	if len(rules) != 1 {
		return nil
	}
	rule, isDefault := rules[0].(*DefaultHeadlessRule)
	if !isDefault || rule.invert || rule.ruleSetItem != nil {
		return nil
	}
	return rule
}

func matchAnyHeadlessRule(rules []adapter.HeadlessRule, metadata *adapter.InboundContext) bool {
	var (
		matched        bool
		deferredGroups uint8
	)
	snapshot := snapshotRuleMatch(metadata)
	for _, rule := range rules {
		metadata.ResetRuleMatchCache()
		if rule.Match(metadata) {
			if metadata.DeferredIPCIDRMatchGroups == 0 {
				snapshot.restore(metadata)
				return true
			}
			matched = true
			deferredGroups |= metadata.DeferredIPCIDRMatchGroups
		}
	}
	snapshot.restore(metadata)
	if matched {
		metadata.DeferredIPCIDRMatchGroups |= deferredGroups
	}
	return matched
}

func (r *RuleSetItem) ContainsDestinationIPCIDRRule() bool {
	if r.ipCidrMatchSource {
		return false
	}
	return common.Any(r.setList, func(ruleSet adapter.RuleSet) bool {
		return ruleSet.Metadata().ContainsIPCIDRRule
	})
}

func (r *RuleSetItem) String() string {
	if len(r.tagList) == 1 {
		return F.ToString("rule_set=", r.tagList[0])
	} else {
		return F.ToString("rule_set=[", strings.Join(r.tagList, " "), "]")
	}
}