mirror of
https://github.com/ton-blockchain/ton
synced 2025-03-09 15:40:10 +00:00
113 lines
3.7 KiB
C++
113 lines
3.7 KiB
C++
/*
|
|
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-2020 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"
|
|
|
|
#include <bitset>
|
|
|
|
namespace td {
|
|
class UdpServer;
|
|
}
|
|
|
|
namespace ton {
|
|
|
|
namespace adnl {
|
|
|
|
class AdnlPeerTable;
|
|
|
|
using AdnlCategoryMask = std::bitset<256>;
|
|
|
|
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, AdnlCategoryMask cat_mask, 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, AdnlCategoryMask cat_mask, td::uint32 priority) = 0;
|
|
virtual void add_proxy_addr(td::IPAddress addr, td::uint16 local_port, std::shared_ptr<AdnlProxy> proxy,
|
|
AdnlCategoryMask cat_mask, 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;
|
|
virtual void set_local_id_category(AdnlNodeIdShort id, td::uint8 cat) = 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
|