mirror of
https://github.com/MetaCubeX/mihomo.git
synced 2024-09-12 11:52:33 +00:00
140 lines
2.9 KiB
Go
140 lines
2.9 KiB
Go
package resolver
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"math/rand"
|
|
"net"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/Dreamacro/clash/component/trie"
|
|
)
|
|
|
|
var (
|
|
// DefaultResolver aim to resolve ip
|
|
DefaultResolver Resolver
|
|
|
|
// DisableIPv6 means don't resolve ipv6 host
|
|
// default value is true
|
|
DisableIPv6 = true
|
|
|
|
// DefaultHosts aim to resolve hosts
|
|
DefaultHosts = trie.New()
|
|
|
|
// DefaultDNSTimeout defined the default dns request timeout
|
|
DefaultDNSTimeout = time.Second * 5
|
|
)
|
|
|
|
var (
|
|
ErrIPNotFound = errors.New("couldn't find ip")
|
|
ErrIPVersion = errors.New("ip version error")
|
|
ErrIPv6Disabled = errors.New("ipv6 disabled")
|
|
)
|
|
|
|
type Resolver interface {
|
|
ResolveIP(host string) (ip net.IP, err error)
|
|
ResolveIPv4(host string) (ip net.IP, err error)
|
|
ResolveIPv6(host string) (ip net.IP, err error)
|
|
}
|
|
|
|
// ResolveIPv4 with a host, return ipv4
|
|
func ResolveIPv4(host string) (net.IP, error) {
|
|
if node := DefaultHosts.Search(host); node != nil {
|
|
if ip := node.Data.(net.IP).To4(); ip != nil {
|
|
return ip, nil
|
|
}
|
|
}
|
|
|
|
ip := net.ParseIP(host)
|
|
if ip != nil {
|
|
if !strings.Contains(host, ":") {
|
|
return ip, nil
|
|
}
|
|
return nil, ErrIPVersion
|
|
}
|
|
|
|
if DefaultResolver != nil {
|
|
return DefaultResolver.ResolveIPv4(host)
|
|
}
|
|
|
|
ctx, cancel := context.WithTimeout(context.Background(), DefaultDNSTimeout)
|
|
defer cancel()
|
|
ipAddrs, err := net.DefaultResolver.LookupIP(ctx, "ip4", host)
|
|
if err != nil {
|
|
return nil, err
|
|
} else if len(ipAddrs) == 0 {
|
|
return nil, ErrIPNotFound
|
|
}
|
|
|
|
return ipAddrs[rand.Intn(len(ipAddrs))], nil
|
|
}
|
|
|
|
// ResolveIPv6 with a host, return ipv6
|
|
func ResolveIPv6(host string) (net.IP, error) {
|
|
if DisableIPv6 {
|
|
return nil, ErrIPv6Disabled
|
|
}
|
|
|
|
if node := DefaultHosts.Search(host); node != nil {
|
|
if ip := node.Data.(net.IP).To16(); ip != nil {
|
|
return ip, nil
|
|
}
|
|
}
|
|
|
|
ip := net.ParseIP(host)
|
|
if ip != nil {
|
|
if strings.Contains(host, ":") {
|
|
return ip, nil
|
|
}
|
|
return nil, ErrIPVersion
|
|
}
|
|
|
|
if DefaultResolver != nil {
|
|
return DefaultResolver.ResolveIPv6(host)
|
|
}
|
|
|
|
ctx, cancel := context.WithTimeout(context.Background(), DefaultDNSTimeout)
|
|
defer cancel()
|
|
ipAddrs, err := net.DefaultResolver.LookupIP(ctx, "ip6", host)
|
|
if err != nil {
|
|
return nil, err
|
|
} else if len(ipAddrs) == 0 {
|
|
return nil, ErrIPNotFound
|
|
}
|
|
|
|
return ipAddrs[rand.Intn(len(ipAddrs))], nil
|
|
}
|
|
|
|
// ResolveIPWithResolver same as ResolveIP, but with a resolver
|
|
func ResolveIPWithResolver(host string, r Resolver) (net.IP, error) {
|
|
if node := DefaultHosts.Search(host); node != nil {
|
|
return node.Data.(net.IP), nil
|
|
}
|
|
|
|
if r != nil {
|
|
if DisableIPv6 {
|
|
return r.ResolveIPv4(host)
|
|
}
|
|
return r.ResolveIP(host)
|
|
} else if DisableIPv6 {
|
|
return ResolveIPv4(host)
|
|
}
|
|
|
|
ip := net.ParseIP(host)
|
|
if ip != nil {
|
|
return ip, nil
|
|
}
|
|
|
|
ipAddr, err := net.ResolveIPAddr("ip", host)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return ipAddr.IP, nil
|
|
}
|
|
|
|
// ResolveIP with a host, return ip
|
|
func ResolveIP(host string) (net.IP, error) {
|
|
return ResolveIPWithResolver(host, DefaultResolver)
|
|
}
|