Add postStateRoot and withdrawRoot when rebuilding batch cache (#412)

node/core/batch.go

@@ -107,6 +107,8 @@ func (e *Executor) CalculateCapWithProposalBlock(currentBlockBytes []byte, curre
 		var l1TxHashes []common.Hash
 		var totalL1MessagePopped = parentBatchHeader.TotalL1MessagePopped
 		var lastHeightBeforeCurrentBatch uint64
+		var lastBlockStateRoot common.Hash
+		var lastBlockWithdrawRoot common.Hash
 		var l2TxNum int
 
 		for i, blockBz := range blocks {
@@ -119,6 +121,11 @@ func (e *Executor) CalculateCapWithProposalBlock(currentBlockBytes []byte, curre
 				lastHeightBeforeCurrentBatch = wBlock.Number - 1
 			}
 
+			if i == len(blocks)-1 {
+				lastBlockStateRoot = wBlock.StateRoot
+				lastBlockWithdrawRoot = wBlock.WithdrawTrieRoot
+			}
+
 			totalL1MessagePoppedBefore := totalL1MessagePopped
 			txsPayload, l1TxHashes, totalL1MessagePopped, skippedBitmap, l2TxNum, err = ParsingTxs(transactions[i], parentBatchHeader.TotalL1MessagePopped, totalL1MessagePoppedBefore, skippedBitmap)
 			if err != nil {
@@ -148,6 +155,8 @@ func (e *Executor) CalculateCapWithProposalBlock(currentBlockBytes []byte, curre
 			return false, 0, err
 		}
 		e.batchingCache.prevStateRoot = header.Root
+		e.batchingCache.postStateRoot = lastBlockStateRoot
+		e.batchingCache.withdrawRoot = lastBlockWithdrawRoot
 
 		// initialize latest batch index
 		e.metrics.BatchIndex.Set(float64(e.batchingCache.parentBatchHeader.BatchIndex))