From d2c10f31b0a911128b3f2099c9ffc221bad49a3c Mon Sep 17 00:00:00 2001 From: Arturo Hernandez Date: Fri, 31 Aug 2018 20:17:28 -0400 Subject: [PATCH] better order on fix of small TCP DATA --- zk/base.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/zk/base.py b/zk/base.py index b54d64b..aa387da 100644 --- a/zk/base.py +++ b/zk/base.py @@ -1340,8 +1340,8 @@ class ZK(object): if self.verbose: print ("_rc_DATA! is {} bytes, tcp length is {}".format(len(self.__data), self.__tcp_length)) if len(self.__data) < (self.__tcp_length - 8): need = (self.__tcp_length - 8) - len(self.__data) - more_data = self.__recieve_raw_data(need) if self.verbose: print ("need more data: {}".format(need)) + more_data = self.__recieve_raw_data(need) return b''.join([self.__data, more_data]) else: #enough data if self.verbose: print ("Enough data") @@ -1438,8 +1438,8 @@ class ZK(object): if self.verbose: print ("DATA! is {} bytes, tcp length is {}".format(len(self.__data), self.__tcp_length)) if len(self.__data) < (self.__tcp_length - 8): need = (self.__tcp_length - 8) - len(self.__data) - more_data = self.__recieve_raw_data(need) if self.verbose: print ("need more data: {}".format(need)) + more_data = self.__recieve_raw_data(need) return b''.join([self.__data, more_data]), len(self.__data) + len(more_data) else: #enough data if self.verbose: print ("Enough data")