TCP流提取BMP图像显示错位问题排查求助

1. 修正Base64填充逻辑 #

你当前的Base64处理逻辑存在错误：直接移除所有=后仅添加一个=，不符合Base64的填充规则。Base64要求编码后数据长度必须是4的倍数，需根据长度模4的结果补充0-2个=。正确处理代码如下：

import base64

# 清理原始数据：移除换行、空格
tst = sec_orig.replace('\n', '').replace(' ', '')
# 计算所需填充符数量
padding = 4 - len(tst) % 4
if padding != 4:
    tst += '=' * padding

decoded_data = base64.b64decode(tst, altchars=None, validate=True)

错误的填充会导致解码后的二进制数据字节偏移，直接引发BMP像素排列错位。

2. 验证TCP段重组的完整性 #

• 确认TCP段按顺序完整拼接：TCP为流式传输，段的顺序不能混乱，也不能丢包。可对比解码后数据长度与BMP文件头记录的总文件大小是否一致。
• 核对BMP文件头有效性：解码后前2字节必须为BM（十六进制42 4D），若不符合则说明数据提取或解码过程出错。用十六进制编辑器查看decoded_data前54字节（文件头+信息头），重点验证以下字段：
  • 第2-5字节：文件总大小（需与len(decoded_data)相等）
  • 第18-21字节：图像宽度
  • 第22-25字节：图像高度
  • 第28-29字节：位深度（常见24、32等）

3. 检查BMP行对齐补位 #

BMP格式要求每行像素数据字节数为4的倍数，不足部分用0填充。若解码后像素数据缺失这些补位字节，会导致后续行像素偏移。可通过以下代码验证：

import struct

# 从BMP信息头读取宽度、位深度
width = struct.unpack('<I', decoded_data[18:22])[0]
bit_depth = struct.unpack('<H', decoded_data[28:30])[0]

# 计算理论每行字节数与实际每行字节数
bytes_per_pixel = bit_depth // 8
expected_row_bytes = ((width * bytes_per_pixel) + 3) // 4 * 4
# 图像数据从第54字节开始
pixel_data = decoded_data[54:]
actual_row_bytes = len(pixel_data) // abs(struct.unpack('<I', decoded_data[22:26])[0])

if expected_row_bytes != actual_row_bytes:
    print(f"行字节数不匹配：预期{expected_row_bytes}，实际{actual_row_bytes}")

若输出不匹配，说明数据提取或解码时丢失了补位字节，需重新检查TCP段的完整性。

内容的提问来源于stack exchange，提问作者NiceBerg