commit 128657d64525475a14902d2c11c3013cf47ba5b7
author Manuel Pégourié-Gonnard <mpg@elzevir.fr> Thu Dec 18 16:35:52 2014 +0000
committer Manuel Pégourié-Gonnard <mpg@elzevir.fr> Mon Feb 16 17:24:57 2015 +0000
tree 8a3cba671eae870b9dc4d2c99ebfa9513625d814
parent b8ca72315411ea2b659c5d7dba0c36097f0574a4

Use memory_buffer_alloc() in benchmark if available

Allows to measure memory by primitive.

programs/test/benchmark.c

diff --git a/programs/test/benchmark.c b/programs/test/benchmark.c
index 675547b..c0c7a11 100644
--- a/programs/test/benchmark.c
+++ b/programs/test/benchmark.c
@@ -61,9 +61,8 @@
 #include "polarssl/ecdh.h"
 #include "polarssl/error.h"
 
-#include <stdio.h>
-#include <stdlib.h>
-#include <string.h>
+#if defined(POLARSSL_MEMORY_BUFFER_ALLOC_C)
+#include "polarssl/memory_buffer_alloc.h"
 #endif
 
 #if defined _MSC_VER && !defined snprintf
@@ -182,6 +181,9 @@
     unsigned char tmp[200];
     char title[TITLE_LEN];
     todo_list todo;
+#if defined(POLARSSL_MEMORY_BUFFER_ALLOC_C)
+    unsigned char malloc_buf[1000000] = { 0 };
+#endif
 
     if( argc == 1 )
         memset( &todo, 1, sizeof( todo ) );
@@ -243,6 +245,9 @@
 
     polarssl_printf( "\n" );
 
+#if defined(POLARSSL_MEMORY_BUFFER_ALLOC_C)
+    memory_buffer_alloc_init( malloc_buf, sizeof( malloc_buf ) );
+#endif
     memset( buf, 0xAA, sizeof( buf ) );
     memset( tmp, 0xBB, sizeof( tmp ) );
 
@@ -642,6 +647,13 @@
 #endif
     polarssl_printf( "\n" );
 
+#if defined(POLARSSL_MEMORY_BUFFER_ALLOC_C)
+#if defined(POLARSSL_MEMORY_DEBUG)
+    memory_buffer_alloc_status();
+#endif
+    memory_buffer_alloc_free();
+#endif
+
 #if defined(_WIN32)
     polarssl_printf( "  Press Enter to exit this program.\n" );
     fflush( stdout ); getchar();