diff --git a/phpseclib/Math/BigInteger.php b/phpseclib/Math/BigInteger.php index 4fd88909..71fe5eef 100644 --- a/phpseclib/Math/BigInteger.php +++ b/phpseclib/Math/BigInteger.php @@ -3088,19 +3088,31 @@ class Math_BigInteger /** * Generate a random number * - * @param optional Integer $min - * @param optional Integer $max + * Returns a random number between $min and $max where $min and $max + * can be defined using one of the two methods: + * + * $min->random($max) + * $max->random($min) + * + * @param Math_BigInteger $arg1 + * @param optional Math_BigInteger $arg2 * @return Math_BigInteger * @access public + * @internal The API for creating random numbers used to be $a->random($min, $max), where $a was a Math_BigInteger object. + * That method is still supported for BC purposes. */ - function random($min = false, $max = false) + function random($arg1, $arg2 = false) { - if ($min === false) { - $min = new Math_BigInteger(0); + if ($arg1 === false) { + return false; } - if ($max === false) { - $max = new Math_BigInteger(0x7FFFFFFF); + if ($arg2 === false) { + $max = $arg1; + $min = $this; + } else { + $min = $arg1; + $max = $arg2; } $compare = $max->compare($min); @@ -3163,21 +3175,25 @@ class Math_BigInteger * If there's not a prime within the given range, false will be returned. If more than $timeout seconds have elapsed, * give up and return false. * - * @param optional Integer $min - * @param optional Integer $max + * @param Math_BigInteger $arg1 + * @param optional Math_BigInteger $arg2 * @param optional Integer $timeout - * @return Math_BigInteger + * @return Mixed * @access public * @internal See {@link http://www.cacr.math.uwaterloo.ca/hac/about/chap4.pdf#page=15 HAC 4.44}. */ - function randomPrime($min = false, $max = false, $timeout = false) + function randomPrime($arg1, $arg2 = false, $timeout = false) { - if ($min === false) { - $min = new Math_BigInteger(0); + if ($arg1 === false) { + return false; } - if ($max === false) { - $max = new Math_BigInteger(0x7FFFFFFF); + if ($arg2 === false) { + $max = $arg1; + $min = $this; + } else { + $min = $arg1; + $max = $arg2; } $compare = $max->compare($min); @@ -3289,7 +3305,7 @@ class Math_BigInteger * $t parameter is distributability. Math_BigInteger::randomPrime() can be distributed across multiple pageloads * on a website instead of just one. * - * @param optional Integer $t + * @param optional Math_BigInteger $t * @return Boolean * @access public * @internal Uses the diff --git a/tests/Unit/Math/BigInteger/TestCase.php b/tests/Unit/Math/BigInteger/TestCase.php index d233dfb4..87addfd4 100644 --- a/tests/Unit/Math/BigInteger/TestCase.php +++ b/tests/Unit/Math/BigInteger/TestCase.php @@ -266,6 +266,34 @@ abstract class Unit_Math_BigInteger_TestCase extends PhpseclibTestCase $this->assertSame('18446744073709551616', (string) $y); } + public function testRandomTwoArgument() + { + $min = $this->getInstance(0); + $max = $this->getInstance('18446744073709551616'); + + $rand1 = $min->random($min, $max); + // technically $rand1 can equal $min but with the $min and $max we've + // chosen it's just not that likely + $this->assertTrue($rand1->compare($min) > 0); + $this->assertTrue($rand1->compare($max) < 0); + } + + public function testRandomOneArgument() + { + $min = $this->getInstance(0); + $max = $this->getInstance('18446744073709551616'); + + $rand1 = $min->random($max); + $this->assertTrue($rand1->compare($min) > 0); + $this->assertTrue($rand1->compare($max) < 0); + + $rand2 = $max->random($min); + $this->assertTrue($rand2->compare($min) > 0); + $this->assertTrue($rand2->compare($max) < 0); + + $this->assertFalse($rand1->equals($rand2)); + } + /** * @group github279 */