forked from cerc-io/plugeth
61 lines
2.3 KiB
Go
61 lines
2.3 KiB
Go
package tracers
|
|
|
|
import (
|
|
"reflect"
|
|
"math/big"
|
|
|
|
"github.com/ethereum/go-ethereum/core/state"
|
|
"github.com/ethereum/go-ethereum/core/vm"
|
|
"github.com/ethereum/go-ethereum/log"
|
|
"github.com/ethereum/go-ethereum/plugins"
|
|
"github.com/ethereum/go-ethereum/plugins/interfaces"
|
|
"github.com/ethereum/go-ethereum/plugins/wrappers"
|
|
"github.com/openrelayxyz/plugeth-utils/core"
|
|
)
|
|
|
|
func GetPluginTracer(pl *plugins.PluginLoader, name string) (func(*state.StateDB, vm.BlockContext) interfaces.TracerResult, bool) {
|
|
tracers := pl.Lookup("Tracers", func(item interface{}) bool {
|
|
_, ok := item.(*map[string]func(core.StateDB) core.TracerResult)
|
|
_, ok2 := item.(*map[string]func(core.StateDB, core.BlockContext) core.TracerResult)
|
|
if !(ok || ok2) {
|
|
log.Warn("Found tracer that did not match type", "tracer", reflect.TypeOf(item))
|
|
}
|
|
return ok || ok2
|
|
})
|
|
|
|
for _, tmap := range tracers {
|
|
if tracerMap, ok := tmap.(*map[string]func(core.StateDB) core.TracerResult); ok {
|
|
if tracer, ok := (*tracerMap)[name]; ok {
|
|
return func(sdb *state.StateDB, vmctx vm.BlockContext) interfaces.TracerResult {
|
|
return wrappers.NewWrappedTracer(tracer(wrappers.NewWrappedStateDB(sdb)))
|
|
}, true
|
|
}
|
|
}
|
|
if tracerMap, ok := tmap.(*map[string]func(core.StateDB, core.BlockContext) core.TracerResult); ok {
|
|
if tracer, ok := (*tracerMap)[name]; ok {
|
|
return func(sdb *state.StateDB, vmctx vm.BlockContext) interfaces.TracerResult {
|
|
return wrappers.NewWrappedTracer(tracer(wrappers.NewWrappedStateDB(sdb), core.BlockContext{
|
|
Coinbase: core.Address(vmctx.Coinbase),
|
|
GasLimit: vmctx.GasLimit,
|
|
BlockNumber: vmctx.BlockNumber,
|
|
// casting int64 zero as bigInt is foundation PluGeth only, it is being done to preserve compatability with other networks
|
|
Time: new(big.Int).SetInt64(int64(vmctx.Time)),
|
|
Difficulty: vmctx.Difficulty,
|
|
BaseFee: vmctx.BaseFee,
|
|
}))
|
|
}, true
|
|
}
|
|
}
|
|
}
|
|
log.Info("Tracer not found", "name", name, "tracers", len(tracers))
|
|
return nil, false
|
|
}
|
|
|
|
func getPluginTracer(name string) (func(*state.StateDB, vm.BlockContext) interfaces.TracerResult, bool) {
|
|
if plugins.DefaultPluginLoader == nil {
|
|
log.Warn("Attempting GetPluginTracer, but default PluginLoader has not been initialized")
|
|
return nil, false
|
|
}
|
|
return GetPluginTracer(plugins.DefaultPluginLoader, name)
|
|
}
|