commit 0963e6cfac2230d68c6ed1aa220ac41f096796ff
author Andres Amaya Garcia <andres.amayagarcia@arm.com> Thu Jul 20 14:34:08 2017 +0100
committer Andres Amaya Garcia <andres.amayagarcia@arm.com> Thu Jul 20 14:41:10 2017 +0100
tree 183f37db4a4d362594d6ea6c26b7b6201e89ef71
parent 94682d1d7d4a8492b0e832318bad670b427167b8

Fix possible memory leak in <MD>_ext()

library/sha256.c

diff --git a/library/sha256.c b/library/sha256.c
index 16a2f0b..fb03cd1 100644
--- a/library/sha256.c
+++ b/library/sha256.c
@@ -355,17 +355,18 @@
     mbedtls_sha256_init( &ctx );
 
     if( ( ret = mbedtls_sha256_starts_ext( &ctx, is224 ) ) != 0 )
-            return( ret );
+        goto exit;
 
     if( ( ret = mbedtls_sha256_update_ext( &ctx, input, ilen ) ) != 0 )
-            return( ret );
+        goto exit;
 
     if( ( ret = mbedtls_sha256_finish_ext( &ctx, output ) ) != 0 )
-            return( ret );
+        goto exit;
 
+exit:
     mbedtls_sha256_free( &ctx );
 
-    return( 0 );
+    return( ret );
 }
 
 #if defined(MBEDTLS_SELF_TEST)