All files / src/crypter AES.ts

100% Statements 23/23
100% Branches 6/6
100% Functions 5/5
100% Lines 21/21

Press n or j to go to the next uncovered block, b, p or k for the previous block.

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 391x 1x 1x 1x   1x       26x       26x     26x 26x 2x 24x       25x 25x 25x 96x 25x       25x 25x 25x 114x 25x      
import { Block } from 'aes-ts';
import { ICrypter } from './ICrypter';
import { CBCMode, type IModeOfOperation } from './modeOfOperation';
import { PKCS7Padding, type IPadding } from './padding';
 
export class AES extends ICrypter {
 
    private readonly blockCrypter: Block;
 
    protected readonly blockSize = 16;
 
    public constructor(
        key: Uint8Array,
        private readonly modeOfOperation: IModeOfOperation = new CBCMode(),
        padding: IPadding = new PKCS7Padding()
    ) {
        super(padding);
        if (key.length !== 16 && key.length !== 24 && key.length !== 32)
            throw new Error('AES key must be 16, 24 or 32 bytes long.');
        this.blockCrypter = new Block(key);
    }
 
    protected encryptBlocks(blocks: Uint8Array[]): Uint8Array {
        const modeOfOperation = this.modeOfOperation.encryption;
        let encrypted = modeOfOperation.start();
        for (const block of blocks)
            encrypted = new Uint8Array([...encrypted, ...modeOfOperation.combine(block, block => this.blockCrypter.encrypt(block))]);
        return new Uint8Array([...encrypted, ...modeOfOperation.finish()]);
    }
 
    protected decryptBlocks(blocks: Uint8Array[]): Uint8Array {
        const modeOfOperation = this.modeOfOperation.decryption;
        let decrypted = modeOfOperation.start();
        for (const block of blocks)
            decrypted = new Uint8Array([...decrypted, ...modeOfOperation.combine(block, block => this.blockCrypter.decrypt(block))]);
        return new Uint8Array([...decrypted, ...modeOfOperation.finish()]);
    }
}