fix: Handle 2fa enforcement earlier

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

Author: nickvergessen

lib/private/User/Session.php

@@ -402,10 +402,8 @@ public function logClientIn($user,
 			return false;
 		}
 
-		if (!$isTokenPassword && $this->isTokenAuthEnforced()) {
-			throw new PasswordLoginForbiddenException();
-		}
-		if (!$isTokenPassword && $this->isTwoFactorEnforced($user)) {
+		if (!$isTokenPassword && ($this->isTokenAuthEnforced() || $this->isTwoFactorEnforced($user))) {
+			$this->handleLoginFailed($throttler, $currentDelay, $remoteAddress, $user, $password);
 			throw new PasswordLoginForbiddenException();
 		}
 
@@ -586,7 +584,8 @@ public function tryBasicAuthLogin(IRequest $request,
 				// If credentials were provided, they need to be valid, otherwise we do boom
 				throw new LoginException();
 			} catch (PasswordLoginForbiddenException $ex) {
-				// Nothing to do
+				// If credentials were provided, they need to be valid, otherwise we do boom
+				throw new LoginException(previous: $ex);
 			}
 		}
 		return false;

tests/lib/User/SessionTest.php

@@ -441,7 +441,7 @@ public function testLogClientInNoTokenPasswordWith2fa() {
 			->method('getRemoteAddress')
 			->willReturn('192.168.0.1');
 		$this->throttler
-			->expects($this->once())
+			->expects($this->exactly(2))
 			->method('sleepDelayOrThrowOnMax')
 			->with('192.168.0.1');
 		$this->throttler
@@ -450,6 +450,15 @@ public function testLogClientInNoTokenPasswordWith2fa() {
 			->with('192.168.0.1')
 			->willReturn(0);
 
+		$this->throttler
+			->expects($this->once())
+			->method('registerAttempt')
+			->with('login', '192.168.0.1', ['user' => 'john']);
+		$this->dispatcher
+			->expects($this->once())
+			->method('dispatchTyped')
+			->with(new LoginFailed('john', 'doe'));
+
 		$userSession->logClientIn('john', 'doe', $request, $this->throttler);
 	}
 
@@ -553,7 +562,7 @@ public function testLogClientInNoTokenPasswordNo2fa() {
 			->method('getRemoteAddress')
 			->willReturn('192.168.0.1');
 		$this->throttler
-			->expects($this->once())
+			->expects($this->exactly(2))
 			->method('sleepDelayOrThrowOnMax')
 			->with('192.168.0.1');
 		$this->throttler
@@ -562,6 +571,15 @@ public function testLogClientInNoTokenPasswordNo2fa() {
 			->with('192.168.0.1')
 			->willReturn(0);
 
+		$this->throttler
+			->expects($this->once())
+			->method('registerAttempt')
+			->with('login', '192.168.0.1', ['user' => 'john']);
+		$this->dispatcher
+			->expects($this->once())
+			->method('dispatchTyped')
+			->with(new LoginFailed('john', 'doe'));
+
 		$userSession->logClientIn('john', 'doe', $request, $this->throttler);
 	}