Prevent writing to media hasher when hasher stream has closed.

This commit is contained in:
DebaucheryLibrarian 2021-02-05 03:13:10 +01:00
parent 647e9bb186
commit 3e813ca251
1 changed files with 6 additions and 2 deletions

View File

@ -504,10 +504,10 @@ async function fetchSource(source, baseMedia) {
const maxAttempts = source.attempts || 3; const maxAttempts = source.attempts || 3;
logger.silly(`Fetching media from ${source.src}`); logger.silly(`Fetching media from ${source.src}`);
// attempts
async function attempt(attempts = 1) { async function attempt(attempts = 1) {
const hasher = new blake2.Hash('blake2b', { digestLength: 24 }); const hasher = new blake2.Hash('blake2b', { digestLength: 24 });
let hasherReady = true;
hasher.setEncoding('hex'); hasher.setEncoding('hex');
try { try {
@ -519,7 +519,10 @@ async function fetchSource(source, baseMedia) {
hashStream.on('data', (chunk) => { hashStream.on('data', (chunk) => {
size += chunk.length; size += chunk.length;
if (hasherReady) {
hasher.write(chunk); hasher.write(chunk);
}
}); });
const { mimetype } = source.stream const { mimetype } = source.stream
@ -551,6 +554,7 @@ async function fetchSource(source, baseMedia) {
}, },
}; };
} catch (error) { } catch (error) {
hasherReady = false;
hasher.end(); hasher.end();
if (error.code !== 'VERIFY_TYPE') { if (error.code !== 'VERIFY_TYPE') {