Add initialize2 for rollup to set state root (#641)

Co-authored-by: Agnar Song <[email protected]>
morph-l2 · Nov 20, 2024 · 375d1f0 · 375d1f0
1 parent 278ac64
commit 375d1f0
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 state root with bytes32(0)!");
+
+        if (committedStateRoots[lastCommittedBatchIndex] == bytes32(0)) {
+            committedStateRoots[lastCommittedBatchIndex] = _prevStateRoot;
+        }
+    }
+
     /************************
      * Restricted Functions *
      ************************/