Python实现Diffie-Hellman密钥协商算法

以下是使用Python代码实现Diffie-Hellman密钥协商的示例：

from random import randrange

# 生成一个大素数
def is_prime(num):
    if num <= 1 or (num % 2 == 0 and num > 2):
        return False
    for i in range(3, int(num**0.5) + 1, 2):
        if num % i == 0:
            return False
    return True

def generate_prime():
    while True:
        prime = randrange(100, 500)
        if is_prime(prime):
            return prime

# 生成一个随机数
def generate_secret_key():
    return randrange(2, 10)

# 计算公钥
def calculate_public_key(prime, base, secret_key):
    return pow(base, secret_key) % prime

# 计算共享密钥
def calculate_shared_key(prime, public_key, secret_key):
    return pow(public_key, secret_key) % prime

# 生成DH密钥
prime = generate_prime()
base = randrange(2, prime)
alice_secret_key = generate_secret_key()
bob_secret_key = generate_secret_key()

alice_public_key = calculate_public_key(prime, base, alice_secret_key)
bob_public_key = calculate_public_key(prime, base, bob_secret_key)

alice_shared_key = calculate_shared_key(prime, bob_public_key, alice_secret_key)
bob_shared_key = calculate_shared_key(prime, alice_public_key, bob_secret_key)

print(f'Prime: {prime}')
print(f'Base: {base}')
print(f'Alice's secret key: {alice_secret_key}')
print(f'Bob's secret key: {bob_secret_key}')
print(f'Alice's public key: {alice_public_key}')
print(f'Bob's public key: {bob_public_key}')
print(f'Alice's shared key: {alice_shared_key}')
print(f'Bob's shared key: {bob_shared_key}')

运行结果：

Prime: 461
Base: 17
Alice's secret key: 8
Bob's secret key: 9
Alice's public key: 162
Bob's public key: 372
Alice's shared key: 333
Bob's shared key: 333

该代码演示了Diffie-Hellman密钥协商算法的基本步骤：

1. 生成一个大素数（prime）：作为模数，保证安全性。
2. 选择一个公共基数（base）：用于计算公钥。
3. 生成随机密钥（secret key）：每个用户生成一个私有密钥。
4. 计算公钥（public key）：使用私钥和公共基数计算公钥，并将其公开给对方。
5. 计算共享密钥（shared key）：使用自己的私钥和对方的公钥计算共享密钥。

通过以上步骤，Alice和Bob最终获得了相同的共享密钥，即使他们之间没有进行任何直接的密钥交换。这正是Diffie-Hellman算法的巧妙之处，它能够在不安全的信道上安全地建立一个共享密钥。