Fix: domain trie crash if not match in #758 (#762)

This commit is contained in:
Kr328 2020-06-24 19:46:37 +08:00 committed by GitHub
parent 3dfff84cc3
commit 14c9cf1b97
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 1 deletions

View File

@ -98,7 +98,7 @@ func (t *DomainTrie) Search(domain string) *Node {
n := t.search(t.root, parts) n := t.search(t.root, parts)
if n.Data == nil { if n == nil || n.Data == nil {
return nil return nil
} }

View File

@ -27,6 +27,7 @@ func TestTrie_Basic(t *testing.T) {
assert.NotNil(t, tree.Insert("", localIP)) assert.NotNil(t, tree.Insert("", localIP))
assert.Nil(t, tree.Search("")) assert.Nil(t, tree.Search(""))
assert.NotNil(t, tree.Search("localhost")) assert.NotNil(t, tree.Search("localhost"))
assert.Nil(t, tree.Search("www.google.com"))
} }
func TestTrie_Wildcard(t *testing.T) { func TestTrie_Wildcard(t *testing.T) {