M-Labs
/
artiq
mirror of https://github.com/m-labs/artiq.git
2
0
Fork 2
Code Issues Projects Releases Wiki Activity

coredevice/comm_tcp: raise exception on connection closed

This commit is contained in:
Sebastien Bourdeauducq 2015-04-29 11:18:51 +08:00
parent 283695e8aa
commit e5f7fcc339
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
artiq/coredevice/comm_tcp.py
View File

@ -30,7 +30,10 @@ class Comm(CommGeneric, AutoDB):
def read(self, length): def read(self, length):
r = bytes() r = bytes()
while len(r) < length: while len(r) < length:
r += self.socket.recv(min(8192, length - len(r))) rn = self.socket.recv(min(8192, length - len(r)))
if not rn:
raise IOError("Connection closed")
r += rn
return r return r
def write(self, data): def write(self, data):