267 lines
10 KiB
PHP
267 lines
10 KiB
PHP
|
<?php
|
||
|
declare(strict_types=1);
|
||
|
/**
|
||
|
* @copyright Copyright (c) 2018 Roeland Jago Douma <roeland@famdouma.nl>
|
||
|
*
|
||
|
* @author Roeland Jago Douma <roeland@famdouma.nl>
|
||
|
*
|
||
|
* @license GNU AGPL version 3 or any later version
|
||
|
*
|
||
|
* This program is free software: you can redistribute it and/or modify
|
||
|
* it under the terms of the GNU Affero General Public License as
|
||
|
* published by the Free Software Foundation, either version 3 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 Affero General Public License for more details.
|
||
|
*
|
||
|
* You should have received a copy of the GNU Affero General Public License
|
||
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||
|
*
|
||
|
*/
|
||
|
|
||
|
namespace Test\Authentication\Token;
|
||
|
|
||
|
use OC;
|
||
|
use OC\Authentication\Token\PublicKeyToken;
|
||
|
use OC\Authentication\Token\PublicKeyTokenMapper;
|
||
|
use OC\Authentication\Token\IToken;
|
||
|
use OCP\DB\QueryBuilder\IQueryBuilder;
|
||
|
use OCP\IDBConnection;
|
||
|
use OCP\IUser;
|
||
|
use Test\TestCase;
|
||
|
|
||
|
/**
|
||
|
* @group DB
|
||
|
*/
|
||
|
class PublicKeyTokenMapperTest extends TestCase {
|
||
|
|
||
|
/** @var PublicKeyTokenMapper */
|
||
|
private $mapper;
|
||
|
|
||
|
/** @var IDBConnection */
|
||
|
private $dbConnection;
|
||
|
|
||
|
/** @var int */
|
||
|
private $time;
|
||
|
|
||
|
protected function setUp() {
|
||
|
parent::setUp();
|
||
|
|
||
|
$this->dbConnection = OC::$server->getDatabaseConnection();
|
||
|
$this->time = time();
|
||
|
$this->resetDatabase();
|
||
|
|
||
|
$this->mapper = new PublicKeyTokenMapper($this->dbConnection);
|
||
|
}
|
||
|
|
||
|
private function resetDatabase() {
|
||
|
$qb = $this->dbConnection->getQueryBuilder();
|
||
|
$qb->delete('authtoken')->execute();
|
||
|
$qb->insert('authtoken')->values([
|
||
|
'uid' => $qb->createNamedParameter('user1'),
|
||
|
'login_name' => $qb->createNamedParameter('User1'),
|
||
|
'password' => $qb->createNamedParameter('a75c7116460c082912d8f6860a850904|3nz5qbG1nNSLLi6V|c55365a0e54cfdfac4a175bcf11a7612aea74492277bba6e5d96a24497fa9272488787cb2f3ad34d8b9b8060934fce02f008d371df3ff3848f4aa61944851ff0'),
|
||
|
'name' => $qb->createNamedParameter('Firefox on Linux'),
|
||
|
'token' => $qb->createNamedParameter('9c5a2e661482b65597408a6bb6c4a3d1af36337381872ac56e445a06cdb7fea2b1039db707545c11027a4966919918b19d875a8b774840b18c6cbb7ae56fe206'),
|
||
|
'type' => $qb->createNamedParameter(IToken::TEMPORARY_TOKEN),
|
||
|
'last_activity' => $qb->createNamedParameter($this->time - 120, IQueryBuilder::PARAM_INT), // Two minutes ago
|
||
|
'last_check' => $this->time - 60 * 10, // 10mins ago
|
||
|
'public_key' => $qb->createNamedParameter('public key'),
|
||
|
'private_key' => $qb->createNamedParameter('private key'),
|
||
|
'version' => $qb->createNamedParameter(2),
|
||
|
])->execute();
|
||
|
$qb->insert('authtoken')->values([
|
||
|
'uid' => $qb->createNamedParameter('user2'),
|
||
|
'login_name' => $qb->createNamedParameter('User2'),
|
||
|
'password' => $qb->createNamedParameter('971a337057853344700bbeccf836519f|UwOQwyb34sJHtqPV|036d4890f8c21d17bbc7b88072d8ef049a5c832a38e97f3e3d5f9186e896c2593aee16883f617322fa242728d0236ff32d163caeb4bd45e14ca002c57a88665f'),
|
||
|
'name' => $qb->createNamedParameter('Firefox on Android'),
|
||
|
'token' => $qb->createNamedParameter('1504445f1524fc801035448a95681a9378ba2e83930c814546c56e5d6ebde221198792fd900c88ed5ead0555780dad1ebce3370d7e154941cd5de87eb419899b'),
|
||
|
'type' => $qb->createNamedParameter(IToken::TEMPORARY_TOKEN),
|
||
|
'last_activity' => $qb->createNamedParameter($this->time - 60 * 60 * 24 * 3, IQueryBuilder::PARAM_INT), // Three days ago
|
||
|
'last_check' => $this->time - 10, // 10secs ago
|
||
|
'public_key' => $qb->createNamedParameter('public key'),
|
||
|
'private_key' => $qb->createNamedParameter('private key'),
|
||
|
'version' => $qb->createNamedParameter(2),
|
||
|
])->execute();
|
||
|
$qb->insert('authtoken')->values([
|
||
|
'uid' => $qb->createNamedParameter('user1'),
|
||
|
'login_name' => $qb->createNamedParameter('User1'),
|
||
|
'password' => $qb->createNamedParameter('063de945d6f6b26862d9b6f40652f2d5|DZ/z520tfdXPtd0T|395f6b89be8d9d605e409e20b9d9abe477fde1be38a3223f9e508f979bf906e50d9eaa4dca983ca4fb22a241eb696c3f98654e7775f78c4caf13108f98642b53'),
|
||
|
'name' => $qb->createNamedParameter('Iceweasel on Linux'),
|
||
|
'token' => $qb->createNamedParameter('47af8697ba590fb82579b5f1b3b6e8066773a62100abbe0db09a289a62f5d980dc300fa3d98b01d7228468d1ab05c1aa14c8d14bd5b6eee9cdf1ac14864680c3'),
|
||
|
'type' => $qb->createNamedParameter(IToken::TEMPORARY_TOKEN),
|
||
|
'last_activity' => $qb->createNamedParameter($this->time - 120, IQueryBuilder::PARAM_INT), // Two minutes ago
|
||
|
'last_check' => $this->time - 60 * 10, // 10mins ago
|
||
|
'public_key' => $qb->createNamedParameter('public key'),
|
||
|
'private_key' => $qb->createNamedParameter('private key'),
|
||
|
'version' => $qb->createNamedParameter(2),
|
||
|
])->execute();
|
||
|
}
|
||
|
|
||
|
private function getNumberOfTokens() {
|
||
|
$qb = $this->dbConnection->getQueryBuilder();
|
||
|
$result = $qb->select($qb->createFunction('count(*) as `count`'))
|
||
|
->from('authtoken')
|
||
|
->execute()
|
||
|
->fetch();
|
||
|
return (int) $result['count'];
|
||
|
}
|
||
|
|
||
|
public function testInvalidate() {
|
||
|
$token = '9c5a2e661482b65597408a6bb6c4a3d1af36337381872ac56e445a06cdb7fea2b1039db707545c11027a4966919918b19d875a8b774840b18c6cbb7ae56fe206';
|
||
|
|
||
|
$this->mapper->invalidate($token);
|
||
|
|
||
|
$this->assertSame(2, $this->getNumberOfTokens());
|
||
|
}
|
||
|
|
||
|
public function testInvalidateInvalid() {
|
||
|
$token = 'youwontfindthisoneinthedatabase';
|
||
|
|
||
|
$this->mapper->invalidate($token);
|
||
|
|
||
|
$this->assertSame(3, $this->getNumberOfTokens());
|
||
|
}
|
||
|
|
||
|
public function testInvalidateOld() {
|
||
|
$olderThan = $this->time - 60 * 60; // One hour
|
||
|
|
||
|
$this->mapper->invalidateOld($olderThan);
|
||
|
|
||
|
$this->assertSame(2, $this->getNumberOfTokens());
|
||
|
}
|
||
|
|
||
|
public function testGetToken() {
|
||
|
$token = new PublicKeyToken();
|
||
|
$token->setUid('user2');
|
||
|
$token->setLoginName('User2');
|
||
|
$token->setPassword('971a337057853344700bbeccf836519f|UwOQwyb34sJHtqPV|036d4890f8c21d17bbc7b88072d8ef049a5c832a38e97f3e3d5f9186e896c2593aee16883f617322fa242728d0236ff32d163caeb4bd45e14ca002c57a88665f');
|
||
|
$token->setName('Firefox on Android');
|
||
|
$token->setToken('1504445f1524fc801035448a95681a9378ba2e83930c814546c56e5d6ebde221198792fd900c88ed5ead0555780dad1ebce3370d7e154941cd5de87eb419899b');
|
||
|
$token->setType(IToken::TEMPORARY_TOKEN);
|
||
|
$token->setRemember(IToken::DO_NOT_REMEMBER);
|
||
|
$token->setLastActivity($this->time - 60 * 60 * 24 * 3);
|
||
|
$token->setLastCheck($this->time - 10);
|
||
|
$token->setPublicKey('public key');
|
||
|
$token->setPrivateKey('private key');
|
||
|
|
||
|
$dbToken = $this->mapper->getToken($token->getToken());
|
||
|
|
||
|
$token->setId($dbToken->getId()); // We don't know the ID
|
||
|
$token->resetUpdatedFields();
|
||
|
|
||
|
$this->assertEquals($token, $dbToken);
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* @expectedException \OCP\AppFramework\Db\DoesNotExistException
|
||
|
*/
|
||
|
public function testGetInvalidToken() {
|
||
|
$token = 'thisisaninvalidtokenthatisnotinthedatabase';
|
||
|
|
||
|
$this->mapper->getToken($token);
|
||
|
}
|
||
|
|
||
|
public function testGetTokenById() {
|
||
|
$token = new PublicKeyToken();
|
||
|
$token->setUid('user2');
|
||
|
$token->setLoginName('User2');
|
||
|
$token->setPassword('971a337057853344700bbeccf836519f|UwOQwyb34sJHtqPV|036d4890f8c21d17bbc7b88072d8ef049a5c832a38e97f3e3d5f9186e896c2593aee16883f617322fa242728d0236ff32d163caeb4bd45e14ca002c57a88665f');
|
||
|
$token->setName('Firefox on Android');
|
||
|
$token->setToken('1504445f1524fc801035448a95681a9378ba2e83930c814546c56e5d6ebde221198792fd900c88ed5ead0555780dad1ebce3370d7e154941cd5de87eb419899b');
|
||
|
$token->setType(IToken::TEMPORARY_TOKEN);
|
||
|
$token->setRemember(IToken::DO_NOT_REMEMBER);
|
||
|
$token->setLastActivity($this->time - 60 * 60 * 24 * 3);
|
||
|
$token->setLastCheck($this->time - 10);
|
||
|
$token->setPublicKey('public key');
|
||
|
$token->setPrivateKey('private key');
|
||
|
|
||
|
$dbToken = $this->mapper->getToken($token->getToken());
|
||
|
$token->setId($dbToken->getId()); // We don't know the ID
|
||
|
$token->resetUpdatedFields();
|
||
|
|
||
|
$dbToken = $this->mapper->getTokenById($token->getId());
|
||
|
$this->assertEquals($token, $dbToken);
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* @expectedException \OCP\AppFramework\Db\DoesNotExistException
|
||
|
*/
|
||
|
public function testGetTokenByIdNotFound() {
|
||
|
$this->mapper->getTokenById(-1);
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* @expectedException \OCP\AppFramework\Db\DoesNotExistException
|
||
|
*/
|
||
|
public function testGetInvalidTokenById() {
|
||
|
$id = '42';
|
||
|
|
||
|
$this->mapper->getToken($id);
|
||
|
}
|
||
|
|
||
|
public function testGetTokenByUser() {
|
||
|
/** @var IUser|\PHPUnit_Framework_MockObject_MockObject $user */
|
||
|
$user = $this->createMock(IUser::class);
|
||
|
$user->expects($this->once())
|
||
|
->method('getUID')
|
||
|
->will($this->returnValue('user1'));
|
||
|
|
||
|
$this->assertCount(2, $this->mapper->getTokenByUser($user));
|
||
|
}
|
||
|
|
||
|
public function testGetTokenByUserNotFound() {
|
||
|
/** @var IUser|\PHPUnit_Framework_MockObject_MockObject $user */
|
||
|
$user = $this->createMock(IUser::class);
|
||
|
$user->expects($this->once())
|
||
|
->method('getUID')
|
||
|
->will($this->returnValue('user1000'));
|
||
|
|
||
|
$this->assertCount(0, $this->mapper->getTokenByUser($user));
|
||
|
}
|
||
|
|
||
|
public function testDeleteById() {
|
||
|
/** @var IUser|\PHPUnit_Framework_MockObject_MockObject $user */
|
||
|
$user = $this->createMock(IUser::class);
|
||
|
$qb = $this->dbConnection->getQueryBuilder();
|
||
|
$qb->select('id')
|
||
|
->from('authtoken')
|
||
|
->where($qb->expr()->eq('token', $qb->createNamedParameter('9c5a2e661482b65597408a6bb6c4a3d1af36337381872ac56e445a06cdb7fea2b1039db707545c11027a4966919918b19d875a8b774840b18c6cbb7ae56fe206')));
|
||
|
$result = $qb->execute();
|
||
|
$id = $result->fetch()['id'];
|
||
|
$user->expects($this->once())
|
||
|
->method('getUID')
|
||
|
->will($this->returnValue('user1'));
|
||
|
|
||
|
$this->mapper->deleteById($user, (int)$id);
|
||
|
$this->assertEquals(2, $this->getNumberOfTokens());
|
||
|
}
|
||
|
|
||
|
public function testDeleteByIdWrongUser() {
|
||
|
/** @var IUser|\PHPUnit_Framework_MockObject_MockObject $user */
|
||
|
$user = $this->createMock(IUser::class);
|
||
|
$id = 33;
|
||
|
$user->expects($this->once())
|
||
|
->method('getUID')
|
||
|
->will($this->returnValue('user10000'));
|
||
|
|
||
|
$this->mapper->deleteById($user, $id);
|
||
|
$this->assertEquals(3, $this->getNumberOfTokens());
|
||
|
}
|
||
|
|
||
|
public function testDeleteByName() {
|
||
|
$qb = $this->dbConnection->getQueryBuilder();
|
||
|
$qb->select('name')
|
||
|
->from('authtoken')
|
||
|
->where($qb->expr()->eq('token', $qb->createNamedParameter('9c5a2e661482b65597408a6bb6c4a3d1af36337381872ac56e445a06cdb7fea2b1039db707545c11027a4966919918b19d875a8b774840b18c6cbb7ae56fe206')));
|
||
|
$result = $qb->execute();
|
||
|
$name = $result->fetch()['name'];
|
||
|
$this->mapper->deleteByName($name);
|
||
|
$this->assertEquals(2, $this->getNumberOfTokens());
|
||
|
}
|
||
|
|
||
|
}
|