Skip to content

Commit

Permalink
Adjusted buf allocated from heap in ssl_client2 to be as small as pos…
Browse files Browse the repository at this point in the history
…sible
  • Loading branch information
Teppo Järvelin committed Oct 25, 2019
1 parent 614efc2 commit 7800719
Showing 1 changed file with 17 additions and 3 deletions.
20 changes: 17 additions & 3 deletions programs/ssl/ssl_client2.c
Original file line number Diff line number Diff line change
Expand Up @@ -1522,9 +1522,23 @@ int main( int argc, char *argv[] )
goto usage;
}

main_buf_len = MAX_REQUEST_SIZE + 1;
buf = mbedtls_calloc( 1, MAX_REQUEST_SIZE + 1 );
if( buf == NULL ) {
/* try to use as small buf from the heap as possible */
if( opt.request_size <= 0 )
{
main_buf_len = MBEDTLS_SSL_MAX_CONTENT_LEN + 1;
}
else if( opt.request_size < (int)sizeof(GET_REQUEST) )
{
main_buf_len = sizeof(GET_REQUEST) + 1;
}
else
{
main_buf_len = opt.request_size + 1;
}

buf = mbedtls_calloc( 1, main_buf_len );
if( buf == NULL )
{
mbedtls_printf( "buf allocation failed!\n" );
goto exit;
}
Expand Down

0 comments on commit 7800719

Please sign in to comment.