core: Modified to extract DH secret length from modulus instead of pu…

…blic key length Signed-off-by: Katsunori Kikuchi <Katsunori.Kikuchi@sony.com>
OP-TEE · Jul 12, 2023 · 54c317e · 54c317e
1 parent a871924
commit 54c317e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/core/drivers/crypto/crypto_api/acipher/dh.c b/core/drivers/crypto/crypto_api/acipher/dh.c
@@ -71,7 +71,7 @@ TEE_Result crypto_acipher_dh_shared_secret(struct dh_keypair *private_key,
 	dh = drvcrypt_get_ops(CRYPTO_DH);
 	if (dh) {
 		/* Allocate the binary Secret buffer */
-		sdata.secret.length = crypto_bignum_num_bytes(public_key);
+		sdata.secret.length = crypto_bignum_num_bytes(private_key->p);
 		secret_buf = malloc(sdata.secret.length);
 		if (!secret_buf)
 			return TEE_ERROR_OUT_OF_MEMORY;