Browse Source

Add metadata to \OCP\AppFramework\Http\Response::throttle

Fixes https://github.com/nextcloud/server/issues/5891

Signed-off-by: Lukas Reschke <lukas@statuscode.ch>
tags/v13.0.0beta1
Lukas Reschke 6 years ago
parent
commit
f22ab3e665
No account linked to committer's email address

+ 1
- 1
core/Controller/LoginController.php View File

@@ -248,7 +248,7 @@ class LoginController extends Controller {
$args['redirect_url'] = $redirect_url;
}
$response = new RedirectResponse($this->urlGenerator->linkToRoute('core.login.showLoginForm', $args));
$response->throttle();
$response->throttle(['user' => $user]);
$this->session->set('loginMessages', [
['invalidpassword'], []
]);

+ 1
- 1
lib/private/AppFramework/Middleware/Security/BruteForceMiddleware.php View File

@@ -75,7 +75,7 @@ class BruteForceMiddleware extends Middleware {
$action = $this->reflector->getAnnotationParameter('BruteForceProtection', 'action');
$ip = $this->request->getRemoteAddress();
$this->throttler->sleepDelay($ip, $action);
$this->throttler->registerAttempt($action, $ip);
$this->throttler->registerAttempt($action, $ip, $response->getThrottleMetadata());
}

return parent::afterController($controller, $methodName, $response);

+ 15
- 1
lib/public/AppFramework/Http/Response.php View File

@@ -83,6 +83,8 @@ class Response {

/** @var bool */
private $throttled = false;
/** @var array */
private $throttleMetadata = [];

/**
* Caches the response
@@ -328,10 +330,22 @@ class Response {
* Marks the response as to throttle. Will be throttled when the
* @BruteForceProtection annotation is added.
*
* @param array $metadata
* @since 12.0.0
*/
public function throttle() {
public function throttle(array $metadata = []) {
$this->throttled = true;
$this->throttleMetadata = $metadata;
}

/**
* Returns the throttle metadata, defaults to empty array
*
* @return array
* @since 13.0.0
*/
public function getThrottleMetadata() {
return $this->throttleMetadata;
}

/**

+ 2
- 2
tests/Core/Controller/LoginControllerTest.php View File

@@ -307,7 +307,7 @@ class LoginControllerTest extends TestCase {
->method('deleteUserValue');

$expected = new \OCP\AppFramework\Http\RedirectResponse($loginPageUrl);
$expected->throttle();
$expected->throttle(['user' => 'MyUserName']);
$this->assertEquals($expected, $this->loginController->tryLogin($user, $password, '/apps/files'));
}

@@ -634,7 +634,7 @@ class LoginControllerTest extends TestCase {
->method('createRememberMeToken');

$expected = new RedirectResponse('');
$expected->throttle();
$expected->throttle(['user' => 'john']);
$this->assertEquals($expected, $this->loginController->tryLogin('john@doe.com', 'just wrong', null));
}
}

+ 5
- 0
tests/lib/AppFramework/Http/ResponseTest.php View File

@@ -269,4 +269,9 @@ class ResponseTest extends \Test\TestCase {
$this->childResponse->throttle();
$this->assertTrue($this->childResponse->isThrottled());
}

public function testGetThrottleMetadata() {
$this->childResponse->throttle(['foo' => 'bar']);
$this->assertSame(['foo' => 'bar'], $this->childResponse->getThrottleMetadata());
}
}

+ 4
- 0
tests/lib/AppFramework/Middleware/Security/BruteForceMiddlewareTest.php View File

@@ -112,6 +112,10 @@ class BruteForceMiddlewareTest extends TestCase {
->expects($this->once())
->method('isThrottled')
->willReturn(true);
$response
->expects($this->once())
->method('getThrottleMetadata')
->willReturn([]);
$this->reflector
->expects($this->once())
->method('getAnnotationParameter')

Loading…
Cancel
Save