@maci-protocol/circuits 0.0.0-ci.4c6d4e8 → 0.0.0-ci.52ce07e

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (48) hide show
  1. package/LICENSE +1 -2
  2. package/build/ts/{genZkeys.d.ts → generateZkeys.d.ts} +1 -1
  3. package/build/ts/generateZkeys.d.ts.map +1 -0
  4. package/build/ts/{genZkeys.js → generateZkeys.js} +1 -1
  5. package/build/ts/generateZkeys.js.map +1 -0
  6. package/build/ts/types.d.ts +11 -11
  7. package/build/ts/types.d.ts.map +1 -1
  8. package/build/tsconfig.build.tsbuildinfo +1 -1
  9. package/circom/circuits.json +23 -7
  10. package/circom/coordinator/full/MessageProcessor.circom +253 -0
  11. package/circom/coordinator/full/SingleMessageProcessor.circom +204 -0
  12. package/circom/coordinator/non-qv/processMessages.circom +110 -112
  13. package/circom/coordinator/non-qv/tallyVotes.circom +49 -46
  14. package/circom/coordinator/qv/processMessages.circom +111 -110
  15. package/circom/coordinator/qv/tallyVotes.circom +63 -63
  16. package/circom/utils/{calculateTotal.circom → CalculateTotal.circom} +8 -6
  17. package/circom/utils/{verifySignature.circom → EdDSAPoseidonVerifier.circom} +40 -66
  18. package/circom/utils/MessageHasher.circom +57 -0
  19. package/circom/utils/MessageToCommand.circom +107 -0
  20. package/circom/utils/PoseidonHasher.circom +29 -0
  21. package/circom/utils/{privToPubKey.circom → PrivateToPublicKey.circom} +12 -10
  22. package/circom/utils/VerifySignature.circom +39 -0
  23. package/circom/utils/full/MessageValidator.circom +91 -0
  24. package/circom/utils/full/StateLeafAndBallotTransformer.circom +122 -0
  25. package/circom/utils/non-qv/{messageValidator.circom → MessageValidator.circom} +17 -15
  26. package/circom/utils/non-qv/{stateLeafAndBallotTransformer.circom → StateLeafAndBallotTransformer.circom} +36 -36
  27. package/circom/utils/qv/{messageValidator.circom → MessageValidator.circom} +17 -15
  28. package/circom/utils/qv/{stateLeafAndBallotTransformer.circom → StateLeafAndBallotTransformer.circom} +36 -36
  29. package/circom/utils/trees/BinaryMerkleRoot.circom +11 -3
  30. package/circom/utils/trees/CheckRoot.circom +18 -14
  31. package/circom/utils/trees/LeafExists.circom +3 -3
  32. package/circom/utils/trees/{MerkleGeneratePathIndices.circom → MerklePathIndicesGenerator.circom} +11 -7
  33. package/circom/utils/trees/MerkleTreeInclusionProof.circom +10 -9
  34. package/circom/utils/trees/QuinaryCheckRoot.circom +54 -0
  35. package/circom/utils/trees/QuinaryGeneratePathIndices.circom +44 -0
  36. package/circom/utils/trees/QuinaryLeafExists.circom +30 -0
  37. package/circom/utils/trees/QuinarySelector.circom +42 -0
  38. package/circom/utils/trees/QuinaryTreeInclusionProof.circom +55 -0
  39. package/circom/utils/trees/Splicer.circom +76 -0
  40. package/circom/voter/PollJoined.circom +43 -0
  41. package/circom/voter/PollJoining.circom +54 -0
  42. package/package.json +15 -12
  43. package/build/ts/genZkeys.d.ts.map +0 -1
  44. package/build/ts/genZkeys.js.map +0 -1
  45. package/circom/utils/hashers.circom +0 -78
  46. package/circom/utils/messageToCommand.circom +0 -78
  47. package/circom/utils/trees/incrementalQuinaryTree.circom +0 -287
  48. package/circom/voter/poll.circom +0 -92
@@ -0,0 +1,43 @@
1
+ pragma circom 2.0.0;
2
+
3
+ // local imports
4
+ include "../utils/PoseidonHasher.circom";
5
+ include "../utils/PrivateToPublicKey.circom";
6
+ include "../utils/trees/BinaryMerkleRoot.circom";
7
+
8
+ // Poll Joined Circuit
9
+ // Allows a user to prove that they have joined a MACI poll.
10
+ // This is to be used with the MACI offchain implementation to allow
11
+ // users to authenticate to the relayer service (to reduce spamming).
12
+ template PollJoined(stateTreeDepth) {
13
+ // Constants defining the tree structure
14
+ var STATE_TREE_ARITY = 2;
15
+
16
+ // User's private key
17
+ signal input privateKey;
18
+ // User's voice credits balance
19
+ signal input voiceCreditsBalance;
20
+ // Path elements
21
+ signal input pathElements[stateTreeDepth][STATE_TREE_ARITY - 1];
22
+ // Path indices
23
+ signal input pathIndices[stateTreeDepth];
24
+ // Poll State tree root which proves the user is joined
25
+ signal input stateRoot;
26
+ // The actual tree depth (might be <= stateTreeDepth) Used in BinaryMerkleRoot
27
+ signal input actualStateTreeDepth;
28
+
29
+ // User private to public key
30
+ var derivedPublicKey[2] = PrivateToPublicKey()(privateKey);
31
+
32
+ var stateLeaf = PoseidonHasher(3)([derivedPublicKey[0], derivedPublicKey[1], voiceCreditsBalance]);
33
+
34
+ // Inclusion proof
35
+ var calculatedRoot = BinaryMerkleRoot(stateTreeDepth)(
36
+ stateLeaf,
37
+ actualStateTreeDepth,
38
+ pathIndices,
39
+ pathElements
40
+ );
41
+
42
+ calculatedRoot === stateRoot;
43
+ }
@@ -0,0 +1,54 @@
1
+ pragma circom 2.0.0;
2
+
3
+ // local imports
4
+ include "../utils/PoseidonHasher.circom";
5
+ include "../utils/PrivateToPublicKey.circom";
6
+ include "../utils/trees/BinaryMerkleRoot.circom";
7
+
8
+ // Poll Joining Circuit
9
+ // Allows a user to prove knowledge of a private key that is signed up to
10
+ // a MACI contract.
11
+ template PollJoining(stateTreeDepth) {
12
+ // Constants defining the tree structure
13
+ var STATE_TREE_ARITY = 2;
14
+
15
+ // User's private key
16
+ signal input privateKey;
17
+ // Poll's public key
18
+ signal input pollPublicKey[2];
19
+ // Siblings
20
+ signal input siblings[stateTreeDepth][STATE_TREE_ARITY - 1];
21
+ // Indices
22
+ signal input indices[stateTreeDepth];
23
+ // User's hashed private key
24
+ signal input nullifier;
25
+ // MACI State tree root which proves the user is signed up
26
+ signal input stateRoot;
27
+ // The actual tree depth (might be <= stateTreeDepth) used in BinaryMerkleRoot
28
+ signal input actualStateTreeDepth;
29
+ // The poll id
30
+ signal input pollId;
31
+
32
+ // Compute the nullifier (hash of private key and poll id)
33
+ var computedNullifier = PoseidonHasher(2)([privateKey, pollId]);
34
+ nullifier === computedNullifier;
35
+
36
+ // User private to public key
37
+ var derivedPublicKey[2] = PrivateToPublicKey()(privateKey);
38
+ // Hash the public key
39
+ var publicKeyHash = PoseidonHasher(2)([derivedPublicKey[0], derivedPublicKey[1]]);
40
+
41
+ // Ensure the poll public key is the same as the maci one (public input)
42
+ derivedPublicKey[0] === pollPublicKey[0];
43
+ derivedPublicKey[1] === pollPublicKey[1];
44
+
45
+ // Inclusion proof
46
+ var calculatedRoot = BinaryMerkleRoot(stateTreeDepth)(
47
+ publicKeyHash,
48
+ actualStateTreeDepth,
49
+ indices,
50
+ siblings
51
+ );
52
+
53
+ calculatedRoot === stateRoot;
54
+ }
package/package.json CHANGED
@@ -1,6 +1,6 @@
1
1
  {
2
2
  "name": "@maci-protocol/circuits",
3
- "version": "0.0.0-ci.4c6d4e8",
3
+ "version": "0.0.0-ci.52ce07e",
4
4
  "private": false,
5
5
  "description": "zk-SNARK circuits for MACI",
6
6
  "main": "build/ts/index.js",
@@ -18,7 +18,7 @@
18
18
  "scripts": {
19
19
  "build-test-circuits-c": "ts-node ./ts/compile.ts --cWitness",
20
20
  "build-test-circuits-wasm": "ts-node ./ts/compile.ts",
21
- "gen-zkeys": "ts-node ./ts/genZkeys.ts",
21
+ "generate-zkeys": "ts-node ./ts/generateZkeys.ts",
22
22
  "info": "NODE_OPTIONS=--max-old-space-size=4096 ts-node ./ts/info.ts",
23
23
  "watch": "tsc --watch",
24
24
  "build": "tsc -p tsconfig.build.json",
@@ -28,12 +28,15 @@
28
28
  "mocha-test": "NODE_OPTIONS=--max-old-space-size=4096 ts-mocha --exit -g '^(?!.*\\[fuzz\\]).*$'",
29
29
  "test": "pnpm run mocha-test ts/__tests__/*.test.ts",
30
30
  "test:fuzz": "NODE_OPTIONS=--max-old-space-size=4096 ts-mocha --exit -g '\\[fuzz\\]' ./ts/__tests__/*.test.ts",
31
- "test:hasher": "pnpm run mocha-test ts/__tests__/Hasher.test.ts",
31
+ "test:poseidonHasher": "pnpm run mocha-test ts/__tests__/PoseidonHasher.test.ts",
32
+ "test:messageHasher": "pnpm run mocha-test ts/__tests__/MessageHasher.test.ts",
32
33
  "test:slAndBallotTransformer": "pnpm run mocha-test ts/__tests__/StateLeafAndBallotTransformer.test.ts",
34
+ "test:slAndBallotTransformerFull": "pnpm run mocha-test ts/__tests__/StateLeafAndBallotTransformerFull.test.ts",
33
35
  "test:messageToCommand": "pnpm run mocha-test ts/__tests__/MessageToCommand.test.ts",
34
36
  "test:messageValidator": "pnpm run mocha-test ts/__tests__/MessageValidator.test.ts",
37
+ "test:messageValidatorFull": "pnpm run mocha-test ts/__tests__/MessageValidatorFull.test.ts",
35
38
  "test:verifySignature": "pnpm run mocha-test ts/__tests__/VerifySignature.test.ts",
36
- "test:privToPubKey": "pnpm run mocha-test ts/__tests__/PrivToPubKey.test.ts",
39
+ "test:privateToPublicKey": "pnpm run mocha-test ts/__tests__/PrivateToPublicKey.test.ts",
37
40
  "test:calculateTotal": "pnpm run mocha-test ts/__tests__/CalculateTotal.test.ts",
38
41
  "test:processMessages": "pnpm run mocha-test ts/__tests__/ProcessMessages.test.ts",
39
42
  "test:tallyVotes": "pnpm run mocha-test ts/__tests__/TallyVotes.test.ts",
@@ -43,10 +46,10 @@
43
46
  "test:pollJoined": "pnpm run mocha-test ts/__tests__/PollJoined.test.ts"
44
47
  },
45
48
  "dependencies": {
46
- "@maci-protocol/core": "0.0.0-ci.4c6d4e8",
47
- "@maci-protocol/crypto": "0.0.0-ci.4c6d4e8",
48
- "@maci-protocol/domainobjs": "0.0.0-ci.4c6d4e8",
49
- "@maci-protocol/sdk": "0.0.0-ci.4c6d4e8",
49
+ "@maci-protocol/core": "0.0.0-ci.52ce07e",
50
+ "@maci-protocol/crypto": "0.0.0-ci.52ce07e",
51
+ "@maci-protocol/domainobjs": "0.0.0-ci.52ce07e",
52
+ "@maci-protocol/sdk": "0.0.0-ci.52ce07e",
50
53
  "@zk-kit/circuits": "^0.4.0",
51
54
  "circomkit": "^0.3.2",
52
55
  "circomlib": "^2.0.5"
@@ -55,17 +58,17 @@
55
58
  "@types/chai": "^4.3.11",
56
59
  "@types/chai-as-promised": "^7.1.8",
57
60
  "@types/mocha": "^10.0.10",
58
- "@types/node": "^22.14.0",
61
+ "@types/node": "^22.15.17",
59
62
  "@types/snarkjs": "^0.7.9",
60
63
  "@zk-kit/baby-jubjub": "^1.0.3",
61
64
  "chai": "^4.3.10",
62
65
  "chai-as-promised": "^7.1.2",
63
66
  "fast-check": "^4.1.1",
64
- "glob": "^11.0.1",
65
- "mocha": "^11.1.0",
67
+ "glob": "^11.0.2",
68
+ "mocha": "^11.2.2",
66
69
  "ts-mocha": "^11.1.0",
67
70
  "ts-node": "^10.9.1",
68
71
  "typescript": "^5.8.3"
69
72
  },
70
- "gitHead": "314f41986d667b5968be17c3884459dddc7fd547"
73
+ "gitHead": "7bc383d813b0574ea9f0898cd32babac830cdc2b"
71
74
  }
@@ -1 +0,0 @@
1
- {"version":3,"file":"genZkeys.d.ts","sourceRoot":"","sources":["../../ts/genZkeys.ts"],"names":[],"mappings":"AAQA;;;;;;GAMG;AACH,eAAO,MAAM,aAAa,GAAU,aAAa,MAAM,KAAG,OAAO,CAAC,IAAI,CA2CrE,CAAC"}
@@ -1 +0,0 @@
1
- {"version":3,"file":"genZkeys.js","sourceRoot":"","sources":["../../ts/genZkeys.ts"],"names":[],"mappings":";;;;;;AAAA,4CAAkD;AAClD,yCAAgF;AAEhF,4CAAoB;AACpB,gDAAwB;AAIxB;;;;;;GAMG;AACI,MAAM,aAAa,GAAG,KAAK,EAAE,UAAmB,EAAiB,EAAE;IACxE,8BAA8B;IAC9B,MAAM,cAAc,GAAG,cAAI,CAAC,OAAO,CAAC,SAAS,EAAE,IAAI,EAAE,gBAAgB,CAAC,CAAC;IACvE,MAAM,eAAe,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,YAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC,cAAc,EAAE,OAAO,CAAC,CAAoB,CAAC;IAC3G,MAAM,kBAAkB,GAAG,cAAI,CAAC,OAAO,CAAC,SAAS,EAAE,IAAI,EAAE,QAAQ,EAAE,eAAe,CAAC,CAAC;IACpF,MAAM,qBAAqB,GAAG,IAAI,CAAC,KAAK,CACtC,MAAM,YAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC,kBAAkB,EAAE,OAAO,CAAC,CACZ,CAAC;IAC9C,MAAM,eAAe,GAA4B,MAAM,CAAC,OAAO,CAAC,qBAAqB,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,EAAE,MAAM,CAAC,EAAE,EAAE,CAAC,CAAC;QAC9G,IAAI;QACJ,GAAG,MAAM;KACV,CAAC,CAAC,CAAC;IAEJ,MAAM,OAAO,GAAG,UAAU,CAAC,CAAC,CAAC,cAAI,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IAClE,8BAA8B;IAC9B,IAAI,OAAO,EAAE,CAAC;QACZ,eAAe,CAAC,QAAQ,GAAG,OAAO,CAAC;QACnC,eAAe,CAAC,OAAO,GAAG,OAAO,CAAC;IACpC,CAAC;IAED,MAAM,iBAAiB,GAAG,IAAI,qBAAS,CAAC;QACtC,GAAG,eAAe;QAClB,OAAO,EAAE,KAAK;KACf,CAAC,CAAC;IAEH,oDAAoD;IACpD,4DAA4D;IAC5D,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,eAAe,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC;QACnD,MAAM,OAAO,GAAG,eAAe,CAAC,CAAC,CAAC,CAAC;QAEnC,sCAAsC;QACtC,OAAO,CAAC,GAAG,CAAC,uBAAuB,OAAO,CAAC,IAAI,KAAK,CAAC,CAAC;QAEtD,4CAA4C;QAC5C,MAAM,EAAE,aAAa,EAAE,GAAG,MAAM,iBAAiB,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QACtE,kBAAkB;QAClB,MAAM,QAAQ,GAAG,cAAI,CAAC,OAAO,CAAC,eAAe,CAAC,QAAQ,EAAE,OAAO,CAAC,IAAI,EAAE,GAAG,OAAO,CAAC,IAAI,SAAS,CAAC,CAAC;QAChG,4CAA4C;QAC5C,MAAM,YAAE,CAAC,QAAQ,CAAC,MAAM,CAAC,aAAa,EAAE,QAAQ,CAAC,CAAC;IACpD,CAAC;IAED,sCAAsC;IACtC,MAAM,IAAA,kBAAY,GAAE,CAAC;AACvB,CAAC,CAAC;AA3CW,QAAA,aAAa,iBA2CxB;AAEF,IAAI,OAAO,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;IAC5B,CAAC,KAAK,IAAI,EAAE;QACV,MAAM,eAAe,GAAG,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;QAC1D,IAAI,eAAe,KAAK,CAAC,CAAC,EAAE,CAAC;YAC3B,MAAM,IAAA,qBAAa,GAAE,CAAC;QACxB,CAAC;aAAM,CAAC;YACN,MAAM,YAAY,GAAG,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC;YACzE,MAAM,IAAA,qBAAa,EAAC,YAAY,CAAC,CAAC;QACpC,CAAC;IACH,CAAC,CAAC,EAAE,CAAC;AACP,CAAC"}
@@ -1,78 +0,0 @@
1
- pragma circom 2.0.0;
2
-
3
- // zk-kit imports
4
- include "./poseidon-cipher.circom";
5
-
6
- /**
7
- * Computes the Poseidon hash for an array of n inputs, including a default initial state
8
- * of zero not counted in n. First, extends the inputs by prepending a zero, creating an array [0, inputs].
9
- * Then, the Poseidon hash of the extended inputs is calculated, with the first element of the
10
- * result assigned as the output.
11
- */
12
- template PoseidonHasher(n) {
13
- signal input inputs[n];
14
- signal output out;
15
-
16
- // [0, inputs].
17
- var computedExtendedInputs[n + 1];
18
- computedExtendedInputs[0] = 0;
19
-
20
- for (var i = 0; i < n; i++) {
21
- computedExtendedInputs[i + 1] = inputs[i];
22
- }
23
-
24
- // Compute the Poseidon hash of the extended inputs.
25
- var computedPoseidonPerm[n + 1];
26
- computedPoseidonPerm = PoseidonPerm(n + 1)(computedExtendedInputs);
27
-
28
- out <== computedPoseidonPerm[0];
29
- }
30
-
31
- /**
32
- * Hashes a MACI message and the public key used for message encryption.
33
- * This template processes 10 message inputs and a 2-element public key
34
- * combining them using the Poseidon hash function. The hashing process involves two stages:
35
- * 1. hashing message parts in groups of five and,
36
- * 2. hashing the grouped results alongside the first message input and
37
- * the encryption public key to produce a final hash output.
38
- */
39
- template MessageHasher() {
40
- // The MACI message is composed of 10 parts.
41
- signal input in[10];
42
- // the public key used to encrypt the message.
43
- signal input encPubKey[2];
44
- // we output an hash.
45
- signal output hash;
46
-
47
- // Hasher4(
48
- // Hasher5_1(in[1], in[2], in[3], in[4], in[5]),
49
- // Hasher5_2(in[6], in[7], in[8], in[9], in[10])
50
- // in[11],
51
- // in[12]
52
- // )
53
-
54
- var computedHasher5_1;
55
- computedHasher5_1 = PoseidonHasher(5)([
56
- in[0],
57
- in[1],
58
- in[2],
59
- in[3],
60
- in[4]
61
- ]);
62
-
63
- var computedHasher5_2;
64
- computedHasher5_2 = PoseidonHasher(5)([
65
- in[5],
66
- in[6],
67
- in[7],
68
- in[8],
69
- in[9]
70
- ]);
71
-
72
- hash <== PoseidonHasher(4)([
73
- computedHasher5_1,
74
- computedHasher5_2,
75
- encPubKey[0],
76
- encPubKey[1]
77
- ]);
78
- }
@@ -1,78 +0,0 @@
1
- pragma circom 2.0.0;
2
-
3
- // circomlib import
4
- include "./bitify.circom";
5
- // zk-kit imports
6
- include "./ecdh.circom";
7
- include "./unpack-element.circom";
8
- include "./poseidon-cipher.circom";
9
- // local imports
10
- include "./hashers.circom";
11
-
12
- /**
13
- * Converts a MACI message to a command by decrypting it.
14
- * Processes encrypted MACI messages into structured MACI commands
15
- * by decrypting using a shared key derived from ECDH. After decryption,
16
- * unpacks and assigns decrypted values to specific command components.
17
- */
18
- template MessageToCommand() {
19
- var MSG_LENGTH = 7;
20
- var PACKED_CMD_LENGTH = 4;
21
- var UNPACKED_CMD_LENGTH = 8;
22
- var UNPACK_ELEM_LENGTH = 5;
23
- var DECRYPTED_LENGTH = 9;
24
- var MESSAGE_PARTS = 10;
25
-
26
- // The message is an array of 10 parts.
27
- signal input message[MESSAGE_PARTS];
28
- signal input encPrivKey;
29
- signal input encPubKey[2];
30
-
31
- // Command parts.
32
- signal output stateIndex;
33
- signal output newPubKey[2];
34
- signal output voteOptionIndex;
35
- signal output newVoteWeight;
36
- signal output nonce;
37
- signal output pollId;
38
- signal output salt;
39
- signal output sigR8[2];
40
- signal output sigS;
41
- // Packed command.
42
- signal output packedCommandOut[PACKED_CMD_LENGTH];
43
-
44
- // Generate the shared key for decrypting the message.
45
- var computedEcdh[2] = Ecdh()(encPrivKey, encPubKey);
46
-
47
- // Decrypt the message using Poseidon decryption.
48
- var computedDecryptor[DECRYPTED_LENGTH] = PoseidonDecryptWithoutCheck(MSG_LENGTH)(
49
- message,
50
- 0,
51
- computedEcdh
52
- );
53
-
54
- // Save the decrypted message into a packed command signal.
55
- signal packedCommand[PACKED_CMD_LENGTH];
56
- for (var i = 0; i < PACKED_CMD_LENGTH; i++) {
57
- packedCommand[i] <== computedDecryptor[i];
58
- }
59
-
60
- var computedUnpackElement[UNPACK_ELEM_LENGTH] = UnpackElement(UNPACK_ELEM_LENGTH)(packedCommand[0]);
61
-
62
- // Everything below were packed into the first element.
63
- stateIndex <== computedUnpackElement[4];
64
- voteOptionIndex <== computedUnpackElement[3];
65
- newVoteWeight <== computedUnpackElement[2];
66
- nonce <== computedUnpackElement[1];
67
- pollId <== computedUnpackElement[0];
68
-
69
- newPubKey[0] <== packedCommand[1];
70
- newPubKey[1] <== packedCommand[2];
71
- salt <== packedCommand[3];
72
-
73
- sigR8[0] <== computedDecryptor[4];
74
- sigR8[1] <== computedDecryptor[5];
75
- sigS <== computedDecryptor[6];
76
-
77
- packedCommandOut <== packedCommand;
78
- }
@@ -1,287 +0,0 @@
1
- pragma circom 2.0.0;
2
-
3
- // circomlib imports
4
- include "./bitify.circom";
5
- include "./mux1.circom";
6
- // zk-kit import
7
- include "./safe-comparators.circom";
8
- // local imports
9
- include "../calculateTotal.circom";
10
- include "../hashers.circom";
11
-
12
- // Incremental Quintary Merkle Tree (IQT) verification circuits.
13
- // Since each node contains 5 leaves, we are using PoseidonT6 for hashing them.
14
- //
15
- // nb. circom has some particularities which limit the code patterns we can use:
16
- // - You can only assign a value to a signal once.
17
- // - A component's input signal must only be wired to another component's output signal.
18
- // - Variables can store linear combinations, and can also be used for loops,
19
- // declaring sizes of things, and anything that is not related to inputs of a circuit.
20
- // - The compiler fails whenever you try to mix invalid elements.
21
- // - You can't use a signal as a list index.
22
-
23
- /**
24
- * Selects an item from a list based on the given index.
25
- * It verifies the index is within the valid range and then iterates over the inputs to find the match.
26
- * For each item, it checks if its position equals the given index and if so, multiplies the item
27
- * by the result of the equality check, effectively selecting it.
28
- * The sum of these results yields the selected item, ensuring only the item at the specified index be the output.
29
- *
30
- * nb. The number of items must be less than 8, and the index must be less than the number of items.
31
- */
32
- template QuinSelector(choices) {
33
- signal input in[choices];
34
- signal input index;
35
- signal output out;
36
-
37
- // Ensure that index < choices.
38
- var computedLtIndex = SafeLessThan(3)([index, choices]);
39
- computedLtIndex === 1;
40
-
41
- // Initialize an array to hold the results of equality checks.
42
- var computedResults[choices];
43
-
44
- // For each item, check whether its index equals the input index.
45
- // The result is multiplied by the corresponding input value.
46
- for (var i = 0; i < choices; i++) {
47
- var computedIsIndexEqual = IsEqual()([i, index]);
48
-
49
- computedResults[i] = computedIsIndexEqual * in[i];
50
- }
51
-
52
- // Calculate the total sum of the results array.
53
- out <== CalculateTotal(choices)(computedResults);
54
- }
55
-
56
- /**
57
- * The output array contains the input items, with the leaf inserted at the
58
- * specified index. For example, if input = [0, 20, 30, 40], index = 3, and
59
- * leaf = 10, the output will be [0, 20, 30, 10, 40].
60
- */
61
- template Splicer(numItems) {
62
- // The number of output items (because only one item is inserted).
63
- var NUM_OUTPUT_ITEMS = numItems + 1;
64
-
65
- signal input in[numItems];
66
- signal input leaf;
67
- signal input index;
68
- signal output out[NUM_OUTPUT_ITEMS];
69
-
70
- // There is a loop where the goal is to assign values to the output signal.
71
- //
72
- // | output[0] | output[1] | output[2] | ...
73
- //
74
- // We can either assign the leaf, or an item from the `items` signal, to the output, using Mux1().
75
- // The Mux1's selector is 0 or 1 depending on whether the index is equal to the loop counter.
76
- //
77
- // i --> [IsEqual] <-- index
78
- // |
79
- // v
80
- // leaf --> [Mux1] <-- <item from in>
81
- // |
82
- // v
83
- // output[m]
84
- //
85
- // To obtain the value from <item from in>, we need to compute an item
86
- // index (let it be `s`).
87
- // 1. if index = 2 and i = 0, then s = 0
88
- // 2. if index = 2 and i = 1, then s = 1
89
- // 3. if index = 2 and i = 2, then s = 2
90
- // 4. if index = 2 and i = 3, then s = 2
91
- // 5. if index = 2 and i = 4, then s = 3
92
- // We then wire `s`, as well as each item in `in` to a QuinSelector.
93
- // The output signal from the QuinSelector is <item from in> and gets
94
- // wired to Mux1 (as above).
95
-
96
- var inputs[NUM_OUTPUT_ITEMS];
97
-
98
- for (var i = 0; i < numItems; i++) {
99
- inputs[i] = in[i];
100
- }
101
- inputs[NUM_OUTPUT_ITEMS - 1] = 0;
102
-
103
- for (var i = 0; i < NUM_OUTPUT_ITEMS; i++) {
104
- // Determines if current index is greater than the insertion index.
105
- var computedIsIndexAfterInsertPoint = SafeGreaterThan(3)([i, index]);
106
-
107
- // Calculates correct index for original items, adjusting for leaf insertion.
108
- var computedAdjustedIndex = i - computedIsIndexAfterInsertPoint;
109
-
110
- // Selects item from the original array or the leaf for insertion.
111
- var computedQuinSelected = QuinSelector(NUM_OUTPUT_ITEMS)(inputs, computedAdjustedIndex);
112
- var computedIsIndexEqual = IsEqual()([index, i]);
113
- var mux = Mux1()([computedQuinSelected, leaf], computedIsIndexEqual);
114
-
115
- out[i] <== mux;
116
- }
117
- }
118
-
119
- /**
120
- * Computes the root of an IQT given a leaf, its path, and sibling nodes at each level of the tree.
121
- * It iteratively incorporates the leaf or the hash from the previous level with sibling nodes using
122
- * the Splicer to place the leaf or hash at the correct position based on path_index.
123
- * Then, it hashes these values together with PoseidonHasher to move up the tree.
124
- * This process repeats for each level (levels) of the tree, culminating in the computation of the tree's root.
125
- */
126
- template QuinTreeInclusionProof(levels) {
127
- var LEAVES_PER_NODE = 5;
128
- var LEAVES_PER_PATH_LEVEL = LEAVES_PER_NODE - 1;
129
-
130
- signal input leaf;
131
- signal input path_index[levels];
132
- signal input path_elements[levels][LEAVES_PER_PATH_LEVEL];
133
- signal output root;
134
-
135
- var currentLeaf = leaf;
136
-
137
- // Iteratively hash each level of path_elements with the leaf or previous hash
138
- for (var i = 0; i < levels; i++) {
139
- var elements[LEAVES_PER_PATH_LEVEL];
140
-
141
- for (var j = 0; j < LEAVES_PER_PATH_LEVEL; j++) {
142
- elements[j] = path_elements[i][j];
143
- }
144
-
145
- var computedSplicedLeaf[LEAVES_PER_NODE] = Splicer(LEAVES_PER_PATH_LEVEL)(
146
- elements,
147
- currentLeaf,
148
- path_index[i]
149
- );
150
-
151
- currentLeaf = PoseidonHasher(5)([
152
- computedSplicedLeaf[0],
153
- computedSplicedLeaf[1],
154
- computedSplicedLeaf[2],
155
- computedSplicedLeaf[3],
156
- computedSplicedLeaf[4]
157
- ]);
158
- }
159
-
160
- root <== currentLeaf;
161
- }
162
-
163
- /**
164
- * Verifies if a given leaf exists within an IQT.
165
- * Takes a leaf, its path to the root (specified by indices and path elements),
166
- * and the root itself, to verify the leaf's inclusion within the tree.
167
- */
168
- template QuinLeafExists(levels){
169
- var LEAVES_PER_NODE = 5;
170
- var LEAVES_PER_PATH_LEVEL = LEAVES_PER_NODE - 1;
171
-
172
- signal input leaf;
173
- signal input path_index[levels];
174
- signal input path_elements[levels][LEAVES_PER_PATH_LEVEL];
175
- signal input root;
176
-
177
- // Verify the Merkle path.
178
- var computedRoot = QuinTreeInclusionProof(levels)(leaf, path_index, path_elements);
179
-
180
- root === computedRoot;
181
- }
182
-
183
- /**
184
- * Checks if a list of leaves exists within an IQT, leveraging the PoseidonT6
185
- * circuit for hashing. This can be used to verify the presence of multiple leaves.
186
- */
187
- template QuinBatchLeavesExists(levels, batchLevels) {
188
- var LEAVES_PER_NODE = 5;
189
- var LEAVES_PER_PATH_LEVEL = LEAVES_PER_NODE - 1;
190
- var LEAVES_PER_BATCH = LEAVES_PER_NODE ** batchLevels;
191
-
192
- signal input root;
193
- signal input leaves[LEAVES_PER_BATCH];
194
- signal input path_index[levels - batchLevels];
195
- signal input path_elements[levels - batchLevels][LEAVES_PER_PATH_LEVEL];
196
-
197
- // Compute the subroot (= leaf).
198
- var computedQuinSubroot = QuinCheckRoot(batchLevels)(leaves);
199
-
200
- // Check if the Merkle path is valid
201
- QuinLeafExists(levels - batchLevels)(computedQuinSubroot, path_index, path_elements, root);
202
- }
203
-
204
- /**
205
- * Calculates the path indices required for Merkle proof verifications (e.g., QuinTreeInclusionProof, QuinLeafExists).
206
- * Given a node index within an IQT and the total tree levels, it outputs the path indices leading to that node.
207
- * The template handles the modulo and division operations to break down the tree index into its constituent path indices.
208
- * e.g., if the index is 30 and the number of levels is 4, the output should be [0, 1, 1, 0].
209
- */
210
- template QuinGeneratePathIndices(levels) {
211
- var BASE = 5;
212
-
213
- signal input in;
214
- signal output out[levels];
215
-
216
- var m = in;
217
- var computedResults[levels];
218
-
219
- for (var i = 0; i < levels; i++) {
220
- // circom's best practices suggests to avoid using <-- unless you
221
- // are aware of what's going on. This is the only way to do modulo operation.
222
- out[i] <-- m % BASE;
223
- m = m \ BASE;
224
-
225
- // Check that each output element is less than the base.
226
- var computedIsOutputElementLessThanBase = SafeLessThan(3)([out[i], BASE]);
227
- computedIsOutputElementLessThanBase === 1;
228
-
229
- // Re-compute the total sum.
230
- computedResults[i] = out[i] * (BASE ** i);
231
- }
232
-
233
- // Check that the total sum matches the index.
234
- var computedCalculateTotal = CalculateTotal(levels)(computedResults);
235
-
236
- computedCalculateTotal === in;
237
- }
238
-
239
- /**
240
- * Computes the root of a quintary Merkle tree given a list of leaves.
241
- * This template constructs a Merkle tree with each node having 5 children (quintary)
242
- * and computes the root by hashing with Poseidon the leaves and intermediate nodes in the given order.
243
- * The computation is performed by first hashing groups of 5 leaves to form the bottom layer of nodes,
244
- * then recursively hashing groups of these nodes to form the next layer, and so on, until the root is computed.
245
- */
246
- template QuinCheckRoot(levels) {
247
- var LEAVES_PER_NODE = 5;
248
- var totalLeaves = LEAVES_PER_NODE ** levels;
249
- var numLeafHashers = LEAVES_PER_NODE ** (levels - 1);
250
-
251
- signal input leaves[totalLeaves];
252
- signal output root;
253
-
254
- // Determine the total number of hashers.
255
- var numHashers = 0;
256
- for (var i = 0; i < levels; i++) {
257
- numHashers += LEAVES_PER_NODE ** i;
258
- }
259
-
260
- var computedHashers[numHashers];
261
-
262
- // Initialize hashers for the leaves.
263
- for (var i = 0; i < numLeafHashers; i++) {
264
- computedHashers[i] = PoseidonHasher(5)([
265
- leaves[i * LEAVES_PER_NODE + 0],
266
- leaves[i * LEAVES_PER_NODE + 1],
267
- leaves[i * LEAVES_PER_NODE + 2],
268
- leaves[i * LEAVES_PER_NODE + 3],
269
- leaves[i * LEAVES_PER_NODE + 4]
270
- ]);
271
- }
272
-
273
- // Initialize hashers for intermediate nodes and compute the root.
274
- var k = 0;
275
- for (var i = numLeafHashers; i < numHashers; i++) {
276
- computedHashers[i] = PoseidonHasher(5)([
277
- computedHashers[k * LEAVES_PER_NODE + 0],
278
- computedHashers[k * LEAVES_PER_NODE + 1],
279
- computedHashers[k * LEAVES_PER_NODE + 2],
280
- computedHashers[k * LEAVES_PER_NODE + 3],
281
- computedHashers[k * LEAVES_PER_NODE + 4]
282
- ]);
283
- k++;
284
- }
285
-
286
- root <== computedHashers[numHashers - 1];
287
- }