Merge pull request #1250 from mmerickel/fix/1211

do not assume that read will consume the number of bytes requested
This commit is contained in:
Joffrey F 2016-11-28 14:00:32 -08:00 committed by GitHub
commit 52376e4b24
1 changed files with 7 additions and 3 deletions

View File

@ -69,7 +69,11 @@ def frames_iter(socket):
"""
Returns a generator of frames read from socket
"""
n = next_frame_size(socket)
while n > 0:
yield read(socket, n)
while True:
n = next_frame_size(socket)
if n == 0:
break
while n > 0:
result = read(socket, n)
n -= len(result)
yield result