使用 Python 生成与 Java 兼容的 Diffie-Hellman答案

【问题标题】：Generate a Java compatible Diffie-Hellman using Python使用 Python 生成与 Java 兼容的 Diffie-Hellman
【发布时间】：2021-01-31 14:37:38
【问题描述】：

我正在尝试用 Python 重写以下代码。原始代码是使用 sjcl 库用 Javascript 编写的。

// Inputs
var serverPubX = "WIUBDotrk02Rk/apL11jQPbmX0quyaYz2EIkGUlVf7s=";
var serverPubY = "diZ2CbfSUy5Kr82OIfd4Ajusq2K+/kjGZ7ymcqVwn2k=";

// The code
var serverPubXBits = sjcl.codec.base64.toBits(serverPubX);
var serverPubYBits = sjcl.codec.base64.toBits(serverPubY);
var serverPubKeyPointBits = serverPubXBits.concat(serverPubYBits);

var serverPubKey = new sjcl.ecc.elGamal.publicKey(
    sjcl.ecc.curves.c256, serverPubKeyPointBits);

var clientKeys = sjcl.ecc.elGamal.generateKeys(256, 1);

// What I need:
var sharedKey = clientKeys.sec.dhJavaEc(serverPubKey);

我的主要问题是dhJavaEc 函数。根据 sjcl 文档，它是Java compatible Diffie-Hellmann function。但我在 pycryptodome 库中找不到任何等价物。

我检查了dhJavaEc 代码，这是它的作用：

// Looks like it converts the server key to Jacobian and then multiply it by client key
serverPubKey.J.toJac().mult(clientKeys.sec.I, serverPubKey.J).toAffine().x.toBits()

// serverPubKey.J is the X and Y keys concatenated:
sjcl.codec.base64.fromBits(serverPubKey.J.toBits())
"WIUBDotrk02Rk/apL11jQPbmX0quyaYz2EIkGUlVf7t2JnYJt9JTLkqvzY4h93gCO6yrYr7+SMZnvKZypXCfaQ=="

// In my example, clientKeys.sec.I is this:
sjcl.codec.base64.fromBits(clientKeys.sec.I.toBits())
"zIhDVlFUpWQiRP+bjyEIhSLq8rcB8+XInXGhm6JGcVI="

// And the calculated key is:
sjcl.codec.base64.fromBits(sharedKey)
"ZBin/RV1qnfKoIuel+5fzv1y8rn3UZkMPO3pXva3VzQ="

如何使用 Python 生成等效的“sharedKey”？

【问题讨论】：

标签： python cryptography elliptic-curve diffie-hellman

【解决方案1】：

PyCryptodome 目前似乎不支持 ECDH，请参阅 Future plans。另一种方法是 Cryptography 库，请参阅 Elliptic Curve Key Exchange algorithm。

库期望私钥为int，未压缩格式的公钥为bytes object。未压缩格式由连接的 x 和 y 坐标组成，前面有一个 0x04 字节。

sjcl.ecc.curves.c256 定义了 secp256r1（又名 prime256v1 又名 NIST P-256）。

那么在 Python 中使用 密码学 的可能实现是：

from cryptography.hazmat.backends import default_backend
from cryptography.hazmat.primitives.asymmetric import ec
import base64

curve = ec.SECP256R1()

clientKey = 'zIhDVlFUpWQiRP+bjyEIhSLq8rcB8+XInXGhm6JGcVI='
privateKeyInt = int.from_bytes(base64.b64decode(clientKey), byteorder='big', signed=False)
privateKey = ec.derive_private_key(privateKeyInt, curve, default_backend())

serverPubKey = 'WIUBDotrk02Rk/apL11jQPbmX0quyaYz2EIkGUlVf7t2JnYJt9JTLkqvzY4h93gCO6yrYr7+SMZnvKZypXCfaQ=='
publicKeyUncomp = b'\x04' + base64.b64decode(serverPubKey)
publicKey = ec.EllipticCurvePublicKey.from_encoded_point(curve, publicKeyUncomp)

sharedSecret = privateKey.exchange(ec.ECDH(), publicKey)
print(base64.b64encode(sharedSecret).decode('utf8')) # ZBin/RV1qnfKoIuel+5fzv1y8rn3UZkMPO3pXva3VzQ=

生成与 JavaScript 代码相同的共享密钥。

【讨论】：