forked from mirror/go-ethereum
eth/tracers: add golang 4byte tracer (#23882)
* native 4byte tracer * Update eth/tracers/native/4byte.go Co-authored-by: Martin Holst Swende <martin@swende.se> * Update eth/tracers/native/4byte.go Co-authored-by: Martin Holst Swende <martin@swende.se> * goimports * eth/tracers: make 4byte tracer not care about create Co-authored-by: Martin Holst Swende <martin@swende.se>verkle/onleaf
Ward Bradt
3 years ago
committed by
GitHub
parent
5358e491f3
commit
e9294a7fe9
@ -0,0 +1,148 @@ |
|||||||
|
// Copyright 2021 The go-ethereum Authors
|
||||||
|
// This file is part of the go-ethereum library.
|
||||||
|
//
|
||||||
|
// The go-ethereum library is free software: you can redistribute it and/or modify
|
||||||
|
// it under the terms of the GNU Lesser General Public License as published by
|
||||||
|
// the Free Software Foundation, either version 3 of the License, or
|
||||||
|
// (at your option) any later version.
|
||||||
|
//
|
||||||
|
// The go-ethereum library 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 Lesser General Public License for more details.
|
||||||
|
//
|
||||||
|
// You should have received a copy of the GNU Lesser General Public License
|
||||||
|
// along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.
|
||||||
|
|
||||||
|
package native |
||||||
|
|
||||||
|
import ( |
||||||
|
"encoding/json" |
||||||
|
"math/big" |
||||||
|
"strconv" |
||||||
|
"sync/atomic" |
||||||
|
"time" |
||||||
|
|
||||||
|
"github.com/ethereum/go-ethereum/common" |
||||||
|
"github.com/ethereum/go-ethereum/core/vm" |
||||||
|
"github.com/ethereum/go-ethereum/eth/tracers" |
||||||
|
) |
||||||
|
|
||||||
|
func init() { |
||||||
|
register("4byte", newFourByteTracer) |
||||||
|
} |
||||||
|
|
||||||
|
// fourByteTracer searches for 4byte-identifiers, and collects them for post-processing.
|
||||||
|
// It collects the methods identifiers along with the size of the supplied data, so
|
||||||
|
// a reversed signature can be matched against the size of the data.
|
||||||
|
//
|
||||||
|
// Example:
|
||||||
|
// > debug.traceTransaction( "0x214e597e35da083692f5386141e69f47e973b2c56e7a8073b1ea08fd7571e9de", {tracer: "4byte"})
|
||||||
|
// {
|
||||||
|
// 0x27dc297e-128: 1,
|
||||||
|
// 0x38cc4831-0: 2,
|
||||||
|
// 0x524f3889-96: 1,
|
||||||
|
// 0xadf59f99-288: 1,
|
||||||
|
// 0xc281d19e-0: 1
|
||||||
|
// }
|
||||||
|
type fourByteTracer struct { |
||||||
|
env *vm.EVM |
||||||
|
ids map[string]int // ids aggregates the 4byte ids found
|
||||||
|
interrupt uint32 // Atomic flag to signal execution interruption
|
||||||
|
reason error // Textual reason for the interruption
|
||||||
|
activePrecompiles []common.Address // Updated on CaptureStart based on given rules
|
||||||
|
} |
||||||
|
|
||||||
|
// newFourByteTracer returns a native go tracer which collects
|
||||||
|
// 4 byte-identifiers of a tx, and implements vm.EVMLogger.
|
||||||
|
func newFourByteTracer() tracers.Tracer { |
||||||
|
t := &fourByteTracer{ |
||||||
|
ids: make(map[string]int), |
||||||
|
} |
||||||
|
return t |
||||||
|
} |
||||||
|
|
||||||
|
// isPrecompiled returns whether the addr is a precompile. Logic borrowed from newJsTracer in eth/tracers/js/tracer.go
|
||||||
|
func (t *fourByteTracer) isPrecompiled(addr common.Address) bool { |
||||||
|
for _, p := range t.activePrecompiles { |
||||||
|
if p == addr { |
||||||
|
return true |
||||||
|
} |
||||||
|
} |
||||||
|
return false |
||||||
|
} |
||||||
|
|
||||||
|
// store saves the given identifier and datasize.
|
||||||
|
func (t *fourByteTracer) store(id []byte, size int) { |
||||||
|
key := bytesToHex(id) + "-" + strconv.Itoa(size) |
||||||
|
t.ids[key] += 1 |
||||||
|
} |
||||||
|
|
||||||
|
// CaptureStart implements the EVMLogger interface to initialize the tracing operation.
|
||||||
|
func (t *fourByteTracer) CaptureStart(env *vm.EVM, from common.Address, to common.Address, create bool, input []byte, gas uint64, value *big.Int) { |
||||||
|
t.env = env |
||||||
|
|
||||||
|
// Update list of precompiles based on current block
|
||||||
|
rules := env.ChainConfig().Rules(env.Context.BlockNumber) |
||||||
|
t.activePrecompiles = vm.ActivePrecompiles(rules) |
||||||
|
|
||||||
|
// Save the outer calldata also
|
||||||
|
if len(input) >= 4 { |
||||||
|
t.store(input[0:4], len(input)-4) |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// CaptureState implements the EVMLogger interface to trace a single step of VM execution.
|
||||||
|
func (t *fourByteTracer) CaptureState(pc uint64, op vm.OpCode, gas, cost uint64, scope *vm.ScopeContext, rData []byte, depth int, err error) { |
||||||
|
} |
||||||
|
|
||||||
|
// CaptureEnter is called when EVM enters a new scope (via call, create or selfdestruct).
|
||||||
|
func (t *fourByteTracer) CaptureEnter(op vm.OpCode, from common.Address, to common.Address, input []byte, gas uint64, value *big.Int) { |
||||||
|
// Skip if tracing was interrupted
|
||||||
|
if atomic.LoadUint32(&t.interrupt) > 0 { |
||||||
|
t.env.Cancel() |
||||||
|
return |
||||||
|
} |
||||||
|
if len(input) < 4 { |
||||||
|
return |
||||||
|
} |
||||||
|
// primarily we want to avoid CREATE/CREATE2/SELFDESTRUCT
|
||||||
|
if op != vm.DELEGATECALL && op != vm.STATICCALL && |
||||||
|
op != vm.CALL && op != vm.CALLCODE { |
||||||
|
return |
||||||
|
} |
||||||
|
// Skip any pre-compile invocations, those are just fancy opcodes
|
||||||
|
if t.isPrecompiled(to) { |
||||||
|
return |
||||||
|
} |
||||||
|
t.store(input[0:4], len(input)-4) |
||||||
|
} |
||||||
|
|
||||||
|
// CaptureExit is called when EVM exits a scope, even if the scope didn't
|
||||||
|
// execute any code.
|
||||||
|
func (t *fourByteTracer) CaptureExit(output []byte, gasUsed uint64, err error) { |
||||||
|
} |
||||||
|
|
||||||
|
// CaptureFault implements the EVMLogger interface to trace an execution fault.
|
||||||
|
func (t *fourByteTracer) CaptureFault(pc uint64, op vm.OpCode, gas, cost uint64, scope *vm.ScopeContext, depth int, err error) { |
||||||
|
} |
||||||
|
|
||||||
|
// CaptureEnd is called after the call finishes to finalize the tracing.
|
||||||
|
func (t *fourByteTracer) CaptureEnd(output []byte, gasUsed uint64, _ time.Duration, err error) { |
||||||
|
} |
||||||
|
|
||||||
|
// GetResult returns the json-encoded nested list of call traces, and any
|
||||||
|
// error arising from the encoding or forceful termination (via `Stop`).
|
||||||
|
func (t *fourByteTracer) GetResult() (json.RawMessage, error) { |
||||||
|
res, err := json.Marshal(t.ids) |
||||||
|
if err != nil { |
||||||
|
return nil, err |
||||||
|
} |
||||||
|
return res, t.reason |
||||||
|
} |
||||||
|
|
||||||
|
// Stop terminates execution of the tracer at the first opportune moment.
|
||||||
|
func (t *fourByteTracer) Stop(err error) { |
||||||
|
t.reason = err |
||||||
|
atomic.StoreUint32(&t.interrupt, 1) |
||||||
|
} |
||||||
Loading…
Reference in new issue