initial commit

adnl/adnl-network-manager.h

@@ -0,0 +1,107 @@
+/*
+    This file is part of TON Blockchain Library.
+
+    TON Blockchain Library is free software: you can redistribute it and/or modify
+    it under the terms of the GNU Lesser General Public License as published by
+    the Free Software Foundation, either version 2 of the License, or
+    (at your option) any later version.
+
+    TON Blockchain Library 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 Lesser General Public License for more details.
+
+    You should have received a copy of the GNU Lesser General Public License
+    along with TON Blockchain Library.  If not, see <http://www.gnu.org/licenses/>.
+
+    Copyright 2017-2019 Telegram Systems LLP
+*/
+#pragma once
+
+#include "td/actor/actor.h"
+
+#include "td/actor/PromiseFuture.h"
+#include "td/utils/port/IPAddress.h"
+
+#include "adnl-node-id.hpp"
+#include "adnl-proxy-types.h"
+
+namespace td {
+class UdpServer;
+}
+
+namespace ton {
+
+namespace adnl {
+
+class AdnlPeerTable;
+
+class AdnlNetworkConnection : public td::actor::Actor {
+ public:
+  class Callback {
+   public:
+    virtual void on_change_state(bool ready) = 0;
+    virtual ~Callback() = default;
+  };
+  virtual void send(AdnlNodeIdShort src, AdnlNodeIdShort dst, td::uint32 priority, td::BufferSlice message) = 0;
+  virtual bool is_alive() const = 0;
+  virtual bool is_active() const = 0;
+  virtual ~AdnlNetworkConnection() = default;
+};
+
+class AdnlNetworkManager : public td::actor::Actor {
+ public:
+  //using ConnHandle = td::uint64;
+  class Callback {
+   public:
+    virtual ~Callback() = default;
+    //virtual void receive_packet(td::IPAddress addr, ConnHandle conn_handle, td::BufferSlice data) = 0;
+    virtual void receive_packet(td::IPAddress addr, td::BufferSlice data) = 0;
+  };
+  static td::actor::ActorOwn<AdnlNetworkManager> create(td::uint16 out_port);
+
+  virtual ~AdnlNetworkManager() = default;
+
+  virtual void install_callback(std::unique_ptr<Callback> callback) = 0;
+
+  virtual void add_self_addr(td::IPAddress addr, td::uint32 priority) = 0;
+  virtual void add_proxy_addr(td::IPAddress addr, std::shared_ptr<AdnlProxy> proxy, td::uint32 priority) = 0;
+  virtual void send_udp_packet(AdnlNodeIdShort src_id, AdnlNodeIdShort dst_id, td::IPAddress dst_addr,
+                               td::uint32 priority, td::BufferSlice data) = 0;
+  //virtual void send_tcp_packet(AdnlNodeIdShort src_id, AdnlNodeIdShort dst_id, td::IPAddress dst_addr,
+  //                             td::uint32 priority, td::BufferSlice data) = 0;
+  //virtual void send_answer_packet(AdnlNodeIdShort src_id, AdnlNodeIdShort dst_id, td::IPAddress dst_addr,
+  //                             ConnHandle conn_handle, td::uint32 priority, td::BufferSlice data) = 0;
+
+  static constexpr td::uint32 get_mtu() {
+    return 1440;
+  }
+
+  struct PrintId {};
+  PrintId print_id() const {
+    return PrintId{};
+  }
+};
+
+}  // namespace adnl
+
+}  // namespace ton
+
+namespace td {
+
+inline td::StringBuilder &operator<<(td::StringBuilder &sb, const ton::adnl::AdnlNetworkManager::PrintId &id) {
+  sb << "[networkmanager]";
+  return sb;
+}
+
+inline td::StringBuilder &operator<<(td::StringBuilder &sb, const ton::adnl::AdnlNetworkManager &manager) {
+  sb << manager.print_id();
+  return sb;
+}
+
+inline td::StringBuilder &operator<<(td::StringBuilder &sb, const ton::adnl::AdnlNetworkManager *manager) {
+  sb << manager->print_id();
+  return sb;
+}
+
+}  // namespace td