61 行
1.9 KiB
PHP
61 行
1.9 KiB
PHP
<?php
|
|
|
|
namespace MediaWiki\Extension\RNRSHook;
|
|
|
|
use MediaWiki\Extension\AbuseFilter\Hooks\AbuseFilterAlterVariablesHook;
|
|
use MediaWiki\Extension\AbuseFilter\Hooks\AbuseFilterBuilderHook;
|
|
use MediaWiki\Extension\AbuseFilter\Variables\VariableHolder;
|
|
use Title;
|
|
use User;
|
|
use RequestContext;
|
|
use Wikimedia\IPUtils;
|
|
|
|
class AbuseFilterHooks implements AbuseFilterAlterVariablesHook, AbuseFilterBuilderHook
|
|
{
|
|
/**
|
|
* Determine if a given IP is an internal server node
|
|
*
|
|
* @param string|null $ip The IP address to check, or null to use the request IP
|
|
* @return bool True if an internal server node, false otherwise
|
|
*/
|
|
public static function isInternalNode($ip = null)
|
|
{
|
|
if ($ip == null) {
|
|
$ip = RequestContext::getMain()->getRequest()->getIP();
|
|
}
|
|
|
|
// IPv4 only
|
|
$low_ip_1 = '172.20.0.0';
|
|
$high_ip_1 = '172.20.0.255';
|
|
$low_ip_2 = '10.0.0.0';
|
|
$high_ip_2 = '10.0.16.255';
|
|
$low_ip_3 = '127.0.0.0';
|
|
$high_ip_3 = '127.255.255.255';
|
|
|
|
$result = (version_compare($low_ip_1, IPUtils::sanitizeIP($ip)) + version_compare(IPUtils::sanitizeIP($ip), $high_ip_1) === -2 ||
|
|
version_compare($low_ip_2, IPUtils::sanitizeIP($ip)) + version_compare(IPUtils::sanitizeIP($ip), $high_ip_2) === -2 ||
|
|
version_compare($low_ip_3, IPUtils::sanitizeIP($ip)) + version_compare(IPUtils::sanitizeIP($ip), $high_ip_3) === -2
|
|
);
|
|
|
|
return (bool)$result;
|
|
}
|
|
|
|
/**
|
|
* @inheritDoc
|
|
*/
|
|
public function onAbuseFilterAlterVariables(VariableHolder &$vars, Title $title, User $user)
|
|
{
|
|
$vars->setVar('internal_node', self::isInternalNode());
|
|
return true;
|
|
}
|
|
|
|
/**
|
|
* @inheritDoc
|
|
*/
|
|
public function onAbuseFilter_builder(array &$realValues)
|
|
{
|
|
$realValues['vars']['internal_node'] = 'internal-node';
|
|
return true;
|
|
}
|
|
}
|