chore: allow disabled system hosts by environment variable DISABLE_SYSTEM_HOSTS

This commit is contained in:
wwqgtxx 2024-03-13 15:32:26 +08:00
parent dceb8ee535
commit b3db113b1b

View File

@ -3,6 +3,8 @@ package resolver
import (
"errors"
"net/netip"
"os"
"strconv"
"strings"
_ "unsafe"
@ -11,6 +13,8 @@ import (
"github.com/zhangyunhao116/fastrand"
)
var DisableSystemHosts, _ = strconv.ParseBool(os.Getenv("DISABLE_SYSTEM_HOSTS"))
type Hosts struct {
*trie.DomainTrie[HostValue]
}
@ -47,7 +51,7 @@ func (h *Hosts) Search(domain string, isDomain bool) (*HostValue, bool) {
return &hostValue, false
}
if !isDomain {
if !isDomain && !DisableSystemHosts {
addr, _ := lookupStaticHost(domain)
if hostValue, err := NewHostValue(addr); err == nil {
return &hostValue, true