TOB-GARDEN-7 resolution

gardenfi · Mar 13, 2024 · 43e8eb4 · 43e8eb4
1 parent 16eb2a9
commit 43e8eb4
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/contracts/stake/DelegateManager.sol b/contracts/stake/DelegateManager.sol
@@ -175,6 +175,8 @@ abstract contract DelegateManager is BaseStaker {
      * @return voteCount The total number of votes delegated to the specified filler address.
      */
     function getVotes(address filler) external view returns (uint256 voteCount) {
+        _checkRole(FILLER, filler);
+
         bytes32[] memory delegates = fillers[filler].delegateStakeIDs._inner._values;
         uint256 delegateLength = delegates.length;
 

diff --git a/contracts/stake/FillerManager.sol b/contracts/stake/FillerManager.sol
@@ -71,7 +71,9 @@ abstract contract FillerManager is BaseStaker {
         require(filler.deregisteredAt != 0, "FillerManager: not registered");
         require(filler.deregisteredAt + FILLER_COOL_DOWN < block.number, "FillerManager: cooldown not passed");
 
-        delete (fillers[filler_]);
+        fillers[filler_].feeInBips = 0;
+        fillers[filler_].stake = 0;
+        fillers[filler_].deregisteredAt = 0;
 
         SEED.safeTransfer(filler_, FILLER_STAKE);