Encrypt stream in chunks

This commit is contained in:
Belma Gutlic 2019-12-24 13:46:50 +01:00
parent 5167df18a5
commit 128aba164d

View File

@ -1,22 +1,35 @@
import { Handshake } from "./handshake";
import { Buffer } from "buffer";
import { Handshake } from "./handshake";
interface ReturnEncryptionWrapper {
(source: Iterable<Uint8Array>): AsyncIterableIterator<Uint8Array>;
}
const maxPlaintextLength = 65519;
// Returns generator that encrypts payload from the user
export function encryptStream(handshake: Handshake): ReturnEncryptionWrapper {
return async function * (source) {
for await (const chunk of source) {
const chunkBuffer = Buffer.from(chunk);
const data = await handshake.encrypt(chunkBuffer, handshake.session);
yield data;
for (let i = 0; i < chunkBuffer.length; i += maxPlaintextLength) {
let end = i + maxPlaintextLength;
if (end > chunkBuffer.length) {
end = chunkBuffer.length;
}
const data = handshake.encrypt(chunkBuffer.slice(i, end), handshake.session);
yield data;
}
}
}
}
// Decrypt received payload to the user
export function decryptStream(handshake: Handshake): ReturnEncryptionWrapper {
return async function * (source) {