Browse Source

Fix DNS over QUIC stream close

dyhkwong 6 months ago
parent
commit
63d3e9f6e5
1 changed files with 2 additions and 2 deletions
  1. 2 2
      dns/transport/quic/quic.go

+ 2 - 2
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)
 }