123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081 |
- import assert from "assert";
- import fs from "fs";
- import { join, resolve } from "path";
- import * as aes from "./index.js";
- 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 = JSON.parse(json);
- function getCrypter(key, test) {
- 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");
- });
- }
- });
- //# sourceMappingURL=tests.js.map
|