status-im
/
dagger-contracts
mirror of https://github.com/status-im/dagger-contracts.git
2
0
Fork 0
Code Issues Projects Releases Wiki Activity

Collateral invariant

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