mirrors/dae
1
0
Fork 0
mirror of https://github.com/daeuniverse/dae.git synced 2025-07-12 08:49:53 +07:00
Code Issues Packages Projects Releases Wiki Activity

fix: add _ to valid domain chars set (#365)

Browse Source
This commit is contained in:
mzz
2023-12-26 19:27:18 +08:00
committed by GitHub
parent a794c4ca61
commit aeaf2fc661
3 changed files with 8 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
component/dns/response_routing.go
View File

@ -9,7 +9,6 @@ import (
"fmt"
"net/netip"
"strconv"
"strings"
"github.com/daeuniverse/dae/common/consts"
"github.com/daeuniverse/dae/component/routing"
@ -225,7 +224,6 @@ func (m *ResponseMatcher) Match(
if qName == "" {
return 0, fmt.Errorf("qName cannot be empty")
}
qName = strings.TrimSuffix(strings.ToLower(qName), ".")
domainMatchBitmap := m.domainMatcher.MatchDomainBitmap(qName)
bin128 := make([]string, 0, len(ips))
for _, ip := range ips {

11
component/routing/domain_matcher/ahocorasick_slimtrie.go
View File

@ -7,15 +7,16 @@ package domain_matcher
import (
"fmt"
"regexp"
"strings"
"github.com/daeuniverse/dae/common/consts"
"github.com/daeuniverse/dae/pkg/trie"
"github.com/sirupsen/logrus"
"github.com/v2rayA/ahocorasick-domain"
"regexp"
"strings"
)
var ValidDomainChars = trie.NewValidChars([]byte("0123456789abcdefghijklmnopqrstuvwxyz-.^"))
var ValidDomainChars = trie.NewValidChars([]byte("0123456789abcdefghijklmnopqrstuvwxyz-.^_"))
type AhocorasickSlimtrie struct {
log *logrus.Logger
@ -52,7 +53,7 @@ nextPattern:
case consts.RoutingDomainKey_Full:
for _, r := range []byte(d) {
if !ValidDomainChars.IsValidChar(r) {
n.log.Warnf("DomainMatcher: skip bad full domain: %v: unexpected chat: %v", d, r)
n.log.Warnf("DomainMatcher: skip bad full domain: %v: unexpected char: %v", d, string(r))
continue nextPattern
}
}
@ -60,7 +61,7 @@ nextPattern:
case consts.RoutingDomainKey_Suffix:
for _, r := range []byte(d) {
if !ValidDomainChars.IsValidChar(r) {
n.log.Warnf("DomainMatcher: skip bad suffix domain: %v: unexpected chat: %v", d, r)
n.log.Warnf("DomainMatcher: skip bad suffix domain: %v: unexpected char: %v", d, string(r))
continue nextPattern
}
}

3
pkg/trie/trie_test.go
View File

@ -94,7 +94,8 @@ func TestTrie(t *testing.T) {
"zib.fmc^",
"zk.ytamlacbci.",
"zk.ytamlacbci^",
}, NewValidChars([]byte("0123456789abcdefghijklmnopqrstuvwxyz-.^")))
"nc.ude.ctsu.srorrim.pct_.sptth_", // https://github.com/daeuniverse/daed/issues/400
}, NewValidChars([]byte("0123456789abcdefghijklmnopqrstuvwxyz-.^_")))
if err != nil {
t.Fatal(err)
}