mirror of
https://we.phorge.it/source/phorge.git
synced 2024-12-23 22:10:55 +01:00
3f53718d10
Summary: Depends on D18702. Ref T13008. This replaces the old hard-coded single rate limit with multiple flexible limits, and defines two types of limits: - Rate: reject requests if a client has completed too many requests recently. - Connection: reject requests if a client has too many more connections than disconnections recently. The connection limit adds +1 to the score for each connection, then adds -1 for each disconnection. So the overall number is how many open connections they have, at least approximately. Supporting multiple limits will let us do limiting by Hostname and by remote address (e.g., a specific IP can't exceed a low limit, and all requests to a hostname can't exceed a higher limit). Configuring the new limits looks something like this: ``` PhabricatorStartup::addRateLimit(new PhabricatorClientRateLimit()) ->setLimitKey('rate') ->setClientKey($_SERVER['REMOTE_ADDR']) ->setLimit(5); PhabricatorStartup::addRateLimit(new PhabricatorClientConnectionLimit()) ->setLimitKey('conn') ->setClientKey($_SERVER['REMOTE_ADDR']) ->setLimit(2); ``` Test Plan: - Configured limits as above. - Made a lot of requests, got cut off by the rate limit. - Used `curl --limit-rate -F 'data=@the_letter_m.txt' ...` to upload files really slowly. Got cut off by the connection limit. With `enable_post_data_reading` off, this correctly killed the connections //before// the uploads finished. - I'll send this stuff to `secure` before production to give it more of a chance. Reviewers: amckinley Reviewed By: amckinley Maniphest Tasks: T13008 Differential Revision: https://secure.phabricator.com/D18703
58 lines
1.2 KiB
PHP
58 lines
1.2 KiB
PHP
<?php
|
|
|
|
final class PhabricatorClientRateLimit
|
|
extends PhabricatorClientLimit {
|
|
|
|
protected function getBucketDuration() {
|
|
return 60;
|
|
}
|
|
|
|
protected function getBucketCount() {
|
|
return 5;
|
|
}
|
|
|
|
protected function shouldRejectConnection($score) {
|
|
$limit = $this->getLimit();
|
|
|
|
// Reject connections if the average score across all buckets exceeds the
|
|
// limit.
|
|
$average_score = $score / $this->getBucketCount();
|
|
|
|
return ($average_score > $limit);
|
|
}
|
|
|
|
protected function getConnectScore() {
|
|
return 0;
|
|
}
|
|
|
|
protected function getPenaltyScore() {
|
|
return 1;
|
|
}
|
|
|
|
protected function getDisconnectScore(array $request_state) {
|
|
$score = 1;
|
|
|
|
// If the user was logged in, let them make more requests.
|
|
if (isset($request_state['viewer'])) {
|
|
$viewer = $request_state['viewer'];
|
|
if ($viewer->isLoggedIn()) {
|
|
$score = 0.25;
|
|
}
|
|
}
|
|
|
|
return $score;
|
|
}
|
|
|
|
protected function getRateLimitReason($score) {
|
|
$client_key = $this->getClientKey();
|
|
|
|
// NOTE: This happens before we load libraries, so we can not use pht()
|
|
// here.
|
|
|
|
return
|
|
"TOO MANY REQUESTS\n".
|
|
"You (\"{$client_key}\") are issuing too many requests ".
|
|
"too quickly.\n";
|
|
}
|
|
|
|
}
|