upstream: close plain connections

AdguardTeam · Jun 8, 2023 · 6687dac · 6687dac
1 parent 4162e21
commit 6687dac
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/upstream/upstream_plain.go b/upstream/upstream_plain.go
@@ -106,13 +106,15 @@ func (p *plainDNS) dialExchange(
 	if err != nil {
 		return nil, fmt.Errorf("dialing %s over %s: %w", p.addr.Host, network, err)
 	}
+	defer func(c net.Conn) { err = errors.WithDeferred(err, c.Close()) }(conn.Conn)
 
 	resp, _, err = client.ExchangeWithConn(req, conn)
 	if isExpectedConnErr(err) {
 		conn.Conn, err = dial(ctx, string(network), "")
 		if err != nil {
 			return nil, fmt.Errorf("dialing %s over %s again: %w", p.addr.Host, network, err)
 		}
+		defer func(c net.Conn) { err = errors.WithDeferred(err, c.Close()) }(conn.Conn)
 
 		resp, _, err = client.ExchangeWithConn(req, conn)
 	}