tests.ts•3.11 kB
import assert from "assert"
import fs from "fs";
import { join, resolve } from "path";
import * as aes from "./index.js";
interface TestCase {
modeOfOperation: string;
iv: Array<number>;
key: Array<number>;
plaintext: Array<Array<number>>;
encrypted: Array<Array<number>>;
segmentSize: number;
}
const root = (function() {
let root = process.cwd();
while (true) {
if (fs.existsSync(join(root, "package.json"))) { return root; }
const parent = join(root, "..");
if (parent === root) { break; }
root = parent;
}
throw new Error("could not find root");
})();
describe("Tests Encrypting and Decrypting", function() {
const json = fs.readFileSync(resolve(root, "./test/test-vectors.json")).toString()
const tests: Array<TestCase> = JSON.parse(json);
function getCrypter(key: Uint8Array, test: TestCase): null | aes.ModeOfOperation {
switch (test.modeOfOperation) {
case "ctr":
return new aes.CTR(key, 0);
case "cbc":
return new aes.CBC(key, Buffer.from(test.iv));
case "cfb":
return new aes.CFB(key, Buffer.from(test.iv), test.segmentSize * 8);
case "ecb":
return new aes.ECB(key);
case "ofb":
return new aes.OFB(key, Buffer.from(test.iv));
}
return null;
}
tests.forEach((test, index) => {
it(`tests encrypting: ${ test. modeOfOperation}.${ index }`, function() {
const encrypter = getCrypter(Buffer.from(test.key), test);
if (!encrypter) { this.skip(); }
for (let i = 0; i < test.plaintext.length; i++) {
const plaintext = Buffer.from(test.plaintext[i]);
const ciphertext = Buffer.from(test.encrypted[i]);
const result = Buffer.from(encrypter.encrypt(plaintext));
assert.ok(ciphertext.equals(result), "encrypting failed");
}
});
it(`tests decrypting: ${ test. modeOfOperation}.${ index }`, function() {
const decrypter = getCrypter(Buffer.from(test.key), test);
if (!decrypter) { this.skip(); }
for (let i = 0; i < test.plaintext.length; i++) {
const plaintext = Buffer.from(test.plaintext[i]);
const ciphertext = Buffer.from(test.encrypted[i]);
const result = Buffer.from(decrypter.decrypt(ciphertext));
assert.ok(plaintext.equals(result), "decrypting failed");
}
});
});
});
describe("Tests Padding", function() {
for (let size = 0; size < 100; size++) {
it(`tests padding: length=${ size }`, function() {
// Create a random piece of data
const data = new Uint8Array(size);
data.fill(42);
// Pad it
const padded = aes.pkcs7Pad(data);
assert.ok((padded.length % 16) === 0, "Failed to pad to block size");
assert.ok(data.length <= padded.length && padded.length <= data.length + 16, "Padding went awry");
assert.ok(padded[padded.length - 1] >= 1 && padded[padded.length - 1] <= 16, "Failed to pad to block size");
// Trim it
const trimmed = aes.pkcs7Strip(padded);
assert.ok(Buffer.from(data).equals(trimmed), "Failed to trim to original data");
});
}
});