TARAXA
Loading...
Searching...
No Matches
taraxa_capability.hpp
Go to the documentation of this file.
1#pragma once
2
3#include <libp2p/Capability.h>
4#include <libp2p/Common.h>
5#include <libp2p/Host.h>
6#include <libp2p/Session.h>
7
8#include <memory>
9
11#include "config/config.hpp"
16#include "pbft/pbft_chain.hpp"
18
19namespace taraxa {
20class DbStorage;
21class PbftManager;
22class PbftChain;
23class VoteManager;
24class DagManager;
25class TransactionManager;
26class SlashingManager;
27enum class TransactionStatus;
28
29namespace pillar_chain {
30class PillarChainManager;
31}
32
33} // namespace taraxa
34
36
37class ISyncPacketHandler;
38class IVotePacketHandler;
39class IPillarVotePacketHandler;
40class IGetPillarVotesBundlePacketHandler;
41class ITransactionPacketHandler;
42class IDagBlockPacketHandler;
43
44class PbftSyncingState;
45class TaraxaPeer;
46
48 public:
52 using InitPacketsHandlers = std::function<std::shared_ptr<PacketsHandler>(
53 const std::string &logs_prefix, const FullNodeConfig &config, const h256 &genesis_hash,
54 const std::shared_ptr<PeersState> &peers_state, const std::shared_ptr<PbftSyncingState> &pbft_syncing_state,
55
56 const std::shared_ptr<tarcap::TimePeriodPacketsStats> &packets_stats, const std::shared_ptr<DbStorage> &db,
57 const std::shared_ptr<PbftManager> &pbft_mgr, const std::shared_ptr<PbftChain> &pbft_chain,
58 const std::shared_ptr<VoteManager> &vote_mgr, const std::shared_ptr<DagManager> &dag_mgr,
59 const std::shared_ptr<TransactionManager> &trx_mgr, const std::shared_ptr<SlashingManager> &slashing_manager,
60 const std::shared_ptr<pillar_chain::PillarChainManager> &pillar_chain_mgr, TarcapVersion version)>;
61
67
68 public:
69 TaraxaCapability(TarcapVersion version, const FullNodeConfig &conf, const h256 &genesis_hash,
70 std::weak_ptr<dev::p2p::Host> host,
71 std::shared_ptr<network::threadpool::PacketsThreadPool> threadpool,
72 std::shared_ptr<TimePeriodPacketsStats> packets_stats,
73 std::shared_ptr<PbftSyncingState> syncing_state, std::shared_ptr<DbStorage> db,
74 std::shared_ptr<PbftManager> pbft_mgr, std::shared_ptr<PbftChain> pbft_chain,
75 std::shared_ptr<VoteManager> vote_mgr, std::shared_ptr<DagManager> dag_mgr,
76 std::shared_ptr<TransactionManager> trx_mgr, std::shared_ptr<SlashingManager> slashing_manager,
77 std::shared_ptr<pillar_chain::PillarChainManager> pillar_chain_mgr,
78 InitPacketsHandlers init_packets_handlers = kInitLatestVersionHandlers);
79
80 virtual ~TaraxaCapability() = default;
85
86 // CapabilityFace implemented interface
87 std::string name() const override;
88 TarcapVersion version() const override;
89 unsigned messageCount() const override;
90 void onConnect(std::weak_ptr<dev::p2p::Session> session, u256 const &) override;
91 void onDisconnect(dev::p2p::NodeID const &_nodeID) override;
92 void interpretCapabilityPacket(std::weak_ptr<dev::p2p::Session> session, unsigned _id, dev::RLP const &_r) override;
93 std::string packetTypeToString(unsigned _packetType) const override;
94
95 const std::shared_ptr<PeersState> &getPeersState();
96
104 template <typename PacketHandlerType>
105 std::shared_ptr<PacketHandlerType> getSpecificHandler(SubprotocolPacketType packet_type) const;
106
107 private:
109 void handlePacketQueueOverLimit(std::shared_ptr<dev::p2p::Host> host, dev::p2p::NodeID node_id, size_t tp_queue_size);
110
111 private:
112 // Capability version
114
115 // Packets stats per time period
116 std::shared_ptr<TimePeriodPacketsStats> all_packets_stats_;
117
118 // Node config
120
121 // Peers state
122 std::shared_ptr<PeersState> peers_state_;
123
124 // Syncing state + syncing handler
125 std::shared_ptr<PbftSyncingState> pbft_syncing_state_;
126
127 // Packets handlers
128 std::shared_ptr<PacketsHandler> packets_handlers_;
129
130 // Main Threadpool for processing packets
131 std::shared_ptr<threadpool::PacketsThreadPool> thread_pool_;
132
133 // Last disconnect time and number of peers
134 std::chrono::system_clock::time_point last_ddos_disconnect_time_ = {};
135 std::chrono::system_clock::time_point queue_over_limit_start_time_ = {};
136 bool queue_over_limit_ = false;
138
140};
141
142template <typename PacketHandlerType>
143std::shared_ptr<PacketHandlerType> TaraxaCapability::getSpecificHandler(SubprotocolPacketType packet_type) const {
144 // Note: Allow to manually cast only to known base classes types.
145 // We support multiple taraxa capabilities, which can contain different versions of packet handlers and casting
146 // directly to final classes types breaks the functionality...
147 switch (packet_type) {
150 if (!std::is_same<ISyncPacketHandler, PacketHandlerType>::value) {
151 assert(false);
152 }
153 break;
154
156 if (!std::is_same<ITransactionPacketHandler, PacketHandlerType>::value) {
157 assert(false);
158 }
159 break;
160
163 if (!std::is_same<IVotePacketHandler, PacketHandlerType>::value) {
164 assert(false);
165 }
166 break;
167
169 if (!std::is_same<IPillarVotePacketHandler, PacketHandlerType>::value) {
170 assert(false);
171 }
172 break;
173
175 if (!std::is_same<IGetPillarVotesBundlePacketHandler, PacketHandlerType>::value) {
176 assert(false);
177 }
178 break;
179
181 if (!std::is_same<IDagBlockPacketHandler, PacketHandlerType>::value) {
182 assert(false);
183 }
184 break;
185
186 default:
187 assert(false);
188 return nullptr;
189 }
190
191 auto handler = packets_handlers_->getSpecificHandler(packet_type);
192 return std::dynamic_pointer_cast<PacketHandlerType>(handler);
193}
194
195} // namespace taraxa::network::tarcap
Definition RLP.h:56
The Capability interface. This interface has to be implemented to support any devp2p subprotocol.
Definition Capability.h:16
Definition taraxa_capability.hpp:47
std::string packetTypeToString(unsigned _packetType) const override
Convert supplied packet type to string - used for logging purposes.
Definition taraxa_capability.cpp:121
TarcapVersion version_
Definition taraxa_capability.hpp:113
std::string name() const override
Subprotocol name, used in negotiating common capabilities with the peers.
Definition taraxa_capability.cpp:58
void onDisconnect(dev::p2p::NodeID const &_nodeID) override
Definition taraxa_capability.cpp:102
void handlePacketQueueOverLimit(std::shared_ptr< dev::p2p::Host > host, dev::p2p::NodeID node_id, size_t tp_queue_size)
Definition taraxa_capability.cpp:203
const FullNodeConfig & kConf
Definition taraxa_capability.hpp:119
TaraxaCapability & operator=(const TaraxaCapability &ro)=delete
uint32_t last_disconnect_number_of_peers_
Definition taraxa_capability.hpp:137
std::shared_ptr< PbftSyncingState > pbft_syncing_state_
Definition taraxa_capability.hpp:125
std::shared_ptr< PacketHandlerType > getSpecificHandler(SubprotocolPacketType packet_type) const
templated getSpecificHandler method for getting specific packet handler based on packet_type
Definition taraxa_capability.hpp:143
std::function< std::shared_ptr< PacketsHandler >(const std::string &logs_prefix, const FullNodeConfig &config, const h256 &genesis_hash, const std::shared_ptr< PeersState > &peers_state, const std::shared_ptr< PbftSyncingState > &pbft_syncing_state, const std::shared_ptr< tarcap::TimePeriodPacketsStats > &packets_stats, const std::shared_ptr< DbStorage > &db, const std::shared_ptr< PbftManager > &pbft_mgr, const std::shared_ptr< PbftChain > &pbft_chain, const std::shared_ptr< VoteManager > &vote_mgr, const std::shared_ptr< DagManager > &dag_mgr, const std::shared_ptr< TransactionManager > &trx_mgr, const std::shared_ptr< SlashingManager > &slashing_manager, const std::shared_ptr< pillar_chain::PillarChainManager > &pillar_chain_mgr, TarcapVersion version)> InitPacketsHandlers
Function signature for creating taraxa capability packets handlers.
Definition taraxa_capability.hpp:60
static const InitPacketsHandlers kInitLatestVersionHandlers
Default InitPacketsHandlers function definition with the latest version of packets handlers.
Definition taraxa_capability.hpp:65
std::chrono::system_clock::time_point queue_over_limit_start_time_
Definition taraxa_capability.hpp:135
static const InitPacketsHandlers kInitV5VersionHandlers
Definition taraxa_capability.hpp:66
TaraxaCapability(TaraxaCapability &&ro)=delete
TarcapVersion version() const override
Definition taraxa_capability.cpp:60
void onConnect(std::weak_ptr< dev::p2p::Session > session, u256 const &) override
Definition taraxa_capability.cpp:64
std::shared_ptr< PacketsHandler > packets_handlers_
Definition taraxa_capability.hpp:128
void interpretCapabilityPacket(std::weak_ptr< dev::p2p::Session > session, unsigned _id, dev::RLP const &_r) override
Definition taraxa_capability.cpp:125
std::shared_ptr< TimePeriodPacketsStats > all_packets_stats_
Definition taraxa_capability.hpp:116
const std::shared_ptr< PeersState > & getPeersState()
Definition taraxa_capability.cpp:254
TaraxaCapability(const TaraxaCapability &ro)=delete
logger::Logger logger_
Definition taraxa_capability.hpp:139
std::shared_ptr< threadpool::PacketsThreadPool > thread_pool_
Definition taraxa_capability.hpp:131
TaraxaCapability & operator=(TaraxaCapability &&ro)=delete
bool queue_over_limit_
Definition taraxa_capability.hpp:136
std::shared_ptr< PeersState > peers_state_
Definition taraxa_capability.hpp:122
unsigned messageCount() const override
Number of messages supported by the subprotocol version.
Definition taraxa_capability.cpp:62
bool filterSyncIrrelevantPackets(SubprotocolPacketType packet_type) const
Definition taraxa_capability.cpp:243
std::chrono::system_clock::time_point last_ddos_disconnect_time_
Definition taraxa_capability.hpp:134
TransactionStatus
TransactionStatus enum class defines current transaction status.
Definition transaction_manager.hpp:23
std::shared_ptr< spdlog::logger > Logger
Definition logging.hpp:12
Definition vote_manager.hpp:24
unsigned TarcapVersion
Definition tarcap_version.hpp:4
SubprotocolPacketType
SubprotocolPacketType is used in networking layer to differentiate packet types.
Definition packet_types.hpp:12
@ kGetPillarVotesBundlePacket
Definition packet_types.hpp:33
@ kStatusPacket
Definition packet_types.hpp:28
@ kDagBlockPacket
Definition packet_types.hpp:21
@ kVotePacket
Definition packet_types.hpp:15
@ kPbftSyncPacket
Definition packet_types.hpp:30
@ kVotesBundlePacket
Definition packet_types.hpp:17
@ kTransactionPacket
Definition packet_types.hpp:24
@ kPillarVotePacket
Definition packet_types.hpp:32
Definition app.hpp:16
Definition config.hpp:41