check for stream truncation in saveToTempFile

1 files changed, 9 insertions, 2 deletions

diff --git a/packages/backend/src/misc/create-temp.ts b/packages/backend/src/misc/create-temp.ts
index f2138abf66..9ba95cff42 100644
--- a/packages/backend/src/misc/create-temp.ts
+++ b/packages/backend/src/misc/create-temp.ts
@@ -30,13 +30,20 @@ export function createTempDir(): Promise<[string, () => void]> {
 	});
 }
 
-export async function saveToTempFile(stream: NodeJS.ReadableStream): Promise<[string, () => void]> {
+export async function saveToTempFile(stream: NodeJS.ReadableStream & { truncated?: boolean }): Promise<[string, () => void]> {
 	const [filepath, cleanup] = await createTemp();
+
 	try {
 		await pipeline(stream, fs.createWriteStream(filepath));
-		return [filepath, cleanup];
 	} catch (e) {
 		cleanup();
 		throw e;
 	}
+
+	if (stream.truncated) {
+		cleanup();
+		throw new Error('Read failed: input stream truncated');
+	}
+
+	return [filepath, cleanup];
 }