Abort processing when user-connection aborted

Signed-off-by: Alexander Brandscheidt <[email protected]>
nextcloud · Dec 18, 2023 · 4a00f33 · 4a00f33
1 parent 0ff6746
commit 4a00f33
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/lib/private/Streamer.php b/lib/private/Streamer.php
@@ -28,6 +28,7 @@
  */
 namespace OC;
 
+use Icewind\Streams\CallBackWrapper;
 use OC\Files\Filesystem;
 use OCP\Files\File;
 use OCP\Files\Folder;
@@ -126,6 +127,7 @@ public function addDirRecursive(string $dir, string $internalDir = ''): void {
 		/** @var LoggerInterface $logger */
 		$logger = \OC::$server->query(LoggerInterface::class);
 		foreach ($files as $file) {
+			if(connection_status() !== CONNECTION_NORMAL) return;
 			if ($file instanceof File) {
 				try {
 					$fh = $file->fopen('r');
@@ -161,6 +163,15 @@ public function addDirRecursive(string $dir, string $internalDir = ''): void {
 	 * @return bool $success
 	 */
 	public function addFileFromStream($stream, string $internalName, int|float $size, $time): bool {
+		if(connection_status() !== CONNECTION_NORMAL) return false;
+		// Close file-stream when user-connection closed
+		$stream = CallbackWrapper::wrap($stream,
+			function ($count) use ($stream) {
+				if (connection_status() !== CONNECTION_NORMAL) {
+					fclose($stream);
+				}
+			});
+
 		$options = [];
 		if ($time) {
 			$options = [