Check if the root domain is an IP address

This commit is contained in:
Nabeel Shahzad
2020-08-15 13:50:58 -04:00
parent d3e6d2c49a
commit be72571185
2 changed files with 11 additions and 1 deletions

View File

@@ -118,6 +118,14 @@ class Utils
$extract = new Extract(); $extract = new Extract();
$result = $extract->parse($url); $result = $extract->parse($url);
return $result->getRegistrableDomain(); $val = $result->getRegistrableDomain();
if (!empty($val)) {
return $val;
}
// Couldn't validate a domain, see if this is an IP address?
if (filter_var($url, FILTER_VALIDATE_IP)) {
return $url;
}
} }
} }

View File

@@ -86,6 +86,7 @@ class UtilsTest extends TestCase
$tests = [ $tests = [
'http://phpvms.net', 'http://phpvms.net',
'https://phpvms.net', 'https://phpvms.net',
'https://phpvms.net/',
'phpvms.net', 'phpvms.net',
'https://phpvms.net/index.php', 'https://phpvms.net/index.php',
'https://demo.phpvms.net', 'https://demo.phpvms.net',
@@ -98,5 +99,6 @@ class UtilsTest extends TestCase
$this->assertEquals('phpvms.co.uk', Utils::getRootDomain('http://phpvms.co.uk')); $this->assertEquals('phpvms.co.uk', Utils::getRootDomain('http://phpvms.co.uk'));
$this->assertEquals('phpvms.co.uk', Utils::getRootDomain('http://www.phpvms.co.uk')); $this->assertEquals('phpvms.co.uk', Utils::getRootDomain('http://www.phpvms.co.uk'));
$this->assertEquals('52.249.193.184', Utils::getRootDomain('http://52.249.193.184'));
} }
} }