joker
/
godns
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

File hosts support widlcard #31

This commit is contained in:
kenshinx 2017-05-18 18:10:33 +08:00
parent 829c7af1bc
commit 5f42911d12
1 changed files with 23 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
hosts.go
View File

@ -10,6 +10,7 @@ import (
"time" "time"
"github.com/hoisie/redis" "github.com/hoisie/redis"
"golang.org/x/net/publicsuffix"
) )
type Hosts struct { type Hosts struct {
@ -149,14 +150,32 @@ type FileHosts struct {
} }
func (f *FileHosts) Get(domain string) ([]string, bool) { func (f *FileHosts) Get(domain string) ([]string, bool) {
domain = strings.ToLower(domain)
f.mu.RLock() f.mu.RLock()
defer f.mu.RUnlock()
domain = strings.ToLower(domain)
ip, ok := f.hosts[domain] ip, ok := f.hosts[domain]
f.mu.RUnlock() if ok {
if !ok { return []string{ip}, true
}
sld, err := publicsuffix.EffectiveTLDPlusOne(domain)
if err != nil {
return nil, false return nil, false
} }
return []string{ip}, true
for host, ip := range f.hosts {
if strings.HasPrefix(host, "*.") {
old, err := publicsuffix.EffectiveTLDPlusOne(host)
if err != nil {
continue
}
if sld == old {
return []string{ip}, true
}
}
}
return nil, false
} }
func (f *FileHosts) Refresh() { func (f *FileHosts) Refresh() {