@maci-protocol/circuits 0.0.0-ci.2d2f5fb → 0.0.0-ci.2d57ba7

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 (33) hide show
  1. package/README.md +2 -2
  2. package/build/ts/types.d.ts +4 -4
  3. package/build/ts/types.d.ts.map +1 -1
  4. package/build/tsconfig.build.tsbuildinfo +1 -1
  5. package/circom/circuits.json +28 -12
  6. package/circom/coordinator/full/MessageProcessor.circom +253 -0
  7. package/circom/coordinator/full/SingleMessageProcessor.circom +203 -0
  8. package/circom/coordinator/non-qv/{processMessages.circom → MessageProcessor.circom} +12 -206
  9. package/circom/coordinator/non-qv/SingleMessageProcessor.circom +199 -0
  10. package/circom/coordinator/non-qv/{tallyVotes.circom → VoteTally.circom} +18 -95
  11. package/circom/coordinator/qv/{processMessages.circom → MessageProcessor.circom} +13 -214
  12. package/circom/coordinator/qv/SingleMessageProcessor.circom +207 -0
  13. package/circom/coordinator/qv/{tallyVotes.circom → VoteTally.circom} +14 -114
  14. package/circom/coordinator/qv/VoteTallyWithIndividualCounts.circom +226 -0
  15. package/circom/utils/CalculateTotal.circom +6 -6
  16. package/circom/utils/full/MessageValidator.circom +2 -4
  17. package/circom/utils/non-qv/MessageValidator.circom +2 -2
  18. package/circom/utils/non-qv/ResultCommitmentVerifier.circom +84 -0
  19. package/circom/utils/qv/MessageValidator.circom +2 -2
  20. package/circom/utils/qv/ResultCommitmentVerifier.circom +107 -0
  21. package/circom/utils/trees/BinaryMerkleRoot.circom +6 -3
  22. package/circom/utils/trees/LeafExists.circom +2 -2
  23. package/circom/utils/trees/MerkleTreeInclusionProof.circom +4 -4
  24. package/circom/utils/trees/QuinaryCheckRoot.circom +54 -0
  25. package/circom/utils/trees/{MerklePathIndicesGenerator.circom → QuinaryGeneratePathIndices.circom} +18 -18
  26. package/circom/utils/trees/QuinaryLeafExists.circom +30 -0
  27. package/circom/utils/trees/QuinarySelector.circom +42 -0
  28. package/circom/utils/trees/QuinaryTreeInclusionProof.circom +55 -0
  29. package/circom/utils/trees/Splicer.circom +76 -0
  30. package/circom/voter/PollJoined.circom +3 -3
  31. package/circom/voter/PollJoining.circom +3 -3
  32. package/package.json +16 -16
  33. package/circom/utils/trees/incrementalQuinaryTree.circom +0 -287
@@ -0,0 +1,226 @@
1
+ pragma circom 2.0.0;
2
+
3
+ // circomlib import
4
+ include "./comparators.circom";
5
+ // zk-kit import
6
+ include "./unpack-element.circom";
7
+ // local imports
8
+ include "../../utils/trees/CheckRoot.circom";
9
+ include "../../utils/trees/LeafExists.circom";
10
+ include "../../utils/trees/QuinaryCheckRoot.circom";
11
+ include "../../utils/qv/ResultCommitmentVerifier.circom";
12
+ include "../../utils/CalculateTotal.circom";
13
+ include "../../utils/PoseidonHasher.circom";
14
+
15
+ /**
16
+ * Processes batches of votes and verifies their validity in a Merkle tree structure.
17
+ * This template supports Quadratic Voting (QV) with individual vote counting.
18
+ * Note: this circuit is not using right now, this is a part of individual vote counts functionality.
19
+ * Not finished yet, don't use it in production. It is kept here for future use.
20
+ */
21
+ template VoteTallyWithIndividualCountsQv(
22
+ stateTreeDepth,
23
+ tallyProcessingStateTreeDepth,
24
+ voteOptionTreeDepth
25
+ ) {
26
+ // Ensure there's at least one level in the vote option tree.
27
+ assert(voteOptionTreeDepth > 0);
28
+ // Ensure the intermediate state tree has at least one level.
29
+ assert(tallyProcessingStateTreeDepth > 0);
30
+ // The intermediate state tree must be smaller than the full state tree.
31
+ assert(tallyProcessingStateTreeDepth < stateTreeDepth);
32
+
33
+ // Number of children per node in the tree, defining the tree's branching factor.
34
+ var TREE_ARITY = 5;
35
+ var BALLOT_TREE_ARITY = 2;
36
+ var VOTE_COUNTS_TREE_ARITY = 2;
37
+
38
+ // The number of ballots processed at once, determined by the depth of the intermediate state tree.
39
+ var ballotBatchSize = BALLOT_TREE_ARITY ** tallyProcessingStateTreeDepth;
40
+ var voteCountsBatchSize = VOTE_COUNTS_TREE_ARITY ** tallyProcessingStateTreeDepth;
41
+ // Number of voting options available, determined by the depth of the vote option tree.
42
+ var totalVoteOptions = TREE_ARITY ** voteOptionTreeDepth;
43
+
44
+ // Number of elements in each ballot.
45
+ var BALLOT_LENGTH = 2;
46
+ // Index for the nonce in the ballot array.
47
+ var BALLOT_NONCE_INDEX = 0;
48
+ // Index for the voting option root in the ballot array.
49
+ var BALLOT_VOTE_OPTION_ROOT_INDEX = 1;
50
+ // Difference in tree depths, used in path calculations.
51
+ var STATE_TREE_DEPTH_DIFFERENCE = stateTreeDepth - tallyProcessingStateTreeDepth;
52
+ // Number of elements in each vote count leaf.
53
+ var VOTE_COUNTS_LENGTH = 2;
54
+ // Index for the voting option index.
55
+ var VOTE_COUNTS_NONCE_INDEX = 0;
56
+ // Index for root of the vote count per option.
57
+ var VOTE_COUNTS_ROOT_INDEX = 1;
58
+
59
+ // Root of the state Merkle tree, representing the overall state before voting.
60
+ signal input stateRoot;
61
+ // Root of the ballot Merkle tree, representing the submitted ballots.
62
+ signal input ballotRoot;
63
+ // Root of the vote counts Merkle tree, representing the counts of votes for each option.
64
+ signal input voteCountsRoot;
65
+ // Salt used in commitment to secure the ballot data.
66
+ signal input sbSalt;
67
+ // Commitment to the state and ballots.
68
+ signal input sbCommitment;
69
+ // Commitment to the current tally before this batch.
70
+ signal input currentTallyCommitment;
71
+ // Commitment to the new tally after processing this batch.
72
+ signal input newTallyCommitment;
73
+ // Start index of given batch
74
+ signal input index;
75
+ // Number of users that signup
76
+ signal input totalSignups;
77
+ // Ballots and their corresponding path elements for verification in the tree.
78
+ signal input ballots[ballotBatchSize][BALLOT_LENGTH];
79
+ signal input ballotPathElements[STATE_TREE_DEPTH_DIFFERENCE][BALLOT_TREE_ARITY - 1];
80
+ signal input votes[ballotBatchSize][totalVoteOptions];
81
+ // Individual vote count tree and their corresponding path elements for verification in the tree.
82
+ signal input voteCounts[voteCountsBatchSize][VOTE_COUNTS_LENGTH];
83
+ signal input voteCountsPathElements[STATE_TREE_DEPTH_DIFFERENCE][VOTE_COUNTS_TREE_ARITY - 1];
84
+ signal input voteCountsData[voteCountsBatchSize][totalVoteOptions];
85
+ // Current results for each vote option.
86
+ signal input currentResults[totalVoteOptions];
87
+ // Salt for the root of the current results.
88
+ signal input currentResultsRootSalt;
89
+ // Total voice credits spent so far.
90
+ signal input currentSpentVoiceCreditSubtotal;
91
+ // Salt for the total spent voice credits.
92
+ signal input currentSpentVoiceCreditSubtotalSalt;
93
+ // Spent voice credits per vote option.
94
+ signal input currentPerVoteOptionSpentVoiceCredits[totalVoteOptions];
95
+ // Salt for the root of spent credits per option.
96
+ signal input currentPerVoteOptionSpentVoiceCreditsRootSalt;
97
+ // Salt for the root of the new results.
98
+ signal input newResultsRootSalt;
99
+ // Salt for the new spent credits per vote option root.
100
+ signal input newPerVoteOptionSpentVoiceCreditsRootSalt;
101
+ // Salt for the new total spent voice credits root.
102
+ signal input newSpentVoiceCreditSubtotalSalt;
103
+
104
+ // Verify sbCommitment.
105
+ var computedSbCommitment = PoseidonHasher(3)([stateRoot, ballotRoot, sbSalt]);
106
+ computedSbCommitment === sbCommitment;
107
+
108
+ // Validates that the index is within the valid range of sign-ups.
109
+ var totalSignupsValid = LessEqThan(50)([index, totalSignups]);
110
+ totalSignupsValid === 1;
111
+
112
+ // Hashes each ballot for subroot generation, and checks the existence of the leaf in the Merkle tree.
113
+ var computedBallotHashers[ballotBatchSize];
114
+ var computedVoteCountsHashers[voteCountsBatchSize];
115
+
116
+ for (var i = 0; i < ballotBatchSize; i++) {
117
+ computedBallotHashers[i] = PoseidonHasher(2)([
118
+ ballots[i][BALLOT_NONCE_INDEX],
119
+ ballots[i][BALLOT_VOTE_OPTION_ROOT_INDEX]
120
+ ]);
121
+ }
122
+
123
+ for (var i = 0; i < voteCountsBatchSize; i++) {
124
+ computedVoteCountsHashers[i] = PoseidonHasher(2)([
125
+ voteCounts[i][VOTE_COUNTS_NONCE_INDEX],
126
+ voteCounts[i][VOTE_COUNTS_ROOT_INDEX]
127
+ ]);
128
+ }
129
+
130
+ var computedBallotSubroot = CheckRoot(tallyProcessingStateTreeDepth)(computedBallotHashers);
131
+ var computedBallotPathIndices[STATE_TREE_DEPTH_DIFFERENCE] = Num2Bits(STATE_TREE_DEPTH_DIFFERENCE)(index / ballotBatchSize);
132
+
133
+ var computedVoteCountsSubroot = CheckRoot(tallyProcessingStateTreeDepth)(computedVoteCountsHashers);
134
+ var computedVoteCountsPathIndices[STATE_TREE_DEPTH_DIFFERENCE] = Num2Bits(STATE_TREE_DEPTH_DIFFERENCE)(index / voteCountsBatchSize);
135
+
136
+ // Verifies each ballot's existence within the ballot tree.
137
+ LeafExists(STATE_TREE_DEPTH_DIFFERENCE)(
138
+ computedBallotSubroot,
139
+ ballotPathElements,
140
+ computedBallotPathIndices,
141
+ ballotRoot
142
+ );
143
+
144
+ // Verifies each vote count's existence within the vote count tree.
145
+ LeafExists(STATE_TREE_DEPTH_DIFFERENCE)(
146
+ computedVoteCountsSubroot,
147
+ voteCountsPathElements,
148
+ computedVoteCountsPathIndices,
149
+ voteCountsRoot
150
+ );
151
+
152
+ // Processes vote options, verifying each against its declared root.
153
+ var computedVoteTree[ballotBatchSize];
154
+
155
+ for (var i = 0; i < ballotBatchSize; i++) {
156
+ computedVoteTree[i] = QuinaryCheckRoot(voteOptionTreeDepth)(votes[i]);
157
+ computedVoteTree[i] === ballots[i][BALLOT_VOTE_OPTION_ROOT_INDEX];
158
+ }
159
+
160
+ // Processes vote counts, verifying each against its declared root.
161
+ var computedVoteCountsTree[voteCountsBatchSize];
162
+
163
+ for (var i = 0; i < voteCountsBatchSize; i++) {
164
+ computedVoteCountsTree[i] = QuinaryCheckRoot(voteOptionTreeDepth)(voteCountsData[i]);
165
+ computedVoteCountsTree[i] === voteCounts[i][VOTE_COUNTS_ROOT_INDEX];
166
+ }
167
+
168
+ // Calculates new results and spent voice credits based on the current and incoming votes.
169
+ var computedIsFirstBatch = IsZero()(index);
170
+ var computedIsZero = IsZero()(computedIsFirstBatch);
171
+
172
+ // Tally the new results.
173
+ var computedCalculateTotalResult[totalVoteOptions];
174
+ for (var i = 0; i < totalVoteOptions; i++) {
175
+ var numsRC[ballotBatchSize + 1];
176
+ numsRC[ballotBatchSize] = currentResults[i] * computedIsZero;
177
+ for (var j = 0; j < ballotBatchSize; j++) {
178
+ numsRC[j] = votes[j][i];
179
+ }
180
+
181
+ computedCalculateTotalResult[i] = CalculateTotal(ballotBatchSize + 1)(numsRC);
182
+ }
183
+
184
+ // Tally the new spent voice credit total.
185
+ var numsSVC[ballotBatchSize * totalVoteOptions + 1];
186
+ numsSVC[ballotBatchSize * totalVoteOptions] = currentSpentVoiceCreditSubtotal * computedIsZero;
187
+ for (var i = 0; i < ballotBatchSize; i++) {
188
+ for (var j = 0; j < totalVoteOptions; j++) {
189
+ numsSVC[i * totalVoteOptions + j] = votes[i][j] * votes[i][j];
190
+ }
191
+ }
192
+
193
+ var computedNewSpentVoiceCreditSubtotal = CalculateTotal(ballotBatchSize * totalVoteOptions + 1)(numsSVC);
194
+
195
+ // Tally the spent voice credits per vote option.
196
+ var computedNewPerVOSpentVoiceCredits[totalVoteOptions];
197
+
198
+ for (var i = 0; i < totalVoteOptions; i++) {
199
+ var computedTotalVoiceCreditSpent[ballotBatchSize + 1];
200
+ computedTotalVoiceCreditSpent[ballotBatchSize] = currentPerVoteOptionSpentVoiceCredits[i] * computedIsZero;
201
+ for (var j = 0; j < ballotBatchSize; j++) {
202
+ computedTotalVoiceCreditSpent[j] = votes[j][i] * votes[j][i];
203
+ }
204
+
205
+ computedNewPerVOSpentVoiceCredits[i] = CalculateTotal(ballotBatchSize + 1)(computedTotalVoiceCreditSpent);
206
+ }
207
+
208
+ // Verifies the updated results and spent credits, ensuring consistency and correctness of tally updates.
209
+ ResultCommitmentVerifierQv(voteOptionTreeDepth)(
210
+ computedIsFirstBatch,
211
+ currentTallyCommitment,
212
+ newTallyCommitment,
213
+ currentResults,
214
+ currentResultsRootSalt,
215
+ computedCalculateTotalResult,
216
+ newResultsRootSalt,
217
+ currentSpentVoiceCreditSubtotal,
218
+ currentSpentVoiceCreditSubtotalSalt,
219
+ computedNewSpentVoiceCreditSubtotal,
220
+ newSpentVoiceCreditSubtotalSalt,
221
+ currentPerVoteOptionSpentVoiceCredits,
222
+ currentPerVoteOptionSpentVoiceCreditsRootSalt,
223
+ computedNewPerVOSpentVoiceCredits,
224
+ newPerVoteOptionSpentVoiceCreditsRootSalt
225
+ );
226
+ }
@@ -1,24 +1,24 @@
1
1
  pragma circom 2.0.0;
2
2
 
3
3
  /**
4
- * Computes the cumulative sum of an array of n input signals.
4
+ * Computes the cumulative sum of an array of length input signals.
5
5
  * It iterates through each input, aggregating the sum up to that point,
6
6
  * and outputs the total sum of all inputs. This template is useful for
7
7
  * operations requiring the total sum of multiple signals, ensuring the
8
8
  * final output reflects the cumulative total of the inputs provided.
9
9
  */
10
- template CalculateTotal(n) {
10
+ template CalculateTotal(length) {
11
11
  // Array of values.
12
- signal input nums[n];
12
+ signal input nums[length];
13
13
  // Total sum.
14
14
  signal output sum;
15
15
 
16
- signal sums[n];
16
+ signal sums[length];
17
17
  sums[0] <== nums[0];
18
18
 
19
- for (var i = 1; i < n; i++) {
19
+ for (var i = 1; i < length; i++) {
20
20
  sums[i] <== sums[i - 1] + nums[i];
21
21
  }
22
22
 
23
- sum <== sums[n - 1];
23
+ sum <== sums[length - 1];
24
24
  }
@@ -1,7 +1,5 @@
1
1
  pragma circom 2.0.0;
2
2
 
3
- // circomlib import
4
- include "./mux1.circom";
5
3
  // zk-kit imports
6
4
  include "./safe-comparators.circom";
7
5
  // local imports
@@ -28,7 +26,7 @@ template MessageValidatorFull() {
28
26
  // Ballot nonce.
29
27
  signal input originalNonce;
30
28
  // Command nonce.
31
- signal input nonce;
29
+ signal input commandNonce;
32
30
  // Packed command.
33
31
  signal input command[PACKED_COMMAND_LENGTH];
34
32
  // Public key of the state leaf (user).
@@ -61,7 +59,7 @@ template MessageValidatorFull() {
61
59
  var computedIsVoteOptionIndexValid = SafeLessThan(252)([voteOptionIndex, voteOptions]);
62
60
 
63
61
  // Check (3) - The nonce must be correct.
64
- var computedIsNonceValid = IsEqual()([originalNonce + 1, nonce]);
62
+ var computedIsNonceValid = IsEqual()([originalNonce + 1, commandNonce]);
65
63
 
66
64
  // Check (4) - The signature must be correct.
67
65
  var computedIsSignatureValid = VerifySignature()(publicKey, signaturePoint, signatureScalar, command);
@@ -26,7 +26,7 @@ template MessageValidatorNonQv() {
26
26
  // Ballot nonce.
27
27
  signal input originalNonce;
28
28
  // Command nonce.
29
- signal input nonce;
29
+ signal input commandNonce;
30
30
  // Packed command.
31
31
  signal input command[PACKED_COMMAND_LENGTH];
32
32
  // Public key of the state leaf (user).
@@ -59,7 +59,7 @@ template MessageValidatorNonQv() {
59
59
  var computedIsVoteOptionIndexValid = SafeLessThan(252)([voteOptionIndex, voteOptions]);
60
60
 
61
61
  // Check (3) - The nonce must be correct.
62
- var computedIsNonceValid = IsEqual()([originalNonce + 1, nonce]);
62
+ var computedIsNonceValid = IsEqual()([originalNonce + 1, commandNonce]);
63
63
 
64
64
  // Check (4) - The signature must be correct.
65
65
  var computedIsSignatureValid = VerifySignature()(publicKey, signaturePoint, signatureScalar, command);
@@ -0,0 +1,84 @@
1
+ pragma circom 2.0.0;
2
+
3
+ // circomlib import
4
+ include "./comparators.circom";
5
+ // local imports
6
+ include "../trees/QuinaryCheckRoot.circom";
7
+ include "../PoseidonHasher.circom";
8
+
9
+ /**
10
+ * Performs verifications and computations related to current voting results.
11
+ * Also, computes and outputs a commitment to the new results.
12
+ * This template does not support Quadratic Voting (QV).
13
+ */
14
+ template ResultCommitmentVerifierNonQv(voteOptionTreeDepth) {
15
+ // Number of children per node in the tree, defining the tree's branching factor.
16
+ var TREE_ARITY = 5;
17
+ // Number of voting options available, determined by the depth of the vote option tree.
18
+ var totalVoteOptions = TREE_ARITY ** voteOptionTreeDepth;
19
+
20
+ // Equal to 1 if this is the first batch, otherwise 0.
21
+ signal input isFirstBatch;
22
+ // Commitment to the current tally before this batch.
23
+ signal input currentTallyCommitment;
24
+ // Commitment to the new tally after processing this batch.
25
+ signal input newTallyCommitment;
26
+
27
+ // Current results for each vote option.
28
+ signal input currentResults[totalVoteOptions];
29
+ // Salt for the root of the current results.
30
+ signal input currentResultsRootSalt;
31
+
32
+ // New results for each vote option.
33
+ signal input newResults[totalVoteOptions];
34
+ // Salt for the root of the new results.
35
+ signal input newResultsRootSalt;
36
+
37
+ // Total voice credits spent so far.
38
+ signal input currentSpentVoiceCreditSubtotal;
39
+ // Salt for the total spent voice credits.
40
+ signal input currentSpentVoiceCreditSubtotalSalt;
41
+
42
+ // Total new voice credits spent.
43
+ signal input newSpentVoiceCreditSubtotal;
44
+ // Salt for the new total spent voice credits.
45
+ signal input newSpentVoiceCreditSubtotalSalt;
46
+
47
+ // Compute the commitment to the current results.
48
+ var computedCurrentResultsRoot = QuinaryCheckRoot(voteOptionTreeDepth)(currentResults);
49
+
50
+ // Verify currentResultsCommitmentHash.
51
+ var computedCurrentResultsCommitment = PoseidonHasher(2)([computedCurrentResultsRoot, currentResultsRootSalt]);
52
+
53
+ // Compute the commitment to the current spent voice credits.
54
+ var computedCurrentSpentVoiceCreditsCommitment = PoseidonHasher(2)([currentSpentVoiceCreditSubtotal, currentSpentVoiceCreditSubtotalSalt]);
55
+
56
+ // Commit to the current tally
57
+ var computedCurrentTallyCommitment = PoseidonHasher(2)([computedCurrentResultsCommitment, computedCurrentSpentVoiceCreditsCommitment]);
58
+
59
+ // Check if the current tally commitment is correct only if this is not the first batch.
60
+ // computedIsZero.out is 1 if this is not the first batch.
61
+ // computedIsZero.out is 0 if this is the first batch.
62
+ var computedIsZero = IsZero()(isFirstBatch);
63
+
64
+ // isFirstCommitment is 0 if this is the first batch, currentTallyCommitment should be 0 if this is the first batch.
65
+ // isFirstCommitment is 1 if this is not the first batch, currentTallyCommitment should not be 0 if this is the first batch.
66
+ signal isFirstCommitment;
67
+ isFirstCommitment <== computedIsZero * computedCurrentTallyCommitment;
68
+ isFirstCommitment === currentTallyCommitment;
69
+
70
+ // Compute the root of the new results.
71
+ var computedNewResultsRoot = QuinaryCheckRoot(voteOptionTreeDepth)(newResults);
72
+ var computedNewResultsCommitment = PoseidonHasher(2)([computedNewResultsRoot, newResultsRootSalt]);
73
+
74
+ // Compute the commitment to the new spent voice credits value.
75
+ var computedNewSpentVoiceCreditsCommitment = PoseidonHasher(2)([newSpentVoiceCreditSubtotal, newSpentVoiceCreditSubtotalSalt]);
76
+
77
+ // Commit to the new tally.
78
+ var computedNewTallyCommitment = PoseidonHasher(2)([
79
+ computedNewResultsCommitment,
80
+ computedNewSpentVoiceCreditsCommitment
81
+ ]);
82
+
83
+ computedNewTallyCommitment === newTallyCommitment;
84
+ }
@@ -26,7 +26,7 @@ template MessageValidator() {
26
26
  // Ballot nonce.
27
27
  signal input originalNonce;
28
28
  // Command nonce.
29
- signal input nonce;
29
+ signal input commandNonce;
30
30
  // Packed command.
31
31
  signal input command[PACKED_COMMAND_LENGTH];
32
32
  // Public key of the state leaf (user).
@@ -59,7 +59,7 @@ template MessageValidator() {
59
59
  var computedIsVoteOptionIndexValid = SafeLessThan(252)([voteOptionIndex, voteOptions]);
60
60
 
61
61
  // Check (3) - The nonce must be correct.
62
- var computedIsNonceValid = IsEqual()([originalNonce + 1, nonce]);
62
+ var computedIsNonceValid = IsEqual()([originalNonce + 1, commandNonce]);
63
63
 
64
64
  // Check (4) - The signature must be correct.
65
65
  var computedIsSignatureValid = VerifySignature()(publicKey, signaturePoint, signatureScalar, command);
@@ -0,0 +1,107 @@
1
+ pragma circom 2.0.0;
2
+
3
+ // circomlib import
4
+ include "./comparators.circom";
5
+ // local imports
6
+ include "../trees/QuinaryCheckRoot.circom";
7
+ include "../PoseidonHasher.circom";
8
+
9
+ /**
10
+ * Performs verifications and computations related to current voting results.
11
+ * Also, computes and outputs a commitment to the new results.
12
+ * This template supports the Quadratic Voting (QV).
13
+ */
14
+ template ResultCommitmentVerifierQv(voteOptionTreeDepth) {
15
+ // Number of children per node in the tree, defining the tree's branching factor.
16
+ var TREE_ARITY = 5;
17
+ // Number of voting options available, determined by the depth of the vote option tree.
18
+ var totalVoteOptions = TREE_ARITY ** voteOptionTreeDepth;
19
+
20
+ // Equal to 1 if this is the first batch, otherwise 0.
21
+ signal input isFirstBatch;
22
+ // Commitment to the current tally before this batch.
23
+ signal input currentTallyCommitment;
24
+ // Commitment to the new tally after processing this batch.
25
+ signal input newTallyCommitment;
26
+
27
+ // Current results for each vote option.
28
+ signal input currentResults[totalVoteOptions];
29
+ // Salt for the root of the current results.
30
+ signal input currentResultsRootSalt;
31
+
32
+ // New results for each vote option.
33
+ signal input newResults[totalVoteOptions];
34
+ // Salt for the root of the new results.
35
+ signal input newResultsRootSalt;
36
+
37
+ // Total voice credits spent so far.
38
+ signal input currentSpentVoiceCreditSubtotal;
39
+ // Salt for the total spent voice credits.
40
+ signal input currentSpentVoiceCreditSubtotalSalt;
41
+
42
+ // Total new voice credits spent.
43
+ signal input newSpentVoiceCreditSubtotal;
44
+ // Salt for the new total spent voice credits.
45
+ signal input newSpentVoiceCreditSubtotalSalt;
46
+
47
+ // Spent voice credits per vote option.
48
+ signal input currentPerVoteOptionSpentVoiceCredits[totalVoteOptions];
49
+ // Salt for the root of spent credits per option.
50
+ signal input currentPerVoteOptionSpentVoiceCreditsRootSalt;
51
+
52
+ // New spent voice credits per vote option.
53
+ signal input newPerVoteOptionSpentVoiceCredits[totalVoteOptions];
54
+ // Salt for the root of new spent credits per option.
55
+ signal input newPerVoteOptionSpentVoiceCreditsRootSalt;
56
+
57
+ // Compute the commitment to the current results.
58
+ var computedCurrentResultsRoot = QuinaryCheckRoot(voteOptionTreeDepth)(currentResults);
59
+
60
+ // Verify currentResultsCommitmentHash.
61
+ var computedCurrentResultsCommitment = PoseidonHasher(2)([computedCurrentResultsRoot, currentResultsRootSalt]);
62
+
63
+ // Compute the commitment to the current spent voice credits.
64
+ var computedCurrentSpentVoiceCreditsCommitment = PoseidonHasher(2)([currentSpentVoiceCreditSubtotal, currentSpentVoiceCreditSubtotalSalt]);
65
+
66
+ // Compute the root of the spent voice credits per vote option.
67
+ var computedCurrentPerVoteOptionSpentVoiceCreditsRoot = QuinaryCheckRoot(voteOptionTreeDepth)(currentPerVoteOptionSpentVoiceCredits);
68
+ var computedCurrentPerVoteOptionSpentVoiceCreditsCommitment = PoseidonHasher(2)([computedCurrentPerVoteOptionSpentVoiceCreditsRoot, currentPerVoteOptionSpentVoiceCreditsRootSalt]);
69
+
70
+ // Commit to the current tally.
71
+ var computedCurrentTallyCommitment = PoseidonHasher(3)([
72
+ computedCurrentResultsCommitment,
73
+ computedCurrentSpentVoiceCreditsCommitment,
74
+ computedCurrentPerVoteOptionSpentVoiceCreditsCommitment
75
+ ]);
76
+
77
+ // Check if the current tally commitment is correct only if this is not the first batch.
78
+ // computedIsZero.out is 1 if this is not the first batch.
79
+ // computedIsZero.out is 0 if this is the first batch.
80
+ var computedIsZero = IsZero()(isFirstBatch);
81
+
82
+ // isFirstCommitment is 0 if this is the first batch, currentTallyCommitment should be 0 if this is the first batch.
83
+ // isFirstCommitment is 1 if this is not the first batch, currentTallyCommitment should not be 0 if this is the first batch.
84
+ signal isFirstCommitment;
85
+ isFirstCommitment <== computedIsZero * computedCurrentTallyCommitment;
86
+ isFirstCommitment === currentTallyCommitment;
87
+
88
+ // Compute the root of the new results.
89
+ var computedNewResultsRoot = QuinaryCheckRoot(voteOptionTreeDepth)(newResults);
90
+ var computedNewResultsCommitment = PoseidonHasher(2)([computedNewResultsRoot, newResultsRootSalt]);
91
+
92
+ // Compute the commitment to the new spent voice credits value.
93
+ var computedNewSpentVoiceCreditsCommitment = PoseidonHasher(2)([newSpentVoiceCreditSubtotal, newSpentVoiceCreditSubtotalSalt]);
94
+
95
+ // Compute the root of the spent voice credits per vote option.
96
+ var computedNewPerVoteOptionSpentVoiceCreditsRoot = QuinaryCheckRoot(voteOptionTreeDepth)(newPerVoteOptionSpentVoiceCredits);
97
+ var computedNewPerVoteOptionSpentVoiceCreditsCommitment = PoseidonHasher(2)([computedNewPerVoteOptionSpentVoiceCreditsRoot, newPerVoteOptionSpentVoiceCreditsRootSalt]);
98
+
99
+ // Commit to the new tally.
100
+ var computedNewTallyCommitment = PoseidonHasher(3)([
101
+ computedNewResultsCommitment,
102
+ computedNewSpentVoiceCreditsCommitment,
103
+ computedNewPerVoteOptionSpentVoiceCreditsCommitment
104
+ ]);
105
+
106
+ computedNewTallyCommitment === newTallyCommitment;
107
+ }
@@ -7,7 +7,7 @@ include "./comparators.circom";
7
7
  include "../PoseidonHasher.circom";
8
8
 
9
9
  // @note taken from @zk-kit/circuits
10
- // if used directly in processMessages circom complains about duplicated
10
+ // if used directly in MessageProcessor circom complains about duplicated
11
11
  // templates (Ark, Poseidon, etc.)
12
12
  // This circuit is designed to calculate the root of a binary Merkle
13
13
  // tree given a leaf, its depth, and the necessary sibling
@@ -25,14 +25,17 @@ template BinaryMerkleRoot(MAX_DEPTH) {
25
25
  signal input leaf;
26
26
  // The depth of the Merkle tree.
27
27
  signal input depth;
28
- // The indices of the leaf node in the Merkle tree.
29
- signal input indices[MAX_DEPTH];
28
+ // The index of the leaf node in the Merkle tree.
29
+ signal input index;
30
30
  // The sibling nodes of the leaf node in the Merkle tree.
31
31
  signal input siblings[MAX_DEPTH][1];
32
32
 
33
33
  // The output of the Merkle tree root.
34
34
  signal output out;
35
35
 
36
+ // The indices of the leaf node in the Merkle tree.
37
+ signal indices[MAX_DEPTH] <== Num2Bits(MAX_DEPTH)(index);
38
+
36
39
  signal nodes[MAX_DEPTH + 1];
37
40
  nodes[0] <== leaf;
38
41
 
@@ -13,13 +13,13 @@ template LeafExists(levels) {
13
13
  // The elements along the path needed for the inclusion proof.
14
14
  signal input path_elements[levels][1];
15
15
  // The indices indicating the path taken through the tree for the leaf.
16
- signal input path_index[levels];
16
+ signal input path_indices[levels];
17
17
  // The root of the Merkle tree, against which the inclusion is verified.
18
18
  signal input root;
19
19
 
20
20
  var computedMerkleRoot = MerkleTreeInclusionProof(levels)(
21
21
  leaf,
22
- path_index,
22
+ path_indices,
23
23
  path_elements
24
24
  );
25
25
 
@@ -12,7 +12,7 @@ template MerkleTreeInclusionProof(n_levels) {
12
12
  // The leaf node from which the Merkle root is calculated.
13
13
  signal input leaf;
14
14
  // Indices indicating left or right child for each level of the tree.
15
- signal input path_index[n_levels];
15
+ signal input path_indices[n_levels];
16
16
  // Sibling node values required to compute the hash at each level.
17
17
  signal input path_elements[n_levels][1];
18
18
 
@@ -25,8 +25,8 @@ template MerkleTreeInclusionProof(n_levels) {
25
25
  levelHashes[0] <== leaf;
26
26
 
27
27
  for (var i = 0; i < n_levels; i++) {
28
- // Validate path_index to be either 0 or 1, ensuring no other values.
29
- path_index[i] * (1 - path_index[i]) === 0;
28
+ // Validate path_indices to be either 0 or 1, ensuring no other values.
29
+ path_indices[i] * (1 - path_indices[i]) === 0;
30
30
 
31
31
  // Configure the multiplexer based on the path index for the current level.
32
32
  var multiplexer[2][2] = [
@@ -36,7 +36,7 @@ template MerkleTreeInclusionProof(n_levels) {
36
36
 
37
37
  var multiplexerResult[2] = MultiMux1(2)(
38
38
  multiplexer,
39
- path_index[i]
39
+ path_indices[i]
40
40
  );
41
41
 
42
42
  var computedLevelHash = PoseidonHasher(2)([multiplexerResult[0], multiplexerResult[1]]);
@@ -0,0 +1,54 @@
1
+ pragma circom 2.0.0;
2
+
3
+ // local imports
4
+ include "../PoseidonHasher.circom";
5
+
6
+ /**
7
+ * Computes the root of a quintary Merkle tree given a list of leaves.
8
+ * This template constructs a Merkle tree with each node having 5 children (quintary)
9
+ * and computes the root by hashing with Poseidon the leaves and intermediate nodes in the given order.
10
+ * The computation is performed by first hashing groups of 5 leaves to form the bottom layer of nodes,
11
+ * then recursively hashing groups of these nodes to form the next layer, and so on, until the root is computed.
12
+ */
13
+ template QuinaryCheckRoot(levels) {
14
+ var LEAVES_PER_NODE = 5;
15
+ var totalLeaves = LEAVES_PER_NODE ** levels;
16
+ var numLeafHashers = LEAVES_PER_NODE ** (levels - 1);
17
+
18
+ signal input leaves[totalLeaves];
19
+ signal output root;
20
+
21
+ // Determine the total number of hashers.
22
+ var numHashers = 0;
23
+ for (var i = 0; i < levels; i++) {
24
+ numHashers += LEAVES_PER_NODE ** i;
25
+ }
26
+
27
+ var computedHashers[numHashers];
28
+
29
+ // Initialize hashers for the leaves.
30
+ for (var i = 0; i < numLeafHashers; i++) {
31
+ computedHashers[i] = PoseidonHasher(5)([
32
+ leaves[i * LEAVES_PER_NODE + 0],
33
+ leaves[i * LEAVES_PER_NODE + 1],
34
+ leaves[i * LEAVES_PER_NODE + 2],
35
+ leaves[i * LEAVES_PER_NODE + 3],
36
+ leaves[i * LEAVES_PER_NODE + 4]
37
+ ]);
38
+ }
39
+
40
+ // Initialize hashers for intermediate nodes and compute the root.
41
+ var k = 0;
42
+ for (var i = numLeafHashers; i < numHashers; i++) {
43
+ computedHashers[i] = PoseidonHasher(5)([
44
+ computedHashers[k * LEAVES_PER_NODE + 0],
45
+ computedHashers[k * LEAVES_PER_NODE + 1],
46
+ computedHashers[k * LEAVES_PER_NODE + 2],
47
+ computedHashers[k * LEAVES_PER_NODE + 3],
48
+ computedHashers[k * LEAVES_PER_NODE + 4]
49
+ ]);
50
+ k++;
51
+ }
52
+
53
+ root <== computedHashers[numHashers - 1];
54
+ }