don't double-eof

This commit is contained in:
ed 2021-09-09 00:54:09 +02:00
parent 524a3ba566
commit 6136b9bf9c

View file

@ -192,7 +192,7 @@ class _Unrecv(object):
def recv_ex(self, nbytes): def recv_ex(self, nbytes):
"""read an exact number of bytes""" """read an exact number of bytes"""
ret = self.recv(nbytes) ret = self.recv(nbytes)
while len(ret) < nbytes: while ret and len(ret) < nbytes:
buf = self.recv(nbytes - len(ret)) buf = self.recv(nbytes - len(ret))
if not buf: if not buf:
break break
@ -233,7 +233,7 @@ class _LUnrecv(object):
def recv_ex(self, nbytes): def recv_ex(self, nbytes):
"""read an exact number of bytes""" """read an exact number of bytes"""
ret = self.recv(nbytes) ret = self.recv(nbytes)
while len(ret) < nbytes: while ret and len(ret) < nbytes:
buf = self.recv(nbytes - len(ret)) buf = self.recv(nbytes - len(ret))
if not buf: if not buf:
break break