aboutsummaryrefslogtreecommitdiffstats
path: root/netlib/tcp.py
diff options
context:
space:
mode:
Diffstat (limited to 'netlib/tcp.py')
-rw-r--r--netlib/tcp.py9
1 files changed, 6 insertions, 3 deletions
diff --git a/netlib/tcp.py b/netlib/tcp.py
index c8545d4f..f6179faa 100644
--- a/netlib/tcp.py
+++ b/netlib/tcp.py
@@ -225,9 +225,12 @@ class Reader(_FileLike):
"""
result = self.read(length)
if length != -1 and len(result) != length:
- raise NetLibIncomplete(
- "Expected %s bytes, got %s" % (length, len(result))
- )
+ if not result:
+ raise NetLibDisconnect()
+ else:
+ raise NetLibIncomplete(
+ "Expected %s bytes, got %s" % (length, len(result))
+ )
return result