From 4f3368e31e7c6443ccf9567e08a428ed353aa851 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Manuel=20P=C3=A9gouri=C3=A9-Gonnard?= <mpg@elzevir.fr>
Date: Sun, 19 Jul 2015 15:01:28 +0200
Subject: [PATCH] Fix bug in benchmark.c with DHM params

---
 ChangeLog                 |  5 +++++
 programs/test/benchmark.c | 10 ++++------
 2 files changed, 9 insertions(+), 6 deletions(-)

diff --git a/ChangeLog b/ChangeLog
index 9b438b130..0d2e2a12b 100644
--- a/ChangeLog
+++ b/ChangeLog
@@ -1,5 +1,10 @@
 mbed TLS ChangeLog (Sorted per branch, date)
 
+= mbed TLS 2 branch
+
+Bugfix
+   * Fix segfault in the benchmark program when benchmarking DHM.
+
 = mbed TLS 2.0.0 released 2015-07-13
 
 Features
diff --git a/programs/test/benchmark.c b/programs/test/benchmark.c
index 9838b84da..f1bd52354 100644
--- a/programs/test/benchmark.c
+++ b/programs/test/benchmark.c
@@ -91,8 +91,6 @@ int main( void )
 #define HEADER_FORMAT   "  %-24s :  "
 #define TITLE_LEN       25
 
-#define DHM_SIZES 3
-
 #define OPTIONS                                                         \
     "md4, md5, ripemd160, sha1, sha256, sha512,\n"                      \
     "arc4, des3, des, aes_cbc, aes_gcm, aes_ccm, camellia, blowfish,\n" \
@@ -611,19 +609,19 @@ int main( int argc, char *argv[] )
 #if defined(MBEDTLS_DHM_C) && defined(MBEDTLS_BIGNUM_C)
     if( todo.dhm )
     {
-        int dhm_sizes[DHM_SIZES] = { 2048, 3072 };
-        const char *dhm_P[DHM_SIZES] = {
+        int dhm_sizes[] = { 2048, 3072 };
+        const char *dhm_P[] = {
             MBEDTLS_DHM_RFC3526_MODP_2048_P,
             MBEDTLS_DHM_RFC3526_MODP_3072_P,
         };
-        const char *dhm_G[DHM_SIZES] = {
+        const char *dhm_G[] = {
             MBEDTLS_DHM_RFC3526_MODP_2048_G,
             MBEDTLS_DHM_RFC3526_MODP_3072_G,
         };
 
         mbedtls_dhm_context dhm;
         size_t olen;
-        for( i = 0; i < DHM_SIZES; i++ )
+        for( i = 0; (size_t) i < sizeof( dhm_sizes ) / sizeof( dhm_sizes[0] ); i++ )
         {
             mbedtls_dhm_init( &dhm );