mirrors/barkey
1
0
Fork 0
mirror of https://codeberg.org/yeentown/barkey.git synced 2025-07-07 20:44:34 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

check for stream truncation in saveToTempFile

Browse source
This commit is contained in:
Hazelnoot 2025-05-08 16:43:52 -04:00
parent e75e4f11a2
commit 7bfe16cbb0
1 changed files with 9 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
packages/backend/src/misc/create-temp.ts
View file

@ -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(); const [filepath, cleanup] = await createTemp();
try { try {
await pipeline(stream, fs.createWriteStream(filepath)); await pipeline(stream, fs.createWriteStream(filepath));
return [filepath, cleanup];
} catch (e) { } catch (e) {
cleanup(); cleanup();
throw e; throw e;
} }
if (stream.truncated) {
cleanup();
throw new Error('Read failed: input stream truncated');
}
return [filepath, cleanup];
} }