From c22210da47a64f9c27d62eb648e75d8d56642096 Mon Sep 17 00:00:00 2001
From: dyhkwong <50692134+dyhkwong@users.noreply.github.com>
Date: Thu, 27 Mar 2025 10:19:57 +0800
Subject: [PATCH] Fix DNS over QUIC stream close

---
 dns/transport/quic/quic.go | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/dns/transport/quic/quic.go b/dns/transport/quic/quic.go
index fc5101ee..18f8b9fe 100644
--- a/dns/transport/quic/quic.go
+++ b/dns/transport/quic/quic.go
@@ -140,12 +140,12 @@ func (t *Transport) exchange(ctx context.Context, message *mDNS.Msg, conn quic.C
 	if err != nil {
 		return nil, err
 	}
-	defer stream.Close()
-	defer stream.CancelRead(0)
 	err = transport.WriteMessage(stream, 0, message)
 	if err != nil {
+		stream.Close()
 		return nil, err
 	}
+	stream.Close()
 	return transport.ReadMessage(stream)
 }