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