mirror of
https://github.com/vacp2p/staking-reward-streamer.git
synced 2026-01-14 23:48:07 -05:00
9374025924
This commit introduces changes related to vault registrations in the stake manager. The stake manager needs to keep track of the vaults a users creates so it can aggregate accumulated MP across vaults for any given user. The `StakeVault` now comes with a `register()` function which needs to be called to register itself with the stake manager. `StakeManager` has a new `onlyRegisteredVault` modifier that ensures only registered vaults can actually `stake` and `unstake`. Closes #70
81 lines
3.5 KiB
Ruby
81 lines
3.5 KiB
Ruby
using RewardsStreamerMP as streamer;
|
|
using ERC20A as staked;
|
|
|
|
methods {
|
|
function emergencyModeEnabled() external returns (bool) envfree;
|
|
}
|
|
|
|
definition isViewFunction(method f) returns bool = (
|
|
f.selector == sig:streamer.STAKING_TOKEN().selector ||
|
|
f.selector == sig:streamer.SCALE_FACTOR().selector ||
|
|
f.selector == sig:streamer.MP_RATE_PER_YEAR().selector ||
|
|
f.selector == sig:streamer.MIN_LOCKUP_PERIOD().selector ||
|
|
f.selector == sig:streamer.MAX_LOCKUP_PERIOD().selector ||
|
|
f.selector == sig:streamer.MAX_MULTIPLIER().selector ||
|
|
f.selector == sig:streamer.rewardIndex().selector ||
|
|
f.selector == sig:streamer.lastMPUpdatedTime().selector ||
|
|
f.selector == sig:streamer.owner().selector ||
|
|
f.selector == sig:streamer.totalStaked().selector ||
|
|
f.selector == sig:streamer.totalMaxMP().selector ||
|
|
f.selector == sig:streamer.totalMP().selector ||
|
|
f.selector == sig:streamer.accounts(address).selector ||
|
|
f.selector == sig:streamer.emergencyModeEnabled().selector ||
|
|
f.selector == sig:streamer.getStakedBalance(address).selector ||
|
|
f.selector == sig:streamer.getAccount(address).selector ||
|
|
f.selector == sig:streamer.rewardsBalanceOf(address).selector ||
|
|
f.selector == sig:streamer.totalRewardsSupply().selector ||
|
|
f.selector == sig:streamer.calculateAccountRewards(address).selector ||
|
|
f.selector == sig:streamer.lastRewardTime().selector ||
|
|
f.selector == sig:streamer.rewardAmount().selector ||
|
|
f.selector == sig:streamer.totalRewardsAccrued().selector ||
|
|
f.selector == sig:streamer.rewardStartTime().selector ||
|
|
f.selector == sig:streamer.rewardEndTime().selector ||
|
|
f.selector == sig:streamer.getUserTotalMP(address).selector ||
|
|
f.selector == sig:streamer.getUserTotalMaxMP(address).selector ||
|
|
f.selector == sig:streamer.getUserTotalStakedBalance(address).selector ||
|
|
f.selector == sig:streamer.vaults(address,uint256).selector ||
|
|
f.selector == sig:streamer.vaultOwners(address).selector ||
|
|
f.selector == sig:streamer.registerVault().selector ||
|
|
f.selector == sig:streamer.getUserVaults(address).selector
|
|
);
|
|
|
|
definition isOwnableFunction(method f) returns bool = (
|
|
f.selector == sig:streamer.renounceOwnership().selector ||
|
|
f.selector == sig:streamer.transferOwnership(address).selector ||
|
|
f.selector == sig:streamer.setReward(uint256, uint256).selector
|
|
);
|
|
|
|
definition isTrustedCodehashAccessFunction(method f) returns bool = (
|
|
f.selector == sig:streamer.setTrustedCodehash(bytes32, bool).selector ||
|
|
f.selector == sig:streamer.isTrustedCodehash(bytes32).selector
|
|
);
|
|
|
|
definition isInitializerFunction(method f) returns bool = (
|
|
f.selector == sig:streamer.initialize(address,address).selector
|
|
);
|
|
|
|
definition isUUPSUpgradeableFunction(method f) returns bool = (
|
|
f.selector == sig:streamer.proxiableUUID().selector ||
|
|
f.selector == sig:streamer.UPGRADE_INTERFACE_VERSION().selector ||
|
|
f.selector == sig:streamer.upgradeToAndCall(address, bytes).selector ||
|
|
f.selector == sig:streamer.__TrustedCodehashAccess_init(address).selector
|
|
);
|
|
|
|
rule accountCanOnlyLeaveInEmergencyMode(method f) {
|
|
env e;
|
|
calldataarg args;
|
|
|
|
require emergencyModeEnabled() == true;
|
|
|
|
f@withrevert(e, args);
|
|
bool isReverted = lastReverted;
|
|
|
|
assert !isReverted => f.selector == sig:streamer.leave().selector ||
|
|
isViewFunction(f) ||
|
|
isOwnableFunction(f) ||
|
|
isTrustedCodehashAccessFunction(f) ||
|
|
isInitializerFunction(f) ||
|
|
isUUPSUpgradeableFunction(f);
|
|
}
|
|
|