2019-10-09 15:32:15 -07:00
|
|
|
package zcash
|
|
|
|
|
|
|
|
import (
|
|
|
|
"log"
|
|
|
|
"net"
|
|
|
|
"os"
|
|
|
|
"sync"
|
|
|
|
"time"
|
|
|
|
|
|
|
|
"github.com/btcsuite/btcd/peer"
|
|
|
|
"github.com/btcsuite/btcd/wire"
|
|
|
|
|
|
|
|
"github.com/gtank/coredns-zcash/zcash/network"
|
|
|
|
|
|
|
|
"github.com/pkg/errors"
|
|
|
|
)
|
|
|
|
|
2019-10-12 16:18:25 -07:00
|
|
|
var (
|
|
|
|
ErrRepeatConnection = errors.New("attempted repeat connection to existing peer")
|
2019-10-12 19:00:20 -07:00
|
|
|
ErrNoSuchPeer = errors.New("no record of requested peer")
|
2019-10-12 16:18:25 -07:00
|
|
|
)
|
|
|
|
|
2019-10-09 15:32:15 -07:00
|
|
|
var defaultPeerConfig = &peer.Config{
|
|
|
|
UserAgentName: "MagicBean",
|
|
|
|
UserAgentVersion: "2.0.7",
|
|
|
|
ChainParams: nil,
|
|
|
|
Services: 0,
|
|
|
|
TrickleInterval: time.Second * 10,
|
|
|
|
ProtocolVersion: 170009, // Blossom
|
|
|
|
}
|
|
|
|
|
|
|
|
type Seeder struct {
|
|
|
|
peer *peer.Peer
|
|
|
|
config *peer.Config
|
2019-10-12 09:42:16 -07:00
|
|
|
logger *log.Logger
|
2019-10-09 15:32:15 -07:00
|
|
|
|
2019-10-12 16:18:25 -07:00
|
|
|
handshakeSignals *sync.Map
|
|
|
|
pendingPeers *sync.Map
|
|
|
|
livePeers *sync.Map
|
2019-10-09 15:32:15 -07:00
|
|
|
|
2019-10-12 19:00:20 -07:00
|
|
|
addrRecvChan chan *wire.NetAddress
|
|
|
|
|
2019-10-09 15:32:15 -07:00
|
|
|
// For mutating the above
|
|
|
|
peerState sync.RWMutex
|
|
|
|
}
|
|
|
|
|
2019-10-12 09:42:16 -07:00
|
|
|
func NewSeeder(network network.Network) (*Seeder, error) {
|
|
|
|
config, err := newSeederPeerConfig(network, defaultPeerConfig)
|
|
|
|
if err != nil {
|
|
|
|
return nil, errors.Wrap(err, "could not construct seeder")
|
|
|
|
}
|
|
|
|
|
|
|
|
logger := log.New(os.Stdout, "zcash_seeder: ", log.Ldate|log.Ltime|log.Lshortfile|log.LUTC)
|
|
|
|
|
|
|
|
newSeeder := Seeder{
|
|
|
|
config: config,
|
|
|
|
logger: logger,
|
2019-10-12 16:18:25 -07:00
|
|
|
handshakeSignals: new(sync.Map),
|
|
|
|
pendingPeers: new(sync.Map),
|
|
|
|
livePeers: new(sync.Map),
|
2019-10-12 19:00:20 -07:00
|
|
|
addrRecvChan: make(chan *wire.NetAddress, 100),
|
2019-10-12 09:42:16 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
newSeeder.config.Listeners.OnVerAck = newSeeder.onVerAck
|
2019-10-12 19:00:20 -07:00
|
|
|
newSeeder.config.Listeners.OnAddr = newSeeder.onAddr
|
2019-10-12 09:42:16 -07:00
|
|
|
|
|
|
|
return &newSeeder, nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func newTestSeeder(network network.Network) (*Seeder, error) {
|
|
|
|
config, err := newSeederPeerConfig(network, defaultPeerConfig)
|
|
|
|
if err != nil {
|
|
|
|
return nil, errors.Wrap(err, "could not construct seeder")
|
|
|
|
}
|
|
|
|
|
|
|
|
sink, _ := os.OpenFile(os.DevNull, os.O_WRONLY, 0666)
|
|
|
|
logger := log.New(sink, "zcash_seeder: ", log.Ldate|log.Ltime|log.Lshortfile|log.LUTC)
|
|
|
|
|
|
|
|
// Allows connections to self for easy mocking
|
|
|
|
config.AllowSelfConns = true
|
|
|
|
|
|
|
|
newSeeder := Seeder{
|
|
|
|
config: config,
|
|
|
|
logger: logger,
|
2019-10-12 16:18:25 -07:00
|
|
|
handshakeSignals: new(sync.Map),
|
|
|
|
pendingPeers: new(sync.Map),
|
|
|
|
livePeers: new(sync.Map),
|
2019-10-12 19:00:20 -07:00
|
|
|
addrRecvChan: make(chan *wire.NetAddress, 100),
|
2019-10-12 09:42:16 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
newSeeder.config.Listeners.OnVerAck = newSeeder.onVerAck
|
2019-10-12 19:00:20 -07:00
|
|
|
newSeeder.config.Listeners.OnAddr = newSeeder.onAddr
|
2019-10-12 09:42:16 -07:00
|
|
|
|
|
|
|
return &newSeeder, nil
|
|
|
|
}
|
|
|
|
|
2019-10-09 15:32:15 -07:00
|
|
|
func newSeederPeerConfig(magic network.Network, template *peer.Config) (*peer.Config, error) {
|
|
|
|
var newPeerConfig peer.Config
|
|
|
|
|
|
|
|
// Load the default values
|
|
|
|
if template != nil {
|
|
|
|
newPeerConfig = *template
|
|
|
|
}
|
|
|
|
|
|
|
|
params, err := network.GetNetworkParams(magic)
|
|
|
|
if err != nil {
|
|
|
|
return nil, errors.Wrap(err, "couldn't construct peer config")
|
|
|
|
}
|
|
|
|
newPeerConfig.ChainParams = params
|
|
|
|
|
|
|
|
return &newPeerConfig, nil
|
|
|
|
}
|
|
|
|
|
2019-10-16 18:05:42 -07:00
|
|
|
// GetNetworkDefaultPort returns the default port of the network this seeder is configured for.
|
|
|
|
func (s *Seeder) GetNetworkDefaultPort() string {
|
|
|
|
return s.config.ChainParams.DefaultPort
|
|
|
|
}
|
|
|
|
|
2019-10-12 09:42:16 -07:00
|
|
|
func (s *Seeder) onVerAck(p *peer.Peer, msg *wire.MsgVerAck) {
|
2019-10-12 16:18:25 -07:00
|
|
|
// Check if we're expecting to hear from this peer
|
|
|
|
_, ok := s.pendingPeers.Load(p.Addr())
|
2019-10-09 15:32:15 -07:00
|
|
|
|
2019-10-12 16:18:25 -07:00
|
|
|
if !ok {
|
2019-10-12 09:42:16 -07:00
|
|
|
s.logger.Printf("Got verack from unexpected peer %s", p.Addr())
|
|
|
|
return
|
|
|
|
}
|
2019-10-09 15:32:15 -07:00
|
|
|
|
2019-10-12 16:18:25 -07:00
|
|
|
// Add to set of live peers
|
|
|
|
s.livePeers.Store(p.Addr(), p)
|
2019-10-09 15:32:15 -07:00
|
|
|
|
2019-10-12 16:18:25 -07:00
|
|
|
// Remove from set of pending peers
|
|
|
|
s.pendingPeers.Delete(p.Addr())
|
2019-10-09 15:32:15 -07:00
|
|
|
|
2019-10-12 16:18:25 -07:00
|
|
|
// Signal successful connection
|
|
|
|
if signal, ok := s.handshakeSignals.Load(p.Addr()); ok {
|
|
|
|
signal.(chan struct{}) <- struct{}{}
|
|
|
|
return
|
2019-10-09 15:32:15 -07:00
|
|
|
}
|
|
|
|
|
2019-10-12 16:18:25 -07:00
|
|
|
s.logger.Printf("Got verack from peer without a callback channel: %s", p.Addr())
|
2019-10-09 15:32:15 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
// ConnectToPeer attempts to connect to a peer on the default port at the
|
2019-10-16 18:05:42 -07:00
|
|
|
// specified address. It returns an error if it can't complete handshake with
|
|
|
|
// the peer. Otherwise it returns nil and adds the peer to the list of live
|
|
|
|
// connections and known-good addresses.
|
|
|
|
func (s *Seeder) ConnectOnDefaultPort(addr string) error {
|
|
|
|
return s.Connect(addr, s.config.ChainParams.DefaultPort)
|
|
|
|
}
|
2019-10-09 15:32:15 -07:00
|
|
|
|
2019-10-16 18:05:42 -07:00
|
|
|
func (s *Seeder) Connect(addr, port string) error {
|
|
|
|
connectionString := net.JoinHostPort(addr, port)
|
2019-10-09 15:32:15 -07:00
|
|
|
p, err := peer.NewOutboundPeer(s.config, connectionString)
|
|
|
|
if err != nil {
|
2019-10-12 09:42:16 -07:00
|
|
|
return errors.Wrap(err, "constructing outbound peer")
|
2019-10-09 15:32:15 -07:00
|
|
|
}
|
|
|
|
|
2019-10-12 16:18:25 -07:00
|
|
|
_, alreadyPending := s.pendingPeers.LoadOrStore(p.Addr(), p)
|
|
|
|
_, alreadyHandshaking := s.handshakeSignals.LoadOrStore(p.Addr(), make(chan struct{}, 1))
|
|
|
|
_, alreadyLive := s.livePeers.Load(p.Addr())
|
|
|
|
|
|
|
|
if alreadyPending || alreadyHandshaking || alreadyLive {
|
|
|
|
s.logger.Printf("Attempted repeat connection to peer %s", p.Addr())
|
|
|
|
return ErrRepeatConnection
|
|
|
|
}
|
|
|
|
|
2019-10-09 15:32:15 -07:00
|
|
|
conn, err := net.Dial("tcp", p.Addr())
|
|
|
|
if err != nil {
|
2019-10-12 09:42:16 -07:00
|
|
|
return errors.Wrap(err, "dialing new peer address")
|
2019-10-09 15:32:15 -07:00
|
|
|
}
|
|
|
|
|
2019-10-12 09:42:16 -07:00
|
|
|
// Begin connection negotiation.
|
|
|
|
s.logger.Printf("Handshake initated with new peer %s", p.Addr())
|
2019-10-09 15:32:15 -07:00
|
|
|
p.AssociateConnection(conn)
|
|
|
|
|
2019-10-12 19:00:20 -07:00
|
|
|
// TODO: handle disconnect during this
|
2019-10-12 16:18:25 -07:00
|
|
|
handshakeChan, _ := s.handshakeSignals.Load(p.Addr())
|
|
|
|
|
|
|
|
select {
|
|
|
|
case <-handshakeChan.(chan struct{}):
|
|
|
|
s.logger.Printf("Handshake completed with new peer %s", p.Addr())
|
|
|
|
s.handshakeSignals.Delete(p.Addr())
|
|
|
|
return nil
|
|
|
|
case <-time.After(time.Second * 1):
|
|
|
|
return errors.New("peer handshake timed out")
|
2019-10-09 15:32:15 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
panic("This should be unreachable")
|
|
|
|
}
|
|
|
|
|
2019-10-16 18:05:42 -07:00
|
|
|
// GetPeer returns a live peer identified by "host:port" string, or an error if
|
|
|
|
// we aren't connected to that peer.
|
2019-10-12 09:42:16 -07:00
|
|
|
func (s *Seeder) GetPeer(addr string) (*peer.Peer, error) {
|
2019-10-16 18:05:42 -07:00
|
|
|
p, ok := s.livePeers.Load(addr)
|
2019-10-09 15:32:15 -07:00
|
|
|
|
2019-10-12 16:18:25 -07:00
|
|
|
if ok {
|
|
|
|
return p.(*peer.Peer), nil
|
2019-10-09 15:32:15 -07:00
|
|
|
}
|
|
|
|
|
2019-10-12 19:00:20 -07:00
|
|
|
return nil, ErrNoSuchPeer
|
|
|
|
}
|
|
|
|
|
2019-10-16 18:05:42 -07:00
|
|
|
// DisconnectPeer disconnects from a live peer identified by "host:port"
|
|
|
|
// string. It returns an error if we aren't connected to that peer.
|
2019-10-12 19:00:20 -07:00
|
|
|
func (s *Seeder) DisconnectPeer(addr string) error {
|
2019-10-16 18:05:42 -07:00
|
|
|
p, ok := s.livePeers.Load(addr)
|
2019-10-12 19:00:20 -07:00
|
|
|
|
|
|
|
if !ok {
|
|
|
|
return ErrNoSuchPeer
|
|
|
|
}
|
|
|
|
|
|
|
|
// TODO: type safety and error handling
|
|
|
|
|
|
|
|
v := p.(*peer.Peer)
|
|
|
|
v.Disconnect()
|
|
|
|
v.WaitForDisconnect()
|
2019-10-16 18:05:42 -07:00
|
|
|
s.livePeers.Delete(addr)
|
2019-10-12 19:00:20 -07:00
|
|
|
|
|
|
|
return nil
|
2019-10-12 09:42:16 -07:00
|
|
|
}
|
|
|
|
|
2019-10-16 18:05:42 -07:00
|
|
|
// DisconnectAllPeers terminates the connections to all live and pending peers.
|
2019-10-12 09:42:16 -07:00
|
|
|
func (s *Seeder) DisconnectAllPeers() {
|
2019-10-12 16:18:25 -07:00
|
|
|
s.pendingPeers.Range(func(key, value interface{}) bool {
|
|
|
|
p, ok := value.(*peer.Peer)
|
|
|
|
if !ok {
|
|
|
|
s.logger.Printf("Invalid peer in pendingPeers")
|
|
|
|
return false
|
2019-10-12 09:42:16 -07:00
|
|
|
}
|
2019-10-12 16:18:25 -07:00
|
|
|
s.logger.Printf("Disconnecting from pending peer %s", p.Addr())
|
|
|
|
p.Disconnect()
|
|
|
|
p.WaitForDisconnect()
|
|
|
|
s.pendingPeers.Delete(key)
|
|
|
|
return true
|
|
|
|
})
|
|
|
|
|
|
|
|
s.livePeers.Range(func(key, value interface{}) bool {
|
|
|
|
p, ok := value.(*peer.Peer)
|
|
|
|
if !ok {
|
|
|
|
s.logger.Printf("Invalid peer in livePeers")
|
|
|
|
return false
|
2019-10-12 09:42:16 -07:00
|
|
|
}
|
2019-10-12 16:18:25 -07:00
|
|
|
s.logger.Printf("Disconnecting from live peer %s", p.Addr())
|
2019-10-12 19:00:20 -07:00
|
|
|
s.DisconnectPeer(p.Addr())
|
|
|
|
return true
|
|
|
|
})
|
|
|
|
}
|
|
|
|
|
|
|
|
func (s *Seeder) RequestAddresses() {
|
|
|
|
s.livePeers.Range(func(key, value interface{}) bool {
|
|
|
|
p, ok := value.(*peer.Peer)
|
|
|
|
if !ok {
|
|
|
|
s.logger.Printf("Invalid peer in livePeers")
|
|
|
|
return false
|
|
|
|
}
|
|
|
|
s.logger.Printf("Requesting addresses from peer %s", p.Addr())
|
|
|
|
p.QueueMessage(wire.NewMsgGetAddr(), nil)
|
2019-10-12 16:18:25 -07:00
|
|
|
return true
|
|
|
|
})
|
2019-10-09 15:32:15 -07:00
|
|
|
}
|
2019-10-12 16:18:25 -07:00
|
|
|
|
2019-10-12 19:00:20 -07:00
|
|
|
func (s *Seeder) WaitForMoreAddresses() {
|
|
|
|
<-s.addrRecvChan
|
|
|
|
}
|
|
|
|
|
|
|
|
func (s *Seeder) onAddr(p *peer.Peer, msg *wire.MsgAddr) {
|
|
|
|
if len(msg.AddrList) == 0 {
|
|
|
|
s.logger.Printf("Got empty addr message from peer %s. Disconnecting.", p.Addr())
|
|
|
|
s.DisconnectPeer(p.Addr())
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
s.logger.Printf("Got %d addrs from peer %s", len(msg.AddrList), p.Addr())
|
|
|
|
for _, addr := range msg.AddrList {
|
|
|
|
s.addrRecvChan <- addr
|
|
|
|
}
|
|
|
|
}
|