ioutil.py

# ioutil.py - Just some simple utilities for reading/writing C-style data.
# Every project has to have it's own little utility functions. It sucks but
# it's better than repeating yourself ad nauseum.

import struct

# read_struct reads struct data from a file-like object
def read_struct(file, fmt):
    size = struct.calcsize(fmt)
    data = file.read(size)

    if len(data) < size:
        print('warning: short struct read (%i < %i)' % (len(data), size))

    return struct.unpack(fmt, data)


# write_struct writes struct data to a file-like object
def write_struct(file, fmt, *args):
    file.write(struct.pack(fmt, *args))


# read_cstr reads a C-style string (and returns a `bytes` object.)
def read_cstr(file):
    buf = bytearray()
    while True:
        b = file.read(1)
        if b == '' or b == b'\x00':
            return bytes(buf)
        else:
            buf.append(ord(b))

# write_cstr writes a C-style string
def write_cstr(file, buf):
    file.write(buf + '\x00')