我正在使用构造2.8 对由一些久违的 Pascal 程序创建的一些文件的头文件进行反向工程。
标题由许多不同的记录组成,其中一些是可选的,我不确定顺序是否固定。
例如,其中两条记录如下所示:
header_record_filetype = cs.Struct(
'record_type' / cs.Int8ub,
'file_type' / cs.PascalString(cs.Int16ub),
'unknown' / cs.Int8ub
)
header_record_user = cs.Struct(
'record_type' / cs.Int8ub,
'user' / cs.PascalString(cs.Int16ub)
)
我已经确定了六个。
我将如何让解析器根据record_type
成员为未知数量的记录选择正确的记录类型,直到遇到类型为 0 的记录(或到达文件末尾)?
我是这样解决的:
header = cs.Struct(
'record_type' / cs.Int8ub,
'record' / cs.Switch(cs.this.record_type, {header_record_type_0x01: header_record_0x01,
header_record_type_filename: header_record_filename,
header_record_type_filetype: header_record_filetype,
header_record_type_user: header_record_user,
header_record_type_end: header_record_end,
header_record_type_image_metadata: header_record_image_metadata},
default=header_record_end
),
'offset' / cs.Tell
)
with open(sys.argv[1], 'rb') as f:
h = f.read(2048)
index = 0
record_type = h[index]
while record_type != 0:
record = header.parse(h[index:])
print(record)
index += record.offset
record_type = record.record_type
但我不知道这是否是最好的*方法。
*对于“最佳”的某些价值。
我发现 RepeatUntil() 构造隐藏在帮助页面的底部。所以现在我有这个:
header = cs.Struct(
'type' / cs.Enum(cs.Int8ub,
file_metadata=0x01,
filename=0x02,
file_type=0x03,
user=0x0A,
image_metadata=0x10,
end=0xFF),
'record' / cs.Switch(cs.this.type, {'file_metadata': header_record_file_metadata,
'filename': header_record_filename,
'file_type': header_record_filetype,
'user': header_record_user,
'end': header_record_end,
'image_metadata': header_record_image_metadata}),
'size' / cs.Tell
)
with open(sys.argv[1], 'rb') as f:
h = f.read(2048)
records = cs.RepeatUntil(lambda obj, lst, ctx: obj.type == 'end', header).parse(h)
print(records)
这感觉更干净,更符合construct的声明性。
本文收集自互联网,转载请注明来源。
如有侵权,请联系 [email protected] 删除。
我来说两句