feat(security): Allow to opt-out of ratelimit protection, e.g. for testing on CI

Signed-off-by: Joas Schilling <coding@schilljs.com>

diff --git a/config/config.sample.php b/config/config.sample.php
index 9d79b154120c8c6ae6ae9796013dffd0de6b0562..cb9e5b0f5a63910b8137db16db54c26e0dcd8ee0 100644 (file)
--- a/config/config.sample.php
+++ b/config/config.sample.php
@@ -295,7 +295,7 @@ $CONFIG = [
 
 /**
  * The interval at which token activity should be updated.
- * Increasing this value means that the last activty on the security page gets
+ * Increasing this value means that the last activity on the security page gets
  * more outdated.
  *
  * Tokens are still checked every 5 minutes for validity

diff --git a/lib/private/Security/RateLimiting/Backend/DatabaseBackend.php b/lib/private/Security/RateLimiting/Backend/DatabaseBackend.php
index ea4bd87d6cd9ae676162336ec3b5284d52e33d94..d1631a8d0ae2d17d927ff3da88ad9f98b3baa830 100644 (file)
--- a/lib/private/Security/RateLimiting/Backend/DatabaseBackend.php
+++ b/lib/private/Security/RateLimiting/Backend/DatabaseBackend.php
@@ -3,8 +3,10 @@
 declare(strict_types=1);
 
 /**
+ * @copyright Copyright (c) 2023 Joas Schilling <coding@schilljs.com>
  * @copyright Copyright (c) 2021 Lukas Reschke <lukas@statuscode.ch>
  *
+ * @author Joas Schilling <coding@schilljs.com>
  * @author Lukas Reschke <lukas@statuscode.ch>
  *
  * @license GNU AGPL version 3 or any later version
@@ -27,24 +29,25 @@ namespace OC\Security\RateLimiting\Backend;
 
 use OCP\AppFramework\Utility\ITimeFactory;
 use OCP\DB\QueryBuilder\IQueryBuilder;
+use OCP\IConfig;
 use OCP\IDBConnection;
 
 class DatabaseBackend implements IBackend {
        private const TABLE_NAME = 'ratelimit_entries';
 
+       /** @var IConfig */
+       private $config;
        /** @var IDBConnection */
        private $dbConnection;
        /** @var ITimeFactory */
        private $timeFactory;
 
-       /**
-        * @param IDBConnection $dbConnection
-        * @param ITimeFactory $timeFactory
-        */
        public function __construct(
+               IConfig $config,
                IDBConnection $dbConnection,
                ITimeFactory $timeFactory
        ) {
+               $this->config = $config;
                $this->dbConnection = $dbConnection;
                $this->timeFactory = $timeFactory;
        }
@@ -115,7 +118,12 @@ class DatabaseBackend implements IBackend {
                        ->values([
                                'hash' => $qb->createNamedParameter($identifier, IQueryBuilder::PARAM_STR),
                                'delete_after' => $qb->createNamedParameter($deleteAfter, IQueryBuilder::PARAM_DATE),
-                       ])
-                       ->executeStatement();
+                       ]);
+
+               if (!$this->config->getSystemValueBool('ratelimit.protection.enabled', true)) {
+                       return;
+               }
+
+               $qb->executeStatement();
        }
 }

diff --git a/lib/private/Security/RateLimiting/Backend/MemoryCacheBackend.php b/lib/private/Security/RateLimiting/Backend/MemoryCacheBackend.php
index f4880fb239c16ecd3c9a470a284f4dc86ff67a0a..4bcb459c64e0ff566b0924e5460132745e9c6437 100644 (file)
--- a/lib/private/Security/RateLimiting/Backend/MemoryCacheBackend.php
+++ b/lib/private/Security/RateLimiting/Backend/MemoryCacheBackend.php
@@ -3,9 +3,11 @@
 declare(strict_types=1);
 
 /**
+ * @copyright Copyright (c) 2023 Joas Schilling <coding@schilljs.com>
  * @copyright Copyright (c) 2017 Lukas Reschke <lukas@statuscode.ch>
  *
  * @author Christoph Wurst <christoph@winzerhof-wurst.at>
+ * @author Joas Schilling <coding@schilljs.com>
  * @author Lukas Reschke <lukas@statuscode.ch>
  * @author Morris Jobke <hey@morrisjobke.de>
  * @author Roeland Jago Douma <roeland@famdouma.nl>
@@ -31,6 +33,7 @@ namespace OC\Security\RateLimiting\Backend;
 use OCP\AppFramework\Utility\ITimeFactory;
 use OCP\ICache;
 use OCP\ICacheFactory;
+use OCP\IConfig;
 
 /**
  * Class MemoryCacheBackend uses the configured distributed memory cache for storing
@@ -39,17 +42,18 @@ use OCP\ICacheFactory;
  * @package OC\Security\RateLimiting\Backend
  */
 class MemoryCacheBackend implements IBackend {
+       /** @var IConfig */
+       private $config;
        /** @var ICache */
        private $cache;
        /** @var ITimeFactory */
        private $timeFactory;
 
-       /**
-        * @param ICacheFactory $cacheFactory
-        * @param ITimeFactory $timeFactory
-        */
-       public function __construct(ICacheFactory $cacheFactory,
-                                                               ITimeFactory $timeFactory) {
+       public function __construct(
+               IConfig $config,
+               ICacheFactory $cacheFactory,
+               ITimeFactory $timeFactory) {
+               $this->config = $config;
                $this->cache = $cacheFactory->createDistributed(__CLASS__);
                $this->timeFactory = $timeFactory;
        }
@@ -121,6 +125,11 @@ class MemoryCacheBackend implements IBackend {
 
                // Store the new attempt
                $existingAttempts[] = (string)($currentTime + $period);
+
+               if (!$this->config->getSystemValueBool('ratelimit.protection.enabled', true)) {
+                       return;
+               }
+
                $this->cache->set($identifier, json_encode($existingAttempts));
        }
 }

diff --git a/lib/private/Server.php b/lib/private/Server.php
index 9846822a5089145810cebc662af9c2fca545b07c..9e51d32cd56cba40ca911a59deb36a3cc8b4138d 100644 (file)
--- a/lib/private/Server.php
+++ b/lib/private/Server.php
@@ -841,11 +841,13 @@ class Server extends ServerContainer implements IServerContainer {
                        $cacheFactory = $c->get(ICacheFactory::class);
                        if ($cacheFactory->isAvailable()) {
                                $backend = new \OC\Security\RateLimiting\Backend\MemoryCacheBackend(
+                                       $c->get(AllConfig::class),
                                        $this->get(ICacheFactory::class),
                                        new \OC\AppFramework\Utility\TimeFactory()
                                );
                        } else {
                                $backend = new \OC\Security\RateLimiting\Backend\DatabaseBackend(
+                                       $c->get(AllConfig::class),
                                        $c->get(IDBConnection::class),
                                        new \OC\AppFramework\Utility\TimeFactory()
                                );

diff --git a/tests/lib/Security/RateLimiting/Backend/MemoryCacheBackendTest.php b/tests/lib/Security/RateLimiting/Backend/MemoryCacheBackendTest.php
index e7cb4a93a299ef97d58dca4555a7046cd560780a..a48b1211587b106f06d0eb1a2e92d83e053baa90 100644 (file)
--- a/tests/lib/Security/RateLimiting/Backend/MemoryCacheBackendTest.php
+++ b/tests/lib/Security/RateLimiting/Backend/MemoryCacheBackendTest.php
@@ -28,9 +28,12 @@ use OC\Security\RateLimiting\Backend\MemoryCacheBackend;
 use OCP\AppFramework\Utility\ITimeFactory;
 use OCP\ICache;
 use OCP\ICacheFactory;
+use OCP\IConfig;
 use Test\TestCase;
 
 class MemoryCacheBackendTest extends TestCase {
+       /** @var IConfig|\PHPUnit\Framework\MockObject\MockObject */
+       private $config;
        /** @var ICacheFactory|\PHPUnit\Framework\MockObject\MockObject */
        private $cacheFactory;
        /** @var ITimeFactory|\PHPUnit\Framework\MockObject\MockObject */
@@ -43,6 +46,7 @@ class MemoryCacheBackendTest extends TestCase {
        protected function setUp(): void {
                parent::setUp();
 
+               $this->config = $this->createMock(IConfig::class);
                $this->cacheFactory = $this->createMock(ICacheFactory::class);
                $this->timeFactory = $this->createMock(ITimeFactory::class);
                $this->cache = $this->createMock(ICache::class);
@@ -53,7 +57,12 @@ class MemoryCacheBackendTest extends TestCase {
                        ->with('OC\Security\RateLimiting\Backend\MemoryCacheBackend')
                        ->willReturn($this->cache);
 
+               $this->config->method('getSystemValueBool')
+                       ->with('ratelimit.protection.enabled')
+                       ->willReturn(true);
+
                $this->memoryCache = new MemoryCacheBackend(
+                       $this->config,
                        $this->cacheFactory,
                        $this->timeFactory
                );