repermit

orbs-network · Jan 15, 2024 · 1ab4e10 · 1ab4e10
1 parent c6d8c67
commit 1ab4e10
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/src/RePermit.sol b/src/RePermit.sol
@@ -78,15 +78,15 @@ contract RePermit is EIP712, IEIP712 {
         bytes calldata signature
     ) external {
         if (block.timestamp > permit.deadline) revert SignatureExpired(permit.deadline);
-        uint256 _s = spent[signer][permit.permitted.token][msg.sender][permit.nonce];
-        if (_s + request.amount > permit.permitted.amount) {
-            revert InsufficientAllowance(_s);
-        }
 
         bytes32 hash = _hashTypedDataV4(RePermitLib.hashWithWitness(permit, witness, witnessTypeString, msg.sender));
         if (!SignatureChecker.isValidSignatureNow(signer, hash, signature)) revert InvalidSignature();
 
-        spent[signer][permit.permitted.token][msg.sender][permit.nonce] += request.amount;
+        uint256 _spent = (spent[signer][permit.permitted.token][msg.sender][permit.nonce] += request.amount);
+        if (_spent > permit.permitted.amount) {
+            revert InsufficientAllowance(_spent - request.amount);
+        }
+
         IERC20(permit.permitted.token).safeTransferFrom(signer, request.to, request.amount);
     }
 }