1 files changed, 204 insertions, 0 deletions
diff --git a/src/p2p/net_node.h b/src/p2p/net_node.h
new file mode 100644
index 000000000..32249a6d1
--- /dev/null
+++ b/src/p2p/net_node.h
@@ -0,0 +1,204 @@
+// Copyright (c) 2012-2013 The Cryptonote developers
+// Distributed under the MIT/X11 software license, see the accompanying
+// file COPYING or http://www.opensource.org/licenses/mit-license.php.
+
+#pragma once
+#include <boost/thread.hpp>
+#include <boost/bind.hpp>
+#include <boost/foreach.hpp>
+#include <boost/bimap.hpp>
+#include <boost/multi_index_container.hpp>
+#include <boost/multi_index/ordered_index.hpp>
+#include <boost/multi_index/identity.hpp>
+#include <boost/multi_index/member.hpp>
+#include <boost/multi_index/global_fun.hpp>
+#include <boost/program_options/options_description.hpp>
+#include <boost/program_options/variables_map.hpp>
+#include <boost/serialization/version.hpp>
+
+#include "warnings.h"
+#include "net/levin_server_cp2.h"
+#include "p2p_protocol_defs.h"
+#include "storages/levin_abstract_invoke2.h"
+#include "net_peerlist.h"
+#include "p2p_networks.h"
+#include "math_helper.h"
+#include "net_node_common.h"
+
+using namespace epee;
+
+PUSH_WARNINGS
+DISABLE_VS_WARNINGS(4355)
+
+namespace nodetool
+{
+  template<class base_type>
+  struct p2p_connection_context_t: base_type //t_payload_net_handler::connection_context //public net_utils::connection_context_base
+  {
+    peerid_type peer_id;
+  };
+
+  template<class t_payload_net_handler>
+  class node_server: public levin::levin_commands_handler<p2p_connection_context_t<typename t_payload_net_handler::connection_context> >,
+                     public i_p2p_endpoint<typename t_payload_net_handler::connection_context>
+  {
+    struct by_conn_id{};
+    struct by_peer_id{};
+    struct by_addr{};
+
+    typedef p2p_connection_context_t<typename t_payload_net_handler::connection_context> p2p_connection_context;
+
+    typedef COMMAND_HANDSHAKE_T<typename t_payload_net_handler::payload_type> COMMAND_HANDSHAKE;
+    typedef COMMAND_TIMED_SYNC_T<typename t_payload_net_handler::payload_type> COMMAND_TIMED_SYNC;
+
+  public:
+    typedef t_payload_net_handler payload_net_handler;
+    // Some code
+    node_server(t_payload_net_handler& payload_handler):m_payload_handler(payload_handler), m_allow_local_ip(false), m_hide_my_port(false)
+    {}
+
+    static void init_options(boost::program_options::options_description& desc);
+
+    bool run();
+    bool init(const boost::program_options::variables_map& vm);
+    bool deinit();
+    bool send_stop_signal();
+    uint32_t get_this_peer_port(){return m_listenning_port;}
+    t_payload_net_handler& get_payload_object();
+
+    template <class Archive, class t_version_type>
+    void serialize(Archive &a,  const t_version_type ver)
+    {
+      a & m_peerlist;
+      a & m_config.m_peer_id;
+    }
+    // debug functions
+    bool log_peerlist();
+    bool log_connections();
+    virtual uint64_t get_connections_count();
+    size_t get_outgoing_connections_count();
+    peerlist_manager& get_peerlist_manager(){return m_peerlist;}
+  private:
+    typedef COMMAND_REQUEST_STAT_INFO_T<typename t_payload_net_handler::stat_info> COMMAND_REQUEST_STAT_INFO;
+
+    CHAIN_LEVIN_INVOKE_MAP2(p2p_connection_context); //move levin_commands_handler interface invoke(...) callbacks into invoke map
+    CHAIN_LEVIN_NOTIFY_MAP2(p2p_connection_context); //move levin_commands_handler interface notify(...) callbacks into nothing
+
+    BEGIN_INVOKE_MAP2(node_server)
+      HANDLE_INVOKE_T2(COMMAND_HANDSHAKE, &node_server::handle_handshake)
+      HANDLE_INVOKE_T2(COMMAND_TIMED_SYNC, &node_server::handle_timed_sync)
+      HANDLE_INVOKE_T2(COMMAND_PING, &node_server::handle_ping)
+#ifdef ALLOW_DEBUG_COMMANDS
+      HANDLE_INVOKE_T2(COMMAND_REQUEST_STAT_INFO, &node_server::handle_get_stat_info)
+      HANDLE_INVOKE_T2(COMMAND_REQUEST_NETWORK_STATE, &node_server::handle_get_network_state)
+      HANDLE_INVOKE_T2(COMMAND_REQUEST_PEER_ID, &node_server::handle_get_peer_id)
+#endif
+      CHAIN_INVOKE_MAP_TO_OBJ_FORCE_CONTEXT(m_payload_handler, typename t_payload_net_handler::connection_context&)
+    END_INVOKE_MAP2()
+
+    //----------------- commands handlers ----------------------------------------------
+    int handle_handshake(int command, typename COMMAND_HANDSHAKE::request& arg, typename COMMAND_HANDSHAKE::response& rsp, p2p_connection_context& context);
+    int handle_timed_sync(int command, typename COMMAND_TIMED_SYNC::request& arg, typename COMMAND_TIMED_SYNC::response& rsp, p2p_connection_context& context);
+    int handle_ping(int command, COMMAND_PING::request& arg, COMMAND_PING::response& rsp, p2p_connection_context& context);
+#ifdef ALLOW_DEBUG_COMMANDS
+    int handle_get_stat_info(int command, typename COMMAND_REQUEST_STAT_INFO::request& arg, typename COMMAND_REQUEST_STAT_INFO::response& rsp, p2p_connection_context& context);
+    int handle_get_network_state(int command, COMMAND_REQUEST_NETWORK_STATE::request& arg, COMMAND_REQUEST_NETWORK_STATE::response& rsp, p2p_connection_context& context);
+    int handle_get_peer_id(int command, COMMAND_REQUEST_PEER_ID::request& arg, COMMAND_REQUEST_PEER_ID::response& rsp, p2p_connection_context& context);
+#endif
+    bool init_config();
+    bool make_default_config();
+    bool store_config();
+    bool check_trust(const proof_of_trust& tr);
+
+
+    //----------------- levin_commands_handler -------------------------------------------------------------
+    virtual void on_connection_new(p2p_connection_context& context);
+    virtual void on_connection_close(p2p_connection_context& context);
+    virtual void callback(p2p_connection_context& context);
+    //----------------- i_p2p_endpoint -------------------------------------------------------------
+    virtual bool relay_notify_to_all(int command, const std::string& data_buff, const epee::net_utils::connection_context_base& context);
+    virtual bool invoke_command_to_peer(int command, const std::string& req_buff, std::string& resp_buff, const epee::net_utils::connection_context_base& context);
+    virtual bool invoke_notify_to_peer(int command, const std::string& req_buff, const epee::net_utils::connection_context_base& context);
+    virtual bool drop_connection(const epee::net_utils::connection_context_base& context);
+    virtual void request_callback(const epee::net_utils::connection_context_base& context);
+    virtual void for_each_connection(std::function<bool(typename t_payload_net_handler::connection_context&)> f);
+    //-----------------------------------------------------------------------------------------------
+    bool parse_peer_from_string(nodetool::net_address& pe, const std::string& node_addr);
+    bool handle_command_line(const boost::program_options::variables_map& vm);
+    bool idle_worker();
+    bool handle_remote_peerlist(const std::list<peerlist_entry>& peerlist, time_t local_time, const net_utils::connection_context_base& context);
+    bool get_local_node_data(basic_node_data& node_data);
+    //bool get_local_handshake_data(handshake_data& hshd);
+
+    bool merge_peerlist_with_local(const std::list<peerlist_entry>& bs);
+    bool fix_time_delta(std::list<peerlist_entry>& local_peerlist, time_t local_time, int64_t& delta);
+
+    bool connections_maker();
+    bool peer_sync_idle_maker();
+    bool do_handshake_with_peer(peerid_type& pi, p2p_connection_context& context, bool just_take_peerlist = false);
+    bool do_peer_timed_sync(const net_utils::connection_context_base& context, peerid_type peer_id);
+
+    bool make_new_connection_from_peerlist(bool use_white_list);
+    bool try_to_connect_and_handshake_with_new_peer(const net_address& na, bool just_take_peerlist = false, uint64_t last_seen_stamp = 0, bool white = true);
+    size_t get_random_index_with_fixed_probability(size_t max_index);
+    bool is_peer_used(const peerlist_entry& peer);
+    bool is_addr_connected(const net_address& peer);  
+    template<class t_callback>
+    bool try_ping(basic_node_data& node_data, p2p_connection_context& context, t_callback cb);
+    bool make_expected_connections_count(bool white_list, size_t expected_connections);
+
+    //debug functions
+    std::string print_connections_container();
+
+
+    typedef net_utils::boosted_tcp_server<levin::async_protocol_handler<p2p_connection_context> > net_server;
+
+    struct config
+    {
+      network_config m_net_config;
+      uint64_t m_peer_id;
+
+      BEGIN_KV_SERIALIZE_MAP()
+        KV_SERIALIZE(m_net_config)
+        KV_SERIALIZE(m_peer_id)
+      END_KV_SERIALIZE_MAP()
+    };
+
+    config m_config;
+    std::string m_config_folder;
+
+    bool m_have_address;
+    bool m_first_connection_maker_call;
+    uint32_t m_listenning_port;
+    uint32_t m_external_port;
+    uint32_t m_ip_address;
+    bool m_allow_local_ip;
+    bool m_hide_my_port;
+
+    //critical_section m_connections_lock;
+    //connections_indexed_container m_connections;
+
+    t_payload_net_handler& m_payload_handler;
+    peerlist_manager m_peerlist;
+
+    math_helper::once_a_time_seconds<P2P_DEFAULT_HANDSHAKE_INTERVAL> m_peer_handshake_idle_maker_interval;
+    math_helper::once_a_time_seconds<1> m_connections_maker_interval;
+    math_helper::once_a_time_seconds<60*30, false> m_peerlist_store_interval;
+
+    std::string m_bind_ip;
+    std::string m_port;
+#ifdef ALLOW_DEBUG_COMMANDS
+    uint64_t m_last_stat_request_time;
+#endif
+    std::list<net_address>   m_priority_peers;
+    std::vector<net_address> m_seed_nodes;
+    std::list<nodetool::peerlist_entry> m_command_line_peers;
+    uint64_t m_peer_livetime;
+    //keep connections to initiate some interactions
+    net_server m_net_server;
+  };
+}
+
+#include "net_node.inl"
+
+POP_WARNINGS