mirror of
https://github.com/MetaCubeX/mihomo.git
synced 2024-09-27 02:17:54 +00:00
b3db113b1b
Some checks failed
Build / build (map[abi:1 goarch:loong64 goos:linux output:loong64-abi1]) (push) Has been cancelled
Build / build (map[abi:2 goarch:loong64 goos:linux output:loong64-abi2]) (push) Has been cancelled
Build / build (map[goamd64:v1 goarch:amd64 goos:darwin goversion:1.20 output:amd64-compatible-go120]) (push) Has been cancelled
Build / build (map[goamd64:v1 goarch:amd64 goos:darwin output:amd64-compatible]) (push) Has been cancelled
Build / build (map[goamd64:v1 goarch:amd64 goos:freebsd output:amd64-compatible]) (push) Has been cancelled
Build / build (map[goamd64:v1 goarch:amd64 goos:linux goversion:1.20 output:amd64-compatible-go120 test:test]) (push) Has been cancelled
Build / build (map[goamd64:v1 goarch:amd64 goos:linux output:amd64-compatible test:test]) (push) Has been cancelled
Build / build (map[goamd64:v1 goarch:amd64 goos:windows goversion:1.20 output:amd64-compatible-go120]) (push) Has been cancelled
Build / build (map[goamd64:v1 goarch:amd64 goos:windows output:amd64-compatible]) (push) Has been cancelled
Build / build (map[goamd64:v3 goarch:amd64 goos:darwin goversion:1.20 output:amd64-go120]) (push) Has been cancelled
Build / build (map[goamd64:v3 goarch:amd64 goos:darwin output:amd64]) (push) Has been cancelled
Build / build (map[goamd64:v3 goarch:amd64 goos:freebsd output:amd64]) (push) Has been cancelled
Build / build (map[goamd64:v3 goarch:amd64 goos:linux goversion:1.20 output:amd64-go120]) (push) Has been cancelled
Build / build (map[goamd64:v3 goarch:amd64 goos:linux output:amd64]) (push) Has been cancelled
Build / build (map[goamd64:v3 goarch:amd64 goos:windows goversion:1.20 output:amd64-go120]) (push) Has been cancelled
Build / build (map[goamd64:v3 goarch:amd64 goos:windows output:amd64]) (push) Has been cancelled
Build / build (map[goarch:386 goos:android ndk:i686-linux-android34 output:386]) (push) Has been cancelled
Build / build (map[goarch:386 goos:freebsd output:386]) (push) Has been cancelled
Build / build (map[goarch:386 goos:linux goversion:1.20 output:386-go120]) (push) Has been cancelled
Build / build (map[goarch:386 goos:linux output:386]) (push) Has been cancelled
Build / build (map[goarch:386 goos:windows goversion:1.20 output:386-go120]) (push) Has been cancelled
Build / build (map[goarch:386 goos:windows output:386]) (push) Has been cancelled
Build / build (map[goarch:amd64 goos:android ndk:x86_64-linux-android34 output:amd64]) (push) Has been cancelled
Build / build (map[goarch:arm goarm:7 goos:linux output:armv7]) (push) Has been cancelled
Build / build (map[goarch:arm goarm:7 goos:windows output:armv7]) (push) Has been cancelled
Build / build (map[goarch:arm goos:android ndk:armv7a-linux-androideabi34 output:armv7]) (push) Has been cancelled
Build / build (map[goarch:arm64 goos:android ndk:aarch64-linux-android34 output:arm64-v8]) (push) Has been cancelled
Build / build (map[goarch:arm64 goos:darwin goversion:1.20 output:arm64-go120]) (push) Has been cancelled
Build / build (map[goarch:arm64 goos:darwin output:arm64]) (push) Has been cancelled
Build / build (map[goarch:arm64 goos:freebsd output:arm64]) (push) Has been cancelled
Build / build (map[goarch:arm64 goos:linux output:arm64]) (push) Has been cancelled
Build / build (map[goarch:arm64 goos:windows output:arm64]) (push) Has been cancelled
Build / build (map[goarch:mips goos:linux mips:hardfloat output:mips-hardfloat]) (push) Has been cancelled
Build / build (map[goarch:mips goos:linux mips:softfloat output:mips-softfloat]) (push) Has been cancelled
Build / build (map[goarch:mips64 goos:linux output:mips64]) (push) Has been cancelled
Build / build (map[goarch:mips64le goos:linux output:mips64le]) (push) Has been cancelled
Build / build (map[goarch:mipsle goos:linux mips:hardfloat output:mipsle-hardfloat]) (push) Has been cancelled
Build / build (map[goarch:mipsle goos:linux mips:softfloat output:mipsle-softfloat]) (push) Has been cancelled
Build / build (map[goarch:riscv64 goos:linux output:riscv64]) (push) Has been cancelled
Build / build (map[goarch:s390x goos:linux output:s390x]) (push) Has been cancelled
Trigger CMFA Update / trigger-CMFA-update (push) Has been cancelled
Build / Upload-Prerelease (push) Has been cancelled
Build / Upload-Release (push) Has been cancelled
Build / Docker (push) Has been cancelled
129 lines
2.8 KiB
Go
129 lines
2.8 KiB
Go
package resolver
|
|
|
|
import (
|
|
"errors"
|
|
"net/netip"
|
|
"os"
|
|
"strconv"
|
|
"strings"
|
|
_ "unsafe"
|
|
|
|
"github.com/metacubex/mihomo/common/utils"
|
|
"github.com/metacubex/mihomo/component/trie"
|
|
"github.com/zhangyunhao116/fastrand"
|
|
)
|
|
|
|
var DisableSystemHosts, _ = strconv.ParseBool(os.Getenv("DISABLE_SYSTEM_HOSTS"))
|
|
|
|
type Hosts struct {
|
|
*trie.DomainTrie[HostValue]
|
|
}
|
|
|
|
func NewHosts(hosts *trie.DomainTrie[HostValue]) Hosts {
|
|
return Hosts{
|
|
hosts,
|
|
}
|
|
}
|
|
|
|
// lookupStaticHost looks up the addresses and the canonical name for the given host from /etc/hosts.
|
|
//
|
|
//go:linkname lookupStaticHost net.lookupStaticHost
|
|
func lookupStaticHost(host string) ([]string, string)
|
|
|
|
// Return the search result and whether to match the parameter `isDomain`
|
|
func (h *Hosts) Search(domain string, isDomain bool) (*HostValue, bool) {
|
|
if value := h.DomainTrie.Search(domain); value != nil {
|
|
hostValue := value.Data()
|
|
for {
|
|
if isDomain && hostValue.IsDomain {
|
|
return &hostValue, true
|
|
} else {
|
|
if node := h.DomainTrie.Search(hostValue.Domain); node != nil {
|
|
hostValue = node.Data()
|
|
} else {
|
|
break
|
|
}
|
|
}
|
|
}
|
|
if isDomain == hostValue.IsDomain {
|
|
return &hostValue, true
|
|
}
|
|
|
|
return &hostValue, false
|
|
}
|
|
if !isDomain && !DisableSystemHosts {
|
|
addr, _ := lookupStaticHost(domain)
|
|
if hostValue, err := NewHostValue(addr); err == nil {
|
|
return &hostValue, true
|
|
}
|
|
}
|
|
return nil, false
|
|
}
|
|
|
|
type HostValue struct {
|
|
IsDomain bool
|
|
IPs []netip.Addr
|
|
Domain string
|
|
}
|
|
|
|
func NewHostValue(value any) (HostValue, error) {
|
|
isDomain := true
|
|
ips := make([]netip.Addr, 0)
|
|
domain := ""
|
|
if valueArr, err := utils.ToStringSlice(value); err != nil {
|
|
return HostValue{}, err
|
|
} else {
|
|
if len(valueArr) > 1 {
|
|
isDomain = false
|
|
for _, str := range valueArr {
|
|
if ip, err := netip.ParseAddr(str); err == nil {
|
|
ips = append(ips, ip)
|
|
} else {
|
|
return HostValue{}, err
|
|
}
|
|
}
|
|
} else if len(valueArr) == 1 {
|
|
host := valueArr[0]
|
|
if ip, err := netip.ParseAddr(host); err == nil {
|
|
ips = append(ips, ip)
|
|
isDomain = false
|
|
} else {
|
|
domain = host
|
|
}
|
|
}
|
|
}
|
|
if isDomain {
|
|
return NewHostValueByDomain(domain)
|
|
} else {
|
|
return NewHostValueByIPs(ips)
|
|
}
|
|
}
|
|
|
|
func NewHostValueByIPs(ips []netip.Addr) (HostValue, error) {
|
|
if len(ips) == 0 {
|
|
return HostValue{}, errors.New("ip list is empty")
|
|
}
|
|
return HostValue{
|
|
IsDomain: false,
|
|
IPs: ips,
|
|
}, nil
|
|
}
|
|
|
|
func NewHostValueByDomain(domain string) (HostValue, error) {
|
|
domain = strings.Trim(domain, ".")
|
|
item := strings.Split(domain, ".")
|
|
if len(item) < 2 {
|
|
return HostValue{}, errors.New("invaild domain")
|
|
}
|
|
return HostValue{
|
|
IsDomain: true,
|
|
Domain: domain,
|
|
}, nil
|
|
}
|
|
|
|
func (hv HostValue) RandIP() (netip.Addr, error) {
|
|
if hv.IsDomain {
|
|
return netip.Addr{}, errors.New("value type is error")
|
|
}
|
|
return hv.IPs[fastrand.Intn(len(hv.IPs))], nil
|
|
}
|