mirror of
https://github.com/kaspanet/kaspad.git
synced 2026-02-26 05:10:02 +00:00
8fdb5aa024
* [NOD-1123] Bubble bad-message errors up to the protocol level. * [NOD-1123] Implement Banning. * [NOD-1123] Properly use &stopped. * [NOD-1123] Ban by IP rather than IP and port. * [NOD-1123] Don't initiate connections to banned peers. * [NOD-1123] Fix infinite loop in checkOutgoingConnections. * [NOD-1123] Fix bannedAddresses key. * [NOD-1123] Rename onBadMessageHandler to onInvalidMessageHandler.
64 lines
2.2 KiB
Go
64 lines
2.2 KiB
Go
package flowcontext
|
|
|
|
import (
|
|
"github.com/kaspanet/kaspad/addrmgr"
|
|
"github.com/kaspanet/kaspad/blockdag"
|
|
"github.com/kaspanet/kaspad/config"
|
|
"github.com/kaspanet/kaspad/connmanager"
|
|
"github.com/kaspanet/kaspad/mempool"
|
|
"github.com/kaspanet/kaspad/netadapter"
|
|
"github.com/kaspanet/kaspad/netadapter/id"
|
|
"github.com/kaspanet/kaspad/protocol/flows/blockrelay"
|
|
"github.com/kaspanet/kaspad/protocol/flows/relaytransactions"
|
|
peerpkg "github.com/kaspanet/kaspad/protocol/peer"
|
|
"github.com/kaspanet/kaspad/util"
|
|
"github.com/kaspanet/kaspad/util/daghash"
|
|
"sync"
|
|
"time"
|
|
)
|
|
|
|
// FlowContext holds state that is relevant to more than one flow or one peer, and allows communication between
|
|
// different flows that can be associated to different peers.
|
|
type FlowContext struct {
|
|
cfg *config.Config
|
|
netAdapter *netadapter.NetAdapter
|
|
txPool *mempool.TxPool
|
|
addedTransactions []*util.Tx
|
|
dag *blockdag.BlockDAG
|
|
addressManager *addrmgr.AddrManager
|
|
connectionManager *connmanager.ConnectionManager
|
|
|
|
transactionsToRebroadcastLock sync.Mutex
|
|
transactionsToRebroadcast map[daghash.TxID]*util.Tx
|
|
lastRebroadcastTime time.Time
|
|
sharedRequestedTransactions *relaytransactions.SharedRequestedTransactions
|
|
|
|
sharedRequestedBlocks *blockrelay.SharedRequestedBlocks
|
|
|
|
isInIBD uint32
|
|
startIBDMutex sync.Mutex
|
|
ibdPeer *peerpkg.Peer
|
|
|
|
peers map[*id.ID]*peerpkg.Peer
|
|
peersMutex sync.RWMutex
|
|
}
|
|
|
|
// New returns a new instance of FlowContext.
|
|
func New(cfg *config.Config, dag *blockdag.BlockDAG, addressManager *addrmgr.AddrManager,
|
|
txPool *mempool.TxPool, netAdapter *netadapter.NetAdapter,
|
|
connectionManager *connmanager.ConnectionManager) *FlowContext {
|
|
|
|
return &FlowContext{
|
|
cfg: cfg,
|
|
netAdapter: netAdapter,
|
|
dag: dag,
|
|
addressManager: addressManager,
|
|
connectionManager: connectionManager,
|
|
txPool: txPool,
|
|
sharedRequestedTransactions: relaytransactions.NewSharedRequestedTransactions(),
|
|
sharedRequestedBlocks: blockrelay.NewSharedRequestedBlocks(),
|
|
peers: make(map[*id.ID]*peerpkg.Peer),
|
|
transactionsToRebroadcast: make(map[daghash.TxID]*util.Tx),
|
|
}
|
|
}
|