#include <lnv3/lnv3.h>
#include <nttl/modSqrt.h>
#include <nttl/randomPrime.h>


#include "validate.h"
#include "funcList.h"


bool
ValidateModSqrt( const string& operation )
{
    if( operation != "modSqrt" ) return false;

    ln a, p, s;
    cin >> a >> p >> s;

    ln sqrt = Sqrt( a, p );
    CheckForError( sqrt, s, "modSqrt" );

    return true;
}


static TestFunctionList RegFunc1( ValidateModSqrt );


bool
ValidateModSqrtRandom( const string& operation )
{
    if( operation != "modSqrtRandom" ) return false;

    long count, size;
    cin >> count >> size;

    for( long i=0 ; i<count ; i++ )
    {
        ln p;
        RandomPrime( &p, size );
        ln a = ln( ).Random( size );
        a %= p;

        ln r = Sqrt( a, p );

        if( Jacobi( a, p ) == -1 )
            CheckForError( r, ln(0), "modSqrtRandom" );
        else
            CheckForError( r * r % p, a, "modSqrtRandom" );
    }

    return true;
}


static TestFunctionList RegFunc2( ValidateModSqrtRandom );