Fix: domain dns crash

This commit is contained in:
Dreamacro 2020-02-17 22:13:15 +08:00
parent df0ab6aa8e
commit 46edae9896
4 changed files with 32 additions and 23 deletions

View File

@ -441,21 +441,21 @@ func parseHosts(cfg *RawConfig) (*trie.Trie, error) {
return tree, nil return tree, nil
} }
func hostWithDefaultPort(host string, defPort string) (string, string, error) { func hostWithDefaultPort(host string, defPort string) (string, error) {
if !strings.Contains(host, ":") { if !strings.Contains(host, ":") {
host += ":" host += ":"
} }
hostname, port, err := net.SplitHostPort(host) hostname, port, err := net.SplitHostPort(host)
if err != nil { if err != nil {
return "", "", err return "", err
} }
if port == "" { if port == "" {
port = defPort port = defPort
} }
return net.JoinHostPort(hostname, port), hostname, nil return net.JoinHostPort(hostname, port), nil
} }
func parseNameServer(servers []string) ([]dns.NameServer, error) { func parseNameServer(servers []string) ([]dns.NameServer, error) {
@ -471,21 +471,20 @@ func parseNameServer(servers []string) ([]dns.NameServer, error) {
return nil, fmt.Errorf("DNS NameServer[%d] format error: %s", idx, err.Error()) return nil, fmt.Errorf("DNS NameServer[%d] format error: %s", idx, err.Error())
} }
var addr, dnsNetType, host string var addr, dnsNetType string
switch u.Scheme { switch u.Scheme {
case "udp": case "udp":
addr, host, err = hostWithDefaultPort(u.Host, "53") addr, err = hostWithDefaultPort(u.Host, "53")
dnsNetType = "" // UDP dnsNetType = "" // UDP
case "tcp": case "tcp":
addr, host, err = hostWithDefaultPort(u.Host, "53") addr, err = hostWithDefaultPort(u.Host, "53")
dnsNetType = "tcp" // TCP dnsNetType = "tcp" // TCP
case "tls": case "tls":
addr, host, err = hostWithDefaultPort(u.Host, "853") addr, err = hostWithDefaultPort(u.Host, "853")
dnsNetType = "tcp-tls" // DNS over TLS dnsNetType = "tcp-tls" // DNS over TLS
case "https": case "https":
clearURL := url.URL{Scheme: "https", Host: u.Host, Path: u.Path} clearURL := url.URL{Scheme: "https", Host: u.Host, Path: u.Path}
addr = clearURL.String() addr = clearURL.String()
_, host, err = hostWithDefaultPort(u.Host, "853")
dnsNetType = "https" // DNS over HTTPS dnsNetType = "https" // DNS over HTTPS
default: default:
return nil, fmt.Errorf("DNS NameServer[%d] unsupport scheme: %s", idx, u.Scheme) return nil, fmt.Errorf("DNS NameServer[%d] unsupport scheme: %s", idx, u.Scheme)
@ -500,7 +499,6 @@ func parseNameServer(servers []string) ([]dns.NameServer, error) {
dns.NameServer{ dns.NameServer{
Net: dnsNetType, Net: dnsNetType,
Addr: addr, Addr: addr,
Host: host,
}, },
) )
} }
@ -552,7 +550,8 @@ func parseDNS(cfg RawDNS) (*DNS, error) {
} }
// check default nameserver is pure ip addr // check default nameserver is pure ip addr
for _, ns := range dnsCfg.DefaultNameserver { for _, ns := range dnsCfg.DefaultNameserver {
if net.ParseIP(ns.Host) == nil { host, _, err := net.SplitHostPort(ns.Addr)
if err != nil || net.ParseIP(host) == nil {
return nil, errors.New("default nameserver should be pure IP") return nil, errors.New("default nameserver should be pure IP")
} }
} }

View File

@ -2,6 +2,8 @@ package dns
import ( import (
"context" "context"
"fmt"
"net"
"strings" "strings"
"github.com/Dreamacro/clash/component/dialer" "github.com/Dreamacro/clash/component/dialer"
@ -12,7 +14,7 @@ import (
type client struct { type client struct {
*D.Client *D.Client
r *Resolver r *Resolver
addr string port string
host string host string
} }
@ -21,18 +23,24 @@ func (c *client) Exchange(m *D.Msg) (msg *D.Msg, err error) {
} }
func (c *client) ExchangeContext(ctx context.Context, m *D.Msg) (msg *D.Msg, err error) { func (c *client) ExchangeContext(ctx context.Context, m *D.Msg) (msg *D.Msg, err error) {
network := "udp" var ip net.IP
if strings.HasPrefix(c.Client.Net, "tcp") { if c.r == nil {
network = "tcp" // a default ip dns
} ip = net.ParseIP(c.host)
} else {
ip, err := c.r.ResolveIP(c.host) var err error
if err != nil { if ip, err = c.r.ResolveIP(c.host); err != nil {
return nil, err println("?")
return nil, fmt.Errorf("use default dns resolve failed: %w", err)
}
} }
d := dialer.Dialer() d := dialer.Dialer()
if dialer.DialHook != nil { if dialer.DialHook != nil {
network := "udp"
if strings.HasPrefix(c.Client.Net, "tcp") {
network = "tcp"
}
dialer.DialHook(d, network, ip) dialer.DialHook(d, network, ip)
} }
@ -46,7 +54,7 @@ func (c *client) ExchangeContext(ctx context.Context, m *D.Msg) (msg *D.Msg, err
} }
ch := make(chan result, 1) ch := make(chan result, 1)
go func() { go func() {
msg, _, err := c.Client.Exchange(m, c.addr) msg, _, err := c.Client.Exchange(m, net.JoinHostPort(ip.String(), c.port))
ch <- result{msg, err} ch <- result{msg, err}
}() }()

View File

@ -263,7 +263,6 @@ func (r *Resolver) asyncExchange(client []dnsClient, msg *D.Msg) <-chan *result
type NameServer struct { type NameServer struct {
Net string Net string
Addr string Addr string
Host string
} }
type FallbackFilter struct { type FallbackFilter struct {

View File

@ -4,6 +4,7 @@ import (
"crypto/tls" "crypto/tls"
"encoding/json" "encoding/json"
"errors" "errors"
"net"
"time" "time"
"github.com/Dreamacro/clash/common/cache" "github.com/Dreamacro/clash/common/cache"
@ -125,6 +126,7 @@ func transform(servers []NameServer, resolver *Resolver) []dnsClient {
continue continue
} }
host, port, _ := net.SplitHostPort(s.Addr)
ret = append(ret, &client{ ret = append(ret, &client{
Client: &D.Client{ Client: &D.Client{
Net: s.Net, Net: s.Net,
@ -136,8 +138,9 @@ func transform(servers []NameServer, resolver *Resolver) []dnsClient {
UDPSize: 4096, UDPSize: 4096,
Timeout: 5 * time.Second, Timeout: 5 * time.Second,
}, },
addr: s.Addr, port: port,
host: s.Host, host: host,
r: resolver,
}) })
} }
return ret return ret