logos-storage/logos-storage-contracts-eth
1
0
Fork 0
mirror of https://github.com/logos-storage/logos-storage-contracts-eth.git synced 2026-01-04 06:13:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

vault: check Lock invariant before writing

Browse Source
This commit is contained in:
Mark Spanbroek 2025-01-28 15:00:20 +01:00
parent 220b648758
commit f0da7cd918
1 changed files with 13 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
contracts/vault/VaultBase.sol
View File

@ -149,10 +149,12 @@ abstract contract VaultBase {
Timestamp expiry,
Timestamp maximum
) internal {
require(expiry <= maximum, ExpiryPastMaximum());
Lock memory existing = _locks[controller][context];
require(existing.maximum == Timestamp.wrap(0), AlreadyLocked());
_locks[controller][context] = Lock({expiry: expiry, maximum: maximum});
Lock memory lock = _locks[controller][context];
require(lock.maximum == Timestamp.wrap(0), AlreadyLocked());
lock.expiry = expiry;
lock.maximum = maximum;
_checkLockInvariant(lock);
_locks[controller][context] = lock;
}
function _extendLock(
@ -163,8 +165,9 @@ abstract contract VaultBase {
Lock memory lock = _locks[controller][context];
require(lock.isLocked(), LockRequired());
require(lock.expiry <= expiry, InvalidExpiry());
require(expiry <= lock.maximum, ExpiryPastMaximum());
_locks[controller][context].expiry = expiry;
lock.expiry = expiry;
_checkLockInvariant(lock);
_locks[controller][context] = lock;
}
function _flow(
@ -198,6 +201,10 @@ abstract contract VaultBase {
_flows[controller][context][to] = receiverFlow;
}
function _checkLockInvariant(Lock memory lock) private pure {
require(lock.expiry <= lock.maximum, ExpiryPastMaximum());
}
function _checkFlowInvariant(
Balance memory balance,
Lock memory lock,