plugeth/core/chain_makers.go

146 lines
4.5 KiB
Go
Raw Normal View History

package core
import (
"fmt"
2015-03-03 16:55:23 +00:00
"math/big"
2015-03-16 10:27:38 +00:00
"github.com/ethereum/go-ethereum/common"
2015-04-01 21:58:26 +00:00
"github.com/ethereum/go-ethereum/core/state"
2015-03-16 22:48:18 +00:00
"github.com/ethereum/go-ethereum/core/types"
"github.com/ethereum/go-ethereum/event"
"github.com/ethereum/go-ethereum/pow"
)
// So we can generate blocks easily
type FakePow struct{}
func (f FakePow) Search(block pow.Block, stop <-chan struct{}) (uint64, []byte) {
return 0, nil
2015-02-28 19:58:37 +00:00
}
func (f FakePow) Verify(block pow.Block) bool { return true }
func (f FakePow) GetHashrate() int64 { return 0 }
func (f FakePow) Turbo(bool) {}
// So we can deterministically seed different blockchains
var (
CanonicalSeed = 1
ForkSeed = 2
)
2015-02-28 19:58:37 +00:00
// Utility functions for making chains on the fly
// Exposed for sake of testing from other packages (eg. go-ethash)
2015-03-16 22:48:18 +00:00
func NewBlockFromParent(addr common.Address, parent *types.Block) *types.Block {
return newBlockFromParent(addr, parent)
}
2015-03-16 10:27:38 +00:00
func MakeBlock(bman *BlockProcessor, parent *types.Block, i int, db common.Database, seed int) *types.Block {
2015-02-28 19:58:37 +00:00
return makeBlock(bman, parent, i, db, seed)
}
2015-03-16 10:27:38 +00:00
func MakeChain(bman *BlockProcessor, parent *types.Block, max int, db common.Database, seed int) types.Blocks {
2015-02-28 19:58:37 +00:00
return makeChain(bman, parent, max, db, seed)
}
2015-03-16 10:27:38 +00:00
func NewChainMan(block *types.Block, eventMux *event.TypeMux, db common.Database) *ChainManager {
return newChainManager(block, eventMux, db)
}
func NewBlockProc(db common.Database, cman *ChainManager, eventMux *event.TypeMux) *BlockProcessor {
return newBlockProcessor(db, cman, eventMux)
}
2015-03-16 10:27:38 +00:00
func NewCanonical(n int, db common.Database) (*BlockProcessor, error) {
return newCanonical(n, db)
}
2015-02-28 19:58:37 +00:00
// block time is fixed at 10 seconds
2015-03-16 22:48:18 +00:00
func newBlockFromParent(addr common.Address, parent *types.Block) *types.Block {
block := types.NewBlock(parent.Hash(), addr, parent.Root(), common.BigPow(2, 32), 0, nil)
block.SetUncles(nil)
block.SetTransactions(nil)
block.SetReceipts(nil)
header := block.Header()
2015-03-04 09:49:56 +00:00
header.Difficulty = CalcDifficulty(block.Header(), parent.Header())
2015-03-16 10:27:38 +00:00
header.Number = new(big.Int).Add(parent.Header().Number, common.Big1)
2015-02-28 19:58:37 +00:00
header.Time = parent.Header().Time + 10
header.GasLimit = CalcGasLimit(parent)
block.Td = parent.Td
return block
}
// Actually make a block by simulating what miner would do
2015-02-28 19:58:37 +00:00
// we seed chains by the first byte of the coinbase
2015-03-16 10:27:38 +00:00
func makeBlock(bman *BlockProcessor, parent *types.Block, i int, db common.Database, seed int) *types.Block {
2015-03-16 22:48:18 +00:00
var addr common.Address
addr[0], addr[19] = byte(seed), byte(i)
block := newBlockFromParent(addr, parent)
state := state.New(block.Root(), db)
cbase := state.GetOrNewStateObject(addr)
cbase.SetGasPool(CalcGasLimit(parent))
cbase.AddBalance(BlockReward)
2015-04-01 21:58:26 +00:00
state.Update()
block.SetRoot(state.Root())
return block
}
// Make a chain with real blocks
// Runs ProcessWithParent to get proper state roots
2015-03-16 10:27:38 +00:00
func makeChain(bman *BlockProcessor, parent *types.Block, max int, db common.Database, seed int) types.Blocks {
bman.bc.currentBlock = parent
blocks := make(types.Blocks, max)
for i := 0; i < max; i++ {
2015-02-28 19:58:37 +00:00
block := makeBlock(bman, parent, i, db, seed)
_, err := bman.processWithParent(block, parent)
if err != nil {
fmt.Println("process with parent failed", err)
panic(err)
}
2015-05-15 10:24:29 +00:00
block.Td = CalcTD(block, parent)
blocks[i] = block
parent = block
}
return blocks
}
// Create a new chain manager starting from given block
// Effectively a fork factory
2015-03-16 10:27:38 +00:00
func newChainManager(block *types.Block, eventMux *event.TypeMux, db common.Database) *ChainManager {
genesis := GenesisBlock(0, db)
2015-05-18 15:46:47 +00:00
bc := &ChainManager{blockDb: db, stateDb: db, genesisBlock: genesis, eventMux: eventMux, pow: FakePow{}}
2015-04-22 11:09:59 +00:00
bc.txState = state.ManageState(state.New(genesis.Root(), db))
bc.futureBlocks = NewBlockCache(1000)
if block == nil {
bc.Reset()
} else {
bc.currentBlock = block
bc.td = block.Td
}
return bc
}
// block processor with fake pow
func newBlockProcessor(db common.Database, cman *ChainManager, eventMux *event.TypeMux) *BlockProcessor {
chainMan := newChainManager(nil, eventMux, db)
bman := NewBlockProcessor(db, db, FakePow{}, chainMan, eventMux)
return bman
}
2015-02-28 19:58:37 +00:00
// Make a new, deterministic canonical chain by running InsertChain
// on result of makeChain
2015-03-16 10:27:38 +00:00
func newCanonical(n int, db common.Database) (*BlockProcessor, error) {
eventMux := &event.TypeMux{}
bman := newBlockProcessor(db, newChainManager(nil, eventMux, db), eventMux)
bman.bc.SetProcessor(bman)
parent := bman.bc.CurrentBlock()
if n == 0 {
return bman, nil
}
2015-02-28 19:58:37 +00:00
lchain := makeChain(bman, parent, n, db, CanonicalSeed)
_, err := bman.bc.InsertChain(lchain)
2015-02-28 19:58:37 +00:00
return bman, err
}