mirror of
https://github.com/ethereum/go-ethereum.git
synced 2026-07-26 06:36:43 +00:00
fix pre: separate implementation for TraceMany (#26)
This commit is contained in:
Zhang Guyu
GitHub
parent
cccf0b020d
commit
25a9ed71e2
1 changed files with 158 additions and 107 deletions
|
|
@ -1,25 +1,178 @@
|
|||
package eth
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"context"
|
||||
"fmt"
|
||||
"hash"
|
||||
"math"
|
||||
"math/big"
|
||||
|
||||
txtrace "github.com/DeBankDeFi/etherlib/pkg/txtracev1"
|
||||
"github.com/ethereum/go-ethereum/common"
|
||||
"github.com/ethereum/go-ethereum/common/hexutil"
|
||||
"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/ethereum/go-ethereum/internal/ethapi"
|
||||
"github.com/ethereum/go-ethereum/rpc"
|
||||
|
||||
txtrace "github.com/DeBankDeFi/etherlib/pkg/txtracev1"
|
||||
// "github.com/DeBankDeFi/eth/txtrace"
|
||||
"golang.org/x/crypto/sha3"
|
||||
)
|
||||
|
||||
type helpHash struct {
|
||||
hashed hash.Hash
|
||||
}
|
||||
|
||||
func newHash() *helpHash {
|
||||
|
||||
return &helpHash{hashed: sha3.NewLegacyKeccak256()}
|
||||
}
|
||||
|
||||
func (h *helpHash) Reset() {
|
||||
h.hashed.Reset()
|
||||
}
|
||||
|
||||
func (h *helpHash) Update(key, val []byte) {
|
||||
h.hashed.Write(key)
|
||||
h.hashed.Write(val)
|
||||
}
|
||||
|
||||
func (h *helpHash) Hash() common.Hash {
|
||||
return common.BytesToHash(h.hashed.Sum(nil))
|
||||
}
|
||||
|
||||
type PreExecTx struct {
|
||||
ChainId *big.Int
|
||||
From, To, Data, Value, Gas, GasPrice, Nonce string
|
||||
}
|
||||
|
||||
type preData struct {
|
||||
block *types.Block
|
||||
tx *types.Transaction
|
||||
msg types.Message
|
||||
stateDb *state.StateDB
|
||||
header *types.Header
|
||||
}
|
||||
|
||||
// PreExecAPI provides pre exec info for rpc
|
||||
type PreExecAPI struct {
|
||||
e *Ethereum
|
||||
}
|
||||
|
||||
func NewPreExecAPI(e *Ethereum) *PreExecAPI {
|
||||
return &PreExecAPI{e: e}
|
||||
}
|
||||
|
||||
func (api *PreExecAPI) getBlockAndMsg(origin *PreExecTx, number *big.Int) (*types.Block, types.Message) {
|
||||
fromAddr := common.HexToAddress(origin.From)
|
||||
toAddr := common.HexToAddress(origin.To)
|
||||
|
||||
tx := types.NewTx(&types.LegacyTx{
|
||||
Nonce: hexutil.MustDecodeUint64(origin.Nonce),
|
||||
To: &toAddr,
|
||||
Value: hexutil.MustDecodeBig(origin.Value),
|
||||
Gas: hexutil.MustDecodeUint64(origin.Gas),
|
||||
GasPrice: hexutil.MustDecodeBig(origin.GasPrice),
|
||||
Data: hexutil.MustDecode(origin.Data),
|
||||
})
|
||||
|
||||
number.Add(number, big.NewInt(1))
|
||||
block := types.NewBlock(
|
||||
&types.Header{Number: number},
|
||||
[]*types.Transaction{tx}, nil, nil, newHash())
|
||||
|
||||
msg := types.NewMessage(
|
||||
fromAddr,
|
||||
&toAddr,
|
||||
hexutil.MustDecodeUint64(origin.Nonce),
|
||||
hexutil.MustDecodeBig(origin.Value),
|
||||
hexutil.MustDecodeUint64(origin.Gas),
|
||||
hexutil.MustDecodeBig(origin.GasPrice),
|
||||
tx.GasFeeCap(),
|
||||
tx.GasTipCap(),
|
||||
hexutil.MustDecode(origin.Data),
|
||||
nil, false, true,
|
||||
)
|
||||
return block, msg
|
||||
}
|
||||
|
||||
func (api *PreExecAPI) prepareData(ctx context.Context, origin *PreExecTx) (*preData, error) {
|
||||
var (
|
||||
d preData
|
||||
err error
|
||||
)
|
||||
bc := api.e.blockchain
|
||||
d.header, err = api.e.APIBackend.HeaderByNumber(ctx, rpc.LatestBlockNumber)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
latestNumber := d.header.Number
|
||||
parent := api.e.blockchain.GetBlockByNumber(latestNumber.Uint64())
|
||||
d.stateDb, err = state.New(parent.Header().Root, bc.StateCache(), bc.Snapshots())
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
d.block, d.msg = api.getBlockAndMsg(origin, latestNumber)
|
||||
d.tx = d.block.Transactions()[0]
|
||||
return &d, nil
|
||||
}
|
||||
|
||||
func (api *PreExecAPI) GetLogs(ctx context.Context, origin *PreExecTx) (*types.Receipt, error) {
|
||||
var (
|
||||
bc = api.e.blockchain
|
||||
)
|
||||
d, err := api.prepareData(ctx, origin)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
gas := d.tx.Gas()
|
||||
gp := new(core.GasPool).AddGas(gas)
|
||||
|
||||
d.stateDb.Prepare(d.tx.Hash(), 0)
|
||||
receipt, err := core.ApplyTransactionForPreExec(
|
||||
bc.Config(), bc, nil, gp, d.stateDb, d.header, d.tx, d.msg, &gas, *bc.GetVMConfig())
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return receipt, receipt.Err
|
||||
}
|
||||
|
||||
// TraceTransaction tracing pre-exec transaction object.
|
||||
func (api *PreExecAPI) TraceTransaction(ctx context.Context, origin *PreExecTx) (interface{}, error) {
|
||||
var (
|
||||
bc = api.e.blockchain
|
||||
tracer *txtrace.OeTracer
|
||||
err error
|
||||
)
|
||||
d, err := api.prepareData(ctx, origin)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
txContext := core.NewEVMTxContext(d.msg)
|
||||
txIndex := 0
|
||||
|
||||
tracer = txtrace.NewOeTracer(nil)
|
||||
// Run the transaction with tracing enabled.
|
||||
vmenv := vm.NewEVM(core.NewEVMBlockContext(d.header, bc, nil), txContext, d.stateDb, bc.Config(), vm.Config{Debug: true, Tracer: tracer})
|
||||
vmenv.Context.BaseFee = big.NewInt(0)
|
||||
|
||||
// Call Prepare to clear out the statedb access list
|
||||
d.stateDb.Prepare(d.tx.Hash(), txIndex)
|
||||
|
||||
tracer.SetMessage(d.block.Number(), d.block.Hash(), d.tx.Hash(), uint(txIndex), d.msg.From(), d.msg.To(), *d.msg.Value())
|
||||
|
||||
_, err = core.ApplyMessage(vmenv, d.msg, new(core.GasPool).AddGas(d.msg.Gas()))
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("tracing failed: %v", err)
|
||||
}
|
||||
|
||||
tracer.Finalize()
|
||||
return tracer.GetResult(), nil
|
||||
}
|
||||
|
||||
type PreArgs struct {
|
||||
ChainId *big.Int `json:"chainId,omitempty"`
|
||||
From *common.Address `json:"from"`
|
||||
To *common.Address `json:"to"`
|
||||
|
|
@ -33,114 +186,12 @@ type PreExecTx struct {
|
|||
Input *hexutil.Bytes `json:"input"`
|
||||
}
|
||||
|
||||
type PreExecResult struct {
|
||||
type PreResult struct {
|
||||
Trace *[]txtrace.ActionTrace `json:"trace"`
|
||||
Logs []*types.Log `json:"logs"`
|
||||
}
|
||||
|
||||
// PreExecAPI provides pre exec info for rpc
|
||||
type PreExecAPI struct {
|
||||
e *Ethereum
|
||||
}
|
||||
|
||||
func NewPreExecAPI(e *Ethereum) *PreExecAPI {
|
||||
return &PreExecAPI{e: e}
|
||||
}
|
||||
|
||||
func (api *PreExecAPI) GetLogs(ctx context.Context, origin *PreExecTx) (*types.Receipt, error) {
|
||||
state, header, err := api.e.APIBackend.StateAndHeaderByNumberOrHash(ctx, rpc.BlockNumberOrHashWithNumber(rpc.LatestBlockNumber))
|
||||
if state == nil || err != nil {
|
||||
return nil, err
|
||||
}
|
||||
txArgs := ethapi.TransactionArgs{
|
||||
ChainID: (*hexutil.Big)(big.NewInt(1)),
|
||||
From: origin.From,
|
||||
To: origin.To,
|
||||
Gas: origin.Gas,
|
||||
GasPrice: origin.GasPrice,
|
||||
MaxFeePerGas: origin.MaxFeePerGas,
|
||||
MaxPriorityFeePerGas: origin.MaxPriorityFeePerGas,
|
||||
Value: origin.Value,
|
||||
Nonce: origin.Nonce,
|
||||
Data: origin.Data,
|
||||
Input: origin.Input,
|
||||
}
|
||||
// Get a new instance of the EVM.
|
||||
msg, err := txArgs.ToMessage(0, header.BaseFee)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
evm, vmError, err := api.e.APIBackend.GetEVM(ctx, msg, state, header, &vm.Config{NoBaseFee: true})
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
// Execute the message.
|
||||
gp := new(core.GasPool).AddGas(math.MaxUint64)
|
||||
result, err := core.ApplyMessage(evm, msg, gp)
|
||||
if err := vmError(); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("err: %w (supplied gas %d)", err, msg.Gas())
|
||||
}
|
||||
receipt := types.NewReceipt(bytes.NewBufferString("").Bytes(), result.Failed(), result.UsedGas)
|
||||
receipt.GasUsed = result.UsedGas
|
||||
receipt.Logs = state.Logs()
|
||||
receipt.BlockHash = header.Hash()
|
||||
receipt.BlockNumber = header.Number
|
||||
return receipt, nil
|
||||
}
|
||||
|
||||
// TraceTransaction tracing pre-exec transaction object.
|
||||
func (api *PreExecAPI) TraceTransaction(ctx context.Context, origin *PreExecTx) (interface{}, error) {
|
||||
state, header, err := api.e.APIBackend.StateAndHeaderByNumberOrHash(ctx, rpc.BlockNumberOrHashWithNumber(rpc.LatestBlockNumber))
|
||||
if state == nil || err != nil {
|
||||
return nil, err
|
||||
}
|
||||
txArgs := ethapi.TransactionArgs{
|
||||
ChainID: (*hexutil.Big)(big.NewInt(1)),
|
||||
From: origin.From,
|
||||
To: origin.To,
|
||||
Gas: origin.Gas,
|
||||
GasPrice: origin.GasPrice,
|
||||
MaxFeePerGas: origin.MaxFeePerGas,
|
||||
MaxPriorityFeePerGas: origin.MaxPriorityFeePerGas,
|
||||
Value: origin.Value,
|
||||
Nonce: origin.Nonce,
|
||||
Data: origin.Data,
|
||||
Input: origin.Input,
|
||||
}
|
||||
// Get a new instance of the EVM.
|
||||
msg, err := txArgs.ToMessage(0, header.BaseFee)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
tracer := txtrace.NewOeTracer(nil)
|
||||
tracer.SetFrom(msg.From())
|
||||
tracer.SetTo(msg.To())
|
||||
tracer.SetValue(*msg.Value())
|
||||
tracer.SetTxIndex(uint(0))
|
||||
tracer.SetBlockNumber(header.Number)
|
||||
tracer.SetBlockHash(header.Hash())
|
||||
evm, vmError, err := api.e.APIBackend.GetEVM(ctx, msg, state, header, &vm.Config{NoBaseFee: true, Debug: true, Tracer: tracer})
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
// Execute the message.
|
||||
gp := new(core.GasPool).AddGas(math.MaxUint64)
|
||||
result, err := core.ApplyMessage(evm, msg, gp)
|
||||
if err := vmError(); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("err: %w (supplied gas %d)", err, msg.Gas())
|
||||
}
|
||||
tracer.SetGasUsed(result.UsedGas)
|
||||
tracer.Finalize()
|
||||
return tracer.GetResult(), nil
|
||||
}
|
||||
|
||||
func (api *PreExecAPI) TraceMany(ctx context.Context, origins []PreExecTx) (*PreExecResult, error) {
|
||||
func (api *PreExecAPI) TraceMany(ctx context.Context, origins []PreArgs) (*PreResult, error) {
|
||||
state, header, err := api.e.APIBackend.StateAndHeaderByNumberOrHash(ctx, rpc.BlockNumberOrHashWithNumber(rpc.LatestBlockNumber))
|
||||
if state == nil || err != nil {
|
||||
return nil, err
|
||||
|
|
@ -222,5 +273,5 @@ func (api *PreExecAPI) TraceMany(ctx context.Context, origins []PreExecTx) (*Pre
|
|||
}
|
||||
tracer.SetGasUsed(result.UsedGas)
|
||||
tracer.Finalize()
|
||||
return &PreExecResult{Trace: tracer.GetResult(), Logs: state.GetLogs(txHash, header.Hash())}, nil
|
||||
return &PreResult{Trace: tracer.GetResult(), Logs: state.GetLogs(txHash, header.Hash())}, nil
|
||||
}
|
||||
|
|
|
|||
Loading…
Reference in a new issue