From d4117504ebd7b93f97bf02b38e51d3bcf3ba675a Mon Sep 17 00:00:00 2001 From: Toby Date: Mon, 28 Feb 2022 14:26:52 -0800 Subject: [PATCH] fix: incorrect UDP frag sendErr handling --- pkg/core/server_client.go | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/pkg/core/server_client.go b/pkg/core/server_client.go index eff83e2..73a1ba3 100644 --- a/pkg/core/server_client.go +++ b/pkg/core/server_client.go @@ -343,9 +343,9 @@ func (c *serverClient) handleUDP(stream quic.Stream) { } // try no frag first _ = struc.Pack(&msgBuf, &msg) - err = c.CS.SendMessage(msgBuf.Bytes()) - if err != nil { - if errSize, ok := err.(quic.ErrMessageToLarge); ok { + sendErr := c.CS.SendMessage(msgBuf.Bytes()) + if sendErr != nil { + if errSize, ok := sendErr.(quic.ErrMessageToLarge); ok { // need to frag msg.MsgID = uint16(rand.Intn(0xFFFF)) + 1 // msgID must be > 0 when fragCount > 1 fragMsgs := fragUDPMessage(msg, int(errSize))