From 38f3a3807373a19cacc0171dd7d5fd11f44dbd35 Mon Sep 17 00:00:00 2001 From: troido Date: Sun, 31 Dec 2017 00:10:14 +0100 Subject: merged roomedge into portal, moved tcommunicate and utils to common folder and fixed package when starting main directory --- asciifarm/common/tcommunicate.py | 32 ++++++++++++++++++++++++++++++++ 1 file changed, 32 insertions(+) create mode 100644 asciifarm/common/tcommunicate.py (limited to 'asciifarm/common/tcommunicate.py') diff --git a/asciifarm/common/tcommunicate.py b/asciifarm/common/tcommunicate.py new file mode 100644 index 0000000..b1fc1b0 --- /dev/null +++ b/asciifarm/common/tcommunicate.py @@ -0,0 +1,32 @@ + +HEADER_SIZE = 4 + + +# this module is for sending discree messages over TCP +# this is achieved by prefixing all messages with their length +# calls to send and recv will also keep attempting to send all data unless this proves impossible + + +def send(sock, msg): + length = len(msg) + header = length.to_bytes(4, byteorder="big") + totalmsg = header + msg + sock.sendall(totalmsg) + +def receive(sock): + header = recvall(sock, 4) #sock.recv(4) + length = int.from_bytes(header, byteorder="big") + return recvall(sock, length) + +def recvall(sock, length): + chunks = [] + bytes_recd = 0 + while bytes_recd < length: + chunk = sock.recv(min(length - bytes_recd, 4096)) + if chunk == b'': + break + #raise RuntimeError("socket connection broken") + chunks.append(chunk) + bytes_recd = bytes_recd + len(chunk) + return b''.join(chunks) + -- cgit