From 5fe4ad835d192cd95bb0040e1d00abaf44a5a46c Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?C=C3=B4me=20Chilliet?= <come.chilliet@nextcloud.com>
Date: Thu, 28 Mar 2024 17:30:22 +0100
Subject: [PATCH] fix: Show errors in encryption:migrate-key-storage-format and
 continue to other files
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Signed-off-by: Côme Chilliet <come.chilliet@nextcloud.com>
---
 core/Command/Encryption/MigrateKeyStorage.php | 30 ++++++++++++-------
 1 file changed, 20 insertions(+), 10 deletions(-)

diff --git a/core/Command/Encryption/MigrateKeyStorage.php b/core/Command/Encryption/MigrateKeyStorage.php
index 183b98f93dfc3..321b554c6f37d 100644
--- a/core/Command/Encryption/MigrateKeyStorage.php
+++ b/core/Command/Encryption/MigrateKeyStorage.php
@@ -55,7 +55,7 @@ protected function execute(InputInterface $input, OutputInterface $output): int
 	protected function updateKeys(string $root, OutputInterface $output): bool {
 		$output->writeln("Start to update the keys:");
 
-		$this->updateSystemKeys($root);
+		$this->updateSystemKeys($root, $output);
 		$this->updateUsersKeys($root, $output);
 		$this->config->deleteSystemValue('encryption.key_storage_migrated');
 		return true;
@@ -64,15 +64,15 @@ protected function updateKeys(string $root, OutputInterface $output): bool {
 	/**
 	 * Move system key folder
 	 */
-	protected function updateSystemKeys(string $root): void {
+	protected function updateSystemKeys(string $root, OutputInterface $output): void {
 		if (!$this->rootView->is_dir($root . '/files_encryption')) {
 			return;
 		}
 
-		$this->traverseKeys($root . '/files_encryption', null);
+		$this->traverseKeys($root . '/files_encryption', null, $output);
 	}
 
-	private function traverseKeys(string $folder, ?string $uid): void {
+	private function traverseKeys(string $folder, ?string $uid, OutputInterface $output): void {
 		$listing = $this->rootView->getDirectoryContent($folder);
 
 		foreach ($listing as $node) {
@@ -88,6 +88,11 @@ private function traverseKeys(string $folder, ?string $uid): void {
 
 				$content = $this->rootView->file_get_contents($path);
 
+				if ($content === false) {
+					$output->writeln("<error>Failed to open path $path</error>");
+					continue;
+				}
+
 				try {
 					$this->crypto->decrypt($content);
 					continue;
@@ -106,12 +111,12 @@ private function traverseKeys(string $folder, ?string $uid): void {
 		}
 	}
 
-	private function traverseFileKeys(string $folder): void {
+	private function traverseFileKeys(string $folder, OutputInterface $output): void {
 		$listing = $this->rootView->getDirectoryContent($folder);
 
 		foreach ($listing as $node) {
 			if ($node['mimetype'] === 'httpd/unix-directory') {
-				$this->traverseFileKeys($folder . '/' . $node['name']);
+				$this->traverseFileKeys($folder . '/' . $node['name'], $output);
 			} else {
 				$endsWith = function (string $haystack, string $needle): bool {
 					$length = strlen($needle);
@@ -130,6 +135,11 @@ private function traverseFileKeys(string $folder): void {
 
 					$content = $this->rootView->file_get_contents($path);
 
+					if ($content === false) {
+						$output->writeln("<error>Failed to open path $path</error>");
+						continue;
+					}
+
 					try {
 						$this->crypto->decrypt($content);
 						continue;
@@ -173,7 +183,7 @@ protected function updateUsersKeys(string $root, OutputInterface $output): void
 				foreach ($users as $user) {
 					$progress->advance();
 					$this->setupUserFS($user);
-					$this->updateUserKeys($root, $user);
+					$this->updateUserKeys($root, $user, $output);
 				}
 				$offset += $limit;
 			} while (count($users) >= $limit);
@@ -186,16 +196,16 @@ protected function updateUsersKeys(string $root, OutputInterface $output): void
 	 *
 	 * @throws \Exception
 	 */
-	protected function updateUserKeys(string $root, string $user): void {
+	protected function updateUserKeys(string $root, string $user, OutputInterface $output): void {
 		if ($this->userManager->userExists($user)) {
 			$source = $root . '/' . $user . '/files_encryption/OC_DEFAULT_MODULE';
 			if ($this->rootView->is_dir($source)) {
-				$this->traverseKeys($source, $user);
+				$this->traverseKeys($source, $user, $output);
 			}
 
 			$source = $root . '/' . $user . '/files_encryption/keys';
 			if ($this->rootView->is_dir($source)) {
-				$this->traverseFileKeys($source);
+				$this->traverseFileKeys($source, $output);
 			}
 		}
 	}