101 lines
4.5 KiB
C++
101 lines
4.5 KiB
C++
#pragma once
|
|
|
|
#include <nano/lib/network_filter.hpp>
|
|
#include <nano/node/endpoint.hpp>
|
|
#include <nano/node/messages.hpp>
|
|
|
|
#include <memory>
|
|
#include <vector>
|
|
|
|
namespace nano
|
|
{
|
|
namespace transport
|
|
{
|
|
enum class parse_status
|
|
{
|
|
none,
|
|
success,
|
|
insufficient_work,
|
|
invalid_header,
|
|
invalid_message_type,
|
|
invalid_keepalive_message,
|
|
invalid_publish_message,
|
|
invalid_confirm_req_message,
|
|
invalid_confirm_ack_message,
|
|
invalid_node_id_handshake_message,
|
|
invalid_telemetry_req_message,
|
|
invalid_telemetry_ack_message,
|
|
invalid_bulk_pull_message,
|
|
invalid_bulk_pull_account_message,
|
|
invalid_frontier_req_message,
|
|
invalid_asc_pull_req_message,
|
|
invalid_asc_pull_ack_message,
|
|
invalid_network,
|
|
outdated_version,
|
|
duplicate_publish_message,
|
|
duplicate_confirm_ack_message,
|
|
message_size_too_big,
|
|
};
|
|
|
|
class message_deserializer : public std::enable_shared_from_this<nano::transport::message_deserializer>
|
|
{
|
|
public:
|
|
using callback_type = std::function<void (boost::system::error_code, std::unique_ptr<nano::message>)>;
|
|
|
|
parse_status status{ parse_status::none };
|
|
|
|
using read_query = std::function<void (std::shared_ptr<std::vector<uint8_t>> const &, size_t, std::function<void (boost::system::error_code const &, std::size_t)>)>;
|
|
|
|
message_deserializer (nano::network_constants const &, nano::network_filter &, nano::block_uniquer &, nano::vote_uniquer &, read_query read_op);
|
|
|
|
/*
|
|
* Asynchronously read next message from the channel_read_fn.
|
|
* If an irrecoverable error is encountered callback will be called with an error code set and null message.
|
|
* If a 'soft' error is encountered (eg. duplicate block publish) error won't be set but message will be null. In that case, `status` field will be set to code indicating reason for failure.
|
|
* If message is received successfully, error code won't be set and message will be non-null. `status` field will be set to `success`.
|
|
* Should not be called until the previous invocation finishes and calls the callback.
|
|
*/
|
|
void read (callback_type const && callback);
|
|
|
|
private:
|
|
void received_header (callback_type const && callback);
|
|
void received_message (nano::message_header header, std::size_t payload_size, callback_type const && callback);
|
|
|
|
/*
|
|
* Deserializes message using data in `read_buffer`.
|
|
* @return If successful returns non-null message, otherwise sets `status` to error appropriate code and returns nullptr
|
|
*/
|
|
std::unique_ptr<nano::message> deserialize (nano::message_header header, std::size_t payload_size);
|
|
std::unique_ptr<nano::keepalive> deserialize_keepalive (nano::stream &, nano::message_header const &);
|
|
std::unique_ptr<nano::publish> deserialize_publish (nano::stream &, nano::message_header const &, nano::network_filter::digest_t const & digest);
|
|
std::unique_ptr<nano::confirm_req> deserialize_confirm_req (nano::stream &, nano::message_header const &);
|
|
std::unique_ptr<nano::confirm_ack> deserialize_confirm_ack (nano::stream &, nano::message_header const &, nano::network_filter::digest_t const & digest);
|
|
std::unique_ptr<nano::node_id_handshake> deserialize_node_id_handshake (nano::stream &, nano::message_header const &);
|
|
std::unique_ptr<nano::telemetry_req> deserialize_telemetry_req (nano::stream &, nano::message_header const &);
|
|
std::unique_ptr<nano::telemetry_ack> deserialize_telemetry_ack (nano::stream &, nano::message_header const &);
|
|
std::unique_ptr<nano::bulk_pull> deserialize_bulk_pull (nano::stream &, nano::message_header const &);
|
|
std::unique_ptr<nano::bulk_pull_account> deserialize_bulk_pull_account (nano::stream &, nano::message_header const &);
|
|
std::unique_ptr<nano::bulk_push> deserialize_bulk_push (nano::stream &, nano::message_header const &);
|
|
std::unique_ptr<nano::frontier_req> deserialize_frontier_req (nano::stream &, nano::message_header const &);
|
|
std::unique_ptr<nano::asc_pull_req> deserialize_asc_pull_req (nano::stream &, nano::message_header const &);
|
|
std::unique_ptr<nano::asc_pull_ack> deserialize_asc_pull_ack (nano::stream &, nano::message_header const &);
|
|
|
|
private:
|
|
std::shared_ptr<std::vector<uint8_t>> read_buffer;
|
|
|
|
private: // Constants
|
|
static constexpr std::size_t HEADER_SIZE = 8;
|
|
static constexpr std::size_t MAX_MESSAGE_SIZE = 1024 * 65;
|
|
|
|
private: // Dependencies
|
|
nano::network_constants const & network_constants_m;
|
|
nano::network_filter & network_filter_m;
|
|
nano::block_uniquer & block_uniquer_m;
|
|
nano::vote_uniquer & vote_uniquer_m;
|
|
read_query read_op;
|
|
};
|
|
|
|
nano::stat::detail to_stat_detail (parse_status);
|
|
std::string_view to_string (parse_status);
|
|
}
|
|
}
|