Merge pull request #196 from zkemail/helpers/poseidon-large

Add poseidonLarge JS version
zkemail · May 27, 2024 · 1b11429 · 1b11429
2 parents 3b9ffd9 + a520e3c
commit 1b11429
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 9 deletions.
diff --git a/packages/circuits/tests/email-verifier.test.ts b/packages/circuits/tests/email-verifier.test.ts
@@ -5,7 +5,7 @@ import path from "path";
 import { DKIMVerificationResult } from "@zk-email/helpers/src/dkim";
 import { generateEmailVerifierInputsFromDKIMResult } from "@zk-email/helpers/src/input-generators";
 import { verifyDKIMSignature } from "@zk-email/helpers/src/dkim";
-import { bigIntToChunkedBytes } from "@zk-email/helpers/src/binary-format";
+import { poseidonLarge } from "@zk-email/helpers/src/hash";
 
 
 describe("EmailVerifier", () => {
@@ -169,15 +169,13 @@ describe("EmailVerifier", () => {
     });
 
     // Calculate the Poseidon hash with pubkey chunked to 9*242 like in circuit
-    const poseidon = await buildPoseidon();
-    const pubkeyChunked = bigIntToChunkedBytes(dkimResult.publicKey, 242, 9);
-    const hash = poseidon(pubkeyChunked);
+    const poseidonHash = await poseidonLarge(dkimResult.publicKey, 9, 242);
 
     // Calculate the hash using the circuit
     const witness = await circuit.calculateWitness(emailVerifierInputs);
 
     await circuit.assertOut(witness, {
-      pubkeyHash: poseidon.F.toObject(hash),
+      pubkeyHash: poseidonHash,
     });
   });
 });

diff --git a/packages/helpers/src/hash.ts b/packages/helpers/src/hash.ts
@@ -0,0 +1,10 @@
+import { buildPoseidon } from 'circomlibjs';
+import { bigIntToChunkedBytes } from './binary-format';
+
+export async function poseidonLarge(input: bigint, numChunks: number, bitsPerChunk: number) {
+  const poseidon = await buildPoseidon();
+  const pubkeyChunked = bigIntToChunkedBytes(input, bitsPerChunk, numChunks);
+  const hash = poseidon(pubkeyChunked);
+
+  return poseidon.F.toObject(hash) as Promise<bigint>;
+}
diff --git a/scripts/dkim/update-dkim-registry.ts b/scripts/dkim/update-dkim-registry.ts
@@ -6,6 +6,7 @@ import forge from "node-forge";
 import { bigIntToChunkedBytes } from "@zk-email/helpers/src/binaryFormat";
 const fs = require("fs");
 import { abi } from "../abis/DKIMRegistry.json";
+import { poseidonLarge } from "@zk-email/helpers/src/hash";
 require("dotenv").config();
 
 async function updateContract(domain: string, pubkeyHashes: string[]) {
@@ -252,16 +253,15 @@ async function updateDKIMRegistry({
 
   // Generate pub key hash using 242 * 9 chunks (Poseidon lib don't take more than 16 inputs)
   const domainHashedPubKeyMap: { [key: string]: string[] } = {};
-  const poseidon = await buildPoseidon();
+
   for (let domain of Object.keys(domainPubKeyMap)) {
     for (let { publicKey } of domainPubKeyMap[domain]) {
-      const pubkeyChunked = bigIntToChunkedBytes(BigInt(publicKey), 242, 9);
-      const hash = poseidon(pubkeyChunked);
+      const poseidonHash = await poseidonLarge(BigInt(publicKey), 9, 242);
 
       if (!domainHashedPubKeyMap[domain]) {
         domainHashedPubKeyMap[domain] = [];
       }
-      domainHashedPubKeyMap[domain].push(poseidon.F.toObject(hash).toString());
+      domainHashedPubKeyMap[domain].push(poseidonHash.toString());
     }
   }
   _writeToFile("dkim-keys-hashed.json", domainHashedPubKeyMap);