erigon-pulse/cmd/evm/staterunner.go
ledgerwatch ee83447fa1
CL lightclient to create memdb in the tmpdir which gets cleaned up on… (#6829)
… startup

---------

Co-authored-by: Alexey Sharp <alexeysharp@Alexeys-iMac.local>
Co-authored-by: Alex Sharp <alexsharp@Alexs-MacBook-Pro-2.local>
2023-02-11 20:44:51 +00:00

186 lines
5.3 KiB
Go

// Copyright 2017 The go-ethereum Authors
// This file is part of go-ethereum.
//
// go-ethereum is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// go-ethereum is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with go-ethereum. If not, see <http://www.gnu.org/licenses/>.
package main
import (
"context"
"encoding/json"
"errors"
"fmt"
"os"
libcommon "github.com/ledgerwatch/erigon-lib/common"
"github.com/ledgerwatch/erigon-lib/kv/memdb"
"github.com/ledgerwatch/log/v3"
"github.com/urfave/cli/v2"
"github.com/ledgerwatch/erigon/core/state"
"github.com/ledgerwatch/erigon/core/vm"
"github.com/ledgerwatch/erigon/eth/tracers/logger"
"github.com/ledgerwatch/erigon/tests"
"github.com/ledgerwatch/erigon/turbo/trie"
)
var stateTestCommand = cli.Command{
Action: stateTestCmd,
Name: "statetest",
Usage: "executes the given state tests",
ArgsUsage: "<file>",
}
// StatetestResult contains the execution status after running a state test, any
// error that might have occurred and a dump of the final state if requested.
type StatetestResult struct {
Name string `json:"name"`
Pass bool `json:"pass"`
Root *libcommon.Hash `json:"stateRoot,omitempty"`
Fork string `json:"fork"`
Error string `json:"error,omitempty"`
State *state.Dump `json:"state,omitempty"`
}
func stateTestCmd(ctx *cli.Context) error {
if len(ctx.Args().First()) == 0 {
return errors.New("path-to-test argument required")
}
// Configure the go-ethereum logger
log.Root().SetHandler(log.LvlFilterHandler(log.LvlDebug, log.StderrHandler))
// Configure the EVM logger
config := &logger.LogConfig{
DisableMemory: ctx.Bool(DisableMemoryFlag.Name),
DisableStack: ctx.Bool(DisableStackFlag.Name),
DisableStorage: ctx.Bool(DisableStorageFlag.Name),
DisableReturnData: ctx.Bool(DisableReturnDataFlag.Name),
}
var (
tracer vm.EVMLogger
debugger *logger.StructLogger
)
switch {
case ctx.Bool(MachineFlag.Name):
tracer = logger.NewJSONLogger(config, os.Stderr)
case ctx.Bool(DebugFlag.Name):
debugger = logger.NewStructLogger(config)
tracer = debugger
default:
debugger = logger.NewStructLogger(config)
}
// Load the test content from the input file
src, err := os.ReadFile(ctx.Args().First())
if err != nil {
return err
}
var stateTests map[string]tests.StateTest
if err = json.Unmarshal(src, &stateTests); err != nil {
return err
}
// Iterate over all the stateTests, run them and aggregate the results
results, err := aggregateResultsFromStateTests(ctx, stateTests, tracer, debugger)
if err != nil {
return err
}
out, _ := json.MarshalIndent(results, "", " ")
fmt.Println(string(out))
return nil
}
func aggregateResultsFromStateTests(
ctx *cli.Context,
stateTests map[string]tests.StateTest,
tracer vm.EVMLogger,
debugger *logger.StructLogger,
) ([]StatetestResult, error) {
// Iterate over all the stateTests, run them and aggregate the results
cfg := vm.Config{
Tracer: tracer,
Debug: ctx.Bool(DebugFlag.Name) || ctx.Bool(MachineFlag.Name),
}
db := memdb.New("")
defer db.Close()
tx, txErr := db.BeginRw(context.Background())
if txErr != nil {
return nil, txErr
}
defer tx.Rollback()
results := make([]StatetestResult, 0, len(stateTests))
for key, test := range stateTests {
for _, st := range test.Subtests() {
// Run the test and aggregate the result
result := &StatetestResult{Name: key, Fork: st.Fork, Pass: true}
var root libcommon.Hash
var calcRootErr error
statedb, err := test.Run(tx, st, cfg)
// print state root for evmlab tracing
root, calcRootErr = trie.CalcRoot("", tx)
if err == nil && calcRootErr != nil {
err = calcRootErr
}
if err != nil {
// Test failed, mark as so and dump any state to aid debugging
result.Pass, result.Error = false, err.Error()
}
/*
if result.Error != "" {
if ctx.GlobalBool(DumpFlag.Name) && statedb != nil {
tx, err1 := tds.Database().Begin(context.Background(), ethdb.RO)
if err1 != nil {
return fmt.Errorf("transition cannot open tx: %v", err1)
}
dump := state.NewDumper(tx, tds.GetBlockNr()).DefaultRawDump()
tx.Rollback()
result.State = &dump
}
}
*/
// print state root for evmlab tracing
if statedb != nil {
result.Root = &root
if ctx.Bool(MachineFlag.Name) {
_, printErr := fmt.Fprintf(os.Stderr, "{\"stateRoot\": \"%#x\"}\n", root.Bytes())
if printErr != nil {
log.Warn("Failed to write to stderr", "err", printErr)
}
}
}
results = append(results, *result)
// Print any structured logs collected
if ctx.Bool(DebugFlag.Name) {
if debugger != nil {
_, printErr := fmt.Fprintln(os.Stderr, "#### TRACE ####")
if printErr != nil {
log.Warn("Failed to write to stderr", "err", printErr)
}
logger.WriteTrace(os.Stderr, debugger.StructLogs())
}
}
}
}
return results, nil
}