commit cbe3d0d5cc78d1a6367a628028c108111e9b2320
author Paul Bakker <p.j.bakker@polarssl.org> Thu Apr 17 16:00:59 2014 +0200
committer Paul Bakker <p.j.bakker@polarssl.org> Thu Apr 17 16:00:59 2014 +0200
tree 0f796a735e69507f1c3f56d3ab21ac4969003652
parent 4f42c11846bcd8783d2a07ad30ce979eb23add22

Added return value checking for correctness in programs

programs/pkey/dh_genprime.c

diff --git a/programs/pkey/dh_genprime.c b/programs/pkey/dh_genprime.c
index f51465a..1f9d730 100644
--- a/programs/pkey/dh_genprime.c
+++ b/programs/pkey/dh_genprime.c
@@ -65,7 +65,12 @@
     ((void) argv);
 
     mpi_init( &G ); mpi_init( &P ); mpi_init( &Q );
-    mpi_read_string( &G, 10, GENERATOR );
+
+    if( ( ret = mpi_read_string( &G, 10, GENERATOR ) ) != 0 )
+    {
+        printf( " failed\n  ! mpi_read_string returned %d\n", ret );
+        goto exit;
+    }
 
     printf( "\nWARNING: You should not generate and use your own DHM primes\n" );
     printf( "         unless you are very certain of what you are doing!\n" );