commit 3b58700d3895971c9500fc34b7b9d494a011a337
author Christoph M. Wintersteiger <cwinter@microsoft.com> Wed Feb 20 17:26:42 2019 +0000
committer Janos Follath <janos.follath@arm.com> Mon Aug 19 13:37:46 2019 +0100
tree da457cd2b0ff7de2e07d4ed1321f526415029a53
parent bfc8eb2b78f2b7eaedebea5b9070cf10ef6f73de

ECDH: Fix use of ECDH API in full handshake benchmark

programs/test/benchmark.c

diff --git a/programs/test/benchmark.c b/programs/test/benchmark.c
index e05470a..ada42df 100644
--- a/programs/test/benchmark.c
+++ b/programs/test/benchmark.c
@@ -1022,13 +1022,13 @@
                 continue;
 
             mbedtls_ecdh_init( &ecdh_srv );
-            mbedtls_ecdh_init( &ecdh_cli );
-            CHECK_AND_CONTINUE( mbedtls_ecdh_setup( &ecdh_srv, curve_info->grp_id ) );
-            CHECK_AND_CONTINUE( mbedtls_ecdh_setup( &ecdh_cli, curve_info->grp_id ) );
+            mbedtls_ecdh_init( &ecdh_cli );            
 
             mbedtls_snprintf( title, sizeof( title ), "ECDHE-%s", curve_info->name );
             TIME_PUBLIC( title, "full handshake",
                 const unsigned char * p_srv = buf_srv;
+                
+                CHECK_AND_CONTINUE( mbedtls_ecdh_setup( &ecdh_srv, curve_info->grp_id ) );
                 CHECK_AND_CONTINUE( mbedtls_ecdh_make_params( &ecdh_srv, &olen, buf_srv, sizeof( buf_srv ), myrand, NULL ) );
 
                 CHECK_AND_CONTINUE( mbedtls_ecdh_read_params( &ecdh_cli, &p_srv, p_srv + olen ) );
@@ -1038,10 +1038,11 @@
                 CHECK_AND_CONTINUE( mbedtls_ecdh_calc_secret( &ecdh_srv, &olen, buf_srv, sizeof( buf_srv ), myrand, NULL ) );
 
                 CHECK_AND_CONTINUE( mbedtls_ecdh_calc_secret( &ecdh_cli, &olen, buf_cli, sizeof( buf_cli ), myrand, NULL ) );
+                mbedtls_ecdh_free( &ecdh_cli );
+                
+                mbedtls_ecdh_free( &ecdh_srv );
             );
 
-            mbedtls_ecdh_free( &ecdh_srv );
-            mbedtls_ecdh_free( &ecdh_cli );
         }
     }
 #endif