2014-12-10 06:12:25 +02:00
package eth
import (
"math"
"math/big"
2014-12-14 20:08:18 +02:00
"math/rand"
"sort"
2014-12-10 06:12:25 +02:00
"sync"
"time"
"github.com/ethereum/go-ethereum/core/types"
"github.com/ethereum/go-ethereum/ethutil"
ethlogger "github.com/ethereum/go-ethereum/logger"
2014-12-10 06:12:25 +02:00
"github.com/ethereum/go-ethereum/pow"
2014-12-10 06:12:25 +02:00
)
var poolLogger = ethlogger . NewLogger ( "Blockpool" )
const (
2014-12-14 20:08:18 +02:00
blockHashesBatchSize = 256
blockBatchSize = 64
blocksRequestInterval = 10 // seconds
blocksRequestRepetition = 1
blockHashesRequestInterval = 10 // seconds
blocksRequestMaxIdleRounds = 10
cacheTimeout = 3 // minutes
blockTimeout = 5 // minutes
2014-12-10 06:12:25 +02:00
)
type poolNode struct {
2014-12-14 20:08:18 +02:00
lock sync . RWMutex
hash [ ] byte
block * types . Block
child * poolNode
parent * poolNode
section * section
knownParent bool
peer string
source string
complete bool
2014-12-10 06:12:25 +02:00
}
type BlockPool struct {
2014-12-14 20:08:18 +02:00
lock sync . RWMutex
2014-12-10 06:12:25 +02:00
pool map [ string ] * poolNode
2014-12-14 20:08:18 +02:00
peersLock sync . RWMutex
2014-12-10 06:12:25 +02:00
peers map [ string ] * peerInfo
peer * peerInfo
quit chan bool
wg sync . WaitGroup
running bool
2014-12-14 20:08:18 +02:00
// the minimal interface with blockchain
hasBlock func ( hash [ ] byte ) bool
insertChain func ( types . Blocks ) error
2014-12-14 20:08:18 +02:00
<< << << < HEAD
2014-12-10 06:12:25 +02:00
verifyPoW func ( pow . Block ) bool
2014-12-14 20:08:18 +02:00
== == == =
verifyPoW func ( * types . Block ) bool
>> >> >> > blockpool rewritten , tests broken FIXME
2014-12-10 06:12:25 +02:00
}
type peerInfo struct {
2014-12-14 20:08:18 +02:00
lock sync . RWMutex
td * big . Int
currentBlock [ ] byte
id string
2014-12-10 06:12:25 +02:00
requestBlockHashes func ( [ ] byte ) error
requestBlocks func ( [ ] [ ] byte ) error
2014-12-14 20:08:18 +02:00
peerError func ( int , string , ... interface { } )
2014-12-10 06:12:25 +02:00
2014-12-14 20:08:18 +02:00
sections map [ string ] * section
roots [ ] * poolNode
quitC chan bool
2014-12-10 06:12:25 +02:00
}
2014-12-14 20:08:18 +02:00
<< << << < HEAD
2014-12-10 06:12:25 +02:00
func NewBlockPool ( hasBlock func ( hash [ ] byte ) bool , insertChain func ( types . Blocks ) error , verifyPoW func ( pow . Block ) bool ,
2014-12-14 20:08:18 +02:00
== == == =
func NewBlockPool ( hasBlock func ( hash [ ] byte ) bool , insertChain func ( types . Blocks ) error , verifyPoW func ( * types . Block ) bool ,
>> >> >> > blockpool rewritten , tests broken FIXME
2014-12-14 20:08:18 +02:00
) * BlockPool {
2014-12-10 06:12:25 +02:00
return & BlockPool {
2014-12-14 20:08:18 +02:00
hasBlock : hasBlock ,
insertChain : insertChain ,
verifyPoW : verifyPoW ,
2014-12-10 06:12:25 +02:00
}
}
2014-12-14 20:08:18 +02:00
// allows restart
func ( self * BlockPool ) Start ( ) {
2014-12-10 06:12:25 +02:00
self . lock . Lock ( )
2014-12-14 20:08:18 +02:00
if self . running {
2014-12-10 06:12:25 +02:00
self . lock . Unlock ( )
return
}
2014-12-14 20:08:18 +02:00
self . running = true
self . quit = make ( chan bool )
self . pool = make ( map [ string ] * poolNode )
2014-12-10 06:12:25 +02:00
self . lock . Unlock ( )
2014-12-14 20:08:18 +02:00
self . peersLock . Lock ( )
self . peers = make ( map [ string ] * peerInfo )
self . peersLock . Unlock ( )
2014-12-10 06:12:25 +02:00
2014-12-14 20:08:18 +02:00
poolLogger . Infoln ( "Started" )
2014-12-10 06:12:25 +02:00
}
2014-12-14 20:08:18 +02:00
func ( self * BlockPool ) Stop ( ) {
self . lock . Lock ( )
if ! self . running {
self . lock . Unlock ( )
2014-12-10 06:12:25 +02:00
return
}
2014-12-14 20:08:18 +02:00
self . running = false
2014-12-10 06:12:25 +02:00
self . lock . Unlock ( )
2014-12-14 20:08:18 +02:00
poolLogger . Infoln ( "Stopping" )
2014-12-10 06:12:25 +02:00
2014-12-14 20:08:18 +02:00
close ( self . quit )
self . lock . Lock ( )
2014-12-10 06:12:25 +02:00
self . peersLock . Lock ( )
2014-12-14 20:08:18 +02:00
self . peers = nil
self . pool = nil
self . peer = nil
self . wg . Wait ( )
self . lock . Unlock ( )
2014-12-10 06:12:25 +02:00
self . peersLock . Unlock ( )
2014-12-14 20:08:18 +02:00
poolLogger . Infoln ( "Stopped" )
2014-12-10 06:12:25 +02:00
}
// AddPeer is called by the eth protocol instance running on the peer after
// the status message has been received with total difficulty and current block hash
// AddPeer can only be used once, RemovePeer needs to be called when the peer disconnects
2014-12-14 20:08:18 +02:00
func ( self * BlockPool ) AddPeer ( td * big . Int , currentBlock [ ] byte , peerId string , requestBlockHashes func ( [ ] byte ) error , requestBlocks func ( [ ] [ ] byte ) error , peerError func ( int , string , ... interface { } ) ) bool {
2014-12-10 06:12:25 +02:00
self . peersLock . Lock ( )
defer self . peersLock . Unlock ( )
if self . peers [ peerId ] != nil {
panic ( "peer already added" )
}
2014-12-14 20:08:18 +02:00
peer := & peerInfo {
2014-12-10 06:12:25 +02:00
td : td ,
currentBlock : currentBlock ,
id : peerId , //peer.Identity().Pubkey()
requestBlockHashes : requestBlockHashes ,
requestBlocks : requestBlocks ,
2014-12-14 20:08:18 +02:00
peerError : peerError ,
2014-12-10 06:12:25 +02:00
}
2014-12-14 20:08:18 +02:00
self . peers [ peerId ] = peer
2014-12-10 06:12:25 +02:00
poolLogger . Debugf ( "add new peer %v with td %v" , peerId , td )
currentTD := ethutil . Big0
if self . peer != nil {
currentTD = self . peer . td
}
if td . Cmp ( currentTD ) > 0 {
2014-12-14 20:08:18 +02:00
self . peer . stop ( peer )
peer . start ( self . peer )
2014-12-10 06:12:25 +02:00
poolLogger . Debugf ( "peer %v promoted to best peer" , peerId )
2014-12-14 20:08:18 +02:00
self . peer = peer
2014-12-10 06:12:25 +02:00
return true
}
return false
}
// RemovePeer is called by the eth protocol when the peer disconnects
func ( self * BlockPool ) RemovePeer ( peerId string ) {
self . peersLock . Lock ( )
defer self . peersLock . Unlock ( )
2014-12-14 20:08:18 +02:00
peer := self . peers [ peerId ]
if peer == nil {
return
2014-12-10 06:12:25 +02:00
}
self . peers [ peerId ] = nil
poolLogger . Debugf ( "remove peer %v" , peerId [ 0 : 4 ] )
// if current best peer is removed, need find a better one
2014-12-14 20:08:18 +02:00
if self . peer != nil && peerId == self . peer . id {
2014-12-10 06:12:25 +02:00
var newPeer * peerInfo
max := ethutil . Big0
// peer with the highest self-acclaimed TD is chosen
for _ , info := range self . peers {
if info . td . Cmp ( max ) > 0 {
max = info . td
newPeer = info
}
}
2014-12-14 20:08:18 +02:00
self . peer . stop ( peer )
peer . start ( self . peer )
2014-12-10 06:12:25 +02:00
if newPeer != nil {
2014-12-14 20:08:18 +02:00
poolLogger . Debugf ( "peer %v with td %v promoted to best peer" , newPeer . id [ 0 : 4 ] , newPeer . td )
2014-12-10 06:12:25 +02:00
} else {
poolLogger . Warnln ( "no peers left" )
}
}
}
2014-12-14 20:08:18 +02:00
// Entry point for eth protocol to add block hashes received via BlockHashesMsg
// only hashes from the best peer is handled
// this method is always responsible to initiate further hash requests until
// a known parent is reached unless cancelled by a peerChange event
// this process also launches all request processes on each chain section
// this function needs to run asynchronously for one peer since the message is discarded???
func ( self * BlockPool ) AddBlockHashes ( next func ( ) ( [ ] byte , bool ) , peerId string ) {
// check if this peer is the best
peer , best := self . getPeer ( peerId )
if ! best {
return
}
// peer is still the best
var child * poolNode
var depth int
// iterate using next (rlp stream lazy decoder) feeding hashesC
self . wg . Add ( 1 )
go func ( ) {
for {
select {
case <- self . quit :
return
case <- peer . quitC :
// if the peer is demoted, no more hashes taken
break
default :
hash , ok := next ( )
if ! ok {
// message consumed chain skeleton built
break
}
// check if known block connecting the downloaded chain to our blockchain
if self . hasBlock ( hash ) {
poolLogger . Infof ( "known block (%x...)\n" , hash [ 0 : 4 ] )
if child != nil {
child . Lock ( )
// mark child as absolute pool root with parent known to blockchain
child . knownParent = true
child . Unlock ( )
}
break
}
//
var parent * poolNode
// look up node in pool
parent = self . get ( hash )
if parent != nil {
// reached a known chain in the pool
// request blocks on the newly added part of the chain
if child != nil {
self . link ( parent , child )
// activate the current chain
self . activateChain ( parent , peer , true )
poolLogger . Debugf ( "potential chain of %v blocks added, reached blockpool, activate chain" , depth )
break
}
// if this is the first hash, we expect to find it
parent . RLock ( )
grandParent := parent . parent
parent . RUnlock ( )
if grandParent != nil {
// activate the current chain
self . activateChain ( parent , peer , true )
poolLogger . Debugf ( "block hash found, activate chain" )
break
}
// the first node is the root of a chain in the pool, rejoice and continue
}
// if node does not exist, create it and index in the pool
section := & section { }
if child == nil {
section . top = parent
}
parent = & poolNode {
hash : hash ,
child : child ,
section : section ,
peer : peerId ,
}
self . set ( hash , parent )
poolLogger . Debugf ( "create potential block for %x..." , hash [ 0 : 4 ] )
depth ++
child = parent
}
}
if child != nil {
poolLogger . Debugf ( "chain of %v hashes added" , depth )
// start a processSection on the last node, but switch off asking
// hashes and blocks until next peer confirms this chain
section := self . processSection ( child )
peer . addSection ( child . hash , section )
section . start ( )
}
} ( )
}
// AddBlock is the entry point for the eth protocol when blockmsg is received upon requests
// It has a strict interpretation of the protocol in that if the block received has not been requested, it results in an error (which can be ignored)
// block is checked for PoW
// only the first PoW-valid block for a hash is considered legit
func ( self * BlockPool ) AddBlock ( block * types . Block , peerId string ) {
hash := block . Hash ( )
node := self . get ( hash )
node . RLock ( )
b := node . block
node . RUnlock ( )
if b != nil {
return
}
if node == nil && ! self . hasBlock ( hash ) {
self . peerError ( peerId , ErrUnrequestedBlock , "%x" , hash )
return
}
// validate block for PoW
if ! self . verifyPoW ( block ) {
self . peerError ( peerId , ErrInvalidPoW , "%x" , hash )
}
node . Lock ( )
node . block = block
node . source = peerId
node . Unlock ( )
}
// iterates down a known poolchain and activates fetching processes
// on each chain section for the peer
// stops if the peer is demoted
// registers last section root as root for the peer (in case peer is promoted a second time, to remember)
func ( self * BlockPool ) activateChain ( node * poolNode , peer * peerInfo , on bool ) {
self . wg . Add ( 1 )
go func ( ) {
for {
node . sectionRLock ( )
bottom := node . section . bottom
if bottom == nil { // the chain section is being created or killed
break
}
// register this section with the peer
if peer != nil {
peer . addSection ( bottom . hash , bottom . section )
if on {
bottom . section . start ( )
} else {
bottom . section . start ( )
}
}
if bottom . parent == nil {
node = bottom
break
}
// if peer demoted stop activation
select {
case <- peer . quitC :
break
default :
}
node = bottom . parent
bottom . sectionRUnlock ( )
}
// remember root for this peer
peer . addRoot ( node )
self . wg . Done ( )
} ( )
}
// main worker thread on each section in the poolchain
// - kills the section if there are blocks missing after an absolute time
// - kills the section if there are maxIdleRounds of idle rounds of block requests with no response
// - periodically polls the chain section for missing blocks which are then requested from peers
// - registers the process controller on the peer so that if the peer is promoted as best peer the second time (after a disconnect of a better one), all active processes are switched back on unless they expire and killed ()
// - when turned off (if peer disconnects and new peer connects with alternative chain), no blockrequests are made but absolute expiry timer is ticking
// - when turned back on it recursively calls itself on the root of the next chain section
// - when exits, signals to
func ( self * BlockPool ) processSection ( node * poolNode ) * section {
// absolute time after which sub-chain is killed if not complete (some blocks are missing)
suicideTimer := time . After ( blockTimeout * time . Minute )
var blocksRequestTimer , blockHashesRequestTimer <- chan time . Time
var nodeC , missingC , processC chan * poolNode
controlC := make ( chan bool )
resetC := make ( chan bool )
var hashes [ ] [ ] byte
var i , total , missing , lastMissing , depth int
var blockHashesRequests , blocksRequests int
var idle int
var init , alarm , done , same , running , once bool
orignode := node
hash := node . hash
node . sectionLock ( )
defer node . sectionUnlock ( )
section := & section { controlC : controlC , resetC : resetC }
node . section = section
go func ( ) {
self . wg . Add ( 1 )
for {
node . sectionRLock ( )
controlC = node . section . controlC
node . sectionRUnlock ( )
if init {
// missing blocks read from nodeC
// initialized section
if depth == 0 {
break
}
// enable select case to read missing block when ready
processC = missingC
missingC = make ( chan * poolNode , lastMissing )
nodeC = nil
// only do once
init = false
} else {
if ! once {
missingC = nil
processC = nil
i = 0
total = 0
lastMissing = 0
}
}
// went through all blocks in section
if i != 0 && i == lastMissing {
if len ( hashes ) > 0 {
// send block requests to peers
self . requestBlocks ( blocksRequests , hashes )
}
blocksRequests ++
poolLogger . Debugf ( "[%x] block request attempt %v: missing %v/%v/%v" , hash [ 0 : 4 ] , blocksRequests , missing , total , depth )
if missing == lastMissing {
// idle round
if same {
// more than once
idle ++
// too many idle rounds
if idle > blocksRequestMaxIdleRounds {
poolLogger . Debugf ( "[%x] block requests had %v idle rounds (%v total attempts): missing %v/%v/%v\ngiving up..." , hash [ 0 : 4 ] , idle , blocksRequests , missing , total , depth )
self . killChain ( node , nil )
break
}
} else {
idle = 0
}
same = true
} else {
if missing == 0 {
// no missing nodes
poolLogger . Debugf ( "block request process complete on section %x... (%v total blocksRequests): missing %v/%v/%v" , hash [ 0 : 4 ] , blockHashesRequests , blocksRequests , missing , total , depth )
node . Lock ( )
orignode . complete = true
node . Unlock ( )
blocksRequestTimer = nil
if blockHashesRequestTimer == nil {
// not waiting for hashes any more
poolLogger . Debugf ( "hash request on root %x... successful (%v total attempts)\nquitting..." , hash [ 0 : 4 ] , blockHashesRequests )
break
} // otherwise suicide if no hashes coming
}
same = false
}
lastMissing = missing
i = 0
missing = 0
// ready for next round
done = true
}
if done && alarm {
poolLogger . Debugf ( "start checking if new blocks arrived (attempt %v): missing %v/%v/%v" , blocksRequests , missing , total , depth )
blocksRequestTimer = time . After ( blocksRequestInterval * time . Second )
alarm = false
done = false
// processC supposed to be empty and never closed so just swap, no need to allocate
tempC := processC
processC = missingC
missingC = tempC
}
select {
case <- self . quit :
break
case <- suicideTimer :
self . killChain ( node , nil )
poolLogger . Warnf ( "[%x] timeout. (%v total attempts): missing %v/%v/%v" , hash [ 0 : 4 ] , blocksRequests , missing , total , depth )
break
case <- blocksRequestTimer :
alarm = true
case <- blockHashesRequestTimer :
orignode . RLock ( )
parent := orignode . parent
orignode . RUnlock ( )
if parent != nil {
// if not root of chain, switch off
poolLogger . Debugf ( "[%x] parent found, hash requests deactivated (after %v total attempts)\n" , hash [ 0 : 4 ] , blockHashesRequests )
blockHashesRequestTimer = nil
} else {
blockHashesRequests ++
poolLogger . Debugf ( "[%x] hash request on root (%v total attempts)\n" , hash [ 0 : 4 ] , blockHashesRequests )
self . requestBlockHashes ( parent . hash )
blockHashesRequestTimer = time . After ( blockHashesRequestInterval * time . Second )
}
case r , ok := <- controlC :
if ! ok {
break
}
if running && ! r {
poolLogger . Debugf ( "process on section %x... (%v total attempts): missing %v/%v/%v" , hash [ 0 : 4 ] , blocksRequests , missing , total , depth )
alarm = false
blocksRequestTimer = nil
blockHashesRequestTimer = nil
processC = nil
}
if ! running && r {
poolLogger . Debugf ( "[%x] on" , hash [ 0 : 4 ] )
orignode . RLock ( )
parent := orignode . parent
complete := orignode . complete
knownParent := orignode . knownParent
orignode . RUnlock ( )
if ! complete {
poolLogger . Debugf ( "[%x] activate block requests" , hash [ 0 : 4 ] )
blocksRequestTimer = time . After ( 0 )
}
if parent == nil && ! knownParent {
// if no parent but not connected to blockchain
poolLogger . Debugf ( "[%x] activate block hashes requests" , hash [ 0 : 4 ] )
blockHashesRequestTimer = time . After ( 0 )
} else {
blockHashesRequestTimer = nil
}
alarm = true
processC = missingC
if ! once {
// if not run at least once fully, launch iterator
processC = make ( chan * poolNode )
missingC = make ( chan * poolNode )
self . foldUp ( orignode , processC )
once = true
}
}
total = lastMissing
case <- resetC :
once = false
init = false
done = false
case node , ok := <- processC :
if ! ok {
// channel closed, first iteration finished
init = true
once = true
continue
}
i ++
// if node has no block
node . RLock ( )
block := node . block
nhash := node . hash
knownParent := node . knownParent
node . RUnlock ( )
if ! init {
depth ++
}
if block == nil {
missing ++
if ! init {
total ++
}
hashes = append ( hashes , nhash )
if len ( hashes ) == blockBatchSize {
self . requestBlocks ( blocksRequests , hashes )
hashes = nil
}
missingC <- node
} else {
// block is found
if knownParent {
// connected to the blockchain, insert the longest chain of blocks
var blocks types . Blocks
child := node
parent := node
node . sectionRLock ( )
for child != nil && child . block != nil {
parent = child
blocks = append ( blocks , parent . block )
child = parent . child
}
node . sectionRUnlock ( )
poolLogger . Debugf ( "[%x] insert %v blocks into blockchain" , hash [ 0 : 4 ] , len ( blocks ) )
if err := self . insertChain ( blocks ) ; err != nil {
// TODO: not clear which peer we need to address
// peerError should dispatch to peer if still connected and disconnect
self . peerError ( node . source , ErrInvalidBlock , "%v" , err )
poolLogger . Debugf ( "invalid block %v" , node . hash )
poolLogger . Debugf ( "penalise peers %v (hash), %v (block)" , node . peer , node . source )
// penalise peer in node.source
self . killChain ( node , nil )
// self.disconnect()
break
}
// if suceeded mark the next one (no block yet) as connected to blockchain
if child != nil {
child . Lock ( )
child . knownParent = true
child . Unlock ( )
}
// reset starting node to first node with missing block
orignode = child
// pop the inserted ancestors off the channel
for i := 1 ; i < len ( blocks ) ; i ++ {
<- processC
}
// delink inserted chain section
self . killChain ( node , parent )
}
}
}
}
poolLogger . Debugf ( "[%x] quit after\n%v block hashes requests\n%v block requests: missing %v/%v/%v" , hash [ 0 : 4 ] , blockHashesRequests , blocksRequests , missing , total , depth )
self . wg . Done ( )
node . sectionLock ( )
node . section . controlC = nil
node . sectionUnlock ( )
// this signals that controller not available
} ( )
return section
}
func ( self * BlockPool ) peerError ( peerId string , code int , format string , params ... interface { } ) {
self . peersLock . RLock ( )
defer self . peersLock . RUnlock ( )
peer , ok := self . peers [ peerId ]
if ok {
peer . peerError ( code , format , params ... )
}
}
func ( self * BlockPool ) requestBlockHashes ( hash [ ] byte ) {
self . peersLock . Lock ( )
defer self . peersLock . Unlock ( )
if self . peer != nil {
self . peer . requestBlockHashes ( hash )
}
}
func ( self * BlockPool ) requestBlocks ( attempts int , hashes [ ] [ ] byte ) {
// distribute block request among known peers
2014-12-10 06:12:25 +02:00
self . peersLock . Lock ( )
defer self . peersLock . Unlock ( )
2014-12-14 20:08:18 +02:00
peerCount := len ( self . peers )
// on first attempt use the best peer
if attempts == 0 {
self . peer . requestBlocks ( hashes )
return
}
repetitions := int ( math . Min ( float64 ( peerCount ) , float64 ( blocksRequestRepetition ) ) )
poolLogger . Debugf ( "request %v missing blocks from %v/%v peers" , len ( hashes ) , repetitions , peerCount )
i := 0
indexes := rand . Perm ( peerCount ) [ 0 : ( repetitions - 1 ) ]
sort . Ints ( indexes )
for _ , peer := range self . peers {
if i == indexes [ 0 ] {
peer . requestBlocks ( hashes )
indexes = indexes [ 1 : ]
if len ( indexes ) == 0 {
break
}
}
i ++
}
}
func ( self * BlockPool ) getPeer ( peerId string ) ( * peerInfo , bool ) {
self . peersLock . RLock ( )
defer self . peersLock . RUnlock ( )
if self . peer != nil && self . peer . id == peerId {
2014-12-10 06:12:25 +02:00
return self . peer , true
}
info , ok := self . peers [ peerId ]
if ! ok {
panic ( "unknown peer" )
}
return info , false
}
2014-12-14 20:08:18 +02:00
func ( self * peerInfo ) addSection ( hash [ ] byte , section * section ) {
self . lock . Lock ( )
defer self . lock . Unlock ( )
self . sections [ string ( hash ) ] = section
}
func ( self * peerInfo ) addRoot ( node * poolNode ) {
self . lock . Lock ( )
defer self . lock . Unlock ( )
self . roots = append ( self . roots , node )
}
2014-12-10 06:12:25 +02:00
2014-12-14 20:08:18 +02:00
// (re)starts processes registered for this peer (self)
func ( self * peerInfo ) start ( peer * peerInfo ) {
self . lock . Lock ( )
defer self . lock . Unlock ( )
self . quitC = make ( chan bool )
for _ , root := range self . roots {
root . sectionRLock ( )
if root . section . bottom != nil {
if root . parent == nil {
self . requestBlockHashes ( root . hash )
}
}
root . sectionRUnlock ( )
}
self . roots = nil
self . controlSections ( peer , true )
}
// (re)starts process without requests, only suicide timer
func ( self * peerInfo ) stop ( peer * peerInfo ) {
self . lock . RLock ( )
defer self . lock . RUnlock ( )
close ( self . quitC )
self . controlSections ( peer , false )
}
func ( self * peerInfo ) controlSections ( peer * peerInfo , on bool ) {
if peer != nil {
peer . lock . RLock ( )
defer peer . lock . RUnlock ( )
}
for hash , section := range peer . sections {
if section . done ( ) {
delete ( self . sections , hash )
}
_ , exists := peer . sections [ hash ]
if on || peer == nil || exists {
if on {
// self is best peer
section . start ( )
} else {
// (re)starts process without requests, only suicide timer
section . stop ( )
}
}
}
}
// called when parent is found in pool
// parent and child are guaranteed to be on different sections
2014-12-10 06:12:25 +02:00
func ( self * BlockPool ) link ( parent , child * poolNode ) {
2014-12-14 20:08:18 +02:00
var top bool
parent . sectionLock ( )
if child != nil {
child . sectionLock ( )
}
if parent == parent . section . top && parent . section . top != nil {
top = true
}
var bottom bool
if child == child . section . bottom {
bottom = true
2014-12-10 06:12:25 +02:00
}
if parent . child != child {
orphan := parent . child
2014-12-14 20:08:18 +02:00
if orphan != nil {
// got a fork in the chain
if top {
orphan . lock . Lock ( )
// make old child orphan
orphan . parent = nil
orphan . lock . Unlock ( )
} else { // we are under section lock
// make old child orphan
orphan . parent = nil
// reset section objects above the fork
nchild := orphan . child
node := orphan
section := & section { bottom : orphan }
for node . section == nchild . section {
node = nchild
node . section = section
nchild = node . child
}
section . top = node
// set up a suicide
self . processSection ( orphan ) . stop ( )
}
} else {
// child is on top of a chain need to close section
child . section . bottom = child
}
// adopt new child
2014-12-10 06:12:25 +02:00
parent . child = child
2014-12-14 20:08:18 +02:00
if ! top {
parent . section . top = parent
// restart section process so that shorter section is scanned for blocks
parent . section . reset ( )
}
2014-12-10 06:12:25 +02:00
}
2014-12-14 20:08:18 +02:00
2014-12-10 06:12:25 +02:00
if child != nil {
if child . parent != parent {
2014-12-14 20:08:18 +02:00
stepParent := child . parent
if stepParent != nil {
if bottom {
stepParent . Lock ( )
stepParent . child = nil
stepParent . Unlock ( )
} else {
// we are on the same section
// if it is a aberrant reverse fork,
stepParent . child = nil
node := stepParent
nparent := stepParent . child
section := & section { top : stepParent }
for node . section == nparent . section {
node = nparent
node . section = section
node = node . parent
}
2014-12-10 06:12:25 +02:00
}
2014-12-14 20:08:18 +02:00
} else {
// linking to a root node, ie. parent is under the root of a chain
parent . section . top = parent
}
2014-12-10 06:12:25 +02:00
}
2014-12-14 20:08:18 +02:00
child . parent = parent
child . section . bottom = child
2014-12-10 06:12:25 +02:00
}
2014-12-14 20:08:18 +02:00
// this needed if someone lied about the parent before
child . knownParent = false
2014-12-10 06:12:25 +02:00
2014-12-14 20:08:18 +02:00
parent . sectionUnlock ( )
if child != nil {
child . sectionUnlock ( )
2014-12-10 06:12:25 +02:00
}
2014-12-14 20:08:18 +02:00
}
// this immediately kills the chain from node to end (inclusive) section by section
func ( self * BlockPool ) killChain ( node * poolNode , end * poolNode ) {
poolLogger . Debugf ( "kill chain section with root node %v" , node )
node . sectionLock ( )
node . section . abort ( )
self . set ( node . hash , nil )
child := node . child
top := node . section . top
i := 1
2014-12-10 06:12:25 +02:00
self . wg . Add ( 1 )
2014-12-14 20:08:18 +02:00
go func ( ) {
var quit bool
for node != top && node != end && child != nil {
node = child
select {
case <- self . quit :
quit = true
break
default :
}
self . set ( node . hash , nil )
child = node . child
}
poolLogger . Debugf ( "killed chain section of %v blocks with root node %v" , i , node )
if ! quit {
if node == top {
if node != end && child != nil && end != nil {
//
self . killChain ( child , end )
2014-12-10 06:12:25 +02:00
}
} else {
2014-12-14 20:08:18 +02:00
if child != nil {
// delink rest of this section if ended midsection
child . section . bottom = child
child . parent = nil
}
2014-12-10 06:12:25 +02:00
}
}
2014-12-14 20:08:18 +02:00
node . section . bottom = nil
node . sectionUnlock ( )
self . wg . Done ( )
} ( )
}
// structure to store long range links on chain to skip along
type section struct {
lock sync . RWMutex
bottom * poolNode
top * poolNode
controlC chan bool
resetC chan bool
}
func ( self * section ) start ( ) {
self . lock . RLock ( )
defer self . lock . RUnlock ( )
if self . controlC != nil {
self . controlC <- true
2014-12-10 06:12:25 +02:00
}
}
2014-12-14 20:08:18 +02:00
func ( self * section ) stop ( ) {
self . lock . RLock ( )
defer self . lock . RUnlock ( )
if self . controlC != nil {
self . controlC <- false
}
}
func ( self * section ) reset ( ) {
self . lock . RLock ( )
defer self . lock . RUnlock ( )
if self . controlC != nil {
self . resetC <- true
self . controlC <- false
}
}
func ( self * section ) abort ( ) {
2014-12-10 06:12:25 +02:00
self . lock . Lock ( )
defer self . lock . Unlock ( )
2014-12-14 20:08:18 +02:00
if self . controlC != nil {
close ( self . controlC )
self . controlC = nil
2014-12-10 06:12:25 +02:00
}
2014-12-14 20:08:18 +02:00
}
func ( self * section ) done ( ) bool {
self . lock . Lock ( )
defer self . lock . Unlock ( )
if self . controlC != nil {
return true
2014-12-10 06:12:25 +02:00
}
2014-12-14 20:08:18 +02:00
return false
}
func ( self * BlockPool ) get ( hash [ ] byte ) ( node * poolNode ) {
self . lock . Lock ( )
defer self . lock . Unlock ( )
return self . pool [ string ( hash ) ]
}
func ( self * BlockPool ) set ( hash [ ] byte , node * poolNode ) {
self . lock . Lock ( )
defer self . lock . Unlock ( )
self . pool [ string ( hash ) ] = node
}
// first time for block request, this iteration retrieves nodes of the chain
// from node up to top (all the way if nil) via child links
// copies the controller
// and feeds nodeC channel
// this is performed under section readlock to prevent top from going away
// when
func ( self * BlockPool ) foldUp ( node * poolNode , nodeC chan * poolNode ) {
self . wg . Add ( 1 )
go func ( ) {
node . sectionRLock ( )
defer node . sectionRUnlock ( )
for node != nil {
select {
case <- self . quit :
break
case nodeC <- node :
if node == node . section . top {
break
}
node = node . child
}
}
close ( nodeC )
self . wg . Done ( )
} ( )
}
func ( self * poolNode ) Lock ( ) {
self . sectionLock ( )
self . lock . Lock ( )
}
func ( self * poolNode ) Unlock ( ) {
self . lock . Unlock ( )
self . sectionUnlock ( )
}
func ( self * poolNode ) RLock ( ) {
self . lock . RLock ( )
}
func ( self * poolNode ) RUnlock ( ) {
self . lock . RUnlock ( )
}
func ( self * poolNode ) sectionLock ( ) {
self . lock . RLock ( )
defer self . lock . RUnlock ( )
self . section . lock . Lock ( )
}
func ( self * poolNode ) sectionUnlock ( ) {
self . lock . RLock ( )
defer self . lock . RUnlock ( )
self . section . lock . Unlock ( )
}
func ( self * poolNode ) sectionRLock ( ) {
self . lock . RLock ( )
defer self . lock . RUnlock ( )
self . section . lock . RLock ( )
}
func ( self * poolNode ) sectionRUnlock ( ) {
self . lock . RLock ( )
defer self . lock . RUnlock ( )
self . section . lock . RUnlock ( )
2014-12-10 06:12:25 +02:00
}