add initialize2 for rollup

morph-l2 · Nov 20, 2024 · 836e4a4 · 836e4a4
1 parent c56fb93
commit 836e4a4
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/contracts/contracts/l1/rollup/Rollup.sol b/contracts/contracts/l1/rollup/Rollup.sol
@@ -174,6 +174,15 @@ contract Rollup is IRollup, OwnableUpgradeable, PausableUpgradeable {
         emit UpdateProofRewardPercent(0, _proofRewardPercent);
     }
 
+    function initialize2(bytes32 _prevStateRoot) external reinitializer(2) {
+        require(_getInitializedVersion() == 2, "must have initialized!");
+        require(_prevStateRoot != bytes32(0), "can not set stateroot with bytes32(0)!");
+
+        if (committedStateRoots[lastCommittedBatchIndex] == bytes32(0)) {
+            committedStateRoots[lastCommittedBatchIndex] = _prevStateRoot;
+        }
+    }
+
     /************************
      * Restricted Functions *
      ************************/