2015-07-07 03:54:22 +03:00
|
|
|
// Copyright 2015 The go-ethereum Authors
|
2015-07-22 19:48:40 +03:00
|
|
|
// This file is part of the go-ethereum library.
|
2015-07-07 03:54:22 +03:00
|
|
|
//
|
2015-07-23 19:35:11 +03:00
|
|
|
// The go-ethereum library is free software: you can redistribute it and/or modify
|
2015-07-07 03:54:22 +03:00
|
|
|
// 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.
|
|
|
|
//
|
2015-07-22 19:48:40 +03:00
|
|
|
// The go-ethereum library is distributed in the hope that it will be useful,
|
2015-07-07 03:54:22 +03:00
|
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
2015-07-22 19:48:40 +03:00
|
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
2015-07-07 03:54:22 +03:00
|
|
|
// GNU Lesser General Public License for more details.
|
|
|
|
//
|
|
|
|
// You should have received a copy of the GNU Lesser General Public License
|
2015-07-22 19:48:40 +03:00
|
|
|
// along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.
|
2015-07-07 03:54:22 +03:00
|
|
|
|
2015-05-11 14:26:20 +03:00
|
|
|
// Contains the active peer-set of the downloader, maintaining both failures
|
|
|
|
// as well as reputation metrics to prioritize the block retrievals.
|
|
|
|
|
2015-04-12 13:38:25 +03:00
|
|
|
package downloader
|
|
|
|
|
|
|
|
import (
|
2015-04-12 14:33:42 +03:00
|
|
|
"errors"
|
2015-06-04 14:51:14 +03:00
|
|
|
"fmt"
|
2015-06-03 14:39:21 +03:00
|
|
|
"math"
|
2015-04-12 13:38:25 +03:00
|
|
|
"sync"
|
2015-05-11 14:26:20 +03:00
|
|
|
"sync/atomic"
|
2015-06-03 14:39:21 +03:00
|
|
|
"time"
|
2015-04-12 13:38:25 +03:00
|
|
|
|
|
|
|
"github.com/ethereum/go-ethereum/common"
|
2015-04-18 19:54:57 +03:00
|
|
|
"gopkg.in/fatih/set.v0"
|
2015-04-12 13:38:25 +03:00
|
|
|
)
|
|
|
|
|
2015-08-14 21:25:41 +03:00
|
|
|
// Hash and block fetchers belonging to eth/61 and below
|
2015-06-30 19:05:06 +03:00
|
|
|
type relativeHashFetcherFn func(common.Hash) error
|
|
|
|
type absoluteHashFetcherFn func(uint64, int) error
|
2015-04-13 17:38:32 +03:00
|
|
|
type blockFetcherFn func([]common.Hash) error
|
|
|
|
|
2015-09-28 19:27:31 +03:00
|
|
|
// Block header and body fetchers belonging to eth/62 and above
|
2015-08-14 21:25:41 +03:00
|
|
|
type relativeHeaderFetcherFn func(common.Hash, int, int, bool) error
|
|
|
|
type absoluteHeaderFetcherFn func(uint64, int, int, bool) error
|
|
|
|
type blockBodyFetcherFn func([]common.Hash) error
|
2015-09-28 19:27:31 +03:00
|
|
|
type receiptFetcherFn func([]common.Hash) error
|
2015-10-05 19:37:56 +03:00
|
|
|
type stateFetcherFn func([]common.Hash) error
|
2015-08-14 21:25:41 +03:00
|
|
|
|
2015-05-11 14:26:20 +03:00
|
|
|
var (
|
|
|
|
errAlreadyFetching = errors.New("already fetching blocks from peer")
|
|
|
|
errAlreadyRegistered = errors.New("peer is already registered")
|
|
|
|
errNotRegistered = errors.New("peer is not registered")
|
|
|
|
)
|
2015-04-13 17:38:32 +03:00
|
|
|
|
2015-05-11 14:26:20 +03:00
|
|
|
// peer represents an active peer from which hashes and blocks are retrieved.
|
|
|
|
type peer struct {
|
|
|
|
id string // Unique identifier of the peer
|
|
|
|
head common.Hash // Hash of the peers latest known block
|
|
|
|
|
2015-09-28 19:27:31 +03:00
|
|
|
blockIdle int32 // Current block activity state of the peer (idle = 0, active = 1)
|
|
|
|
receiptIdle int32 // Current receipt activity state of the peer (idle = 0, active = 1)
|
2015-10-05 19:37:56 +03:00
|
|
|
stateIdle int32 // Current node data activity state of the peer (idle = 0, active = 1)
|
2015-09-28 19:27:31 +03:00
|
|
|
rep int32 // Simple peer reputation
|
2015-05-11 14:26:20 +03:00
|
|
|
|
2015-10-05 19:37:56 +03:00
|
|
|
blockCapacity int32 // Number of blocks (bodies) allowed to fetch per request
|
|
|
|
receiptCapacity int32 // Number of receipts allowed to fetch per request
|
|
|
|
stateCapacity int32 // Number of node data pieces allowed to fetch per request
|
|
|
|
|
|
|
|
blockStarted time.Time // Time instance when the last block (body)fetch was started
|
|
|
|
receiptStarted time.Time // Time instance when the last receipt fetch was started
|
|
|
|
stateStarted time.Time // Time instance when the last node data fetch was started
|
2015-05-11 14:26:20 +03:00
|
|
|
|
2015-06-03 14:39:21 +03:00
|
|
|
ignored *set.Set // Set of hashes not to request (didn't have previously)
|
2015-05-11 14:26:20 +03:00
|
|
|
|
2015-08-14 21:25:41 +03:00
|
|
|
getRelHashes relativeHashFetcherFn // [eth/61] Method to retrieve a batch of hashes from an origin hash
|
|
|
|
getAbsHashes absoluteHashFetcherFn // [eth/61] Method to retrieve a batch of hashes from an absolute position
|
|
|
|
getBlocks blockFetcherFn // [eth/61] Method to retrieve a batch of blocks
|
|
|
|
|
|
|
|
getRelHeaders relativeHeaderFetcherFn // [eth/62] Method to retrieve a batch of headers from an origin hash
|
|
|
|
getAbsHeaders absoluteHeaderFetcherFn // [eth/62] Method to retrieve a batch of headers from an absolute position
|
|
|
|
getBlockBodies blockBodyFetcherFn // [eth/62] Method to retrieve a batch of block bodies
|
2015-06-29 17:37:55 +03:00
|
|
|
|
2015-09-28 19:27:31 +03:00
|
|
|
getReceipts receiptFetcherFn // [eth/63] Method to retrieve a batch of block transaction receipts
|
2015-10-05 19:37:56 +03:00
|
|
|
getNodeData stateFetcherFn // [eth/63] Method to retrieve a batch of state trie data
|
2015-09-28 19:27:31 +03:00
|
|
|
|
2015-06-29 17:37:55 +03:00
|
|
|
version int // Eth protocol version number to switch strategies
|
2015-04-18 21:25:55 +03:00
|
|
|
}
|
|
|
|
|
2015-05-11 14:26:20 +03:00
|
|
|
// newPeer create a new downloader peer, with specific hash and block retrieval
|
|
|
|
// mechanisms.
|
2015-08-14 21:25:41 +03:00
|
|
|
func newPeer(id string, version int, head common.Hash,
|
|
|
|
getRelHashes relativeHashFetcherFn, getAbsHashes absoluteHashFetcherFn, getBlocks blockFetcherFn, // eth/61 callbacks, remove when upgrading
|
2015-09-28 19:27:31 +03:00
|
|
|
getRelHeaders relativeHeaderFetcherFn, getAbsHeaders absoluteHeaderFetcherFn, getBlockBodies blockBodyFetcherFn,
|
2015-10-05 19:37:56 +03:00
|
|
|
getReceipts receiptFetcherFn, getNodeData stateFetcherFn) *peer {
|
2015-05-11 14:26:20 +03:00
|
|
|
return &peer{
|
2015-09-28 19:27:31 +03:00
|
|
|
id: id,
|
|
|
|
head: head,
|
|
|
|
blockCapacity: 1,
|
|
|
|
receiptCapacity: 1,
|
2015-10-05 19:37:56 +03:00
|
|
|
stateCapacity: 1,
|
2015-09-28 19:27:31 +03:00
|
|
|
ignored: set.New(),
|
2015-08-14 21:25:41 +03:00
|
|
|
|
2015-06-30 19:05:06 +03:00
|
|
|
getRelHashes: getRelHashes,
|
|
|
|
getAbsHashes: getAbsHashes,
|
|
|
|
getBlocks: getBlocks,
|
2015-08-14 21:25:41 +03:00
|
|
|
|
|
|
|
getRelHeaders: getRelHeaders,
|
|
|
|
getAbsHeaders: getAbsHeaders,
|
|
|
|
getBlockBodies: getBlockBodies,
|
|
|
|
|
2015-09-28 19:27:31 +03:00
|
|
|
getReceipts: getReceipts,
|
2015-10-05 19:37:56 +03:00
|
|
|
getNodeData: getNodeData,
|
2015-09-28 19:27:31 +03:00
|
|
|
|
2015-08-14 21:25:41 +03:00
|
|
|
version: version,
|
2015-04-13 17:38:32 +03:00
|
|
|
}
|
2015-05-11 14:26:20 +03:00
|
|
|
}
|
2015-04-13 17:38:32 +03:00
|
|
|
|
2015-05-11 14:26:20 +03:00
|
|
|
// Reset clears the internal state of a peer entity.
|
|
|
|
func (p *peer) Reset() {
|
2015-09-28 19:27:31 +03:00
|
|
|
atomic.StoreInt32(&p.blockIdle, 0)
|
|
|
|
atomic.StoreInt32(&p.receiptIdle, 0)
|
|
|
|
atomic.StoreInt32(&p.blockCapacity, 1)
|
|
|
|
atomic.StoreInt32(&p.receiptCapacity, 1)
|
2015-10-05 19:37:56 +03:00
|
|
|
atomic.StoreInt32(&p.stateCapacity, 1)
|
2015-05-11 14:26:20 +03:00
|
|
|
p.ignored.Clear()
|
2015-04-13 17:38:32 +03:00
|
|
|
}
|
|
|
|
|
2015-08-14 21:25:41 +03:00
|
|
|
// Fetch61 sends a block retrieval request to the remote peer.
|
|
|
|
func (p *peer) Fetch61(request *fetchRequest) error {
|
2015-05-11 14:26:20 +03:00
|
|
|
// Short circuit if the peer is already fetching
|
2015-09-28 19:27:31 +03:00
|
|
|
if !atomic.CompareAndSwapInt32(&p.blockIdle, 0, 1) {
|
2015-05-11 14:26:20 +03:00
|
|
|
return errAlreadyFetching
|
2015-04-13 17:38:32 +03:00
|
|
|
}
|
2015-09-28 19:27:31 +03:00
|
|
|
p.blockStarted = time.Now()
|
2015-06-03 14:39:21 +03:00
|
|
|
|
2015-05-11 14:26:20 +03:00
|
|
|
// Convert the hash set to a retrievable slice
|
|
|
|
hashes := make([]common.Hash, 0, len(request.Hashes))
|
|
|
|
for hash, _ := range request.Hashes {
|
|
|
|
hashes = append(hashes, hash)
|
|
|
|
}
|
2015-06-15 12:26:05 +03:00
|
|
|
go p.getBlocks(hashes)
|
2015-04-13 17:38:32 +03:00
|
|
|
|
2015-05-11 14:26:20 +03:00
|
|
|
return nil
|
2015-04-13 17:38:32 +03:00
|
|
|
}
|
|
|
|
|
2015-09-28 19:27:31 +03:00
|
|
|
// FetchBodies sends a block body retrieval request to the remote peer.
|
|
|
|
func (p *peer) FetchBodies(request *fetchRequest) error {
|
2015-08-14 21:25:41 +03:00
|
|
|
// Short circuit if the peer is already fetching
|
2015-09-28 19:27:31 +03:00
|
|
|
if !atomic.CompareAndSwapInt32(&p.blockIdle, 0, 1) {
|
2015-08-14 21:25:41 +03:00
|
|
|
return errAlreadyFetching
|
|
|
|
}
|
2015-09-28 19:27:31 +03:00
|
|
|
p.blockStarted = time.Now()
|
2015-08-14 21:25:41 +03:00
|
|
|
|
|
|
|
// Convert the header set to a retrievable slice
|
|
|
|
hashes := make([]common.Hash, 0, len(request.Headers))
|
|
|
|
for _, header := range request.Headers {
|
|
|
|
hashes = append(hashes, header.Hash())
|
|
|
|
}
|
|
|
|
go p.getBlockBodies(hashes)
|
|
|
|
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2015-09-28 19:27:31 +03:00
|
|
|
// FetchReceipts sends a receipt retrieval request to the remote peer.
|
|
|
|
func (p *peer) FetchReceipts(request *fetchRequest) error {
|
|
|
|
// Short circuit if the peer is already fetching
|
|
|
|
if !atomic.CompareAndSwapInt32(&p.receiptIdle, 0, 1) {
|
|
|
|
return errAlreadyFetching
|
2015-06-03 14:39:21 +03:00
|
|
|
}
|
2015-09-28 19:27:31 +03:00
|
|
|
p.receiptStarted = time.Now()
|
2015-06-03 15:43:12 +03:00
|
|
|
|
2015-09-28 19:27:31 +03:00
|
|
|
// Convert the header set to a retrievable slice
|
|
|
|
hashes := make([]common.Hash, 0, len(request.Headers))
|
|
|
|
for _, header := range request.Headers {
|
|
|
|
hashes = append(hashes, header.Hash())
|
2015-06-03 14:39:21 +03:00
|
|
|
}
|
2015-09-28 19:27:31 +03:00
|
|
|
go p.getReceipts(hashes)
|
|
|
|
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2015-10-05 19:37:56 +03:00
|
|
|
// FetchNodeData sends a node state data retrieval request to the remote peer.
|
|
|
|
func (p *peer) FetchNodeData(request *fetchRequest) error {
|
|
|
|
// Short circuit if the peer is already fetching
|
|
|
|
if !atomic.CompareAndSwapInt32(&p.stateIdle, 0, 1) {
|
|
|
|
return errAlreadyFetching
|
|
|
|
}
|
|
|
|
p.stateStarted = time.Now()
|
|
|
|
|
|
|
|
// Convert the hash set to a retrievable slice
|
|
|
|
hashes := make([]common.Hash, 0, len(request.Hashes))
|
|
|
|
for hash, _ := range request.Hashes {
|
|
|
|
hashes = append(hashes, hash)
|
|
|
|
}
|
|
|
|
go p.getNodeData(hashes)
|
|
|
|
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2015-09-28 19:27:31 +03:00
|
|
|
// SetBlocksIdle sets the peer to idle, allowing it to execute new retrieval requests.
|
|
|
|
// Its block retrieval allowance will also be updated either up- or downwards,
|
|
|
|
// depending on whether the previous fetch completed in time or not.
|
|
|
|
func (p *peer) SetBlocksIdle() {
|
|
|
|
p.setIdle(p.blockStarted, blockSoftTTL, blockHardTTL, MaxBlockFetch, &p.blockCapacity, &p.blockIdle)
|
2015-05-11 14:26:20 +03:00
|
|
|
}
|
2015-04-12 13:38:25 +03:00
|
|
|
|
2015-09-28 19:27:31 +03:00
|
|
|
// SetBodiesIdle sets the peer to idle, allowing it to execute new retrieval requests.
|
2015-08-14 21:25:41 +03:00
|
|
|
// Its block body retrieval allowance will also be updated either up- or downwards,
|
|
|
|
// depending on whether the previous fetch completed in time or not.
|
2015-09-28 19:27:31 +03:00
|
|
|
func (p *peer) SetBodiesIdle() {
|
|
|
|
p.setIdle(p.blockStarted, bodySoftTTL, bodyHardTTL, MaxBlockFetch, &p.blockCapacity, &p.blockIdle)
|
|
|
|
}
|
|
|
|
|
|
|
|
// SetReceiptsIdle sets the peer to idle, allowing it to execute new retrieval requests.
|
|
|
|
// Its receipt retrieval allowance will also be updated either up- or downwards,
|
|
|
|
// depending on whether the previous fetch completed in time or not.
|
|
|
|
func (p *peer) SetReceiptsIdle() {
|
|
|
|
p.setIdle(p.receiptStarted, receiptSoftTTL, receiptHardTTL, MaxReceiptFetch, &p.receiptCapacity, &p.receiptIdle)
|
|
|
|
}
|
|
|
|
|
2015-10-05 19:37:56 +03:00
|
|
|
// SetNodeDataIdle sets the peer to idle, allowing it to execute new retrieval
|
|
|
|
// requests. Its node data retrieval allowance will also be updated either up- or
|
|
|
|
// downwards, depending on whether the previous fetch completed in time or not.
|
|
|
|
func (p *peer) SetNodeDataIdle() {
|
|
|
|
p.setIdle(p.stateStarted, stateSoftTTL, stateSoftTTL, MaxStateFetch, &p.stateCapacity, &p.stateIdle)
|
|
|
|
}
|
|
|
|
|
2015-09-28 19:27:31 +03:00
|
|
|
// setIdle sets the peer to idle, allowing it to execute new retrieval requests.
|
|
|
|
// Its data retrieval allowance will also be updated either up- or downwards,
|
|
|
|
// depending on whether the previous fetch completed in time or not.
|
|
|
|
func (p *peer) setIdle(started time.Time, softTTL, hardTTL time.Duration, maxFetch int, capacity, idle *int32) {
|
2015-08-14 21:25:41 +03:00
|
|
|
// Update the peer's download allowance based on previous performance
|
|
|
|
scale := 2.0
|
2015-09-28 19:27:31 +03:00
|
|
|
if time.Since(started) > softTTL {
|
2015-08-14 21:25:41 +03:00
|
|
|
scale = 0.5
|
2015-09-28 19:27:31 +03:00
|
|
|
if time.Since(started) > hardTTL {
|
|
|
|
scale = 1 / float64(maxFetch) // reduces capacity to 1
|
2015-08-14 21:25:41 +03:00
|
|
|
}
|
|
|
|
}
|
|
|
|
for {
|
|
|
|
// Calculate the new download bandwidth allowance
|
2015-09-28 19:27:31 +03:00
|
|
|
prev := atomic.LoadInt32(capacity)
|
|
|
|
next := int32(math.Max(1, math.Min(float64(maxFetch), float64(prev)*scale)))
|
2015-08-14 21:25:41 +03:00
|
|
|
|
|
|
|
// Try to update the old value
|
2015-09-28 19:27:31 +03:00
|
|
|
if atomic.CompareAndSwapInt32(capacity, prev, next) {
|
2015-08-14 21:25:41 +03:00
|
|
|
// If we're having problems at 1 capacity, try to find better peers
|
|
|
|
if next == 1 {
|
|
|
|
p.Demote()
|
|
|
|
}
|
|
|
|
break
|
|
|
|
}
|
|
|
|
}
|
2015-09-28 19:27:31 +03:00
|
|
|
// Set the peer to idle to allow further fetch requests
|
|
|
|
atomic.StoreInt32(idle, 0)
|
|
|
|
}
|
|
|
|
|
|
|
|
// BlockCapacity retrieves the peers block download allowance based on its
|
|
|
|
// previously discovered bandwidth capacity.
|
|
|
|
func (p *peer) BlockCapacity() int {
|
|
|
|
return int(atomic.LoadInt32(&p.blockCapacity))
|
2015-08-14 21:25:41 +03:00
|
|
|
}
|
|
|
|
|
2015-09-28 19:27:31 +03:00
|
|
|
// ReceiptCapacity retrieves the peers block download allowance based on its
|
|
|
|
// previously discovered bandwidth capacity.
|
|
|
|
func (p *peer) ReceiptCapacity() int {
|
|
|
|
return int(atomic.LoadInt32(&p.receiptCapacity))
|
2015-06-03 14:39:21 +03:00
|
|
|
}
|
|
|
|
|
2015-10-05 19:37:56 +03:00
|
|
|
// NodeDataCapacity retrieves the peers block download allowance based on its
|
|
|
|
// previously discovered bandwidth capacity.
|
|
|
|
func (p *peer) NodeDataCapacity() int {
|
|
|
|
return int(atomic.LoadInt32(&p.stateCapacity))
|
|
|
|
}
|
|
|
|
|
2015-05-11 14:26:20 +03:00
|
|
|
// Promote increases the peer's reputation.
|
|
|
|
func (p *peer) Promote() {
|
|
|
|
atomic.AddInt32(&p.rep, 1)
|
|
|
|
}
|
2015-04-12 13:38:25 +03:00
|
|
|
|
2015-05-11 14:26:20 +03:00
|
|
|
// Demote decreases the peer's reputation or leaves it at 0.
|
|
|
|
func (p *peer) Demote() {
|
|
|
|
for {
|
|
|
|
// Calculate the new reputation value
|
|
|
|
prev := atomic.LoadInt32(&p.rep)
|
2015-05-11 16:47:58 +03:00
|
|
|
next := prev / 2
|
|
|
|
|
2015-05-11 14:26:20 +03:00
|
|
|
// Try to update the old value
|
|
|
|
if atomic.CompareAndSwapInt32(&p.rep, prev, next) {
|
|
|
|
return
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2015-04-18 19:54:57 +03:00
|
|
|
|
2015-06-04 14:51:14 +03:00
|
|
|
// String implements fmt.Stringer.
|
|
|
|
func (p *peer) String() string {
|
|
|
|
return fmt.Sprintf("Peer %s [%s]", p.id,
|
|
|
|
fmt.Sprintf("reputation %3d, ", atomic.LoadInt32(&p.rep))+
|
2015-09-28 19:27:31 +03:00
|
|
|
fmt.Sprintf("block cap %3d, ", atomic.LoadInt32(&p.blockCapacity))+
|
|
|
|
fmt.Sprintf("receipt cap %3d, ", atomic.LoadInt32(&p.receiptCapacity))+
|
2015-06-04 14:51:14 +03:00
|
|
|
fmt.Sprintf("ignored %4d", p.ignored.Size()),
|
|
|
|
)
|
|
|
|
}
|
|
|
|
|
2015-05-11 14:26:20 +03:00
|
|
|
// peerSet represents the collection of active peer participating in the block
|
|
|
|
// download procedure.
|
|
|
|
type peerSet struct {
|
|
|
|
peers map[string]*peer
|
|
|
|
lock sync.RWMutex
|
2015-04-12 13:38:25 +03:00
|
|
|
}
|
|
|
|
|
2015-05-11 14:26:20 +03:00
|
|
|
// newPeerSet creates a new peer set top track the active download sources.
|
|
|
|
func newPeerSet() *peerSet {
|
|
|
|
return &peerSet{
|
|
|
|
peers: make(map[string]*peer),
|
2015-04-18 19:54:57 +03:00
|
|
|
}
|
2015-04-12 13:38:25 +03:00
|
|
|
}
|
|
|
|
|
2015-05-11 14:26:20 +03:00
|
|
|
// Reset iterates over the current peer set, and resets each of the known peers
|
|
|
|
// to prepare for a next batch of block retrieval.
|
|
|
|
func (ps *peerSet) Reset() {
|
|
|
|
ps.lock.RLock()
|
|
|
|
defer ps.lock.RUnlock()
|
2015-04-12 13:38:25 +03:00
|
|
|
|
2015-05-11 14:26:20 +03:00
|
|
|
for _, peer := range ps.peers {
|
|
|
|
peer.Reset()
|
2015-04-12 14:33:42 +03:00
|
|
|
}
|
2015-05-11 14:26:20 +03:00
|
|
|
}
|
2015-04-12 14:33:42 +03:00
|
|
|
|
2015-05-11 14:26:20 +03:00
|
|
|
// Register injects a new peer into the working set, or returns an error if the
|
|
|
|
// peer is already known.
|
|
|
|
func (ps *peerSet) Register(p *peer) error {
|
|
|
|
ps.lock.Lock()
|
|
|
|
defer ps.lock.Unlock()
|
2015-05-06 15:32:53 +03:00
|
|
|
|
2015-05-11 14:26:20 +03:00
|
|
|
if _, ok := ps.peers[p.id]; ok {
|
|
|
|
return errAlreadyRegistered
|
2015-05-06 15:32:53 +03:00
|
|
|
}
|
2015-05-11 14:26:20 +03:00
|
|
|
ps.peers[p.id] = p
|
|
|
|
return nil
|
|
|
|
}
|
2015-04-12 14:33:42 +03:00
|
|
|
|
2015-05-11 14:26:20 +03:00
|
|
|
// Unregister removes a remote peer from the active set, disabling any further
|
|
|
|
// actions to/from that particular entity.
|
|
|
|
func (ps *peerSet) Unregister(id string) error {
|
|
|
|
ps.lock.Lock()
|
|
|
|
defer ps.lock.Unlock()
|
|
|
|
|
|
|
|
if _, ok := ps.peers[id]; !ok {
|
|
|
|
return errNotRegistered
|
|
|
|
}
|
|
|
|
delete(ps.peers, id)
|
2015-04-12 14:33:42 +03:00
|
|
|
return nil
|
2015-04-12 13:38:25 +03:00
|
|
|
}
|
2015-04-16 03:16:33 +03:00
|
|
|
|
2015-05-11 14:26:20 +03:00
|
|
|
// Peer retrieves the registered peer with the given id.
|
|
|
|
func (ps *peerSet) Peer(id string) *peer {
|
|
|
|
ps.lock.RLock()
|
|
|
|
defer ps.lock.RUnlock()
|
|
|
|
|
|
|
|
return ps.peers[id]
|
|
|
|
}
|
|
|
|
|
2015-05-11 17:06:42 +03:00
|
|
|
// Len returns if the current number of peers in the set.
|
|
|
|
func (ps *peerSet) Len() int {
|
2015-05-11 14:26:20 +03:00
|
|
|
ps.lock.RLock()
|
|
|
|
defer ps.lock.RUnlock()
|
2015-04-16 03:16:33 +03:00
|
|
|
|
2015-05-11 14:26:20 +03:00
|
|
|
return len(ps.peers)
|
2015-04-16 03:16:33 +03:00
|
|
|
}
|
|
|
|
|
2015-05-11 14:26:20 +03:00
|
|
|
// AllPeers retrieves a flat list of all the peers within the set.
|
|
|
|
func (ps *peerSet) AllPeers() []*peer {
|
|
|
|
ps.lock.RLock()
|
|
|
|
defer ps.lock.RUnlock()
|
2015-04-16 03:16:33 +03:00
|
|
|
|
2015-05-11 14:26:20 +03:00
|
|
|
list := make([]*peer, 0, len(ps.peers))
|
|
|
|
for _, p := range ps.peers {
|
|
|
|
list = append(list, p)
|
2015-04-16 03:16:33 +03:00
|
|
|
}
|
2015-05-11 14:26:20 +03:00
|
|
|
return list
|
2015-04-16 03:16:33 +03:00
|
|
|
}
|
2015-04-18 21:25:55 +03:00
|
|
|
|
2015-09-28 19:27:31 +03:00
|
|
|
// BlockIdlePeers retrieves a flat list of all the currently idle peers within the
|
2015-05-11 16:47:58 +03:00
|
|
|
// active peer set, ordered by their reputation.
|
2015-10-05 19:37:56 +03:00
|
|
|
func (ps *peerSet) BlockIdlePeers() ([]*peer, int) {
|
|
|
|
idle := func(p *peer) bool {
|
|
|
|
return atomic.LoadInt32(&p.blockIdle) == 0
|
2015-05-11 14:26:20 +03:00
|
|
|
}
|
2015-10-05 19:37:56 +03:00
|
|
|
return ps.idlePeers(61, 61, idle)
|
|
|
|
}
|
|
|
|
|
|
|
|
// BodyIdlePeers retrieves a flat list of all the currently body-idle peers within
|
|
|
|
// the active peer set, ordered by their reputation.
|
|
|
|
func (ps *peerSet) BodyIdlePeers() ([]*peer, int) {
|
|
|
|
idle := func(p *peer) bool {
|
|
|
|
return atomic.LoadInt32(&p.blockIdle) == 0
|
2015-05-11 16:47:58 +03:00
|
|
|
}
|
2015-10-05 19:37:56 +03:00
|
|
|
return ps.idlePeers(62, 64, idle)
|
2015-09-28 19:27:31 +03:00
|
|
|
}
|
|
|
|
|
2015-10-05 19:37:56 +03:00
|
|
|
// ReceiptIdlePeers retrieves a flat list of all the currently receipt-idle peers
|
|
|
|
// within the active peer set, ordered by their reputation.
|
2015-09-28 19:27:31 +03:00
|
|
|
func (ps *peerSet) ReceiptIdlePeers() ([]*peer, int) {
|
2015-10-05 19:37:56 +03:00
|
|
|
idle := func(p *peer) bool {
|
|
|
|
return atomic.LoadInt32(&p.receiptIdle) == 0
|
|
|
|
}
|
|
|
|
return ps.idlePeers(63, 64, idle)
|
|
|
|
}
|
|
|
|
|
|
|
|
// NodeDataIdlePeers retrieves a flat list of all the currently node-data-idle
|
|
|
|
// peers within the active peer set, ordered by their reputation.
|
|
|
|
func (ps *peerSet) NodeDataIdlePeers() ([]*peer, int) {
|
|
|
|
idle := func(p *peer) bool {
|
|
|
|
return atomic.LoadInt32(&p.stateIdle) == 0
|
|
|
|
}
|
|
|
|
return ps.idlePeers(63, 64, idle)
|
|
|
|
}
|
|
|
|
|
|
|
|
// idlePeers retrieves a flat list of all currently idle peers satisfying the
|
|
|
|
// protocol version constraints, using the provided function to check idleness.
|
|
|
|
func (ps *peerSet) idlePeers(minProtocol, maxProtocol int, idleCheck func(*peer) bool) ([]*peer, int) {
|
2015-09-28 19:27:31 +03:00
|
|
|
ps.lock.RLock()
|
|
|
|
defer ps.lock.RUnlock()
|
|
|
|
|
|
|
|
idle, total := make([]*peer, 0, len(ps.peers)), 0
|
|
|
|
for _, p := range ps.peers {
|
2015-10-05 19:37:56 +03:00
|
|
|
if p.version >= minProtocol && p.version <= maxProtocol {
|
|
|
|
if idleCheck(p) {
|
2015-09-28 19:27:31 +03:00
|
|
|
idle = append(idle, p)
|
|
|
|
}
|
|
|
|
total++
|
|
|
|
}
|
|
|
|
}
|
|
|
|
for i := 0; i < len(idle); i++ {
|
|
|
|
for j := i + 1; j < len(idle); j++ {
|
|
|
|
if atomic.LoadInt32(&idle[i].rep) < atomic.LoadInt32(&idle[j].rep) {
|
|
|
|
idle[i], idle[j] = idle[j], idle[i]
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return idle, total
|
2015-04-18 21:25:55 +03:00
|
|
|
}
|