commit 66070bc19d510fbe1164666c7cc949fb6a7bc965
author Philippe Antoine <contact@catenacyber.fr> Wed Jan 22 13:54:56 2020 +0100
committer Philippe Antoine <contact@catenacyber.fr> Wed Jan 22 13:54:56 2020 +0100
tree b667a351806626dcaa30106ba9f30d408e1a39f6
parent dbcb44202c9b67184a72afd21c7191c17eee352e

Checks mbedtls_rsa_export return in fuzz targets

programs/fuzz/fuzz_privkey.c

diff --git a/programs/fuzz/fuzz_privkey.c b/programs/fuzz/fuzz_privkey.c
index 533a647..3685592 100644
--- a/programs/fuzz/fuzz_privkey.c
+++ b/programs/fuzz/fuzz_privkey.c
@@ -29,7 +29,9 @@
             mbedtls_mpi_init( &DQ ); mbedtls_mpi_init( &QP );
 
             rsa = mbedtls_pk_rsa( pk );
-            mbedtls_rsa_export( rsa, &N, &P, &Q, &D, &E );
+            if ( mbedtls_rsa_export( rsa, &N, &P, &Q, &D, &E ) != 0 ) {
+                abort();
+            }
             mbedtls_rsa_export_crt( rsa, &DP, &DQ, &QP );
 
             mbedtls_mpi_free( &N ); mbedtls_mpi_free( &P ); mbedtls_mpi_free( &Q );

programs/fuzz/fuzz_pubkey.c

diff --git a/programs/fuzz/fuzz_pubkey.c b/programs/fuzz/fuzz_pubkey.c
index df42f7d..3a59125 100644
--- a/programs/fuzz/fuzz_pubkey.c
+++ b/programs/fuzz/fuzz_pubkey.c
@@ -20,7 +20,9 @@
             mbedtls_mpi_init( &DQ ); mbedtls_mpi_init( &QP );
 
             rsa = mbedtls_pk_rsa( pk );
-            ret = mbedtls_rsa_export( rsa, &N, &P, &Q, &D, &E );
+            if ( mbedtls_rsa_export( rsa, &N, &P, &Q, &D, &E ) != 0 ) {
+                abort();
+            }
             ret = mbedtls_rsa_export_crt( rsa, &DP, &DQ, &QP );
 
             mbedtls_mpi_free( &N ); mbedtls_mpi_free( &P ); mbedtls_mpi_free( &Q );