commit 8e0e48199b6afab71a82e41730c688b188ede30d
author Teppo Järvelin <teppo.jarvelin@arm.com> Mon Oct 21 10:33:11 2019 +0300
committer Teppo Järvelin <teppo.jarvelin@arm.com> Tue Oct 22 14:44:13 2019 +0300
tree c6f5be4ecb684e727c63d437834c594d89c99213
parent 8e0fe19a6ab888025c24c325650a70f45006188e

Changed main buf to be allocated from heap in ssl_client2

programs/ssl/ssl_client2.c

diff --git a/programs/ssl/ssl_client2.c b/programs/ssl/ssl_client2.c
index 066f6a5..be64f6a 100644
--- a/programs/ssl/ssl_client2.c
+++ b/programs/ssl/ssl_client2.c
@@ -935,7 +935,8 @@
     io_ctx_t io_ctx;
 #endif
 
-    unsigned char buf[MAX_REQUEST_SIZE + 1];
+    unsigned char *buf = NULL;
+    unsigned int main_buf_len = 0;
 
 #if defined(MBEDTLS_KEY_EXCHANGE__SOME__PSK_ENABLED)
     unsigned char psk[MBEDTLS_PSK_MAX_LEN];
@@ -1521,6 +1522,13 @@
             goto usage;
     }
 
+    main_buf_len = MAX_REQUEST_SIZE  + 1;
+    buf = mbedtls_calloc( 1, MAX_REQUEST_SIZE  + 1 );
+    if( buf == NULL ) {
+        mbedtls_printf( "buf allocation failed!\n" );
+        goto exit;
+    }
+
     /* Event-driven IO is incompatible with the above custom
      * receive and send functions, as the polling builds on
      * refers to the underlying net_context. */
@@ -2449,7 +2457,7 @@
     mbedtls_printf( "  > Write to server:" );
     fflush( stdout );
 
-    len = mbedtls_snprintf( (char *) buf, sizeof( buf ) - 1, GET_REQUEST,
+    len = mbedtls_snprintf( (char *) buf, main_buf_len - 1, GET_REQUEST,
                             opt.request_page );
     tail_len = (int) strlen( GET_REQUEST_END );
 
@@ -2461,7 +2469,7 @@
         len += opt.request_size - len - tail_len;
     }
 
-    strncpy( (char *) buf + len, GET_REQUEST_END, sizeof( buf ) - len - 1 );
+    strncpy( (char *) buf + len, GET_REQUEST_END, main_buf_len - len - 1 );
     len += tail_len;
 
     /* Truncate if request size is smaller than the "natural" size */
@@ -2577,8 +2585,8 @@
     {
         do
         {
-            len = sizeof( buf ) - 1;
-            memset( buf, 0, sizeof( buf ) );
+            len = main_buf_len - 1;
+            memset( buf, 0, main_buf_len );
             ret = mbedtls_ssl_read( ssl, buf, len );
 
 #if defined(MBEDTLS_ECP_RESTARTABLE)
@@ -2639,8 +2647,8 @@
     }
     else /* Not stream, so datagram */
     {
-        len = sizeof( buf ) - 1;
-        memset( buf, 0, sizeof( buf ) );
+        len = main_buf_len - 1;
+        memset( buf, 0, main_buf_len );
 
         while( 1 )
         {
@@ -2994,6 +3002,7 @@
     mbedtls_free( ssl );
     mbedtls_free( conf );
     mbedtls_free( entropy );
+    mbedtls_free( buf );
 #if defined(MBEDTLS_CTR_DRBG_C)
     mbedtls_free( ctr_drbg );
 #else