chore: applied reviews

zama-ai · Dec 23, 2024 · 9cda628 · 9cda628
1 parent 6d59ffa
commit 9cda628
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 3 deletions.
diff --git a/contracts/codegen/templates.ts b/contracts/codegen/templates.ts
@@ -66,6 +66,7 @@ export function implSol(operators: Operator[]): string {
 
   const coprocessorInterface = generateImplCoprocessorInterface(operators);
   const aclInterface = generateACLInterface();
+  const inputVerifierInterface = generateInputVerifierInterface();
 
   res.push(`
 // SPDX-License-Identifier: BSD-3-Clause-Clear
@@ -79,6 +80,8 @@ ${coprocessorInterface}
 
 ${aclInterface}
 
+${inputVerifierInterface}
+
 library Impl {
   // keccak256(abi.encode(uint256(keccak256("fhevm.storage.FHEVMConfig")) - 1)) & ~bytes32(uint256(0xff))
   bytes32 private constant FHEVMConfigLocation = 0xed8d60e34876f751cc8b014c560745351147d9de11b9347c854e881b128ea600;
@@ -215,6 +218,14 @@ function generateACLInterface(): string {
   `;
 }
 
+function generateInputVerifierInterface(): string {
+  return `
+  interface IInputVerifier {
+    function cleanTransientStorage() external;
+  }
+  `;
+}
+
 export function tfheSol(
   operators: Operator[],
   supportedBits: number[],
@@ -1464,7 +1475,7 @@ function implCustomMethods(): string {
 
     function cleanTransientStorageInputVerifier() internal {
       FHEVMConfig.FHEVMConfigStruct storage $ = getFHEVMConfig();
-      IACL($.InputVerifierAddress).cleanTransientStorage();
+      IInputVerifier($.InputVerifierAddress).cleanTransientStorage();
     }
 
     function isAllowed(uint256 handle, address account) internal view returns (bool) {

diff --git a/contracts/examples/PaymentLimit.sol b/contracts/examples/PaymentLimit.sol
@@ -3,7 +3,6 @@
 pragma solidity ^0.8.24;
 
 import "../lib/TFHE.sol";
-import "@openzeppelin/contracts/access/Ownable2Step.sol";
 
 /// @title PaymentLimit
 /// @notice A contract to demonstrate FHE gas limits in different scenarios

diff --git a/contracts/lib/Impl.sol b/contracts/lib/Impl.sol
@@ -52,6 +52,10 @@ interface IACL {
     function allowForDecryption(uint256[] memory handlesList) external;
 }
 
+interface IInputVerifier {
+    function cleanTransientStorage() external;
+}
+
 library Impl {
     // keccak256(abi.encode(uint256(keccak256("fhevm.storage.FHEVMConfig")) - 1)) & ~bytes32(uint256(0xff))
     bytes32 private constant FHEVMConfigLocation = 0xed8d60e34876f751cc8b014c560745351147d9de11b9347c854e881b128ea600;
@@ -372,7 +376,7 @@ library Impl {
 
     function cleanTransientStorageInputVerifier() internal {
         FHEVMConfig.FHEVMConfigStruct storage $ = getFHEVMConfig();
-        IACL($.InputVerifierAddress).cleanTransientStorage();
+        IInputVerifier($.InputVerifierAddress).cleanTransientStorage();
     }
 
     function isAllowed(uint256 handle, address account) internal view returns (bool) {