MfaControllerTest.php

Go to the documentation of this file.

<?php
 
declare(strict_types=1);
 
/*
 * This file is part of the TYPO3 CMS project.
 *
 * It is free software; you can redistribute it and/or modify it under
 * the terms of the GNU General Public License, either version 2
 * of the License, or any later version.
 *
 * For the full copyright and license information, please read the
 * LICENSE.txt file that was distributed with this source code.
 *
 * The TYPO3 project - inspiring people to share!
 */
 
namespace ‪TYPO3\CMS\Backend\Tests\Functional\Controller;
 
use Prophecy\PhpUnit\ProphecyTrait;
use ‪TYPO3\CMS\Backend\Controller\MfaController;
use ‪TYPO3\CMS\Backend\Routing\UriBuilder;
use ‪TYPO3\CMS\Backend\Template\ModuleTemplateFactory;
use ‪TYPO3\CMS\Backend\View\AuthenticationStyleInformation;
use ‪TYPO3\CMS\Core\Authentication\Mfa\MfaProviderRegistry;
use ‪TYPO3\CMS\Core\Authentication\Mfa\Provider\Totp;
use ‪TYPO3\CMS\Core\Context\Context;
use ‪TYPO3\CMS\Core\Core\Bootstrap;
use ‪TYPO3\CMS\Core\Core\SystemEnvironmentBuilder;
use ‪TYPO3\CMS\Core\FormProtection\FormProtectionFactory;
use ‪TYPO3\CMS\Core\Http\ServerRequest;
use ‪TYPO3\CMS\Core\Log\Logger;
use TYPO3\CMS\Core\Page\PageRenderer;
use ‪TYPO3\CMS\Core\Utility\GeneralUtility;
use TYPO3\TestingFramework\Core\Functional\FunctionalTestCase;
 
class ‪MfaControllerTest extends FunctionalTestCase
{
    use ProphecyTrait;
 
    protected ‪MfaController ‪$subject;
    protected ‪ServerRequest ‪$request;
 
    protected ‪$configurationToUseInTestInstance = [
        'SYS' => [
            'systemMaintainers' => [],
        ],
    ];
 
    protected function ‪setUp(): void
    {
        parent::setUp();
        $this->setUpBackendUserFromFixture(1);
        ‪Bootstrap::initializeLanguageObject();
 
        $container = $this->getContainer();
        $this->subject = new ‪MfaController(
            $container->get(UriBuilder::class),
            $container->get(MfaProviderRegistry::class),
            $container->get(ModuleTemplateFactory::class),
            $container->get(AuthenticationStyleInformation::class),
            $container->get(PageRenderer::class),
        );
        $this->subject->setLogger($this->prophesize(Logger::class)->reveal());
 
        $this->request = (new ‪ServerRequest())
            ->withAttribute('applicationType', ‪SystemEnvironmentBuilder::REQUESTTYPE_BE);
    }
 
    protected function ‪tearDown(): void
    {
        ‪FormProtectionFactory::purgeInstances();
        parent::tearDown();
    }
 
    public function ‪handleRequestThrowsExceptionOnInvalidActionTest(): void
    {
        $this->expectException(\InvalidArgumentException::class);
        $this->expectExceptionCode(1611879244);
 
        ‪$request = $this->request->‪withQueryParams(['action' => 'unknown']);
        ‪$GLOBALS['TYPO3_REQUEST'] = ‪$request;
        $this->subject->handleRequest(‪$request);
    }
 
    public function ‪handleRequestThrowsExceptionOnMissingProviderTest(): void
    {
        $this->expectException(\InvalidArgumentException::class);
        $this->expectExceptionCode(1611879242);
 
        ‪$request = $this->request->‪withQueryParams(['action' => 'auth']);
        ‪$GLOBALS['TYPO3_REQUEST'] = ‪$request;
        $this->subject->handleRequest(‪$request);
    }
 
    public function ‪handleRequestThrowsExceptionOnInactiveProviderTest(): void
    {
        $this->expectException(\InvalidArgumentException::class);
        $this->expectExceptionCode(1611879242);
 
        $queryParams = [
            'action' => 'auth',
            'identifier' => 'totp',
        ];
 
        ‪$request = $this->request->‪withQueryParams($queryParams);
        ‪$GLOBALS['TYPO3_REQUEST'] = ‪$request;
        $this->subject->handleRequest(‪$request);
    }
 
    public function ‪handleRequestReturnsAuthViewTest(): void
    {
        ‪$GLOBALS['BE_USER']->user['mfa'] = json_encode([
            'totp' => ['active' => true, 'secret' => 'KRMVATZTJFZUC53FONXW2ZJB'],
        ]);
 
        $queryParams = [
            'action' => 'auth',
            'identifier' => 'totp',
        ];
 
        ‪$request = $this->request->‪withQueryParams($queryParams);
        ‪$GLOBALS['TYPO3_REQUEST'] = ‪$request;
        $response = $this->subject->handleRequest(‪$request);
 
        self::assertEquals(200, $response->getStatusCode());
 
        $responseContent = $response->getBody()->getContents();
 
        // Auth view for provider is renderer
        self::assertStringContainsString('Time-based one-time password', $responseContent);
        self::assertMatchesRegularExpression('/<form.*name="verify".*id="mfaController">/s', $responseContent);
 
        // Ensure provider specific content is added as well
        self::assertMatchesRegularExpression('/<input.*id="totp"/s', $responseContent);
    }
 
    public function ‪handleRequestReturnsLockedAuthViewTest(): void
    {
        ‪$GLOBALS['BE_USER']->user['mfa'] = json_encode([
            'totp' => ['active' => true, 'secret' => 'KRMVATZTJFZUC53FONXW2ZJB', 'attempts' => 3],
        ]);
 
        $queryParams = [
            'action' => 'auth',
            'identifier' => 'totp',
        ];
 
        ‪$request = $this->request->‪withQueryParams($queryParams);
        ‪$GLOBALS['TYPO3_REQUEST'] = ‪$request;
        $response = $this->subject->handleRequest(‪$request);
 
        self::assertEquals(200, $response->getStatusCode());
        self::assertStringContainsString('This provider is temporarily locked!', $response->getBody()->getContents());
    }
 
    public function ‪handleRequestReturnsAlternativeProvidersInAuthViewTest(): void
    {
        ‪$GLOBALS['BE_USER']->user['mfa'] = json_encode([
            'totp' => ['active' => true, 'secret' => 'KRMVATZTJFZUC53FONXW2ZJB'],
            'recovery-codes' => ['active' => true, 'codes' => ['some-code']],
        ]);
 
        $queryParams = [
            'action' => 'auth',
            'identifier' => 'totp',
        ];
 
        ‪$request = $this->request->‪withQueryParams($queryParams);
        ‪$GLOBALS['TYPO3_REQUEST'] = ‪$request;
        $response = $this->subject->handleRequest(‪$request);
 
        self::assertEquals(200, $response->getStatusCode());
 
        $responseContent = $response->getBody()->getContents();
        self::assertStringContainsString('Alternative providers', $responseContent);
        self::assertMatchesRegularExpression('/<a.*title="Use Recovery codes"/s', $responseContent);
    }
 
    public function ‪handleRequestRedirectsToLoginOnInvalidRequestTest(): void
    {
        ‪$GLOBALS['BE_USER']->user['mfa'] = json_encode([
            'totp' => ['active' => true, 'secret' => 'KRMVATZTJFZUC53FONXW2ZJB'],
        ]);
 
        $queryParams = [
            'action' => 'verify',
            'identifier' => 'totp',
        ];
 
        // The "totp" parameter is missing, therefore the TotpProvider will return false on ->canProcess()
        ‪$request = $this->request->‪withQueryParams($queryParams);
        ‪$GLOBALS['TYPO3_REQUEST'] = ‪$request;
        $response = $this->subject->handleRequest(‪$request);
 
        self::assertEquals(302, $response->getStatusCode());
        self::assertEquals('/typo3/login', parse_url($response->getHeaderLine('location'))['path']);
    }
 
    public function ‪handleRequestRedirectsToLoginOnLockedProviderRequestTest(): void
    {
        ‪$GLOBALS['BE_USER']->user['mfa'] = json_encode([
            'totp' => ['active' => true, 'secret' => 'KRMVATZTJFZUC53FONXW2ZJB', 'attempts' => 3],
        ]);
 
        $queryParams = [
            'action' => 'verify',
            'identifier' => 'totp',
            'totp' => '123456',
        ];
 
        ‪$request = $this->request->‪withQueryParams($queryParams);
        ‪$GLOBALS['TYPO3_REQUEST'] = ‪$request;
        $response = $this->subject->handleRequest(‪$request);
 
        self::assertEquals(302, $response->getStatusCode());
        self::assertEquals('/typo3/login', parse_url($response->getHeaderLine('location'))['path']);
    }
 
    public function ‪handleRequestRedirectsToAuthViewOnUnsuccessfulAuthenticationTest(): void
    {
        ‪$GLOBALS['BE_USER']->user['mfa'] = json_encode([
            'totp' => ['active' => true, 'secret' => 'KRMVATZTJFZUC53FONXW2ZJB'],
        ]);
 
        $queryParams = [
            'action' => 'verify',
            'identifier' => 'totp',
            'totp' => '123456',
        ];
 
        ‪$request = $this->request->‪withQueryParams($queryParams);
        ‪$GLOBALS['TYPO3_REQUEST'] = ‪$request;
        $response = $this->subject->handleRequest(‪$request);
 
        self::assertEquals(302, $response->getStatusCode());
        self::assertEquals('/typo3/auth/mfa', parse_url($response->getHeaderLine('location'))['path']);
    }
 
    public function ‪handleRequestSetsSessionKeyOnSuccessfulAuthenticationTest(): void
    {
        ‪$GLOBALS['BE_USER']->user['mfa'] = json_encode([
            'totp' => ['active' => true, 'secret' => 'KRMVATZTJFZUC53FONXW2ZJB'],
        ]);
 
        $timestamp = GeneralUtility::makeInstance(Context::class)->getPropertyFromAspect('date', 'timestamp');
        $totp = GeneralUtility::makeInstance(
            Totp::class,
            'KRMVATZTJFZUC53FONXW2ZJB'
        )->generateTotp((int)floor($timestamp / 30));
 
        $queryParams = [
            'action' => 'verify',
            'identifier' => 'totp',
            'totp' => $totp,
        ];
 
        // Ensure mfa session key is not set
        self::assertFalse((bool)‪$GLOBALS['BE_USER']->getSessionData('mfa'));
 
        ‪$request = $this->request->‪withQueryParams($queryParams);
        ‪$GLOBALS['TYPO3_REQUEST'] = ‪$request;
        $response = $this->subject->handleRequest(‪$request);
 
        // Session key is set - User is authenticated
        self::assertTrue(‪$GLOBALS['BE_USER']->getSessionData('mfa'));
 
        // Redirect back to login
        self::assertEquals(302, $response->getStatusCode());
        self::assertEquals('/typo3/login', parse_url($response->getHeaderLine('location'))['path']);
    }
 
    public function ‪handleRequestRedirectsToLoginOnCancelTest(): void
    {
        ‪$request = $this->request->‪withQueryParams(['action' => 'cancel']);
        ‪$GLOBALS['TYPO3_REQUEST'] = ‪$request;
        $response = $this->subject->handleRequest(‪$request);
 
        self::assertEquals(302, $response->getStatusCode());
        self::assertEquals('/typo3/login', parse_url($response->getHeaderLine('location'))['path']);
    }
}