123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412 |
- <?php
- /**
- * @author Christoph Wurst <christoph@owncloud.com>
- *
- * @copyright Copyright (c) 2016, Lukas Reschke <lukas@statuscode.ch>
- * @copyright Copyright (c) 2016, ownCloud, Inc.
- * @license AGPL-3.0
- *
- * This code is free software: you can redistribute it and/or modify
- * it under the terms of the GNU Affero General Public License, version 3,
- * as published by the Free Software Foundation.
- *
- * 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, version 3,
- * along with this program. If not, see <http://www.gnu.org/licenses/>
- *
- */
- namespace Test\Authentication\Token;
- use OC\Authentication\Exceptions\InvalidTokenException;
- use OC\Authentication\Token\DefaultToken;
- use OC\Authentication\Token\DefaultTokenProvider;
- use OC\Authentication\Token\IToken;
- use OCP\AppFramework\Db\DoesNotExistException;
- use OCP\AppFramework\Db\Mapper;
- use OCP\AppFramework\Utility\ITimeFactory;
- use OCP\IConfig;
- use OCP\ILogger;
- use OCP\IUser;
- use OCP\Security\ICrypto;
- use Test\TestCase;
- class DefaultTokenProviderTest extends TestCase {
- /** @var DefaultTokenProvider|\PHPUnit_Framework_MockObject_MockObject */
- private $tokenProvider;
- /** @var Mapper|\PHPUnit_Framework_MockObject_MockObject */
- private $mapper;
- /** @var ICrypto|\PHPUnit_Framework_MockObject_MockObject */
- private $crypto;
- /** @var IConfig|\PHPUnit_Framework_MockObject_MockObject */
- private $config;
- /** @var ILogger|\PHPUnit_Framework_MockObject_MockObject */
- private $logger;
- /** @var ITimeFactory|\PHPUnit_Framework_MockObject_MockObject */
- private $timeFactory;
- /** @var int */
- private $time;
- protected function setUp() {
- parent::setUp();
- $this->mapper = $this->getMockBuilder('\OC\Authentication\Token\DefaultTokenMapper')
- ->disableOriginalConstructor()
- ->getMock();
- $this->crypto = $this->createMock(ICrypto::class);
- $this->config = $this->createMock(IConfig::class);
- $this->logger = $this->createMock(ILogger::class);
- $this->timeFactory = $this->createMock(ITimeFactory::class);
- $this->time = 1313131;
- $this->timeFactory->expects($this->any())
- ->method('getTime')
- ->will($this->returnValue($this->time));
- $this->tokenProvider = new DefaultTokenProvider($this->mapper, $this->crypto, $this->config, $this->logger,
- $this->timeFactory);
- }
- public function testGenerateToken() {
- $token = 'token';
- $uid = 'user';
- $user = 'User';
- $password = 'passme';
- $name = 'User-Agent: Mozilla/5.0 (Windows; U; Windows NT 5.1; en-US; rv:1.9.2.12) Gecko/20101026 Firefox/3.6.12'
- . 'User-Agent: Mozilla/5.0 (Windows; U; Windows NT 5.1; en-US; rv:1.9.2.12) Gecko/20101026 Firefox/3.6.12'
- . 'User-Agent: Mozilla/5.0 (Windows; U; Windows NT 5.1; en-US; rv:1.9.2.12) Gecko/20101026 Firefox/3.6.12'
- . 'User-Agent: Mozilla/5.0 (Windows; U; Windows NT 5.1; en-US; rv:1.9.2.12) Gecko/20101026 Firefox/3.6.12';
- $type = IToken::PERMANENT_TOKEN;
- $toInsert = new DefaultToken();
- $toInsert->setUid($uid);
- $toInsert->setLoginName($user);
- $toInsert->setPassword('encryptedpassword');
- $toInsert->setName($name);
- $toInsert->setToken(hash('sha512', $token . '1f4h9s'));
- $toInsert->setType($type);
- $toInsert->setRemember(IToken::DO_NOT_REMEMBER);
- $toInsert->setLastActivity($this->time);
- $this->config->expects($this->any())
- ->method('getSystemValue')
- ->with('secret')
- ->will($this->returnValue('1f4h9s'));
- $this->crypto->expects($this->once())
- ->method('encrypt')
- ->with($password, $token . '1f4h9s')
- ->will($this->returnValue('encryptedpassword'));
- $this->mapper->expects($this->once())
- ->method('insert')
- ->with($this->equalTo($toInsert));
- $actual = $this->tokenProvider->generateToken($token, $uid, $user, $password, $name, $type, IToken::DO_NOT_REMEMBER);
- $this->assertEquals($toInsert, $actual);
- }
- public function testUpdateToken() {
- $tk = new DefaultToken();
- $tk->setLastActivity($this->time - 200);
- $this->mapper->expects($this->once())
- ->method('update')
- ->with($tk);
- $this->tokenProvider->updateTokenActivity($tk);
- $this->assertEquals($this->time, $tk->getLastActivity());
- }
- public function testUpdateTokenDebounce() {
- $tk = new DefaultToken();
- $tk->setLastActivity($this->time - 30);
- $this->mapper->expects($this->never())
- ->method('update')
- ->with($tk);
- $this->tokenProvider->updateTokenActivity($tk);
- }
-
- public function testGetTokenByUser() {
- $user = $this->createMock(IUser::class);
- $this->mapper->expects($this->once())
- ->method('getTokenByUser')
- ->with($user)
- ->will($this->returnValue(['token']));
- $this->assertEquals(['token'], $this->tokenProvider->getTokenByUser($user));
- }
- public function testGetPassword() {
- $token = 'token1234';
- $tk = new DefaultToken();
- $tk->setPassword('someencryptedvalue');
- $this->config->expects($this->once())
- ->method('getSystemValue')
- ->with('secret')
- ->will($this->returnValue('1f4h9s'));
- $this->crypto->expects($this->once())
- ->method('decrypt')
- ->with('someencryptedvalue', $token . '1f4h9s')
- ->will($this->returnValue('passme'));
- $actual = $this->tokenProvider->getPassword($tk, $token);
- $this->assertEquals('passme', $actual);
- }
- /**
- * @expectedException \OC\Authentication\Exceptions\PasswordlessTokenException
- */
- public function testGetPasswordPasswordLessToken() {
- $token = 'token1234';
- $tk = new DefaultToken();
- $tk->setPassword(null);
- $this->tokenProvider->getPassword($tk, $token);
- }
- /**
- * @expectedException \OC\Authentication\Exceptions\InvalidTokenException
- */
- public function testGetPasswordDeletesInvalidToken() {
- $token = 'token1234';
- $tk = new DefaultToken();
- $tk->setPassword('someencryptedvalue');
- /* @var $tokenProvider DefaultTokenProvider */
- $tokenProvider = $this->getMockBuilder('\OC\Authentication\Token\DefaultTokenProvider')
- ->setMethods([
- 'invalidateToken'
- ])
- ->setConstructorArgs([$this->mapper, $this->crypto, $this->config, $this->logger,
- $this->timeFactory])
- ->getMock();
- $this->config->expects($this->once())
- ->method('getSystemValue')
- ->with('secret')
- ->will($this->returnValue('1f4h9s'));
- $this->crypto->expects($this->once())
- ->method('decrypt')
- ->with('someencryptedvalue', $token . '1f4h9s')
- ->will($this->throwException(new \Exception('some crypto error occurred')));
- $tokenProvider->expects($this->once())
- ->method('invalidateToken')
- ->with($token);
- $tokenProvider->getPassword($tk, $token);
- }
- public function testSetPassword() {
- $token = new DefaultToken();
- $tokenId = 'token123';
- $password = '123456';
- $this->config->expects($this->once())
- ->method('getSystemValue')
- ->with('secret')
- ->will($this->returnValue('ocsecret'));
- $this->crypto->expects($this->once())
- ->method('encrypt')
- ->with($password, $tokenId . 'ocsecret')
- ->will($this->returnValue('encryptedpassword'));
- $this->mapper->expects($this->once())
- ->method('update')
- ->with($token);
- $this->tokenProvider->setPassword($token, $tokenId, $password);
- $this->assertEquals('encryptedpassword', $token->getPassword());
- }
- /**
- * @expectedException \OC\Authentication\Exceptions\InvalidTokenException
- */
- public function testSetPasswordInvalidToken() {
- $token = $this->createMock(IToken::class);
- $tokenId = 'token123';
- $password = '123456';
- $this->tokenProvider->setPassword($token, $tokenId, $password);
- }
- public function testInvalidateToken() {
- $this->mapper->expects($this->once())
- ->method('invalidate')
- ->with(hash('sha512', 'token7'));
- $this->tokenProvider->invalidateToken('token7');
- }
- public function testInvaildateTokenById() {
- $id = 123;
- $user = $this->createMock(IUser::class);
- $this->mapper->expects($this->once())
- ->method('deleteById')
- ->with($user, $id);
- $this->tokenProvider->invalidateTokenById($user, $id);
- }
- public function testInvalidateOldTokens() {
- $defaultSessionLifetime = 60 * 60 * 24;
- $defaultRememberMeLifetime = 60 * 60 * 24 * 15;
- $this->config->expects($this->exactly(2))
- ->method('getSystemValue')
- ->will($this->returnValueMap([
- ['session_lifetime', $defaultSessionLifetime, 150],
- ['remember_login_cookie_lifetime', $defaultRememberMeLifetime, 300],
- ]));
- $this->mapper->expects($this->at(0))
- ->method('invalidateOld')
- ->with($this->time - 150);
- $this->mapper->expects($this->at(1))
- ->method('invalidateOld')
- ->with($this->time - 300);
- $this->tokenProvider->invalidateOldTokens();
- }
- public function testRenewSessionTokenWithoutPassword() {
- $token = $this->getMockBuilder(DefaultToken::class)
- ->disableOriginalConstructor()
- ->setMethods(['getUID', 'getLoginName', 'getPassword', 'getName'])
- ->getMock();
- $token
- ->expects($this->at(0))
- ->method('getUID')
- ->willReturn('UserUid');
- $token
- ->expects($this->at(1))
- ->method('getLoginName')
- ->willReturn('UserLoginName');
- $token
- ->expects($this->at(2))
- ->method('getPassword')
- ->willReturn(null);
- $token
- ->expects($this->at(3))
- ->method('getName')
- ->willReturn('MyTokenName');
- $token
- ->expects($this->at(3))
- ->method('getRemember')
- ->willReturn(IToken::DO_NOT_REMEMBER);
- $this->config
- ->expects($this->exactly(2))
- ->method('getSystemValue')
- ->with('secret')
- ->willReturn('MyInstanceSecret');
- $this->mapper
- ->expects($this->at(0))
- ->method('getToken')
- ->with(hash('sha512', 'oldId' . 'MyInstanceSecret'))
- ->willReturn($token);
- $newToken = new DefaultToken();
- $newToken->setUid('UserUid');
- $newToken->setLoginName('UserLoginName');
- $newToken->setName('MyTokenName');
- $newToken->setToken(hash('sha512', 'newId' . 'MyInstanceSecret'));
- $newToken->setType(IToken::TEMPORARY_TOKEN);
- $newToken->setRemember(IToken::DO_NOT_REMEMBER);
- $newToken->setLastActivity(1313131);
- $this->mapper
- ->expects($this->at(1))
- ->method('insert')
- ->with($newToken);
- $this->tokenProvider->renewSessionToken('oldId', 'newId');
- }
- public function testRenewSessionTokenWithPassword() {
- $token = $this->getMockBuilder(DefaultToken::class)
- ->disableOriginalConstructor()
- ->setMethods(['getUID', 'getLoginName', 'getPassword', 'getName'])
- ->getMock();
- $token
- ->expects($this->at(0))
- ->method('getUID')
- ->willReturn('UserUid');
- $token
- ->expects($this->at(1))
- ->method('getLoginName')
- ->willReturn('UserLoginName');
- $token
- ->expects($this->at(2))
- ->method('getPassword')
- ->willReturn('EncryptedPassword');
- $token
- ->expects($this->at(3))
- ->method('getPassword')
- ->willReturn('EncryptedPassword');
- $token
- ->expects($this->at(4))
- ->method('getName')
- ->willReturn('MyTokenName');
- $token
- ->expects($this->at(3))
- ->method('getRemember')
- ->willReturn(IToken::REMEMBER);
- $this->crypto
- ->expects($this->any(0))
- ->method('decrypt')
- ->with('EncryptedPassword', 'oldIdMyInstanceSecret')
- ->willReturn('ClearTextPassword');
- $this->crypto
- ->expects($this->any(1))
- ->method('encrypt')
- ->with('ClearTextPassword', 'newIdMyInstanceSecret')
- ->willReturn('EncryptedPassword');
- $this->config
- ->expects($this->exactly(4))
- ->method('getSystemValue')
- ->with('secret')
- ->willReturn('MyInstanceSecret');
- $this->mapper
- ->expects($this->at(0))
- ->method('getToken')
- ->with(hash('sha512', 'oldId' . 'MyInstanceSecret'))
- ->willReturn($token);
- $newToken = new DefaultToken();
- $newToken->setUid('UserUid');
- $newToken->setLoginName('UserLoginName');
- $newToken->setName('MyTokenName');
- $newToken->setToken(hash('sha512', 'newId' . 'MyInstanceSecret'));
- $newToken->setType(IToken::TEMPORARY_TOKEN);
- $newToken->setRemember(IToken::REMEMBER);
- $newToken->setLastActivity(1313131);
- $newToken->setPassword('EncryptedPassword');
- $this->mapper
- ->expects($this->at(1))
- ->method('insert')
- ->with($this->equalTo($newToken));
- $this->tokenProvider->renewSessionToken('oldId', 'newId');
- }
- public function testGetTokenById() {
- $token = $this->createMock(DefaultToken::class);
- $this->mapper->expects($this->once())
- ->method('getTokenById')
- ->with($this->equalTo(42))
- ->willReturn($token);
- $this->assertSame($token, $this->tokenProvider->getTokenById(42));
- }
- public function testGetInvalidTokenById() {
- $this->expectException(InvalidTokenException::class);
- $this->mapper->expects($this->once())
- ->method('getTokenById')
- ->with($this->equalTo(42))
- ->willThrowException(new DoesNotExistException('nope'));
- $this->tokenProvider->getTokenById(42);
- }
- }
|