Fix QUIC defragger

This commit is contained in:
世界
2023-09-07 19:26:45 +08:00
parent 806f7d0a2b
commit ff209471d8
10 changed files with 26 additions and 78 deletions

View File

@@ -34,7 +34,7 @@ func (c *Client) handleMessage(conn *clientQUICConnection, data []byte) error {
}
switch data[1] {
case CommandPacket:
message := udpMessagePool.Get().(*udpMessage)
message := allocMessage()
err := decodeUDPMessage(message, data[2:])
if err != nil {
message.release()
@@ -82,7 +82,7 @@ func (c *Client) handleUniStream(conn *clientQUICConnection, stream quic.Receive
return E.New("unknown command ", command)
}
reader := io.MultiReader(bufio.NewCachedReader(stream, buffer), stream)
message := udpMessagePool.Get().(*udpMessage)
message := allocMessage()
err = readUDPMessage(message, reader)
if err != nil {
message.release()