techwire/copyparty
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

don't double-eof

Browse Source
This commit is contained in:
ed 2021-09-09 00:54:09 +02:00
parent 524a3ba566
commit 6136b9bf9c
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
copyparty/util.py
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