chore: Introducing Punycode conversion for domain matching

MetaCubeX · Apr 1, 2024 · f3743fc · f3743fc
1 parent 56ed901
commit f3743fc
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 3 deletions.
diff --git a/rules/common/domain.go b/rules/common/domain.go
@@ -4,6 +4,7 @@ import (
 	"strings"
 
 	C "github.com/metacubex/mihomo/constant"
+	"golang.org/x/net/idna"
 )
 
 type Domain struct {
@@ -29,9 +30,10 @@ func (d *Domain) Payload() string {
 }
 
 func NewDomain(domain string, adapter string) *Domain {
+	punycode, _ := idna.ToASCII(strings.ToLower(domain))
 	return &Domain{
 		Base:    &Base{},
-		domain:  strings.ToLower(domain),
+		domain:  punycode,
 		adapter: adapter,
 	}
 }

diff --git a/rules/common/domain_keyword.go b/rules/common/domain_keyword.go
@@ -4,6 +4,7 @@ import (
 	"strings"
 
 	C "github.com/metacubex/mihomo/constant"
+	"golang.org/x/net/idna"
 )
 
 type DomainKeyword struct {
@@ -30,9 +31,10 @@ func (dk *DomainKeyword) Payload() string {
 }
 
 func NewDomainKeyword(keyword string, adapter string) *DomainKeyword {
+	punycode, _ := idna.ToASCII(strings.ToLower(keyword))
 	return &DomainKeyword{
 		Base:    &Base{},
-		keyword: strings.ToLower(keyword),
+		keyword: punycode,
 		adapter: adapter,
 	}
 }

diff --git a/rules/common/domain_suffix.go b/rules/common/domain_suffix.go
@@ -4,6 +4,7 @@ import (
 	"strings"
 
 	C "github.com/metacubex/mihomo/constant"
+	"golang.org/x/net/idna"
 )
 
 type DomainSuffix struct {
@@ -30,9 +31,10 @@ func (ds *DomainSuffix) Payload() string {
 }
 
 func NewDomainSuffix(suffix string, adapter string) *DomainSuffix {
+	punycode, _ := idna.ToASCII(strings.ToLower(suffix))
 	return &DomainSuffix{
 		Base:    &Base{},
-		suffix:  strings.ToLower(suffix),
+		suffix:  punycode,
 		adapter: adapter,
 	}
 }