This commit is contained in:
破娃酱 2017-04-05 22:11:55 +08:00
parent d82c931f8e
commit 862fa890ee

View file

@ -761,7 +761,7 @@ class TCPRelayHandler(object):
def _get_read_size(self, sock, recv_buffer_size):
if self._overhead == 0:
return buffer_size
return recv_buffer_size
buffer_size = len(sock.recv(recv_buffer_size, socket.MSG_PEEK))
if buffer_size == recv_buffer_size:
return buffer_size