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