1
0
mirror of https://github.com/danog/ton.git synced 2024-12-05 21:49:18 +01:00
ton/validator-engine/validator-engine.cpp

2981 lines
113 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-2019 Telegram Systems LLP
*/
#include "validator-engine.hpp"
#include "ton/ton-types.h"
#include "ton/ton-tl.hpp"
#include "common/errorlog.h"
#include "crypto/vm/cp0.h"
#include "crypto/fift/utils.h"
#include "td/utils/filesystem.h"
#include "td/actor/MultiPromise.h"
#include "td/utils/overloaded.h"
#include "td/utils/OptionsParser.h"
#include "td/utils/port/path.h"
#include "td/utils/port/signals.h"
#include "td/utils/port/user.h"
#include "td/utils/ThreadSafeCounter.h"
#include "td/utils/TsFileLog.h"
#include "td/utils/Random.h"
#include "auto/tl/lite_api.h"
#include "memprof/memprof.h"
#include "dht/dht.hpp"
#if TD_DARWIN || TD_LINUX
#include <unistd.h>
#endif
#include <algorithm>
#include <iostream>
#include <sstream>
#include <cstdlib>
#include <set>
Config::Config() {
out_port = 3278;
full_node = ton::PublicKeyHash::zero();
}
Config::Config(ton::ton_api::engine_validator_config &config) {
full_node = ton::PublicKeyHash::zero();
out_port = static_cast<td::uint16>(config.out_port_);
if (!out_port) {
out_port = 3278;
}
for (auto &addr : config.addrs_) {
td::IPAddress in_ip;
td::IPAddress out_ip;
std::shared_ptr<ton::adnl::AdnlProxy> proxy = nullptr;
std::vector<AdnlCategory> categories;
std::vector<AdnlCategory> priority_categories;
ton::ton_api::downcast_call(
*addr.get(),
td::overloaded(
[&](const ton::ton_api::engine_addr &obj) {
in_ip.init_ipv4_port(td::IPAddress::ipv4_to_str(obj.ip_), static_cast<td::uint16>(obj.port_)).ensure();
out_ip = in_ip;
categories = obj.categories_;
priority_categories = obj.priority_categories_;
},
[&](const ton::ton_api::engine_addrProxy &obj) {
in_ip.init_ipv4_port(td::IPAddress::ipv4_to_str(obj.in_ip_), static_cast<td::uint16>(obj.in_port_))
.ensure();
out_ip.init_ipv4_port(td::IPAddress::ipv4_to_str(obj.out_ip_), static_cast<td::uint16>(obj.out_port_))
.ensure();
if (obj.proxy_type_) {
auto R = ton::adnl::AdnlProxy::create(*obj.proxy_type_.get());
R.ensure();
proxy = R.move_as_ok();
categories = obj.categories_;
priority_categories = obj.priority_categories_;
}
}));
config_add_network_addr(in_ip, out_ip, std::move(proxy), categories, priority_categories).ensure();
}
for (auto &adnl : config.adnl_) {
config_add_adnl_addr(ton::PublicKeyHash{adnl->id_}, adnl->category_).ensure();
}
for (auto &dht : config.dht_) {
config_add_dht_node(ton::PublicKeyHash{dht->id_}).ensure();
}
for (auto &val : config.validators_) {
auto key = ton::PublicKeyHash{val->id_};
config_add_validator_permanent_key(key, val->election_date_, val->expire_at_).ensure();
for (auto &temp : val->temp_keys_) {
config_add_validator_temp_key(key, ton::PublicKeyHash{temp->key_}, temp->expire_at_).ensure();
}
for (auto &adnl : val->adnl_addrs_) {
config_add_validator_adnl_id(key, ton::PublicKeyHash{adnl->id_}, adnl->expire_at_).ensure();
}
}
config_add_full_node_adnl_id(ton::PublicKeyHash{config.fullnode_}).ensure();
for (auto &s : config.fullnodeslaves_) {
td::IPAddress ip;
ip.init_ipv4_port(td::IPAddress::ipv4_to_str(s->ip_), static_cast<td::uint16>(s->port_)).ensure();
config_add_full_node_slave(ip, ton::PublicKey{s->adnl_}).ensure();
}
for (auto &s : config.fullnodemasters_) {
config_add_full_node_master(s->port_, ton::PublicKeyHash{s->adnl_}).ensure();
}
for (auto &serv : config.liteservers_) {
config_add_lite_server(ton::PublicKeyHash{serv->id_}, serv->port_).ensure();
}
for (auto &serv : config.control_) {
auto key = ton::PublicKeyHash{serv->id_};
config_add_control_interface(key, serv->port_).ensure();
for (auto &proc : serv->allowed_) {
config_add_control_process(key, serv->port_, ton::PublicKeyHash{proc->id_}, proc->permissions_).ensure();
}
}
if (config.gc_) {
for (auto &gc : config.gc_->ids_) {
config_add_gc(ton::PublicKeyHash{gc}).ensure();
}
}
}
ton::tl_object_ptr<ton::ton_api::engine_validator_config> Config::tl() const {
std::vector<ton::tl_object_ptr<ton::ton_api::engine_Addr>> addrs_vec;
for (auto &x : addrs) {
if (x.second.proxy) {
addrs_vec.push_back(ton::create_tl_object<ton::ton_api::engine_addrProxy>(
static_cast<td::int32>(x.second.in_addr.get_ipv4()), x.second.in_addr.get_port(),
static_cast<td::int32>(x.first.addr.get_ipv4()), x.first.addr.get_port(), x.second.proxy->tl(),
std::vector<td::int32>(x.second.cats.begin(), x.second.cats.end()),
std::vector<td::int32>(x.second.priority_cats.begin(), x.second.priority_cats.end())));
} else {
addrs_vec.push_back(ton::create_tl_object<ton::ton_api::engine_addr>(
static_cast<td::int32>(x.first.addr.get_ipv4()), x.first.addr.get_port(),
std::vector<td::int32>(x.second.cats.begin(), x.second.cats.end()),
std::vector<td::int32>(x.second.priority_cats.begin(), x.second.priority_cats.end())));
}
}
std::vector<ton::tl_object_ptr<ton::ton_api::engine_adnl>> adnl_vec;
for (auto &x : adnl_ids) {
adnl_vec.push_back(ton::create_tl_object<ton::ton_api::engine_adnl>(x.first.tl(), x.second));
}
std::vector<ton::tl_object_ptr<ton::ton_api::engine_dht>> dht_vec;
for (auto &x : dht_ids) {
dht_vec.push_back(ton::create_tl_object<ton::ton_api::engine_dht>(x.tl()));
}
std::vector<ton::tl_object_ptr<ton::ton_api::engine_validator>> val_vec;
for (auto &val : validators) {
std::vector<ton::tl_object_ptr<ton::ton_api::engine_validatorTempKey>> temp_vec;
for (auto &t : val.second.temp_keys) {
temp_vec.push_back(ton::create_tl_object<ton::ton_api::engine_validatorTempKey>(t.first.tl(), t.second));
}
std::vector<ton::tl_object_ptr<ton::ton_api::engine_validatorAdnlAddress>> adnl_val_vec;
for (auto &t : val.second.adnl_ids) {
adnl_val_vec.push_back(ton::create_tl_object<ton::ton_api::engine_validatorAdnlAddress>(t.first.tl(), t.second));
}
val_vec.push_back(ton::create_tl_object<ton::ton_api::engine_validator>(
val.first.tl(), std::move(temp_vec), std::move(adnl_val_vec), val.second.election_date, val.second.expire_at));
}
std::vector<ton::tl_object_ptr<ton::ton_api::engine_validator_fullNodeSlave>> full_node_slaves_vec;
for (auto &x : full_node_slaves) {
full_node_slaves_vec.push_back(ton::create_tl_object<ton::ton_api::engine_validator_fullNodeSlave>(
x.addr.get_ipv4(), x.addr.get_port(), x.key.tl()));
}
std::vector<ton::tl_object_ptr<ton::ton_api::engine_validator_fullNodeMaster>> full_node_masters_vec;
for (auto &x : full_node_masters) {
full_node_masters_vec.push_back(
ton::create_tl_object<ton::ton_api::engine_validator_fullNodeMaster>(x.first, x.second.tl()));
}
std::vector<ton::tl_object_ptr<ton::ton_api::engine_liteServer>> liteserver_vec;
for (auto &x : liteservers) {
liteserver_vec.push_back(ton::create_tl_object<ton::ton_api::engine_liteServer>(x.second.tl(), x.first));
}
std::vector<ton::tl_object_ptr<ton::ton_api::engine_controlInterface>> control_vec;
for (auto &x : controls) {
std::vector<ton::tl_object_ptr<ton::ton_api::engine_controlProcess>> control_proc_vec;
for (auto &y : x.second.clients) {
control_proc_vec.push_back(ton::create_tl_object<ton::ton_api::engine_controlProcess>(y.first.tl(), y.second));
}
control_vec.push_back(ton::create_tl_object<ton::ton_api::engine_controlInterface>(x.second.key.tl(), x.first,
std::move(control_proc_vec)));
}
auto gc_vec = ton::create_tl_object<ton::ton_api::engine_gc>(std::vector<td::Bits256>{});
for (auto &id : gc) {
gc_vec->ids_.push_back(id.tl());
}
return ton::create_tl_object<ton::ton_api::engine_validator_config>(
out_port, std::move(addrs_vec), std::move(adnl_vec), std::move(dht_vec), std::move(val_vec), full_node.tl(),
std::move(full_node_slaves_vec), std::move(full_node_masters_vec), std::move(liteserver_vec),
std::move(control_vec), std::move(gc_vec));
}
td::Result<bool> Config::config_add_network_addr(td::IPAddress in_ip, td::IPAddress out_ip,
std::shared_ptr<ton::adnl::AdnlProxy> proxy,
std::vector<AdnlCategory> cats, std::vector<AdnlCategory> prio_cats) {
Addr addr{out_ip};
auto it = addrs.find(addr);
if (it != addrs.end()) {
bool mod = false;
if (!(it->second.in_addr == in_ip)) {
it->second.in_addr = in_ip;
mod = true;
}
if (it->second.proxy != proxy) {
it->second.proxy = std::move(proxy);
mod = true;
}
for (auto &c : cats) {
if (it->second.cats.insert(c).second) {
mod = true;
}
}
for (auto &c : prio_cats) {
if (it->second.priority_cats.insert(c).second) {
mod = true;
}
}
return mod;
} else {
it = addrs.emplace(std::move(addr), AddrCats{}).first;
it->second.in_addr = in_ip;
it->second.proxy = std::move(proxy);
for (auto &c : cats) {
it->second.cats.insert(c);
}
for (auto &c : prio_cats) {
it->second.priority_cats.insert(c);
}
return true;
}
}
td::Result<bool> Config::config_add_adnl_addr(ton::PublicKeyHash addr, AdnlCategory cat) {
auto it = adnl_ids.find(addr);
if (it != adnl_ids.end()) {
if (it->second != cat) {
it->second = cat;
return true;
} else {
return false;
}
} else {
incref(addr);
adnl_ids.emplace(addr, cat);
return true;
}
}
td::Result<bool> Config::config_add_dht_node(ton::PublicKeyHash id) {
if (dht_ids.count(id) > 0) {
return false;
}
if (adnl_ids.count(id) == 0) {
return td::Status::Error(ton::ErrorCode::notready, "to-be-added dht node not in adnl nodes list");
}
incref(id);
dht_ids.insert(id);
return true;
}
td::Result<bool> Config::config_add_validator_permanent_key(ton::PublicKeyHash id, ton::UnixTime election_date,
ton::UnixTime expire_at) {
for (auto &x : validators) {
if (x.second.election_date == election_date && x.first != id) {
return td::Status::Error(ton::ErrorCode::protoviolation, "duplicate election date");
}
}
auto it = validators.find(id);
if (it != validators.end()) {
if (it->second.election_date != election_date) {
return td::Status::Error(ton::ErrorCode::protoviolation, "election date changed");
}
if (it->second.expire_at != expire_at) {
it->second.expire_at = expire_at;
return true;
} else {
return false;
}
} else {
incref(id);
validators[id] = Validator{{}, {}, election_date, expire_at};
return true;
}
}
td::Result<bool> Config::config_add_validator_temp_key(ton::PublicKeyHash perm_key, ton::PublicKeyHash id,
ton::UnixTime expire_at) {
if (validators.count(perm_key) == 0) {
return td::Status::Error(ton::ErrorCode::notready, "unknown permanent validator key");
}
auto &v = validators[perm_key];
auto it = v.temp_keys.find(id);
if (it != v.temp_keys.end()) {
if (it->second != expire_at) {
it->second = expire_at;
return true;
} else {
return false;
}
} else {
incref(id);
v.temp_keys.emplace(id, expire_at);
return true;
}
}
td::Result<bool> Config::config_add_validator_adnl_id(ton::PublicKeyHash perm_key, ton::PublicKeyHash adnl_id,
ton::UnixTime expire_at) {
if (adnl_ids.count(adnl_id) == 0) {
return td::Status::Error(ton::ErrorCode::notready, "to-be-added validator adnl address not in adnl nodes list");
}
if (validators.count(perm_key) == 0) {
return td::Status::Error(ton::ErrorCode::notready, "unknown permanent validator key");
}
auto &v = validators[perm_key];
auto it = v.adnl_ids.find(adnl_id);
if (it != v.adnl_ids.end()) {
if (it->second != expire_at) {
it->second = expire_at;
return true;
} else {
return false;
}
} else {
incref(adnl_id);
v.adnl_ids.emplace(adnl_id, expire_at);
return true;
}
}
td::Result<bool> Config::config_add_full_node_adnl_id(ton::PublicKeyHash id) {
if (full_node == id) {
return false;
}
if (adnl_ids.count(id) == 0) {
return td::Status::Error(ton::ErrorCode::notready, "to-be-added full node adnl address not in adnl nodes list");
}
if (!full_node.is_zero()) {
decref(full_node);
}
if (!id.is_zero()) {
incref(id);
}
full_node = id;
return true;
}
td::Result<bool> Config::config_add_full_node_slave(td::IPAddress addr, ton::PublicKey id) {
for (auto &s : full_node_slaves) {
if (s.addr == addr) {
if (s.key == id) {
return true;
} else {
return td::Status::Error(ton::ErrorCode::error, "duplicate slave ip");
}
}
}
full_node_slaves.push_back(FullNodeSlave{id, addr});
return true;
}
td::Result<bool> Config::config_add_full_node_master(td::int32 port, ton::PublicKeyHash id) {
if (adnl_ids.count(id) == 0) {
return td::Status::Error(ton::ErrorCode::notready,
"to-be-added full node master adnl address not in adnl nodes list");
}
auto it = full_node_masters.find(port);
if (it != full_node_masters.end()) {
if (it->second == id) {
return false;
} else {
return td::Status::Error("duplicate master port");
}
}
if (liteservers.count(port) > 0 || controls.count(port) > 0) {
return td::Status::Error("duplicate master port");
}
incref(id);
full_node_masters.emplace(port, id);
return true;
}
td::Result<bool> Config::config_add_lite_server(ton::PublicKeyHash key, td::int32 port) {
if (controls.count(port) > 0) {
return td::Status::Error(ton::ErrorCode::error, "duplicate port");
}
auto it = liteservers.find(port);
if (it != liteservers.end()) {
if (it->second == key) {
return false;
} else {
return td::Status::Error(ton::ErrorCode::error, "duplicate port");
}
} else {
incref(key);
liteservers.emplace(port, key);
return true;
}
}
td::Result<bool> Config::config_add_control_interface(ton::PublicKeyHash key, td::int32 port) {
if (liteservers.count(port) > 0) {
return td::Status::Error(ton::ErrorCode::error, "duplicate port");
}
auto it = controls.find(port);
if (it != controls.end()) {
if (it->second.key == key) {
return false;
} else {
return td::Status::Error(ton::ErrorCode::error, "duplicate port");
}
} else {
incref(key);
controls.emplace(port, Control{key, {}});
return true;
}
}
td::Result<bool> Config::config_add_control_process(ton::PublicKeyHash key, td::int32 port, ton::PublicKeyHash id,
td::uint32 permissions) {
if (controls.count(port) == 0) {
return td::Status::Error(ton::ErrorCode::error, "unknown control interface");
}
auto &v = controls[port];
if (v.key != key) {
return td::Status::Error(ton::ErrorCode::error, "unknown control interface");
}
auto it = v.clients.find(id);
if (it != v.clients.end()) {
if (!permissions) {
v.clients.erase(id);
return true;
}
if (it->second != permissions) {
it->second = permissions;
return true;
} else {
return false;
}
} else {
if (!permissions) {
return false;
}
v.clients.emplace(id, permissions);
return true;
}
}
td::Result<bool> Config::config_add_gc(ton::PublicKeyHash key) {
return gc.insert(key).second;
}
void Config::decref(ton::PublicKeyHash key) {
auto v = keys_refcnt[key]--;
CHECK(v > 0);
if (v == 1) {
config_add_gc(key).ensure();
}
}
td::Result<bool> Config::config_del_network_addr(td::IPAddress a, std::vector<AdnlCategory> cats,
std::vector<AdnlCategory> prio_cats) {
Addr addr{a};
auto it = addrs.find(addr);
if (it != addrs.end()) {
bool mod = false;
for (auto &c : cats) {
if (it->second.cats.erase(c)) {
mod = true;
}
}
for (auto &c : prio_cats) {
if (it->second.priority_cats.erase(c)) {
mod = true;
}
}
if (it->second.cats.size() == 0 && it->second.priority_cats.size() == 0) {
addrs.erase(it);
}
return mod;
} else {
return false;
}
}
td::Result<bool> Config::config_del_adnl_addr(ton::PublicKeyHash addr) {
if (adnl_ids.count(addr) == 0) {
return false;
}
if (dht_ids.count(addr)) {
return td::Status::Error(ton::ErrorCode::error, "adnl addr still in use");
}
if (full_node == addr) {
return td::Status::Error(ton::ErrorCode::error, "adnl addr still in use");
}
for (auto &x : validators) {
if (x.second.adnl_ids.count(addr)) {
return td::Status::Error(ton::ErrorCode::error, "adnl addr still in use");
}
}
decref(addr);
adnl_ids.erase(addr);
return true;
}
td::Result<bool> Config::config_del_dht_node(ton::PublicKeyHash id) {
if (dht_ids.count(id) == 0) {
return false;
}
decref(id);
dht_ids.erase(id);
return true;
}
td::Result<bool> Config::config_del_validator_permanent_key(ton::PublicKeyHash id) {
if (validators.count(id) == 0) {
return false;
}
auto &v = validators[id];
for (auto &temp_key : v.temp_keys) {
decref(temp_key.first);
}
for (auto &adnl_id : v.adnl_ids) {
decref(adnl_id.first);
}
decref(id);
validators.erase(id);
return true;
}
td::Result<bool> Config::config_del_validator_temp_key(ton::PublicKeyHash perm_key, ton::PublicKeyHash id) {
if (validators.count(perm_key) == 0) {
return td::Status::Error(ton::ErrorCode::notready, "unknown permanent validator key");
}
auto &v = validators[perm_key];
auto it = v.temp_keys.find(id);
if (it != v.temp_keys.end()) {
decref(id);
v.temp_keys.erase(id);
return true;
} else {
return false;
}
}
td::Result<bool> Config::config_del_validator_adnl_id(ton::PublicKeyHash perm_key, ton::PublicKeyHash adnl_id) {
if (validators.count(perm_key) == 0) {
return td::Status::Error(ton::ErrorCode::notready, "unknown permanent validator key");
}
auto &v = validators[perm_key];
auto it = v.adnl_ids.find(adnl_id);
if (it != v.temp_keys.end()) {
decref(adnl_id);
v.adnl_ids.erase(adnl_id);
return true;
} else {
return false;
}
}
td::Result<bool> Config::config_del_full_node_adnl_id() {
return config_add_full_node_adnl_id(ton::PublicKeyHash::zero());
}
td::Result<bool> Config::config_del_lite_server(td::int32 port) {
auto it = liteservers.find(port);
if (it != liteservers.end()) {
decref(it->second);
liteservers.erase(it);
return true;
} else {
return false;
}
}
td::Result<bool> Config::config_del_control_interface(td::int32 port) {
auto it = controls.find(port);
if (it != controls.end()) {
decref(it->second.key);
controls.erase(it);
return true;
} else {
return false;
}
}
td::Result<bool> Config::config_del_control_process(td::int32 port, ton::PublicKeyHash id) {
auto it = controls.find(port);
if (it != controls.end()) {
return it->second.clients.erase(id);
} else {
return false;
}
}
td::Result<bool> Config::config_del_gc(ton::PublicKeyHash key) {
return gc.erase(key);
}
class ValidatorElectionBidCreator : public td::actor::Actor {
public:
ValidatorElectionBidCreator(td::uint32 date, std::string addr, std::string wallet, std::string dir,
td::actor::ActorId<ValidatorEngine> engine,
td::actor::ActorId<ton::keyring::Keyring> keyring, td::Promise<td::BufferSlice> promise)
: date_(date)
, addr_(addr)
, wallet_(wallet)
, dir_(dir)
, engine_(engine)
, keyring_(keyring)
, promise_(std::move(promise)) {
ttl_ = date_ + 7 * 86400;
}
void start_up() override {
auto pk1 = ton::PrivateKey{ton::privkeys::Ed25519::random()};
perm_key_full_ = pk1.compute_public_key();
perm_key_ = perm_key_full_.compute_short_id();
auto pk2 = ton::PrivateKey{ton::privkeys::Ed25519::random()};
adnl_key_full_ = ton::adnl::AdnlNodeIdFull{pk2.compute_public_key()};
adnl_addr_ = adnl_key_full_.compute_short_id();
td::MultiPromise mp;
auto P = td::PromiseCreator::lambda([SelfId = actor_id(this)](td::Result<td::Unit> R) {
if (R.is_error()) {
td::actor::send_closure(SelfId, &ValidatorElectionBidCreator::abort_query,
R.move_as_error_prefix("keyring fail: "));
} else {
td::actor::send_closure(SelfId, &ValidatorElectionBidCreator::written_keys);
}
});
auto ig = mp.init_guard();
ig.add_promise(std::move(P));
td::actor::send_closure(keyring_, &ton::keyring::Keyring::add_key, std::move(pk1), false, ig.get_promise());
td::actor::send_closure(keyring_, &ton::keyring::Keyring::add_key, std::move(pk2), false, ig.get_promise());
}
void written_keys() {
td::MultiPromise mp;
auto P = td::PromiseCreator::lambda([SelfId = actor_id(this)](td::Result<td::Unit> R) {
if (R.is_error()) {
td::actor::send_closure(SelfId, &ValidatorElectionBidCreator::abort_query,
R.move_as_error_prefix("update config fail: "));
} else {
td::actor::send_closure(SelfId, &ValidatorElectionBidCreator::updated_config);
}
});
auto ig = mp.init_guard();
ig.add_promise(std::move(P));
td::actor::send_closure(engine_, &ValidatorEngine::add_key_to_set, perm_key_full_);
td::actor::send_closure(engine_, &ValidatorEngine::add_key_to_set, adnl_key_full_.pubkey());
td::actor::send_closure(engine_, &ValidatorEngine::try_add_validator_permanent_key, perm_key_, date_, ttl_,
ig.get_promise());
td::actor::send_closure(engine_, &ValidatorEngine::try_add_validator_temp_key, perm_key_, perm_key_, ttl_,
ig.get_promise());
td::actor::send_closure(engine_, &ValidatorEngine::try_add_adnl_node, adnl_addr_.pubkey_hash(), cat_,
ig.get_promise());
td::actor::send_closure(engine_, &ValidatorEngine::try_add_validator_adnl_addr, perm_key_, adnl_addr_.pubkey_hash(),
ttl_, ig.get_promise());
}
void updated_config() {
auto codeR = td::read_file_str(dir_ + "/validator-elect-req.fif");
if (codeR.is_error()) {
abort_query(codeR.move_as_error_prefix("fif not found (validator-elect-req.fif)"));
return;
}
auto R = fift::mem_run_fift(codeR.move_as_ok(),
{"validator-elect-req.fif", wallet_, td::to_string(date_), td::to_string(frac),
adnl_addr_.bits256_value().to_hex(), "OUTPUT"},
dir_ + "/");
if (R.is_error()) {
abort_query(R.move_as_error_prefix("fift fail (validator-elect-req.fif)"));
return;
}
auto P = td::PromiseCreator::lambda([SelfId = actor_id(this)](td::Result<td::BufferSlice> R) {
if (R.is_error()) {
td::actor::send_closure(SelfId, &ValidatorElectionBidCreator::abort_query,
R.move_as_error_prefix("sign fail: "));
} else {
td::actor::send_closure(SelfId, &ValidatorElectionBidCreator::signed_bid, R.move_as_ok());
}
});
auto res = R.move_as_ok();
auto to_signR = res.source_lookup.read_file("OUTPUT");
if (to_signR.is_error()) {
abort_query(td::Status::Error(PSTRING() << "strange error: no to sign file. Output: " << res.output));
return;
}
td::actor::send_closure(keyring_, &ton::keyring::Keyring::sign_message, perm_key_,
td::BufferSlice{to_signR.move_as_ok().data}, std::move(P));
}
void signed_bid(td::BufferSlice signature) {
signature_ = std::move(signature);
auto codeR = td::read_file_str(dir_ + "/validator-elect-signed.fif");
if (codeR.is_error()) {
abort_query(codeR.move_as_error_prefix("fif not found (validator-elect-req.fif)"));
return;
}
auto R = fift::mem_run_fift(
codeR.move_as_ok(),
{"validator-elect-signed.fif", wallet_, td::to_string(date_), td::to_string(frac),
adnl_addr_.bits256_value().to_hex(), td::base64_encode(perm_key_full_.export_as_slice().as_slice()),
td::base64_encode(signature_.as_slice()), "OUTPUT"},
dir_ + "/");
if (R.is_error()) {
abort_query(R.move_as_error_prefix("fift fail (validator-elect-req.fif)"));
return;
}
auto res = R.move_as_ok();
auto dataR = res.source_lookup.read_file("OUTPUT");
if (dataR.is_error()) {
abort_query(td::Status::Error("strage error: no result boc"));
return;
}
result_ = td::BufferSlice(dataR.move_as_ok().data);
finish_query();
}
void abort_query(td::Status error) {
promise_.set_value(ValidatorEngine::create_control_query_error(std::move(error)));
stop();
}
void finish_query() {
promise_.set_value(ton::create_serialize_tl_object<ton::ton_api::engine_validator_electionBid>(
date_, perm_key_.tl(), adnl_addr_.bits256_value(), std::move(result_)));
stop();
}
private:
td::uint32 date_;
std::string addr_;
std::string wallet_;
std::string dir_;
td::actor::ActorId<ValidatorEngine> engine_;
td::actor::ActorId<ton::keyring::Keyring> keyring_;
td::Promise<td::BufferSlice> promise_;
td::uint32 ttl_;
AdnlCategory cat_ = 2;
double frac = 2.7;
ton::PublicKeyHash perm_key_;
ton::PublicKey perm_key_full_;
ton::adnl::AdnlNodeIdShort adnl_addr_;
ton::adnl::AdnlNodeIdFull adnl_key_full_;
td::BufferSlice signature_;
td::BufferSlice result_;
};
class CheckDhtServerStatusQuery : public td::actor::Actor {
public:
void start_up() override {
auto &n = dht_config_->nodes();
result_.resize(n.size(), false);
pending_ = n.size();
for (td::uint32 i = 0; i < n.size(); i++) {
auto P = td::PromiseCreator::lambda([SelfId = actor_id(this), idx = i](td::Result<td::BufferSlice> R) {
td::actor::send_closure(SelfId, &CheckDhtServerStatusQuery::got_result, idx, R.is_ok());
});
auto &E = n.list().at(i);
td::actor::send_closure(adnl_, &ton::adnl::Adnl::add_peer, local_id_, E.adnl_id(), E.addr_list());
td::actor::send_closure(adnl_, &ton::adnl::Adnl::send_query, local_id_, E.adnl_id().compute_short_id(), "ping",
std::move(P), td::Timestamp::in(1.0),
ton::create_serialize_tl_object<ton::ton_api::dht_getSignedAddressList>());
}
}
void got_result(td::uint32 idx, bool result) {
result_[idx] = result;
CHECK(pending_ > 0);
if (!--pending_) {
finish_query();
}
}
void finish_query() {
std::vector<ton::tl_object_ptr<ton::ton_api::engine_validator_dhtServerStatus>> vec;
auto &n = dht_config_->nodes();
for (td::uint32 i = 0; i < n.size(); i++) {
auto &E = n.list().at(i);
vec.push_back(ton::create_tl_object<ton::ton_api::engine_validator_dhtServerStatus>(
E.adnl_id().compute_short_id().bits256_value(), result_[i] ? 1 : 0));
}
promise_.set_value(
ton::create_serialize_tl_object<ton::ton_api::engine_validator_dhtServersStatus>(std::move(vec)));
stop();
}
CheckDhtServerStatusQuery(std::shared_ptr<ton::dht::DhtGlobalConfig> dht_config, ton::adnl::AdnlNodeIdShort local_id,
td::actor::ActorId<ton::adnl::Adnl> adnl, td::Promise<td::BufferSlice> promise)
: dht_config_(std::move(dht_config)), local_id_(local_id), adnl_(adnl), promise_(std::move(promise)) {
}
private:
std::shared_ptr<ton::dht::DhtGlobalConfig> dht_config_;
std::vector<bool> result_;
td::uint32 pending_;
ton::adnl::AdnlNodeIdShort local_id_;
td::actor::ActorId<ton::adnl::Adnl> adnl_;
td::Promise<td::BufferSlice> promise_;
};
void ValidatorEngine::set_local_config(std::string str) {
local_config_ = str;
}
void ValidatorEngine::set_global_config(std::string str) {
global_config_ = str;
}
void ValidatorEngine::set_db_root(std::string db_root) {
db_root_ = db_root;
}
void ValidatorEngine::start_up() {
alarm_timestamp() = td::Timestamp::in(1.0 + td::Random::fast(0, 100) * 0.01);
}
void ValidatorEngine::alarm() {
alarm_timestamp() = td::Timestamp::in(1.0 + td::Random::fast(0, 100) * 0.01);
if (started_) {
if (!validator_manager_.empty()) {
auto P = td::PromiseCreator::lambda(
[SelfId = actor_id(this)](td::Result<td::Ref<ton::validator::MasterchainState>> R) {
if (R.is_ok()) {
td::actor::send_closure(SelfId, &ValidatorEngine::got_state, R.move_as_ok());
}
});
td::actor::send_closure(validator_manager_, &ton::validator::ValidatorManagerInterface::get_top_masterchain_state,
std::move(P));
}
if (state_.not_null()) {
bool need_write = false;
auto val_set = state_->get_total_validator_set(0);
auto e = val_set->export_vector();
std::set<ton::PublicKeyHash> adnl_ids;
for (auto &el : e) {
adnl_ids.insert(ton::PublicKeyHash{el.addr});
}
std::set<ton::PublicKeyHash> to_del;
for (auto &val : config_.validators) {
if (val.second.expire_at < state_->get_unix_time() &&
!val_set->is_validator(ton::NodeIdShort{val.first.bits256_value()})) {
to_del.insert(val.first);
} else {
std::set<ton::PublicKeyHash> to_del_2;
for (auto &x : val.second.adnl_ids) {
if (x.second < state_->get_unix_time() && !adnl_ids.count(x.first)) {
to_del_2.insert(x.first);
}
}
for (auto &x : to_del_2) {
config_.config_del_validator_temp_key(val.first, x);
need_write = true;
}
}
}
for (auto &x : to_del) {
config_.config_del_validator_permanent_key(x);
if (!validator_manager_.empty()) {
td::actor::send_closure(validator_manager_, &ton::validator::ValidatorManagerInterface::del_permanent_key, x,
[](td::Unit) {});
}
if (!full_node_.empty()) {
td::actor::send_closure(full_node_, &ton::validator::fullnode::FullNode::del_permanent_key, x,
[](td::Unit) {});
}
need_write = true;
}
if (need_write) {
write_config([](td::Unit) {});
}
}
for (auto &x : config_.gc) {
if (running_gc_.count(x) == 0) {
running_gc_.insert(x);
keys_.erase(x);
auto P = td::PromiseCreator::lambda([SelfId = actor_id(this), x](td::Result<td::Unit> R) {
R.ensure();
td::actor::send_closure(SelfId, &ValidatorEngine::deleted_key, x);
});
td::actor::send_closure(keyring_, &ton::keyring::Keyring::del_key, x, std::move(P));
}
}
}
}
void ValidatorEngine::deleted_key(ton::PublicKeyHash x) {
CHECK(running_gc_.count(x) == 1);
running_gc_.erase(x);
auto R = config_.config_del_gc(x);
R.ensure();
if (R.move_as_ok()) {
write_config([](td::Unit) {});
}
}
td::Status ValidatorEngine::load_global_config() {
TRY_RESULT_PREFIX(conf_data, td::read_file(global_config_), "failed to read: ");
TRY_RESULT_PREFIX(conf_json, td::json_decode(conf_data.as_slice()), "failed to parse json: ");
ton::ton_api::config_global conf;
TRY_STATUS_PREFIX(ton::ton_api::from_json(conf, conf_json.get_object()), "json does not fit TL scheme: ");
// TODO
// add adnl static nodes
if (conf.adnl_) {
if (conf.adnl_->static_nodes_) {
TRY_RESULT_PREFIX_ASSIGN(adnl_static_nodes_, ton::adnl::AdnlNodesList::create(conf.adnl_->static_nodes_),
"bad static adnl nodes: ");
}
}
if (!conf.dht_) {
return td::Status::Error(ton::ErrorCode::error, "does not contain [dht] section");
}
TRY_RESULT_PREFIX(dht, ton::dht::Dht::create_global_config(std::move(conf.dht_)), "bad [dht] section: ");
dht_config_ = std::move(dht);
if (!conf.validator_) {
return td::Status::Error(ton::ErrorCode::error, "does not contain [validator] section");
}
if (!conf.validator_->zero_state_) {
return td::Status::Error(ton::ErrorCode::error, "[validator] section does not contain [zero_state]");
}
auto zero_state = ton::create_block_id(conf.validator_->zero_state_);
if (zero_state.id.workchain != ton::masterchainId || zero_state.id.shard != ton::shardIdAll ||
zero_state.id.seqno != 0) {
return td::Status::Error(ton::ErrorCode::error, "[validator] section contains invalid [zero_state]");
}
if (zero_state.root_hash.is_zero() || zero_state.file_hash.is_zero()) {
return td::Status::Error(ton::ErrorCode::error, "[validator] section contains incomplete [zero_state]");
}
ton::BlockIdExt init_block;
if (!conf.validator_->init_block_) {
init_block = zero_state;
} else {
init_block = ton::create_block_id(conf.validator_->init_block_);
if (init_block.id.workchain != ton::masterchainId || init_block.id.shard != ton::shardIdAll) {
return td::Status::Error(ton::ErrorCode::error, "[validator] section contains invalid [init_block]");
}
if (init_block.root_hash.is_zero() || init_block.file_hash.is_zero()) {
return td::Status::Error(ton::ErrorCode::error, "[validator] section contains incomplete [init_block]");
}
}
validator_options_ = ton::validator::ValidatorManagerOptions::create(zero_state, init_block);
if (state_ttl_ != 0) {
validator_options_.write().set_state_ttl(state_ttl_);
}
if (block_ttl_ != 0) {
validator_options_.write().set_block_ttl(block_ttl_);
}
if (sync_ttl_ != 0) {
validator_options_.write().set_sync_blocks_before(sync_ttl_);
}
if (archive_ttl_ != 0) {
validator_options_.write().set_archive_ttl(archive_ttl_);
}
if (key_proof_ttl_ != 0) {
validator_options_.write().set_key_proof_ttl(key_proof_ttl_);
}
if (db_depth_ <= 32) {
validator_options_.write().set_filedb_depth(db_depth_);
}
std::vector<ton::BlockIdExt> h;
for (auto &x : conf.validator_->hardforks_) {
auto b = ton::create_block_id(x);
if (!b.is_masterchain()) {
return td::Status::Error(ton::ErrorCode::error,
"[validator/hardforks] section contains not masterchain block id");
}
if (!b.is_valid_full()) {
return td::Status::Error(ton::ErrorCode::error, "[validator/hardforks] section contains invalid block_id");
}
for (auto &y : h) {
if (y.is_valid() && y.seqno() >= b.seqno()) {
y.invalidate();
}
}
h.push_back(b);
}
validator_options_.write().set_hardforks(std::move(h));
return td::Status::OK();
}
void ValidatorEngine::load_empty_local_config(td::Promise<td::Unit> promise) {
auto ret_promise = td::PromiseCreator::lambda(
[SelfId = actor_id(this), promise = std::move(promise)](td::Result<td::Unit> R) mutable {
if (R.is_error()) {
promise.set_error(R.move_as_error());
} else {
td::actor::send_closure(SelfId, &ValidatorEngine::write_config, std::move(promise));
}
});
td::MultiPromise mp;
auto ig = mp.init_guard();
ig.add_promise(std::move(ret_promise));
for (auto &addr : addrs_) {
config_.config_add_network_addr(addr, addr, nullptr, std::vector<td::int32>{0, 1, 2, 3}, std::vector<td::int32>{})
.ensure();
}
{
auto pk = ton::PrivateKey{ton::privkeys::Ed25519::random()};
keys_.emplace(pk.compute_short_id(), pk.compute_public_key());
auto id = pk.compute_short_id();
td::actor::send_closure(keyring_, &ton::keyring::Keyring::add_key, std::move(pk), false, ig.get_promise());
config_.config_add_adnl_addr(id, 0).ensure();
config_.config_add_dht_node(id).ensure();
}
{
auto adnl_pk = ton::PrivateKey{ton::privkeys::Ed25519::random()};
keys_.emplace(adnl_pk.compute_short_id(), adnl_pk.compute_public_key());
auto adnl_short_id = adnl_pk.compute_short_id();
td::actor::send_closure(keyring_, &ton::keyring::Keyring::add_key, std::move(adnl_pk), false, ig.get_promise());
config_.config_add_adnl_addr(adnl_short_id, 1).ensure();
config_.config_add_full_node_adnl_id(adnl_short_id).ensure();
}
}
void ValidatorEngine::load_local_config(td::Promise<td::Unit> promise) {
if (local_config_.size() == 0) {
load_empty_local_config(std::move(promise));
return;
}
auto conf_data_R = td::read_file(local_config_);
if (conf_data_R.is_error()) {
promise.set_error(conf_data_R.move_as_error_prefix("failed to read: "));
return;
}
auto conf_data = conf_data_R.move_as_ok();
auto conf_json_R = td::json_decode(conf_data.as_slice());
if (conf_json_R.is_error()) {
promise.set_error(conf_data_R.move_as_error_prefix("failed to parse json: "));
return;
}
auto conf_json = conf_json_R.move_as_ok();
ton::ton_api::config_local conf;
auto S = ton::ton_api::from_json(conf, conf_json.get_object());
if (S.is_error()) {
promise.set_error(S.move_as_error_prefix("json does not fit TL scheme"));
return;
}
auto ret_promise = td::PromiseCreator::lambda(
[SelfId = actor_id(this), promise = std::move(promise)](td::Result<td::Unit> R) mutable {
if (R.is_error()) {
promise.set_error(R.move_as_error());
} else {
td::actor::send_closure(SelfId, &ValidatorEngine::write_config, std::move(promise));
}
});
td::MultiPromise mp;
auto ig = mp.init_guard();
ig.add_promise(std::move(ret_promise));
for (auto &addr : addrs_) {
config_.config_add_network_addr(addr, addr, nullptr, std::vector<td::int32>{0, 1, 2, 3}, std::vector<td::int32>{})
.ensure();
}
for (auto &local_id : conf.local_ids_) {
ton::PrivateKey pk{local_id->id_};
keys_.emplace(pk.compute_short_id(), pk.compute_public_key());
td::actor::send_closure(keyring_, &ton::keyring::Keyring::add_key, std::move(pk), false, ig.get_promise());
}
td::uint32 max_time = 2000000000;
if (conf.dht_.size() > 0) {
for (auto &d : conf.dht_) {
ton::ton_api::downcast_call(*d.get(), td::overloaded(
[&](ton::ton_api::dht_config_local &obj) {
auto node_id = ton::adnl::AdnlNodeIdShort{obj.id_->id_};
auto it = keys_.find(node_id.pubkey_hash());
if (it == keys_.end()) {
ig.get_promise().set_error(td::Status::Error(
ton::ErrorCode::error, "cannot find private key for dht"));
return;
}
config_.config_add_adnl_addr(node_id.pubkey_hash(), 0).ensure();
config_.config_add_dht_node(node_id.pubkey_hash()).ensure();
},
[&](ton::ton_api::dht_config_random_local &obj) {
for (td::int32 i = 0; i < obj.cnt_; i++) {
auto pk = ton::PrivateKey{ton::privkeys::Ed25519::random()};
keys_.emplace(pk.compute_short_id(), pk.compute_public_key());
auto id = pk.compute_short_id();
td::actor::send_closure(keyring_, &ton::keyring::Keyring::add_key,
std::move(pk), false, ig.get_promise());
config_.config_add_adnl_addr(id, 0).ensure();
config_.config_add_dht_node(id).ensure();
}
}));
}
} else {
auto pk = ton::PrivateKey{ton::privkeys::Ed25519::random()};
keys_.emplace(pk.compute_short_id(), pk.compute_public_key());
auto id = pk.compute_short_id();
td::actor::send_closure(keyring_, &ton::keyring::Keyring::add_key, std::move(pk), false, ig.get_promise());
config_.config_add_adnl_addr(id, 0).ensure();
config_.config_add_dht_node(id).ensure();
}
if (conf.validators_.size() > 0) {
for (auto &val : conf.validators_) {
ton::ton_api::downcast_call(
*val.get(), td::overloaded(
[&](ton::ton_api::validator_config_local &obj) {
auto id = ton::PublicKeyHash{obj.id_->id_};
auto it = keys_.find(id);
if (it == keys_.end()) {
ig.get_promise().set_error(
td::Status::Error(ton::ErrorCode::error, "cannot find private key for dht"));
return;
}
config_.config_add_adnl_addr(id, 2).ensure();
config_.config_add_validator_permanent_key(id, 0, max_time).ensure();
config_.config_add_validator_temp_key(id, id, max_time).ensure();
config_.config_add_validator_adnl_id(id, id, max_time).ensure();
},
[&](ton::ton_api::validator_config_random_local &obj) {
auto pk = ton::PrivateKey{ton::privkeys::Ed25519::random()};
keys_.emplace(pk.compute_short_id(), pk.compute_public_key());
auto id = pk.compute_short_id();
td::actor::send_closure(keyring_, &ton::keyring::Keyring::add_key, std::move(pk), false,
ig.get_promise());
config_.config_add_adnl_addr(id, 2).ensure();
config_.config_add_validator_permanent_key(id, 0, max_time).ensure();
config_.config_add_validator_temp_key(id, id, max_time).ensure();
config_.config_add_validator_adnl_id(id, id, max_time).ensure();
}));
}
} else {
// DO NOTHING
}
{
auto adnl_pk = ton::PrivateKey{ton::privkeys::Ed25519::random()};
keys_.emplace(adnl_pk.compute_short_id(), adnl_pk.compute_public_key());
auto adnl_short_id = adnl_pk.compute_short_id();
td::actor::send_closure(keyring_, &ton::keyring::Keyring::add_key, std::move(adnl_pk), false, ig.get_promise());
config_.config_add_adnl_addr(adnl_short_id, 1).ensure();
config_.config_add_full_node_adnl_id(adnl_short_id).ensure();
}
for (auto &ls : conf.liteservers_) {
ton::ton_api::downcast_call(*ls.get(), td::overloaded(
[&](ton::ton_api::liteserver_config_local &cfg) {
ton::PrivateKey pk{cfg.id_};
keys_.emplace(pk.compute_short_id(), pk.compute_public_key());
auto short_id = pk.compute_short_id();
td::actor::send_closure(keyring_, &ton::keyring::Keyring::add_key,
std::move(pk), false, ig.get_promise());
config_.config_add_lite_server(short_id, cfg.port_).ensure();
},
[&](ton::ton_api::liteserver_config_random_local &cfg) {
auto pk = ton::PrivateKey{ton::privkeys::Ed25519::random()};
auto short_id = pk.compute_short_id();
td::actor::send_closure(keyring_, &ton::keyring::Keyring::add_key,
std::move(pk), false, ig.get_promise());
config_.config_add_lite_server(short_id, cfg.port_).ensure();
}));
}
for (auto &ci : conf.control_) {
ton::PrivateKey pk{ci->priv_};
keys_.emplace(pk.compute_short_id(), pk.compute_public_key());
auto short_id = pk.compute_short_id();
td::actor::send_closure(keyring_, &ton::keyring::Keyring::add_key, std::move(pk), false, ig.get_promise());
config_.config_add_control_interface(short_id, ci->port_).ensure();
config_.config_add_control_process(short_id, ci->port_, ton::PublicKeyHash{ci->pub_}, 0x7fffffff).ensure();
}
}
void ValidatorEngine::load_config(td::Promise<td::Unit> promise) {
if (!config_file_.size()) {
config_file_ = db_root_ + "/config.json";
}
auto conf_data_R = td::read_file(config_file_);
if (conf_data_R.is_error()) {
auto P = td::PromiseCreator::lambda(
[name = local_config_, new_name = config_file_, promise = std::move(promise)](td::Result<td::Unit> R) {
if (R.is_error()) {
LOG(ERROR) << "failed to parse local config '" << name << "': " << R.move_as_error();
std::_Exit(2);
} else {
LOG(ERROR) << "created config file '" << new_name << "'";
LOG(ERROR) << "check it manually before continue";
std::_Exit(0);
}
});
load_local_config(std::move(P));
return;
}
auto conf_data = conf_data_R.move_as_ok();
auto conf_json_R = td::json_decode(conf_data.as_slice());
if (conf_json_R.is_error()) {
promise.set_error(conf_json_R.move_as_error_prefix("failed to parse json: "));
return;
}
auto conf_json = conf_json_R.move_as_ok();
ton::ton_api::engine_validator_config conf;
auto S = ton::ton_api::from_json(conf, conf_json.get_object());
if (S.is_error()) {
promise.set_error(S.move_as_error_prefix("json does not fit TL scheme"));
return;
}
config_ = Config{conf};
td::MultiPromise mp;
auto ig = mp.init_guard();
ig.add_promise(std::move(promise));
for (auto &key : config_.keys_refcnt) {
td::actor::send_closure(keyring_, &ton::keyring::Keyring::add_key_short, key.first, get_key_promise(ig));
}
write_config(ig.get_promise());
}
void ValidatorEngine::write_config(td::Promise<td::Unit> promise) {
auto s = td::json_encode<std::string>(td::ToJson(*config_.tl().get()), true);
auto S = td::write_file(config_file_, s);
if (S.is_ok()) {
promise.set_value(td::Unit());
} else {
promise.set_error(std::move(S));
}
}
td::Promise<ton::PublicKey> ValidatorEngine::get_key_promise(td::MultiPromise::InitGuard &ig) {
auto P = td::PromiseCreator::lambda(
[SelfId = actor_id(this), promise = ig.get_promise()](td::Result<ton::PublicKey> R) mutable {
if (R.is_error()) {
promise.set_error(R.move_as_error());
} else {
td::actor::send_closure(SelfId, &ValidatorEngine::got_key, R.move_as_ok());
promise.set_value(td::Unit());
}
});
return std::move(P);
}
void ValidatorEngine::got_key(ton::PublicKey key) {
keys_[key.compute_short_id()] = key;
}
void ValidatorEngine::start() {
read_config_ = true;
start_adnl();
}
void ValidatorEngine::start_adnl() {
adnl_network_manager_ = ton::adnl::AdnlNetworkManager::create(config_.out_port);
adnl_ = ton::adnl::Adnl::create(db_root_, keyring_.get());
td::actor::send_closure(adnl_, &ton::adnl::Adnl::register_network_manager, adnl_network_manager_.get());
for (auto &addr : config_.addrs) {
add_addr(addr.first, addr.second);
}
for (auto &adnl : config_.adnl_ids) {
add_adnl(adnl.first, adnl.second);
}
td::actor::send_closure(adnl_, &ton::adnl::Adnl::add_static_nodes_from_config, std::move(adnl_static_nodes_));
started_adnl();
}
void ValidatorEngine::add_addr(const Config::Addr &addr, const Config::AddrCats &cats) {
if (!cats.proxy) {
td::actor::send_closure(adnl_network_manager_, &ton::adnl::AdnlNetworkManager::add_self_addr, addr.addr,
cats.cats.size() ? 0 : 1);
} else {
td::actor::send_closure(adnl_network_manager_, &ton::adnl::AdnlNetworkManager::add_proxy_addr, addr.addr,
cats.proxy, cats.cats.size() ? 0 : 1);
}
td::uint32 ts = static_cast<td::uint32>(td::Clocks::system());
for (auto cat : cats.cats) {
CHECK(cat >= 0);
ton::adnl::AdnlAddress x = ton::adnl::AdnlAddressImpl::create(
ton::create_tl_object<ton::ton_api::adnl_address_udp>(cats.in_addr.get_ipv4(), cats.in_addr.get_port()));
addr_lists_[cat].add_addr(std::move(x));
addr_lists_[cat].set_version(ts);
addr_lists_[cat].set_reinit_date(ton::adnl::Adnl::adnl_start_time());
}
for (auto cat : cats.priority_cats) {
CHECK(cat >= 0);
ton::adnl::AdnlAddress x = ton::adnl::AdnlAddressImpl::create(
ton::create_tl_object<ton::ton_api::adnl_address_udp>(cats.in_addr.get_ipv4(), cats.in_addr.get_port()));
prio_addr_lists_[cat].add_addr(std::move(x));
prio_addr_lists_[cat].set_version(ts);
prio_addr_lists_[cat].set_reinit_date(ton::adnl::Adnl::adnl_start_time());
}
}
void ValidatorEngine::add_adnl(ton::PublicKeyHash id, AdnlCategory cat) {
CHECK(keys_.count(id) > 0);
td::actor::send_closure(adnl_, &ton::adnl::Adnl::add_id, ton::adnl::AdnlNodeIdFull{keys_[id]}, addr_lists_[cat]);
}
void ValidatorEngine::started_adnl() {
start_dht();
}
void ValidatorEngine::add_dht(ton::PublicKeyHash id) {
auto D = ton::dht::Dht::create(ton::adnl::AdnlNodeIdShort{id}, db_root_, dht_config_, keyring_.get(), adnl_.get());
D.ensure();
dht_nodes_[id] = D.move_as_ok();
if (default_dht_node_.is_zero()) {
default_dht_node_ = id;
}
}
void ValidatorEngine::start_dht() {
for (auto &dht : config_.dht_ids) {
add_dht(dht);
}
if (default_dht_node_.is_zero()) {
LOG(ERROR) << "trying to work without DHT";
} else {
td::actor::send_closure(adnl_, &ton::adnl::Adnl::register_dht_node, dht_nodes_[default_dht_node_].get());
}
started_dht();
}
void ValidatorEngine::started_dht() {
start_rldp();
}
void ValidatorEngine::start_rldp() {
rldp_ = ton::rldp::Rldp::create(adnl_.get());
started_rldp();
}
void ValidatorEngine::started_rldp() {
start_overlays();
}
void ValidatorEngine::start_overlays() {
if (!default_dht_node_.is_zero()) {
overlay_manager_ =
ton::overlay::Overlays::create(db_root_, keyring_.get(), adnl_.get(), dht_nodes_[default_dht_node_].get());
}
started_overlays();
}
void ValidatorEngine::started_overlays() {
start_validator();
}
void ValidatorEngine::start_validator() {
validator_options_.write().set_allow_blockchain_init(config_.validators.size() > 0);
validator_manager_ = ton::validator::ValidatorManagerFactory::create(
validator_options_, db_root_, keyring_.get(), adnl_.get(), rldp_.get(), overlay_manager_.get());
for (auto &v : config_.validators) {
td::actor::send_closure(validator_manager_, &ton::validator::ValidatorManagerInterface::add_permanent_key, v.first,
[](td::Unit) {});
for (auto &t : v.second.temp_keys) {
td::actor::send_closure(validator_manager_, &ton::validator::ValidatorManagerInterface::add_temp_key, t.first,
[](td::Unit) {});
}
}
started_validator();
}
void ValidatorEngine::started_validator() {
start_full_node();
}
void ValidatorEngine::start_full_node() {
if (!config_.full_node.is_zero() || config_.full_node_slaves.size() > 0) {
auto pk = ton::PrivateKey{ton::privkeys::Ed25519::random()};
auto short_id = pk.compute_short_id();
td::actor::send_closure(keyring_, &ton::keyring::Keyring::add_key, std::move(pk), true, [](td::Unit) {});
if (config_.full_node_slaves.size() > 0) {
std::vector<std::pair<ton::adnl::AdnlNodeIdFull, td::IPAddress>> vec;
for (auto &x : config_.full_node_slaves) {
vec.emplace_back(ton::adnl::AdnlNodeIdFull{x.key}, x.addr);
}
class Cb : public ton::adnl::AdnlExtClient::Callback {
public:
void on_ready() override {
}
void on_stop_ready() override {
}
};
full_node_client_ = ton::adnl::AdnlExtMultiClient::create(std::move(vec), std::make_unique<Cb>());
}
full_node_ = ton::validator::fullnode::FullNode::create(
short_id, ton::adnl::AdnlNodeIdShort{config_.full_node}, validator_options_->zero_block_id().file_hash,
keyring_.get(), adnl_.get(), rldp_.get(),
default_dht_node_.is_zero() ? td::actor::ActorId<ton::dht::Dht>{} : dht_nodes_[default_dht_node_].get(),
overlay_manager_.get(), validator_manager_.get(), full_node_client_.get(), db_root_);
}
for (auto &v : config_.validators) {
td::actor::send_closure(full_node_, &ton::validator::fullnode::FullNode::add_permanent_key, v.first,
[](td::Unit) {});
}
started_full_node();
}
void ValidatorEngine::started_full_node() {
start_lite_server();
}
void ValidatorEngine::add_lite_server(ton::PublicKeyHash id, td::uint16 port) {
td::actor::send_closure(adnl_, &ton::adnl::Adnl::add_id, ton::adnl::AdnlNodeIdFull{keys_[id]},
ton::adnl::AdnlAddressList{});
td::actor::send_closure(validator_manager_, &ton::validator::ValidatorManagerInterface::add_ext_server_id,
ton::adnl::AdnlNodeIdShort{id});
td::actor::send_closure(validator_manager_, &ton::validator::ValidatorManagerInterface::add_ext_server_port, port);
}
void ValidatorEngine::start_lite_server() {
for (auto &s : config_.liteservers) {
add_lite_server(s.second, static_cast<td::uint16>(s.first));
}
started_lite_server();
}
void ValidatorEngine::started_lite_server() {
start_control_interface();
}
void ValidatorEngine::add_control_interface(ton::PublicKeyHash id, td::uint16 port) {
class Callback : public ton::adnl::Adnl::Callback {
public:
void receive_message(ton::adnl::AdnlNodeIdShort src, ton::adnl::AdnlNodeIdShort dst,
td::BufferSlice data) override {
}
void receive_query(ton::adnl::AdnlNodeIdShort src, ton::adnl::AdnlNodeIdShort dst, td::BufferSlice data,
td::Promise<td::BufferSlice> promise) override {
td::actor::send_closure(id_, &ValidatorEngine::process_control_query, port_, src, dst, std::move(data),
std::move(promise));
}
Callback(td::actor::ActorId<ValidatorEngine> id, td::uint16 port) : id_(id), port_(port) {
}
private:
td::actor::ActorId<ValidatorEngine> id_;
td::uint16 port_;
};
td::actor::send_closure(adnl_, &ton::adnl::Adnl::add_id, ton::adnl::AdnlNodeIdFull{keys_[id]},
ton::adnl::AdnlAddressList{});
td::actor::send_closure(adnl_, &ton::adnl::Adnl::subscribe, ton::adnl::AdnlNodeIdShort{id}, std::string(""),
std::make_unique<Callback>(actor_id(this), port));
td::actor::send_closure(control_ext_server_, &ton::adnl::AdnlExtServer::add_local_id, ton::adnl::AdnlNodeIdShort{id});
td::actor::send_closure(control_ext_server_, &ton::adnl::AdnlExtServer::add_tcp_port, port);
}
void ValidatorEngine::add_control_process(ton::PublicKeyHash id, td::uint16 port, ton::PublicKeyHash pub,
td::int32 permissions) {
control_permissions_[CI_key{id, port, pub}] |= permissions;
}
void ValidatorEngine::start_control_interface() {
std::vector<ton::adnl::AdnlNodeIdShort> c_ids;
std::vector<td::uint16> ports;
auto P = td::PromiseCreator::lambda(
[SelfId = actor_id(this)](td::Result<td::actor::ActorOwn<ton::adnl::AdnlExtServer>> R) {
R.ensure();
td::actor::send_closure(SelfId, &ValidatorEngine::started_control_interface, R.move_as_ok());
});
td::actor::send_closure(adnl_, &ton::adnl::Adnl::create_ext_server, std::move(c_ids), std::move(ports), std::move(P));
}
void ValidatorEngine::started_control_interface(td::actor::ActorOwn<ton::adnl::AdnlExtServer> control_ext_server) {
control_ext_server_ = std::move(control_ext_server);
for (auto &s : config_.controls) {
add_control_interface(s.second.key, static_cast<td::uint16>(s.first));
for (auto &p : s.second.clients) {
add_control_process(s.second.key, static_cast<td::uint16>(s.first), p.first, p.second);
}
}
start_full_node_masters();
}
void ValidatorEngine::start_full_node_masters() {
for (auto &x : config_.full_node_masters) {
full_node_masters_.emplace(
static_cast<td::uint16>(x.first),
ton::validator::fullnode::FullNodeMaster::create(
ton::adnl::AdnlNodeIdShort{x.second}, static_cast<td::uint16>(x.first),
validator_options_->zero_block_id().file_hash, keyring_.get(), adnl_.get(), validator_manager_.get()));
}
started_full_node_masters();
}
void ValidatorEngine::started_full_node_masters() {
started();
}
void ValidatorEngine::started() {
started_ = true;
}
void ValidatorEngine::try_add_adnl_node(ton::PublicKeyHash key, AdnlCategory cat, td::Promise<td::Unit> promise) {
if (cat < 0 || static_cast<td::uint32>(cat) > max_cat()) {
promise.set_error(td::Status::Error(ton::ErrorCode::protoviolation, "bad category value"));
return;
}
auto R = config_.config_add_adnl_addr(key, cat);
if (R.is_error()) {
promise.set_error(R.move_as_error());
return;
}
if (!R.move_as_ok()) {
promise.set_value(td::Unit());
return;
}
add_adnl(key, cat);
write_config(std::move(promise));
}
void ValidatorEngine::try_add_dht_node(ton::PublicKeyHash key_hash, td::Promise<td::Unit> promise) {
auto R = config_.config_add_dht_node(key_hash);
if (R.is_error()) {
promise.set_error(R.move_as_error());
return;
}
if (!R.move_as_ok()) {
promise.set_value(td::Unit());
return;
}
add_dht(key_hash);
write_config(std::move(promise));
}
void ValidatorEngine::try_add_validator_permanent_key(ton::PublicKeyHash key_hash, td::uint32 election_date,
td::uint32 ttl, td::Promise<td::Unit> promise) {
auto R = config_.config_add_validator_permanent_key(key_hash, election_date, ttl);
if (R.is_error()) {
promise.set_error(R.move_as_error());
return;
}
if (!R.move_as_ok()) {
promise.set_value(td::Unit());
return;
}
td::MultiPromise mp;
auto ig = mp.init_guard();
ig.add_promise(std::move(promise));
if (!validator_manager_.empty()) {
td::actor::send_closure(validator_manager_, &ton::validator::ValidatorManagerInterface::add_permanent_key, key_hash,
ig.get_promise());
}
if (!full_node_.empty()) {
td::actor::send_closure(full_node_, &ton::validator::fullnode::FullNode::add_permanent_key, key_hash,
ig.get_promise());
}
write_config(ig.get_promise());
}
void ValidatorEngine::try_add_validator_temp_key(ton::PublicKeyHash perm_key, ton::PublicKeyHash temp_key,
td::uint32 ttl, td::Promise<td::Unit> promise) {
auto R = config_.config_add_validator_temp_key(perm_key, temp_key, ttl);
if (R.is_error()) {
promise.set_error(R.move_as_error());
return;
}
if (!R.move_as_ok()) {
promise.set_value(td::Unit());
return;
}
td::MultiPromise mp;
auto ig = mp.init_guard();
ig.add_promise(std::move(promise));
if (!validator_manager_.empty()) {
td::actor::send_closure(validator_manager_, &ton::validator::ValidatorManagerInterface::add_temp_key, temp_key,
ig.get_promise());
}
write_config(ig.get_promise());
}
void ValidatorEngine::try_add_validator_adnl_addr(ton::PublicKeyHash perm_key, ton::PublicKeyHash adnl_id,
td::uint32 ttl, td::Promise<td::Unit> promise) {
auto R = config_.config_add_validator_adnl_id(perm_key, adnl_id, ttl);
if (R.is_error()) {
promise.set_error(R.move_as_error());
return;
}
if (!R.move_as_ok()) {
promise.set_value(td::Unit());
return;
}
write_config(std::move(promise));
}
void ValidatorEngine::try_add_full_node_adnl_addr(ton::PublicKeyHash id, td::Promise<td::Unit> promise) {
auto R = config_.config_add_full_node_adnl_id(id);
if (R.is_error()) {
promise.set_error(R.move_as_error());
return;
}
if (!R.move_as_ok()) {
promise.set_value(td::Unit());
return;
}
if (!full_node_.empty()) {
td::actor::send_closure(full_node_, &ton::validator::fullnode::FullNode::update_adnl_id,
ton::adnl::AdnlNodeIdShort{id}, [](td::Unit) {});
}
write_config(std::move(promise));
}
void ValidatorEngine::try_add_liteserver(ton::PublicKeyHash id, td::int32 port, td::Promise<td::Unit> promise) {
auto R = config_.config_add_lite_server(id, port);
if (R.is_error()) {
promise.set_error(R.move_as_error());
return;
}
if (!R.move_as_ok()) {
promise.set_value(td::Unit());
return;
}
add_lite_server(id, static_cast<td::uint16>(port));
write_config(std::move(promise));
}
void ValidatorEngine::try_add_control_interface(ton::PublicKeyHash id, td::int32 port, td::Promise<td::Unit> promise) {
auto R = config_.config_add_control_interface(id, port);
if (R.is_error()) {
promise.set_error(R.move_as_error());
return;
}
if (!R.move_as_ok()) {
promise.set_value(td::Unit());
return;
}
add_control_interface(id, static_cast<td::uint16>(port));
write_config(std::move(promise));
}
void ValidatorEngine::try_add_control_process(ton::PublicKeyHash id, td::int32 port, ton::PublicKeyHash pub,
td::int32 permissions, td::Promise<td::Unit> promise) {
auto R = config_.config_add_control_process(id, port, pub, permissions);
if (R.is_error()) {
promise.set_error(R.move_as_error());
return;
}
if (!R.move_as_ok()) {
promise.set_value(td::Unit());
return;
}
add_control_process(id, static_cast<td::uint16>(port), pub, permissions);
write_config(std::move(promise));
}
void ValidatorEngine::try_del_adnl_node(ton::PublicKeyHash pub, td::Promise<td::Unit> promise) {
auto R = config_.config_del_adnl_addr(pub);
if (R.is_error()) {
promise.set_error(R.move_as_error());
return;
}
if (!R.move_as_ok()) {
promise.set_value(td::Unit());
return;
}
td::actor::send_closure(adnl_, &ton::adnl::Adnl::del_id, ton::adnl::AdnlNodeIdShort{pub}, [](td::Unit) {});
write_config(std::move(promise));
}
void ValidatorEngine::try_del_dht_node(ton::PublicKeyHash pub, td::Promise<td::Unit> promise) {
if (dht_nodes_.size() == 1 && pub == default_dht_node_) {
promise.set_error(td::Status::Error(ton::ErrorCode::error, "cannot remove last dht node"));
return;
}
auto R = config_.config_del_dht_node(pub);
if (R.is_error()) {
promise.set_error(R.move_as_error());
return;
}
if (!R.move_as_ok()) {
promise.set_value(td::Unit());
return;
}
if (pub == default_dht_node_) {
default_dht_node_ = *config_.dht_ids.begin();
auto d = dht_nodes_[default_dht_node_].get();
CHECK(!d.empty());
td::actor::send_closure(adnl_, &ton::adnl::Adnl::register_dht_node, d);
td::actor::send_closure(overlay_manager_, &ton::overlay::Overlays::update_dht_node, d);
if (!full_node_.empty()) {
td::actor::send_closure(full_node_, &ton::validator::fullnode::FullNode::update_dht_node, d);
}
}
dht_nodes_.erase(pub);
write_config(std::move(promise));
}
void ValidatorEngine::try_del_validator_permanent_key(ton::PublicKeyHash pub, td::Promise<td::Unit> promise) {
auto R = config_.config_del_validator_permanent_key(pub);
if (R.is_error()) {
promise.set_error(R.move_as_error());
return;
}
if (!R.move_as_ok()) {
promise.set_value(td::Unit());
return;
}
if (!validator_manager_.empty()) {
td::actor::send_closure(validator_manager_, &ton::validator::ValidatorManagerInterface::del_permanent_key, pub,
[](td::Unit) {});
}
if (!full_node_.empty()) {
td::actor::send_closure(full_node_, &ton::validator::fullnode::FullNode::del_permanent_key, pub, [](td::Unit) {});
}
write_config(std::move(promise));
}
void ValidatorEngine::try_del_validator_temp_key(ton::PublicKeyHash perm, ton::PublicKeyHash temp_key,
td::Promise<td::Unit> promise) {
auto R = config_.config_del_validator_temp_key(perm, temp_key);
if (R.is_error()) {
promise.set_error(R.move_as_error());
return;
}
if (!R.move_as_ok()) {
promise.set_value(td::Unit());
return;
}
if (!validator_manager_.empty()) {
td::actor::send_closure(validator_manager_, &ton::validator::ValidatorManagerInterface::del_temp_key, temp_key,
[](td::Unit) {});
}
write_config(std::move(promise));
}
void ValidatorEngine::try_del_validator_adnl_addr(ton::PublicKeyHash perm, ton::PublicKeyHash adnl_id,
td::Promise<td::Unit> promise) {
auto R = config_.config_del_validator_adnl_id(perm, adnl_id);
if (R.is_error()) {
promise.set_error(R.move_as_error());
return;
}
if (!R.move_as_ok()) {
promise.set_value(td::Unit());
return;
}
write_config(std::move(promise));
}
void ValidatorEngine::reload_adnl_addrs() {
addr_lists_.clear();
prio_addr_lists_.clear();
for (auto &addr : config_.addrs) {
add_addr(addr.first, addr.second);
}
for (auto &adnl : config_.adnl_ids) {
add_adnl(adnl.first, adnl.second);
}
}
void ValidatorEngine::try_add_listening_port(td::uint32 ip, td::int32 port, std::vector<AdnlCategory> cats,
std::vector<AdnlCategory> prio_cats, td::Promise<td::Unit> promise) {
td::IPAddress a;
a.init_ipv4_port(td::IPAddress::ipv4_to_str(ip), static_cast<td::uint16>(port)).ensure();
auto R = config_.config_add_network_addr(a, a, nullptr, std::move(cats), std::move(prio_cats));
if (R.is_error()) {
promise.set_error(R.move_as_error());
return;
}
if (!R.move_as_ok()) {
promise.set_value(td::Unit());
return;
}
reload_adnl_addrs();
write_config(std::move(promise));
}
void ValidatorEngine::try_del_listening_port(td::uint32 ip, td::int32 port, std::vector<AdnlCategory> cats,
std::vector<AdnlCategory> prio_cats, td::Promise<td::Unit> promise) {
td::IPAddress a;
a.init_ipv4_port(td::IPAddress::ipv4_to_str(ip), static_cast<td::uint16>(port)).ensure();
auto R = config_.config_del_network_addr(a, std::move(cats), std::move(prio_cats));
if (R.is_error()) {
promise.set_error(R.move_as_error());
return;
}
if (!R.move_as_ok()) {
promise.set_value(td::Unit());
return;
}
reload_adnl_addrs();
write_config(std::move(promise));
}
void ValidatorEngine::try_add_proxy(td::uint32 in_ip, td::int32 in_port, td::uint32 out_ip, td::int32 out_port,
std::shared_ptr<ton::adnl::AdnlProxy> proxy, std::vector<AdnlCategory> cats,
std::vector<AdnlCategory> prio_cats, td::Promise<td::Unit> promise) {
td::IPAddress in_addr;
in_addr.init_ipv4_port(td::IPAddress::ipv4_to_str(in_ip), static_cast<td::uint16>(in_port)).ensure();
td::IPAddress out_addr;
out_addr.init_ipv4_port(td::IPAddress::ipv4_to_str(out_ip), static_cast<td::uint16>(out_port)).ensure();
auto R = config_.config_add_network_addr(in_addr, out_addr, std::move(proxy), std::move(cats), std::move(prio_cats));
if (R.is_error()) {
promise.set_error(R.move_as_error());
return;
}
if (!R.move_as_ok()) {
promise.set_value(td::Unit());
return;
}
reload_adnl_addrs();
write_config(std::move(promise));
}
void ValidatorEngine::try_del_proxy(td::uint32 ip, td::int32 port, std::vector<AdnlCategory> cats,
std::vector<AdnlCategory> prio_cats, td::Promise<td::Unit> promise) {
td::IPAddress a;
a.init_ipv4_port(td::IPAddress::ipv4_to_str(ip), static_cast<td::uint16>(port)).ensure();
auto R = config_.config_del_network_addr(a, std::move(cats), std::move(prio_cats));
if (R.is_error()) {
promise.set_error(R.move_as_error());
return;
}
if (!R.move_as_ok()) {
promise.set_value(td::Unit());
return;
}
reload_adnl_addrs();
write_config(std::move(promise));
}
void ValidatorEngine::check_key(ton::PublicKeyHash id, td::Promise<td::Unit> promise) {
if (keys_.count(id) == 1) {
promise.set_value(td::Unit());
return;
}
auto P = td::PromiseCreator::lambda(
[SelfId = actor_id(this), promise = std::move(promise)](td::Result<ton::PublicKey> R) mutable {
if (R.is_error()) {
promise.set_error(R.move_as_error());
} else {
td::actor::send_closure(SelfId, &ValidatorEngine::got_key, R.move_as_ok());
promise.set_value(td::Unit());
}
});
td::actor::send_closure(keyring_, &ton::keyring::Keyring::get_public_key, id, std::move(P));
}
td::BufferSlice ValidatorEngine::create_control_query_error(td::Status error) {
return ton::serialize_tl_object(
ton::create_tl_object<ton::ton_api::engine_validator_controlQueryError>(error.code(), error.message().str()),
true);
}
void ValidatorEngine::run_control_query(ton::ton_api::engine_validator_getTime &query, td::BufferSlice data,
ton::PublicKeyHash src, td::uint32 perm, td::Promise<td::BufferSlice> promise) {
if (!(perm & ValidatorEnginePermissions::vep_default)) {
promise.set_value(create_control_query_error(td::Status::Error(ton::ErrorCode::error, "not authorized")));
return;
}
auto obj = ton::create_tl_object<ton::ton_api::engine_validator_time>(static_cast<td::int32>(td::Clocks::system()));
promise.set_value(ton::serialize_tl_object(obj, true));
}
void ValidatorEngine::run_control_query(ton::ton_api::engine_validator_importPrivateKey &query, td::BufferSlice data,
ton::PublicKeyHash src, td::uint32 perm, td::Promise<td::BufferSlice> promise) {
if (!(perm & ValidatorEnginePermissions::vep_default)) {
promise.set_value(create_control_query_error(td::Status::Error(ton::ErrorCode::error, "not authorized")));
return;
}
if (keyring_.empty()) {
promise.set_value(create_control_query_error(td::Status::Error(ton::ErrorCode::notready, "not started keyring")));
return;
}
auto pk = ton::PrivateKey{query.key_};
auto P = td::PromiseCreator::lambda(
[promise = std::move(promise), hash = pk.compute_short_id()](td::Result<td::Unit> R) mutable {
if (R.is_error()) {
promise.set_value(create_control_query_error(R.move_as_error()));
} else {
promise.set_value(
ton::serialize_tl_object(ton::create_tl_object<ton::ton_api::engine_validator_keyHash>(hash.tl()), true));
}
});
td::actor::send_closure(keyring_, &ton::keyring::Keyring::add_key, std::move(pk), false, std::move(P));
}
void ValidatorEngine::run_control_query(ton::ton_api::engine_validator_exportPrivateKey &query, td::BufferSlice data,
ton::PublicKeyHash src, td::uint32 perm, td::Promise<td::BufferSlice> promise) {
if (!(perm & ValidatorEnginePermissions::vep_unsafe)) {
promise.set_value(create_control_query_error(td::Status::Error(ton::ErrorCode::error, "not authorized")));
return;
}
if (keyring_.empty()) {
promise.set_value(create_control_query_error(td::Status::Error(ton::ErrorCode::notready, "not started keyring")));
return;
}
promise.set_value(create_control_query_error(td::Status::Error(ton::ErrorCode::notready, "not implemented")));
}
void ValidatorEngine::run_control_query(ton::ton_api::engine_validator_exportPublicKey &query, td::BufferSlice data,
ton::PublicKeyHash src, td::uint32 perm, td::Promise<td::BufferSlice> promise) {
if (!(perm & ValidatorEnginePermissions::vep_default)) {
promise.set_value(create_control_query_error(td::Status::Error(ton::ErrorCode::error, "not authorized")));
return;
}
if (keyring_.empty()) {
promise.set_value(create_control_query_error(td::Status::Error(ton::ErrorCode::notready, "not started keyring")));
return;
}
auto P = td::PromiseCreator::lambda([promise = std::move(promise)](td::Result<ton::PublicKey> R) mutable {
if (R.is_error()) {
promise.set_value(create_control_query_error(R.move_as_error()));
} else {
auto pub = R.move_as_ok();
promise.set_value(ton::serialize_tl_object(pub.tl(), true));
}
});
td::actor::send_closure(keyring_, &ton::keyring::Keyring::get_public_key, ton::PublicKeyHash{query.key_hash_},
std::move(P));
}
void ValidatorEngine::run_control_query(ton::ton_api::engine_validator_generateKeyPair &query, td::BufferSlice data,
ton::PublicKeyHash src, td::uint32 perm, td::Promise<td::BufferSlice> promise) {
if (!(perm & ValidatorEnginePermissions::vep_default)) {
promise.set_value(create_control_query_error(td::Status::Error(ton::ErrorCode::error, "not authorized")));
return;
}
if (keyring_.empty()) {
promise.set_value(create_control_query_error(td::Status::Error(ton::ErrorCode::notready, "not started keyring")));
return;
}
auto pk = ton::PrivateKey{ton::privkeys::Ed25519::random()};
auto P = td::PromiseCreator::lambda(
[promise = std::move(promise), hash = pk.compute_short_id()](td::Result<td::Unit> R) mutable {
if (R.is_error()) {
promise.set_value(create_control_query_error(R.move_as_error()));
} else {
promise.set_value(
ton::serialize_tl_object(ton::create_tl_object<ton::ton_api::engine_validator_keyHash>(hash.tl()), true));
}
});
td::actor::send_closure(keyring_, &ton::keyring::Keyring::add_key, std::move(pk), false, std::move(P));
}
void ValidatorEngine::run_control_query(ton::ton_api::engine_validator_addAdnlId &query, td::BufferSlice data,
ton::PublicKeyHash src, td::uint32 perm, td::Promise<td::BufferSlice> promise) {
if (!(perm & ValidatorEnginePermissions::vep_modify)) {
promise.set_value(create_control_query_error(td::Status::Error(ton::ErrorCode::error, "not authorized")));
return;
}
if (!started_) {
promise.set_value(create_control_query_error(td::Status::Error(ton::ErrorCode::notready, "not started")));
return;
}
auto id = ton::PublicKeyHash{query.key_hash_};
auto P = td::PromiseCreator::lambda([SelfId = actor_id(this), id, cat = query.category_,
promise = std::move(promise)](td::Result<td::Unit> R) mutable {
if (R.is_error()) {
promise.set_value(create_control_query_error(R.move_as_error_prefix("failed to get public key: ")));
return;
}
auto P = td::PromiseCreator::lambda([promise = std::move(promise)](td::Result<td::Unit> R) mutable {
if (R.is_error()) {
promise.set_value(create_control_query_error(R.move_as_error_prefix("failed to add adnl node: ")));
} else {
promise.set_value(
ton::serialize_tl_object(ton::create_tl_object<ton::ton_api::engine_validator_success>(), true));
}
});
td::actor::send_closure(SelfId, &ValidatorEngine::try_add_adnl_node, id, cat, std::move(P));
});
check_key(id, std::move(P));
}
void ValidatorEngine::run_control_query(ton::ton_api::engine_validator_addDhtId &query, td::BufferSlice data,
ton::PublicKeyHash src, td::uint32 perm, td::Promise<td::BufferSlice> promise) {
if (!(perm & ValidatorEnginePermissions::vep_modify)) {
promise.set_value(create_control_query_error(td::Status::Error(ton::ErrorCode::error, "not authorized")));
return;
}
if (!started_) {
promise.set_value(create_control_query_error(td::Status::Error(ton::ErrorCode::notready, "not started")));
return;
}
auto id = ton::PublicKeyHash{query.key_hash_};
auto P = td::PromiseCreator::lambda(
[SelfId = actor_id(this), id, promise = std::move(promise)](td::Result<td::Unit> R) mutable {
if (R.is_error()) {
promise.set_value(create_control_query_error(R.move_as_error_prefix("failed to get public key: ")));
return;
}
auto P = td::PromiseCreator::lambda([promise = std::move(promise)](td::Result<td::Unit> R) mutable {
if (R.is_error()) {
promise.set_value(create_control_query_error(R.move_as_error_prefix("failed to add dht node: ")));
} else {
promise.set_value(
ton::serialize_tl_object(ton::create_tl_object<ton::ton_api::engine_validator_success>(), true));
}
});
td::actor::send_closure(SelfId, &ValidatorEngine::try_add_dht_node, id, std::move(P));
});
check_key(id, std::move(P));
}
void ValidatorEngine::run_control_query(ton::ton_api::engine_validator_addValidatorPermanentKey &query,
td::BufferSlice data, ton::PublicKeyHash src, td::uint32 perm,
td::Promise<td::BufferSlice> promise) {
if (!(perm & ValidatorEnginePermissions::vep_modify)) {
promise.set_value(create_control_query_error(td::Status::Error(ton::ErrorCode::error, "not authorized")));
return;
}
if (!started_) {
promise.set_value(create_control_query_error(td::Status::Error(ton::ErrorCode::notready, "not started")));
return;
}
auto id = ton::PublicKeyHash{query.key_hash_};
auto P = td::PromiseCreator::lambda([SelfId = actor_id(this), id, election_date = query.election_date_,
ttl = query.ttl_, promise = std::move(promise)](td::Result<td::Unit> R) mutable {
if (R.is_error()) {
promise.set_value(create_control_query_error(R.move_as_error_prefix("failed to get public key: ")));
return;
}
auto P = td::PromiseCreator::lambda([promise = std::move(promise)](td::Result<td::Unit> R) mutable {
if (R.is_error()) {
promise.set_value(
create_control_query_error(R.move_as_error_prefix("failed to add validator permanent key: ")));
} else {
promise.set_value(
ton::serialize_tl_object(ton::create_tl_object<ton::ton_api::engine_validator_success>(), true));
}
});
td::actor::send_closure(SelfId, &ValidatorEngine::try_add_validator_permanent_key, id, election_date, ttl,
std::move(P));
});
check_key(id, std::move(P));
}
void ValidatorEngine::run_control_query(ton::ton_api::engine_validator_addValidatorTempKey &query, td::BufferSlice data,
ton::PublicKeyHash src, td::uint32 perm, td::Promise<td::BufferSlice> promise) {
if (!(perm & ValidatorEnginePermissions::vep_modify)) {
promise.set_value(create_control_query_error(td::Status::Error(ton::ErrorCode::error, "not authorized")));
return;
}
if (!started_) {
promise.set_value(create_control_query_error(td::Status::Error(ton::ErrorCode::notready, "not started")));
return;
}
auto id = ton::PublicKeyHash{query.key_hash_};
auto P =
td::PromiseCreator::lambda([SelfId = actor_id(this), perm_key = ton::PublicKeyHash{query.permanent_key_hash_}, id,
ttl = query.ttl_, promise = std::move(promise)](td::Result<td::Unit> R) mutable {
if (R.is_error()) {
promise.set_value(create_control_query_error(R.move_as_error_prefix("failed to get public key: ")));
return;
}
auto P = td::PromiseCreator::lambda([promise = std::move(promise)](td::Result<td::Unit> R) mutable {
if (R.is_error()) {
promise.set_value(create_control_query_error(R.move_as_error_prefix("failed to add validator temp key: ")));
} else {
promise.set_value(
ton::serialize_tl_object(ton::create_tl_object<ton::ton_api::engine_validator_success>(), true));
}
});
td::actor::send_closure(SelfId, &ValidatorEngine::try_add_validator_temp_key, perm_key, id, ttl, std::move(P));
});
check_key(id, std::move(P));
}
void ValidatorEngine::run_control_query(ton::ton_api::engine_validator_addValidatorAdnlAddress &query,
td::BufferSlice data, ton::PublicKeyHash src, td::uint32 perm,
td::Promise<td::BufferSlice> promise) {
if (!(perm & ValidatorEnginePermissions::vep_modify)) {
promise.set_value(create_control_query_error(td::Status::Error(ton::ErrorCode::error, "not authorized")));
return;
}
if (!started_) {
promise.set_value(create_control_query_error(td::Status::Error(ton::ErrorCode::notready, "not started")));
return;
}
auto id = ton::PublicKeyHash{query.key_hash_};
auto P = td::PromiseCreator::lambda([SelfId = actor_id(this),
perm_key = ton::PublicKeyHash{query.permanent_key_hash_}, id, ttl = query.ttl_,
promise = std::move(promise)](td::Result<td::Unit> R) mutable {
if (R.is_error()) {
promise.set_value(create_control_query_error(R.move_as_error_prefix("failed to get public key: ")));
return;
}
auto P = td::PromiseCreator::lambda([promise = std::move(promise)](td::Result<td::Unit> R) mutable {
if (R.is_error()) {
promise.set_value(create_control_query_error(R.move_as_error_prefix("failed to add validator adnl address: ")));
} else {
promise.set_value(
ton::serialize_tl_object(ton::create_tl_object<ton::ton_api::engine_validator_success>(), true));
}
});
td::actor::send_closure(SelfId, &ValidatorEngine::try_add_validator_adnl_addr, perm_key, id, ttl, std::move(P));
});
check_key(id, std::move(P));
}
void ValidatorEngine::run_control_query(ton::ton_api::engine_validator_changeFullNodeAdnlAddress &query,
td::BufferSlice data, ton::PublicKeyHash src, td::uint32 perm,
td::Promise<td::BufferSlice> promise) {
if (!(perm & ValidatorEnginePermissions::vep_modify)) {
promise.set_value(create_control_query_error(td::Status::Error(ton::ErrorCode::error, "not authorized")));
return;
}
if (!started_) {
promise.set_value(create_control_query_error(td::Status::Error(ton::ErrorCode::notready, "not started")));
return;
}
auto id = ton::PublicKeyHash{query.adnl_id_};
auto P = td::PromiseCreator::lambda([SelfId = actor_id(this), id,
promise = std::move(promise)](td::Result<td::Unit> R) mutable {
if (R.is_error()) {
promise.set_value(create_control_query_error(R.move_as_error_prefix("failed to get public key: ")));
return;
}
auto P = td::PromiseCreator::lambda([promise = std::move(promise)](td::Result<td::Unit> R) mutable {
if (R.is_error()) {
promise.set_value(create_control_query_error(R.move_as_error_prefix("failed to change full node address: ")));
} else {
promise.set_value(
ton::serialize_tl_object(ton::create_tl_object<ton::ton_api::engine_validator_success>(), true));
}
});
td::actor::send_closure(SelfId, &ValidatorEngine::try_add_full_node_adnl_addr, id, std::move(P));
});
check_key(id, std::move(P));
}
void ValidatorEngine::run_control_query(ton::ton_api::engine_validator_addLiteserver &query, td::BufferSlice data,
ton::PublicKeyHash src, td::uint32 perm, td::Promise<td::BufferSlice> promise) {
if (!(perm & ValidatorEnginePermissions::vep_modify)) {
promise.set_value(create_control_query_error(td::Status::Error(ton::ErrorCode::error, "not authorized")));
return;
}
if (!started_) {
promise.set_value(create_control_query_error(td::Status::Error(ton::ErrorCode::notready, "not started")));
return;
}
auto id = ton::PublicKeyHash{query.key_hash_};
auto P = td::PromiseCreator::lambda([SelfId = actor_id(this), id, port = static_cast<td::uint16>(query.port_),
promise = std::move(promise)](td::Result<td::Unit> R) mutable {
if (R.is_error()) {
promise.set_value(create_control_query_error(R.move_as_error_prefix("failed to get public key: ")));
return;
}
auto P = td::PromiseCreator::lambda([promise = std::move(promise)](td::Result<td::Unit> R) mutable {
if (R.is_error()) {
promise.set_value(create_control_query_error(R.move_as_error_prefix("failed to add liteserver: ")));
} else {
promise.set_value(
ton::serialize_tl_object(ton::create_tl_object<ton::ton_api::engine_validator_success>(), true));
}
});
td::actor::send_closure(SelfId, &ValidatorEngine::try_add_liteserver, id, port, std::move(P));
});
check_key(id, std::move(P));
}
void ValidatorEngine::run_control_query(ton::ton_api::engine_validator_addControlInterface &query, td::BufferSlice data,
ton::PublicKeyHash src, td::uint32 perm, td::Promise<td::BufferSlice> promise) {
if (!(perm & ValidatorEnginePermissions::vep_modify)) {
promise.set_value(create_control_query_error(td::Status::Error(ton::ErrorCode::error, "not authorized")));
return;
}
if (!started_) {
promise.set_value(create_control_query_error(td::Status::Error(ton::ErrorCode::notready, "not started")));
return;
}
auto id = ton::PublicKeyHash{query.key_hash_};
auto P = td::PromiseCreator::lambda([SelfId = actor_id(this), id, port = static_cast<td::uint16>(query.port_),
promise = std::move(promise)](td::Result<td::Unit> R) mutable {
if (R.is_error()) {
promise.set_value(create_control_query_error(R.move_as_error_prefix("failed to get public key: ")));
return;
}
auto P = td::PromiseCreator::lambda([promise = std::move(promise)](td::Result<td::Unit> R) mutable {
if (R.is_error()) {
promise.set_value(create_control_query_error(R.move_as_error_prefix("failed to add control interface: ")));
} else {
promise.set_value(
ton::serialize_tl_object(ton::create_tl_object<ton::ton_api::engine_validator_success>(), true));
}
});
td::actor::send_closure(SelfId, &ValidatorEngine::try_add_control_interface, id, port, std::move(P));
});
check_key(id, std::move(P));
}
void ValidatorEngine::run_control_query(ton::ton_api::engine_validator_delAdnlId &query, td::BufferSlice data,
ton::PublicKeyHash src, td::uint32 perm, td::Promise<td::BufferSlice> promise) {
if (!(perm & ValidatorEnginePermissions::vep_modify)) {
promise.set_value(create_control_query_error(td::Status::Error(ton::ErrorCode::error, "not authorized")));
return;
}
if (!started_) {
promise.set_value(create_control_query_error(td::Status::Error(ton::ErrorCode::notready, "not started")));
return;
}
auto id = ton::PublicKeyHash{query.key_hash_};
auto P = td::PromiseCreator::lambda([promise = std::move(promise)](td::Result<td::Unit> R) mutable {
if (R.is_error()) {
promise.set_value(create_control_query_error(R.move_as_error_prefix("failed to del adnl node: ")));
} else {
promise.set_value(
ton::serialize_tl_object(ton::create_tl_object<ton::ton_api::engine_validator_success>(), true));
}
});
try_del_adnl_node(id, std::move(P));
}
void ValidatorEngine::run_control_query(ton::ton_api::engine_validator_delDhtId &query, td::BufferSlice data,
ton::PublicKeyHash src, td::uint32 perm, td::Promise<td::BufferSlice> promise) {
if (!(perm & ValidatorEnginePermissions::vep_modify)) {
promise.set_value(create_control_query_error(td::Status::Error(ton::ErrorCode::error, "not authorized")));
return;
}
if (!started_) {
promise.set_value(create_control_query_error(td::Status::Error(ton::ErrorCode::notready, "not started")));
return;
}
auto id = ton::PublicKeyHash{query.key_hash_};
auto P = td::PromiseCreator::lambda([promise = std::move(promise)](td::Result<td::Unit> R) mutable {
if (R.is_error()) {
promise.set_value(create_control_query_error(R.move_as_error_prefix("failed to del adnl node: ")));
} else {
promise.set_value(
ton::serialize_tl_object(ton::create_tl_object<ton::ton_api::engine_validator_success>(), true));
}
});
try_del_dht_node(id, std::move(P));
}
void ValidatorEngine::run_control_query(ton::ton_api::engine_validator_delValidatorPermanentKey &query,
td::BufferSlice data, ton::PublicKeyHash src, td::uint32 perm,
td::Promise<td::BufferSlice> promise) {
if (!(perm & ValidatorEnginePermissions::vep_modify)) {
promise.set_value(create_control_query_error(td::Status::Error(ton::ErrorCode::error, "not authorized")));
return;
}
if (!started_) {
promise.set_value(create_control_query_error(td::Status::Error(ton::ErrorCode::notready, "not started")));
return;
}
auto id = ton::PublicKeyHash{query.key_hash_};
auto P = td::PromiseCreator::lambda([promise = std::move(promise)](td::Result<td::Unit> R) mutable {
if (R.is_error()) {
promise.set_value(create_control_query_error(R.move_as_error_prefix("failed to del validator permanent key: ")));
} else {
promise.set_value(
ton::serialize_tl_object(ton::create_tl_object<ton::ton_api::engine_validator_success>(), true));
}
});
try_del_validator_permanent_key(id, std::move(P));
}
void ValidatorEngine::run_control_query(ton::ton_api::engine_validator_delValidatorTempKey &query, td::BufferSlice data,
ton::PublicKeyHash src, td::uint32 perm, td::Promise<td::BufferSlice> promise) {
if (!(perm & ValidatorEnginePermissions::vep_modify)) {
promise.set_value(create_control_query_error(td::Status::Error(ton::ErrorCode::error, "not authorized")));
return;
}
if (!started_) {
promise.set_value(create_control_query_error(td::Status::Error(ton::ErrorCode::notready, "not started")));
return;
}
auto id = ton::PublicKeyHash{query.key_hash_};
auto P = td::PromiseCreator::lambda([promise = std::move(promise)](td::Result<td::Unit> R) mutable {
if (R.is_error()) {
promise.set_value(create_control_query_error(R.move_as_error_prefix("failed to del validator temp key: ")));
} else {
promise.set_value(
ton::serialize_tl_object(ton::create_tl_object<ton::ton_api::engine_validator_success>(), true));
}
});
try_del_validator_temp_key(ton::PublicKeyHash{query.permanent_key_hash_}, id, std::move(P));
}
void ValidatorEngine::run_control_query(ton::ton_api::engine_validator_delValidatorAdnlAddress &query,
td::BufferSlice data, ton::PublicKeyHash src, td::uint32 perm,
td::Promise<td::BufferSlice> promise) {
if (!(perm & ValidatorEnginePermissions::vep_modify)) {
promise.set_value(create_control_query_error(td::Status::Error(ton::ErrorCode::error, "not authorized")));
return;
}
if (!started_) {
promise.set_value(create_control_query_error(td::Status::Error(ton::ErrorCode::notready, "not started")));
return;
}
auto id = ton::PublicKeyHash{query.key_hash_};
auto P = td::PromiseCreator::lambda([promise = std::move(promise)](td::Result<td::Unit> R) mutable {
if (R.is_error()) {
promise.set_value(create_control_query_error(R.move_as_error_prefix("failed to del validator adnl addr: ")));
} else {
promise.set_value(
ton::serialize_tl_object(ton::create_tl_object<ton::ton_api::engine_validator_success>(), true));
}
});
try_del_validator_adnl_addr(ton::PublicKeyHash{query.permanent_key_hash_}, id, std::move(P));
}
void ValidatorEngine::run_control_query(ton::ton_api::engine_validator_addListeningPort &query, td::BufferSlice data,
ton::PublicKeyHash src, td::uint32 perm, td::Promise<td::BufferSlice> promise) {
if (!(perm & ValidatorEnginePermissions::vep_modify)) {
promise.set_value(create_control_query_error(td::Status::Error(ton::ErrorCode::error, "not authorized")));
return;
}
if (!started_) {
promise.set_value(create_control_query_error(td::Status::Error(ton::ErrorCode::notready, "not started")));
return;
}
auto P = td::PromiseCreator::lambda([promise = std::move(promise)](td::Result<td::Unit> R) mutable {
if (R.is_error()) {
promise.set_value(create_control_query_error(R.move_as_error_prefix("failed to add listening port: ")));
} else {
promise.set_value(
ton::serialize_tl_object(ton::create_tl_object<ton::ton_api::engine_validator_success>(), true));
}
});
try_add_listening_port(query.ip_, query.port_, query.categories_, query.priority_categories_, std::move(P));
}
void ValidatorEngine::run_control_query(ton::ton_api::engine_validator_delListeningPort &query, td::BufferSlice data,
ton::PublicKeyHash src, td::uint32 perm, td::Promise<td::BufferSlice> promise) {
if (!(perm & ValidatorEnginePermissions::vep_modify)) {
promise.set_value(create_control_query_error(td::Status::Error(ton::ErrorCode::error, "not authorized")));
return;
}
if (!started_) {
promise.set_value(create_control_query_error(td::Status::Error(ton::ErrorCode::notready, "not started")));
return;
}
auto P = td::PromiseCreator::lambda([promise = std::move(promise)](td::Result<td::Unit> R) mutable {
if (R.is_error()) {
promise.set_value(create_control_query_error(R.move_as_error_prefix("failed to del listening port: ")));
} else {
promise.set_value(
ton::serialize_tl_object(ton::create_tl_object<ton::ton_api::engine_validator_success>(), true));
}
});
try_del_listening_port(query.ip_, query.port_, query.categories_, query.priority_categories_, std::move(P));
}
void ValidatorEngine::run_control_query(ton::ton_api::engine_validator_addProxy &query, td::BufferSlice data,
ton::PublicKeyHash src, td::uint32 perm, td::Promise<td::BufferSlice> promise) {
if (!(perm & ValidatorEnginePermissions::vep_modify)) {
promise.set_value(create_control_query_error(td::Status::Error(ton::ErrorCode::error, "not authorized")));
return;
}
if (!started_) {
promise.set_value(create_control_query_error(td::Status::Error(ton::ErrorCode::notready, "not started")));
return;
}
auto R = ton::adnl::AdnlProxy::create(*query.proxy_.get());
if (R.is_error()) {
promise.set_value(create_control_query_error(R.move_as_error_prefix("bad proxy type: ")));
return;
}
auto P = td::PromiseCreator::lambda([promise = std::move(promise)](td::Result<td::Unit> R) mutable {
if (R.is_error()) {
promise.set_value(create_control_query_error(R.move_as_error_prefix("failed to add listening proxy: ")));
} else {
promise.set_value(
ton::serialize_tl_object(ton::create_tl_object<ton::ton_api::engine_validator_success>(), true));
}
});
try_add_proxy(query.in_ip_, query.in_port_, query.out_ip_, query.out_port_, R.move_as_ok(), query.categories_,
query.priority_categories_, std::move(P));
}
void ValidatorEngine::run_control_query(ton::ton_api::engine_validator_delProxy &query, td::BufferSlice data,
ton::PublicKeyHash src, td::uint32 perm, td::Promise<td::BufferSlice> promise) {
if (!(perm & ValidatorEnginePermissions::vep_modify)) {
promise.set_value(create_control_query_error(td::Status::Error(ton::ErrorCode::error, "not authorized")));
return;
}
if (!started_) {
promise.set_value(create_control_query_error(td::Status::Error(ton::ErrorCode::notready, "not started")));
return;
}
auto P = td::PromiseCreator::lambda([promise = std::move(promise)](td::Result<td::Unit> R) mutable {
if (R.is_error()) {
promise.set_value(create_control_query_error(R.move_as_error_prefix("failed to del listening proxy: ")));
} else {
promise.set_value(
ton::serialize_tl_object(ton::create_tl_object<ton::ton_api::engine_validator_success>(), true));
}
});
try_del_proxy(query.out_ip_, query.out_port_, query.categories_, query.priority_categories_, std::move(P));
}
void ValidatorEngine::run_control_query(ton::ton_api::engine_validator_getConfig &query, td::BufferSlice data,
ton::PublicKeyHash src, td::uint32 perm, td::Promise<td::BufferSlice> promise) {
if (!(perm & ValidatorEnginePermissions::vep_default)) {
promise.set_value(create_control_query_error(td::Status::Error(ton::ErrorCode::error, "not authorized")));
return;
}
auto s = td::json_encode<std::string>(td::ToJson(*config_.tl().get()), true);
promise.set_value(ton::create_serialize_tl_object<ton::ton_api::engine_validator_jsonConfig>(s));
}
void ValidatorEngine::run_control_query(ton::ton_api::engine_validator_sign &query, td::BufferSlice data,
ton::PublicKeyHash src, td::uint32 perm, td::Promise<td::BufferSlice> promise) {
if (!(perm & ValidatorEnginePermissions::vep_unsafe)) {
promise.set_value(create_control_query_error(td::Status::Error(ton::ErrorCode::error, "not authorized")));
return;
}
if (!started_) {
promise.set_value(create_control_query_error(td::Status::Error(ton::ErrorCode::notready, "not started")));
return;
}
LOG(WARNING) << "received sign request: src=" << src.bits256_value().to_hex() << " key=" << query.key_hash_.to_hex()
<< " string=\n"
<< td::base64_encode(query.data_.as_slice());
auto P = td::PromiseCreator::lambda([promise = std::move(promise)](td::Result<td::BufferSlice> R) mutable {
if (R.is_error()) {
promise.set_value(create_control_query_error(R.move_as_error()));
} else {
promise.set_value(ton::serialize_tl_object(
ton::create_tl_object<ton::ton_api::engine_validator_signature>(R.move_as_ok()), true));
}
});
td::actor::send_closure(keyring_, &ton::keyring::Keyring::sign_message, ton::PublicKeyHash{query.key_hash_},
std::move(query.data_), std::move(P));
}
void ValidatorEngine::run_control_query(ton::ton_api::engine_validator_setVerbosity &query, td::BufferSlice data,
ton::PublicKeyHash src, td::uint32 perm, td::Promise<td::BufferSlice> promise) {
if (!(perm & ValidatorEnginePermissions::vep_default)) {
promise.set_value(create_control_query_error(td::Status::Error(ton::ErrorCode::error, "not authorized")));
return;
}
if (query.verbosity_ < 0 || query.verbosity_ > 10) {
promise.set_value(
create_control_query_error(td::Status::Error(ton::ErrorCode::error, "verbosity should be in range [0..10]")));
return;
}
SET_VERBOSITY_LEVEL(VERBOSITY_NAME(ERROR) + query.verbosity_);
promise.set_value(ton::serialize_tl_object(ton::create_tl_object<ton::ton_api::engine_validator_success>(), true));
}
void ValidatorEngine::run_control_query(ton::ton_api::engine_validator_getStats &query, td::BufferSlice data,
ton::PublicKeyHash src, td::uint32 perm, td::Promise<td::BufferSlice> promise) {
if (!(perm & ValidatorEnginePermissions::vep_default)) {
promise.set_value(create_control_query_error(td::Status::Error(ton::ErrorCode::error, "not authorized")));
return;
}
if (validator_manager_.empty()) {
promise.set_value(
create_control_query_error(td::Status::Error(ton::ErrorCode::notready, "validator manager not started")));
return;
}
auto P = td::PromiseCreator::lambda(
[promise = std::move(promise)](td::Result<std::vector<std::pair<std::string, std::string>>> R) mutable {
if (R.is_error()) {
promise.set_value(create_control_query_error(R.move_as_error()));
} else {
auto r = R.move_as_ok();
std::vector<ton::tl_object_ptr<ton::ton_api::engine_validator_oneStat>> vec;
for (auto &s : r) {
vec.push_back(ton::create_tl_object<ton::ton_api::engine_validator_oneStat>(s.first, s.second));
}
promise.set_value(ton::create_serialize_tl_object<ton::ton_api::engine_validator_stats>(std::move(vec)));
}
});
td::actor::send_closure(validator_manager_, &ton::validator::ValidatorManagerInterface::prepare_stats, std::move(P));
}
void ValidatorEngine::run_control_query(ton::ton_api::engine_validator_createElectionBid &query, td::BufferSlice data,
ton::PublicKeyHash src, td::uint32 perm, td::Promise<td::BufferSlice> promise) {
if (!(perm & ValidatorEnginePermissions::vep_default)) {
promise.set_value(create_control_query_error(td::Status::Error(ton::ErrorCode::error, "not authorized")));
return;
}
if (!started_) {
promise.set_value(create_control_query_error(td::Status::Error(ton::ErrorCode::notready, "not started")));
return;
}
if (fift_dir_.empty()) {
promise.set_value(create_control_query_error(td::Status::Error(ton::ErrorCode::notready, "no fift dir")));
return;
}
td::actor::create_actor<ValidatorElectionBidCreator>("bidcreate", query.election_date_, query.election_addr_,
query.wallet_, fift_dir_, actor_id(this), keyring_.get(),
std::move(promise))
.release();
}
void ValidatorEngine::run_control_query(ton::ton_api::engine_validator_checkDhtServers &query, td::BufferSlice data,
ton::PublicKeyHash src, td::uint32 perm, td::Promise<td::BufferSlice> promise) {
if (!(perm & ValidatorEnginePermissions::vep_default)) {
promise.set_value(create_control_query_error(td::Status::Error(ton::ErrorCode::error, "not authorized")));
return;
}
if (keyring_.empty()) {
promise.set_value(create_control_query_error(td::Status::Error(ton::ErrorCode::notready, "keyring not started")));
return;
}
if (!dht_config_) {
promise.set_value(create_control_query_error(td::Status::Error(ton::ErrorCode::notready, "no dht config")));
return;
}
if (config_.adnl_ids.count(ton::PublicKeyHash{query.id_}) == 0) {
promise.set_value(create_control_query_error(td::Status::Error(ton::ErrorCode::notready, "no dht config")));
return;
}
td::actor::create_actor<CheckDhtServerStatusQuery>("pinger", dht_config_, ton::adnl::AdnlNodeIdShort{query.id_},
adnl_.get(), std::move(promise))
.release();
}
void ValidatorEngine::process_control_query(td::uint16 port, ton::adnl::AdnlNodeIdShort src,
ton::adnl::AdnlNodeIdShort dst, td::BufferSlice data,
td::Promise<td::BufferSlice> promise) {
auto it = control_permissions_.find(CI_key{dst.pubkey_hash(), port, src.pubkey_hash()});
if (it == control_permissions_.end()) {
promise.set_value(create_control_query_error(td::Status::Error(ton::ErrorCode::error, "forbidden")));
return;
}
auto E = ton::fetch_tl_object<ton::lite_api::liteServer_query>(data.clone(), true);
if (E.is_ok()) {
if (!started_) {
return;
}
td::actor::send_closure(validator_manager_, &ton::validator::ValidatorManagerInterface::run_ext_query,
std::move(data), std::move(promise));
return;
}
auto G = ton::fetch_tl_object<ton::ton_api::engine_validator_controlQuery>(std::move(data), true);
if (G.is_error()) {
promise.set_value(create_control_query_error(G.move_as_error_prefix("failed to parse validator query: ")));
return;
}
data = std::move(G.move_as_ok()->data_);
auto F = ton::fetch_tl_object<ton::ton_api::Function>(data.clone(), true);
if (F.is_error()) {
promise.set_value(create_control_query_error(F.move_as_error_prefix("failed to parse validator query: ")));
return;
}
auto f = F.move_as_ok();
ton::ton_api::downcast_call(*f.get(), [&](auto &obj) {
run_control_query(obj, std::move(data), src.pubkey_hash(), it->second, std::move(promise));
});
}
void ValidatorEngine::run() {
td::mkdir(db_root_).ensure();
ton::errorlog::ErrorLog::create(db_root_);
auto Sr = load_global_config();
if (Sr.is_error()) {
LOG(ERROR) << "failed to load global config'" << global_config_ << "': " << Sr;
std::_Exit(2);
}
keyring_ = ton::keyring::Keyring::create(db_root_ + "/keyring");
// TODO wait for password
started_keyring_ = true;
auto P = td::PromiseCreator::lambda([SelfId = actor_id(this)](td::Result<td::Unit> R) {
if (R.is_error()) {
LOG(ERROR) << "failed to parse config: " << R.move_as_error();
std::_Exit(2);
} else {
td::actor::send_closure(SelfId, &ValidatorEngine::start);
}
});
load_config(std::move(P));
}
std::atomic<bool> need_stats_flag{false};
void need_stats(int sig) {
need_stats_flag.store(true);
}
std::atomic<bool> rotate_logs_flags{false};
void force_rotate_logs(int sig) {
rotate_logs_flags.store(true);
}
void dump_memory_stats() {
if (!is_memprof_on()) {
return;
}
LOG(WARNING) << "memory_dump";
std::vector<AllocInfo> v;
dump_alloc([&](const AllocInfo &info) { v.push_back(info); });
std::sort(v.begin(), v.end(), [](const AllocInfo &a, const AllocInfo &b) { return a.size > b.size; });
size_t total_size = 0;
size_t other_size = 0;
int cnt = 0;
for (auto &info : v) {
if (cnt++ < 50) {
LOG(WARNING) << td::format::as_size(info.size) << td::format::as_array(info.backtrace);
} else {
other_size += info.size;
}
total_size += info.size;
}
LOG(WARNING) << td::tag("other", td::format::as_size(other_size));
LOG(WARNING) << td::tag("total", td::format::as_size(total_size));
LOG(WARNING) << td::tag("total traces", get_ht_size());
LOG(WARNING) << td::tag("fast_backtrace_success_rate", get_fast_backtrace_success_rate());
}
void dump_stats() {
dump_memory_stats();
LOG(WARNING) << td::NamedThreadSafeCounter::get_default();
}
int main(int argc, char *argv[]) {
SET_VERBOSITY_LEVEL(verbosity_INFO);
td::set_default_failure_signal_handler().ensure();
td::actor::ActorOwn<ValidatorEngine> x;
td::unique_ptr<td::LogInterface> logger_;
SCOPE_EXIT {
td::log_interface = td::default_log_interface;
};
std::vector<std::function<void()>> acts;
td::OptionsParser p;
p.set_description("validator or full node for TON network");
p.add_option('v', "verbosity", "set verbosity level", [&](td::Slice arg) {
int v = VERBOSITY_NAME(FATAL) + (td::to_integer<int>(arg));
SET_VERBOSITY_LEVEL(v);
return td::Status::OK();
});
p.add_option('h', "help", "prints_help", [&]() {
char b[10240];
td::StringBuilder sb(td::MutableSlice{b, 10000});
sb << p;
std::cout << sb.as_cslice().c_str();
std::exit(2);
return td::Status::OK();
});
p.add_option('C', "global-config", "file to read global config", [&](td::Slice fname) {
acts.push_back(
[&x, fname = fname.str()]() { td::actor::send_closure(x, &ValidatorEngine::set_global_config, fname); });
return td::Status::OK();
});
p.add_option('c', "local-config", "file to read local config", [&](td::Slice fname) {
acts.push_back(
[&x, fname = fname.str()]() { td::actor::send_closure(x, &ValidatorEngine::set_local_config, fname); });
return td::Status::OK();
});
p.add_option('I', "ip", "ip:port of instance", [&](td::Slice arg) {
td::IPAddress addr;
TRY_STATUS(addr.init_host_port(arg.str()));
acts.push_back([&x, addr]() { td::actor::send_closure(x, &ValidatorEngine::add_ip, addr); });
return td::Status::OK();
});
p.add_option('D', "db", "root for dbs", [&](td::Slice fname) {
acts.push_back([&x, fname = fname.str()]() { td::actor::send_closure(x, &ValidatorEngine::set_db_root, fname); });
return td::Status::OK();
});
p.add_option('f', "fift-dir", "directory with fift scripts", [&](td::Slice fname) {
acts.push_back([&x, fname = fname.str()]() { td::actor::send_closure(x, &ValidatorEngine::set_fift_dir, fname); });
return td::Status::OK();
});
p.add_option('F', "filedb-depth",
"depth of autodirs for blocks, proofs, etc. Default value is 2. You need to clear the "
"database, if you need to change this option",
[&](td::Slice fname) {
acts.push_back([&x, fname = fname.str()]() {
td::actor::send_closure(x, &ValidatorEngine::set_db_depth, td::to_integer<td::uint32>(fname));
});
return td::Status::OK();
});
p.add_option('d', "daemonize", "set SIGHUP", [&]() {
#if TD_DARWIN || TD_LINUX
close(0);
setsid();
#endif
td::set_signal_handler(td::SignalType::HangUp, force_rotate_logs).ensure();
return td::Status::OK();
});
p.add_option('l', "logname", "log to file", [&](td::Slice fname) {
logger_ = td::TsFileLog::create(fname.str()).move_as_ok();
td::log_interface = logger_.get();
return td::Status::OK();
});
p.add_option('s', "state-ttl", "state will be gc'd after this time (in seconds) default=3600", [&](td::Slice fname) {
auto v = td::to_double(fname);
acts.push_back([&x, v]() { td::actor::send_closure(x, &ValidatorEngine::set_state_ttl, v); });
return td::Status::OK();
});
p.add_option('b', "block-ttl", "blocks will be gc'd after this time (in seconds) default=7*86400",
[&](td::Slice fname) {
auto v = td::to_double(fname);
acts.push_back([&x, v]() { td::actor::send_closure(x, &ValidatorEngine::set_block_ttl, v); });
return td::Status::OK();
});
p.add_option('A', "archive-ttl", "archived blocks will be deleted after this time (in seconds) default=365*86400",
[&](td::Slice fname) {
auto v = td::to_double(fname);
acts.push_back([&x, v]() { td::actor::send_closure(x, &ValidatorEngine::set_archive_ttl, v); });
return td::Status::OK();
});
p.add_option('K', "key-proof-ttl", "key blocks will be deleted after this time (in seconds) default=365*86400*10",
[&](td::Slice fname) {
auto v = td::to_double(fname);
acts.push_back([&x, v]() { td::actor::send_closure(x, &ValidatorEngine::set_key_proof_ttl, v); });
return td::Status::OK();
});
p.add_option('S', "sync-before", "in initial sync download all blocks for last given seconds default=3600",
[&](td::Slice fname) {
auto v = td::to_double(fname);
acts.push_back([&x, v]() { td::actor::send_closure(x, &ValidatorEngine::set_sync_ttl, v); });
return td::Status::OK();
});
td::uint32 threads = 7;
p.add_option('t', "threads", PSTRING() << "number of threads (default=" << threads << ")", [&](td::Slice fname) {
td::int32 v;
try {
v = std::stoi(fname.str());
} catch (...) {
return td::Status::Error(ton::ErrorCode::error, "bad value for --threads: not a number");
}
if (v < 1 || v > 256) {
return td::Status::Error(ton::ErrorCode::error, "bad value for --threads: should be in range [1..256]");
}
threads = v;
return td::Status::OK();
});
p.add_option('u', "user", "change user", [&](td::Slice user) { return td::change_user(user); });
auto S = p.run(argc, argv);
if (S.is_error()) {
LOG(ERROR) << "failed to parse options: " << S.move_as_error();
std::_Exit(2);
}
td::set_runtime_signal_handler(1, need_stats).ensure();
td::actor::Scheduler scheduler({threads});
scheduler.run_in_context([&] {
CHECK(vm::init_op_cp0());
x = td::actor::create_actor<ValidatorEngine>("validator-engine");
for (auto &act : acts) {
act();
}
acts.clear();
td::actor::send_closure(x, &ValidatorEngine::run);
});
while (scheduler.run(1)) {
if (need_stats_flag.exchange(false)) {
dump_stats();
}
if (rotate_logs_flags.exchange(false)) {
if (td::log_interface) {
td::log_interface->rotate();
}
}
}
return 0;
}