KERNEL: initial add of tinysocket

bzr revid: ced-46197cd2df22dc3b22288be8053614f71a95bf05

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))