feat(executor): reset Dns first on closing

igoogolx · Jan 12, 2025 · 9702483 · 9702483
1 parent 316c8f4
commit 9702483
Showing 1 changed file with 7 additions and 6 deletions.
diff --git a/internal/executor/tun.go b/internal/executor/tun.go
@@ -71,6 +71,12 @@ func (c *TunClient) Start() error {
 func (c *TunClient) Close() error {
 	var err error
 
+	if c.config.HijackDns.Enabled {
+		err := dns.Resume(c.config.HijackDns.NetworkService, c.config.HijackDns.AlwaysReset)
+		if err != nil {
+			return err
+		}
+	}
 	statistic.DefaultManager.CloseAllConnections()
 	if err = c.tun.Close(); err != nil {
 		return err
@@ -79,12 +85,7 @@ func (c *TunClient) Close() error {
 	if err != nil {
 		return err
 	}
-	if c.config.HijackDns.Enabled {
-		err := dns.Resume(c.config.HijackDns.NetworkService, c.config.HijackDns.AlwaysReset)
-		if err != nil {
-			return err
-		}
-	}
+
 	if err = c.localserver.Close(); err != nil {
 		return err
 	}