用python写RSA的DH密钥交换代码

以下是一个简单的Python实现RSA的DH密钥交换代码：

from Crypto.PublicKey import RSA
from Crypto.Cipher import PKCS1_OAEP
from Crypto.Util.number import long_to_bytes, bytes_to_long

# 生成RSA密钥对
rsa_key = RSA.generate(2048)
public_key = rsa_key.publickey()
private_key = rsa_key.export_key()

# DH密钥交换
alice_key = public_key.export_key()
bob_key = public_key.export_key()

alice_rsa = RSA.import_key(alice_key)
bob_rsa = RSA.import_key(bob_key)

alice_cipher = PKCS1_OAEP.new(alice_rsa)
bob_cipher = PKCS1_OAEP.new(bob_rsa)

alice_secret = alice_cipher.encrypt(long_to_bytes(1234))
bob_secret = bob_cipher.encrypt(long_to_bytes(5678))

alice_decrypted_secret = bytes_to_long(private_key.decrypt(alice_secret))
bob_decrypted_secret = bytes_to_long(private_key.decrypt(bob_secret))

print(alice_decrypted_secret == bob_decrypted_secret)

其中，我们使用了`Crypto.PublicKey`和`Crypto.Cipher`模块来生成RSA密钥对和加密解密密文，使用`Crypto.Util.number`模块来将长整型转换为字节和反之。在这个例子中，我们生成了两对相同的RSA密钥对，分别用于Alice和Bob。然后，我们使用相应的公钥加密要交换的DH密钥，并使用私钥解密。最后，我们比较Alice和Bob解密后的密钥是否相同，以确保密钥交换成功。