forked from ethereum/go-ethereum
-
Notifications
You must be signed in to change notification settings - Fork 6
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request ethereum#191 from OffchainLabs/recreate-state-for-…
…rpcs Recreate state for rpcs
- Loading branch information
Showing
4 changed files
with
145 additions
and
47 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,103 @@ | ||
| package arbitrum | ||
|
|
||
| import ( | ||
| "context" | ||
| "fmt" | ||
|
|
||
| "github.com/ethereum/go-ethereum/common" | ||
| "github.com/ethereum/go-ethereum/core" | ||
| "github.com/ethereum/go-ethereum/core/state" | ||
| "github.com/ethereum/go-ethereum/core/types" | ||
| "github.com/ethereum/go-ethereum/core/vm" | ||
| "github.com/pkg/errors" | ||
| ) | ||
|
|
||
| var ( | ||
| ErrDepthLimitExceeded = errors.New("state recreation l2 gas depth limit exceeded") | ||
| ) | ||
|
|
||
| type StateBuildingLogFunction func(targetHeader, header *types.Header, hasState bool) | ||
| type StateForHeaderFunction func(header *types.Header) (*state.StateDB, error) | ||
|
|
||
| // finds last available state and header checking it first for targetHeader then looking backwards | ||
| // if maxDepthInL2Gas is positive, it constitutes a limit for cumulative l2 gas used of the traversed blocks | ||
| // else if maxDepthInL2Gas is -1, the traversal depth is not limited | ||
| // otherwise only targetHeader state is checked and no search is performed | ||
| func FindLastAvailableState(ctx context.Context, bc *core.BlockChain, stateFor StateForHeaderFunction, targetHeader *types.Header, logFunc StateBuildingLogFunction, maxDepthInL2Gas int64) (*state.StateDB, *types.Header, error) { | ||
| genesis := bc.Config().ArbitrumChainParams.GenesisBlockNum | ||
| currentHeader := targetHeader | ||
| var state *state.StateDB | ||
| var err error | ||
| var l2GasUsed uint64 | ||
| for ctx.Err() == nil { | ||
| lastHeader := currentHeader | ||
| state, err = stateFor(currentHeader) | ||
| if err == nil { | ||
| break | ||
| } | ||
| if maxDepthInL2Gas > 0 { | ||
| receipts := bc.GetReceiptsByHash(currentHeader.Hash()) | ||
| if receipts == nil { | ||
| return nil, lastHeader, fmt.Errorf("failed to get receipts for hash %v", currentHeader.Hash()) | ||
| } | ||
| for _, receipt := range receipts { | ||
| l2GasUsed += receipt.GasUsed - receipt.GasUsedForL1 | ||
| } | ||
| if l2GasUsed > uint64(maxDepthInL2Gas) { | ||
| return nil, lastHeader, ErrDepthLimitExceeded | ||
| } | ||
| } else if maxDepthInL2Gas != InfiniteMaxRecreateStateDepth { | ||
| return nil, lastHeader, err | ||
| } | ||
| if logFunc != nil { | ||
| logFunc(targetHeader, currentHeader, false) | ||
| } | ||
| if currentHeader.Number.Uint64() <= genesis { | ||
| return nil, lastHeader, errors.Wrap(err, fmt.Sprintf("moved beyond genesis looking for state %d, genesis %d", targetHeader.Number.Uint64(), genesis)) | ||
| } | ||
| currentHeader = bc.GetHeader(currentHeader.ParentHash, currentHeader.Number.Uint64()-1) | ||
| if currentHeader == nil { | ||
| return nil, lastHeader, fmt.Errorf("chain doesn't contain parent of block %d hash %v", lastHeader.Number, lastHeader.Hash()) | ||
| } | ||
| } | ||
| return state, currentHeader, ctx.Err() | ||
| } | ||
|
|
||
| func AdvanceStateByBlock(ctx context.Context, bc *core.BlockChain, state *state.StateDB, targetHeader *types.Header, blockToRecreate uint64, prevBlockHash common.Hash, logFunc StateBuildingLogFunction) (*state.StateDB, *types.Block, error) { | ||
| block := bc.GetBlockByNumber(blockToRecreate) | ||
| if block == nil { | ||
| return nil, nil, fmt.Errorf("block not found while recreating: %d", blockToRecreate) | ||
| } | ||
| if block.ParentHash() != prevBlockHash { | ||
| return nil, nil, fmt.Errorf("reorg detected: number %d expectedPrev: %v foundPrev: %v", blockToRecreate, prevBlockHash, block.ParentHash()) | ||
| } | ||
| if logFunc != nil { | ||
| logFunc(targetHeader, block.Header(), true) | ||
| } | ||
| _, _, _, err := bc.Processor().Process(block, state, vm.Config{}) | ||
| if err != nil { | ||
| return nil, nil, fmt.Errorf("failed recreating state for block %d : %w", blockToRecreate, err) | ||
| } | ||
| return state, block, nil | ||
| } | ||
|
|
||
| func AdvanceStateUpToBlock(ctx context.Context, bc *core.BlockChain, state *state.StateDB, targetHeader *types.Header, lastAvailableHeader *types.Header, logFunc StateBuildingLogFunction) (*state.StateDB, error) { | ||
| returnedBlockNumber := targetHeader.Number.Uint64() | ||
| blockToRecreate := lastAvailableHeader.Number.Uint64() + 1 | ||
| prevHash := lastAvailableHeader.Hash() | ||
| for ctx.Err() == nil { | ||
| state, block, err := AdvanceStateByBlock(ctx, bc, state, targetHeader, blockToRecreate, prevHash, logFunc) | ||
| if err != nil { | ||
| return nil, err | ||
| } | ||
| prevHash = block.Hash() | ||
| if blockToRecreate >= returnedBlockNumber { | ||
| if block.Hash() != targetHeader.Hash() { | ||
| return nil, fmt.Errorf("blockHash doesn't match when recreating number: %d expected: %v got: %v", blockToRecreate, targetHeader.Hash(), block.Hash()) | ||
| } | ||
| return state, nil | ||
| } | ||
| blockToRecreate++ | ||
| } | ||
| return nil, ctx.Err() | ||
| } |