|
|
|
@ -1,14 +1,20 @@
|
|
|
|
|
"""Various I/O and serialization utilities."""
|
|
|
|
|
import io
|
|
|
|
|
import struct
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def send(conn, data, fmt=None):
|
|
|
|
|
if fmt:
|
|
|
|
|
data = struct.pack(fmt, *data)
|
|
|
|
|
def send(conn, data):
|
|
|
|
|
"""Send data blob to connection socket."""
|
|
|
|
|
conn.sendall(data)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def recv(conn, size):
|
|
|
|
|
"""
|
|
|
|
|
Receive bytes from connection socket or stream.
|
|
|
|
|
|
|
|
|
|
If size is struct.calcsize()-compatible format, use it to unpack the data.
|
|
|
|
|
Otherwise, return the plain blob as bytes.
|
|
|
|
|
"""
|
|
|
|
|
try:
|
|
|
|
|
fmt = size
|
|
|
|
|
size = struct.calcsize(fmt)
|
|
|
|
@ -34,11 +40,13 @@ def recv(conn, size):
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def read_frame(conn):
|
|
|
|
|
"""Read size-prefixed frame from connection."""
|
|
|
|
|
size, = recv(conn, '>L')
|
|
|
|
|
return recv(conn, size)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def bytes2num(s):
|
|
|
|
|
"""Convert MSB-first bytes to an unsigned integer."""
|
|
|
|
|
res = 0
|
|
|
|
|
for i, c in enumerate(reversed(bytearray(s))):
|
|
|
|
|
res += c << (i * 8)
|
|
|
|
@ -46,6 +54,7 @@ def bytes2num(s):
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def num2bytes(value, size):
|
|
|
|
|
"""Convert an unsigned integer to MSB-first bytes with specified size."""
|
|
|
|
|
res = []
|
|
|
|
|
for _ in range(size):
|
|
|
|
|
res.append(value & 0xFF)
|
|
|
|
@ -55,10 +64,12 @@ def num2bytes(value, size):
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def pack(fmt, *args):
|
|
|
|
|
"""Serialize MSB-first message."""
|
|
|
|
|
return struct.pack('>' + fmt, *args)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def frame(*msgs):
|
|
|
|
|
"""Serialize MSB-first length-prefixed frame."""
|
|
|
|
|
res = io.BytesIO()
|
|
|
|
|
for msg in msgs:
|
|
|
|
|
res.write(msg)
|
|
|
|
|