解压以ASCIIZ字符串结尾的结构

Question

我试图使用struct.unpack()来拆分以ASCII字符串结尾的数据记录。在

记录（正好是tomtomov2记录）具有以下格式（存储的little endian）：

• 1字节
• 4字节int表示总记录大小（包括此字段）
• 4字节整数
• 4字节整数
• 可变长度字符串，以null结尾

unpack()要求字符串的长度包含在传递给它的格式中。我可以使用第二个字段和记录其余部分的已知大小（13个字节）来获得字符串长度：

str_len = struct.unpack("<xi", record[:5])[0] - 13
fmt = "<biii{0}s".format(str_len)

然后继续进行完整的解包，但是由于字符串是以null结尾的，我真的希望unpack()能帮我完成。如果我在一个不包含自身大小的结构中运行，也会很好。在

我怎么能做到呢？在

Answer 1

实际上，由于struct.calcsize()将告诉您它期望的长度，因此无大小记录相当容易处理。您可以使用它和数据的实际长度为unpack()构造一个新的格式字符串，其中包含正确的字符串长度。在

此函数只是unpack()的包装，允许在最后一个位置添加一个新的格式字符，该字符将删除终端NUL:

import struct
def unpack_with_final_asciiz(fmt, dat):
    """
    Unpack binary data, handling a null-terminated string at the end 
    (and only at the end) automatically.

    The first argument, fmt, is a struct.unpack() format string with the 
    following modfications:
    If fmt's last character is 'z', the returned string will drop the NUL.
    If it is 's' with no length, the string including NUL will be returned.
    If it is 's' with a length, behavior is identical to normal unpack().
    """
    # Just pass on if no special behavior is required
    if fmt[-1] not in ('z', 's') or (fmt[-1] == 's' and fmt[-2].isdigit()):
        return struct.unpack(fmt, dat)

    # Use format string to get size of contained string and rest of record
    non_str_len = struct.calcsize(fmt[:-1])
    str_len = len(dat) - non_str_len

    # Set up new format string
    # If passed 'z', treat terminating NUL as a "pad byte"
    if fmt[-1] == 'z':
        str_fmt = "{0}sx".format(str_len - 1)
    else:
        str_fmt = "{0}s".format(str_len)
    new_fmt = fmt[:-1] + str_fmt

    return struct.unpack(new_fmt, dat)

---

^{pr2}$