diffie-hellman算法用c++实现

Diffie-Hellman算法是一种密钥交换协议，用于在两个通信方之间安全地共享密钥。下面是一个用C++实现Diffie-Hellman算法的示例代码：

#include<iostream>
#include<cmath>
using namespace std;
 
// 根据离散对数算法计算 a^b mod p 的值
int modpow(int a, int b, int p){
    int res = 1;
    while(b > 0){
        if(b & 1) res = (res * a) % p;
        a = (a * a) % p;
        b >>= 1;
    }
    return res;
}
 
// 计算 a 的逆元
int inverse(int a, int p){
    return modpow(a, p-2, p);
}
 
// 计算 g^x mod p 的值
int power(int g, int x, int p){
    return modpow(g, x, p);
}
 
int main(){
    int p, g, a, b;
    cout<<"Enter the value of p and g: ";
    cin>>p>>g;
    cout<<"Enter the value of a and b: ";
    cin>>a>>b;
 
    // 计算 A = g^a mod p 和 B = g^b mod p
    int A = power(g, a, p);
    int B = power(g, b, p);
 
    // 计算共享密钥 K = B^a mod p = A^b mod p
    int K1 = power(B, a, p);
    int K2 = power(A, b, p);
 
    cout<<"A = "<<A<<endl;
    cout<<"B = "<<B<<endl;
    cout<<"K1 = "<<K1<<endl;
    cout<<"K2 = "<<K2<<endl;
 
    return 0;
}

在这个示例中，我们首先输入一个素数p和一个原根g，然后输入两个私钥a和b。接下来，我们计算出A和B，分别是g^a mod p和g^b mod p。然后，我们计算出共享密钥K，它可以通过以下两种方法之一来计算：B^a mod p或A^b mod p。最后，我们输出A、B、K1和K2的值。

需要注意的是，在实际应用中，我们需要使用更大的素数p和私钥a和b来保证安全性。