From bbbe371ea9a827264a92f4efd4fc7f231564cc1a Mon Sep 17 00:00:00 2001
From: Skyxim <noreply@skyxim.com>
Date: Thu, 28 Apr 2022 22:40:06 +0800
Subject: [PATCH] fix: dns specified interface does not change

---
 config/config.go | 2 +-
 dns/client.go    | 7 ++++---
 dns/resolver.go  | 3 ++-
 3 files changed, 7 insertions(+), 5 deletions(-)

diff --git a/config/config.go b/config/config.go
index c6069c69..4a9ce196 100644
--- a/config/config.go
+++ b/config/config.go
@@ -689,7 +689,7 @@ func parseNameServer(servers []string) ([]dns.NameServer, error) {
 				Net:          dnsNetType,
 				Addr:         addr,
 				ProxyAdapter: u.Fragment,
-				Interface:    dialer.DefaultInterface.Load(),
+				Interface:    dialer.DefaultInterface,
 			},
 		)
 	}
diff --git a/dns/client.go b/dns/client.go
index b536f0b1..6415a03a 100644
--- a/dns/client.go
+++ b/dns/client.go
@@ -4,6 +4,7 @@ import (
 	"context"
 	"crypto/tls"
 	"fmt"
+	"go.uber.org/atomic"
 	"net"
 	"net/netip"
 	"strings"
@@ -19,7 +20,7 @@ type client struct {
 	r            *Resolver
 	port         string
 	host         string
-	iface        string
+	iface        *atomic.String
 	proxyAdapter string
 }
 
@@ -49,8 +50,8 @@ func (c *client) ExchangeContext(ctx context.Context, m *D.Msg) (*D.Msg, error)
 	}
 
 	options := []dialer.Option{}
-	if c.iface != "" {
-		options = append(options, dialer.WithInterface(c.iface))
+	if c.iface != nil && c.iface.Load() != "" {
+		options = append(options, dialer.WithInterface(c.iface.Load()))
 	}
 
 	var conn net.Conn
diff --git a/dns/resolver.go b/dns/resolver.go
index 4b4dccc0..b92b500f 100644
--- a/dns/resolver.go
+++ b/dns/resolver.go
@@ -4,6 +4,7 @@ import (
 	"context"
 	"errors"
 	"fmt"
+	"go.uber.org/atomic"
 	"math/rand"
 	"net/netip"
 	"strings"
@@ -356,7 +357,7 @@ func (r *Resolver) HasProxyServer() bool {
 type NameServer struct {
 	Net          string
 	Addr         string
-	Interface    string
+	Interface    *atomic.String
 	ProxyAdapter string
 }