plugeth/eth/backend.go

335 lines
9.2 KiB
Go
Raw Normal View History

package eth
import (
"crypto/ecdsa"
2015-01-04 13:20:16 +00:00
"fmt"
"io/ioutil"
"path"
"strings"
2015-03-03 16:55:23 +00:00
"github.com/ethereum/ethash"
"github.com/ethereum/go-ethereum/accounts"
"github.com/ethereum/go-ethereum/blockpool"
"github.com/ethereum/go-ethereum/core"
"github.com/ethereum/go-ethereum/crypto"
2015-01-04 13:20:16 +00:00
"github.com/ethereum/go-ethereum/ethdb"
"github.com/ethereum/go-ethereum/ethutil"
"github.com/ethereum/go-ethereum/event"
2015-03-02 14:15:28 +00:00
"github.com/ethereum/go-ethereum/logger"
"github.com/ethereum/go-ethereum/miner"
"github.com/ethereum/go-ethereum/p2p"
"github.com/ethereum/go-ethereum/p2p/discover"
"github.com/ethereum/go-ethereum/p2p/nat"
"github.com/ethereum/go-ethereum/vm"
2014-12-15 11:01:55 +00:00
"github.com/ethereum/go-ethereum/whisper"
)
2015-02-13 20:41:13 +00:00
var (
2015-03-10 01:34:52 +00:00
servlogger = logger.NewLogger("SERV")
2015-03-02 14:15:28 +00:00
jsonlogger = logger.NewJsonLogger()
2015-02-13 20:41:13 +00:00
defaultBootNodes = []*discover.Node{
// ETH/DEV cmd/bootnode
2015-02-17 22:14:10 +00:00
discover.MustParseNode("enode://6cdd090303f394a1cac34ecc9f7cda18127eafa2a3a06de39f6d920b0e583e062a7362097c7c65ee490a758b442acd5c80c6fce4b148c6a391e946b45131365b@54.169.166.226:30303"),
// ETH/DEV cpp-ethereum (poc-8.ethdev.com)
discover.MustParseNode("enode://4a44599974518ea5b0f14c31c4463692ac0329cb84851f3435e6d1b18ee4eae4aa495f846a0fa1219bd58035671881d44423876e57db2abd57254d0197da0ebe@5.1.83.226:30303"),
2015-02-13 20:41:13 +00:00
}
)
2015-01-04 13:20:16 +00:00
type Config struct {
2015-02-13 15:02:37 +00:00
Name string
DataDir string
LogFile string
LogLevel int
LogFormat string
VmDebug bool
2015-01-04 13:20:16 +00:00
MaxPeers int
Port string
2015-01-04 13:20:16 +00:00
// This should be a space-separated list of
// discovery node URLs.
BootNodes string
// This key is used to identify the node on the network.
// If nil, an ephemeral key is used.
NodeKey *ecdsa.PrivateKey
NAT nat.Interface
2015-01-05 16:10:42 +00:00
Shh bool
Dial bool
MinerThreads int
2015-03-08 00:52:49 +00:00
AccountManager *accounts.Manager
2015-01-04 13:20:16 +00:00
}
func (cfg *Config) parseBootNodes() []*discover.Node {
2015-02-13 20:41:13 +00:00
if cfg.BootNodes == "" {
return defaultBootNodes
}
var ns []*discover.Node
for _, url := range strings.Split(cfg.BootNodes, " ") {
if url == "" {
continue
}
n, err := discover.ParseNode(url)
if err != nil {
2015-03-10 01:34:52 +00:00
servlogger.Errorf("Bootstrap URL %s: %v\n", url, err)
continue
}
ns = append(ns, n)
}
return ns
}
func (cfg *Config) nodeKey() (*ecdsa.PrivateKey, error) {
// use explicit key from command line args if set
if cfg.NodeKey != nil {
return cfg.NodeKey, nil
}
// use persistent key if present
keyfile := path.Join(cfg.DataDir, "nodekey")
key, err := crypto.LoadECDSA(keyfile)
if err == nil {
return key, nil
}
// no persistent key, generate and store a new one
if key, err = crypto.GenerateKey(); err != nil {
return nil, fmt.Errorf("could not generate server key: %v", err)
}
if err := ioutil.WriteFile(keyfile, crypto.FromECDSA(key), 0600); err != nil {
2015-03-10 01:34:52 +00:00
servlogger.Errorln("could not persist nodekey: ", err)
}
return key, nil
}
type Ethereum struct {
// Channel for shutting down the ethereum
shutdownChan chan bool
// DB interface
blockDb ethutil.Database
stateDb ethutil.Database
2014-12-15 11:01:55 +00:00
//*** SERVICES ***
// State manager for processing new blocks and managing the over all states
2015-01-04 23:18:44 +00:00
blockProcessor *core.BlockProcessor
txPool *core.TxPool
chainManager *core.ChainManager
blockPool *blockpool.BlockPool
2015-03-08 00:52:49 +00:00
accountManager *accounts.Manager
2015-01-04 23:18:44 +00:00
whisper *whisper.Whisper
2015-01-04 13:20:16 +00:00
net *p2p.Server
2014-12-15 11:01:55 +00:00
eventMux *event.TypeMux
txSub event.Subscription
blockSub event.Subscription
miner *miner.Miner
2015-03-02 14:15:28 +00:00
logger logger.LogSystem
Mining bool
DataDir string
}
2015-01-04 13:20:16 +00:00
func New(config *Config) (*Ethereum, error) {
// Boostrap database
2015-03-10 01:34:52 +00:00
servlogger := logger.New(config.DataDir, config.LogFile, config.LogLevel, config.LogFormat)
blockDb, err := ethdb.NewLDBDatabase(path.Join(config.DataDir, "blockchain"))
if err != nil {
return nil, err
}
stateDb, err := ethdb.NewLDBDatabase(path.Join(config.DataDir, "state"))
2015-01-04 13:20:16 +00:00
if err != nil {
return nil, err
}
// Perform database sanity checks
d, _ := blockDb.Get([]byte("ProtocolVersion"))
2015-01-04 13:20:16 +00:00
protov := ethutil.NewValue(d).Uint()
if protov != ProtocolVersion && protov != 0 {
path := path.Join(config.DataDir, "blockchain")
return nil, fmt.Errorf("Database version mismatch. Protocol(%d / %d). `rm -rf %s`", protov, ProtocolVersion, path)
2015-01-04 13:20:16 +00:00
}
saveProtocolVersion(blockDb)
2015-01-07 12:17:48 +00:00
//ethutil.Config.Db = db
eth := &Ethereum{
shutdownChan: make(chan bool),
blockDb: blockDb,
stateDb: stateDb,
eventMux: &event.TypeMux{},
2015-03-10 01:34:52 +00:00
logger: servlogger,
accountManager: config.AccountManager,
DataDir: config.DataDir,
}
cb, err := eth.accountManager.Coinbase()
if err != nil {
2015-03-10 14:42:58 +00:00
return nil, err
}
eth.chainManager = core.NewChainManager(blockDb, stateDb, eth.EventMux())
2015-03-03 16:55:23 +00:00
pow := ethash.New(eth.chainManager)
eth.txPool = core.NewTxPool(eth.EventMux())
eth.blockProcessor = core.NewBlockProcessor(stateDb, pow, eth.txPool, eth.chainManager, eth.EventMux())
2015-01-04 23:18:44 +00:00
eth.chainManager.SetProcessor(eth.blockProcessor)
2014-12-15 11:01:55 +00:00
eth.whisper = whisper.New()
eth.miner = miner.New(cb, eth, pow, config.MinerThreads)
hasBlock := eth.chainManager.HasBlock
insertChain := eth.chainManager.InsertChain
eth.blockPool = blockpool.New(hasBlock, insertChain, pow.Verify)
netprv, err := config.nodeKey()
if err != nil {
return nil, err
}
ethProto := EthProtocol(eth.txPool, eth.chainManager, eth.blockPool)
protocols := []p2p.Protocol{ethProto}
if config.Shh {
protocols = append(protocols, eth.whisper.Protocol())
}
2015-01-04 13:20:16 +00:00
eth.net = &p2p.Server{
PrivateKey: netprv,
Name: config.Name,
MaxPeers: config.MaxPeers,
Protocols: protocols,
NAT: config.NAT,
NoDial: !config.Dial,
BootstrapNodes: config.parseBootNodes(),
2015-01-05 16:10:42 +00:00
}
if len(config.Port) > 0 {
eth.net.ListenAddr = ":" + config.Port
}
vm.Debug = config.VmDebug
return eth, nil
}
2015-03-08 00:52:49 +00:00
func (s *Ethereum) Logger() logger.LogSystem { return s.logger }
func (s *Ethereum) Name() string { return s.net.Name }
func (s *Ethereum) AccountManager() *accounts.Manager { return s.accountManager }
func (s *Ethereum) ChainManager() *core.ChainManager { return s.chainManager }
func (s *Ethereum) BlockProcessor() *core.BlockProcessor { return s.blockProcessor }
func (s *Ethereum) TxPool() *core.TxPool { return s.txPool }
func (s *Ethereum) BlockPool() *blockpool.BlockPool { return s.blockPool }
func (s *Ethereum) Whisper() *whisper.Whisper { return s.whisper }
func (s *Ethereum) EventMux() *event.TypeMux { return s.eventMux }
func (s *Ethereum) BlockDb() ethutil.Database { return s.blockDb }
func (s *Ethereum) StateDb() ethutil.Database { return s.stateDb }
2015-03-08 00:52:49 +00:00
func (s *Ethereum) Miner() *miner.Miner { return s.miner }
func (s *Ethereum) IsListening() bool { return true } // Always listening
func (s *Ethereum) PeerCount() int { return s.net.PeerCount() }
func (s *Ethereum) Peers() []*p2p.Peer { return s.net.Peers() }
func (s *Ethereum) MaxPeers() int { return s.net.MaxPeers }
2015-01-28 17:14:28 +00:00
// Start the ethereum
func (s *Ethereum) Start() error {
2015-03-02 14:15:28 +00:00
jsonlogger.LogJson(&logger.LogStarting{
2015-02-13 15:02:37 +00:00
ClientString: s.net.Name,
2015-02-10 18:21:13 +00:00
ProtocolVersion: ProtocolVersion,
})
2015-01-04 13:20:16 +00:00
err := s.net.Start()
if err != nil {
return err
}
2015-01-05 16:10:42 +00:00
// Start services
s.txPool.Start()
s.blockPool.Start()
2015-01-05 16:10:42 +00:00
if s.whisper != nil {
s.whisper.Start()
}
// broadcast transactions
s.txSub = s.eventMux.Subscribe(core.TxPreEvent{})
go s.txBroadcastLoop()
// broadcast mined blocks
s.blockSub = s.eventMux.Subscribe(core.NewMinedBlockEvent{})
go s.blockBroadcastLoop()
2015-03-10 01:34:52 +00:00
servlogger.Infoln("Server started")
return nil
}
func (s *Ethereum) StartForTest() {
jsonlogger.LogJson(&logger.LogStarting{
ClientString: s.net.Name,
ProtocolVersion: ProtocolVersion,
})
// Start services
s.txPool.Start()
s.blockPool.Start()
}
func (self *Ethereum) SuggestPeer(nodeURL string) error {
n, err := discover.ParseNode(nodeURL)
if err != nil {
return fmt.Errorf("invalid node URL: %v", err)
}
self.net.SuggestPeer(n)
return nil
}
func (s *Ethereum) Stop() {
// Close the database
defer s.blockDb.Close()
defer s.stateDb.Close()
s.txSub.Unsubscribe() // quits txBroadcastLoop
s.blockSub.Unsubscribe() // quits blockBroadcastLoop
s.txPool.Stop()
s.eventMux.Stop()
s.blockPool.Stop()
2015-01-05 16:10:42 +00:00
if s.whisper != nil {
s.whisper.Stop()
}
2015-03-10 01:34:52 +00:00
servlogger.Infoln("Server stopped")
close(s.shutdownChan)
}
// This function will wait for a shutdown and resumes main thread execution
func (s *Ethereum) WaitForShutdown() {
<-s.shutdownChan
}
// now tx broadcasting is taken out of txPool
// handled here via subscription, efficiency?
func (self *Ethereum) txBroadcastLoop() {
// automatically stops if unsubscribe
for obj := range self.txSub.Chan() {
event := obj.(core.TxPreEvent)
2015-01-05 16:10:42 +00:00
self.net.Broadcast("eth", TxMsg, event.Tx.RlpData())
}
}
func (self *Ethereum) blockBroadcastLoop() {
// automatically stops if unsubscribe
2015-01-05 16:10:42 +00:00
for obj := range self.blockSub.Chan() {
switch ev := obj.(type) {
case core.NewMinedBlockEvent:
2015-01-05 16:10:42 +00:00
self.net.Broadcast("eth", NewBlockMsg, ev.Block.RlpData(), ev.Block.Td)
}
}
}
func saveProtocolVersion(db ethutil.Database) {
d, _ := db.Get([]byte("ProtocolVersion"))
protocolVersion := ethutil.NewValue(d).Uint()
if protocolVersion == 0 {
db.Put([]byte("ProtocolVersion"), ethutil.NewValue(ProtocolVersion).Bytes())
}
}