[thrift] branch master updated: fix Out Of Memory when use TNonblockingServer

[je...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

jensg pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/thrift.git


The following commit(s) were added to refs/heads/master by this push:
     new b664cfe25 fix Out Of Memory when use TNonblockingServer
b664cfe25 is described below

commit b664cfe2533e4bbf00fd5e7e0211bf7161ee2a04
Author: shangxu <sh...@domob.cn>
AuthorDate: Fri Nov 13 18:03:01 2020 +0800

    fix Out Of Memory when use TNonblockingServer
---
 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 f62d486eb..ac0649651 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: