python-3.x - 如何在 Python 3 中填充字节序列？

Question

我正在尝试加密 PDF 或可执行文件等文件，但我无法填充它。

我尝试用这种方法读取文件:

with('file', 'rb') as file_read:
    line = file_read.read(n) --> n multple of 16
    encrypt(line, key)

然后我将该行传递给我的 encrypt 函数，但是当我到达不是 16 的倍数的行时，我的程序崩溃了。

如何正确填充字节序列？

Answer 1

这是先前答案的简化(编辑)版本。

---

假设使用 line = f.read(N)(其中 N 是 block 大小)在循环中读取文件，直到 EOF。

1) 平凡的零填充；只需在 read 之后添加:

elen = len(line) % N
if elen:
    line += bytes(N - elen)

零填充是最简单的，但也有缺点。

---

2) PKCS#7 填充，N < 256；在 read 之后添加它并确保循环将在之后退出:

if len(line) < N:
    elen = 1 + (len(line) - 1) % N
    line += bytes(elen for _ in range(elen))

---

请注意，从文件读取与从网络套接字读取不同。需要一个缓冲区来从网络读取固定长度的 block 。