summaryrefslogtreecommitdiffstats
path: root/apps/encryption/lib/Migration.php
diff options
context:
space:
mode:
Diffstat (limited to 'apps/encryption/lib/Migration.php')
-rw-r--r--apps/encryption/lib/Migration.php389
1 files changed, 389 insertions, 0 deletions
diff --git a/apps/encryption/lib/Migration.php b/apps/encryption/lib/Migration.php
new file mode 100644
index 00000000000..db9541d6f5a
--- /dev/null
+++ b/apps/encryption/lib/Migration.php
@@ -0,0 +1,389 @@
+<?php
+/**
+ * @author Björn Schießle <schiessle@owncloud.com>
+ * @author Joas Schilling <nickvergessen@owncloud.com>
+ * @author Robin Appelman <icewind@owncloud.com>
+ *
+ * @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 OCA\Encryption;
+
+
+use OC\Files\View;
+use OCP\IConfig;
+use OCP\IDBConnection;
+use OCP\ILogger;
+
+class Migration {
+
+ private $moduleId;
+ /** @var \OC\Files\View */
+ private $view;
+ /** @var \OCP\IDBConnection */
+ private $connection;
+ /** @var IConfig */
+ private $config;
+ /** @var ILogger */
+ private $logger;
+ /** @var string*/
+ protected $installedVersion;
+
+ /**
+ * @param IConfig $config
+ * @param View $view
+ * @param IDBConnection $connection
+ * @param ILogger $logger
+ */
+ public function __construct(IConfig $config, View $view, IDBConnection $connection, ILogger $logger) {
+ $this->view = $view;
+ $this->view->disableCacheUpdate();
+ $this->connection = $connection;
+ $this->moduleId = \OCA\Encryption\Crypto\Encryption::ID;
+ $this->config = $config;
+ $this->logger = $logger;
+ $this->installedVersion = $this->config->getAppValue('files_encryption', 'installed_version', '-1');
+ }
+
+ public function finalCleanUp() {
+ $this->view->deleteAll('files_encryption/public_keys');
+ $this->updateFileCache();
+ $this->config->deleteAppValue('files_encryption', 'installed_version');
+ }
+
+ /**
+ * update file cache, copy unencrypted_size to the 'size' column
+ */
+ private function updateFileCache() {
+ // make sure that we don't update the file cache multiple times
+ // only update during the first run
+ if ($this->installedVersion !== '-1') {
+ $query = $this->connection->getQueryBuilder();
+ $query->update('filecache')
+ ->set('size', 'unencrypted_size')
+ ->where($query->expr()->eq('encrypted', $query->createParameter('encrypted')))
+ ->setParameter('encrypted', 1);
+ $query->execute();
+ }
+ }
+
+ /**
+ * iterate through users and reorganize the folder structure
+ */
+ public function reorganizeFolderStructure() {
+ $this->reorganizeSystemFolderStructure();
+
+ $limit = 500;
+ $offset = 0;
+ do {
+ $users = \OCP\User::getUsers('', $limit, $offset);
+ foreach ($users as $user) {
+ $this->reorganizeFolderStructureForUser($user);
+ }
+ $offset += $limit;
+ } while (count($users) >= $limit);
+ }
+
+ /**
+ * reorganize system wide folder structure
+ */
+ public function reorganizeSystemFolderStructure() {
+
+ $this->createPathForKeys('/files_encryption');
+
+ // backup system wide folders
+ $this->backupSystemWideKeys();
+
+ // rename system wide mount point
+ $this->renameFileKeys('', '/files_encryption/keys');
+
+ // rename system private keys
+ $this->renameSystemPrivateKeys();
+
+ $storage = $this->view->getMount('')->getStorage();
+ $storage->getScanner()->scan('files_encryption');
+ }
+
+
+ /**
+ * reorganize folder structure for user
+ *
+ * @param string $user
+ */
+ public function reorganizeFolderStructureForUser($user) {
+ // backup all keys
+ \OC_Util::tearDownFS();
+ \OC_Util::setupFS($user);
+ if ($this->backupUserKeys($user)) {
+ // rename users private key
+ $this->renameUsersPrivateKey($user);
+ $this->renameUsersPublicKey($user);
+ // rename file keys
+ $path = '/files_encryption/keys';
+ $this->renameFileKeys($user, $path);
+ $trashPath = '/files_trashbin/keys';
+ if (\OC_App::isEnabled('files_trashbin') && $this->view->is_dir($user . '/' . $trashPath)) {
+ $this->renameFileKeys($user, $trashPath, true);
+ $this->view->deleteAll($trashPath);
+ }
+ // delete old folders
+ $this->deleteOldKeys($user);
+ $this->view->getMount('/' . $user)->getStorage()->getScanner()->scan('files_encryption');
+ }
+ }
+
+ /**
+ * update database
+ */
+ public function updateDB() {
+
+ // make sure that we don't update the file cache multiple times
+ // only update during the first run
+ if ($this->installedVersion === '-1') {
+ return;
+ }
+
+ // delete left-over from old encryption which is no longer needed
+ $this->config->deleteAppValue('files_encryption', 'ocsid');
+ $this->config->deleteAppValue('files_encryption', 'types');
+ $this->config->deleteAppValue('files_encryption', 'enabled');
+
+ $oldAppValues = $this->connection->getQueryBuilder();
+ $oldAppValues->select('*')
+ ->from('appconfig')
+ ->where($oldAppValues->expr()->eq('appid', $oldAppValues->createParameter('appid')))
+ ->setParameter('appid', 'files_encryption');
+ $appSettings = $oldAppValues->execute();
+
+ while ($row = $appSettings->fetch()) {
+ // 'installed_version' gets deleted at the end of the migration process
+ if ($row['configkey'] !== 'installed_version' ) {
+ $this->config->setAppValue('encryption', $row['configkey'], $row['configvalue']);
+ $this->config->deleteAppValue('files_encryption', $row['configkey']);
+ }
+ }
+
+ $oldPreferences = $this->connection->getQueryBuilder();
+ $oldPreferences->select('*')
+ ->from('preferences')
+ ->where($oldPreferences->expr()->eq('appid', $oldPreferences->createParameter('appid')))
+ ->setParameter('appid', 'files_encryption');
+ $preferenceSettings = $oldPreferences->execute();
+
+ while ($row = $preferenceSettings->fetch()) {
+ $this->config->setUserValue($row['userid'], 'encryption', $row['configkey'], $row['configvalue']);
+ $this->config->deleteUserValue($row['userid'], 'files_encryption', $row['configkey']);
+ }
+ }
+
+ /**
+ * create backup of system-wide keys
+ */
+ private function backupSystemWideKeys() {
+ $backupDir = 'encryption_migration_backup_' . date("Y-m-d_H-i-s");
+ $this->view->mkdir($backupDir);
+ $this->view->copy('files_encryption', $backupDir . '/files_encryption');
+ }
+
+ /**
+ * create backup of user specific keys
+ *
+ * @param string $user
+ * @return bool
+ */
+ private function backupUserKeys($user) {
+ $encryptionDir = $user . '/files_encryption';
+ if ($this->view->is_dir($encryptionDir)) {
+ $backupDir = $user . '/encryption_migration_backup_' . date("Y-m-d_H-i-s");
+ $this->view->mkdir($backupDir);
+ $this->view->copy($encryptionDir, $backupDir);
+ return true;
+ }
+ return false;
+ }
+
+ /**
+ * rename system-wide private keys
+ */
+ private function renameSystemPrivateKeys() {
+ $dh = $this->view->opendir('files_encryption');
+ $this->createPathForKeys('/files_encryption/' . $this->moduleId );
+ if (is_resource($dh)) {
+ while (($privateKey = readdir($dh)) !== false) {
+ if (!\OC\Files\Filesystem::isIgnoredDir($privateKey) ) {
+ if (!$this->view->is_dir('/files_encryption/' . $privateKey)) {
+ $this->view->rename('files_encryption/' . $privateKey, 'files_encryption/' . $this->moduleId . '/' . $privateKey);
+ $this->renameSystemPublicKey($privateKey);
+ }
+ }
+ }
+ closedir($dh);
+ }
+ }
+
+ /**
+ * rename system wide public key
+ *
+ * @param string $privateKey private key for which we want to rename the corresponding public key
+ */
+ private function renameSystemPublicKey($privateKey) {
+ $publicKey = substr($privateKey,0 , strrpos($privateKey, '.privateKey')) . '.publicKey';
+ $this->view->rename('files_encryption/public_keys/' . $publicKey, 'files_encryption/' . $this->moduleId . '/' . $publicKey);
+ }
+
+ /**
+ * rename user-specific private keys
+ *
+ * @param string $user
+ */
+ private function renameUsersPrivateKey($user) {
+ $oldPrivateKey = $user . '/files_encryption/' . $user . '.privateKey';
+ $newPrivateKey = $user . '/files_encryption/' . $this->moduleId . '/' . $user . '.privateKey';
+ if ($this->view->file_exists($oldPrivateKey)) {
+ $this->createPathForKeys(dirname($newPrivateKey));
+ $this->view->rename($oldPrivateKey, $newPrivateKey);
+ }
+ }
+
+ /**
+ * rename user-specific public keys
+ *
+ * @param string $user
+ */
+ private function renameUsersPublicKey($user) {
+ $oldPublicKey = '/files_encryption/public_keys/' . $user . '.publicKey';
+ $newPublicKey = $user . '/files_encryption/' . $this->moduleId . '/' . $user . '.publicKey';
+ if ($this->view->file_exists($oldPublicKey)) {
+ $this->createPathForKeys(dirname($newPublicKey));
+ $this->view->rename($oldPublicKey, $newPublicKey);
+ }
+ }
+
+ /**
+ * rename file keys
+ *
+ * @param string $user
+ * @param string $path
+ * @param bool $trash
+ */
+ private function renameFileKeys($user, $path, $trash = false) {
+
+ if ($this->view->is_dir($user . '/' . $path) === false) {
+ $this->logger->info('Skip dir /' . $user . '/' . $path . ': does not exist');
+ return;
+ }
+
+ $dh = $this->view->opendir($user . '/' . $path);
+
+ if (is_resource($dh)) {
+ while (($file = readdir($dh)) !== false) {
+ if (!\OC\Files\Filesystem::isIgnoredDir($file)) {
+ if ($this->view->is_dir($user . '/' . $path . '/' . $file)) {
+ $this->renameFileKeys($user, $path . '/' . $file, $trash);
+ } else {
+ $target = $this->getTargetDir($user, $path, $file, $trash);
+ if ($target !== false) {
+ $this->createPathForKeys(dirname($target));
+ $this->view->rename($user . '/' . $path . '/' . $file, $target);
+ } else {
+ $this->logger->warning(
+ 'did not move key "' . $file
+ . '" could not find the corresponding file in /data/' . $user . '/files.'
+ . 'Most likely the key was already moved in a previous migration run and is already on the right place.');
+ }
+ }
+ }
+ }
+ closedir($dh);
+ }
+ }
+
+ /**
+ * get system mount points
+ * wrap static method so that it can be mocked for testing
+ *
+ * @internal
+ * @return array
+ */
+ protected function getSystemMountPoints() {
+ return \OC_Mount_Config::getSystemMountPoints();
+ }
+
+ /**
+ * generate target directory
+ *
+ * @param string $user
+ * @param string $keyPath
+ * @param string $filename
+ * @param bool $trash
+ * @return string
+ */
+ private function getTargetDir($user, $keyPath, $filename, $trash) {
+ if ($trash) {
+ $filePath = substr($keyPath, strlen('/files_trashbin/keys/'));
+ $targetDir = $user . '/files_encryption/keys/files_trashbin/' . $filePath . '/' . $this->moduleId . '/' . $filename;
+ } else {
+ $filePath = substr($keyPath, strlen('/files_encryption/keys/'));
+ $targetDir = $user . '/files_encryption/keys/files/' . $filePath . '/' . $this->moduleId . '/' . $filename;
+ }
+
+ if ($user === '') {
+ // for system wide mounts we need to check if the mount point really exists
+ $normalized = \OC\Files\Filesystem::normalizePath($filePath);
+ $systemMountPoints = $this->getSystemMountPoints();
+ foreach ($systemMountPoints as $mountPoint) {
+ $normalizedMountPoint = \OC\Files\Filesystem::normalizePath($mountPoint['mountpoint']) . '/';
+ if (strpos($normalized, $normalizedMountPoint) === 0)
+ return $targetDir;
+ }
+ } else if ($trash === false && $this->view->file_exists('/' . $user. '/files/' . $filePath)) {
+ return $targetDir;
+ } else if ($trash === true && $this->view->file_exists('/' . $user. '/files_trashbin/' . $filePath)) {
+ return $targetDir;
+ }
+
+ return false;
+ }
+
+ /**
+ * delete old keys
+ *
+ * @param string $user
+ */
+ private function deleteOldKeys($user) {
+ $this->view->deleteAll($user . '/files_encryption/keyfiles');
+ $this->view->deleteAll($user . '/files_encryption/share-keys');
+ }
+
+ /**
+ * create directories for the keys recursively
+ *
+ * @param string $path
+ */
+ private function createPathForKeys($path) {
+ if (!$this->view->file_exists($path)) {
+ $sub_dirs = explode('/', $path);
+ $dir = '';
+ foreach ($sub_dirs as $sub_dir) {
+ $dir .= '/' . $sub_dir;
+ if (!$this->view->is_dir($dir)) {
+ $this->view->mkdir($dir);
+ }
+ }
+ }
+ }
+}