ThrottlePreAuthenticationProvider.php

Go to the documentation of this file.

<?php
namespace MediaWiki\Auth;

use BagOStuff;
use Config;

class ThrottlePreAuthenticationProvider extends AbstractPreAuthenticationProvider {
    protected $throttleSettings;

    protected $accountCreationThrottle;

    protected $passwordAttemptThrottle;

    protected $cache;

    public function __construct( $params = [] ) {
        $this->throttleSettings = array_intersect_key( $params,
            [ 'accountCreationThrottle' => true, 'passwordAttemptThrottle' => true ] );
        $this->cache = isset( $params['cache'] ) ? $params['cache'] :
            \ObjectCache::getLocalClusterInstance();
    }

    public function setConfig( Config $config ) {
        parent::setConfig( $config );

        $accountCreationThrottle = $this->config->get( 'AccountCreationThrottle' );
        // Handle old $wgAccountCreationThrottle format (number of attempts per 24 hours)
        if ( !is_array( $accountCreationThrottle ) ) {
            $accountCreationThrottle = [ [
                'count' => $accountCreationThrottle,
                'seconds' => 86400,
            ] ];
        }

        // @codeCoverageIgnoreStart
        $this->throttleSettings += [
        // @codeCoverageIgnoreEnd
            'accountCreationThrottle' => $accountCreationThrottle,
            'passwordAttemptThrottle' => $this->config->get( 'PasswordAttemptThrottle' ),
        ];

        if ( !empty( $this->throttleSettings['accountCreationThrottle'] ) ) {
            $this->accountCreationThrottle = new Throttler(
                $this->throttleSettings['accountCreationThrottle'], [
                    'type' => 'acctcreate',
                    'cache' => $this->cache,
                ]
            );
        }
        if ( !empty( $this->throttleSettings['passwordAttemptThrottle'] ) ) {
            $this->passwordAttemptThrottle = new Throttler(
                $this->throttleSettings['passwordAttemptThrottle'], [
                    'type' => 'password',
                    'cache' => $this->cache,
                ]
            );
        }
    }

    public function testForAccountCreation( $user, $creator, array $reqs ) {
        if ( !$this->accountCreationThrottle || !$creator->isPingLimitable() ) {
            return \StatusValue::newGood();
        }

        $ip = $this->manager->getRequest()->getIP();

        if ( !\Hooks::run( 'ExemptFromAccountCreationThrottle', [ $ip ] ) ) {
            $this->logger->debug( __METHOD__ . ": a hook allowed account creation w/o throttle\n" );
            return \StatusValue::newGood();
        }

        $result = $this->accountCreationThrottle->increase( null, $ip, __METHOD__ );
        if ( $result ) {
            $message = wfMessage( 'acct_creation_throttle_hit' )->params( $result['count'] )
                ->durationParams( $result['wait'] );
            return \StatusValue::newFatal( $message );
        }

        return \StatusValue::newGood();
    }

    public function testForAuthentication( array $reqs ) {
        if ( !$this->passwordAttemptThrottle ) {
            return \StatusValue::newGood();
        }

        $ip = $this->manager->getRequest()->getIP();
        try {
            $username = AuthenticationRequest::getUsernameFromRequests( $reqs );
        } catch ( \UnexpectedValueException $e ) {
            $username = '';
        }

        // Get everything this username could normalize to, and throttle each one individually.
        // If nothing uses usernames, just throttle by IP.
        $usernames = $this->manager->normalizeUsername( $username );
        $result = false;
        foreach ( $usernames as $name ) {
            $r = $this->passwordAttemptThrottle->increase( $name, $ip, __METHOD__ );
            if ( $r && ( !$result || $result['wait'] < $r['wait'] ) ) {
                $result = $r;
            }
        }

        if ( $result ) {
            $message = wfMessage( 'login-throttled' )->durationParams( $result['wait'] );
            return \StatusValue::newFatal( $message );
        } else {
            $this->manager->setAuthenticationSessionData( 'LoginThrottle',
                [ 'users' => $usernames, 'ip' => $ip ] );
            return \StatusValue::newGood();
        }
    }

    public function postAuthentication( $user, AuthenticationResponse $response ) {
        if ( $response->status !== AuthenticationResponse::PASS ) {
            return;
        } elseif ( !$this->passwordAttemptThrottle ) {
            return;
        }

        $data = $this->manager->getAuthenticationSessionData( 'LoginThrottle' );
        if ( !$data ) {
            $this->logger->error( 'throttler data not found for {user}', [ 'user' => $user->getName() ] );
            return;
        }

        foreach ( $data['users'] as $name ) {
            $this->passwordAttemptThrottle->clear( $name, $data['ip'] );
        }
    }
}