From 6136b9bf9cfd7ffd40c6abc2affb5fa28b760b98 Mon Sep 17 00:00:00 2001 From: ed Date: Thu, 9 Sep 2021 00:54:09 +0200 Subject: [PATCH] don't double-eof --- copyparty/util.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/copyparty/util.py b/copyparty/util.py index ea2ad0cb..32939382 100644 --- a/copyparty/util.py +++ b/copyparty/util.py @@ -192,7 +192,7 @@ class _Unrecv(object): def recv_ex(self, nbytes): """read an exact number of bytes""" ret = self.recv(nbytes) - while len(ret) < nbytes: + while ret and len(ret) < nbytes: buf = self.recv(nbytes - len(ret)) if not buf: break @@ -233,7 +233,7 @@ class _LUnrecv(object): def recv_ex(self, nbytes): """read an exact number of bytes""" ret = self.recv(nbytes) - while len(ret) < nbytes: + while ret and len(ret) < nbytes: buf = self.recv(nbytes - len(ret)) if not buf: break