DES差分分布表的构造

from CryptoIntro import data


# 构造差分分布表

def format_standardize(s, l):
    for i in range(l - len(s)):
        s = '0' + s
    return s


def show_diff_lat(diff_lat, i):
    print("s[{}]".format(i))
    for i in range(pow(2, 6)):
        for j in range(pow(2, 4)):
            print(diff_lat[i][j], end=' ')
        print()
    print("==================================================")


def generate_diff_lat(i):
    diff_lat = [[0 for i in range(pow(2, 4))] for j in range(pow(2, 6))]
    for input_diff in range(pow(2, 6)):
        for alpha in range(pow(2, 6)):
            beta = format_standardize(bin(alpha ^ input_diff)[2:], 6)
            alpha = format_standardize(bin(alpha)[2:], 6)
            alpha_output = data.s_box_list[i][int(alpha[0] + alpha[5], 2)][int(alpha[1:5], 2)]
            beta_output = data.s_box_list[i][int(beta[0] + beta[5], 2)][int(beta[1:5], 2)]
            output_diff = alpha_output ^ beta_output
            diff_lat[input_diff][output_diff] += 1

    show_diff_lat(diff_lat, i)


if __name__ == "__main__":
    for i in range(8):
        generate_diff_lat(i)