yuzu/mbedtls
1
0
Fork 0
mirror of https://github.com/yuzu-emu/mbedtls.git synced 2025-05-13 17:22:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

Backport dh_genprime update from 2.0

Browse source
This commit is contained in:
Manuel Pégourié-Gonnard 2015-07-03 17:14:06 +02:00
parent 56e245d959
commit f0dd045bbe
1 changed files with 50 additions and 27 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

77
programs/pkey/dh_genprime.c
View file

@ -33,24 +33,6 @@
#define polarssl_printf printf #define polarssl_printf printf
#endif #endif
#if defined(POLARSSL_BIGNUM_C) && defined(POLARSSL_ENTROPY_C) && \
defined(POLARSSL_FS_IO) && defined(POLARSSL_CTR_DRBG_C) && \
defined(POLARSSL_GENPRIME)
#include "polarssl/bignum.h"
#include "polarssl/entropy.h"
#include "polarssl/ctr_drbg.h"
#include <stdio.h>
#include <string.h>
#endif
/*
* Note: G = 4 is always a quadratic residue mod P,
* so it is a generator of order Q (with P = 2*Q+1).
*/
#define DH_P_SIZE 1024
#define GENERATOR "4"
#if !defined(POLARSSL_BIGNUM_C) || !defined(POLARSSL_ENTROPY_C) || \ #if !defined(POLARSSL_BIGNUM_C) || !defined(POLARSSL_ENTROPY_C) || \
!defined(POLARSSL_FS_IO) || !defined(POLARSSL_CTR_DRBG_C) || \ !defined(POLARSSL_FS_IO) || !defined(POLARSSL_CTR_DRBG_C) || \
!defined(POLARSSL_GENPRIME) !defined(POLARSSL_GENPRIME)
@ -62,7 +44,28 @@ int main( void )
return( 0 ); return( 0 );
} }
#else #else
int main( void )
#include "polarssl/bignum.h"
#include "polarssl/entropy.h"
#include "polarssl/ctr_drbg.h"
#include <stdio.h>
#include <string.h>
#define USAGE \
"\n usage: dh_genprime param=<>...\n" \
"\n acceprable parameters:\n" \
" bits=%%d default: 2048\n"
#define DFL_BITS 2048
/*
* Note: G = 4 is always a quadratic residue mod P,
* so it is a generator of order Q (with P = 2*Q+1).
*/
#define GENERATOR "4"
int main( int argc, char **argv )
{ {
int ret = 1; int ret = 1;
mpi G, P, Q; mpi G, P, Q;
@ -70,23 +73,43 @@ int main( void )
ctr_drbg_context ctr_drbg; ctr_drbg_context ctr_drbg;
const char *pers = "dh_genprime"; const char *pers = "dh_genprime";
FILE *fout; FILE *fout;
int nbits = DFL_BITS;
int i;
char *p, *q;
mpi_init( &G ); mpi_init( &P ); mpi_init( &Q ); mpi_init( &G ); mpi_init( &P ); mpi_init( &Q );
entropy_init( &entropy ); entropy_init( &entropy );
if( argc == 0 )
{
usage:
polarssl_printf( USAGE );
return( 1 );
}
for( i = 1; i < argc; i++ )
{
p = argv[i];
if( ( q = strchr( p, '=' ) ) == NULL )
goto usage;
*q++ = '\0';
if( strcmp( p, "bits" ) == 0 )
{
nbits = atoi( q );
if( nbits < 0 || nbits > POLARSSL_MPI_MAX_BITS )
goto usage;
}
else
goto usage;
}
if( ( ret = mpi_read_string( &G, 10, GENERATOR ) ) != 0 ) if( ( ret = mpi_read_string( &G, 10, GENERATOR ) ) != 0 )
{ {
polarssl_printf( " failed\n ! mpi_read_string returned %d\n", ret ); polarssl_printf( " failed\n ! mpi_read_string returned %d\n", ret );
goto exit; goto exit;
} }
polarssl_printf( "\nWARNING: You should not generate and use your own DHM primes\n" );
polarssl_printf( " unless you are very certain of what you are doing!\n" );
polarssl_printf( " Failing to follow this instruction may result in\n" );
polarssl_printf( " weak security for your connections! Use the\n" );
polarssl_printf( " predefined DHM parameters from dhm.h instead!\n\n" );
polarssl_printf( "============================================================\n\n" );
polarssl_printf( " ! Generating large primes may take minutes!\n" ); polarssl_printf( " ! Generating large primes may take minutes!\n" );
polarssl_printf( "\n . Seeding the random number generator..." ); polarssl_printf( "\n . Seeding the random number generator..." );
@ -106,7 +129,7 @@ int main( void )
/* /*
* This can take a long time... * This can take a long time...
*/ */
if( ( ret = mpi_gen_prime( &P, DH_P_SIZE, 1, if( ( ret = mpi_gen_prime( &P, nbits, 1,
ctr_drbg_random, &ctr_drbg ) ) != 0 ) ctr_drbg_random, &ctr_drbg ) ) != 0 )
{ {
polarssl_printf( " failed\n ! mpi_gen_prime returned %d\n\n", ret ); polarssl_printf( " failed\n ! mpi_gen_prime returned %d\n\n", ret );