2012-04-29 13:15:06 -04:00
|
|
|
<?php
|
2021-05-04 04:29:25 -04:00
|
|
|
|
2012-04-29 13:15:06 -04:00
|
|
|
/**
|
2016-07-11 05:58:15 -04:00
|
|
|
* PrivateBin
|
2012-04-29 13:15:06 -04:00
|
|
|
*
|
|
|
|
* a zero-knowledge paste bin
|
|
|
|
*
|
2016-07-11 05:58:15 -04:00
|
|
|
* @link https://github.com/PrivateBin/PrivateBin
|
2012-04-29 13:15:06 -04:00
|
|
|
* @copyright 2012 Sébastien SAUVAGE (sebsauvage.net)
|
2016-07-19 07:56:52 -04:00
|
|
|
* @license https://www.opensource.org/licenses/zlib-license.php The zlib/libpng License
|
2021-04-05 10:44:12 -04:00
|
|
|
* @version 1.3.5
|
2012-04-29 13:15:06 -04:00
|
|
|
*/
|
2016-12-12 12:43:23 -05:00
|
|
|
|
2016-12-12 12:49:08 -05:00
|
|
|
namespace PrivateBin\Persistence;
|
2016-08-09 05:54:42 -04:00
|
|
|
|
|
|
|
use PrivateBin\Configuration;
|
2016-07-21 11:09:48 -04:00
|
|
|
|
2012-04-29 13:15:06 -04:00
|
|
|
/**
|
2016-08-09 05:54:42 -04:00
|
|
|
* TrafficLimiter
|
2012-04-29 13:15:06 -04:00
|
|
|
*
|
|
|
|
* Handles traffic limiting, so no user does more than one call per 10 seconds.
|
|
|
|
*/
|
2016-08-09 05:54:42 -04:00
|
|
|
class TrafficLimiter extends AbstractPersistence
|
2012-04-29 13:15:06 -04:00
|
|
|
{
|
|
|
|
/**
|
2015-08-16 09:55:31 -04:00
|
|
|
* time limit in seconds, defaults to 10s
|
|
|
|
*
|
2012-04-29 13:15:06 -04:00
|
|
|
* @access private
|
|
|
|
* @static
|
|
|
|
* @var int
|
|
|
|
*/
|
|
|
|
private static $_limit = 10;
|
2021-05-04 05:18:06 -04:00
|
|
|
|
2021-05-04 04:29:25 -04:00
|
|
|
/**
|
|
|
|
* listed ips are exempted from limits, defaults to null
|
|
|
|
*
|
|
|
|
* @access private
|
|
|
|
* @static
|
|
|
|
* @var array
|
|
|
|
*/
|
|
|
|
private static $_exemptedIp = null;
|
2012-04-29 13:15:06 -04:00
|
|
|
|
2015-09-26 11:57:46 -04:00
|
|
|
/**
|
|
|
|
* key to fetch IP address
|
|
|
|
*
|
|
|
|
* @access private
|
|
|
|
* @static
|
|
|
|
* @var string
|
|
|
|
*/
|
|
|
|
private static $_ipKey = 'REMOTE_ADDR';
|
|
|
|
|
2012-04-29 13:15:06 -04:00
|
|
|
/**
|
|
|
|
* set the time limit in seconds
|
|
|
|
*
|
|
|
|
* @access public
|
|
|
|
* @static
|
|
|
|
* @param int $limit
|
|
|
|
*/
|
|
|
|
public static function setLimit($limit)
|
|
|
|
{
|
|
|
|
self::$_limit = $limit;
|
|
|
|
}
|
|
|
|
|
2021-05-04 04:29:25 -04:00
|
|
|
/**
|
2021-05-04 05:01:46 -04:00
|
|
|
* set a list of ip(ranges) as string
|
2021-05-04 04:29:25 -04:00
|
|
|
*
|
|
|
|
* @access public
|
|
|
|
* @static
|
2021-05-04 05:01:46 -04:00
|
|
|
* @param string $exemptedIps
|
2021-05-04 04:29:25 -04:00
|
|
|
*/
|
|
|
|
public static function setExemptedIp($exemptedIp)
|
|
|
|
{
|
|
|
|
self::$_exemptedIp = $exemptedIp;
|
|
|
|
}
|
|
|
|
|
2015-09-26 11:57:46 -04:00
|
|
|
/**
|
|
|
|
* set configuration options of the traffic limiter
|
|
|
|
*
|
|
|
|
* @access public
|
|
|
|
* @static
|
2016-08-09 05:54:42 -04:00
|
|
|
* @param Configuration $conf
|
2015-09-26 11:57:46 -04:00
|
|
|
*/
|
2016-08-09 05:54:42 -04:00
|
|
|
public static function setConfiguration(Configuration $conf)
|
2015-09-26 11:57:46 -04:00
|
|
|
{
|
|
|
|
self::setLimit($conf->getKey('limit', 'traffic'));
|
|
|
|
self::setPath($conf->getKey('dir', 'traffic'));
|
2021-05-04 04:29:25 -04:00
|
|
|
self::setExemptedIp($conf->getKey('exemptedIp', 'traffic'));
|
|
|
|
|
2016-07-26 02:19:35 -04:00
|
|
|
if (($option = $conf->getKey('header', 'traffic')) !== null) {
|
2015-09-26 11:57:46 -04:00
|
|
|
$httpHeader = 'HTTP_' . $option;
|
2016-07-26 02:19:35 -04:00
|
|
|
if (array_key_exists($httpHeader, $_SERVER) && !empty($_SERVER[$httpHeader])) {
|
2015-09-26 11:57:46 -04:00
|
|
|
self::$_ipKey = $httpHeader;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
2016-08-10 11:41:46 -04:00
|
|
|
* get a HMAC of the current visitors IP address
|
2015-09-26 11:57:46 -04:00
|
|
|
*
|
|
|
|
* @access public
|
|
|
|
* @static
|
2016-08-10 11:41:46 -04:00
|
|
|
* @param string $algo
|
2015-09-26 11:57:46 -04:00
|
|
|
* @return string
|
|
|
|
*/
|
2016-08-10 11:41:46 -04:00
|
|
|
public static function getHash($algo = 'sha512')
|
2015-09-26 11:57:46 -04:00
|
|
|
{
|
2016-08-10 11:41:46 -04:00
|
|
|
return hash_hmac($algo, $_SERVER[self::$_ipKey], ServerSalt::get());
|
2015-09-26 11:57:46 -04:00
|
|
|
}
|
|
|
|
|
2021-05-06 06:13:03 -04:00
|
|
|
/**
|
|
|
|
* Validate $_ipKey against configured ipranges. If matched ratelimiter will ignore ip
|
|
|
|
*
|
|
|
|
* @access private
|
|
|
|
* @static
|
|
|
|
* @param string $algo
|
|
|
|
* @return string
|
|
|
|
*/
|
|
|
|
private static function matchIp($ipRange = null)
|
|
|
|
{
|
|
|
|
// Match $_ipKey to $ipRange and if it matches it will return with a true
|
|
|
|
$address = \IPLib\Factory::addressFromString($_SERVER[self::$_ipKey]);
|
|
|
|
$range = \IPLib\Factory::rangeFromString(trim($ipRange));
|
|
|
|
// If $range is null something went wrong (possible invalid ip given in config)
|
|
|
|
if ($range == null) {
|
|
|
|
return false;
|
|
|
|
} else {
|
|
|
|
// Ip-lib does throws and exception when something goes wrong, if so we want to catch it and set contained to false
|
|
|
|
try {
|
|
|
|
$contained = $address->matches($range);
|
|
|
|
} catch (\Exception $e) {
|
|
|
|
// If something is wrong with matching the ip, we set $contained to false
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
// Matches return true!
|
|
|
|
if ($contained === true) {
|
|
|
|
return true;
|
|
|
|
} else {
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2012-04-29 13:15:06 -04:00
|
|
|
/**
|
|
|
|
* traffic limiter
|
|
|
|
*
|
|
|
|
* Make sure the IP address makes at most 1 request every 10 seconds.
|
|
|
|
*
|
|
|
|
* @access public
|
|
|
|
* @static
|
2018-11-19 07:07:25 -05:00
|
|
|
* @throws \Exception
|
2012-04-29 13:15:06 -04:00
|
|
|
* @return bool
|
|
|
|
*/
|
2015-09-26 11:57:46 -04:00
|
|
|
public static function canPass()
|
2012-04-29 13:15:06 -04:00
|
|
|
{
|
2015-09-26 11:57:46 -04:00
|
|
|
// disable limits if set to less then 1
|
2016-07-26 02:19:35 -04:00
|
|
|
if (self::$_limit < 1) {
|
|
|
|
return true;
|
|
|
|
}
|
2021-05-06 06:13:03 -04:00
|
|
|
error_reporting(-1);
|
2021-05-04 04:29:25 -04:00
|
|
|
// Check if $_ipKey is exempted from ratelimiting
|
|
|
|
if (!is_null(self::$_exemptedIp)) {
|
2021-05-04 05:14:11 -04:00
|
|
|
$exIp_array = explode(',', self::$_exemptedIp);
|
2021-05-04 04:29:25 -04:00
|
|
|
foreach ($exIp_array as $ipRange) {
|
2021-05-06 06:13:03 -04:00
|
|
|
if (self::matchIp($ipRange) === true)
|
|
|
|
{
|
2021-05-04 04:29:25 -04:00
|
|
|
return true;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2013-10-30 18:54:42 -04:00
|
|
|
|
2013-10-31 20:15:14 -04:00
|
|
|
$file = 'traffic_limiter.php';
|
2018-07-29 09:43:28 -04:00
|
|
|
if (self::_exists($file)) {
|
|
|
|
require self::getPath($file);
|
|
|
|
$tl = $GLOBALS['traffic_limiter'];
|
|
|
|
} else {
|
|
|
|
$tl = array();
|
2012-04-29 13:15:06 -04:00
|
|
|
}
|
|
|
|
|
2016-08-10 11:41:46 -04:00
|
|
|
// purge file of expired hashes to keep it small
|
2018-07-29 09:43:28 -04:00
|
|
|
$now = time();
|
2016-07-26 02:19:35 -04:00
|
|
|
foreach ($tl as $key => $time) {
|
|
|
|
if ($time + self::$_limit < $now) {
|
2012-04-29 13:15:06 -04:00
|
|
|
unset($tl[$key]);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-07-29 09:43:28 -04:00
|
|
|
// this hash is used as an array key, hence a shorter algo is used
|
2016-08-10 11:41:46 -04:00
|
|
|
$hash = self::getHash('sha256');
|
|
|
|
if (array_key_exists($hash, $tl) && ($tl[$hash] + self::$_limit >= $now)) {
|
2012-04-29 13:15:06 -04:00
|
|
|
$result = false;
|
|
|
|
} else {
|
2016-08-10 11:41:46 -04:00
|
|
|
$tl[$hash] = time();
|
2016-08-15 10:45:47 -04:00
|
|
|
$result = true;
|
2012-04-29 13:15:06 -04:00
|
|
|
}
|
2013-10-31 20:15:14 -04:00
|
|
|
self::_store(
|
2012-04-29 13:15:06 -04:00
|
|
|
$file,
|
|
|
|
'<?php' . PHP_EOL .
|
2018-07-29 09:43:28 -04:00
|
|
|
'$GLOBALS[\'traffic_limiter\'] = ' . var_export($tl, true) . ';'
|
2012-04-29 13:15:06 -04:00
|
|
|
);
|
|
|
|
return $result;
|
|
|
|
}
|
|
|
|
}
|