commit f4dfd1c8a5282ea0a9d0641d2fd6dd0649a5c92f
author Minos Galanakis <minos.galanakis@arm.com> Fri Jan 12 16:06:15 2024 +0000
committer Minos Galanakis <minos.galanakis@arm.com> Thu Jan 18 14:29:41 2024 +0000
tree e996be8fcb95178a51130a8b0bae1b9f206f08b3
parent a184fd0516b5a059d9e9a0ab3912345385ff9b72

programs/dh_client/server: Added entropy source to `mbedtls_rsa_pkcs1_sign()`

Signed-off-by: Minos Galanakis <minos.galanakis@arm.com>

programs/pkey/dh_server.c

diff --git a/programs/pkey/dh_server.c b/programs/pkey/dh_server.c
index 11c2b28..7d7618b 100644
--- a/programs/pkey/dh_server.c
+++ b/programs/pkey/dh_server.c
@@ -193,8 +193,9 @@
     buf[n] = (unsigned char) (rsa.MBEDTLS_PRIVATE(len) >> 8);
     buf[n + 1] = (unsigned char) (rsa.MBEDTLS_PRIVATE(len));
 
-    if ((ret = mbedtls_rsa_pkcs1_sign(&rsa, NULL, NULL, MBEDTLS_MD_SHA256,
-                                      MBEDTLS_MD_CAN_SHA256_MAX_SIZE, hash, buf + n + 2)) != 0) {
+    if ((ret = mbedtls_rsa_pkcs1_sign(&rsa, mbedtls_ctr_drbg_random, &ctr_drbg,
+                                      MBEDTLS_MD_SHA256, MBEDTLS_MD_CAN_SHA256_MAX_SIZE,
+                                      hash, buf + n + 2)) != 0) {
         mbedtls_printf(" failed\n  ! mbedtls_rsa_pkcs1_sign returned %d\n\n", ret);
         goto exit;
     }