feat/optimize: add userspace routing and optimize domain routing (#18)

component/routing/domain_matcher/bruteforce.go

@@ -0,0 +1,67 @@
+/*
+ * SPDX-License-Identifier: AGPL-3.0-only
+ * Copyright (c) 2023, v2rayA Organization <team@v2raya.org>
+ */
+
+package domain_matcher
+
+import (
+	"github.com/v2rayA/dae/common/consts"
+	"github.com/v2rayA/dae/component/routing"
+	"regexp"
+	"strings"
+)
+
+type Bruteforce struct {
+	simulatedDomainSet []routing.DomainSet
+	err                error
+}
+
+func NewBruteforce(simulatedDomainSet []routing.DomainSet) *Bruteforce {
+	return &Bruteforce{
+		simulatedDomainSet: simulatedDomainSet,
+	}
+}
+func (n *Bruteforce) AddSet(bitIndex int, patterns []string, typ consts.RoutingDomainKey) {
+}
+func (n *Bruteforce) MatchDomainBitmap(domain string) (bitmap []uint32) {
+	N := len(n.simulatedDomainSet) / 32
+	if len(n.simulatedDomainSet)%32 != 0 {
+		N++
+	}
+	bitmap = make([]uint32, N)
+	for _, s := range n.simulatedDomainSet {
+		for _, d := range s.Domains {
+			var hit bool
+			switch s.Key {
+			case consts.RoutingDomainKey_Suffix:
+				if domain == d || strings.HasSuffix(domain, "."+strings.TrimPrefix(d, ".")) {
+					hit = true
+				}
+			case consts.RoutingDomainKey_Full:
+				if strings.EqualFold(domain, d) {
+					hit = true
+				}
+			case consts.RoutingDomainKey_Keyword:
+				if strings.Contains(strings.ToLower(domain), strings.ToLower(d)) {
+					hit = true
+				}
+			case consts.RoutingDomainKey_Regex:
+				if regexp.MustCompile(d).MatchString(strings.ToLower(domain)) {
+					hit = true
+				}
+			}
+			if hit {
+				bitmap[s.RuleIndex/32] |= 1 << (s.RuleIndex % 32)
+				break
+			}
+		}
+	}
+	return bitmap
+}
+func (n *Bruteforce) Build() error {
+	if n.err != nil {
+		return n.err
+	}
+	return nil
+}