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

Co-authored-by: Agnar Song <[email protected]>
morph-l2 · Nov 20, 2024 · 5eafefe · 5eafefe
1 parent c56fb93
commit 5eafefe
Show file tree

Hide file tree

Showing 4 changed files with 34 additions and 4 deletions.
diff --git a/bindings/bin/rollup_deployed.hex b/bindings/bin/rollup_deployed.hex
diff --git a/bindings/bindings/rollup.go b/bindings/bindings/rollup.go
diff --git a/bindings/bindings/rollup_more.go b/bindings/bindings/rollup_more.go
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 *
      ************************/