Bug fixes for fakeDNS

v2fly · Jan 27, 2021 · 5248416 · 5248416
1 parent a1a4ab1
commit 5248416
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/app/dns/dns.go b/app/dns/dns.go
@@ -190,7 +190,7 @@ func (s *DNS) lookupIPInternal(domain string, option IPOption) ([]net.IP, error)
 
 	// Name servers lookup
 	errs := []error{}
-	ctx := session.ContextWithInbound(context.Background(), &session.Inbound{Tag: s.tag})
+	ctx := session.ContextWithInbound(s.ctx, &session.Inbound{Tag: s.tag})
 	for _, client := range s.sortClients(domain) {
 		ips, err := client.QueryIP(ctx, domain, option)
 		if len(ips) > 0 {

diff --git a/app/dns/nameserver_fakedns.go b/app/dns/nameserver_fakedns.go
@@ -21,9 +21,8 @@ func (f FakeDNSServer) Name() string {
 
 func (f *FakeDNSServer) QueryIP(ctx context.Context, domain string, clientIP net.IP, option IPOption) ([]net.IP, error) {
 	if f.fakeDNSEngine == nil {
-		var fakeDNSEngine dns.FakeDNSEngine
 		if err := core.RequireFeatures(ctx, func(fdns dns.FakeDNSEngine) {
-			fakeDNSEngine = fdns
+			f.fakeDNSEngine = fdns
 		}); err != nil {
 			return nil, newError("Unable to locate a fake DNS Engine").Base(err).AtError()
 		}