diff options
Diffstat (limited to 'bin/tiny_socket.py')
| -rw-r--r-- | bin/tiny_socket.py | 11 |
1 files changed, 9 insertions, 2 deletions
diff --git a/bin/tiny_socket.py b/bin/tiny_socket.py index 3585ccae295..68a81985666 100644 --- a/bin/tiny_socket.py +++ b/bin/tiny_socket.py @@ -9,9 +9,10 @@ class mysocket: socket.AF_INET, socket.SOCK_STREAM) else: self.sock = sock + self.sock.settimeout(50) def connect(self, host, port): self.sock.connect((host, port)) - def mysend(self, msg): + def mysend(self, msg, exception=False): msg = cPickle.dumps(msg) size = len(msg) self.sock.send('%8d' % size) @@ -22,7 +23,13 @@ class mysocket: raise RuntimeError, "socket connection broken" totalsent = totalsent + sent def myreceive(self): - size = int(self.sock.recv(8)) + buf='' + while len(buf) < 8: + chunk = self.sock.recv(8 - len(buf)) + if chunk == '': + raise RuntimeError, "socket connection broken" + buf += chunk + size = int(buf) msg = '' while len(msg) < size: chunk = self.sock.recv(size-len(msg)) |