307 lines
		
	
	
		
			11 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			307 lines
		
	
	
		
			11 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
| // Copyright 2020 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 t8ntool
 | |
| 
 | |
| import (
 | |
| 	"fmt"
 | |
| 	"math/big"
 | |
| 	"os"
 | |
| 
 | |
| 	"github.com/ethereum/go-ethereum/common"
 | |
| 	"github.com/ethereum/go-ethereum/common/math"
 | |
| 	"github.com/ethereum/go-ethereum/consensus/ethash"
 | |
| 	"github.com/ethereum/go-ethereum/consensus/misc"
 | |
| 	"github.com/ethereum/go-ethereum/core"
 | |
| 	"github.com/ethereum/go-ethereum/core/rawdb"
 | |
| 	"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/crypto"
 | |
| 	"github.com/ethereum/go-ethereum/ethdb"
 | |
| 	"github.com/ethereum/go-ethereum/log"
 | |
| 	"github.com/ethereum/go-ethereum/params"
 | |
| 	"github.com/ethereum/go-ethereum/rlp"
 | |
| 	"github.com/ethereum/go-ethereum/trie"
 | |
| 	"golang.org/x/crypto/sha3"
 | |
| )
 | |
| 
 | |
| type Prestate struct {
 | |
| 	Env stEnv             `json:"env"`
 | |
| 	Pre core.GenesisAlloc `json:"pre"`
 | |
| }
 | |
| 
 | |
| // ExecutionResult 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 ExecutionResult struct {
 | |
| 	StateRoot   common.Hash           `json:"stateRoot"`
 | |
| 	TxRoot      common.Hash           `json:"txRoot"`
 | |
| 	ReceiptRoot common.Hash           `json:"receiptsRoot"`
 | |
| 	LogsHash    common.Hash           `json:"logsHash"`
 | |
| 	Bloom       types.Bloom           `json:"logsBloom"        gencodec:"required"`
 | |
| 	Receipts    types.Receipts        `json:"receipts"`
 | |
| 	Rejected    []*rejectedTx         `json:"rejected,omitempty"`
 | |
| 	Difficulty  *math.HexOrDecimal256 `json:"currentDifficulty" gencodec:"required"`
 | |
| 	GasUsed     math.HexOrDecimal64   `json:"gasUsed"`
 | |
| }
 | |
| 
 | |
| type ommer struct {
 | |
| 	Delta   uint64         `json:"delta"`
 | |
| 	Address common.Address `json:"address"`
 | |
| }
 | |
| 
 | |
| //go:generate gencodec -type stEnv -field-override stEnvMarshaling -out gen_stenv.go
 | |
| type stEnv struct {
 | |
| 	Coinbase         common.Address                      `json:"currentCoinbase"   gencodec:"required"`
 | |
| 	Difficulty       *big.Int                            `json:"currentDifficulty"`
 | |
| 	ParentDifficulty *big.Int                            `json:"parentDifficulty"`
 | |
| 	GasLimit         uint64                              `json:"currentGasLimit"   gencodec:"required"`
 | |
| 	Number           uint64                              `json:"currentNumber"     gencodec:"required"`
 | |
| 	Timestamp        uint64                              `json:"currentTimestamp"  gencodec:"required"`
 | |
| 	ParentTimestamp  uint64                              `json:"parentTimestamp,omitempty"`
 | |
| 	BlockHashes      map[math.HexOrDecimal64]common.Hash `json:"blockHashes,omitempty"`
 | |
| 	Ommers           []ommer                             `json:"ommers,omitempty"`
 | |
| 	BaseFee          *big.Int                            `json:"currentBaseFee,omitempty"`
 | |
| 	ParentUncleHash  common.Hash                         `json:"parentUncleHash"`
 | |
| }
 | |
| 
 | |
| type stEnvMarshaling struct {
 | |
| 	Coinbase         common.UnprefixedAddress
 | |
| 	Difficulty       *math.HexOrDecimal256
 | |
| 	ParentDifficulty *math.HexOrDecimal256
 | |
| 	GasLimit         math.HexOrDecimal64
 | |
| 	Number           math.HexOrDecimal64
 | |
| 	Timestamp        math.HexOrDecimal64
 | |
| 	ParentTimestamp  math.HexOrDecimal64
 | |
| 	BaseFee          *math.HexOrDecimal256
 | |
| }
 | |
| 
 | |
| type rejectedTx struct {
 | |
| 	Index int    `json:"index"`
 | |
| 	Err   string `json:"error"`
 | |
| }
 | |
| 
 | |
| // Apply applies a set of transactions to a pre-state
 | |
| func (pre *Prestate) Apply(vmConfig vm.Config, chainConfig *params.ChainConfig,
 | |
| 	txs types.Transactions, miningReward int64,
 | |
| 	getTracerFn func(txIndex int, txHash common.Hash) (tracer vm.EVMLogger, err error)) (*state.StateDB, *ExecutionResult, error) {
 | |
| 
 | |
| 	// Capture errors for BLOCKHASH operation, if we haven't been supplied the
 | |
| 	// required blockhashes
 | |
| 	var hashError error
 | |
| 	getHash := func(num uint64) common.Hash {
 | |
| 		if pre.Env.BlockHashes == nil {
 | |
| 			hashError = fmt.Errorf("getHash(%d) invoked, no blockhashes provided", num)
 | |
| 			return common.Hash{}
 | |
| 		}
 | |
| 		h, ok := pre.Env.BlockHashes[math.HexOrDecimal64(num)]
 | |
| 		if !ok {
 | |
| 			hashError = fmt.Errorf("getHash(%d) invoked, blockhash for that block not provided", num)
 | |
| 		}
 | |
| 		return h
 | |
| 	}
 | |
| 	var (
 | |
| 		statedb     = MakePreState(rawdb.NewMemoryDatabase(), pre.Pre)
 | |
| 		signer      = types.MakeSigner(chainConfig, new(big.Int).SetUint64(pre.Env.Number))
 | |
| 		gaspool     = new(core.GasPool)
 | |
| 		blockHash   = common.Hash{0x13, 0x37}
 | |
| 		rejectedTxs []*rejectedTx
 | |
| 		includedTxs types.Transactions
 | |
| 		gasUsed     = uint64(0)
 | |
| 		receipts    = make(types.Receipts, 0)
 | |
| 		txIndex     = 0
 | |
| 	)
 | |
| 	gaspool.AddGas(pre.Env.GasLimit)
 | |
| 	vmContext := vm.BlockContext{
 | |
| 		CanTransfer: core.CanTransfer,
 | |
| 		Transfer:    core.Transfer,
 | |
| 		Coinbase:    pre.Env.Coinbase,
 | |
| 		BlockNumber: new(big.Int).SetUint64(pre.Env.Number),
 | |
| 		Time:        new(big.Int).SetUint64(pre.Env.Timestamp),
 | |
| 		Difficulty:  pre.Env.Difficulty,
 | |
| 		GasLimit:    pre.Env.GasLimit,
 | |
| 		GetHash:     getHash,
 | |
| 	}
 | |
| 	// If currentBaseFee is defined, add it to the vmContext.
 | |
| 	if pre.Env.BaseFee != nil {
 | |
| 		vmContext.BaseFee = new(big.Int).Set(pre.Env.BaseFee)
 | |
| 	}
 | |
| 	// If DAO is supported/enabled, we need to handle it here. In geth 'proper', it's
 | |
| 	// done in StateProcessor.Process(block, ...), right before transactions are applied.
 | |
| 	if chainConfig.DAOForkSupport &&
 | |
| 		chainConfig.DAOForkBlock != nil &&
 | |
| 		chainConfig.DAOForkBlock.Cmp(new(big.Int).SetUint64(pre.Env.Number)) == 0 {
 | |
| 		misc.ApplyDAOHardFork(statedb)
 | |
| 	}
 | |
| 
 | |
| 	for i, tx := range txs {
 | |
| 		msg, err := tx.AsMessage(signer, pre.Env.BaseFee)
 | |
| 		if err != nil {
 | |
| 			log.Warn("rejected tx", "index", i, "hash", tx.Hash(), "error", err)
 | |
| 			rejectedTxs = append(rejectedTxs, &rejectedTx{i, err.Error()})
 | |
| 			continue
 | |
| 		}
 | |
| 		tracer, err := getTracerFn(txIndex, tx.Hash())
 | |
| 		if err != nil {
 | |
| 			return nil, nil, err
 | |
| 		}
 | |
| 		vmConfig.Tracer = tracer
 | |
| 		vmConfig.Debug = (tracer != nil)
 | |
| 		statedb.Prepare(tx.Hash(), txIndex)
 | |
| 		txContext := core.NewEVMTxContext(msg)
 | |
| 		snapshot := statedb.Snapshot()
 | |
| 		evm := vm.NewEVM(vmContext, txContext, statedb, chainConfig, vmConfig)
 | |
| 
 | |
| 		// (ret []byte, usedGas uint64, failed bool, err error)
 | |
| 		msgResult, err := core.ApplyMessage(evm, msg, gaspool)
 | |
| 		if err != nil {
 | |
| 			statedb.RevertToSnapshot(snapshot)
 | |
| 			log.Info("rejected tx", "index", i, "hash", tx.Hash(), "from", msg.From(), "error", err)
 | |
| 			rejectedTxs = append(rejectedTxs, &rejectedTx{i, err.Error()})
 | |
| 			continue
 | |
| 		}
 | |
| 		includedTxs = append(includedTxs, tx)
 | |
| 		if hashError != nil {
 | |
| 			return nil, nil, NewError(ErrorMissingBlockhash, hashError)
 | |
| 		}
 | |
| 		gasUsed += msgResult.UsedGas
 | |
| 
 | |
| 		// Receipt:
 | |
| 		{
 | |
| 			var root []byte
 | |
| 			if chainConfig.IsByzantium(vmContext.BlockNumber) {
 | |
| 				statedb.Finalise(true)
 | |
| 			} else {
 | |
| 				root = statedb.IntermediateRoot(chainConfig.IsEIP158(vmContext.BlockNumber)).Bytes()
 | |
| 			}
 | |
| 
 | |
| 			// Create a new receipt for the transaction, storing the intermediate root and
 | |
| 			// gas used by the tx.
 | |
| 			receipt := &types.Receipt{Type: tx.Type(), PostState: root, CumulativeGasUsed: gasUsed}
 | |
| 			if msgResult.Failed() {
 | |
| 				receipt.Status = types.ReceiptStatusFailed
 | |
| 			} else {
 | |
| 				receipt.Status = types.ReceiptStatusSuccessful
 | |
| 			}
 | |
| 			receipt.TxHash = tx.Hash()
 | |
| 			receipt.GasUsed = msgResult.UsedGas
 | |
| 
 | |
| 			// If the transaction created a contract, store the creation address in the receipt.
 | |
| 			if msg.To() == nil {
 | |
| 				receipt.ContractAddress = crypto.CreateAddress(evm.TxContext.Origin, tx.Nonce())
 | |
| 			}
 | |
| 
 | |
| 			// Set the receipt logs and create the bloom filter.
 | |
| 			receipt.Logs = statedb.GetLogs(tx.Hash(), blockHash)
 | |
| 			receipt.Bloom = types.CreateBloom(types.Receipts{receipt})
 | |
| 			// These three are non-consensus fields:
 | |
| 			//receipt.BlockHash
 | |
| 			//receipt.BlockNumber
 | |
| 			receipt.TransactionIndex = uint(txIndex)
 | |
| 			receipts = append(receipts, receipt)
 | |
| 		}
 | |
| 
 | |
| 		txIndex++
 | |
| 	}
 | |
| 	statedb.IntermediateRoot(chainConfig.IsEIP158(vmContext.BlockNumber))
 | |
| 	// Add mining reward?
 | |
| 	if miningReward > 0 {
 | |
| 		// Add mining reward. The mining reward may be `0`, which only makes a difference in the cases
 | |
| 		// where
 | |
| 		// - the coinbase suicided, or
 | |
| 		// - there are only 'bad' transactions, which aren't executed. In those cases,
 | |
| 		//   the coinbase gets no txfee, so isn't created, and thus needs to be touched
 | |
| 		var (
 | |
| 			blockReward = big.NewInt(miningReward)
 | |
| 			minerReward = new(big.Int).Set(blockReward)
 | |
| 			perOmmer    = new(big.Int).Div(blockReward, big.NewInt(32))
 | |
| 		)
 | |
| 		for _, ommer := range pre.Env.Ommers {
 | |
| 			// Add 1/32th for each ommer included
 | |
| 			minerReward.Add(minerReward, perOmmer)
 | |
| 			// Add (8-delta)/8
 | |
| 			reward := big.NewInt(8)
 | |
| 			reward.Sub(reward, big.NewInt(0).SetUint64(ommer.Delta))
 | |
| 			reward.Mul(reward, blockReward)
 | |
| 			reward.Div(reward, big.NewInt(8))
 | |
| 			statedb.AddBalance(ommer.Address, reward)
 | |
| 		}
 | |
| 		statedb.AddBalance(pre.Env.Coinbase, minerReward)
 | |
| 	}
 | |
| 	// Commit block
 | |
| 	root, err := statedb.Commit(chainConfig.IsEIP158(vmContext.BlockNumber))
 | |
| 	if err != nil {
 | |
| 		fmt.Fprintf(os.Stderr, "Could not commit state: %v", err)
 | |
| 		return nil, nil, NewError(ErrorEVM, fmt.Errorf("could not commit state: %v", err))
 | |
| 	}
 | |
| 	execRs := &ExecutionResult{
 | |
| 		StateRoot:   root,
 | |
| 		TxRoot:      types.DeriveSha(includedTxs, trie.NewStackTrie(nil)),
 | |
| 		ReceiptRoot: types.DeriveSha(receipts, trie.NewStackTrie(nil)),
 | |
| 		Bloom:       types.CreateBloom(receipts),
 | |
| 		LogsHash:    rlpHash(statedb.Logs()),
 | |
| 		Receipts:    receipts,
 | |
| 		Rejected:    rejectedTxs,
 | |
| 		Difficulty:  (*math.HexOrDecimal256)(vmContext.Difficulty),
 | |
| 		GasUsed:     (math.HexOrDecimal64)(gasUsed),
 | |
| 	}
 | |
| 	return statedb, execRs, nil
 | |
| }
 | |
| 
 | |
| func MakePreState(db ethdb.Database, accounts core.GenesisAlloc) *state.StateDB {
 | |
| 	sdb := state.NewDatabase(db)
 | |
| 	statedb, _ := state.New(common.Hash{}, sdb, nil)
 | |
| 	for addr, a := range accounts {
 | |
| 		statedb.SetCode(addr, a.Code)
 | |
| 		statedb.SetNonce(addr, a.Nonce)
 | |
| 		statedb.SetBalance(addr, a.Balance)
 | |
| 		for k, v := range a.Storage {
 | |
| 			statedb.SetState(addr, k, v)
 | |
| 		}
 | |
| 	}
 | |
| 	// Commit and re-open to start with a clean state.
 | |
| 	root, _ := statedb.Commit(false)
 | |
| 	statedb, _ = state.New(root, sdb, nil)
 | |
| 	return statedb
 | |
| }
 | |
| 
 | |
| func rlpHash(x interface{}) (h common.Hash) {
 | |
| 	hw := sha3.NewLegacyKeccak256()
 | |
| 	rlp.Encode(hw, x)
 | |
| 	hw.Sum(h[:0])
 | |
| 	return h
 | |
| }
 | |
| 
 | |
| // calcDifficulty is based on ethash.CalcDifficulty. This method is used in case
 | |
| // the caller does not provide an explicit difficulty, but instead provides only
 | |
| // parent timestamp + difficulty.
 | |
| // Note: this method only works for ethash engine.
 | |
| func calcDifficulty(config *params.ChainConfig, number, currentTime, parentTime uint64,
 | |
| 	parentDifficulty *big.Int, parentUncleHash common.Hash) *big.Int {
 | |
| 	uncleHash := parentUncleHash
 | |
| 	if uncleHash == (common.Hash{}) {
 | |
| 		uncleHash = types.EmptyUncleHash
 | |
| 	}
 | |
| 	parent := &types.Header{
 | |
| 		ParentHash: common.Hash{},
 | |
| 		UncleHash:  uncleHash,
 | |
| 		Difficulty: parentDifficulty,
 | |
| 		Number:     new(big.Int).SetUint64(number - 1),
 | |
| 		Time:       parentTime,
 | |
| 	}
 | |
| 	return ethash.CalcDifficulty(config, currentTime, parent)
 | |
| }
 |