mirror of
https://we.phorge.it/source/phorge.git
synced 2024-11-14 02:42:40 +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
60 lines
1.9 KiB
PHP
60 lines
1.9 KiB
PHP
<?php
|
|
|
|
phabricator_startup();
|
|
|
|
try {
|
|
PhabricatorStartup::beginStartupPhase('libraries');
|
|
PhabricatorStartup::loadCoreLibraries();
|
|
|
|
PhabricatorStartup::beginStartupPhase('purge');
|
|
PhabricatorCaches::destroyRequestCache();
|
|
|
|
PhabricatorStartup::beginStartupPhase('sink');
|
|
$sink = new AphrontPHPHTTPSink();
|
|
|
|
try {
|
|
PhabricatorStartup::beginStartupPhase('run');
|
|
AphrontApplicationConfiguration::runHTTPRequest($sink);
|
|
} catch (Exception $ex) {
|
|
try {
|
|
$response = new AphrontUnhandledExceptionResponse();
|
|
$response->setException($ex);
|
|
|
|
PhabricatorStartup::endOutputCapture();
|
|
$sink->writeResponse($response);
|
|
} catch (Exception $response_exception) {
|
|
// If we hit a rendering exception, ignore it and throw the original
|
|
// exception. It is generally more interesting and more likely to be
|
|
// the root cause.
|
|
throw $ex;
|
|
}
|
|
}
|
|
} catch (Exception $ex) {
|
|
PhabricatorStartup::didEncounterFatalException('Core Exception', $ex, false);
|
|
}
|
|
|
|
function phabricator_startup() {
|
|
// Load the PhabricatorStartup class itself.
|
|
$t_startup = microtime(true);
|
|
$root = dirname(dirname(__FILE__));
|
|
require_once $root.'/support/startup/PhabricatorStartup.php';
|
|
|
|
// Load client limit classes so the preamble can configure limits.
|
|
require_once $root.'/support/startup/PhabricatorClientLimit.php';
|
|
require_once $root.'/support/startup/PhabricatorClientRateLimit.php';
|
|
require_once $root.'/support/startup/PhabricatorClientConnectionLimit.php';
|
|
|
|
// If the preamble script exists, load it.
|
|
$t_preamble = microtime(true);
|
|
$preamble_path = $root.'/support/preamble.php';
|
|
if (file_exists($preamble_path)) {
|
|
require_once $preamble_path;
|
|
}
|
|
|
|
$t_hook = microtime(true);
|
|
PhabricatorStartup::didStartup($t_startup);
|
|
|
|
PhabricatorStartup::recordStartupPhase('startup.init', $t_startup);
|
|
PhabricatorStartup::recordStartupPhase('preamble', $t_preamble);
|
|
PhabricatorStartup::recordStartupPhase('hook', $t_hook);
|
|
}
|