fumiama/terasu
1
0
Fork 0
mirror of https://github.com/fumiama/terasu.git synced 2026-06-21 12:50:24 +08:00
Code Activity

fix(dns): fast-failed on RST

Browse Source
This commit is contained in:
源文雨
2025-10-03 14:47:40 +08:00
parent 1d9b679c36
commit 9974bdca12
9 changed files with 86 additions and 110 deletions
Download Patch File Download Diff File Expand all files Collapse all files

31
dns/cache.go Normal file
View File

@@ -0,0 +1,31 @@
package dns
import (
"context"
"time"
"github.com/FloatTech/ttl"
"github.com/fumiama/terasu/ip"
)
var lookupTable = ttl.NewCache[string, []string](time.Hour)
// LookupHost use default resolver with its fallback
func LookupHost(ctx context.Context, host string) (addrs []string, err error) {
addrs = lookupTable.Get(host)
if len(addrs) == 0 {
addrs, err = DefaultResolver.LookupHost(ctx, host)
if err != nil {
if ip.IsIPv6Available.Get() {
addrs, err = IPv6Servers.lookupHostDoH(ctx, host)
} else {
addrs, err = IPv4Servers.lookupHostDoH(ctx, host)
}
if err != nil {
return nil, err
}
}
lookupTable.Set(host, addrs)
}
return
}

34
dns/dns.go
View File

@@ -14,15 +14,16 @@ import (
)
var (
// ErrNoDNSAvailable is reported when all servers failed to response
ErrNoDNSAvailable = errors.New("no dns available")
)
var defaultDialer = net.Dialer{
var dnsDialer = net.Dialer{
Timeout: time.Second * 4,
}
func SetTimeout(t time.Duration) {
defaultDialer.Timeout = t
dnsDialer.Timeout = t
}
type dnsstat struct {
@@ -88,7 +89,7 @@ func (ds *DNSList) Add(c *DNSConfig) {
}
}
func (ds *DNSList) LookupHostFallback(ctx context.Context, host string) ([]string, error) {
func (ds *DNSList) lookupHostDoH(ctx context.Context, host string) ([]string, error) {
ds.RLock()
defer ds.RUnlock()
// try to use DoH first
@@ -97,7 +98,7 @@ func (ds *DNSList) LookupHostFallback(ctx context.Context, host string) ([]strin
if !addr.e || !strings.HasPrefix(addr.a, "https://") { // disabled or is not DoH
continue
}
jr, err := lookupdoh(addr.a, host)
jr, err := lookupdoh(ctx, addr.a, host)
if err == nil {
hosts := jr.hosts()
if len(hosts) > 0 {
@@ -110,37 +111,34 @@ func (ds *DNSList) LookupHostFallback(ctx context.Context, host string) ([]strin
if addrs, ok := ds.b[host]; ok {
return addrs, nil
}
return net.DefaultResolver.LookupHost(ctx, host)
return nil, ErrNoDNSAvailable
}
func (ds *DNSList) DialContext(ctx context.Context, dialer *net.Dialer, firstFragmentLen uint8) (tlsConn *tls.Conn, err error) {
err = ErrNoDNSAvailable
if dialer == nil {
dialer = &defaultDialer
dialer = &dnsDialer
}
ds.RLock()
defer ds.RUnlock()
if dialer.Timeout != 0 {
var cancel context.CancelFunc
ctx, cancel = context.WithTimeout(ctx, dialer.Timeout)
defer cancel()
}
if !dialer.Deadline.IsZero() {
var cancel context.CancelFunc
ctx, cancel = context.WithDeadline(ctx, dialer.Deadline)
defer cancel()
}
var conn net.Conn
for host, addrs := range ds.m {
for _, addr := range addrs {
if !addr.e || strings.HasPrefix(addr.a, "https://") { // disabled or is DoH
continue
}
if dialer.Timeout != 0 {
var cancel context.CancelFunc
ctx, cancel = context.WithTimeout(context.Background(), dialer.Timeout)
defer cancel()
} else if !dialer.Deadline.IsZero() {
var cancel context.CancelFunc
ctx, cancel = context.WithDeadline(context.Background(), dialer.Deadline)
defer cancel()
}
conn, err = dialer.DialContext(ctx, "tcp", addr.a)
if err != nil {
addr.e = false // no need to acquire write lock

4
dns/dns_test.go
View File

@@ -28,7 +28,7 @@ func TestResolverFallback(t *testing.T) {
t.Log("IsIPv6Available:", ip.IsIPv6Available.Get())
if ip.IsIPv6Available.Get() {
addrs, err := IPv6Servers.LookupHostFallback(context.TODO(), "huggingface.co")
addrs, err := IPv6Servers.lookupHostDoH(context.TODO(), "huggingface.co")
if err != nil {
t.Fatal(err)
}
@@ -37,7 +37,7 @@ func TestResolverFallback(t *testing.T) {
t.Fail()
}
}
addrs, err := IPv4Servers.LookupHostFallback(context.TODO(), "huggingface.co")
addrs, err := IPv4Servers.lookupHostDoH(context.TODO(), "huggingface.co")
if err != nil {
t.Fatal(err)
}

30
dns/doh.go
View File

@@ -10,9 +10,7 @@ import (
"net/url"
"strconv"
"strings"
"time"
"github.com/FloatTech/ttl"
"golang.org/x/net/http2"
"github.com/fumiama/terasu"
@@ -65,23 +63,9 @@ func (jr *dohjsonresponse) hosts() []string {
return hosts
}
var lookupTable = ttl.NewCache[string, []string](time.Hour)
var trsHTTP2ClientWithSystemDNS = http.Client{
Transport: &http2.Transport{
DialTLSContext: func(ctx context.Context, network, addr string, cfg *tls.Config) (net.Conn, error) {
if defaultDialer.Timeout != 0 {
var cancel context.CancelFunc
ctx, cancel = context.WithTimeout(ctx, defaultDialer.Timeout)
defer cancel()
}
if !defaultDialer.Deadline.IsZero() {
var cancel context.CancelFunc
ctx, cancel = context.WithDeadline(ctx, defaultDialer.Deadline)
defer cancel()
}
host, port, err := net.SplitHostPort(addr)
if err != nil {
return nil, err
@@ -100,7 +84,7 @@ var trsHTTP2ClientWithSystemDNS = http.Client{
var conn net.Conn
var tlsConn *tls.Conn
for _, a := range addrs {
conn, err = defaultDialer.DialContext(ctx, network, net.JoinHostPort(a, port))
conn, err = dnsDialer.DialContext(ctx, network, net.JoinHostPort(a, port))
if err != nil {
continue
}
@@ -111,7 +95,7 @@ var trsHTTP2ClientWithSystemDNS = http.Client{
}
_ = tlsConn.Close()
tlsConn = nil
conn, err = defaultDialer.DialContext(ctx, network, net.JoinHostPort(a, port))
conn, err = dnsDialer.DialContext(ctx, network, net.JoinHostPort(a, port))
if err != nil {
continue
}
@@ -128,18 +112,18 @@ var trsHTTP2ClientWithSystemDNS = http.Client{
},
}
func lookupdoh(server, u string) (jr dohjsonresponse, err error) {
jr, err = lookupdohwithtype(server, u, preferreddohtype())
func lookupdoh(ctx context.Context, server, u string) (jr dohjsonresponse, err error) {
jr, err = lookupdohwithtype(ctx, server, u, preferreddohtype())
if err == nil {
return
}
if ip.IsIPv6Available.Get() {
jr, err = lookupdohwithtype(server, u, recordTypeA)
jr, err = lookupdohwithtype(ctx, server, u, recordTypeA)
}
return
}
func lookupdohwithtype(server, u string, typ recordType) (jr dohjsonresponse, err error) {
func lookupdohwithtype(ctx context.Context, server, u string, typ recordType) (jr dohjsonresponse, err error) {
sb := strings.Builder{}
sb.WriteString(server)
sb.WriteString("?name=")
@@ -148,7 +132,7 @@ func lookupdohwithtype(server, u string, typ recordType) (jr dohjsonresponse, er
sb.WriteString("&type=")
sb.WriteString(strconv.Itoa(int(typ)))
}
req, err := http.NewRequest("GET", sb.String(), nil)
req, err := http.NewRequestWithContext(ctx, "GET", sb.String(), nil)
if err != nil {
return
}