如何在python结构中解压缩可变长度数据

Question

我正在构建一个p2p应用程序，在这里我需要从跟踪器中解压udp公告响应。

announce response:

Offset      Size            Name            
0           32-bit integer  action        
4           32-bit integer  transaction_id
8           32-bit integer  interval
12          32-bit integer  leechers
16          32-bit integer  seeders
20 + 6 * n  32-bit integer  IP address
24 + 6 * n  16-bit integer  TCP port
20 + 6 * N

我需要从上表中读取name字段中列出的所有数据，但是种子器、ip地址和端口的值是可变的。

如何使用python正确地从响应中解压所有数据？

Answer 1

struct模块对于解压缩二进制数据非常有用。

action, transaction_id, interval, leechers, seeders = struct.unpack('>iiiii', announse_response)

然后，您必须遍历其余的数据，以获得所有ip/端口数据：

ip_port_list = []
while True:
    try: ip_port_list.append(struct.unpack('>ih', announse_response))
    except: break

如果没有struct，您将不得不逐字节读取，然后转换大的endian问题。

Answer 2

我也经历过同样的挣扎，所以为此编写了一个简单的包，名为联读。

from binread import formatclass, U32, U16, Array 

@formatclass
class Seeder:
    ip_addr = U32
    tcp_port = U16

@formatclass
class AnnounceResponse:
    action = U32
    transaction_id = U32
    interval = U32
    leechers = U32
    seeders = U32

    # `length` refers to the value of the previous `seeders` field
    seeder_data = Array(Seeder, length="seeders")

# `data` is a bytes object
resp = AnnounceResponse.read(data)

print(resp.transaction_id)
print(len(resp.seeder_data))
print(resp.seeder_data[0].tcp_port)

您可以选择使用@formatclass(byteorder='little') (或'big')为所有值指定endianness。

完全公开，我是宾读的作者。