DeleteLocalPasswords.php

Go to the documentation of this file.

<?php
use MediaWiki\MediaWikiServices;
use Wikimedia\Rdbms\IDatabase;
use Wikimedia\Rdbms\IMaintainableDatabase;
 
require_once __DIR__ . '/../Maintenance.php';
 
class DeleteLocalPasswords extends Maintenance {
    protected $user;
 
    protected $total;
 
    public function __construct() {
        parent::__construct();
        $this->addDescription( "Deletes local password for users." );
        $this->setBatchSize( 1000 );
 
        $this->addOption( 'user', 'If specified, only checks the given user', false, true );
        $this->addOption( 'delete', 'Really delete. To prevent accidents, you must provide this flag.' );
        $this->addOption( 'prefix', "Instead of deleting, make passwords invalid by prefixing with "
            . "':null:'. Make sure PasswordConfig has a 'null' entry. This is meant for testing before "
            . "hard delete." );
        $this->addOption( 'unprefix', 'Instead of deleting, undo the effect of --prefix.' );
    }
 
    protected function initialize() {
        if (
            (int)$this->hasOption( 'delete' ) + (int)$this->hasOption( 'prefix' )
            + (int)$this->hasOption( 'unprefix' ) !== 1
        ) {
            $this->fatalError( "Exactly one of the 'delete', 'prefix', 'unprefix' options must be used\n" );
        }
        if ( $this->hasOption( 'prefix' ) || $this->hasOption( 'unprefix' ) ) {
            $passwordHashTypes = MediaWikiServices::getInstance()->getPasswordFactory()->getTypes();
            if (
                !isset( $passwordHashTypes['null'] )
                || $passwordHashTypes['null']['class'] !== InvalidPassword::class
            ) {
                $this->fatalError(
<<<'ERROR'
'null' password entry missing. To use password prefixing, add
    $wgPasswordConfig['null'] = [ 'class' => InvalidPassword::class ];
to your configuration (and remove once the passwords were deleted).
ERROR
                );
            }
        }
 
        $user = $this->getOption( 'user', false );
        if ( $user !== false ) {
            $userNameUtils = MediaWikiServices::getInstance()->getUserNameUtils();
            $this->user = $userNameUtils->getCanonical( $user );
            if ( $this->user === false ) {
                $this->fatalError( "Invalid user name\n" );
            }
        }
    }
 
    public function execute() {
        $this->initialize();
 
        foreach ( $this->getUserBatches() as $userBatch ) {
            $this->processUsers( $userBatch, $this->getUserDB() );
        }
 
        $this->output( "done. (wrote $this->total rows)\n" );
    }
 
    protected function getUserDB() {
        return $this->getDB( DB_PRIMARY );
    }
 
    protected function processUsers( array $userBatch, IDatabase $dbw ) {
        if ( !$userBatch ) {
            return;
        }
        if ( $this->getOption( 'delete' ) ) {
            $dbw->update( 'user',
                [ 'user_password' => PasswordFactory::newInvalidPassword()->toString() ],
                [ 'user_name' => $userBatch ],
                __METHOD__
            );
        } elseif ( $this->getOption( 'prefix' ) ) {
            $dbw->update( 'user',
                [ 'user_password = ' . $dbw->buildConcat( [ $dbw->addQuotes( ':null:' ),
                        'user_password' ] ) ],
                [
                    'NOT (user_password ' . $dbw->buildLike( ':null:', $dbw->anyString() ) . ')',
                    "user_password != " . $dbw->addQuotes( PasswordFactory::newInvalidPassword()->toString() ),
                    'user_password IS NOT NULL',
                    'user_name' => $userBatch,
                ],
                __METHOD__
            );
        } elseif ( $this->getOption( 'unprefix' ) ) {
            $dbw->update( 'user',
                [ 'user_password = ' . $dbw->buildSubString( 'user_password', strlen( ':null:' ) + 1 ) ],
                [
                    'user_password ' . $dbw->buildLike( ':null:', $dbw->anyString() ),
                    'user_name' => $userBatch,
                ],
                __METHOD__
            );
        }
        $this->total += $dbw->affectedRows();
        MediaWikiServices::getInstance()->getDBLoadBalancerFactory()->waitForReplication();
    }
 
    protected function getUserBatches() {
        if ( $this->user !== null ) {
            $this->output( "\t ... querying '$this->user'\n" );
            yield [ [ $this->user ] ];
            return;
        }
 
        $lastUsername = '';
        $dbw = $this->getDB( DB_PRIMARY );
        do {
            $this->output( "\t ... querying from '$lastUsername'\n" );
            $users = $dbw->selectFieldValues(
                'user',
                'user_name',
                [
                    'user_name > ' . $dbw->addQuotes( $lastUsername ),
                ],
                __METHOD__,
                [
                    'LIMIT' => $this->getBatchSize(),
                    'ORDER BY' => 'user_name ASC',
                ]
            );
            if ( $users ) {
                yield $users;
                $lastUsername = end( $users );
            }
        } while ( count( $users ) === $this->getBatchSize() );
    }
}