feat: decrypt the stream of a file version as returned by IVersionedS…

…torage
owncloud · Oct 18, 2021 · 349f008 · 349f008
1 parent 256cbfe
commit 349f008
Show file tree

Hide file tree

Showing 3 changed files with 87 additions and 7 deletions.
diff --git a/lib/private/Files/Storage/Wrapper/Encryption.php b/lib/private/Files/Storage/Wrapper/Encryption.php
@@ -40,10 +40,11 @@
 use OCP\Encryption\Keys\IStorage;
 use OCP\Files\Mount\IMountPoint;
 use OCP\Files\Storage;
+use OCP\Files\Storage\IVersionedStorage;
 use OCP\ILogger;
 use OCP\Files\Cache\ICacheEntry;
 
-class Encryption extends Wrapper {
+class Encryption extends Wrapper implements Storage\IVersionedStorage {
 	use LocalTempFileTrait;
 
 	/** @var string */
@@ -491,7 +492,8 @@ public function fopen($path, $mode) {
 					$unencryptedSize,
 					$headerSize,
 					$signed,
-					$sourceFileOfRename
+					$sourceFileOfRename,
+					null # TODO: maybe fill it?
 				);
 				unset($this->sourcePath[$path]);
 
@@ -1087,4 +1089,79 @@ protected function isVersion($path) {
 		$normalized = Filesystem::normalizePath($path);
 		return \substr($normalized, 0, \strlen('/files_versions/')) === '/files_versions/';
 	}
+
+	public function getVersions($internalPath) {
+		if ($this->storage instanceof IVersionedStorage) {
+			return $this->storage->getVersions($internalPath);
+		}
+		return null;
+	}
+
+	public function getVersion($internalPath, $versionId) {
+		if ($this->storage instanceof IVersionedStorage) {
+			return $this->storage->getVersion($internalPath, $versionId);
+		}
+		return null;
+	}
+
+	public function getContentOfVersion($internalPath, $versionId) {
+		if ($this->storage instanceof IVersionedStorage) {
+			$stream = $this->storage->getContentOfVersion($internalPath, $versionId);
+			$info = $this->storage->getVersion($internalPath, $versionId);
+			$header = $this->getHeader($stream);
+			$signed = (isset($header['signed']) && $header['signed'] === 'true') ? true : false;
+			$path = $internalPath;
+			# getFullPath returns wrong information here .....
+			$fullPath = $this->getFullPath($path);
+			$encryptionModuleId = $this->util->getEncryptionModuleId($header);
+			$encryptionModule = $this->encryptionManager->getEncryptionModule($encryptionModuleId);
+
+			$size = $this->storage->filesize($internalPath);
+			$unencryptedSize = $size;
+			$fileEncryptionVersion = null;
+			if ($info['size']) {
+				$unencryptedSize = $info['size'];
+			}
+			if ($info['encryptedVersion']) {
+				$fileEncryptionVersion = $info['encryptedVersion'];
+			}
+			$headerSize = $this->getHeaderSize($stream);
+
+			return \OC\Files\Stream\Encryption::wrap(
+				$stream,
+				$path,
+				$fullPath,
+				$header,
+				$this->uid,
+				$encryptionModule,
+				$this->storage,
+				$this,
+				$this->util,
+				$this->fileHelper,
+				'rb',
+				$size,
+				$unencryptedSize,
+				$headerSize,
+				$signed,
+				null,
+				$fileEncryptionVersion
+			);
+		}
+		return null;
+	}
+
+	public function restoreVersion($internalPath, $versionId) {
+		if ($this->storage instanceof IVersionedStorage) {
+			return $this->storage->restoreVersion($internalPath, $versionId);
+		}
+		return null;
+	}
+
+	public function saveVersion($internalPath) {
+		if ($this->storage instanceof IVersionedStorage) {
+			return $this->storage->saveVersion($internalPath);
+		}
+		return null;
+	}
+
 }
diff --git a/lib/private/Files/Stream/Encryption.php b/lib/private/Files/Stream/Encryption.php
@@ -116,7 +116,8 @@ public function __construct() {
 			'encryptionStorage',
 			'headerSize',
 			'signed',
-			'sourceFileOfRename'
+			'sourceFileOfRename',
+			'fileEncryptionVersion'
 		];
 	}
 
@@ -160,7 +161,8 @@ public static function wrap(
 		$unencryptedSize,
 		$headerSize,
 		$signed,
-		$sourceFileOfRename = null,
+		$sourceFileOfRename,
+		$fileEncryptionVersion,
 		$wrapper =  'OC\Files\Stream\Encryption'
 	) {
 		$context = \stream_context_create([
@@ -179,7 +181,8 @@ public static function wrap(
 				'encryptionStorage' => $encStorage,
 				'headerSize' => $headerSize,
 				'signed' => $signed,
-				'sourceFileOfRename' => $sourceFileOfRename
+				'sourceFileOfRename' => $sourceFileOfRename,
+				'fileEncryptionVersion' => $fileEncryptionVersion
 			]
 		]);
 
@@ -259,7 +262,7 @@ public function stream_open($path, $mode, $options, &$opened_path) {
 		}
 
 		$accessList = $this->file->getAccessList($sharePath);
-		$this->newHeader = $this->encryptionModule->begin($this->fullPath, $this->uid, $mode, $this->header, $accessList, $context['sourceFileOfRename']);
+		$this->newHeader = $this->encryptionModule->begin($this->fullPath, $this->uid, $mode, $this->header, $accessList, $context['sourceFileOfRename'], $context['fileEncryptionVersion']);
 		$this->unencryptedBlockSize = $this->encryptionModule->getUnencryptedBlockSize($this->signed);
 
 		if (

diff --git a/lib/public/Encryption/IEncryptionModule.php b/lib/public/Encryption/IEncryptionModule.php
@@ -66,7 +66,7 @@ public function getDisplayName();
 	 *                       or if no additional data is needed return a empty array
 	 * @since 8.1.0
 	 */
-	public function begin($path, $user, $mode, array $header, array $accessList, $sourceFileOfRename);
+	public function begin($path, $user, $mode, array $header, array $accessList, $sourceFileOfRename, $fileEncryptionVersion);
 
 	/**
 	 * last chunk received. This is the place where you can perform some final