logos-co/staking
1
0
Fork 0
mirror of https://github.com/logos-co/staking.git synced 2025-02-09 09:14:43 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(StakeManager): properly init accs and checks init

Browse Source
This commit is contained in:
Ricardo Guilherme Schmidt 2024-02-22 18:23:03 -03:00
parent c9ed9dd833
commit f06168c8ce
1 changed files with 19 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
contracts/StakeManager.sol
View File

@ -62,6 +62,13 @@ contract StakeManager is Ownable {
_; _;
} }
modifier onlyInitialized(address account) {
if (accounts[account].lockUntil == 0) {
revert StakeManager__AccountNotInitialized();
}
_;
}
/** /**
* @notice Only callable when migration is not initialized. * @notice Only callable when migration is not initialized.
*/ */
@ -126,7 +133,14 @@ contract StakeManager is Ownable {
revert StakeManager__InvalidLockupPeriod(); revert StakeManager__InvalidLockupPeriod();
} }
Account storage account = accounts[msg.sender]; Account storage account = accounts[msg.sender];
_processAccount(account, currentEpoch); if (account.lockUntil == 0) {
// account not initialized
account.lockUntil = block.timestamp;
account.epoch = currentEpoch + 1; //starts next epoch
account.rewardAddress = StakeVault(msg.sender).owner();
} else {
_processAccount(account, currentEpoch);
}
_mintIntialMP(account, _time, _amount); _mintIntialMP(account, _time, _amount);
//update storage //update storage
totalSupplyBalance += _amount; totalSupplyBalance += _amount;
@ -137,7 +151,7 @@ contract StakeManager is Ownable {
/** /**
* leaves the staking pool and withdraws all funds; * leaves the staking pool and withdraws all funds;
*/ */
function unstake(uint256 _amount) external onlyVault noMigration processEpoch { function unstake(uint256 _amount) external onlyVault onlyInitialized(msg.sender) noMigration processEpoch {
Account storage account = accounts[msg.sender]; Account storage account = accounts[msg.sender];
if (_amount > account.balance) { if (_amount > account.balance) {
revert("StakeManager: Amount exceeds balance"); revert("StakeManager: Amount exceeds balance");
@ -165,7 +179,7 @@ contract StakeManager is Ownable {
* @dev Reverts when `_time` is bigger than `MAX_LOCKUP_PERIOD` * @dev Reverts when `_time` is bigger than `MAX_LOCKUP_PERIOD`
* @dev Reverts when `_time + block.timestamp` is smaller than current lock time. * @dev Reverts when `_time + block.timestamp` is smaller than current lock time.
*/ */
function lock(uint256 _time) external onlyVault noMigration processEpoch { function lock(uint256 _time) external onlyVault onlyInitialized(msg.sender) noMigration processEpoch {
if (_time > MAX_LOCKUP_PERIOD) { if (_time > MAX_LOCKUP_PERIOD) {
revert StakeManager__InvalidLockupPeriod(); revert StakeManager__InvalidLockupPeriod();
} }
@ -192,7 +206,7 @@ contract StakeManager is Ownable {
* @param _vault Referred account * @param _vault Referred account
* @param _limitEpoch Until what epoch it should be executed * @param _limitEpoch Until what epoch it should be executed
*/ */
function executeAccount(address _vault, uint256 _limitEpoch) external processEpoch { function executeAccount(address _vault, uint256 _limitEpoch) external onlyInitialized(_vault) processEpoch {
_processAccount(accounts[_vault], _limitEpoch); _processAccount(accounts[_vault], _limitEpoch);
} }
@ -251,6 +265,7 @@ contract StakeManager is Ownable {
function migrateTo(bool _acceptMigration) function migrateTo(bool _acceptMigration)
external external
onlyVault onlyVault
onlyInitialized(msg.sender)
onlyMigration onlyMigration
processEpoch processEpoch
returns (StakeManager newManager) returns (StakeManager newManager)