mirror of
https://github.com/danog/ton.git
synced 2024-12-03 09:58:01 +01:00
122 lines
3.3 KiB
C++
122 lines
3.3 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-2019 Telegram Systems LLP
|
|
*/
|
|
#pragma once
|
|
|
|
#include "td/actor/actor.h"
|
|
#include "td/utils/BufferedUdp.h"
|
|
#include "td/net/UdpServer.h"
|
|
#include "td/net/TcpListener.h"
|
|
|
|
#include "td/actor/PromiseFuture.h"
|
|
#include "adnl-network-manager.h"
|
|
|
|
#include <map>
|
|
|
|
namespace td {
|
|
class UdpServer;
|
|
}
|
|
|
|
namespace ton {
|
|
|
|
namespace adnl {
|
|
|
|
class AdnlPeerTable;
|
|
|
|
class AdnlNetworkManagerImpl : public AdnlNetworkManager {
|
|
public:
|
|
struct OutDesc {
|
|
td::IPAddress addr;
|
|
std::shared_ptr<AdnlProxy> proxy;
|
|
|
|
bool is_proxy() const {
|
|
return proxy != nullptr;
|
|
}
|
|
bool operator==(const OutDesc &with) const {
|
|
return addr == with.addr && is_proxy() == with.is_proxy();
|
|
}
|
|
};
|
|
|
|
AdnlNetworkManagerImpl(td::uint16 out_udp_port) : out_udp_port_(out_udp_port) {
|
|
}
|
|
|
|
void install_callback(std::unique_ptr<Callback> callback) override {
|
|
callback_ = std::move(callback);
|
|
}
|
|
|
|
void add_self_addr(td::IPAddress addr, td::uint32 priority) override {
|
|
auto x = OutDesc{addr, nullptr};
|
|
auto &v = out_desc_[priority];
|
|
for (auto &y : v) {
|
|
if (x == y) {
|
|
return;
|
|
}
|
|
}
|
|
out_desc_[priority].push_back(std::move(x));
|
|
add_listening_udp_port(static_cast<td::uint16>(addr.get_port()));
|
|
}
|
|
void add_proxy_addr(td::IPAddress addr, std::shared_ptr<AdnlProxy> proxy, td::uint32 priority) override {
|
|
auto x = OutDesc{addr, std::move(proxy)};
|
|
auto &v = out_desc_[priority];
|
|
for (auto &y : v) {
|
|
if (x == y) {
|
|
return;
|
|
}
|
|
}
|
|
out_desc_[priority].push_back(std::move(x));
|
|
if (!udp_servers_.count(out_udp_port_)) {
|
|
add_listening_udp_port(out_udp_port_);
|
|
}
|
|
}
|
|
void send_udp_packet(AdnlNodeIdShort src_id, AdnlNodeIdShort dst_id, td::IPAddress dst_addr, td::uint32 priority,
|
|
td::BufferSlice data) override;
|
|
|
|
void add_listening_udp_port(td::uint16 port);
|
|
void receive_udp_message(td::UdpMessage message);
|
|
|
|
private:
|
|
std::unique_ptr<Callback> callback_;
|
|
|
|
std::map<td::uint32, std::vector<OutDesc>> out_desc_;
|
|
|
|
td::uint64 received_messages_ = 0;
|
|
td::uint64 sent_messages_ = 0;
|
|
|
|
std::map<td::uint16, td::actor::ActorOwn<td::UdpServer>> udp_servers_;
|
|
|
|
td::uint16 out_udp_port_;
|
|
};
|
|
|
|
} // namespace adnl
|
|
|
|
} // namespace ton
|
|
|
|
namespace td {
|
|
|
|
inline td::StringBuilder &operator<<(td::StringBuilder &sb, const ton::adnl::AdnlNetworkManagerImpl &manager) {
|
|
sb << manager.print_id();
|
|
return sb;
|
|
}
|
|
|
|
inline td::StringBuilder &operator<<(td::StringBuilder &sb, const ton::adnl::AdnlNetworkManagerImpl *manager) {
|
|
sb << manager->print_id();
|
|
return sb;
|
|
}
|
|
|
|
} // namespace td
|