DH秘钥交换实现 java代码

Diffie-Hellman (DH)密钥交换是一种非对称加密技术，用于在两个通信方之间安全地共享秘密密钥，而无需预先共享信息。以下是使用Java实现DH密钥交换的一个简单示例：

import javax.crypto.spec.DHParameterSpec;
import javax.crypto.spec.SecretKeySpec;
import java.math.BigInteger;

public class DHKeyExchange {

    // Diffie-Hellman参数，通常由系统提供
    private static final BigInteger p = new BigInteger("...");
    private static final BigInteger g = new BigInteger("...");

    public static void main(String[] args) {
        try {
            // Alice 和 Bob 分别生成私钥
            BigInteger alicePrivateKey = generatePrivateKey();
            BigInteger bobPrivateKey = generatePrivateKey();

            // Alice 将公钥发送给 Bob
            BigInteger alicePublicKey = calculatePublicDiffieHellman(g, alicePrivateKey, p);
            sendPublicKeyToBob(alicePublicKey);

            // Bob 接收到 Alice 的公钥，并计算共享密钥
            BigInteger sharedSecret = calculateSharedSecret(bobPrivateKey, alicePublicKey, p);

            // 将共享密钥转换为 SecretKeySpec 并存储
            SecretKeySpec secretKey = new SecretKeySpec(sharedSecret.toByteArray(), "AES");
            System.out.println("Alice and Bob shared secret key for AES encryption: " + new String(secretKey.getEncoded()));
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    // 生成随机的大于2的素数作为私钥
    private static BigInteger generatePrivateKey() {
        return BigInteger.probablePrime(512, new Random());
    }

    // 计算公钥
    private static BigInteger calculatePublicDiffieHellman(BigInteger base, BigInteger privateKey, BigInteger p) {
        return base.modPow(privateKey, p);
    }

    // 发送公钥
    private static void sendPublicKeyToBob(BigInteger publicKey) {
        // 这里只是一个简单的表示，实际应用需要通过安全通道传输
        System.out.println("Alice sent her public key to Bob: " + publicKey);
    }

    // 计算共享密钥
    private static BigInteger calculateSharedSecret(BigInteger privateKeyA, BigInteger publicKeyB, BigInteger p) {
        return publicKeyB.modPow(privateKeyA, p);
    }
}

注意这只是一个基本示例，实际应用中你需要处理网络通信、异常处理以及安全地交换公钥等。另外，在生产环境中，你可能还需要依赖专门的安全库如Bouncy Castle来处理DH算法。