mirror of
https://gerrit.wikimedia.org/r/mediawiki/extensions/OATHAuth
synced 2024-11-27 17:50:15 +00:00
0dad2c7031
The current implementation of OATHUserRepository::persist() causes every key to get a new ID when it's saved. This, combined with ::removeKey() which compares keys by ID, means that using recovery codes to disable TOTP is broken since TOTPKey calls persist() to mark the code as saved just before the key is deleted. In this patch I've chosen to add a new ::updateKey() method instead of fixing ::persist(). This is more in line with the other new APIs in OATHUserRepository (namely ::createKey() and ::removeKey()), and is something I've been planning to do eventually - this bug just made that a bit more urgent. ::persist() should be dropped once WebAuthn has been updated too. Tests are also updated - OATHUserRepositoryTest now updates the key before deleting it and there's a new TOTPDisableFormTest to test the entire disabling process. Bug: T363548 Change-Id: I86ddc8e5bfc9cf74c587ffdff523f559c5a3c08c
120 lines
3.9 KiB
PHP
120 lines
3.9 KiB
PHP
<?php
|
|
/**
|
|
* This program is free software; you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation; either version 2 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License along
|
|
* with this program; if not, write to the Free Software Foundation, Inc.,
|
|
* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
|
|
* http://www.gnu.org/copyleft/gpl.html
|
|
*
|
|
* @file
|
|
*/
|
|
|
|
namespace MediaWiki\Extension\OATHAuth\Tests\Integration;
|
|
|
|
use EmptyBagOStuff;
|
|
use MediaWiki\Extension\OATHAuth\Key\TOTPKey;
|
|
use MediaWiki\Extension\OATHAuth\OATHAuthServices;
|
|
use MediaWiki\Extension\OATHAuth\OATHUserRepository;
|
|
use MediaWiki\User\CentralId\CentralIdLookup;
|
|
use MediaWiki\User\CentralId\CentralIdLookupFactory;
|
|
use MediaWikiIntegrationTestCase;
|
|
use Psr\Log\LoggerInterface;
|
|
use Wikimedia\Rdbms\IConnectionProvider;
|
|
use Wikimedia\TestingAccessWrapper;
|
|
|
|
/**
|
|
* @author Taavi Väänänen <hi@taavi.wtf>
|
|
* @group Database
|
|
* @coversDefaultClass \MediaWiki\Extension\OATHAuth\OATHUserRepository
|
|
*/
|
|
class OATHUserRepositoryTest extends MediaWikiIntegrationTestCase {
|
|
/**
|
|
* @covers ::findByUser
|
|
* @covers ::createKey
|
|
* @covers ::updateKey
|
|
* @covers ::remove
|
|
*/
|
|
public function testLookupCreateRemoveKey(): void {
|
|
$user = $this->getTestUser()->getUser();
|
|
|
|
$dbProvider = $this->createMock( IConnectionProvider::class );
|
|
$dbProvider->method( 'getPrimaryDatabase' )->with( 'virtual-oathauth' )->willReturn( $this->db );
|
|
$dbProvider->method( 'getReplicaDatabase' )->with( 'virtual-oathauth' )->willReturn( $this->db );
|
|
|
|
$moduleRegistry = OATHAuthServices::getInstance( $this->getServiceContainer() )->getModuleRegistry();
|
|
$module = $moduleRegistry->getModuleByKey( 'totp' );
|
|
|
|
$lookup = $this->createMock( CentralIdLookup::class );
|
|
$lookup->method( 'centralIdFromLocalUser' )
|
|
->with( $user )
|
|
->willReturn( 12345 );
|
|
$lookupFactory = $this->createMock( CentralIdLookupFactory::class );
|
|
$lookupFactory->method( 'getLookup' )->willReturn( $lookup );
|
|
|
|
$logger = $this->createMock( LoggerInterface::class );
|
|
|
|
$repository = new OATHUserRepository(
|
|
$dbProvider,
|
|
new EmptyBagOStuff(),
|
|
$moduleRegistry,
|
|
$lookupFactory,
|
|
$logger
|
|
);
|
|
|
|
$oathUser = $repository->findByUser( $user );
|
|
$this->assertEquals( 12345, $oathUser->getCentralId() );
|
|
$this->assertEquals( [], $oathUser->getKeys() );
|
|
$this->assertNull( $oathUser->getModule() );
|
|
|
|
/** @var TOTPKey $key */
|
|
$key = $repository->createKey(
|
|
$oathUser,
|
|
$module,
|
|
TOTPKey::newFromRandom()->jsonSerialize(),
|
|
'127.0.0.1'
|
|
);
|
|
|
|
$this->assertNotEmpty(
|
|
$this->db->newSelectQueryBuilder()
|
|
->select( '1' )
|
|
->from( 'oathauth_devices' )
|
|
->where( [ 'oad_user' => $oathUser->getCentralId() ] )
|
|
);
|
|
|
|
$this->assertArrayEquals( [ $key ], $oathUser->getKeys() );
|
|
$this->assertEquals( $module, $oathUser->getModule() );
|
|
|
|
// Test looking it up again from the database
|
|
$this->assertArrayEquals( [ $key ], $repository->findByUser( $user )->getKeys() );
|
|
|
|
// Use a scratch code, which causes the key to be updated.
|
|
TestingAccessWrapper::newFromObject( $key )->recoveryCodes = [ 'new scratch tokens' ];
|
|
$repository->updateKey( $oathUser, $key );
|
|
|
|
$this->assertEquals(
|
|
[ 'new scratch tokens' ],
|
|
$repository->findByUser( $user )->getKeys()[0]->getScratchTokens()
|
|
);
|
|
|
|
$repository->removeKey(
|
|
$oathUser,
|
|
$key,
|
|
'127.0.0.1',
|
|
true
|
|
);
|
|
|
|
$this->assertEquals( [], $oathUser->getKeys() );
|
|
$this->assertNull( $oathUser->getModule() );
|
|
$this->assertEquals( [], $repository->findByUser( $user )->getKeys() );
|
|
}
|
|
}
|