feat: implement IterativeAes128MessagePublic with test

README.md

@@ -24,8 +24,6 @@ This work was spawned from the Mina Grant Starter Program and aims to provide a
 
 - AES-128 encryption
 - Static and dynamic message sizes
-- Counter Mode (CTR)
-- Galois Counter Mode (GCM) (coming soon)
 
 ## Installation and Quick Start
 
@@ -77,6 +75,14 @@ npm run build
 node ./build/test/circuitSummary.js
 ```
 
+## Code Breakdown
+The main entrypoint of the code is contained within `src/implementations/IterativeAES128.ts` which has the following implemented.
+    - `IterativeAes128` is responsible for verifying that a cipher has been encrypted using AES with an arbitrary key and message
+    - `IterativeAes128MessagePublic` is responsible for verifying that a cipher **and** a message have been encrypted using AES with an arbitrary key.
+    - `computeIterativeAes128Encryption()` which can be inlined within circuits in order to proof AES encryption.
+
+Additionally, `Byte16` is used to represent 256-bit numbers and is commonly used as inputs to functions and circuits.
+
 ## Circuit Breakdown
 
 ### AES128 Iterative Summary
@@ -90,13 +96,6 @@ node ./build/test/circuitSummary.js
 | Rot64        | 4800   |
 | RangeCheck0  | 4800   |
 
-### Core:
-- Implementing block mode: **Counter Mode (CTR)**.
-
-### Optional:
-
-- User authentication block mode: **Galois Counter Mode (GCM)**.
-
 # Contributing
 Everyone is welcome to contribute, file an issue or submit a pull request if you think there is something worth mentioning.
 

src/implementations/IterativeAES128.ts

@@ -67,6 +67,31 @@ const IterativeAes128 = ZkProgram({
   },
 });
 
+class IterativeAES128MessagePublicInput extends Struct({
+  cipher: Byte16,
+  message: Byte16,
+}) {}
+
+/**
+ * A zkProgram that verifies a proof that a message and cipher are conncted via AES-128 using the given key.
+ */
+const IterativeAes128MessagePublic = ZkProgram({
+  name: "aes-verify-iterative-decrypt",
+  publicInput: IterativeAES128MessagePublicInput,
+
+  methods: {
+    verifyAES128: {
+      privateInputs: [Byte16],
+
+      async method(input: IterativeAES128MessagePublicInput, key: Byte16) {
+        const message = input.message;
+        const state = computeIterativeAes128Encryption(message, key);
+        state.assertEquals(input.cipher);
+      },
+    },
+  },
+});
+
 /**
  * Generates a proof that the given message was encrypted with AES-128 using the given key.
  * The key must be in hex form.
@@ -77,7 +102,6 @@ const IterativeAes128 = ZkProgram({
  * @throws If the message is not 16 characters long or the key is not 32 characters long
  * @throws If the proof generation fails
  */
-// NO TEST NOW AS IT WILL CHANGE SOON
 async function generateIterativeAes128Proof(
   message: string,
   keyHex: string, // Should we allow non hex strings?
@@ -107,4 +131,6 @@ export {
   generateIterativeAes128Proof,
   IterativeAes128,
   IterativeAES128PublicInput,
+  IterativeAes128MessagePublic,
+  IterativeAES128MessagePublicInput,
 };

src/index.ts

@@ -1,8 +1,16 @@
 import {
   IterativeAes128,
   IterativeAES128PublicInput,
+  IterativeAes128MessagePublic,
+  IterativeAES128MessagePublicInput,
 } from "./implementations/IterativeAES128";
 import { Byte16 } from "./primitives/Bytes.js";
 
-export { IterativeAes128, IterativeAES128PublicInput, Byte16 };
+export {
+  IterativeAes128,
+  IterativeAES128PublicInput,
+  Byte16,
+  IterativeAes128MessagePublic,
+  IterativeAES128MessagePublicInput,
+};
 export { generateIterativeAes128Proof as generateAes128Proof } from "./implementations/IterativeAES128";

test/implementations/verifyAES128.test.ts

@@ -2,6 +2,8 @@ import {
   computeIterativeAes128Encryption,
   IterativeAes128,
   IterativeAES128PublicInput as AESPublicInput,
+  IterativeAes128MessagePublic,
+  IterativeAES128MessagePublicInput,
 } from "../../src/implementations/IterativeAES128.js";
 import { encryptAES128 } from "../../src/utils/crypto.js";
 import { Byte16 } from "../../src/primitives/Bytes.js";
@@ -53,3 +55,24 @@ describe("Iterative AES128 Encryption", () => {
     },
   );
 });
+
+describe("Iterative AES128 Decryption", () => {
+  (RUN_ZK_TESTS ? it : it.skip)(
+    "should verify the proof using the zkProgram",
+    async () => {
+      const { verificationKey } = await IterativeAes128MessagePublic.compile();
+      const { plaintext, key } = testVectorToByte16(testVector1);
+      const cipher = Byte16.fromHex(getCipherText(testVector1));
+      const input = new IterativeAES128MessagePublicInput({
+        cipher,
+        message: plaintext,
+      });
+      const { proof } = await IterativeAes128MessagePublic.verifyAES128(
+        input,
+        key,
+      );
+      const isValid = await verify(proof, verificationKey);
+      expect(isValid).toBe(true);
+    },
+  );
+});