commit d55f776cb778c85cc017de4a3fbde059ad673691
author Manuel Pégourié-Gonnard <manuel.pegourie-gonnard@arm.com> Fri Aug 18 17:40:15 2017 +0200
committer Manuel Pégourié-Gonnard <manuel.pegourie-gonnard@arm.com> Fri Aug 18 17:40:15 2017 +0200
tree 3258def49e6a9298dddf65c5c0980a70609e23d5
parent aaa9814879d5a99d386a18b67a868d1f3e718cf4

Skip context allocation if restart disabled

library/pk.c

diff --git a/library/pk.c b/library/pk.c
index fae517a..9a6e862 100644
--- a/library/pk.c
+++ b/library/pk.c
@@ -241,7 +241,10 @@
         return( MBEDTLS_ERR_PK_BAD_INPUT_DATA );
 
 #if defined(MBEDTLS_ECDSA_C) && defined(MBEDTLS_ECP_RESTARTABLE)
-    if( rs_ctx != NULL && ctx->pk_info->verify_rs_func != NULL )
+    /* optimization: use non-restartable version if restart disabled */
+    if( rs_ctx != NULL &&
+        mbedtls_ecp_restart_enabled() &&
+        ctx->pk_info->verify_rs_func != NULL )
     {
         int ret;
 
@@ -351,7 +354,10 @@
         return( MBEDTLS_ERR_PK_BAD_INPUT_DATA );
 
 #if defined(MBEDTLS_ECDSA_C) && defined(MBEDTLS_ECP_RESTARTABLE)
-    if( rs_ctx != NULL && ctx->pk_info->sign_rs_func != NULL )
+    /* optimization: use non-restartable version if restart disabled */
+    if( rs_ctx != NULL &&
+        mbedtls_ecp_restart_enabled() &&
+        ctx->pk_info->sign_rs_func != NULL )
     {
         int ret;