node.hpp

// Copyright (C) 2018 Bluzelle
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU Affero General Public License, version 3,
// as published by the Free Software Foundation.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU Affero General Public License for more details.
//
// You should have received a copy of the GNU Affero General Public License
// along with this program. If not, see <http://www.gnu.org/licenses/>.

#pragma once

#include <include/boost_asio_beast.hpp>
#include <node/node_base.hpp>
#include <chaos/chaos_base.hpp>
#include <crypto/crypto_base.hpp>
#include <options/options_base.hpp>
#include <json/json.h>
#include <mutex>
#include <atomic>

#include <gtest/gtest_prod.h>


namespace bzn
{
    class node final : public bzn::node_base, public std::enable_shared_from_this<node>
    {
    public:
        node(std::shared_ptr<bzn::asio::io_context_base> io_context, std::shared_ptr<bzn::beast::websocket_base> websocket, std::shared_ptr<bzn::chaos_base> chaos, const std::chrono::milliseconds& ws_idle_timeout,
            const boost::asio::ip::tcp::endpoint& ep, std::shared_ptr<bzn::crypto_base> crypto, std::shared_ptr<bzn::options_base> options);

        bool register_for_message(const std::string& msg_type, bzn::message_handler msg_handler) override;

        bool register_for_message(const bzn_envelope::PayloadCase type, bzn::protobuf_handler msg_handler) override;

        void start() override;

        void send_message_json(const boost::asio::ip::tcp::endpoint& ep, std::shared_ptr<bzn::json_message> msg) override;

        void send_message(const boost::asio::ip::tcp::endpoint& ep, std::shared_ptr<bzn_envelope> msg, bool close_session) override;

        void send_message_str(const boost::asio::ip::tcp::endpoint& ep, std::shared_ptr<bzn::encoded_message> msg, bool close_session) override;

    private:
        FRIEND_TEST(node, test_that_registered_message_handler_is_invoked);
        FRIEND_TEST(node, test_that_wrongly_signed_messages_are_dropped);

        void do_accept();

        void priv_msg_handler(const bzn::json_message& msg, std::shared_ptr<bzn::session_base> session);
        void priv_protobuf_handler(const bzn_envelope& msg, std::shared_ptr<bzn::session_base> session);

        std::unique_ptr<bzn::asio::tcp_acceptor_base> tcp_acceptor;
        std::shared_ptr<bzn::asio::io_context_base>   io_context;
        std::unique_ptr<bzn::asio::tcp_socket_base>   acceptor_socket;
        std::shared_ptr<bzn::beast::websocket_base>   websocket;
        std::shared_ptr<bzn::chaos_base>              chaos;
        const std::chrono::milliseconds               ws_idle_timeout;

        std::unordered_map<std::string, bzn::message_handler> message_map;
        std::unordered_map<bzn_envelope::PayloadCase, bzn::protobuf_handler> protobuf_map;
        std::mutex message_map_mutex;

        std::once_flag start_once;

        std::atomic<bzn::session_id> session_id_counter = 0;

        std::shared_ptr<bzn::crypto_base> crypto;
        std::shared_ptr<bzn::options_base> options;
    };

} // bzn