commit bbafd34ebb94b67ccd86b972dac266ccb563a0b3
author Andres Amaya Garcia <Andres.AmayaGarcia@arm.com> Wed Jul 05 14:25:21 2017 +0100
committer Andres Amaya Garcia <Andres.AmayaGarcia@arm.com> Wed Jul 05 14:25:21 2017 +0100
tree c152b2cbe2acf152405b57f66ed63773976b6d9a
parent af0b31d76faa9f1a23bd46a2afc2cbf020b7361c

Set len var to 0 when buf is freed in ssl_tls.c

library/ssl_tls.c

diff --git a/library/ssl_tls.c b/library/ssl_tls.c
index 9b5fccb..c85cc72 100644
--- a/library/ssl_tls.c
+++ b/library/ssl_tls.c
@@ -6049,13 +6049,19 @@
         return( MBEDTLS_ERR_SSL_BAD_INPUT_DATA );
     }
 
-    if( conf->psk != NULL || conf->psk_identity != NULL )
+    if( conf->psk != NULL )
     {
         mbedtls_zeroize( conf->psk, conf->psk_len );
+
         mbedtls_free( conf->psk );
-        mbedtls_free( conf->psk_identity );
         conf->psk = NULL;
+        conf->psk_len = 0;
+    }
+    if( conf->psk_identity != NULL )
+    {
+        mbedtls_free( conf->psk_identity );
         conf->psk_identity = NULL;
+        conf->psk_identity_len = 0;
     }
 
     if( ( conf->psk = mbedtls_calloc( 1, psk_len ) ) == NULL ||
@@ -6090,6 +6096,7 @@
     {
         mbedtls_zeroize( ssl->handshake->psk, ssl->handshake->psk_len );
         mbedtls_free( ssl->handshake->psk );
+        ssl->handshake->psk_len = 0;
     }
 
     if( ( ssl->handshake->psk = mbedtls_calloc( 1, psk_len ) ) == NULL )