mirror of
https://github.com/ton-blockchain/ton
synced 2025-03-09 15:40:10 +00:00
219 lines
9.2 KiB
C++
219 lines
9.2 KiB
C++
/*
|
|
This file is part of TON Blockchain source code.
|
|
|
|
TON Blockchain is free software; you can redistribute it and/or
|
|
modify it under the terms of the GNU 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 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 General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with TON Blockchain. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
In addition, as a special exception, the copyright holders give permission
|
|
to link the code of portions of this program with the OpenSSL library.
|
|
You must obey the GNU General Public License in all respects for all
|
|
of the code used other than OpenSSL. If you modify file(s) with this
|
|
exception, you may extend this exception to your version of the file(s),
|
|
but you are not obligated to do so. If you do not wish to do so, delete this
|
|
exception statement from your version. If you delete this exception statement
|
|
from all source files in the program, then also delete it here.
|
|
|
|
Copyright 2017-2020 Telegram Systems LLP
|
|
*/
|
|
#include "adnl/adnl.h"
|
|
#include "dht/dht.h"
|
|
|
|
#include "td/actor/MultiPromise.h"
|
|
|
|
#include "ton/ton-types.h"
|
|
|
|
#include "auto/tl/ton_api_json.h"
|
|
#include "auto/tl/ton_api.hpp"
|
|
|
|
#include <set>
|
|
#include <map>
|
|
|
|
enum DhtServerPermissions : td::uint32 { vep_default = 1, vep_modify = 2, vep_unsafe = 4 };
|
|
|
|
using AdnlCategory = td::int8;
|
|
|
|
struct Config {
|
|
struct Addr {
|
|
td::IPAddress addr;
|
|
|
|
bool operator<(const Addr &with) const {
|
|
return addr < with.addr;
|
|
}
|
|
};
|
|
struct AddrCats {
|
|
td::IPAddress in_addr;
|
|
std::shared_ptr<ton::adnl::AdnlProxy> proxy;
|
|
std::set<AdnlCategory> cats;
|
|
std::set<AdnlCategory> priority_cats;
|
|
};
|
|
struct Control {
|
|
ton::PublicKeyHash key;
|
|
std::map<ton::PublicKeyHash, td::uint32> clients;
|
|
};
|
|
|
|
std::map<ton::PublicKeyHash, td::uint32> keys_refcnt;
|
|
td::uint16 out_port;
|
|
std::map<Addr, AddrCats> addrs;
|
|
std::map<ton::PublicKeyHash, AdnlCategory> adnl_ids;
|
|
std::set<ton::PublicKeyHash> dht_ids;
|
|
std::map<td::int32, Control> controls;
|
|
std::set<ton::PublicKeyHash> gc;
|
|
|
|
void decref(ton::PublicKeyHash key);
|
|
void incref(ton::PublicKeyHash key) {
|
|
keys_refcnt[key]++;
|
|
}
|
|
|
|
td::Result<bool> config_add_network_addr(td::IPAddress in_addr, td::IPAddress out_addr,
|
|
std::shared_ptr<ton::adnl::AdnlProxy> proxy, std::vector<AdnlCategory> cats,
|
|
std::vector<AdnlCategory> prio_cats);
|
|
td::Result<bool> config_add_adnl_addr(ton::PublicKeyHash addr, AdnlCategory cat);
|
|
td::Result<bool> config_add_dht_node(ton::PublicKeyHash id);
|
|
td::Result<bool> config_add_control_interface(ton::PublicKeyHash key, td::int32 port);
|
|
td::Result<bool> config_add_control_process(ton::PublicKeyHash key, td::int32 port, ton::PublicKeyHash id,
|
|
td::uint32 permissions);
|
|
td::Result<bool> config_add_gc(ton::PublicKeyHash key);
|
|
td::Result<bool> config_del_network_addr(td::IPAddress addr, std::vector<AdnlCategory> cats,
|
|
std::vector<AdnlCategory> prio_cats);
|
|
td::Result<bool> config_del_adnl_addr(ton::PublicKeyHash addr);
|
|
td::Result<bool> config_del_dht_node(ton::PublicKeyHash id);
|
|
td::Result<bool> config_del_control_interface(td::int32 port);
|
|
td::Result<bool> config_del_control_process(td::int32 port, ton::PublicKeyHash id);
|
|
td::Result<bool> config_del_gc(ton::PublicKeyHash key);
|
|
|
|
ton::tl_object_ptr<ton::ton_api::engine_validator_config> tl() const;
|
|
|
|
Config();
|
|
Config(ton::ton_api::engine_validator_config &config);
|
|
};
|
|
|
|
class DhtServer : public td::actor::Actor {
|
|
private:
|
|
td::actor::ActorOwn<ton::keyring::Keyring> keyring_;
|
|
td::actor::ActorOwn<ton::adnl::AdnlNetworkManager> adnl_network_manager_;
|
|
td::actor::ActorOwn<ton::adnl::Adnl> adnl_;
|
|
std::map<ton::PublicKeyHash, td::actor::ActorOwn<ton::dht::Dht>> dht_nodes_;
|
|
ton::PublicKeyHash default_dht_node_ = ton::PublicKeyHash::zero();
|
|
td::actor::ActorOwn<ton::adnl::AdnlExtServer> control_ext_server_;
|
|
|
|
std::string local_config_ = "";
|
|
std::string global_config_ = "ton-global.config";
|
|
std::string config_file_;
|
|
std::string temp_config_file() const {
|
|
return config_file_ + ".tmp";
|
|
}
|
|
|
|
std::string db_root_ = "/var/ton-work/db/";
|
|
|
|
std::vector<td::IPAddress> addrs_;
|
|
std::vector<td::IPAddress> proxy_addrs_;
|
|
|
|
std::shared_ptr<ton::dht::DhtGlobalConfig> dht_config_;
|
|
Config config_;
|
|
|
|
std::set<ton::PublicKeyHash> running_gc_;
|
|
|
|
std::map<ton::PublicKeyHash, ton::PublicKey> keys_;
|
|
|
|
td::Promise<ton::PublicKey> get_key_promise(td::MultiPromise::InitGuard &ig);
|
|
void got_key(ton::PublicKey key);
|
|
void deleted_key(ton::PublicKeyHash key);
|
|
|
|
void write_config(td::Promise<td::Unit> promise);
|
|
|
|
std::map<td::uint32, ton::adnl::AdnlAddressList> addr_lists_;
|
|
std::map<td::uint32, ton::adnl::AdnlAddressList> prio_addr_lists_;
|
|
|
|
std::map<ton::PublicKeyHash, td::uint32> control_permissions_;
|
|
|
|
bool read_config_ = false;
|
|
bool started_keyring_ = false;
|
|
bool started_ = false;
|
|
|
|
public:
|
|
static constexpr td::uint32 max_cat() {
|
|
return 256;
|
|
}
|
|
|
|
void set_local_config(std::string str);
|
|
void set_global_config(std::string str);
|
|
void set_db_root(std::string db_root);
|
|
void add_ip(td::IPAddress addr) {
|
|
addrs_.push_back(addr);
|
|
}
|
|
void start_up() override;
|
|
DhtServer() {
|
|
}
|
|
|
|
// load config
|
|
td::Status load_global_config();
|
|
void load_empty_local_config(td::Promise<td::Unit> promise);
|
|
void load_local_config(td::Promise<td::Unit> promise);
|
|
void load_config(td::Promise<td::Unit> promise);
|
|
|
|
void start();
|
|
|
|
void start_adnl();
|
|
void add_addr(const Config::Addr &addr, const Config::AddrCats &cats);
|
|
void add_adnl(ton::PublicKeyHash id, AdnlCategory cat);
|
|
void started_adnl();
|
|
|
|
void start_dht();
|
|
void add_dht(td::actor::ActorOwn<ton::dht::Dht> dht, td::Promise<td::Unit> promise);
|
|
void started_dht();
|
|
|
|
void start_control_interface();
|
|
void started_control_interface(td::actor::ActorOwn<ton::adnl::AdnlExtServer> control_ext_server);
|
|
|
|
void started();
|
|
|
|
void alarm() override;
|
|
void run();
|
|
|
|
void add_adnl_node(ton::PublicKey pub, AdnlCategory cat, td::Promise<td::Unit> promise);
|
|
void add_dht_node(ton::PublicKeyHash pub, td::Promise<td::Unit> promise);
|
|
void add_control_interface(ton::PublicKeyHash id, td::int32 port, td::Promise<td::Unit> promise);
|
|
void add_control_process(ton::PublicKeyHash id, td::int32 port, ton::PublicKeyHash pub, td::int32 permissions,
|
|
td::Promise<td::Unit> promise);
|
|
void del_adnl_node(ton::PublicKeyHash pub, td::Promise<td::Unit> promise);
|
|
void del_dht_node(ton::PublicKeyHash pub, td::Promise<td::Unit> promise);
|
|
|
|
static td::BufferSlice create_control_query_error(td::Status error);
|
|
|
|
void run_control_query(ton::ton_api::engine_validator_getTime &query, td::BufferSlice data, ton::PublicKeyHash src,
|
|
td::uint32 perm, td::Promise<td::BufferSlice> promise);
|
|
void run_control_query(ton::ton_api::engine_validator_importPrivateKey &query, td::BufferSlice data,
|
|
ton::PublicKeyHash src, td::uint32 perm, td::Promise<td::BufferSlice> promise);
|
|
void run_control_query(ton::ton_api::engine_validator_exportPrivateKey &query, td::BufferSlice data,
|
|
ton::PublicKeyHash src, td::uint32 perm, td::Promise<td::BufferSlice> promise);
|
|
void run_control_query(ton::ton_api::engine_validator_exportPublicKey &query, td::BufferSlice data,
|
|
ton::PublicKeyHash src, td::uint32 perm, td::Promise<td::BufferSlice> promise);
|
|
void run_control_query(ton::ton_api::engine_validator_generateKeyPair &query, td::BufferSlice data,
|
|
ton::PublicKeyHash src, td::uint32 perm, td::Promise<td::BufferSlice> promise);
|
|
void run_control_query(ton::ton_api::engine_validator_addAdnlId &query, td::BufferSlice data, ton::PublicKeyHash src,
|
|
td::uint32 perm, td::Promise<td::BufferSlice> promise);
|
|
void run_control_query(ton::ton_api::engine_validator_addDhtId &query, td::BufferSlice data, ton::PublicKeyHash src,
|
|
td::uint32 perm, td::Promise<td::BufferSlice> promise);
|
|
void run_control_query(ton::ton_api::engine_validator_getConfig &query, td::BufferSlice data, ton::PublicKeyHash src,
|
|
td::uint32 perm, td::Promise<td::BufferSlice> promise);
|
|
void run_control_query(ton::ton_api::engine_validator_sign &query, td::BufferSlice data, ton::PublicKeyHash src,
|
|
td::uint32 perm, td::Promise<td::BufferSlice> promise);
|
|
template <class T>
|
|
void run_control_query(T &query, td::BufferSlice data, ton::PublicKeyHash src, td::uint32 perm,
|
|
td::Promise<td::BufferSlice> promise) {
|
|
promise.set_value(
|
|
create_control_query_error(td::Status::Error(ton::ErrorCode::protoviolation, "query not supported")));
|
|
}
|
|
void process_control_query(ton::adnl::AdnlNodeIdShort src, ton::adnl::AdnlNodeIdShort dst, td::BufferSlice data,
|
|
td::Promise<td::BufferSlice> promise);
|
|
};
|