mbedtls/tests/suites/test_suite_mpi.function

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
{
    mpi X;
    char str[1000];
    int len = 1000;

    mpi_init(&X, NULL);

    TEST_ASSERT( mpi_read_string( &X, {radix_X}, {input_X} ) == 0 );
    TEST_ASSERT( mpi_write_string( &X, {radix_A}, str, &len ) == 0 );
    TEST_ASSERT( strcmp( str, {input_A} ) == 0 );
}
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_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_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

/* 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 );
*/