commit b72b4edec1f742a42962e25cda9414d2d15b471a
author Manuel Pégourié-Gonnard <mpg@elzevir.fr> Fri Sep 13 13:55:26 2013 +0200
committer Manuel Pégourié-Gonnard <mpg@elzevir.fr> Fri Sep 13 13:55:26 2013 +0200
tree a1cf9a230a8d2a80f6d5ea3a62f4c39cf3b9116d
parent 4fe9200f471a5fa47900418eaf41b8f7535dfbc4

Fix memory leak in DHM

library/dhm.c

diff --git a/library/dhm.c b/library/dhm.c
index b95f899..f43b047 100644
--- a/library/dhm.c
+++ b/library/dhm.c
@@ -98,7 +98,7 @@
 {
     int ret;
 
-    memset( ctx, 0, sizeof( dhm_context ) );
+    dhm_free( ctx );
 
     if( ( ret = dhm_read_bignum( &ctx->P,  p, end ) ) != 0 ||
         ( ret = dhm_read_bignum( &ctx->G,  p, end ) ) != 0 ||
@@ -364,10 +364,12 @@
  */
 void dhm_free( dhm_context *ctx )
 {
-    mpi_free( &ctx->Vi ); mpi_free( &ctx->Vf );
+    mpi_free( &ctx->_X); mpi_free( &ctx->Vf ); mpi_free( &ctx->Vi );
     mpi_free( &ctx->RP ); mpi_free( &ctx->K ); mpi_free( &ctx->GY );
     mpi_free( &ctx->GX ); mpi_free( &ctx->X ); mpi_free( &ctx->G );
     mpi_free( &ctx->P );
+
+    memset( ctx, 0, sizeof( dhm_context ) );
 }
 
 #if defined(POLARSSL_SELF_TEST)