2018-06-20 07:06:27 -05:00
// Copyright 2017 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 network
import (
"bytes"
"fmt"
"math/rand"
"strings"
"sync"
"time"
"github.com/ethereum/go-ethereum/common"
"github.com/ethereum/go-ethereum/swarm/log"
"github.com/ethereum/go-ethereum/swarm/pot"
2019-01-24 05:35:10 -06:00
sv "github.com/ethereum/go-ethereum/swarm/version"
2018-06-20 07:06:27 -05:00
)
/ *
Taking the proximity order relative to a fix point x classifies the points in
the space ( n byte long byte sequences ) into bins . Items in each are at
most half as distant from x as items in the previous bin . Given a sample of
uniformly distributed items ( a hash function over arbitrary sequence ) the
proximity scale maps onto series of subsets with cardinalities on a negative
exponential scale .
It also has the property that any two item belonging to the same bin are at
most half as distant from each other as they are from x .
If we think of random sample of items in the bins as connections in a network of
interconnected nodes then relative proximity can serve as the basis for local
decisions for graph traversal where the task is to find a route between two
points . Since in every hop , the finite distance halves , there is
a guaranteed constant maximum limit on the number of hops needed to reach one
node from the other .
* /
2018-12-21 23:53:30 -06:00
var Pof = pot . DefaultPof ( 256 )
2018-06-20 07:06:27 -05:00
// KadParams holds the config params for Kademlia
type KadParams struct {
// adjustable parameters
2019-01-10 05:33:51 -06:00
MaxProxDisplay int // number of rows the table shows
NeighbourhoodSize int // nearest neighbour core minimum cardinality
MinBinSize int // minimum number of peers in a row
MaxBinSize int // maximum number of peers in a row before pruning
RetryInterval int64 // initial interval before a peer is first redialed
RetryExponent int // exponent to multiply retry intervals with
MaxRetries int // maximum number of redial attempts
2018-06-20 07:06:27 -05:00
// function to sanction or prevent suggesting a peer
2018-12-21 23:53:30 -06:00
Reachable func ( * BzzAddr ) bool ` json:"-" `
2018-06-20 07:06:27 -05:00
}
// NewKadParams returns a params struct with default values
func NewKadParams ( ) * KadParams {
return & KadParams {
2019-01-10 05:33:51 -06:00
MaxProxDisplay : 16 ,
NeighbourhoodSize : 2 ,
MinBinSize : 2 ,
MaxBinSize : 4 ,
RetryInterval : 4200000000 , // 4.2 sec
MaxRetries : 42 ,
RetryExponent : 2 ,
2018-06-20 07:06:27 -05:00
}
}
// Kademlia is a table of live peers and a db of known peers (node records)
type Kademlia struct {
lock sync . RWMutex
2018-12-21 23:53:30 -06:00
* KadParams // Kademlia configuration parameters
base [ ] byte // immutable baseaddress of the table
addrs * pot . Pot // pots container for known peer addresses
conns * pot . Pot // pots container for live peer connections
depth uint8 // stores the last current depth of saturation
nDepth int // stores the last neighbourhood depth
nDepthC chan int // returned by DepthC function to signal neighbourhood depth change
addrCountC chan int // returned by AddrCountC function to signal peer count change
2018-06-20 07:06:27 -05:00
}
// NewKademlia creates a Kademlia table for base address addr
// with parameters as in params
// if params is nil, it uses default values
func NewKademlia ( addr [ ] byte , params * KadParams ) * Kademlia {
if params == nil {
params = NewKadParams ( )
}
return & Kademlia {
base : addr ,
KadParams : params ,
addrs : pot . NewPot ( nil , 0 ) ,
conns : pot . NewPot ( nil , 0 ) ,
}
}
2018-09-12 04:24:56 -05:00
// entry represents a Kademlia table entry (an extension of BzzAddr)
2018-06-20 07:06:27 -05:00
type entry struct {
2018-09-12 04:24:56 -05:00
* BzzAddr
conn * Peer
2018-06-20 07:06:27 -05:00
seenAt time . Time
retries int
}
2018-09-12 04:24:56 -05:00
// newEntry creates a kademlia peer from a *Peer
func newEntry ( p * BzzAddr ) * entry {
2018-06-20 07:06:27 -05:00
return & entry {
2018-09-12 04:24:56 -05:00
BzzAddr : p ,
seenAt : time . Now ( ) ,
2018-06-20 07:06:27 -05:00
}
}
// Label is a short tag for the entry for debug
func Label ( e * entry ) string {
return fmt . Sprintf ( "%s (%d)" , e . Hex ( ) [ : 4 ] , e . retries )
}
// Hex is the hexadecimal serialisation of the entry address
func ( e * entry ) Hex ( ) string {
2018-09-12 04:24:56 -05:00
return fmt . Sprintf ( "%x" , e . Address ( ) )
2018-06-20 07:06:27 -05:00
}
2018-09-12 04:24:56 -05:00
// Register enters each address as kademlia peer record into the
2018-06-20 07:06:27 -05:00
// database of known peer addresses
2018-09-12 04:24:56 -05:00
func ( k * Kademlia ) Register ( peers ... * BzzAddr ) error {
2018-06-20 07:06:27 -05:00
k . lock . Lock ( )
defer k . lock . Unlock ( )
var known , size int
for _ , p := range peers {
// error if self received, peer should know better
// and should be punished for this
if bytes . Equal ( p . Address ( ) , k . base ) {
return fmt . Errorf ( "add peers: %x is self" , k . base )
}
var found bool
2018-12-21 23:53:30 -06:00
k . addrs , _ , found , _ = pot . Swap ( k . addrs , p , Pof , func ( v pot . Val ) pot . Val {
2018-06-20 07:06:27 -05:00
// if not found
if v == nil {
// insert new offline peer into conns
return newEntry ( p )
}
// found among known peers, do nothing
return v
} )
if found {
known ++
}
size ++
}
// send new address count value only if there are new addresses
if k . addrCountC != nil && size - known > 0 {
k . addrCountC <- k . addrs . Size ( )
}
k . sendNeighbourhoodDepthChange ( )
return nil
}
2019-01-17 00:29:34 -06:00
// SuggestPeer returns an unconnected peer address as a peer suggestion for connection
func ( k * Kademlia ) SuggestPeer ( ) ( suggestedPeer * BzzAddr , saturationDepth int , changed bool ) {
2018-06-20 07:06:27 -05:00
k . lock . Lock ( )
defer k . lock . Unlock ( )
2019-01-17 00:29:34 -06:00
radius := neighbourhoodRadiusForPot ( k . conns , k . NeighbourhoodSize , k . base )
// collect undersaturated bins in ascending order of number of connected peers
// and from shallow to deep (ascending order of PO)
// insert them in a map of bin arrays, keyed with the number of connected peers
saturation := make ( map [ int ] [ ] int )
var lastPO int // the last non-empty PO bin in the iteration
saturationDepth = - 1 // the deepest PO such that all shallower bins have >= k.MinBinSize peers
var pastDepth bool // whether po of iteration >= depth
k . conns . EachBin ( k . base , Pof , 0 , func ( po , size int , f func ( func ( val pot . Val ) bool ) bool ) bool {
// process skipped empty bins
for ; lastPO < po ; lastPO ++ {
// find the lowest unsaturated bin
if saturationDepth == - 1 {
saturationDepth = lastPO
}
// if there is an empty bin, depth is surely passed
pastDepth = true
saturation [ 0 ] = append ( saturation [ 0 ] , lastPO )
2018-06-20 07:06:27 -05:00
}
2019-01-17 00:29:34 -06:00
lastPO = po + 1
// past radius, depth is surely passed
if po >= radius {
pastDepth = true
2018-09-12 04:24:56 -05:00
}
2019-01-17 00:29:34 -06:00
// beyond depth the bin is treated as unsaturated even if size >= k.MinBinSize
// in order to achieve full connectivity to all neighbours
if pastDepth && size >= k . MinBinSize {
size = k . MinBinSize - 1
2018-06-20 07:06:27 -05:00
}
2019-01-17 00:29:34 -06:00
// process non-empty unsaturated bins
if size < k . MinBinSize {
// find the lowest unsaturated bin
if saturationDepth == - 1 {
saturationDepth = po
}
saturation [ size ] = append ( saturation [ size ] , po )
2018-06-20 07:06:27 -05:00
}
2019-01-17 00:29:34 -06:00
return true
} )
// to trigger peer requests for peers closer than closest connection, include
// all bins from nearest connection upto nearest address as unsaturated
var nearestAddrAt int
k . addrs . EachNeighbour ( k . base , Pof , func ( _ pot . Val , po int ) bool {
nearestAddrAt = po
return false
2018-06-20 07:06:27 -05:00
} )
2019-01-17 00:29:34 -06:00
// including bins as size 0 has the effect that requesting connection
// is prioritised over non-empty shallower bins
for ; lastPO <= nearestAddrAt ; lastPO ++ {
saturation [ 0 ] = append ( saturation [ 0 ] , lastPO )
}
// all PO bins are saturated, ie., minsize >= k.MinBinSize, no peer suggested
if len ( saturation ) == 0 {
2018-06-20 07:06:27 -05:00
return nil , 0 , false
}
2019-01-17 00:29:34 -06:00
// find the first callable peer in the address book
// starting from the bins with smallest size proceeding from shallow to deep
// for each bin (up until neighbourhood radius) we find callable candidate peers
for size := 0 ; size < k . MinBinSize && suggestedPeer == nil ; size ++ {
bins , ok := saturation [ size ]
if ! ok {
// no bin with this size
continue
2018-06-20 07:06:27 -05:00
}
2019-01-17 00:29:34 -06:00
cur := 0
curPO := bins [ 0 ]
k . addrs . EachBin ( k . base , Pof , curPO , func ( po , _ int , f func ( func ( pot . Val ) bool ) bool ) bool {
curPO = bins [ cur ]
// find the next bin that has size size
if curPO == po {
cur ++
} else {
// skip bins that have no addresses
for ; cur < len ( bins ) && curPO < po ; cur ++ {
curPO = bins [ cur ]
}
if po < curPO {
cur --
return true
}
// stop if there are no addresses
if curPO < po {
return false
}
2018-09-12 04:24:56 -05:00
}
2019-01-17 00:29:34 -06:00
// curPO found
// find a callable peer out of the addresses in the unsaturated bin
// stop if found
f ( func ( val pot . Val ) bool {
e := val . ( * entry )
if k . callable ( e ) {
suggestedPeer = e . BzzAddr
return false
}
return true
} )
return cur < len ( bins ) && suggestedPeer == nil
2018-06-20 07:06:27 -05:00
} )
}
2019-01-17 00:29:34 -06:00
if uint8 ( saturationDepth ) < k . depth {
k . depth = uint8 ( saturationDepth )
return suggestedPeer , saturationDepth , true
2018-06-20 07:06:27 -05:00
}
2019-01-17 00:29:34 -06:00
return suggestedPeer , 0 , false
2018-06-20 07:06:27 -05:00
}
2019-01-24 05:02:18 -06:00
// On inserts the peer as a kademlia peer into the live peers
2018-09-12 04:24:56 -05:00
func ( k * Kademlia ) On ( p * Peer ) ( uint8 , bool ) {
2018-06-20 07:06:27 -05:00
k . lock . Lock ( )
defer k . lock . Unlock ( )
var ins bool
2018-12-21 23:53:30 -06:00
k . conns , _ , _ , _ = pot . Swap ( k . conns , p , Pof , func ( v pot . Val ) pot . Val {
2018-06-20 07:06:27 -05:00
// if not found live
if v == nil {
ins = true
// insert new online peer into conns
2018-09-12 04:24:56 -05:00
return p
2018-06-20 07:06:27 -05:00
}
// found among live peers, do nothing
return v
} )
2018-11-07 13:33:36 -06:00
if ins && ! p . BzzPeer . LightNode {
2018-09-12 04:24:56 -05:00
a := newEntry ( p . BzzAddr )
a . conn = p
2018-06-20 07:06:27 -05:00
// insert new online peer into addrs
2018-12-21 23:53:30 -06:00
k . addrs , _ , _ , _ = pot . Swap ( k . addrs , p , Pof , func ( v pot . Val ) pot . Val {
2018-09-12 04:24:56 -05:00
return a
2018-06-20 07:06:27 -05:00
} )
// send new address count value only if the peer is inserted
if k . addrCountC != nil {
k . addrCountC <- k . addrs . Size ( )
}
}
log . Trace ( k . string ( ) )
// calculate if depth of saturation changed
2018-12-21 23:53:30 -06:00
depth := uint8 ( k . saturation ( ) )
2018-06-20 07:06:27 -05:00
var changed bool
if depth != k . depth {
changed = true
k . depth = depth
}
k . sendNeighbourhoodDepthChange ( )
return k . depth , changed
}
// NeighbourhoodDepthC returns the channel that sends a new kademlia
// neighbourhood depth on each change.
// Not receiving from the returned channel will block On function
// when the neighbourhood depth is changed.
2018-11-26 06:52:04 -06:00
// TODO: Why is this exported, and if it should be; why can't we have more subscribers than one?
2018-06-20 07:06:27 -05:00
func ( k * Kademlia ) NeighbourhoodDepthC ( ) <- chan int {
2018-09-12 04:24:56 -05:00
k . lock . Lock ( )
defer k . lock . Unlock ( )
2018-06-20 07:06:27 -05:00
if k . nDepthC == nil {
k . nDepthC = make ( chan int )
}
return k . nDepthC
}
2019-02-20 07:45:25 -06:00
// CloseNeighbourhoodDepthC closes the channel returned by
// NeighbourhoodDepthC and stops sending neighbourhood change.
func ( k * Kademlia ) CloseNeighbourhoodDepthC ( ) {
k . lock . Lock ( )
defer k . lock . Unlock ( )
if k . nDepthC != nil {
close ( k . nDepthC )
k . nDepthC = nil
}
}
2018-06-20 07:06:27 -05:00
// sendNeighbourhoodDepthChange sends new neighbourhood depth to k.nDepth channel
// if it is initialized.
func ( k * Kademlia ) sendNeighbourhoodDepthChange ( ) {
// nDepthC is initialized when NeighbourhoodDepthC is called and returned by it.
// It provides signaling of neighbourhood depth change.
// This part of the code is sending new neighbourhood depth to nDepthC if that condition is met.
if k . nDepthC != nil {
2019-01-10 05:33:51 -06:00
nDepth := depthForPot ( k . conns , k . NeighbourhoodSize , k . base )
2018-06-20 07:06:27 -05:00
if nDepth != k . nDepth {
k . nDepth = nDepth
k . nDepthC <- nDepth
}
}
}
// AddrCountC returns the channel that sends a new
// address count value on each change.
// Not receiving from the returned channel will block Register function
// when address count value changes.
func ( k * Kademlia ) AddrCountC ( ) <- chan int {
2019-02-13 06:03:23 -06:00
k . lock . Lock ( )
defer k . lock . Unlock ( )
2018-06-20 07:06:27 -05:00
if k . addrCountC == nil {
k . addrCountC = make ( chan int )
}
return k . addrCountC
}
2019-02-20 07:45:25 -06:00
// CloseAddrCountC closes the channel returned by
// AddrCountC and stops sending address count change.
func ( k * Kademlia ) CloseAddrCountC ( ) {
k . lock . Lock ( )
defer k . lock . Unlock ( )
if k . addrCountC != nil {
close ( k . addrCountC )
k . addrCountC = nil
}
}
2018-06-20 07:06:27 -05:00
// Off removes a peer from among live peers
2018-09-12 04:24:56 -05:00
func ( k * Kademlia ) Off ( p * Peer ) {
2018-06-20 07:06:27 -05:00
k . lock . Lock ( )
defer k . lock . Unlock ( )
var del bool
2018-11-07 13:33:36 -06:00
if ! p . BzzPeer . LightNode {
2018-12-21 23:53:30 -06:00
k . addrs , _ , _ , _ = pot . Swap ( k . addrs , p , Pof , func ( v pot . Val ) pot . Val {
2018-11-07 13:33:36 -06:00
// v cannot be nil, must check otherwise we overwrite entry
if v == nil {
panic ( fmt . Sprintf ( "connected peer not found %v" , p ) )
}
del = true
return newEntry ( p . BzzAddr )
} )
} else {
2018-06-20 07:06:27 -05:00
del = true
2018-11-07 13:33:36 -06:00
}
2018-06-20 07:06:27 -05:00
if del {
2018-12-21 23:53:30 -06:00
k . conns , _ , _ , _ = pot . Swap ( k . conns , p , Pof , func ( _ pot . Val ) pot . Val {
2018-06-20 07:06:27 -05:00
// v cannot be nil, but no need to check
return nil
} )
// send new address count value only if the peer is deleted
if k . addrCountC != nil {
k . addrCountC <- k . addrs . Size ( )
}
k . sendNeighbourhoodDepthChange ( )
}
}
// EachConn is an iterator with args (base, po, f) applies f to each live peer
// that has proximity order po or less as measured from the base
// if base is nil, kademlia base address is used
2019-01-09 20:36:19 -06:00
func ( k * Kademlia ) EachConn ( base [ ] byte , o int , f func ( * Peer , int ) bool ) {
2018-06-20 07:06:27 -05:00
k . lock . RLock ( )
defer k . lock . RUnlock ( )
k . eachConn ( base , o , f )
}
2019-01-09 20:36:19 -06:00
func ( k * Kademlia ) eachConn ( base [ ] byte , o int , f func ( * Peer , int ) bool ) {
2018-06-20 07:06:27 -05:00
if len ( base ) == 0 {
base = k . base
}
2018-12-21 23:53:30 -06:00
k . conns . EachNeighbour ( base , Pof , func ( val pot . Val , po int ) bool {
2018-06-20 07:06:27 -05:00
if po > o {
return true
}
2019-01-09 20:36:19 -06:00
return f ( val . ( * Peer ) , po )
2018-06-20 07:06:27 -05:00
} )
}
// EachAddr called with (base, po, f) is an iterator applying f to each known peer
2018-12-21 23:53:30 -06:00
// that has proximity order o or less as measured from the base
2018-06-20 07:06:27 -05:00
// if base is nil, kademlia base address is used
2019-01-09 20:36:19 -06:00
func ( k * Kademlia ) EachAddr ( base [ ] byte , o int , f func ( * BzzAddr , int ) bool ) {
2018-06-20 07:06:27 -05:00
k . lock . RLock ( )
defer k . lock . RUnlock ( )
k . eachAddr ( base , o , f )
}
2019-01-09 20:36:19 -06:00
func ( k * Kademlia ) eachAddr ( base [ ] byte , o int , f func ( * BzzAddr , int ) bool ) {
2018-06-20 07:06:27 -05:00
if len ( base ) == 0 {
base = k . base
}
2018-12-21 23:53:30 -06:00
k . addrs . EachNeighbour ( base , Pof , func ( val pot . Val , po int ) bool {
2018-06-20 07:06:27 -05:00
if po > o {
return true
}
2019-01-09 20:36:19 -06:00
return f ( val . ( * entry ) . BzzAddr , po )
2018-06-20 07:06:27 -05:00
} )
}
2019-01-17 00:29:34 -06:00
// NeighbourhoodDepth returns the depth for the pot, see depthForPot
2018-11-26 06:52:04 -06:00
func ( k * Kademlia ) NeighbourhoodDepth ( ) ( depth int ) {
k . lock . RLock ( )
defer k . lock . RUnlock ( )
2019-01-10 05:33:51 -06:00
return depthForPot ( k . conns , k . NeighbourhoodSize , k . base )
2018-11-26 06:52:04 -06:00
}
2019-01-17 00:29:34 -06:00
// neighbourhoodRadiusForPot returns the neighbourhood radius of the kademlia
// neighbourhood radius encloses the nearest neighbour set with size >= neighbourhoodSize
// i.e., neighbourhood radius is the deepest PO such that all bins not shallower altogether
// contain at least neighbourhoodSize connected peers
// if there is altogether less than neighbourhoodSize peers connected, it returns 0
2018-11-26 10:13:59 -06:00
// caller must hold the lock
2019-01-17 00:29:34 -06:00
func neighbourhoodRadiusForPot ( p * pot . Pot , neighbourhoodSize int , pivotAddr [ ] byte ) ( depth int ) {
2019-01-10 05:33:51 -06:00
if p . Size ( ) <= neighbourhoodSize {
2018-06-20 07:06:27 -05:00
return 0
}
2018-11-26 10:13:59 -06:00
// total number of peers in iteration
2018-06-20 07:06:27 -05:00
var size int
f := func ( v pot . Val , i int ) bool {
2018-11-26 10:13:59 -06:00
// po == 256 means that addr is the pivot address(self)
if i == 256 {
return true
}
2018-06-20 07:06:27 -05:00
size ++
2018-11-26 10:13:59 -06:00
// this means we have all nn-peers.
// depth is by default set to the bin of the farthest nn-peer
2019-01-10 05:33:51 -06:00
if size == neighbourhoodSize {
2019-01-17 00:29:34 -06:00
depth = i
2018-11-26 10:13:59 -06:00
return false
}
2018-12-21 23:53:30 -06:00
2018-11-26 10:13:59 -06:00
return true
}
2018-12-21 23:53:30 -06:00
p . EachNeighbour ( pivotAddr , Pof , f )
2019-01-17 00:29:34 -06:00
return depth
}
// depthForPot returns the depth for the pot
// depth is the radius of the minimal extension of nearest neighbourhood that
// includes all empty PO bins. I.e., depth is the deepest PO such that
// - it is not deeper than neighbourhood radius
// - all bins shallower than depth are not empty
// caller must hold the lock
func depthForPot ( p * pot . Pot , neighbourhoodSize int , pivotAddr [ ] byte ) ( depth int ) {
if p . Size ( ) <= neighbourhoodSize {
return 0
}
// determining the depth is a two-step process
// first we find the proximity bin of the shallowest of the neighbourhoodSize peers
// the numeric value of depth cannot be higher than this
maxDepth := neighbourhoodRadiusForPot ( p , neighbourhoodSize , pivotAddr )
2018-12-21 23:53:30 -06:00
// the second step is to test for empty bins in order from shallowest to deepest
// if an empty bin is found, this will be the actual depth
// we stop iterating if we hit the maxDepth determined in the first step
2019-01-15 04:51:33 -06:00
p . EachBin ( pivotAddr , Pof , 0 , func ( po int , _ int , f func ( func ( pot . Val ) bool ) bool ) bool {
2018-12-21 23:53:30 -06:00
if po == depth {
if maxDepth == depth {
return false
}
depth ++
return true
}
return false
} )
2018-11-26 10:13:59 -06:00
2018-06-20 07:06:27 -05:00
return depth
}
2018-09-12 04:24:56 -05:00
// callable decides if an address entry represents a callable peer
func ( k * Kademlia ) callable ( e * entry ) bool {
2018-06-20 07:06:27 -05:00
// not callable if peer is live or exceeded maxRetries
2018-09-12 04:24:56 -05:00
if e . conn != nil || e . retries > k . MaxRetries {
return false
2018-06-20 07:06:27 -05:00
}
// calculate the allowed number of retries based on time lapsed since last seen
timeAgo := int64 ( time . Since ( e . seenAt ) )
div := int64 ( k . RetryExponent )
div += ( 150000 - rand . Int63n ( 300000 ) ) * div / 1000000
var retries int
for delta := timeAgo ; delta > k . RetryInterval ; delta /= div {
retries ++
}
// this is never called concurrently, so safe to increment
// peer can be retried again
if retries < e . retries {
log . Trace ( fmt . Sprintf ( "%08x: %v long time since last try (at %v) needed before retry %v, wait only warrants %v" , k . BaseAddr ( ) [ : 4 ] , e , timeAgo , e . retries , retries ) )
2018-09-12 04:24:56 -05:00
return false
2018-06-20 07:06:27 -05:00
}
// function to sanction or prevent suggesting a peer
2018-09-12 04:24:56 -05:00
if k . Reachable != nil && ! k . Reachable ( e . BzzAddr ) {
2018-06-20 07:06:27 -05:00
log . Trace ( fmt . Sprintf ( "%08x: peer %v is temporarily not callable" , k . BaseAddr ( ) [ : 4 ] , e ) )
2018-09-12 04:24:56 -05:00
return false
2018-06-20 07:06:27 -05:00
}
e . retries ++
log . Trace ( fmt . Sprintf ( "%08x: peer %v is callable" , k . BaseAddr ( ) [ : 4 ] , e ) )
2018-09-12 04:24:56 -05:00
return true
2018-06-20 07:06:27 -05:00
}
// BaseAddr return the kademlia base address
func ( k * Kademlia ) BaseAddr ( ) [ ] byte {
return k . base
}
// String returns kademlia table + kaddb table displayed with ascii
func ( k * Kademlia ) String ( ) string {
k . lock . RLock ( )
defer k . lock . RUnlock ( )
return k . string ( )
}
2018-09-12 04:24:56 -05:00
// string returns kademlia table + kaddb table displayed with ascii
// caller must hold the lock
2018-06-20 07:06:27 -05:00
func ( k * Kademlia ) string ( ) string {
wsrow := " "
var rows [ ] string
rows = append ( rows , "=========================================================================" )
2019-01-24 05:35:10 -06:00
if len ( sv . GitCommit ) > 0 {
rows = append ( rows , fmt . Sprintf ( "commit hash: %s" , sv . GitCommit ) )
}
2018-06-20 07:06:27 -05:00
rows = append ( rows , fmt . Sprintf ( "%v KΛÐΞMLIΛ hive: queen's address: %x" , time . Now ( ) . UTC ( ) . Format ( time . UnixDate ) , k . BaseAddr ( ) [ : 3 ] ) )
2019-01-10 05:33:51 -06:00
rows = append ( rows , fmt . Sprintf ( "population: %d (%d), NeighbourhoodSize: %d, MinBinSize: %d, MaxBinSize: %d" , k . conns . Size ( ) , k . addrs . Size ( ) , k . NeighbourhoodSize , k . MinBinSize , k . MaxBinSize ) )
2018-06-20 07:06:27 -05:00
liverows := make ( [ ] string , k . MaxProxDisplay )
peersrows := make ( [ ] string , k . MaxProxDisplay )
2019-01-10 05:33:51 -06:00
depth := depthForPot ( k . conns , k . NeighbourhoodSize , k . base )
2018-06-20 07:06:27 -05:00
rest := k . conns . Size ( )
2019-01-15 04:51:33 -06:00
k . conns . EachBin ( k . base , Pof , 0 , func ( po , size int , f func ( func ( val pot . Val ) bool ) bool ) bool {
2018-06-20 07:06:27 -05:00
var rowlen int
if po >= k . MaxProxDisplay {
po = k . MaxProxDisplay - 1
}
row := [ ] string { fmt . Sprintf ( "%2d" , size ) }
rest -= size
2019-01-15 04:51:33 -06:00
f ( func ( val pot . Val ) bool {
2018-09-12 04:24:56 -05:00
e := val . ( * Peer )
2018-06-20 07:06:27 -05:00
row = append ( row , fmt . Sprintf ( "%x" , e . Address ( ) [ : 2 ] ) )
rowlen ++
return rowlen < 4
} )
r := strings . Join ( row , " " )
r = r + wsrow
liverows [ po ] = r [ : 31 ]
return true
} )
2019-01-15 04:51:33 -06:00
k . addrs . EachBin ( k . base , Pof , 0 , func ( po , size int , f func ( func ( val pot . Val ) bool ) bool ) bool {
2018-06-20 07:06:27 -05:00
var rowlen int
if po >= k . MaxProxDisplay {
po = k . MaxProxDisplay - 1
}
if size < 0 {
panic ( "wtf" )
}
row := [ ] string { fmt . Sprintf ( "%2d" , size ) }
// we are displaying live peers too
2019-01-15 04:51:33 -06:00
f ( func ( val pot . Val ) bool {
2018-06-20 07:06:27 -05:00
e := val . ( * entry )
row = append ( row , Label ( e ) )
rowlen ++
return rowlen < 4
} )
peersrows [ po ] = strings . Join ( row , " " )
return true
} )
for i := 0 ; i < k . MaxProxDisplay ; i ++ {
if i == depth {
rows = append ( rows , fmt . Sprintf ( "============ DEPTH: %d ==========================================" , i ) )
}
left := liverows [ i ]
right := peersrows [ i ]
if len ( left ) == 0 {
left = " 0 "
}
if len ( right ) == 0 {
right = " 0"
}
rows = append ( rows , fmt . Sprintf ( "%03d %v | %v" , i , left , right ) )
}
rows = append ( rows , "=========================================================================" )
return "\n" + strings . Join ( rows , "\n" )
}
2018-12-21 23:53:30 -06:00
// PeerPot keeps info about expected nearest neighbours
2018-06-20 07:06:27 -05:00
// used for testing only
2018-12-21 23:53:30 -06:00
// TODO move to separate testing tools file
2018-06-20 07:06:27 -05:00
type PeerPot struct {
2019-02-14 12:01:50 -06:00
NNSet [ ] [ ] byte
PeersPerBin [ ] int
2018-06-20 07:06:27 -05:00
}
2018-09-12 04:24:56 -05:00
// NewPeerPotMap creates a map of pot record of *BzzAddr with keys
2018-06-20 07:06:27 -05:00
// as hexadecimal representations of the address.
2019-01-10 05:33:51 -06:00
// the NeighbourhoodSize of the passed kademlia is used
2018-09-12 04:24:56 -05:00
// used for testing only
2018-12-21 23:53:30 -06:00
// TODO move to separate testing tools file
2019-01-10 05:33:51 -06:00
func NewPeerPotMap ( neighbourhoodSize int , addrs [ ] [ ] byte ) map [ string ] * PeerPot {
2018-11-26 10:13:59 -06:00
2018-06-20 07:06:27 -05:00
// create a table of all nodes for health check
np := pot . NewPot ( nil , 0 )
for _ , addr := range addrs {
2018-12-21 23:53:30 -06:00
np , _ , _ = pot . Add ( np , addr , Pof )
2018-06-20 07:06:27 -05:00
}
ppmap := make ( map [ string ] * PeerPot )
2018-12-21 23:53:30 -06:00
// generate an allknowing source of truth for connections
// for every kademlia passed
2018-06-20 07:06:27 -05:00
for i , a := range addrs {
2018-11-26 10:13:59 -06:00
// actual kademlia depth
2019-01-10 05:33:51 -06:00
depth := depthForPot ( np , neighbourhoodSize , a )
2018-11-26 10:13:59 -06:00
// all nn-peers
2018-06-20 07:06:27 -05:00
var nns [ ] [ ] byte
2019-02-14 12:01:50 -06:00
peersPerBin := make ( [ ] int , depth )
2018-11-26 10:13:59 -06:00
2018-12-21 23:53:30 -06:00
// iterate through the neighbours, going from the deepest to the shallowest
np . EachNeighbour ( a , Pof , func ( val pot . Val , po int ) bool {
2018-11-26 10:13:59 -06:00
addr := val . ( [ ] byte )
// po == 256 means that addr is the pivot address(self)
2018-12-21 23:53:30 -06:00
// we do not include self in the map
2018-06-20 07:06:27 -05:00
if po == 256 {
return true
}
2018-12-21 23:53:30 -06:00
// append any neighbors found
// a neighbor is any peer in or deeper than the depth
2018-11-26 10:13:59 -06:00
if po >= depth {
nns = append ( nns , addr )
2019-02-14 12:01:50 -06:00
} else {
// for peers < depth, we just count the number in each bin
// the bin is the index of the slice
peersPerBin [ po ] ++
2018-06-20 07:06:27 -05:00
}
2019-02-14 12:01:50 -06:00
return true
2018-06-20 07:06:27 -05:00
} )
2018-11-26 10:13:59 -06:00
2019-02-14 12:01:50 -06:00
log . Trace ( fmt . Sprintf ( "%x PeerPotMap NNS: %s, peersPerBin" , addrs [ i ] [ : 4 ] , LogAddrs ( nns ) ) )
2018-12-21 23:53:30 -06:00
ppmap [ common . Bytes2Hex ( a ) ] = & PeerPot {
2019-02-14 12:01:50 -06:00
NNSet : nns ,
PeersPerBin : peersPerBin ,
2018-12-21 23:53:30 -06:00
}
2018-06-20 07:06:27 -05:00
}
return ppmap
}
2019-01-17 00:29:34 -06:00
// saturation returns the smallest po value in which the node has less than MinBinSize peers
// if the iterator reaches neighbourhood radius, then the last bin + 1 is returned
2018-12-21 23:53:30 -06:00
func ( k * Kademlia ) saturation ( ) int {
2018-06-20 07:06:27 -05:00
prev := - 1
2019-01-17 00:29:34 -06:00
radius := neighbourhoodRadiusForPot ( k . conns , k . NeighbourhoodSize , k . base )
k . conns . EachBin ( k . base , Pof , 0 , func ( po , size int , f func ( func ( val pot . Val ) bool ) bool ) bool {
2018-06-20 07:06:27 -05:00
prev ++
2019-01-17 00:29:34 -06:00
if po >= radius {
return false
}
2018-12-21 23:53:30 -06:00
return prev == po && size >= k . MinBinSize
2018-06-20 07:06:27 -05:00
} )
2019-01-17 00:29:34 -06:00
if prev < 0 {
return 0
2018-06-20 07:06:27 -05:00
}
return prev
}
2019-02-14 12:01:50 -06:00
// isSaturated returns true if the kademlia is considered saturated, or false if not.
// It checks this by checking an array of ints called unsaturatedBins; each item in that array corresponds
// to the bin which is unsaturated (number of connections < k.MinBinSize).
// The bin is considered unsaturated only if there are actual peers in that PeerPot's bin (peersPerBin)
// (if there is no peer for a given bin, then no connection could ever be established;
// in a God's view this is relevant as no more peers will ever appear on that bin)
func ( k * Kademlia ) isSaturated ( peersPerBin [ ] int , depth int ) bool {
// depth could be calculated from k but as this is called from `GetHealthInfo()`,
// the depth has already been calculated so we can require it as a parameter
// early check for depth
if depth != len ( peersPerBin ) {
return false
}
unsaturatedBins := make ( [ ] int , 0 )
k . conns . EachBin ( k . base , Pof , 0 , func ( po , size int , f func ( func ( val pot . Val ) bool ) bool ) bool {
if po >= depth {
return false
}
log . Trace ( "peers per bin" , "peersPerBin[po]" , peersPerBin [ po ] , "po" , po )
// if there are actually peers in the PeerPot who can fulfill k.MinBinSize
if size < k . MinBinSize && size < peersPerBin [ po ] {
log . Trace ( "connections for po" , "po" , po , "size" , size )
unsaturatedBins = append ( unsaturatedBins , po )
}
return true
} )
log . Trace ( "list of unsaturated bins" , "unsaturatedBins" , unsaturatedBins )
return len ( unsaturatedBins ) == 0
}
2018-12-21 23:53:30 -06:00
// knowNeighbours tests if all neighbours in the peerpot
// are found among the peers known to the kademlia
2018-09-12 04:24:56 -05:00
// It is used in Healthy function for testing only
2018-12-21 23:53:30 -06:00
// TODO move to separate testing tools file
func ( k * Kademlia ) knowNeighbours ( addrs [ ] [ ] byte ) ( got bool , n int , missing [ ] [ ] byte ) {
2018-06-20 07:06:27 -05:00
pm := make ( map [ string ] bool )
2019-01-10 05:33:51 -06:00
depth := depthForPot ( k . conns , k . NeighbourhoodSize , k . base )
2018-12-21 23:53:30 -06:00
// create a map with all peers at depth and deeper known in the kademlia
2019-01-09 20:36:19 -06:00
k . eachAddr ( nil , 255 , func ( p * BzzAddr , po int ) bool {
// in order deepest to shallowest compared to the kademlia base address
// all bins (except self) are included (0 <= bin <= 255)
2018-12-21 23:53:30 -06:00
if po < depth {
2018-06-20 07:06:27 -05:00
return false
}
2018-12-21 23:53:30 -06:00
pk := common . Bytes2Hex ( p . Address ( ) )
2018-06-20 07:06:27 -05:00
pm [ pk ] = true
return true
} )
2018-12-21 23:53:30 -06:00
// iterate through nearest neighbors in the peerpot map
// if we can't find the neighbor in the map we created above
// then we don't know all our neighbors
// (which sadly is all too common in modern society)
var gots int
var culprits [ ] [ ] byte
for _ , p := range addrs {
pk := common . Bytes2Hex ( p )
if pm [ pk ] {
gots ++
} else {
log . Trace ( fmt . Sprintf ( "%08x: known nearest neighbour %s not found" , k . base , pk ) )
culprits = append ( culprits , p )
2018-06-20 07:06:27 -05:00
}
}
2018-12-21 23:53:30 -06:00
return gots == len ( addrs ) , gots , culprits
2018-06-20 07:06:27 -05:00
}
2018-12-21 23:53:30 -06:00
// connectedNeighbours tests if all neighbours in the peerpot
// are currently connected in the kademlia
2018-09-12 04:24:56 -05:00
// It is used in Healthy function for testing only
2018-12-21 23:53:30 -06:00
func ( k * Kademlia ) connectedNeighbours ( peers [ ] [ ] byte ) ( got bool , n int , missing [ ] [ ] byte ) {
2018-06-20 07:06:27 -05:00
pm := make ( map [ string ] bool )
2019-01-10 05:33:51 -06:00
// create a map with all peers at depth and deeper that are connected in the kademlia
// in order deepest to shallowest compared to the kademlia base address
// all bins (except self) are included (0 <= bin <= 255)
depth := depthForPot ( k . conns , k . NeighbourhoodSize , k . base )
2019-01-09 20:36:19 -06:00
k . eachConn ( nil , 255 , func ( p * Peer , po int ) bool {
2018-12-21 23:53:30 -06:00
if po < depth {
2018-06-20 07:06:27 -05:00
return false
}
2018-12-21 23:53:30 -06:00
pk := common . Bytes2Hex ( p . Address ( ) )
2018-06-20 07:06:27 -05:00
pm [ pk ] = true
return true
} )
2018-12-21 23:53:30 -06:00
// iterate through nearest neighbors in the peerpot map
// if we can't find the neighbor in the map we created above
// then we don't know all our neighbors
2018-06-20 07:06:27 -05:00
var gots int
var culprits [ ] [ ] byte
for _ , p := range peers {
2018-12-21 23:53:30 -06:00
pk := common . Bytes2Hex ( p )
2018-06-20 07:06:27 -05:00
if pm [ pk ] {
gots ++
} else {
2018-12-21 23:53:30 -06:00
log . Trace ( fmt . Sprintf ( "%08x: ExpNN: %s not found" , k . base , pk ) )
2018-06-20 07:06:27 -05:00
culprits = append ( culprits , p )
}
}
return gots == len ( peers ) , gots , culprits
}
// Health state of the Kademlia
2018-09-12 04:24:56 -05:00
// used for testing only
2018-06-20 07:06:27 -05:00
type Health struct {
2018-12-21 23:53:30 -06:00
KnowNN bool // whether node knows all its neighbours
CountKnowNN int // amount of neighbors known
MissingKnowNN [ ] [ ] byte // which neighbours we should have known but we don't
ConnectNN bool // whether node is connected to all its neighbours
CountConnectNN int // amount of neighbours connected to
MissingConnectNN [ ] [ ] byte // which neighbours we should have been connected to but we're not
2019-02-14 12:01:50 -06:00
// Saturated: if in all bins < depth number of connections >= MinBinsize or,
// if number of connections < MinBinSize, to the number of available peers in that bin
Saturated bool
Hive string
2018-06-20 07:06:27 -05:00
}
2019-02-14 12:01:50 -06:00
// GetHealthInfo reports the health state of the kademlia connectivity
2018-12-21 23:53:30 -06:00
//
// The PeerPot argument provides an all-knowing view of the network
// The resulting Health object is a result of comparisons between
// what is the actual composition of the kademlia in question (the receiver), and
// what SHOULD it have been when we take all we know about the network into consideration.
//
2018-09-12 04:24:56 -05:00
// used for testing only
2019-02-14 12:01:50 -06:00
func ( k * Kademlia ) GetHealthInfo ( pp * PeerPot ) * Health {
2018-06-20 07:06:27 -05:00
k . lock . RLock ( )
defer k . lock . RUnlock ( )
2019-01-17 00:29:34 -06:00
if len ( pp . NNSet ) < k . NeighbourhoodSize {
log . Warn ( "peerpot NNSet < NeighbourhoodSize" )
}
2018-12-21 23:53:30 -06:00
gotnn , countgotnn , culpritsgotnn := k . connectedNeighbours ( pp . NNSet )
knownn , countknownn , culpritsknownn := k . knowNeighbours ( pp . NNSet )
2019-01-10 05:33:51 -06:00
depth := depthForPot ( k . conns , k . NeighbourhoodSize , k . base )
2019-02-14 12:01:50 -06:00
// check saturation
saturated := k . isSaturated ( pp . PeersPerBin , depth )
2018-12-21 23:53:30 -06:00
log . Trace ( fmt . Sprintf ( "%08x: healthy: knowNNs: %v, gotNNs: %v, saturated: %v\n" , k . base , knownn , gotnn , saturated ) )
return & Health {
KnowNN : knownn ,
CountKnowNN : countknownn ,
MissingKnowNN : culpritsknownn ,
ConnectNN : gotnn ,
CountConnectNN : countgotnn ,
MissingConnectNN : culpritsgotnn ,
Saturated : saturated ,
Hive : k . string ( ) ,
2018-06-20 07:06:27 -05:00
}
}
2019-02-14 12:01:50 -06:00
// Healthy return the strict interpretation of `Healthy` given a `Health` struct
// definition of strict health: all conditions must be true:
// - we at least know one peer
// - we know all neighbors
// - we are connected to all known neighbors
// - it is saturated
func ( h * Health ) Healthy ( ) bool {
return h . KnowNN && h . ConnectNN && h . CountKnowNN > 0 && h . Saturated
}