yuzu-emu/mbedtls
1
0
Fork 0
mirror of https://github.com/yuzu-emu/mbedtls.git synced 2024-11-26 06:45:44 +01:00
Code Issues Packages Projects Releases Wiki Activity

Adjusted buf allocated from heap in ssl_client2 to be as small as possible

Browse Source
This commit is contained in:
Teppo Järvelin 2019-10-25 14:30:33 +03:00
parent 614efc268e
commit 7800719551
1 changed files with 17 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
programs/ssl/ssl_client2.c
View File

@ -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;
}