commit 216b40de9b5b9cf820a783d8886db78108bffd7d
author Andrzej Kurek <andrzej.kurek@arm.com> Wed Jun 20 07:07:55 2018 -0400
committer Simon Butcher <simon.butcher@arm.com> Wed Sep 26 22:53:12 2018 +0100
tree fe75b80a42cd4591994025b9f0546675fdd6d951
parent d731a6348ab4445421c688c9e6e66ff77118b51d

ssl_server2: add buffer overhead for a termination character

Switch to mbedtls style of memory allocation

programs/ssl/ssl_server2.c

diff --git a/programs/ssl/ssl_server2.c b/programs/ssl/ssl_server2.c
index 1a57a79..6736bcc 100644
--- a/programs/ssl/ssl_server2.c
+++ b/programs/ssl/ssl_server2.c
@@ -1355,7 +1355,7 @@
 #if defined(MBEDTLS_DEBUG_C)
     mbedtls_debug_set_threshold( opt.debug_level );
 #endif
-    buf = malloc( opt.buffer_size );
+    buf = mbedtls_calloc( 1, opt.buffer_size + 1 );
     if( buf == NULL )
     {
         mbedtls_printf("Could not allocate %u bytes\n", opt.buffer_size);
@@ -2545,6 +2545,7 @@
     mbedtls_memory_buffer_alloc_free();
 #endif
 
+    mbedtls_free(buf);
     mbedtls_printf( " done.\n" );
 
 #if defined(_WIN32)