Fix bug: block number needs to be updated when setting reward speed a…

…gain after it has been unset Bug Rari-Capital#2 described in this post: https://www.comp.xyz/t/comptroller-compspeed-bug/2111
sriyantra · Sep 23, 2021 · 335cd82 · 335cd82
1 parent 7276d2d
commit 335cd82
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/contracts/RewardsDistributor.sol b/contracts/RewardsDistributor.sol
@@ -179,6 +179,9 @@ contract RewardsDistributor is ExponentialNoError {
                 if (compBorrowState[address(cToken)].index == 0) {
                     allMarkets.push(cToken);
                 }
+            } else {
+                // Update block number to ensure extra interest is not accrued during the prior period
+                compSupplyState[address(cToken)].block = safe32(getBlockNumber(), "block number exceeds 32 bits");
             }
         }
 
@@ -214,6 +217,9 @@ contract RewardsDistributor is ExponentialNoError {
                 if (compSupplyState[address(cToken)].index == 0) {
                     allMarkets.push(cToken);
                 }
+            } else {
+                // Update block number to ensure extra interest is not accrued during the prior period
+                compBorrowState[address(cToken)].block = safe32(getBlockNumber(), "block number exceeds 32 bits");
             }
         }