From 9598ed2311ea03dd7ccbdfe5aa5548d9f551fc20 Mon Sep 17 00:00:00 2001 From: shangxu Date: Fri, 13 Nov 2020 18:03:01 +0800 Subject: [PATCH] [PATCH] fix Out Of Memory when use TNonblockingServer Gbp-Pq: Name fix_out_of_memory.patch --- lib/py/src/server/TNonblockingServer.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/lib/py/src/server/TNonblockingServer.py b/lib/py/src/server/TNonblockingServer.py index f62d486..ac06496 100644 --- a/lib/py/src/server/TNonblockingServer.py +++ b/lib/py/src/server/TNonblockingServer.py @@ -151,6 +151,10 @@ class Connection(object): while len(self._rbuf) >= self._reading.end: if self._reading.is_header: mlen, = struct.unpack('!i', self._rbuf[:4]) + if mlen < 0: + logger.error('could not read the head from frame') + self.close() + break self._reading = Message(self._reading.end, mlen, False) self.status = WAIT_MESSAGE else: -- 2.30.2