<?php

namespace Mdanter\Ecc\Tests\Curves;

use Mdanter\Ecc\Math\GmpMathInterface;
use Mdanter\Ecc\Tests\AbstractTestCase;
use Mdanter\Ecc\EccFactory;
use Mdanter\Ecc\Crypto\Signature\Signer;
use Mdanter\Ecc\Random\RandomGeneratorFactory;

class SecCurveTest extends AbstractTestCase
{

    public function getCurveParams()
    {
        return $this->_getAdapters([
            [ 'curve256k1', '0', '7', '115792089237316195423570985008687907853269984665640564039457584007908834671663' ],
            [ 'curve256r1', '115792089210356248762697446949407573530086143415290314195533631308867097853948', '41058363725152142129326129780047268409114441015993725554835256314039467401291', '115792089210356248762697446949407573530086143415290314195533631308867097853951' ],
            [ 'curve384r1', '39402006196394479212279040100143613805079739270465446667948293404245721771496870329047266088258938001861606973112316', '27580193559959705877849011840389048093056905856361568521428707301988689241309860865136260764883745107765439761230575', '39402006196394479212279040100143613805079739270465446667948293404245721771496870329047266088258938001861606973112319' ],
        ]);
    }

    /**
     *
     * @dataProvider getCurveParams
     */
    public function testCurveGeneration(GmpMathInterface $math, $function, $a, $b, $prime)
    {
        $factory = EccFactory::getSecgCurves($math);
        $curve = $factory->{$function}();

        $this->assertInstanceOf($this->classCurveFpInterface, $curve);
        $this->assertEquals($a, $math->toString($curve->getA()));
        $this->assertEquals($b, $math->toString($curve->getB()));
        $this->assertEquals($prime, $math->toString($curve->getPrime()));
    }

    public function getGeneratorParams()
    {
        return $this->_getAdapters([
            [ 'generator256k1', '115792089237316195423570985008687907852837564279074904382605163141518161494337', '115792089237316195423570985008687907853269984665640564039457584007908834671663' ],
            [ 'generator256r1', '115792089210356248762697446949407573529996955224135760342422259061068512044369', '115792089210356248762697446949407573530086143415290314195533631308867097853951' ],
            [ 'generator384r1', '39402006196394479212279040100143613805079739270465446667946905279627659399113263569398956308152294913554433653942643', '39402006196394479212279040100143613805079739270465446667948293404245721771496870329047266088258938001861606973112319' ],
        ]);
    }

    /**
     *
     * @dataProvider getGeneratorParams
     */
    public function testGeneratorGeneration(GmpMathInterface $math, $function, $order, $prime)
    {
        $factory = EccFactory::getSecgCurves($math);
        $generator = $factory->{$function}();

        $this->assertInstanceOf($this->classPointInterface, $generator);
        $this->assertEquals($order, $math->toString($generator->getOrder()));
        $this->assertEquals($prime, $math->toString($generator->getCurve()->getPrime()));
    }

    /**
     *
     * @dataProvider getAdapters
     */
    public function testSecp256r1EquivalenceToNistP256(GmpMathInterface $adapter)
    {
        $secpFactory = EccFactory::getSecgCurves($adapter);
        $nistFactory = EccFactory::getNistCurves($adapter);

        $signer = new Signer($adapter);

        $secret = gmp_init('DC51D3866A15BACDE33D96F992FCA99DA7E6EF0934E7097559C27F1614C88A7F', 16);

        $secpKey = $secpFactory->generator256r1()->getPrivateKeyFrom($secret);
        $nistKey = $nistFactory->generator256()->getPrivateKeyFrom($secret);

        $randomK = RandomGeneratorFactory::getRandomGenerator()->generate($secpKey->getPoint()->getOrder());
        $message = RandomGeneratorFactory::getRandomGenerator()->generate($secpKey->getPoint()->getOrder());

        $sigSecp = $signer->sign($secpKey, $message, $randomK);
        $sigNist = $signer->sign($nistKey, $message, $randomK);

        $this->assertTrue($adapter->equals($sigNist->getR(), $sigSecp->getR()));
        $this->assertTrue($adapter->equals($sigNist->getS(), $sigSecp->getS()));
    }

    /**
     * @dataProvider getAdapters
     */
    public function testSecp384r1EquivalenceToNistP384(GmpMathInterface $adapter)
    {
        $secpFactory = EccFactory::getSecgCurves($adapter);
        $nistFactory = EccFactory::getNistCurves($adapter);

        $signer = new Signer($adapter);

        $secret = gmp_init('DC51D3866A15BACDE33D96F992FCA99DA7E6EF0934E7097559C27F1614C88A7F', 16);

        $secpKey = $secpFactory->generator384r1()->getPrivateKeyFrom($secret);
        $nistKey = $nistFactory->generator384()->getPrivateKeyFrom($secret);

        $randomK = RandomGeneratorFactory::getRandomGenerator()->generate($secpKey->getPoint()->getOrder());
        $message = RandomGeneratorFactory::getRandomGenerator()->generate($secpKey->getPoint()->getOrder());

        $sigSecp = $signer->sign($secpKey, $message, $randomK);
        $sigNist = $signer->sign($nistKey, $message, $randomK);

        $this->assertTrue($adapter->equals($sigNist->getR(), $sigSecp->getR()));
        $this->assertTrue($adapter->equals($sigNist->getS(), $sigSecp->getS()));
    }
}
