diff --git a/packages/backend/src/misc/create-temp.ts b/packages/backend/src/misc/create-temp.ts index fda63c7a9d..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 { +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; } catch (e) { cleanup(); throw e; } + + if (stream.truncated) { + cleanup(); + throw new Error('Read failed: input stream truncated'); + } + + return [filepath, cleanup]; } diff --git a/packages/backend/src/server/ServerUtilityService.ts b/packages/backend/src/server/ServerUtilityService.ts index c2a3132489..00eb97f679 100644 --- a/packages/backend/src/server/ServerUtilityService.ts +++ b/packages/backend/src/server/ServerUtilityService.ts @@ -54,7 +54,7 @@ export class ServerUtilityService { } } else { // Otherwise it's a file try { - const filepath = await saveToTempFile(part.file); + const [filepath] = await saveToTempFile(part.file); const tmpUploads = (request.tmpUploads ??= []); tmpUploads.push(filepath);