package eth import ( "encoding/json" "net" "path" "sync" "github.com/ethereum/go-ethereum/core" "github.com/ethereum/go-ethereum/crypto" "github.com/ethereum/go-ethereum/ethutil" "github.com/ethereum/go-ethereum/event" ethlogger "github.com/ethereum/go-ethereum/logger" "github.com/ethereum/go-ethereum/p2p" "github.com/ethereum/go-ethereum/pow/ezp" "github.com/ethereum/go-ethereum/rpc" "github.com/ethereum/go-ethereum/state" ) const ( seedNodeAddress = "poc-7.ethdev.com:30300" ) var logger = ethlogger.NewLogger("SERV") type Ethereum struct { // Channel for shutting down the ethereum shutdownChan chan bool quit chan bool // DB interface db ethutil.Database // State manager for processing new blocks and managing the over all states blockManager *core.BlockManager // The transaction pool. Transaction can be pushed on this pool // for later including in the blocks txPool *core.TxPool // The canonical chain chainManager *core.ChainManager // The block pool blockPool *BlockPool // Event eventMux *event.TypeMux // Nonce Nonce uint64 ListenAddr string blacklist p2p.Blacklist server *p2p.Server txSub event.Subscription blockSub event.Subscription // Capabilities for outgoing peers // serverCaps Caps peersFile string Mining bool RpcServer *rpc.JsonRpcServer keyManager *crypto.KeyManager clientIdentity p2p.ClientIdentity synclock sync.Mutex syncGroup sync.WaitGroup filterMu sync.RWMutex filterId int filters map[int]*core.Filter } func New(db ethutil.Database, identity p2p.ClientIdentity, keyManager *crypto.KeyManager, nat p2p.NAT, port string, maxPeers int) (*Ethereum, error) { saveProtocolVersion(db) ethutil.Config.Db = db // FIXME: blacklist := p2p.NewBlacklist() // Sorry Py person. I must blacklist. you perform badly blacklist.Put(ethutil.Hex2Bytes("64656330303561383532336435376331616537643864663236623336313863373537353163636634333530626263396330346237336262623931383064393031")) peersFile := path.Join(ethutil.Config.ExecPath, "known_peers.json") nonce, _ := ethutil.RandomUint64() listenAddr := ":" + port eth := &Ethereum{ shutdownChan: make(chan bool), quit: make(chan bool), db: db, Nonce: nonce, // serverCaps: caps, peersFile: peersFile, ListenAddr: listenAddr, keyManager: keyManager, clientIdentity: identity, blacklist: blacklist, eventMux: &event.TypeMux{}, filters: make(map[int]*core.Filter), } eth.txPool = core.NewTxPool(eth) eth.chainManager = core.NewChainManager(eth.EventMux()) eth.blockManager = core.NewBlockManager(eth) eth.chainManager.SetProcessor(eth.blockManager) hasBlock := eth.chainManager.HasBlock insertChain := eth.chainManager.InsertChain pow := ezp.New() verifyPoW := pow.Verify eth.blockPool = NewBlockPool(hasBlock, insertChain, verifyPoW) // Start the tx pool eth.txPool.Start() ethProto := EthProtocol(eth.txPool, eth.chainManager, eth.blockPool) protocols := []p2p.Protocol{ethProto} server := &p2p.Server{ Identity: identity, MaxPeers: maxPeers, Protocols: protocols, ListenAddr: listenAddr, Blacklist: blacklist, NAT: nat, } eth.server = server return eth, nil } func (s *Ethereum) KeyManager() *crypto.KeyManager { return s.keyManager } func (s *Ethereum) ClientIdentity() p2p.ClientIdentity { return s.clientIdentity } func (s *Ethereum) ChainManager() *core.ChainManager { return s.chainManager } func (s *Ethereum) BlockManager() *core.BlockManager { return s.blockManager } func (s *Ethereum) TxPool() *core.TxPool { return s.txPool } func (s *Ethereum) BlockPool() *BlockPool { return s.blockPool } func (s *Ethereum) EventMux() *event.TypeMux { return s.eventMux } func (self *Ethereum) Db() ethutil.Database { return self.db } func (s *Ethereum) IsMining() bool { return s.Mining } func (s *Ethereum) IsListening() bool { if s.ListenAddr == "" { return false } else { return true } } func (s *Ethereum) PeerCount() int { return s.server.PeerCount() } func (s *Ethereum) Peers() []*p2p.Peer { return s.server.Peers() } // Start the ethereum func (s *Ethereum) Start(seed bool) error { err := s.server.Start() if err != nil { return err } s.blockPool.Start() go s.filterLoop() // 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() // TODO: read peers here if seed { logger.Infof("Connect to seed node %v", seedNodeAddress) if err := s.SuggestPeer(seedNodeAddress); err != nil { return err } } logger.Infoln("Server started") return nil } func (self *Ethereum) SuggestPeer(addr string) error { netaddr, err := net.ResolveTCPAddr("tcp", addr) if err != nil { logger.Errorf("couldn't resolve %s:", addr, err) return err } self.server.SuggestPeer(netaddr.IP, netaddr.Port, nil) return nil } func (s *Ethereum) Stop() { // Close the database defer s.db.Close() // // WritePeers(s.peersFile, s.server.PeerAddresses()) close(s.quit) s.txSub.Unsubscribe() // quits txBroadcastLoop s.blockSub.Unsubscribe() // quits blockBroadcastLoop if s.RpcServer != nil { s.RpcServer.Stop() } s.txPool.Stop() s.eventMux.Stop() s.blockPool.Stop() logger.Infoln("Server stopped") close(s.shutdownChan) } // This function will wait for a shutdown and resumes main thread execution func (s *Ethereum) WaitForShutdown() { <-s.shutdownChan } func WritePeers(path string, addresses []string) { if len(addresses) > 0 { data, _ := json.MarshalIndent(addresses, "", " ") ethutil.WriteFile(path, data) } } func ReadPeers(path string) (ips []string, err error) { var data string data, err = ethutil.ReadAllFile(path) if err != nil { json.Unmarshal([]byte(data), &ips) } return } // 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) self.server.Broadcast("eth", TxMsg, []interface{}{event.Tx.RlpData()}) } } func (self *Ethereum) blockBroadcastLoop() { // automatically stops if unsubscribe for obj := range self.txSub.Chan() { event := obj.(core.NewMinedBlockEvent) self.server.Broadcast("eth", NewBlockMsg, event.Block.Value().Val) } } 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()) } } // InstallFilter adds filter for blockchain events. // The filter's callbacks will run for matching blocks and messages. // The filter should not be modified after it has been installed. func (self *Ethereum) InstallFilter(filter *core.Filter) (id int) { self.filterMu.Lock() id = self.filterId self.filters[id] = filter self.filterId++ self.filterMu.Unlock() return id } func (self *Ethereum) UninstallFilter(id int) { self.filterMu.Lock() delete(self.filters, id) self.filterMu.Unlock() } // GetFilter retrieves a filter installed using InstallFilter. // The filter may not be modified. func (self *Ethereum) GetFilter(id int) *core.Filter { self.filterMu.RLock() defer self.filterMu.RUnlock() return self.filters[id] } func (self *Ethereum) filterLoop() { // Subscribe to events events := self.eventMux.Subscribe(core.NewBlockEvent{}, state.Messages(nil)) for event := range events.Chan() { switch event.(type) { case core.NewBlockEvent: self.filterMu.RLock() for _, filter := range self.filters { if filter.BlockCallback != nil { e := event.(core.NewBlockEvent) filter.BlockCallback(e.Block) } } self.filterMu.RUnlock() case state.Messages: self.filterMu.RLock() for _, filter := range self.filters { if filter.MessageCallback != nil { e := event.(state.Messages) msgs := filter.FilterMessages(e) if len(msgs) > 0 { filter.MessageCallback(msgs) } } } self.filterMu.RUnlock() } } }