commit b5b20f19e7191f50a88edadde217f64d8a58e192
author Paul Bakker <p.j.bakker@polarssl.org> Sun Sep 16 15:07:49 2012 +0000
committer Paul Bakker <p.j.bakker@polarssl.org> Sun Sep 16 15:07:49 2012 +0000
tree d1311a71e015b0fb12b253212f19945f234d7388
parent 0c93d126bcd6a61fdb1d7fda08e99d3379f93ce5

 - Extra sanity check for input added

library/dhm.c

diff --git a/library/dhm.c b/library/dhm.c
index 4a59571..90e5b4b 100644
--- a/library/dhm.c
+++ b/library/dhm.c
@@ -128,6 +128,9 @@
     size_t n1, n2, n3;
     unsigned char *p;
 
+    if( mpi_cmp_int( &ctx->P, 0 ) == 0 )
+        return( POLARSSL_ERR_DHM_BAD_INPUT_DATA );
+
     /*
      * Generate X as large as possible ( < P )
      */
@@ -211,6 +214,9 @@
     if( ctx == NULL || olen < 1 || olen > ctx->len )
         return( POLARSSL_ERR_DHM_BAD_INPUT_DATA );
 
+    if( mpi_cmp_int( &ctx->P, 0 ) == 0 )
+        return( POLARSSL_ERR_DHM_BAD_INPUT_DATA );
+
     /*
      * generate X and calculate GX = G^X mod P
      */