Montecarlo模拟AES扩散

Question

我需要对AE进行蒙特卡洛模拟来观察扩散的效果（如果我更改了一位明文，那么密文中的一半位应该会改变）...我认为我认为我在这里做了正确的一切，但是在AES（AES = AES.New（key，aes.mode_ecb）的定义中，说语法错误）。我在哪里犯了一个错误？在此先感谢您的帮助c：

import numpy as np
import random
from Cryptodome.Cipher import AES



def flip_bit(text, ibit):
    flipped_text = bytearray(text) 
    flipped_text[ibit//8] ^= 1 << (ibit%8) 
    return bytes(flipped_text)

def text_distance(textA, textB):
    AxorB = bytes(a ^ b for (a, b) in zip(textA, textB))
    distance = sum([bin(byte).count('1') for byte in AxorB])
    return distance

def mcs_diffusion(key_length):
    
    assert key_length in AES.key_size
    
 
    key = bytes(random.randint( key_length)
    aes = AES.new(key, AES.MODE_ECB)             
   

    x = bytes(random.randint(AES.block_size))
    x1 = flip_bit(x, random.randint(8*AES.block_size))

    y = aes.encrypt(x)
    y1 = aes.encrypt(x1)
    distance = text_distance(y, y1)
    
    return distance


N = 10000 
key_length = 16

results = np.zeros(N)
for i in range(N):
    results[i] = mcs_diffusion(key_length)
    
print('average: {}'.format(np.mean(results)))
print('min, max: {}, {}'.format(min(results), max(results)))