9bb89ab323* Improve const correctness and adhere to 'const' specifier positioning style Co-authored-by: Mario Ortiz Manero <marioortizmanero@gmail.com>
72 lines
3 KiB
C++
72 lines
3 KiB
C++
#pragma once
|
|
|
|
#include <nano/node/common.hpp>
|
|
#include <nano/node/socket.hpp>
|
|
|
|
#include <atomic>
|
|
#include <queue>
|
|
|
|
namespace nano
|
|
{
|
|
class bootstrap_server;
|
|
class bootstrap_listener final
|
|
{
|
|
public:
|
|
bootstrap_listener (uint16_t, nano::node &);
|
|
void start ();
|
|
void stop ();
|
|
void accept_action (boost::system::error_code const &, std::shared_ptr<nano::socket> const &);
|
|
std::size_t connection_count ();
|
|
|
|
nano::mutex mutex;
|
|
std::unordered_map<nano::bootstrap_server *, std::weak_ptr<nano::bootstrap_server>> connections;
|
|
nano::tcp_endpoint endpoint ();
|
|
nano::node & node;
|
|
std::shared_ptr<nano::server_socket> listening_socket;
|
|
bool on{ false };
|
|
std::atomic<std::size_t> bootstrap_count{ 0 };
|
|
std::atomic<std::size_t> realtime_count{ 0 };
|
|
|
|
private:
|
|
uint16_t port;
|
|
};
|
|
|
|
std::unique_ptr<container_info_component> collect_container_info (bootstrap_listener & bootstrap_listener, std::string const & name);
|
|
|
|
class message;
|
|
class bootstrap_server final : public std::enable_shared_from_this<nano::bootstrap_server>
|
|
{
|
|
public:
|
|
bootstrap_server (std::shared_ptr<nano::socket> const &, std::shared_ptr<nano::node> const &);
|
|
~bootstrap_server ();
|
|
void stop ();
|
|
void receive ();
|
|
void receive_header_action (boost::system::error_code const &, std::size_t);
|
|
void receive_bulk_pull_action (boost::system::error_code const &, std::size_t, nano::message_header const &);
|
|
void receive_bulk_pull_account_action (boost::system::error_code const &, std::size_t, nano::message_header const &);
|
|
void receive_frontier_req_action (boost::system::error_code const &, std::size_t, nano::message_header const &);
|
|
void receive_keepalive_action (boost::system::error_code const &, std::size_t, nano::message_header const &);
|
|
void receive_publish_action (boost::system::error_code const &, std::size_t, nano::message_header const &);
|
|
void receive_confirm_req_action (boost::system::error_code const &, std::size_t, nano::message_header const &);
|
|
void receive_confirm_ack_action (boost::system::error_code const &, std::size_t, nano::message_header const &);
|
|
void receive_node_id_handshake_action (boost::system::error_code const &, std::size_t, nano::message_header const &);
|
|
void receive_telemetry_ack_action (boost::system::error_code const & ec, std::size_t size_a, nano::message_header const & header_a);
|
|
void add_request (std::unique_ptr<nano::message>);
|
|
void finish_request ();
|
|
void finish_request_async ();
|
|
void timeout ();
|
|
void run_next (nano::unique_lock<nano::mutex> & lock_a);
|
|
bool is_bootstrap_connection ();
|
|
bool is_realtime_connection ();
|
|
std::shared_ptr<std::vector<uint8_t>> receive_buffer;
|
|
std::shared_ptr<nano::socket> const socket;
|
|
std::shared_ptr<nano::node> node;
|
|
nano::mutex mutex;
|
|
std::queue<std::unique_ptr<nano::message>> requests;
|
|
std::atomic<bool> stopped{ false };
|
|
// Remote enpoint used to remove response channel even after socket closing
|
|
nano::tcp_endpoint remote_endpoint{ boost::asio::ip::address_v6::any (), 0 };
|
|
nano::account remote_node_id{};
|
|
std::chrono::steady_clock::time_point last_telemetry_req{ std::chrono::steady_clock::time_point () };
|
|
};
|
|
}
|