50 lines
920 B
Go
50 lines
920 B
Go
package dns
|
|
|
|
import (
|
|
"net"
|
|
"strings"
|
|
|
|
"github.com/Dreamacro/clash/component/mmdb"
|
|
"github.com/Dreamacro/clash/component/trie"
|
|
)
|
|
|
|
type fallbackIPFilter interface {
|
|
Match(net.IP) bool
|
|
}
|
|
|
|
type geoipFilter struct {
|
|
code string
|
|
}
|
|
|
|
func (gf *geoipFilter) Match(ip net.IP) bool {
|
|
record, _ := mmdb.Instance().Country(ip)
|
|
return !strings.EqualFold(record.Country.IsoCode, gf.code) && !ip.IsPrivate()
|
|
}
|
|
|
|
type ipnetFilter struct {
|
|
ipnet *net.IPNet
|
|
}
|
|
|
|
func (inf *ipnetFilter) Match(ip net.IP) bool {
|
|
return inf.ipnet.Contains(ip)
|
|
}
|
|
|
|
type fallbackDomainFilter interface {
|
|
Match(domain string) bool
|
|
}
|
|
|
|
type domainFilter struct {
|
|
tree *trie.DomainTrie
|
|
}
|
|
|
|
func NewDomainFilter(domains []string) *domainFilter {
|
|
df := domainFilter{tree: trie.New()}
|
|
for _, domain := range domains {
|
|
df.tree.Insert(domain, "")
|
|
}
|
|
return &df
|
|
}
|
|
|
|
func (df *domainFilter) Match(domain string) bool {
|
|
return df.tree.Search(domain) != nil
|
|
}
|