xmengnet/mihomo
1
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fix(domain-trie): crash when insert

Browse source
This commit is contained in:
Dreamacro 2019-09-14 21:42:40 +08:00
parent 09917a2a95
commit 3dd9ea52d8
2 changed files with 5 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
component/domain-trie/tire.go
View file

@ -22,7 +22,7 @@ type Trie struct {
} }
func isValidDomain(domain string) bool { func isValidDomain(domain string) bool {
return domain[0] != '.' && domain[len(domain)-1] != '.' return domain != "" && domain[0] != '.' && domain[len(domain)-1] != '.'
} }
// Insert adds a node to the trie. // Insert adds a node to the trie.

4
component/domain-trie/trie_test.go
View file

@ -26,6 +26,10 @@ func TestTrie_Basic(t *testing.T) {
if !node.Data.(net.IP).Equal(localIP) { if !node.Data.(net.IP).Equal(localIP) {
t.Error("should equal 127.0.0.1") t.Error("should equal 127.0.0.1")
} }
if tree.Insert("", localIP) == nil {
t.Error("should return error")
}
} }
func TestTrie_Wildcard(t *testing.T) { func TestTrie_Wildcard(t *testing.T) {