mirror of
https://github.com/ton-blockchain/ton
synced 2025-03-09 15:40:10 +00:00
332 lines
12 KiB
C++
332 lines
12 KiB
C++
/*
|
|
This file is part of TON Blockchain Library.
|
|
|
|
TON Blockchain Library is free software: you can redistribute it and/or modify
|
|
it under the terms of the GNU Lesser General Public License as published by
|
|
the Free Software Foundation, either version 2 of the License, or
|
|
(at your option) any later version.
|
|
|
|
TON Blockchain Library is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU Lesser General Public License for more details.
|
|
|
|
You should have received a copy of the GNU Lesser General Public License
|
|
along with TON Blockchain Library. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
Copyright 2017-2019 Telegram Systems LLP
|
|
*/
|
|
#include "accept-block.hpp"
|
|
#include "adnl/utils.hpp"
|
|
#include "validator/interfaces/validator-manager.h"
|
|
#include "ton/ton-tl.hpp"
|
|
#include "ton/ton-io.hpp"
|
|
|
|
#include "validator/fabric.h"
|
|
#include "validator/invariants.hpp"
|
|
|
|
#include "top-shard-description.hpp"
|
|
|
|
namespace ton {
|
|
|
|
namespace validator {
|
|
|
|
namespace dummy0 {
|
|
|
|
void AcceptBlockQuery::abort_query(td::Status reason) {
|
|
if (promise_) {
|
|
LOG(WARNING) << "aborting accept block " << id_ << " query: " << reason;
|
|
promise_.set_error(std::move(reason));
|
|
}
|
|
stop();
|
|
}
|
|
|
|
void AcceptBlockQuery::finish_query() {
|
|
ValidatorInvariants::check_post_accept(handle_);
|
|
if (promise_) {
|
|
promise_.set_value(td::Unit());
|
|
}
|
|
stop();
|
|
}
|
|
|
|
void AcceptBlockQuery::alarm() {
|
|
abort_query(td::Status::Error(ErrorCode::timeout, "timeout"));
|
|
}
|
|
|
|
void AcceptBlockQuery::start_up() {
|
|
alarm_timestamp() = timeout_;
|
|
|
|
auto P = td::PromiseCreator::lambda([SelfId = actor_id(this)](td::Result<BlockHandle> R) {
|
|
if (R.is_error()) {
|
|
td::actor::send_closure(SelfId, &AcceptBlockQuery::abort_query, R.move_as_error());
|
|
} else {
|
|
td::actor::send_closure(SelfId, &AcceptBlockQuery::got_block_handle, R.move_as_ok());
|
|
}
|
|
});
|
|
|
|
td::actor::send_closure(manager_, &ValidatorManager::get_block_handle, id_, true, std::move(P));
|
|
}
|
|
|
|
void AcceptBlockQuery::got_block_handle(BlockHandle handle) {
|
|
handle_ = std::move(handle);
|
|
if (handle_->processed() && handle_->received() && handle_->received_state() && handle_->inited_signatures() &&
|
|
handle_->inited_split_after() && handle_->inited_merge_before() && handle_->inited_prev() &&
|
|
(id_.is_masterchain() ? handle_->inited_proof() : handle_->inited_proof_link())) {
|
|
send_block_description();
|
|
return;
|
|
}
|
|
if (data_.not_null() && !handle_->received()) {
|
|
auto P = td::PromiseCreator::lambda([SelfId = actor_id(this)](td::Result<td::Unit> R) {
|
|
if (R.is_error()) {
|
|
td::actor::send_closure(SelfId, &AcceptBlockQuery::abort_query, R.move_as_error());
|
|
} else {
|
|
td::actor::send_closure(SelfId, &AcceptBlockQuery::written_block_data);
|
|
}
|
|
});
|
|
|
|
td::actor::send_closure(manager_, &ValidatorManager::set_block_data, handle_, data_, std::move(P));
|
|
} else {
|
|
written_block_data();
|
|
}
|
|
}
|
|
|
|
void AcceptBlockQuery::written_block_data() {
|
|
if (handle_->inited_signatures()) {
|
|
written_block_signatures();
|
|
return;
|
|
}
|
|
auto P = td::PromiseCreator::lambda([SelfId = actor_id(this)](td::Result<td::Unit> R) {
|
|
if (R.is_error()) {
|
|
td::actor::send_closure(SelfId, &AcceptBlockQuery::abort_query, R.move_as_error());
|
|
} else {
|
|
td::actor::send_closure(SelfId, &AcceptBlockQuery::written_block_signatures);
|
|
}
|
|
});
|
|
|
|
td::actor::send_closure(manager_, &ValidatorManager::set_block_signatures, handle_, signatures_, std::move(P));
|
|
}
|
|
|
|
void AcceptBlockQuery::written_block_signatures() {
|
|
if (prev_.size() == 2) {
|
|
handle_->set_merge(true);
|
|
} else {
|
|
handle_->set_merge(false);
|
|
}
|
|
|
|
for (auto &p : prev_) {
|
|
handle_->set_prev(p);
|
|
}
|
|
|
|
if (handle_->need_flush()) {
|
|
auto P = td::PromiseCreator::lambda([SelfId = actor_id(this)](td::Result<td::Unit> R) {
|
|
if (R.is_error()) {
|
|
td::actor::send_closure(SelfId, &AcceptBlockQuery::abort_query, R.move_as_error());
|
|
} else {
|
|
td::actor::send_closure(SelfId, &AcceptBlockQuery::written_block_info);
|
|
}
|
|
});
|
|
|
|
handle_->flush(manager_, handle_, std::move(P));
|
|
} else {
|
|
written_block_info();
|
|
}
|
|
}
|
|
|
|
void AcceptBlockQuery::written_block_info() {
|
|
LOG(WARNING) << "written block info";
|
|
if (data_.not_null()) {
|
|
auto P = td::PromiseCreator::lambda([SelfId = actor_id(this)](td::Result<td::Ref<ShardState>> R) {
|
|
if (R.is_error()) {
|
|
td::actor::send_closure(SelfId, &AcceptBlockQuery::abort_query, R.move_as_error());
|
|
} else {
|
|
td::actor::send_closure(SelfId, &AcceptBlockQuery::got_prev_state, R.move_as_ok());
|
|
}
|
|
});
|
|
|
|
CHECK(prev_.size() <= 2);
|
|
td::actor::send_closure(manager_, &ValidatorManager::wait_prev_block_state, handle_, timeout_, std::move(P));
|
|
} else {
|
|
auto P = td::PromiseCreator::lambda([SelfId = actor_id(this)](td::Result<td::Ref<BlockData>> R) {
|
|
if (R.is_error()) {
|
|
td::actor::send_closure(SelfId, &AcceptBlockQuery::failed_to_get_block_candidate);
|
|
} else {
|
|
td::actor::send_closure(SelfId, &AcceptBlockQuery::got_block_data, R.move_as_ok());
|
|
}
|
|
});
|
|
|
|
td::actor::send_closure(manager_, &ValidatorManager::get_block_candidate_data_from_db, id_, std::move(P));
|
|
}
|
|
}
|
|
|
|
void AcceptBlockQuery::failed_to_get_block_candidate() {
|
|
auto P = td::PromiseCreator::lambda([SelfId = actor_id(this)](td::Result<td::Ref<BlockData>> R) {
|
|
if (R.is_error()) {
|
|
td::actor::send_closure(SelfId, &AcceptBlockQuery::abort_query, R.move_as_error());
|
|
} else {
|
|
td::actor::send_closure(SelfId, &AcceptBlockQuery::got_block_data, R.move_as_ok());
|
|
}
|
|
});
|
|
|
|
td::actor::send_closure(manager_, &ValidatorManager::wait_block_data, handle_, timeout_, std::move(P));
|
|
}
|
|
|
|
void AcceptBlockQuery::got_block_data(td::Ref<BlockData> data) {
|
|
data_ = std::move(data);
|
|
if (handle_->received()) {
|
|
written_block_info();
|
|
} else {
|
|
auto P = td::PromiseCreator::lambda([SelfId = actor_id(this)](td::Result<td::Unit> R) {
|
|
if (R.is_error()) {
|
|
td::actor::send_closure(SelfId, &AcceptBlockQuery::abort_query, R.move_as_error());
|
|
} else {
|
|
td::actor::send_closure(SelfId, &AcceptBlockQuery::written_block_data);
|
|
}
|
|
});
|
|
|
|
td::actor::send_closure(manager_, &ValidatorManager::set_block_data, handle_, data_, std::move(P));
|
|
}
|
|
}
|
|
|
|
void AcceptBlockQuery::got_prev_state(td::Ref<ShardState> state) {
|
|
LOG(WARNING) << "got prev state";
|
|
state_ = std::move(state);
|
|
|
|
state_.write().apply_block(id_, data_).ensure();
|
|
|
|
handle_->set_split(state_->before_split());
|
|
|
|
auto P = td::PromiseCreator::lambda([SelfId = actor_id(this)](td::Result<td::Unit> R) {
|
|
if (R.is_error()) {
|
|
td::actor::send_closure(SelfId, &AcceptBlockQuery::abort_query, R.move_as_error());
|
|
} else {
|
|
td::actor::send_closure(SelfId, &AcceptBlockQuery::written_state);
|
|
}
|
|
});
|
|
|
|
td::actor::send_closure(manager_, &ValidatorManager::set_block_state, handle_, state_, std::move(P));
|
|
}
|
|
|
|
void AcceptBlockQuery::written_state() {
|
|
LOG(WARNING) << "written state";
|
|
|
|
// generate proof
|
|
|
|
CHECK(prev_.size() == 1);
|
|
std::vector<tl_object_ptr<ton_api::tonNode_blockIdExt>> prev;
|
|
for (auto &p : prev_) {
|
|
prev.emplace_back(create_tl_block_id(p));
|
|
}
|
|
|
|
auto proof_link = create_tl_object<ton_api::test0_proofLink>(
|
|
create_tl_block_id(id_), std::move(prev), Bits256_2_UInt256(state_->root_hash()), handle_->split_after());
|
|
proof_link_ = create_proof_link(serialize_tl_object(proof_link, true)).move_as_ok();
|
|
|
|
if (id_.is_masterchain()) {
|
|
auto proof = create_tl_object<ton_api::test0_proof>(
|
|
std::move(proof_link), catchain_seqno_, validator_set_hash_,
|
|
fetch_tl_object<ton_api::test0_blockSignatures>(signatures_->serialize(), true).move_as_ok());
|
|
proof_ = create_proof(prev_[0], serialize_tl_object(proof, true)).move_as_ok();
|
|
}
|
|
|
|
if (handle_->id().is_masterchain()) {
|
|
CHECK(prev_.size() == 1);
|
|
auto P = td::PromiseCreator::lambda([SelfId = actor_id(this)](td::Result<td::Unit> R) {
|
|
if (R.is_error()) {
|
|
td::actor::send_closure(SelfId, &AcceptBlockQuery::abort_query, R.move_as_error());
|
|
} else {
|
|
td::actor::send_closure(SelfId, &AcceptBlockQuery::written_block_next);
|
|
}
|
|
});
|
|
|
|
td::actor::send_closure(manager_, &ValidatorManager::set_next_block, prev_[0], id_, std::move(P));
|
|
} else {
|
|
written_block_next();
|
|
}
|
|
}
|
|
|
|
void AcceptBlockQuery::written_block_next() {
|
|
auto P = td::PromiseCreator::lambda([SelfId = actor_id(this)](td::Result<td::Unit> R) {
|
|
if (R.is_error()) {
|
|
td::actor::send_closure(SelfId, &AcceptBlockQuery::abort_query, R.move_as_error());
|
|
} else {
|
|
td::actor::send_closure(SelfId, &AcceptBlockQuery::written_block_proof);
|
|
}
|
|
});
|
|
|
|
if (id_.is_masterchain()) {
|
|
td::actor::send_closure(manager_, &ValidatorManager::set_block_proof, handle_, proof_, std::move(P));
|
|
} else {
|
|
td::actor::send_closure(manager_, &ValidatorManager::set_block_proof_link, handle_, proof_link_, std::move(P));
|
|
}
|
|
}
|
|
|
|
void AcceptBlockQuery::written_block_proof() {
|
|
if (handle_->need_flush()) {
|
|
auto P = td::PromiseCreator::lambda([SelfId = actor_id(this)](td::Result<td::Unit> R) {
|
|
if (R.is_error()) {
|
|
td::actor::send_closure(SelfId, &AcceptBlockQuery::abort_query, R.move_as_error());
|
|
} else {
|
|
td::actor::send_closure(SelfId, &AcceptBlockQuery::written_block_info_2);
|
|
}
|
|
});
|
|
|
|
handle_->flush(manager_, handle_, std::move(P));
|
|
} else {
|
|
written_block_info_2();
|
|
}
|
|
}
|
|
|
|
void AcceptBlockQuery::written_block_info_2() {
|
|
auto P = td::PromiseCreator::lambda([SelfId = actor_id(this)](td::Result<td::Unit> R) {
|
|
if (R.is_error()) {
|
|
td::actor::send_closure(SelfId, &AcceptBlockQuery::abort_query, R.move_as_error());
|
|
} else {
|
|
td::actor::send_closure(SelfId, &AcceptBlockQuery::send_block_description);
|
|
}
|
|
});
|
|
|
|
td::actor::send_closure(manager_, &ValidatorManager::new_block, handle_, state_, std::move(P));
|
|
}
|
|
|
|
void AcceptBlockQuery::send_block_description() {
|
|
if (!handle_->id().is_masterchain()) {
|
|
bool after_split = prev_.size() == 1 && handle_->id().id.shard != prev_[0].id.shard;
|
|
if (after_split) {
|
|
CHECK(shard_parent(handle_->id().shard_full()) == prev_[0].shard_full());
|
|
}
|
|
bool after_merge = prev_.size() == 2;
|
|
|
|
auto desc = td::Ref<ShardTopBlockDescriptionImpl>{true,
|
|
handle_->id(),
|
|
after_split,
|
|
after_merge,
|
|
handle_->split_after(),
|
|
catchain_seqno_,
|
|
validator_set_hash_,
|
|
signatures_->serialize()};
|
|
td::actor::send_closure(manager_, &ValidatorManager::send_top_shard_block_description, std::move(desc));
|
|
}
|
|
finish_query();
|
|
}
|
|
|
|
AcceptBlockQuery::AcceptBlockQuery(BlockIdExt id, td::Ref<BlockData> data, std::vector<BlockIdExt> prev,
|
|
UnixTime catchain_seqno, td::uint32 validator_set_hash,
|
|
td::Ref<BlockSignatureSet> signatures, bool send_broadcast,
|
|
td::actor::ActorId<ValidatorManager> manager, td::Promise<td::Unit> promise)
|
|
: id_(id)
|
|
, data_(std::move(data))
|
|
, prev_(std::move(prev))
|
|
, catchain_seqno_(catchain_seqno)
|
|
, validator_set_hash_(validator_set_hash)
|
|
, signatures_(std::move(signatures))
|
|
, send_broadcast_(send_broadcast)
|
|
, manager_(manager)
|
|
, promise_(std::move(promise)) {
|
|
CHECK(prev_.size() > 0);
|
|
}
|
|
|
|
} // namespace dummy0
|
|
|
|
} // namespace validator
|
|
|
|
} // namespace ton
|