81 lines
2.7 KiB
PHP
81 lines
2.7 KiB
PHP
|
<?php
|
||
|
|
/**
|
||
|
|
* These helpful tests were lifted from the Symfony library
|
||
|
|
* https://github.com/symfony/http-foundation/blob/master/LICENSE
|
||
|
|
*
|
||
|
|
* (c) Fabien Potencier <fabien@symfony.com>
|
||
|
|
*
|
||
|
|
* For the full copyright and license information, please view the LICENSE
|
||
|
|
* file that was distributed with this source code.
|
||
|
|
*/
|
||
|
|
use SilverStripe\Control\Util\IPUtils;
|
||
|
|
|
||
|
|
class IPUtilsTest extends SapphireTest {
|
||
|
|
/**
|
||
|
|
* @dataProvider testIPv4Provider
|
||
|
|
*/
|
||
|
|
public function testIPv4($matches, $remoteAddr, $cidr)
|
||
|
|
{
|
||
|
|
$this->assertSame($matches, IPUtils::checkIP($remoteAddr, $cidr));
|
||
|
|
}
|
||
|
|
|
||
|
|
public function testIPv4Provider()
|
||
|
|
{
|
||
|
|
return array(
|
||
|
|
array(true, '192.168.1.1', '192.168.1.1'),
|
||
|
|
array(true, '192.168.1.1', '192.168.1.1/1'),
|
||
|
|
array(true, '192.168.1.1', '192.168.1.0/24'),
|
||
|
|
array(false, '192.168.1.1', '1.2.3.4/1'),
|
||
|
|
array(false, '192.168.1.1', '192.168.1.1/33'), // invalid subnet
|
||
|
|
array(true, '192.168.1.1', array('1.2.3.4/1', '192.168.1.0/24')),
|
||
|
|
array(true, '192.168.1.1', array('192.168.1.0/24', '1.2.3.4/1')),
|
||
|
|
array(false, '192.168.1.1', array('1.2.3.4/1', '4.3.2.1/1')),
|
||
|
|
array(true, '1.2.3.4', '0.0.0.0/0'),
|
||
|
|
array(true, '1.2.3.4', '192.168.1.0/0'),
|
||
|
|
array(false, '1.2.3.4', '256.256.256/0'), // invalid CIDR notation
|
||
|
|
array(false, 'an_invalid_ip', '192.168.1.0/24'),
|
||
|
|
);
|
||
|
|
}
|
||
|
|
|
||
|
|
/**
|
||
|
|
* @dataProvider testIPv6Provider
|
||
|
|
*/
|
||
|
|
public function testIPv6($matches, $remoteAddr, $cidr)
|
||
|
|
{
|
||
|
|
if (!defined('AF_INET6')) {
|
||
|
|
$this->markTestSkipped('Only works when PHP is compiled without the option "disable-ipv6".');
|
||
|
|
}
|
||
|
|
|
||
|
|
$this->assertSame($matches, IPUtils::checkIP($remoteAddr, $cidr));
|
||
|
|
}
|
||
|
|
|
||
|
|
public function testIPv6Provider()
|
||
|
|
{
|
||
|
|
return array(
|
||
|
|
array(true, '2a01:198:603:0:396e:4789:8e99:890f', '2a01:198:603:0::/65'),
|
||
|
|
array(false, '2a00:198:603:0:396e:4789:8e99:890f', '2a01:198:603:0::/65'),
|
||
|
|
array(false, '2a01:198:603:0:396e:4789:8e99:890f', '::1'),
|
||
|
|
array(true, '0:0:0:0:0:0:0:1', '::1'),
|
||
|
|
array(false, '0:0:603:0:396e:4789:8e99:0001', '::1'),
|
||
|
|
array(true, '2a01:198:603:0:396e:4789:8e99:890f', array('::1', '2a01:198:603:0::/65')),
|
||
|
|
array(true, '2a01:198:603:0:396e:4789:8e99:890f', array('2a01:198:603:0::/65', '::1')),
|
||
|
|
array(false, '2a01:198:603:0:396e:4789:8e99:890f', array('::1', '1a01:198:603:0::/65')),
|
||
|
|
array(false, '}__test|O:21:"JDatabaseDriverMysqli":3:{s:2', '::1'),
|
||
|
|
array(false, '2a01:198:603:0:396e:4789:8e99:890f', 'unknown'),
|
||
|
|
);
|
||
|
|
}
|
||
|
|
|
||
|
|
/**
|
||
|
|
* @expectedException \RuntimeException
|
||
|
|
* @requires extension sockets
|
||
|
|
*/
|
||
|
|
public function testAnIPv6WithOptionDisabledIPv6()
|
||
|
|
{
|
||
|
|
if (defined('AF_INET6')) {
|
||
|
|
$this->markTestSkipped('Only works when PHP is compiled with the option "disable-ipv6".');
|
||
|
|
}
|
||
|
|
|
||
|
|
IPUtils::checkIP('2a01:198:603:0:396e:4789:8e99:890f', '2a01:198:603:0::/65');
|
||
|
|
}
|
||
|
|
}
|