fix: ensure one time address == precomputed address

Grateful-protocol · Nov 26, 2024 · ef629bc · ef629bc
1 parent f40c1f6
commit ef629bc
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/contracts/Grateful.sol b/src/contracts/Grateful.sol
@@ -269,6 +269,11 @@ contract Grateful is IGrateful, Ownable2Step, ReentrancyGuard {
     oneTimePayments[_precomputed] = true;
     oneTime =
       new OneTime{salt: bytes32(_salt)}(IGrateful(address(this)), _tokens, _merchant, _amount, _paymentId, _yieldFunds);
+
+    if (address(oneTime) != _precomputed) {
+      revert Grateful_PrecomputedAddressMismatch();
+    }
+
     emit OneTimePaymentCreated(_merchant, _tokens, _amount);
   }