Add delegate to lockProof()

Signed-off-by: Jim Zhang <[email protected]>
hyperledger-labs · Sep 26, 2024 · 5ae387b · 5ae387b
1 parent 713dcce
commit 5ae387b
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/solidity/contracts/lib/zeto_common.sol b/solidity/contracts/lib/zeto_common.sol
@@ -43,9 +43,17 @@ abstract contract ZetoCommon is OwnableUpgradeable {
     // should be called by escrow contracts that will use uploaded proofs
     // to execute transactions, in order to prevent the proof from being used
     // by parties other than the escrow contract
-    function lockProof(Commonlib.Proof calldata proof) public {
+    function lockProof(
+        Commonlib.Proof calldata proof,
+        address delegate
+    ) public {
         bytes32 proofHash = Commonlib.getProofHash(proof);
-        lockedProofs[proofHash] = msg.sender;
+        require(lockedProofs[proofHash] == address(0), "Proof already locked");
+        if (delegate != address(0)) {
+            lockedProofs[proofHash] = delegate;
+        } else {
+            lockedProofs[proofHash] = msg.sender;
+        }
     }
 
     function sortInputsAndOutputs(

diff --git a/solidity/contracts/zkDvP.sol b/solidity/contracts/zkDvP.sol
@@ -206,8 +206,10 @@ contract zkDvP {
         bytes32 proofHash = getProofHash(proof);
         if (trade.paymentProofHash == proofHash) {
             trade.paymentProof = proof;
+            paymentToken.lockProof(proof, address(this));
         } else if (trade.assetProofHash == proofHash) {
             trade.assetProof = proof;
+            assetToken.lockProof(proof, address(this));
         } else {
             revert("Invalid proof");
         }