codex-storage/codex-contracts-eth
1
0
Fork 0
mirror of https://github.com/codex-storage/codex-contracts-eth.git synced 2025-02-10 10:26:32 +00:00
Code Issues Packages Projects Releases Wiki Activity

Collateral invariant

Browse Source
This commit is contained in:
Mark Spanbroek 2022-02-14 16:19:47 +01:00 committed by markspanbroek
parent 6e0aded775
commit 3d50e56fe7
1 changed files with 12 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
contracts/Collateral.sol
View File

@ -7,7 +7,10 @@ contract Collateral {
IERC20 private immutable token;
mapping(address => uint256) private balances;
constructor(IERC20 _token) {
uint256 private totalDeposited;
uint256 private totalBalance;
constructor(IERC20 _token) invariant {
token = _token;
}
@ -15,8 +18,15 @@ contract Collateral {
return balances[account];
}
function deposit(uint256 amount) public {
function deposit(uint256 amount) public invariant {
token.transferFrom(msg.sender, address(this), amount);
totalDeposited += amount;
balances[msg.sender] += amount;
totalBalance += amount;
}
modifier invariant() {
_;
assert(totalDeposited == totalBalance);
}
}