From a01de02f9a44a9ed6cd5ab9d1621be50d995a5b8 Mon Sep 17 00:00:00 2001 From: hebo Date: Sun, 29 Sep 2019 09:21:55 +0800 Subject: [PATCH] fix bug of read short packet --- session-dealer/mysql/session.go | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/session-dealer/mysql/session.go b/session-dealer/mysql/session.go index 11c0706..6640479 100644 --- a/session-dealer/mysql/session.go +++ b/session-dealer/mysql/session.go @@ -143,14 +143,14 @@ func (ms *MysqlSession) readFromClient(seqID int64, bytes []byte) { contentSize := int64(len(bytes)) if ms.expectReceiveSize == -1 { - ms.expectReceiveSize = extractMysqlPayloadSize(bytes[:4]) - // ignore too big mysql packet - if ms.expectReceiveSize >= MaxMysqlPacketLen { + // ignore invalid head package + if len(bytes) <= 4{ return } - // ignore invalid head package - if len(bytes) <= 4{ + ms.expectReceiveSize = extractMysqlPayloadSize(bytes[:4]) + // ignore too big mysql packet + if ms.expectReceiveSize >= MaxMysqlPacketLen { return }