Apply suggestions from code review

Co-authored-by: Qiang Zhu <[email protected]>
ethstorage · Sep 24, 2024 · b1e2dfc · b1e2dfc
1 parent 70937b7
commit b1e2dfc
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/contracts/StorageContract.sol b/contracts/StorageContract.sol
@@ -284,7 +284,13 @@ abstract contract StorageContract is DecentralizedKV, ReentrancyGuardTransient {
                 uint256 prepaidAmountCap =
                     STORAGE_COST * ((1 << SHARD_ENTRY_BITS) - kvEntryCount % (1 << SHARD_ENTRY_BITS));
                 if (prepaidAmountCap > prepaidAmount) {
-                    prepaidAmountSaved = _paymentIn(prepaidAmountCap, prepaidLastMineTime, _minedTs)
+if (prepaidLastMineTime < _minedTs) {
+    fullReward = _paymentIn(STORAGE_COST << SHARD_ENTRY_BITS, info.lastMineTime, _minedTs);
+    prepaidAmountIn = _paymentIn(prepaidAmount, prepaidLastMineTime, _minedTs);
+    if (prepaidAmountIn > fullReward - reward) {
+        prepaidAmountIn = fullReward - reward;
+    }
+}
                         - _paymentIn(prepaidAmount, prepaidLastMineTime, _minedTs);
                     prepaidAmountCap = prepaidAmount;
                 }