This commit is contained in:
Larry Hastings 2014-05-18 23:19:55 -07:00
commit debcb9dd54
1 changed files with 7 additions and 5 deletions

View File

@ -204,13 +204,15 @@ length message::
totalsent = totalsent + sent
def myreceive(self):
msg = b''
while len(msg) < MSGLEN:
chunk = self.sock.recv(MSGLEN-len(msg))
chunks = []
bytes_recd = 0
while bytes_recd < MSGLEN:
chunk = self.sock.recv(min(MSGLEN - bytes_recd, 2048))
if chunk == b'':
raise RuntimeError("socket connection broken")
msg = msg + chunk
return msg
chucks.append(chunk)
bytes_recd = bytes_recd + len(chunk)
return b''.join(chunks)
The sending code here is usable for almost any messaging scheme - in Python you
send strings, and you can use ``len()`` to determine its length (even if it has