python - 使用通配符在 bytearray 中查找 bytearray

Question

我有一些大的二进制文件，我需要在其中搜索特定的字节序列，例如:

find_bytes = bytearray(base64.b16decode('a226fb42'))
with open(filename, "rb") as f:
    file_bytes = bytearray(f.read())
    found_pos = file_bytes.find(find_bytes, 0)

这很好用，除了现在我希望能够将序列中的特定字节值(例如 00 或 FF)表示为匹配任何字节的通配符，因此例如 a2000042 应该匹配任何以 a2 开始并以 42 结束的 4 字节序列。

有没有办法扩展find 方法来做到这一点，或者有更好的解决方案？

使用 Python 2.7 但愿意在必要时切换..

Answer 1

您可以使用正则表达式(它们适用于字节数组):

>>> import re
>>> bytes = bytearray('\x01\x02\x03\x04\x05')
>>> re.search(b'\x02.\x04',bytes).group(0)
'\x02\x03\x04'

只需使用“.”作为通配符。

但是对于非常大的文件可能会导致问题，因为需要先将整个文件加载到一个字符串中。