mirror of
https://github.com/ethereum/go-ethereum.git
synced 2026-07-27 15:16:43 +00:00
325 lines
8.3 KiB
Go
325 lines
8.3 KiB
Go
// Copyright 2016 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 adapters
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"math"
|
|
"net"
|
|
"sync"
|
|
|
|
"github.com/ethereum/go-ethereum/event"
|
|
"github.com/ethereum/go-ethereum/log"
|
|
"github.com/ethereum/go-ethereum/node"
|
|
"github.com/ethereum/go-ethereum/p2p"
|
|
"github.com/ethereum/go-ethereum/p2p/discover"
|
|
"github.com/ethereum/go-ethereum/rpc"
|
|
)
|
|
|
|
// SimAdapter is a NodeAdapter which creates in-memory nodes and connects them
|
|
// using an in-memory p2p.MsgReadWriter pipe
|
|
type SimAdapter struct {
|
|
mtx sync.RWMutex
|
|
nodes map[discover.NodeID]*SimNode
|
|
services map[string]ServiceFunc
|
|
}
|
|
|
|
// NewSimAdapter creates a SimAdapter which is capable of running in-memory
|
|
// nodes running any of the given services (the service to run on a particular
|
|
// node is passed to the NewNode function in the NodeConfig)
|
|
func NewSimAdapter(services map[string]ServiceFunc) *SimAdapter {
|
|
return &SimAdapter{
|
|
nodes: make(map[discover.NodeID]*SimNode),
|
|
services: services,
|
|
}
|
|
}
|
|
|
|
// Name returns the name of the adapter for logging purpoeses
|
|
func (s *SimAdapter) Name() string {
|
|
return "sim-adapter"
|
|
}
|
|
|
|
// NewNode returns a new SimNode using the given config
|
|
func (s *SimAdapter) NewNode(config *NodeConfig) (Node, error) {
|
|
var nodeprotos []p2p.Protocol
|
|
|
|
s.mtx.Lock()
|
|
defer s.mtx.Unlock()
|
|
|
|
// check a node with the ID doesn't already exist
|
|
id := config.Id
|
|
if _, exists := s.nodes[id.NodeID]; exists {
|
|
return nil, fmt.Errorf("node already exists: %s", id)
|
|
}
|
|
|
|
// check the service is valid and initialize it
|
|
/*
|
|
serviceFunc, exists := s.services[config.Service]
|
|
if !exists {
|
|
return nil, fmt.Errorf("unknown node service %q", config.Service)
|
|
}
|
|
|
|
node := &SimNode{
|
|
Id: id,
|
|
config: config,
|
|
adapter: s,
|
|
serviceFunc: serviceFunc,
|
|
*/
|
|
//serviceFunc, exists := s.services[config.Service]
|
|
|
|
//if !exists {
|
|
// return nil, fmt.Errorf("unknown node service %q", config.Service)
|
|
//}
|
|
//service := serviceFunc(id)
|
|
|
|
_, err := node.New(&node.Config{
|
|
P2P: p2p.Config{
|
|
PrivateKey: config.PrivateKey,
|
|
MaxPeers: math.MaxInt32,
|
|
NoDiscovery: true,
|
|
Protocols: nodeprotos,
|
|
Dialer: s,
|
|
EnableMsgEvents: true,
|
|
},
|
|
})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
servicefuncs := make(map[string]ServiceFunc)
|
|
|
|
for name, servicefunc := range s.services {
|
|
service := servicefunc(id, nil)
|
|
/*if err := n.Register(func(ctx *node.ServiceContext) (node.Service, error) {
|
|
return service, err
|
|
}); err != nil {
|
|
return nil, err
|
|
}*/
|
|
for _, proto := range service.Protocols() {
|
|
nodeprotos = append(nodeprotos, proto)
|
|
}
|
|
servicefuncs[name] = servicefunc
|
|
}
|
|
|
|
simnode := &SimNode{
|
|
//node: n,
|
|
Id: id,
|
|
serviceFuncs: servicefuncs,
|
|
adapter: s,
|
|
config: config,
|
|
running: make(map[string]node.Service),
|
|
}
|
|
s.nodes[id.NodeID] = simnode
|
|
return simnode, nil
|
|
}
|
|
|
|
func (s *SimAdapter) Dial(dest *discover.Node) (conn net.Conn, err error) {
|
|
node, ok := s.GetNode(dest.ID)
|
|
if !ok {
|
|
return nil, fmt.Errorf("unknown node: %s", dest.ID)
|
|
}
|
|
srv := node.Server()
|
|
if srv == nil {
|
|
return nil, fmt.Errorf("node not running: %s", dest.ID)
|
|
}
|
|
pipe1, pipe2 := net.Pipe()
|
|
go srv.SetupConn(pipe1, 0, nil)
|
|
return pipe2, nil
|
|
}
|
|
|
|
// GetNode returns the node with the given ID if it exists
|
|
func (s *SimAdapter) GetNode(id discover.NodeID) (*SimNode, bool) {
|
|
s.mtx.RLock()
|
|
defer s.mtx.RUnlock()
|
|
node, ok := s.nodes[id]
|
|
return node, ok
|
|
}
|
|
|
|
// SimNode is an in-memory node which connects to other SimNodes using an
|
|
// in-memory p2p.MsgReadWriter pipe, running an underlying service protocol
|
|
// directly over that pipe.
|
|
//
|
|
// It implements the p2p.Server interface so it can be used transparently
|
|
// by the underlying service.
|
|
type SimNode struct {
|
|
lock sync.RWMutex
|
|
Id *NodeId
|
|
config *NodeConfig
|
|
adapter *SimAdapter
|
|
serviceFuncs map[string]ServiceFunc
|
|
node *node.Node
|
|
running map[string]node.Service
|
|
client *rpc.Client
|
|
rpcMux *rpcMux
|
|
}
|
|
|
|
// Addr returns the node's discovery address
|
|
func (self *SimNode) Addr() []byte {
|
|
return []byte(self.Node().String())
|
|
}
|
|
|
|
// Node returns a discover.Node representing the SimNode
|
|
func (self *SimNode) Node() *discover.Node {
|
|
return discover.NewNode(self.Id.NodeID, net.IP{127, 0, 0, 1}, 30303, 30303)
|
|
}
|
|
|
|
// Client returns an rpc.Client which can be used to communicate with the
|
|
// underlying service (it is set once the node has started)
|
|
func (self *SimNode) Client() (*rpc.Client, error) {
|
|
self.lock.Lock()
|
|
defer self.lock.Unlock()
|
|
if self.client == nil {
|
|
return nil, errors.New("RPC not started")
|
|
}
|
|
return self.client, nil
|
|
}
|
|
|
|
// ServeRPC serves RPC requests over the given connection using the node's
|
|
// RPC multiplexer
|
|
func (self *SimNode) ServeRPC(conn net.Conn) error {
|
|
self.lock.Lock()
|
|
mux := self.rpcMux
|
|
self.lock.Unlock()
|
|
if mux == nil {
|
|
return errors.New("RPC not started")
|
|
}
|
|
mux.Serve(conn)
|
|
return nil
|
|
}
|
|
|
|
// Snapshot creates a snapshot of the service state by calling the
|
|
// simulation_snapshot RPC method
|
|
func (self *SimNode) Snapshot() ([]byte, error) {
|
|
self.lock.Lock()
|
|
defer self.lock.Unlock()
|
|
if self.client == nil {
|
|
return nil, errors.New("RPC not started")
|
|
}
|
|
var snapshot []byte
|
|
return snapshot, self.client.Call(&snapshot, "simulation_snapshot")
|
|
}
|
|
|
|
// Start starts the RPC handler and the underlying service
|
|
func (self *SimNode) Start(snapshot []byte) error {
|
|
self.lock.Lock()
|
|
defer self.lock.Unlock()
|
|
if self.node != nil {
|
|
return errors.New("node already started")
|
|
}
|
|
|
|
services := []node.ServiceConstructor{}
|
|
|
|
|
|
// so we can control the order of the services if we need
|
|
for _, name := range self.config.Services {
|
|
service := self.serviceFuncs[name](self.Id, snapshot)
|
|
services = append(services, func(ctx *node.ServiceContext) (node.Service, error) {
|
|
self.running[name] = service
|
|
return service, nil
|
|
})
|
|
}
|
|
|
|
node, err := node.New(&node.Config{
|
|
P2P: p2p.Config{
|
|
PrivateKey: self.config.PrivateKey,
|
|
MaxPeers: math.MaxInt32,
|
|
NoDiscovery: true,
|
|
Dialer: self.adapter,
|
|
EnableMsgEvents: false,
|
|
},
|
|
NoUSB: true,
|
|
})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
for _, service := range services {
|
|
log.Debug("registering service", "service", service)
|
|
if err := node.Register(service); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
if err := node.Start(); err != nil {
|
|
return err
|
|
}
|
|
|
|
handler, err := node.RPCHandler()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// create an in-process RPC multiplexer
|
|
pipe1, pipe2 := net.Pipe()
|
|
go handler.ServeCodec(rpc.NewJSONCodec(pipe1), rpc.OptionMethodInvocation|rpc.OptionSubscriptions)
|
|
self.rpcMux = newRPCMux(pipe2)
|
|
|
|
// create an in-process RPC client
|
|
self.client = self.rpcMux.Client()
|
|
|
|
self.node = node
|
|
|
|
return nil
|
|
}
|
|
|
|
func (self *SimNode) Stop() error {
|
|
self.lock.Lock()
|
|
defer self.lock.Unlock()
|
|
if self.node == nil {
|
|
return nil
|
|
}
|
|
if err := self.node.Stop(); err != nil {
|
|
return err
|
|
}
|
|
self.node = nil
|
|
return nil
|
|
}
|
|
|
|
func (self *SimNode) Server() *p2p.Server {
|
|
self.lock.Lock()
|
|
defer self.lock.Unlock()
|
|
if self.node == nil {
|
|
return nil
|
|
}
|
|
return self.node.Server()
|
|
}
|
|
|
|
// Service returns a underlying node.Service of the speficied type
|
|
func (self *SimNode) GetService(servicename string) node.Service {
|
|
log.Warn("retrieving service", "name", servicename)
|
|
return self.running[servicename]
|
|
}
|
|
|
|
func (self *SimNode) SubscribeEvents(ch chan *p2p.PeerEvent) event.Subscription {
|
|
srv := self.Server()
|
|
if srv == nil {
|
|
panic("node not running")
|
|
}
|
|
return srv.SubscribeEvents(ch)
|
|
}
|
|
|
|
func (self *SimNode) NodeInfo() *p2p.NodeInfo {
|
|
server := self.Server()
|
|
if server == nil {
|
|
return &p2p.NodeInfo{
|
|
ID: self.Id.String(),
|
|
Enode: self.Node().String(),
|
|
}
|
|
}
|
|
return server.NodeInfo()
|
|
}
|