yuzu-emu/mbedtls
2
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

pkwrite: add a safety check before calculating the buffer size

Browse source
This commit is contained in:
Andrzej Kurek 2018-11-19 18:09:59 -05:00
parent 2f31122585
commit 16d6000577
1 changed files with 6 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
library/pkwrite.c
View file

@ -167,8 +167,13 @@ int mbedtls_pk_write_pubkey( unsigned char **p, unsigned char *start,
#if defined(MBEDTLS_USE_PSA_CRYPTO) #if defined(MBEDTLS_USE_PSA_CRYPTO)
if( mbedtls_pk_get_type( key ) == MBEDTLS_PK_OPAQUE ) if( mbedtls_pk_get_type( key ) == MBEDTLS_PK_OPAQUE )
{ {
size_t buffer_size = *p - start; size_t buffer_size;
psa_key_slot_t* key_slot = (psa_key_slot_t*) key->pk_ctx; psa_key_slot_t* key_slot = (psa_key_slot_t*) key->pk_ctx;
if ( *p < start )
return( MBEDTLS_ERR_PK_BAD_INPUT_DATA );
buffer_size = *p - start;
if ( psa_export_public_key( *key_slot, start, buffer_size, &len ) if ( psa_export_public_key( *key_slot, start, buffer_size, &len )
!= PSA_SUCCESS ) != PSA_SUCCESS )
{ {