126 lines
3.6 KiB
PHP
126 lines
3.6 KiB
PHP
<?php
|
|
|
|
/**
|
|
* SPDX-FileCopyrightText: 2017-2024 Nextcloud GmbH and Nextcloud contributors
|
|
* SPDX-FileCopyrightText: 2016 ownCloud, Inc.
|
|
* SPDX-License-Identifier: AGPL-3.0-or-later
|
|
*/
|
|
|
|
namespace Test\Traits;
|
|
|
|
use OC\Encryption\EncryptionWrapper;
|
|
use OC\Files\SetupManager;
|
|
use OC\Memcache\ArrayCache;
|
|
use OCA\Encryption\AppInfo\Application;
|
|
use OCA\Encryption\Crypto\Encryption;
|
|
use OCA\Encryption\KeyManager;
|
|
use OCA\Encryption\Users\Setup;
|
|
use OCP\App\IAppManager;
|
|
use OCP\Encryption\IManager;
|
|
use OCP\IConfig;
|
|
use OCP\IUserManager;
|
|
use OCP\IUserSession;
|
|
use OCP\Server;
|
|
use Psr\Log\LoggerInterface;
|
|
|
|
/**
|
|
* Enables encryption
|
|
*/
|
|
trait EncryptionTrait {
|
|
// from MountProviderTrait
|
|
abstract protected function registerStorageWrapper($name, $wrapper);
|
|
|
|
// from phpunit
|
|
abstract protected static function markTestSkipped(string $message = ''): void;
|
|
abstract protected static function assertTrue($condition, string $message = ''): void;
|
|
|
|
private $encryptionWasEnabled;
|
|
|
|
private $originalEncryptionModule;
|
|
|
|
/** @var IUserManager */
|
|
private $userManager;
|
|
/** @var SetupManager */
|
|
private $setupManager;
|
|
|
|
/**
|
|
* @var IConfig
|
|
*/
|
|
private $config;
|
|
|
|
/**
|
|
* @var Application
|
|
*/
|
|
private $encryptionApp;
|
|
|
|
protected function loginWithEncryption($user = '') {
|
|
\OC_Util::tearDownFS();
|
|
\OC_User::setUserId('');
|
|
// needed for fully logout
|
|
Server::get(IUserSession::class)->setUser(null);
|
|
|
|
$this->setupManager->tearDown();
|
|
|
|
\OC_User::setUserId($user);
|
|
$this->postLogin();
|
|
\OC_Util::setupFS($user);
|
|
if ($this->userManager->userExists($user)) {
|
|
\OC::$server->getUserFolder($user);
|
|
}
|
|
}
|
|
|
|
protected function setupForUser($name, $password) {
|
|
$this->setupManager->tearDown();
|
|
$this->setupManager->setupForUser($this->userManager->get($name));
|
|
|
|
$container = $this->encryptionApp->getContainer();
|
|
/** @var KeyManager $keyManager */
|
|
$keyManager = $container->query(KeyManager::class);
|
|
/** @var Setup $userSetup */
|
|
$userSetup = $container->query(Setup::class);
|
|
$userSetup->setupUser($name, $password);
|
|
$encryptionManager = $container->query(IManager::class);
|
|
$this->encryptionApp->setUp($encryptionManager);
|
|
$keyManager->init($name, $password);
|
|
}
|
|
|
|
protected function postLogin() {
|
|
$encryptionWrapper = new EncryptionWrapper(
|
|
new ArrayCache(),
|
|
Server::get(\OCP\Encryption\IManager::class),
|
|
Server::get(LoggerInterface::class)
|
|
);
|
|
|
|
$this->registerStorageWrapper('oc_encryption', [$encryptionWrapper, 'wrapStorage']);
|
|
}
|
|
|
|
protected function setUpEncryptionTrait() {
|
|
$isReady = Server::get(\OCP\Encryption\IManager::class)->isReady();
|
|
if (!$isReady) {
|
|
$this->markTestSkipped('Encryption not ready');
|
|
}
|
|
|
|
$this->userManager = Server::get(IUserManager::class);
|
|
$this->setupManager = Server::get(SetupManager::class);
|
|
|
|
Server::get(IAppManager::class)->loadApp('encryption');
|
|
|
|
$this->encryptionApp = new Application([], $isReady);
|
|
|
|
$this->config = Server::get(IConfig::class);
|
|
$this->encryptionWasEnabled = $this->config->getAppValue('core', 'encryption_enabled', 'no');
|
|
$this->originalEncryptionModule = $this->config->getAppValue('core', 'default_encryption_module');
|
|
$this->config->setAppValue('core', 'default_encryption_module', Encryption::ID);
|
|
$this->config->setAppValue('core', 'encryption_enabled', 'yes');
|
|
$this->assertTrue(Server::get(\OCP\Encryption\IManager::class)->isEnabled());
|
|
}
|
|
|
|
protected function tearDownEncryptionTrait() {
|
|
if ($this->config) {
|
|
$this->config->setAppValue('core', 'encryption_enabled', $this->encryptionWasEnabled);
|
|
$this->config->setAppValue('core', 'default_encryption_module', $this->originalEncryptionModule);
|
|
$this->config->deleteAppValue('encryption', 'useMasterKey');
|
|
}
|
|
}
|
|
}
|