forked from cerc-io/plugeth
2b44ef5f93
This change introduces a breaking change to miner.etherbase is configured. Previously, users did not need to explicitly set the etherbase address via flag, since 'first' local account was used as etherbase automatically. This change removes the "default first account" feature. In Proof-of-stake world, the fee recipient address is provided by CL, and not configured in Geth any more - meaning that miner.etherbase is mostly for legacy networks(pow, clique networks etc).
284 lines
8.4 KiB
Go
284 lines
8.4 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 miner implements Ethereum block creation and mining.
|
|
package miner
|
|
|
|
import (
|
|
"errors"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/ethereum/go-ethereum/common"
|
|
"github.com/ethereum/go-ethereum/consensus/clique"
|
|
"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/txpool"
|
|
"github.com/ethereum/go-ethereum/core/types"
|
|
"github.com/ethereum/go-ethereum/core/vm"
|
|
"github.com/ethereum/go-ethereum/eth/downloader"
|
|
"github.com/ethereum/go-ethereum/event"
|
|
"github.com/ethereum/go-ethereum/trie"
|
|
)
|
|
|
|
type mockBackend struct {
|
|
bc *core.BlockChain
|
|
txPool *txpool.TxPool
|
|
}
|
|
|
|
func NewMockBackend(bc *core.BlockChain, txPool *txpool.TxPool) *mockBackend {
|
|
return &mockBackend{
|
|
bc: bc,
|
|
txPool: txPool,
|
|
}
|
|
}
|
|
|
|
func (m *mockBackend) BlockChain() *core.BlockChain {
|
|
return m.bc
|
|
}
|
|
|
|
func (m *mockBackend) TxPool() *txpool.TxPool {
|
|
return m.txPool
|
|
}
|
|
|
|
func (m *mockBackend) StateAtBlock(block *types.Block, reexec uint64, base *state.StateDB, checkLive bool, preferDisk bool) (statedb *state.StateDB, err error) {
|
|
return nil, errors.New("not supported")
|
|
}
|
|
|
|
type testBlockChain struct {
|
|
statedb *state.StateDB
|
|
gasLimit uint64
|
|
chainHeadFeed *event.Feed
|
|
}
|
|
|
|
func (bc *testBlockChain) CurrentBlock() *types.Block {
|
|
return types.NewBlock(&types.Header{
|
|
GasLimit: bc.gasLimit,
|
|
}, nil, nil, nil, trie.NewStackTrie(nil))
|
|
}
|
|
|
|
func (bc *testBlockChain) GetBlock(hash common.Hash, number uint64) *types.Block {
|
|
return bc.CurrentBlock()
|
|
}
|
|
|
|
func (bc *testBlockChain) StateAt(common.Hash) (*state.StateDB, error) {
|
|
return bc.statedb, nil
|
|
}
|
|
|
|
func (bc *testBlockChain) SubscribeChainHeadEvent(ch chan<- core.ChainHeadEvent) event.Subscription {
|
|
return bc.chainHeadFeed.Subscribe(ch)
|
|
}
|
|
|
|
func TestMiner(t *testing.T) {
|
|
miner, mux, cleanup := createMiner(t)
|
|
defer cleanup(false)
|
|
|
|
miner.Start()
|
|
waitForMiningState(t, miner, true)
|
|
// Start the downloader
|
|
mux.Post(downloader.StartEvent{})
|
|
waitForMiningState(t, miner, false)
|
|
// Stop the downloader and wait for the update loop to run
|
|
mux.Post(downloader.DoneEvent{})
|
|
waitForMiningState(t, miner, true)
|
|
|
|
// Subsequent downloader events after a successful DoneEvent should not cause the
|
|
// miner to start or stop. This prevents a security vulnerability
|
|
// that would allow entities to present fake high blocks that would
|
|
// stop mining operations by causing a downloader sync
|
|
// until it was discovered they were invalid, whereon mining would resume.
|
|
mux.Post(downloader.StartEvent{})
|
|
waitForMiningState(t, miner, true)
|
|
|
|
mux.Post(downloader.FailedEvent{})
|
|
waitForMiningState(t, miner, true)
|
|
}
|
|
|
|
// TestMinerDownloaderFirstFails tests that mining is only
|
|
// permitted to run indefinitely once the downloader sees a DoneEvent (success).
|
|
// An initial FailedEvent should allow mining to stop on a subsequent
|
|
// downloader StartEvent.
|
|
func TestMinerDownloaderFirstFails(t *testing.T) {
|
|
miner, mux, cleanup := createMiner(t)
|
|
defer cleanup(false)
|
|
|
|
miner.Start()
|
|
waitForMiningState(t, miner, true)
|
|
// Start the downloader
|
|
mux.Post(downloader.StartEvent{})
|
|
waitForMiningState(t, miner, false)
|
|
|
|
// Stop the downloader and wait for the update loop to run
|
|
mux.Post(downloader.FailedEvent{})
|
|
waitForMiningState(t, miner, true)
|
|
|
|
// Since the downloader hasn't yet emitted a successful DoneEvent,
|
|
// we expect the miner to stop on next StartEvent.
|
|
mux.Post(downloader.StartEvent{})
|
|
waitForMiningState(t, miner, false)
|
|
|
|
// Downloader finally succeeds.
|
|
mux.Post(downloader.DoneEvent{})
|
|
waitForMiningState(t, miner, true)
|
|
|
|
// Downloader starts again.
|
|
// Since it has achieved a DoneEvent once, we expect miner
|
|
// state to be unchanged.
|
|
mux.Post(downloader.StartEvent{})
|
|
waitForMiningState(t, miner, true)
|
|
|
|
mux.Post(downloader.FailedEvent{})
|
|
waitForMiningState(t, miner, true)
|
|
}
|
|
|
|
func TestMinerStartStopAfterDownloaderEvents(t *testing.T) {
|
|
miner, mux, cleanup := createMiner(t)
|
|
defer cleanup(false)
|
|
|
|
miner.Start()
|
|
waitForMiningState(t, miner, true)
|
|
// Start the downloader
|
|
mux.Post(downloader.StartEvent{})
|
|
waitForMiningState(t, miner, false)
|
|
|
|
// Downloader finally succeeds.
|
|
mux.Post(downloader.DoneEvent{})
|
|
waitForMiningState(t, miner, true)
|
|
|
|
miner.Stop()
|
|
waitForMiningState(t, miner, false)
|
|
|
|
miner.Start()
|
|
waitForMiningState(t, miner, true)
|
|
|
|
miner.Stop()
|
|
waitForMiningState(t, miner, false)
|
|
}
|
|
|
|
func TestStartWhileDownload(t *testing.T) {
|
|
miner, mux, cleanup := createMiner(t)
|
|
defer cleanup(false)
|
|
waitForMiningState(t, miner, false)
|
|
miner.Start()
|
|
waitForMiningState(t, miner, true)
|
|
// Stop the downloader and wait for the update loop to run
|
|
mux.Post(downloader.StartEvent{})
|
|
waitForMiningState(t, miner, false)
|
|
// Starting the miner after the downloader should not work
|
|
miner.Start()
|
|
waitForMiningState(t, miner, false)
|
|
}
|
|
|
|
func TestStartStopMiner(t *testing.T) {
|
|
miner, _, cleanup := createMiner(t)
|
|
defer cleanup(false)
|
|
waitForMiningState(t, miner, false)
|
|
miner.Start()
|
|
waitForMiningState(t, miner, true)
|
|
miner.Stop()
|
|
waitForMiningState(t, miner, false)
|
|
}
|
|
|
|
func TestCloseMiner(t *testing.T) {
|
|
miner, _, cleanup := createMiner(t)
|
|
defer cleanup(true)
|
|
waitForMiningState(t, miner, false)
|
|
miner.Start()
|
|
waitForMiningState(t, miner, true)
|
|
// Terminate the miner and wait for the update loop to run
|
|
miner.Close()
|
|
waitForMiningState(t, miner, false)
|
|
}
|
|
|
|
// TestMinerSetEtherbase checks that etherbase becomes set even if mining isn't
|
|
// possible at the moment
|
|
func TestMinerSetEtherbase(t *testing.T) {
|
|
miner, mux, cleanup := createMiner(t)
|
|
defer cleanup(false)
|
|
miner.Start()
|
|
waitForMiningState(t, miner, true)
|
|
// Start the downloader
|
|
mux.Post(downloader.StartEvent{})
|
|
waitForMiningState(t, miner, false)
|
|
// Now user tries to configure proper mining address
|
|
miner.Start()
|
|
// Stop the downloader and wait for the update loop to run
|
|
mux.Post(downloader.DoneEvent{})
|
|
waitForMiningState(t, miner, true)
|
|
|
|
coinbase := common.HexToAddress("0xdeedbeef")
|
|
miner.SetEtherbase(coinbase)
|
|
if addr := miner.worker.etherbase(); addr != coinbase {
|
|
t.Fatalf("Unexpected etherbase want %x got %x", coinbase, addr)
|
|
}
|
|
}
|
|
|
|
// waitForMiningState waits until either
|
|
// * the desired mining state was reached
|
|
// * a timeout was reached which fails the test
|
|
func waitForMiningState(t *testing.T, m *Miner, mining bool) {
|
|
t.Helper()
|
|
|
|
var state bool
|
|
for i := 0; i < 100; i++ {
|
|
time.Sleep(10 * time.Millisecond)
|
|
if state = m.Mining(); state == mining {
|
|
return
|
|
}
|
|
}
|
|
t.Fatalf("Mining() == %t, want %t", state, mining)
|
|
}
|
|
|
|
func createMiner(t *testing.T) (*Miner, *event.TypeMux, func(skipMiner bool)) {
|
|
// Create Ethash config
|
|
config := Config{
|
|
Etherbase: common.HexToAddress("123456789"),
|
|
}
|
|
// Create chainConfig
|
|
chainDB := rawdb.NewMemoryDatabase()
|
|
genesis := core.DeveloperGenesisBlock(15, 11_500_000, common.HexToAddress("12345"))
|
|
chainConfig, _, err := core.SetupGenesisBlock(chainDB, trie.NewDatabase(chainDB), genesis)
|
|
if err != nil {
|
|
t.Fatalf("can't create new chain config: %v", err)
|
|
}
|
|
// Create consensus engine
|
|
engine := clique.New(chainConfig.Clique, chainDB)
|
|
// Create Ethereum backend
|
|
bc, err := core.NewBlockChain(chainDB, nil, genesis, nil, engine, vm.Config{}, nil, nil)
|
|
if err != nil {
|
|
t.Fatalf("can't create new chain %v", err)
|
|
}
|
|
statedb, _ := state.New(common.Hash{}, state.NewDatabase(chainDB), nil)
|
|
blockchain := &testBlockChain{statedb, 10000000, new(event.Feed)}
|
|
|
|
pool := txpool.NewTxPool(testTxPoolConfig, chainConfig, blockchain)
|
|
backend := NewMockBackend(bc, pool)
|
|
// Create event Mux
|
|
mux := new(event.TypeMux)
|
|
// Create Miner
|
|
miner := New(backend, &config, chainConfig, mux, engine, nil)
|
|
cleanup := func(skipMiner bool) {
|
|
bc.Stop()
|
|
engine.Close()
|
|
pool.Stop()
|
|
if !skipMiner {
|
|
miner.Close()
|
|
}
|
|
}
|
|
return miner, mux, cleanup
|
|
}
|