如何用编程实现deffie hellman密钥交换

Deffie-Hellman密钥交换是一种通过非安全通道交换密钥的加密算法，以下是Python实现：

import random

# 选择素数及原根
def select_prime_and_primitive_root():
    primes = [i for i in range(2, 100) if all(i % j != 0 for j in range(2, i))]  # 筛选素数
    p = random.choice(primes)  # 随机选择一个素数
    for i in range(2, p):
        roots = []
        for j in range(1, p):
            root = (i ** j) % p
            if root not in roots:
                roots.append(root)
        if len(roots) == p - 1:  # 判断i是否为p的原根
            return p, i

# 密钥交换
def key_exchange():
    p, g = select_prime_and_primitive_root()
    a = random.randint(1, p - 1)  # Alice生成密钥
    A = pow(g, a, p)  # Alice发送A给Bob
    b = random.randint(1, p - 1)  # Bob生成密钥
    B = pow(g, b, p)  # Bob发送B给Alice
    S_A = pow(B, a, p)  # Alice计算出密钥
    S_B = pow(A, b, p)  # Bob计算出密钥
    assert S_A == S_B
    return S_A

key = key_exchange()
print("密钥为：", key)

在该实现中，首先选择一个素数及其原根，然后Alice和Bob分别生成自己的密钥，并将其公钥发送给对方。最后双方根据对方的公钥计算出密钥并进行比对，确保密钥一致。