用Python读写二进制文件

Question

以下代码似乎没有正确读/写二进制形式.它应该读取一个二进制文件,逐位XOR数据并将其写回文件.没有任何语法错误,但数据无法验证,我已通过其他工具测试源数据以确认xor密钥.

更新:根据评论中的反馈,这很可能是由于我正在测试的系统的字节序.

xortools.py:

def four_byte_xor(buf, key):
    out = ''
    for i in range(0,len(buf)/4):
        c = struct.unpack("=I", buf[(i*4):(i*4)+4])[0]
        c ^= key
        out += struct.pack("=I", c)
    return out

调用xortools.py:

from xortools import four_byte_xor
in_buf = open('infile.bin','rb').read()
out_buf = open('outfile.bin','wb')
out_buf.write(four_byte_xor(in_buf, 0x01010101))
out_buf.close()

看来我需要读取每个答案的字节数.由于上面的函数操作多个字节,上面的函数如何合并到下面？或者没关系？我需要使用struct吗？

with open("myfile", "rb") as f:
    byte = f.read(1)
    while byte:
        # Do stuff with byte.
        byte = f.read(1)

例如,以下文件有4个重复字节,01020304:

使用01020304的密钥对数据进行异或操作,该密钥将原始字节归零:

这是对原始函数的尝试,在这种情况下,05010501的结果是不正确的:

Answer 1

试试这个功能：

def four_byte_xor(buf, key):
    outl = []
    for i in range(0, len(buf), 4):
        chunk = buf[i:i+4]
        v = struct.unpack(b"=I", chunk)[0]
        v ^= key
        outl.append(struct.pack(b"=I", v))
    return b"".join(outl)

我不确定你是否真的接受了 4 个字节的输入，但我没有尝试破译它。假设您的输入可以被 4 整除。

编辑，基于新输入的新功能：

def four_byte_xor(buf, key):
    key = struct.pack(b">I", key)
    buf = bytearray(buf)
    for offset in range(0, len(buf), 4):
        for i, byte in enumerate(key):
            buf[offset + i] = chr(buf[offset + i] ^ ord(byte))
    return str(buf)

这可能可以改进，但它确实提供了正确的输出。