From 7bfe16cbb0ba0562586b471a8d7ae24bfe3dacfa Mon Sep 17 00:00:00 2001 From: Hazelnoot Date: Thu, 8 May 2025 16:43:52 -0400 Subject: [PATCH] check for stream truncation in saveToTempFile --- packages/backend/src/misc/create-temp.ts | 11 +++++++++-- 1 file 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]; }