mbedtls/tests/suites/test_suite_mpi.function

566 lines
15 KiB
Plaintext

BEGIN_HEADER
#include <polarssl/bignum.h>
static int myrand( void *rng_state )
{
if( rng_state != NULL )
rng_state = NULL;
return( rand() );
}
END_HEADER
BEGIN_CASE
mpi_read_write_string:radix_X:input_X:radix_A:input_A:output_size:result_read:result_write
{
mpi X;
char str[1000];
int len = {output_size};
mpi_init(&X, NULL);
TEST_ASSERT( mpi_read_string( &X, {radix_X}, {input_X} ) == {result_read} );
if( {result_read} == 0 )
{
TEST_ASSERT( mpi_write_string( &X, {radix_A}, str, &len ) == {result_write} );
if( {result_write} == 0 )
{
TEST_ASSERT( strcasecmp( str, {input_A} ) == 0 );
}
}
}
END_CASE
BEGIN_CASE
mpi_read_binary:input_X:radix_A:input_A
{
mpi X;
unsigned char str[1000];
unsigned char buf[1000];
int len = 1000;
int input_len;
mpi_init(&X, NULL);
input_len = unhexify( buf, {input_X} );
TEST_ASSERT( mpi_read_binary( &X, buf, input_len ) == 0 );
TEST_ASSERT( mpi_write_string( &X, {radix_A}, (char *) str, &len ) == 0 );
TEST_ASSERT( strcmp( (char *) str, {input_A} ) == 0 );
}
END_CASE
BEGIN_CASE
mpi_write_binary:radix_X:input_X:input_A:output_size:result
{
mpi X;
unsigned char str[1000];
unsigned char buf[1000];
int buflen;
memset( buf, 0x00, 1000 );
memset( str, 0x00, 1000 );
mpi_init(&X, NULL);
TEST_ASSERT( mpi_read_string( &X, {radix_X}, {input_X} ) == 0 );
buflen = mpi_size( &X );
if( buflen > {output_size} )
buflen = {output_size};
TEST_ASSERT( mpi_write_binary( &X, buf, buflen ) == {result} );
if( {result} == 0)
{
hexify( str, buf, buflen );
TEST_ASSERT( strcasecmp( (char *) str, {input_A} ) == 0 );
}
}
END_CASE
BEGIN_CASE
mpi_read_file:radix_X:input_file:input_A:result
{
mpi X;
unsigned char str[1000];
unsigned char buf[1000];
int buflen;
FILE *file;
memset( buf, 0x00, 1000 );
memset( str, 0x00, 1000 );
mpi_init(&X, NULL);
file = fopen( {input_file}, "r" );
TEST_ASSERT( mpi_read_file( &X, {radix_X}, file ) == {result} );
fclose(file);
if( {result} == 0 )
{
buflen = mpi_size( &X );
TEST_ASSERT( mpi_write_binary( &X, buf, buflen ) == 0 );
hexify( str, buf, buflen );
TEST_ASSERT( strcasecmp( (char *) str, {input_A} ) == 0 );
}
}
END_CASE
BEGIN_CASE
mpi_write_file:radix_X:input_X:output_radix:output_file
{
mpi X, Y;
FILE *file_out, *file_in;
mpi_init( &X, &Y, NULL );
TEST_ASSERT( mpi_read_string( &X, {radix_X}, {input_X} ) == 0 );
file_out = fopen( {output_file}, "w" );
TEST_ASSERT( mpi_write_file( NULL, &X, {output_radix}, file_out ) == 0 );
fclose(file_out);
file_in = fopen( {output_file}, "r" );
TEST_ASSERT( mpi_read_file( &Y, {output_radix}, file_in ) == 0 );
fclose(file_in);
TEST_ASSERT( mpi_cmp_mpi( &X, &Y ) == 0 );
}
END_CASE
BEGIN_CASE
mpi_lsb:radix_X:input_X:nr_bits
{
mpi X;
mpi_init(&X, NULL);
TEST_ASSERT( mpi_read_string( &X, {radix_X}, {input_X} ) == 0 );
TEST_ASSERT( mpi_lsb( &X ) == {nr_bits} );
}
END_CASE
BEGIN_CASE
mpi_msb:radix_X:input_X:nr_bits
{
mpi X;
mpi_init(&X, NULL);
TEST_ASSERT( mpi_read_string( &X, {radix_X}, {input_X} ) == 0 );
TEST_ASSERT( mpi_msb( &X ) == {nr_bits} );
}
END_CASE
BEGIN_CASE
mpi_gcd:radix_X:input_X:radix_Y:input_Y:radix_A:input_A
{
mpi A, X, Y, Z;
mpi_init(&A, &X, &Y, &Z, NULL);
TEST_ASSERT( mpi_read_string( &X, {radix_X}, {input_X} ) == 0 );
TEST_ASSERT( mpi_read_string( &Y, {radix_Y}, {input_Y} ) == 0 );
TEST_ASSERT( mpi_read_string( &A, {radix_A}, {input_A} ) == 0 );
TEST_ASSERT( mpi_gcd( &Z, &X, &Y ) == 0 );
TEST_ASSERT( mpi_cmp_mpi( &Z, &A ) == 0 );
}
END_CASE
BEGIN_CASE
mpi_cmp_int:input_X:input_A:result_CMP
{
mpi X;
mpi_init(&X, NULL);
TEST_ASSERT( mpi_lset( &X, {input_X} ) == 0);
TEST_ASSERT( mpi_cmp_int( &X, {input_A} ) == {result_CMP});
}
END_CASE
BEGIN_CASE
mpi_cmp_mpi:radix_X:input_X:radix_Y:input_Y:input_A
{
mpi X, Y;
mpi_init(&X, &Y, NULL);
TEST_ASSERT( mpi_read_string( &X, {radix_X}, {input_X} ) == 0 );
TEST_ASSERT( mpi_read_string( &Y, {radix_Y}, {input_Y} ) == 0 );
TEST_ASSERT( mpi_cmp_mpi( &X, &Y ) == {input_A} );
}
END_CASE
BEGIN_CASE
mpi_cmp_abs:radix_X:input_X:radix_Y:input_Y:input_A
{
mpi X, Y;
mpi_init(&X, &Y, NULL);
TEST_ASSERT( mpi_read_string( &X, {radix_X}, {input_X} ) == 0 );
TEST_ASSERT( mpi_read_string( &Y, {radix_Y}, {input_Y} ) == 0 );
TEST_ASSERT( mpi_cmp_abs( &X, &Y ) == {input_A} );
}
END_CASE
BEGIN_CASE
mpi_copy:input_X:input_A
{
mpi X, Y, A;
mpi_init(&X, &Y, &A, NULL);
TEST_ASSERT( mpi_lset( &X, {input_X} ) == 0 );
TEST_ASSERT( mpi_lset( &Y, {input_A} ) == 0 );
TEST_ASSERT( mpi_lset( &A, {input_A} ) == 0 );
TEST_ASSERT( mpi_cmp_mpi( &X, &Y ) != 0 );
TEST_ASSERT( mpi_cmp_mpi( &Y, &A ) == 0 );
TEST_ASSERT( mpi_copy( &Y, &X ) == 0 );
TEST_ASSERT( mpi_cmp_mpi( &X, &Y ) == 0 );
TEST_ASSERT( mpi_cmp_mpi( &Y, &A ) != 0 );
}
END_CASE
BEGIN_CASE
mpi_copy_self:input_X
{
mpi X;
mpi_init(&X, NULL);
TEST_ASSERT( mpi_lset( &X, {input_X} ) == 0 );
TEST_ASSERT( mpi_copy( &X, &X ) == 0 );
TEST_ASSERT( mpi_cmp_int( &X, {input_X} ) == 0 );
}
END_CASE
BEGIN_CASE
mpi_swap:input_X:input_Y
{
mpi X, Y, A;
mpi_init(&X, &Y, &A, NULL);
TEST_ASSERT( mpi_lset( &X, {input_X} ) == 0 );
TEST_ASSERT( mpi_lset( &Y, {input_Y} ) == 0 );
TEST_ASSERT( mpi_lset( &A, {input_X} ) == 0 );
TEST_ASSERT( mpi_cmp_mpi( &X, &Y ) != 0 );
TEST_ASSERT( mpi_cmp_mpi( &X, &A ) == 0 );
mpi_swap( &X, &Y );
TEST_ASSERT( mpi_cmp_mpi( &X, &Y ) != 0 );
TEST_ASSERT( mpi_cmp_mpi( &Y, &A ) == 0 );
}
END_CASE
BEGIN_CASE
mpi_add_mpi:radix_X:input_X:radix_Y:input_Y:radix_A:input_A
{
mpi X, Y, Z, A;
mpi_init(&X, &Y, &Z, &A, NULL);
TEST_ASSERT( mpi_read_string( &X, {radix_X}, {input_X} ) == 0 );
TEST_ASSERT( mpi_read_string( &Y, {radix_Y}, {input_Y} ) == 0 );
TEST_ASSERT( mpi_read_string( &A, {radix_A}, {input_A} ) == 0 );
TEST_ASSERT( mpi_add_mpi( &Z, &X, &Y ) == 0 );
TEST_ASSERT( mpi_cmp_mpi( &Z, &A ) == 0 );
}
END_CASE
BEGIN_CASE
mpi_add_abs:radix_X:input_X:radix_Y:input_Y:radix_A:input_A
{
mpi X, Y, Z, A;
mpi_init(&X, &Y, &Z, &A, NULL);
TEST_ASSERT( mpi_read_string( &X, {radix_X}, {input_X} ) == 0 );
TEST_ASSERT( mpi_read_string( &Y, {radix_Y}, {input_Y} ) == 0 );
TEST_ASSERT( mpi_read_string( &A, {radix_A}, {input_A} ) == 0 );
TEST_ASSERT( mpi_add_abs( &Z, &X, &Y ) == 0 );
TEST_ASSERT( mpi_cmp_mpi( &Z, &A ) == 0 );
}
END_CASE
BEGIN_CASE
mpi_add_abs_add_first:radix_X:input_X:radix_Y:input_Y:radix_A:input_A
{
mpi X, Y, A;
mpi_init(&X, &Y, &A, NULL);
TEST_ASSERT( mpi_read_string( &X, {radix_X}, {input_X} ) == 0 );
TEST_ASSERT( mpi_read_string( &Y, {radix_Y}, {input_Y} ) == 0 );
TEST_ASSERT( mpi_read_string( &A, {radix_A}, {input_A} ) == 0 );
TEST_ASSERT( mpi_add_abs( &X, &X, &Y ) == 0 );
TEST_ASSERT( mpi_cmp_mpi( &X, &A ) == 0 );
}
END_CASE
BEGIN_CASE
mpi_add_abs_add_second:radix_X:input_X:radix_Y:input_Y:radix_A:input_A
{
mpi X, Y, A;
mpi_init(&X, &Y, &A, NULL);
TEST_ASSERT( mpi_read_string( &X, {radix_X}, {input_X} ) == 0 );
TEST_ASSERT( mpi_read_string( &Y, {radix_Y}, {input_Y} ) == 0 );
TEST_ASSERT( mpi_read_string( &A, {radix_A}, {input_A} ) == 0 );
TEST_ASSERT( mpi_add_abs( &Y, &X, &Y ) == 0 );
TEST_ASSERT( mpi_cmp_mpi( &Y, &A ) == 0 );
}
END_CASE
BEGIN_CASE
mpi_add_int:radix_X:input_X:input_Y:radix_A:input_A
{
mpi X, Z, A;
mpi_init(&X, &Z, &A, NULL);
TEST_ASSERT( mpi_read_string( &X, {radix_X}, {input_X} ) == 0 );
TEST_ASSERT( mpi_read_string( &A, {radix_A}, {input_A} ) == 0 );
TEST_ASSERT( mpi_add_int( &Z, &X, {input_Y} ) == 0 );
TEST_ASSERT( mpi_cmp_mpi( &Z, &A ) == 0 );
}
END_CASE
BEGIN_CASE
mpi_sub_mpi:radix_X:input_X:radix_Y:input_Y:radix_A:input_A
{
mpi X, Y, Z, A;
mpi_init(&X, &Y, &Z, &A, NULL);
TEST_ASSERT( mpi_read_string( &X, {radix_X}, {input_X} ) == 0 );
TEST_ASSERT( mpi_read_string( &Y, {radix_Y}, {input_Y} ) == 0 );
TEST_ASSERT( mpi_read_string( &A, {radix_A}, {input_A} ) == 0 );
TEST_ASSERT( mpi_sub_mpi( &Z, &X, &Y ) == 0 );
TEST_ASSERT( mpi_cmp_mpi( &Z, &A ) == 0 );
}
END_CASE
BEGIN_CASE
mpi_sub_abs:radix_X:input_X:radix_Y:input_Y:radix_A:input_A:sub_result
{
mpi X, Y, Z, A;
int res;
mpi_init(&X, &Y, &Z, &A, NULL);
TEST_ASSERT( mpi_read_string( &X, {radix_X}, {input_X} ) == 0 );
TEST_ASSERT( mpi_read_string( &Y, {radix_Y}, {input_Y} ) == 0 );
TEST_ASSERT( mpi_read_string( &A, {radix_A}, {input_A} ) == 0 );
res = mpi_sub_abs( &Z, &X, &Y );
TEST_ASSERT( res == {sub_result} );
if( res == 0 )
TEST_ASSERT( mpi_cmp_mpi( &Z, &A ) == 0 );
}
END_CASE
BEGIN_CASE
mpi_sub_int:radix_X:input_X:input_Y:radix_A:input_A
{
mpi X, Z, A;
mpi_init(&X, &Z, &A, NULL);
TEST_ASSERT( mpi_read_string( &X, {radix_X}, {input_X} ) == 0 );
TEST_ASSERT( mpi_read_string( &A, {radix_A}, {input_A} ) == 0 );
TEST_ASSERT( mpi_sub_int( &Z, &X, {input_Y} ) == 0 );
TEST_ASSERT( mpi_cmp_mpi( &Z, &A ) == 0 );
}
END_CASE
BEGIN_CASE
mpi_mul_mpi:radix_X:input_X:radix_Y:input_Y:radix_A:input_A
{
mpi X, Y, Z, A;
mpi_init(&X, &Y, &Z, &A, NULL);
TEST_ASSERT( mpi_read_string( &X, {radix_X}, {input_X} ) == 0 );
TEST_ASSERT( mpi_read_string( &Y, {radix_Y}, {input_Y} ) == 0 );
TEST_ASSERT( mpi_read_string( &A, {radix_A}, {input_A} ) == 0 );
TEST_ASSERT( mpi_mul_mpi( &Z, &X, &Y ) == 0 );
TEST_ASSERT( mpi_cmp_mpi( &Z, &A ) == 0 );
}
END_CASE
BEGIN_CASE
mpi_mul_int:radix_X:input_X:input_Y:radix_A:input_A:result_comparison
{
mpi X, Z, A;
mpi_init(&X, &Z, &A, NULL);
TEST_ASSERT( mpi_read_string( &X, {radix_X}, {input_X} ) == 0 );
TEST_ASSERT( mpi_read_string( &A, {radix_A}, {input_A} ) == 0 );
TEST_ASSERT( mpi_mul_int( &Z, &X, {input_Y} ) == 0 );
TEST_ASSERT( mpi_cmp_mpi( &Z, &A ) {result_comparison} 0 );
}
END_CASE
BEGIN_CASE
mpi_div_mpi:radix_X:input_X:radix_Y:input_Y:radix_A:input_A:radix_B:input_B:div_result
{
mpi X, Y, Q, R, A, B;
int res;
mpi_init(&X, &Y, &Q, &R, &A, &B, NULL);
TEST_ASSERT( mpi_read_string( &X, {radix_X}, {input_X} ) == 0 );
TEST_ASSERT( mpi_read_string( &Y, {radix_Y}, {input_Y} ) == 0 );
TEST_ASSERT( mpi_read_string( &A, {radix_A}, {input_A} ) == 0 );
TEST_ASSERT( mpi_read_string( &B, {radix_B}, {input_B} ) == 0 );
res = mpi_div_mpi( &Q, &R, &X, &Y );
TEST_ASSERT( res == {div_result} );
if( res == 0 )
{
TEST_ASSERT( mpi_cmp_mpi( &Q, &A ) == 0 );
TEST_ASSERT( mpi_cmp_mpi( &R, &B ) == 0 );
}
}
END_CASE
BEGIN_CASE
mpi_div_int:radix_X:input_X:input_Y:radix_A:input_A:radix_B:input_B:div_result
{
mpi X, Q, R, A, B;
int res;
mpi_init(&X, &Q, &R, &A, &B, NULL);
TEST_ASSERT( mpi_read_string( &X, {radix_X}, {input_X} ) == 0 );
TEST_ASSERT( mpi_read_string( &A, {radix_A}, {input_A} ) == 0 );
TEST_ASSERT( mpi_read_string( &B, {radix_B}, {input_B} ) == 0 );
res = mpi_div_int( &Q, &R, &X, {input_Y} );
TEST_ASSERT( res == {div_result} );
if( res == 0 )
{
TEST_ASSERT( mpi_cmp_mpi( &Q, &A ) == 0 );
TEST_ASSERT( mpi_cmp_mpi( &R, &B ) == 0 );
}
}
END_CASE
BEGIN_CASE
mpi_mod_mpi:radix_X:input_X:radix_Y:input_Y:radix_A:input_A:div_result
{
mpi X, Y, Z, A;
int res;
mpi_init(&X, &Y, &Z, &A, NULL);
TEST_ASSERT( mpi_read_string( &X, {radix_X}, {input_X} ) == 0 );
TEST_ASSERT( mpi_read_string( &Y, {radix_Y}, {input_Y} ) == 0 );
TEST_ASSERT( mpi_read_string( &A, {radix_A}, {input_A} ) == 0 );
res = mpi_mod_mpi( &Z, &X, &Y );
TEST_ASSERT( res == {div_result} );
if( res == 0 )
{
TEST_ASSERT( mpi_cmp_mpi( &Z, &A ) == 0 );
}
}
END_CASE
BEGIN_CASE
mpi_mod_int:radix_X:input_X:input_Y:input_A:div_result
{
mpi X;
int res;
t_int r;
mpi_init(&X, NULL);
TEST_ASSERT( mpi_read_string( &X, {radix_X}, {input_X} ) == 0 );
res = mpi_mod_int( &r, &X, {input_Y} );
TEST_ASSERT( res == {div_result} );
if( res == 0 )
{
TEST_ASSERT( r == {input_A} );
}
}
END_CASE
BEGIN_CASE
mpi_exp_mod:radix_A:input_A:radix_E:input_E:radix_N:input_N:radix_RR:input_RR:radix_X:input_X:div_result
{
mpi A, E, N, RR, Z, X;
int res;
mpi_init(&A, &E, &N, &RR, &Z, &X, NULL);
TEST_ASSERT( mpi_read_string( &A, {radix_A}, {input_A} ) == 0 );
TEST_ASSERT( mpi_read_string( &E, {radix_E}, {input_E} ) == 0 );
TEST_ASSERT( mpi_read_string( &N, {radix_N}, {input_N} ) == 0 );
TEST_ASSERT( mpi_read_string( &X, {radix_X}, {input_X} ) == 0 );
if( strlen( {input_RR} ) )
TEST_ASSERT( mpi_read_string( &RR, {radix_RR}, {input_RR} ) == 0 );
res = mpi_exp_mod( &Z, &A, &E, &N, &RR );
TEST_ASSERT( res == {div_result} );
if( res == 0 )
{
TEST_ASSERT( mpi_cmp_mpi( &Z, &X ) == 0 );
}
}
END_CASE
BEGIN_CASE
mpi_inv_mod:radix_X:input_X:radix_Y:input_Y:radix_A:input_A:div_result
{
mpi X, Y, Z, A;
int res;
mpi_init(&X, &Y, &Z, &A, NULL);
TEST_ASSERT( mpi_read_string( &X, {radix_X}, {input_X} ) == 0 );
TEST_ASSERT( mpi_read_string( &Y, {radix_Y}, {input_Y} ) == 0 );
TEST_ASSERT( mpi_read_string( &A, {radix_A}, {input_A} ) == 0 );
res = mpi_inv_mod( &Z, &X, &Y );
TEST_ASSERT( res == {div_result} );
if( res == 0 )
{
TEST_ASSERT( mpi_cmp_mpi( &Z, &A ) == 0 );
}
}
END_CASE
BEGIN_CASE
mpi_is_prime:radix_X:input_X:div_result
{
mpi X;
int res;
mpi_init(&X, NULL);
TEST_ASSERT( mpi_read_string( &X, {radix_X}, {input_X} ) == 0 );
res = mpi_is_prime( &X, myrand, NULL );
TEST_ASSERT( res == {div_result} );
}
END_CASE
BEGIN_CASE
mpi_shift_l:radix_X:input_X:shift_X:radix_A:input_A
{
mpi X, A;
mpi_init(&X, &A, NULL);
TEST_ASSERT( mpi_read_string( &X, {radix_X}, {input_X} ) == 0 );
TEST_ASSERT( mpi_read_string( &A, {radix_A}, {input_A} ) == 0 );
TEST_ASSERT( mpi_shift_l( &X, {shift_X} ) == 0 );
TEST_ASSERT( mpi_cmp_mpi( &X, &A ) == 0 );
}
END_CASE
BEGIN_CASE
mpi_shift_r:radix_X:input_X:shift_X:radix_A:input_A
{
mpi X, A;
mpi_init(&X, &A, NULL);
TEST_ASSERT( mpi_read_string( &X, {radix_X}, {input_X} ) == 0 );
TEST_ASSERT( mpi_read_string( &A, {radix_A}, {input_A} ) == 0 );
TEST_ASSERT( mpi_shift_r( &X, {shift_X} ) == 0 );
TEST_ASSERT( mpi_cmp_mpi( &X, &A ) == 0 );
}
END_CASE
BEGIN_CASE
mpi_selftest:
{
TEST_ASSERT( mpi_self_test( 0 ) == 0 );
}
END_CASE
/* Helper Code
char str[1000];
int len = 1000;
mpi_write_string(&Z, 10, str, &len);
printf("Z: %d %s\n", Z.s, str);
TEST_ASSERT( mpi_cmp_mpi( &Z, &A ) == 0 );
*/