Python中生成AES密钥调度的标准加密库及实现方法咨询

1. 现成库说明 #

• 主流Python密码学库（如pycryptodome、cryptography）均将AES密钥调度封装在内部，不会暴露单独调用接口——这类库的设计目标是提供安全易用的加解密功能，而非开放算法内部细节。
• 若聚焦密钥调度的研究与耗时测试，手动实现是最可控的方案，能完全自定义初始子密钥，也方便精准测试性能。

2. 手动实现AES密钥调度（支持预定义初始密钥） #

AES密钥调度核心包含轮常量（Rcon）、字节替换（S盒）、循环移位等操作。以下是AES-128的实现（可扩展至192/256位），支持将任意预定义字符串（如密码）转为初始子密钥：

基础工具函数 #

# AES标准S盒
S_BOX = [
    0x63, 0x7C, 0x77, 0x7B, 0xF2, 0x6B, 0x6F, 0xC5, 0x30, 0x01, 0x67, 0x2B, 0xFE, 0xD7, 0xAB, 0x76,
    0xCA, 0x82, 0xC9, 0x7D, 0xFA, 0x59, 0x47, 0xF0, 0xAD, 0xD4, 0xA2, 0xAF, 0x9C, 0xA4, 0x72, 0xC0,
    0xB7, 0xFD, 0x93, 0x26, 0x36, 0x3F, 0xF7, 0xCC, 0x34, 0xA5, 0xE5, 0xF1, 0x71, 0xD8, 0x31, 0x15,
    0x04, 0xC7, 0x23, 0xC3, 0x18, 0x96, 0x05, 0x9A, 0x07, 0x12, 0x80, 0xE2, 0xEB, 0x27, 0xB2, 0x75,
    0x09, 0x83, 0x2C, 0x1A, 0x1B, 0x6E, 0x5A, 0xA0, 0x52, 0x3B, 0xD6, 0xB3, 0x29, 0xE3, 0x2F, 0x84,
    0x53, 0xD1, 0x00, 0xED, 0x20, 0xFC, 0xB1, 0x5B, 0x6A, 0xCB, 0xBE, 0x39, 0x4A, 0x4C, 0x58, 0xCF,
    0xD0, 0xEF, 0xAA, 0xFB, 0x43, 0x4D, 0x33, 0x85, 0x45, 0xF9, 0x02, 0x7F, 0x50, 0x3C, 0x9F, 0xA8,
    0x51, 0xA3, 0x40, 0x8F, 0x92, 0x9D, 0x38, 0xF5, 0xBC, 0xB6, 0xDA, 0x21, 0x10, 0xFF, 0xF3, 0xD2,
    0xCD, 0x0C, 0x13, 0xEC, 0x5F, 0x97, 0x44, 0x17, 0xC4, 0xA7, 0x7E, 0x3D, 0x64, 0x5D, 0x19, 0x73,
    0x60, 0x81, 0x4F, 0xDC, 0x22, 0x2A, 0x90, 0x88, 0x46, 0xEE, 0xB8, 0x14, 0xDE, 0x5E, 0x0B, 0xDB,
    0xE0, 0x32, 0x3A, 0x0A, 0x49, 0x06, 0x24, 0x5C, 0xC2, 0xD3, 0xAC, 0x62, 0x91, 0x95, 0xE4, 0x79,
    0xE7, 0xC8, 0x37, 0x6D, 0x8D, 0xD5, 0x4E, 0xA9, 0x6C, 0x56, 0xF4, 0xEA, 0x65, 0x7A, 0xAE, 0x08,
    0xBA, 0x78, 0x25, 0x2E, 0x1C, 0xA6, 0xB4, 0xC6, 0xE8, 0xDD, 0x74, 0x1F, 0x4B, 0xBD, 0x8B, 0x8A,
    0x70, 0x3E, 0xB5, 0x66, 0x48, 0x03, 0xF6, 0x0E, 0x61, 0x35, 0x57, 0xB9, 0x86, 0xC1, 0x1D, 0x9E,
    0xE1, 0xF8, 0x98, 0x11, 0x69, 0xD9, 0x8E, 0x94, 0x9B, 0x1E, 0x87, 0xE9, 0xCE, 0x55, 0x28, 0xDF,
    0x8C, 0xA1, 0x89, 0x0D, 0xBF, 0xE6, 0x42, 0x68, 0x41, 0x99, 0x2D, 0x0F, 0xB0, 0x54, 0xBB, 0x16
]

# 循环左移1字节
def rot_word(word):
    return word[1:] + word[:1]

# 字节替换+轮常量异或
def sub_word(word, rcon):
    return bytes([S_BOX[b] ^ rcon[i] for i, b in enumerate(word)])

密钥调度核心函数 #

def aes_key_schedule(initial_key_bytes, key_size=128):
    """
    生成AES轮子密钥
    :param initial_key_bytes: 初始子密钥字节串（需匹配密钥长度：16字节=128位，24=192，32=256）
    :param key_size: AES密钥长度，可选128/192/256
    :return: 轮子密钥列表，每个元素为16字节的轮密钥
    """
    if key_size == 128:
        num_rounds = 10
        key_words = 4
    elif key_size == 192:
        num_rounds = 12
        key_words = 6
    elif key_size == 256:
        num_rounds = 14
        key_words = 8
    else:
        raise ValueError("无效密钥长度，仅支持128/192/256位")
    
    # 初始密钥拆分为4字节的word单元
    words = [initial_key_bytes[i*4 : (i+1)*4] for i in range(key_words)]
    
    # 生成剩余轮子密钥
    rcon = [0x01, 0x02, 0x04, 0x08, 0x10, 0x20, 0x40, 0x80, 0x1B, 0x36]
    for i in range(key_words, (num_rounds + 1) * 4):
        temp = words[i-1]
        if i % key_words == 0:
            temp = sub_word(rot_word(temp), [rcon[i//key_words - 1], 0x00, 0x00, 0x00])
        elif key_size == 256 and i % key_words == 4:
            temp = bytes([S_BOX[b] for b in temp])
        # 异或前一个word生成当前word
        new_word = bytes([a ^ b for a, b in zip(words[i - key_words], temp)])
        words.append(new_word)
    
    # 组合word为完整轮密钥
    round_keys = [b''.join(words[i*4 : (i+1)*4]) for i in range(num_rounds + 1)]
    return round_keys

预定义字符串作为初始密钥的使用示例 #

# 示例：用自定义密码生成初始密钥，调整为16字节（AES-128要求）
password = "my_research_key_001"
# 截断或补全到16字节（实际场景建议用PBKDF2等KDF生成密钥，此处为演示直接处理）
initial_key = password.encode('utf-8')[:16].ljust(16, b'\x00')

# 生成轮子密钥
round_keys = aes_key_schedule(initial_key, key_size=128)

# 验证输出
print("初始轮密钥:", round_keys[0].hex())
print("第1轮密钥:", round_keys[1].hex())

3. 密钥调度耗时测试 #

使用Python内置timeit模块可精准测试生成耗时，示例如下：

import timeit

def test_key_schedule_speed():
    password = "benchmark_test_key"
    initial_key = password.encode('utf-8')[:16].ljust(16, b'\x00')
    # 运行10000次取平均耗时
    total_time = timeit.timeit(lambda: aes_key_schedule(initial_key), number=10000)
    print(f"10000次AES-128密钥调度总耗时: {total_time:.4f}秒")
    print(f"单次平均耗时: {total_time/10000:.8f}秒")

test_key_schedule_speed()

补充说明 #

• 若需支持AES-192/256，只需调整key_size参数，并确保初始密钥字节长度为24或32字节。
• 如需更深入的性能分析，可结合cProfile模块查看函数内部耗时分布。

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