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 22:33:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

vault: disallow designating of flowing tokens

Browse Source
This commit is contained in:
Mark Spanbroek 2025-01-30 13:36:46 +01:00
parent 4c6c68f9d9
commit d0b8f961b2
2 changed files with 17 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
contracts/vault/VaultBase.sol
View File

@ -150,10 +150,17 @@ abstract contract VaultBase {
Recipient recipient,
uint128 amount
) internal {
Balance storage balance = _balances[controller][context][recipient];
Balance memory balance = _balances[controller][context][recipient];
require(amount <= balance.available, InsufficientBalance());
balance.available -= amount;
balance.designated += amount;
Flow memory flow = _flows[controller][context][recipient];
Lock memory lock = _locks[controller][context];
_checkFlowInvariant(balance, lock, flow);
_balances[controller][context][recipient] = balance;
}
function _lockup(

9
test/Vault.tests.js
View File

@ -599,6 +599,15 @@ describe("Vault", function () {
).to.be.revertedWith("InsufficientBalance")
})
it("cannot designate tokens that are flowing", async function () {
await vault.flow(context, sender, receiver, 5)
setAutomine(true)
await vault.designate(context, sender, 500)
await expect(vault.designate(context, sender, 1)).to.be.revertedWith(
"InsufficientBalance"
)
})
it("cannot burn tokens that are flowing", async function () {
await vault.flow(context, sender, receiver, 5)
setAutomine(true)