AES加密算法实战：从位级雪崩测试到性能基准对比

1次阅读

共计 2444 个字符，预计需要花费 7 分钟才能阅读完成。

AES（Advanced Encryption Standard）是一种广泛使用的对称加密算法，由 NIST 于 2001 年发布。它支持 128、192 和 256 位密钥长度，采用替换 - 置换网络（SPN）结构，具有加密强度高、实现效率好的特点。在 HTTPS、VPN、文件加密等场景中都有广泛应用。

许多开发者在实现加密系统时容易陷入两个误区：

只关注功能实现，忽视算法特性验证 ：加密算法需要满足严格的数学特性，其中位级雪崩效应（改变 1 位输入导致约 50% 输出位变化）是衡量扩散特性的重要指标
缺乏性能基准数据 ：不同实现方式（如 ECB/CBC 模式）和不同平台上的性能可能差异显著，影响系统设计决策

# 安装依赖库
pip install pycryptodome

from Crypto.Cipher import AES
from Crypto.Util.Padding import pad, unpad
from Crypto.Random import get_random_bytes

class AESCipher:
    def __init__(self, key=None):
        self.key = key if key else get_random_bytes(16)  # 默认 128 位密钥
        self.block_size = AES.block_size

    def encrypt(self, plaintext, mode=AES.MODE_CBC):
        iv = get_random_bytes(self.block_size)
        cipher = AES.new(self.key, mode, iv)
        ciphertext = cipher.encrypt(pad(plaintext, self.block_size))
        return iv + ciphertext

    def decrypt(self, ciphertext, mode=AES.MODE_CBC):
        iv = ciphertext[:self.block_size]
        cipher = AES.new(self.key, mode, iv)
        plaintext = unpad(cipher.decrypt(ciphertext[self.block_size:]), self.block_size)
        return plaintext

def avalanche_test(cipher, plaintext, iterations=1000):
    """
    计算平均雪崩效应比例
    :return: 平均变化比例 (0.0~1.0)
    """
    total_changed = 0
    original = cipher.encrypt(plaintext)

    for _ in range(iterations):
        # 翻转 1 个随机位
        modified = bytearray(plaintext)
        byte_pos = getrandbits(8) % len(modified)
        bit_mask = 1 << (getrandbits(8) % 8)
        modified[byte_pos] ^= bit_mask

        # 计算变化位数
        changed = sum(bin(a^b).count('1') for a,b in zip(
            original, 
            cipher.encrypt(bytes(modified))
        ))
        total_changed += changed / (len(original)*8)  # 变化比例

    return total_changed / iterations

# 测试示例
aes = AESCipher()
test_data = b'This is avalanche test data'
print(f"雪崩效应比例: {avalanche_test(aes, test_data):.2%}")
# 期望输出: 雪崩效应比例: 48.25%~51.75%

import timeit
from tabulate import tabulate

def benchmark():
    data = get_random_bytes(1024*1024)  # 1MB 测试数据
    modes = [AES.MODE_ECB, AES.MODE_CBC, AES.MODE_GCM]
    results = []

    for mode in modes:
        cipher = AESCipher()
        # 预热
        cipher.encrypt(b'test', mode)
        # 正式测试
        time = timeit.timeit(lambda: cipher.encrypt(data, mode),
            number=100
        )
        results.append((mode, time/100))

    print(tabulate(
        results, 
        headers=['Mode', 'Time(s)'],
        floatfmt='.4f'
    ))

# 示例输出
# Mode     Time(s)
# ------   -------
# ECB      0.0214
# CBC      0.0231
# GCM      0.0287