<?php

/**
 * Tests for the Diffie-Hellman key exchange implementation in the
 * OpenID library.
 *
 * PHP versions 4 and 5
 *
 * LICENSE: See the COPYING file included in this distribution.
 *
 * @package OpenID
 * @author JanRain, Inc. <openid@janrain.com>
 * @copyright 2005-2008 Janrain, Inc.
 * @license http://www.apache.org/licenses/LICENSE-2.0 Apache
 */

require_once 'Auth/OpenID/DiffieHellman.php';
require_once 'Tests/Auth/OpenID/TestUtil.php';

class Tests_Auth_OpenID_DiffieHellman_CheckCases extends PHPUnit_Framework_TestCase
{
	function __construct($cases, $n)
	{
		$this->cases = $cases;
		$this->n = $n;
	}

	function runTest()
	{
		$this->assertEquals($this->n, count($this->cases));
	}
}

class Tests_Auth_OpenID_DiffieHellman_Private extends PHPUnit_Framework_TestCase
{
	function __construct($name, $input, $expected)
	{
		$this->setName("$name");
		$this->input = $input;
		$this->expected = $expected;
	}

	function runTest()
	{
		$lib =& Auth_OpenID_getMathLib();
		$dh = new Auth_OpenID_DiffieHellman(null, null, $this->input);
		$this->assertEquals($lib->cmp($this->expected, $dh->getPublicKey()), 0);
	}
}

class Tests_Auth_OpenID_DiffieHellman_Exch extends PHPUnit_Framework_TestCase
{
	function __construct($name, $p1, $p2, $shared)
	{
		$this->setName("$name");
		$this->p1 = $p1;
		$this->p2 = $p2;
		$this->shared = $shared;
	}

	function runTest()
	{
		$lib = Auth_OpenID_getMathLib();
		$shared = $lib->init($this->shared);
		$dh1 = new Auth_OpenID_DiffieHellman(null, null, $this->p1);
		$dh2 = new Auth_OpenID_DiffieHellman(null, null, $this->p2);
		$sh1 = $dh1->getSharedSecret($dh2->getPublicKey());
		$sh2 = $dh2->getSharedSecret($dh1->getPublicKey());
		$this->assertEquals($lib->cmp($shared, $sh1), 0);
		$this->assertEquals($lib->cmp($shared, $sh2), 0);
	}
}

class Tests_Auth_OpenID_DiffieHellman extends PHPUnit_Framework_TestSuite
{
	function __construct($name)
	{
		$this->setName($name);

		$priv_cases = Tests_Auth_OpenID_DiffieHellman::_readPrivateTestCases();
		$sanity = new Tests_Auth_OpenID_DiffieHellman_CheckCases(
			$priv_cases, 29);
		$sanity->setName('Check parsing of priv test data');
		$this->addTest($sanity);

		$exch_cases = Tests_Auth_OpenID_DiffieHellman::_readExchTestCases();
		$sanity = new Tests_Auth_OpenID_DiffieHellman_CheckCases(
			$exch_cases, 25);
		$sanity->setName('Check parsing of exch test data');
		$this->addTest($sanity);

		if (!defined('Auth_OpenID_NO_MATH_SUPPORT')) {
			if (defined('Tests_Auth_OpenID_thorough')) {
				$npriv = count($priv_cases);
				$nexch = count($exch_cases);
			} else {
				$npriv = 1;
				$nexch = 3;
			}

			for ($i = 0; $i < $npriv; $i++) {
				list($input, $expected) = $priv_cases[$i];
				$one = new Tests_Auth_OpenID_DiffieHellman_Private(
					"DHPriv $i", $input, $expected);
				$this->addTest($one);
			}

			for ($i = 0; $i < $nexch; $i++) {
				$case = $exch_cases[$i];
				$one = new Tests_Auth_OpenID_DiffieHellman_Exch(
					$i, $case[0], $case[1], $case[2]);
				$this->addTest($one);
			}
		}
	}

	function _readPrivateTestCases()
	{
		$lines = Tests_Auth_OpenID_readlines('dhpriv');
		$cases = array();
		foreach ($lines as $line) {
			$case = array();
			if (!preg_match('/^(\d+) (\d+)\n$/', $line, $case)) {
				trigger_error("Bad test input: $line", E_USER_ERROR);
			}

			$c = count($case);
			if ($c != 3) {
				trigger_error("Wrong number of elements in parsed case: $c",
					E_USER_ERROR);
			}

			array_shift($case);
			$cases[] = $case;
		}

		return $cases;
	}

	function _readExchTestCases()
	{
		$lines = Tests_Auth_OpenID_readlines('dhexch');
		$cases = array();
		foreach ($lines as $line) {
			$case = array();
			if (!preg_match('/^(\d+) (\d+) (\d+)\n$/', $line, $case)) {
				trigger_error("Bad test input: $line", E_USER_ERROR);
			}

			$c = count($case);
			if ($c != 4) {
				trigger_error("Wrong number of elements in parsed case: $c",
					E_USER_ERROR);
			}

			array_shift($case);
			$cases[] = $case;
		}
		return $cases;
	}
}


