swtor/ssn
0
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

♻ Convert decrypt stream into transform

Browse source
This commit is contained in:
C-3PO 2018-07-05 11:32:29 +02:00
parent 9688c12797
commit 933134eb35
Signed by: c3po
GPG key ID: 62993C4BB4D86F24
2 changed files with 17 additions and 22 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
src/ssn/extractFileStream.ts
View file

@ -37,8 +37,8 @@ export default function extractFileStream(file: ISsnFileEntry, inputStream: stre
//pipe into decryption if file is encrypted
if (file.decryptionKeys !== undefined) {
const decryptTransform = decryptStream(curStream, [...file.decryptionKeys] as [number, number, number]);
curStream = decryptTransform;
const decryptTransform = decryptStream([...file.decryptionKeys] as [number, number, number]);
curStream = curStream.pipe(decryptTransform);
}
//pipe into decompression
@ -46,8 +46,7 @@ export default function extractFileStream(file: ISsnFileEntry, inputStream: stre
decompressTransform.on('error', (error) => {
throw new Error(`Error during decompression: ${error.message}`);
});
curStream.pipe(decompressTransform);
curStream = decompressTransform;
curStream = curStream.pipe(decompressTransform);
//set max length
const maxLength2 = streamSetMaxLength(curStream, file.size);

32
src/ssn/streams/decryptStream.ts
View file

@ -1,26 +1,22 @@
import * as stream from 'stream';
import getDecryptor from '../decryption/decryptChunk';
export default function decryptStream(inputStream: stream.Readable, decryptionKeys: [number, number, number]): stream.Readable {
export default function decryptTransform(decryptionKeys: [number, number, number]): stream.Transform {
const decryptor = getDecryptor(decryptionKeys);
const transform = new stream.Transform();
const outStream = new stream.Readable({
read(size) {
//Decrypt chunk
const encryptedChunk: Buffer = inputStream.read(size);
if (encryptedChunk === null) {
//If end has been reached, stop
this.push(null);
} else {
const decryptedChunk = decryptor(encryptedChunk);
this.push(decryptedChunk);
}
},
transform.on('readable', () => {
const encryptedChunk = transform.read();
if (encryptedChunk === null) {
transform.end();
return;
}
const decryptedChunk = decryptor(encryptedChunk);
transform.write(decryptedChunk);
});
inputStream.on('end', () => {
outStream.emit('end');
});
return outStream;
return transform;
}