@crisp-e3/contracts 0.0.1-test → 0.0.2-test

package/contracts/CRISPProgram.sol

@@ -10,6 +10,7 @@ import {Ownable} from "@openzeppelin/contracts/access/Ownable.sol";
 import {IE3Program} from "@enclave-e3/contracts/contracts/interfaces/IE3Program.sol";
 import {IInputValidator} from "@enclave-e3/contracts/contracts/interfaces/IInputValidator.sol";
 import {IEnclave} from "@enclave-e3/contracts/contracts/interfaces/IEnclave.sol";
+import {E3} from "@enclave-e3/contracts/contracts/interfaces/IE3.sol";
 import {CRISPInputValidatorFactory} from "./CRISPInputValidatorFactory.sol";
 import {HonkVerifier} from "./CRISPVerifier.sol";
 
@@ -24,6 +25,10 @@ contract CRISPProgram is IE3Program, Ownable {
     HonkVerifier private immutable HONK_VERIFIER;
     bytes32 public imageId;
 
+    /// @notice Half of the largest minimum degree used to fit votes
+    /// inside the plaintext polynomial
+    uint256 public constant HALF_LARGEST_MINIMUM_DEGREE = 28;
+
     // Mappings
     mapping(address => bool) public authorizedContracts;
     mapping(uint256 e3Id => bytes32 paramsHash) public paramsHashes;
@@ -55,10 +60,7 @@ contract CRISPProgram is IE3Program, Ownable {
     ) Ownable(msg.sender) {
         require(address(_enclave) != address(0), EnclaveAddressZero());
         require(address(_verifier) != address(0), VerifierAddressZero());
-        require(
-            address(_inputValidatorFactory) != address(0),
-            InvalidInputValidatorFactory()
-        );
+        require(address(_inputValidatorFactory) != address(0), InvalidInputValidatorFactory());
         require(address(_honkVerifier) != address(0), InvalidHonkVerifier());
 
         enclave = _enclave;
@@ -91,36 +93,68 @@ contract CRISPProgram is IE3Program, Ownable {
     /// @notice Validate the E3 program parameters
     /// @param e3Id The E3 program ID
     /// @param e3ProgramParams The E3 program parameters
-    function validate(
-        uint256 e3Id,
-        uint256,
-        bytes calldata e3ProgramParams,
-        bytes calldata
-    ) external returns (bytes32, IInputValidator inputValidator) {
-        require(
-            authorizedContracts[msg.sender] || msg.sender == owner(),
-            CallerNotAuthorized()
-        );
+    function validate(uint256 e3Id, uint256, bytes calldata e3ProgramParams, bytes calldata)
+        external
+        returns (bytes32, IInputValidator inputValidator)
+    {
+        require(authorizedContracts[msg.sender] || msg.sender == owner(), CallerNotAuthorized());
         require(paramsHashes[e3Id] == bytes32(0), E3AlreadyInitialized());
         paramsHashes[e3Id] = keccak256(e3ProgramParams);
 
         // Deploy a new input validator
-        inputValidator = IInputValidator(
-            INPUT_VALIDATOR_FACTORY.deploy(address(HONK_VERIFIER), owner())
-        );
+        inputValidator = IInputValidator(INPUT_VALIDATOR_FACTORY.deploy(address(HONK_VERIFIER), owner()));
 
         return (ENCRYPTION_SCHEME_ID, inputValidator);
     }
 
+    /// @notice Decode the tally from the plaintext output
+    /// @param e3Id The E3 program ID
+    /// @return yes The number of yes votes
+    /// @return no The number of no votes
+    function decodeTally(uint256 e3Id) public view returns (uint256 yes, uint256 no) {
+        // fetch from enclave
+        E3 memory e3 = enclave.getE3(e3Id);
+
+        // abi decode it into an array of uint256
+        uint256[] memory tally = abi.decode(e3.plaintextOutput, (uint256[]));
+
+        /// @notice We want to completely ignore anything outside of the coefficients
+        /// we agreed to store out votes on.
+        uint256 halfD = tally.length / 2;
+        uint256 START_INDEX_Y = halfD - HALF_LARGEST_MINIMUM_DEGREE;
+        uint256 START_INDEX_N = tally.length - HALF_LARGEST_MINIMUM_DEGREE;
+
+        // first weight (we are converting back from bits to integer)
+        uint256 weight = 2 ** (HALF_LARGEST_MINIMUM_DEGREE - 1);
+
+        // Convert yes votes
+        for (uint256 i = START_INDEX_Y; i < halfD; i++) {
+            yes += tally[i] * weight;
+            weight /= 2; // Right shift equivalent
+        }
+
+        // Reset weight for no votes
+        weight = 2 ** (HALF_LARGEST_MINIMUM_DEGREE - 1);
+
+        // Convert no votes
+        for (uint256 i = START_INDEX_N; i < tally.length; i++) {
+            no += tally[i] * weight;
+            weight /= 2;
+        }
+
+        return (yes, no);
+    }
+
     /// @notice Verify the proof
     /// @param e3Id The E3 program ID
     /// @param ciphertextOutputHash The hash of the ciphertext output
     /// @param proof The proof to verify
-    function verify(
-        uint256 e3Id,
-        bytes32 ciphertextOutputHash,
-        bytes memory proof
-    ) external view override returns (bool) {
+    function verify(uint256 e3Id, bytes32 ciphertextOutputHash, bytes memory proof)
+        external
+        view
+        override
+        returns (bool)
+    {
         require(paramsHashes[e3Id] != bytes32(0), E3DoesNotExist());
         bytes32 inputRoot = bytes32(enclave.getInputRoot(e3Id));
         bytes memory journal = new bytes(396); // (32 + 1) * 4 * 3
@@ -137,11 +171,7 @@ contract CRISPProgram is IE3Program, Ownable {
     /// @param journal The journal to encode into
     /// @param startIndex The start index in the journal
     /// @param hashVal The hash value to encode
-    function encodeLengthPrefixAndHash(
-        bytes memory journal,
-        uint256 startIndex,
-        bytes32 hashVal
-    ) internal pure {
+    function encodeLengthPrefixAndHash(bytes memory journal, uint256 startIndex, bytes32 hashVal) internal pure {
         journal[startIndex] = 0x20;
         startIndex += 4;
         for (uint256 i = 0; i < 32; i++) {

package/contracts/Mocks/MockEnclave.sol

@@ -0,0 +1,39 @@
+// SPDX-License-Identifier: LGPL-3.0-only
+//
+// This file is provided WITHOUT ANY WARRANTY;
+// without even the implied warranty of MERCHANTABILITY
+// or FITNESS FOR A PARTICULAR PURPOSE.
+pragma solidity >=0.8.27;
+
+import {E3} from "@enclave-e3/contracts/contracts/interfaces/IE3.sol";
+import {IE3Program} from "@enclave-e3/contracts/contracts/interfaces/IE3Program.sol";
+import {IInputValidator} from "@enclave-e3/contracts/contracts/interfaces/IInputValidator.sol";
+import {IDecryptionVerifier} from "@enclave-e3/contracts/contracts/interfaces/IDecryptionVerifier.sol";
+
+contract MockEnclave {
+    bytes public plaintextOutput;
+
+    function setPlaintextOutput(uint256[] memory plaintext) external {
+        plaintextOutput = abi.encode(plaintext);
+    }
+
+    function getE3(uint256 e3Id) external view returns (E3 memory) {
+        return E3({
+            seed: 0,
+            threshold: [uint32(1), uint32(2)],
+            requestBlock: 0,
+            startWindow: [uint256(0), uint256(0)],
+            duration: 0,
+            expiration: 0,
+            encryptionSchemeId: bytes32(0),
+            e3Program: IE3Program(address(0)),
+            e3ProgramParams: bytes(""),
+            customParams: bytes(""),
+            inputValidator: IInputValidator(address(0)),
+            decryptionVerifier: IDecryptionVerifier(address(0)),
+            committeePublicKey: bytes32(0),
+            ciphertextOutput: bytes32(0),
+            plaintextOutput: plaintextOutput
+        });
+    }
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@crisp-e3/contracts",
-  "version": "0.0.1-test",
+  "version": "0.0.2-test",
   "type": "module",
   "files": [
     "contracts",
@@ -47,6 +47,7 @@
     "@types/chai": "^4.2.0",
     "@types/mocha": ">=9.1.0",
     "@types/node": "^22.18.0",
+    "chai": "^6.2.0",
     "dotenv": "^16.4.5",
     "ethers": "^6.15.0",
     "forge-std": "github:foundry-rs/forge-std#v1.9.4",
@@ -57,7 +58,9 @@
     "ts-node": "^10.9.2",
     "typechain": "^8.3.0",
     "typescript": "5.8.3",
-    "viem": "2.30.6"
+    "viem": "2.30.6",
+    "@crisp-e3/sdk": "^0.0.2-test",
+    "@crisp-e3/zk-inputs": "^0.0.2-test"
   },
   "scripts": {
     "compile": "hardhat compile",
@@ -66,7 +69,7 @@
     "deploy:contracts": "hardhat run deploy/deploy.ts",
     "deploy:contracts:full": "export DEPLOY_ENCLAVE=true && pnpm deploy:contracts",
     "deploy:contracts:full:mock": "export DEPLOY_ENCLAVE=true && export USE_MOCK_VERIFIER=true && export USE_MOCK_INPUT_VALIDATOR=true && pnpm deploy:contracts",
-    "test": "hardhat test tests/crisp.contracts.test.ts --network localhost",
+    "test": "hardhat test mocha",
     "verify": "hardhat run deploy/verify.ts"
   }
 }