浏览代码

Fix DNS over QUIC stream close

dyhkwong 6 月之前
父节点
当前提交
63d3e9f6e5
共有 1 个文件被更改,包括 2 次插入2 次删除
  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 {
 	if err != nil {
 		return nil, err
 		return nil, err
 	}
 	}
-	defer stream.Close()
-	defer stream.CancelRead(0)
 	err = transport.WriteMessage(stream, 0, message)
 	err = transport.WriteMessage(stream, 0, message)
 	if err != nil {
 	if err != nil {
+		stream.Close()
 		return nil, err
 		return nil, err
 	}
 	}
+	stream.Close()
 	return transport.ReadMessage(stream)
 	return transport.ReadMessage(stream)
 }
 }