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

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

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