feat(redis): set sample max value (#223)

deploy/main/011_deploy_state_changes.ts

@@ -0,0 +1 @@
+import '../test/011_deploy_state_changes';

deploy/test/011_deploy_state_changes.ts

@@ -0,0 +1,17 @@
+import { DeployFunction } from 'hardhat-deploy/types';
+
+// Used for setting states in contracts according to the currently working environment
+const func: DeployFunction = async function ({ deployments, getNamedAccounts }) {
+  const { execute, log } = deployments;
+  const { deployer } = await getNamedAccounts();
+
+  log('Setting Redistribution state "sampleMaxValue"');
+
+  const sampleMaxValue = '3500000000000000000000000000000000000000000000000000000000000000000000000';
+  await execute('Redistribution', { from: deployer }, 'setSampleMaxValue', sampleMaxValue);
+
+  log('----------------------------------------------------');
+};
+
+export default func;
+func.tags = ['main', 'state_changes'];

src/Redistribution.sol

@@ -136,6 +136,10 @@ contract Redistribution is AccessControl, Pausable {
     uint8 private penaltyMultiplierDisagreement = 1;
     uint8 private penaltyMultiplierNonRevealed = 2;
 
+    // alpha=0.097612 beta=0.0716570 k=16
+    uint256 private sampleMaxValue =
+        1284401000000000000000000000000000000000000000000000000000000000000000000;
+
     // The reveal of the winner of the last round.
     Reveal public winner;
 
@@ -145,10 +149,6 @@ contract Redistribution is AccessControl, Pausable {
     // The miniumum stake allowed to be staked using the Staking contract.
     uint64 private constant MIN_STAKE = 100000000000000000;
 
-    // alpha=0.097612 beta=0.0716570 k=16
-    uint256 private constant SAMPLE_MAX_VALUE =
-        1284401000000000000000000000000000000000000000000000000000000000000000000;
-
     // Maximum value of the keccack256 hash.
     bytes32 private constant MAX_H = 0x00000000000000000000000000000000ffffffffffffffffffffffffffffffff;
 
@@ -467,6 +467,8 @@ contract Redistribution is AccessControl, Pausable {
             entryProofLast.proofSegments[0]
         );
 
+        estimateSize(entryProofLast.proofSegments[0]);
+
         PostageContract.withdraw(winnerSelected.owner);
         emit WinnerSelected(winnerSelected);
         emit ChunkCount(PostageContract.validChunkCount());
@@ -625,6 +627,17 @@ contract Redistribution is AccessControl, Pausable {
         penaltyMultiplierNonRevealed = _penaltyMultiplierNonRevealed;
     }
 
+    /**
+     * @notice changes the max sample value used for reserve estimation
+     */
+    function setSampleMaxValue(uint256 _sampleMaxValue) external {
+        if (!hasRole(DEFAULT_ADMIN_ROLE, msg.sender)) {
+            revert NotAdmin();
+        }
+
+        sampleMaxValue = _sampleMaxValue;
+    }
+
     /**
      * @notice Updates the source of randomness. Uses block.difficulty in pre-merge chains, this is substituted
      * to block.prevrandao in post merge chains.
@@ -1091,12 +1104,10 @@ contract Redistribution is AccessControl, Pausable {
                 revert LastElementCheckFailed();
             }
         }
-
-        estimateSize(trALast);
     }
 
-    function estimateSize(bytes32 trALast) internal pure {
-        if (uint256(trALast) >= SAMPLE_MAX_VALUE) {
+    function estimateSize(bytes32 trALast) internal view {
+        if (uint256(trALast) >= sampleMaxValue) {
             revert ReserveCheckFailed(trALast);
         }
     }