mirror of
https://github.com/ton-blockchain/ton
synced 2025-03-09 15:40:10 +00:00
1132 lines
57 KiB
C++
1132 lines
57 KiB
C++
/*
|
|
This file is part of TON Blockchain Library.
|
|
|
|
TON Blockchain Library is free software: you can redistribute it and/or modify
|
|
it under the terms of the GNU Lesser General Public License as published by
|
|
the Free Software Foundation, either version 2 of the License, or
|
|
(at your option) any later version.
|
|
|
|
TON Blockchain Library is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU Lesser General Public License for more details.
|
|
|
|
You should have received a copy of the GNU Lesser General Public License
|
|
along with TON Blockchain Library. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
Copyright 2017-2019 Telegram Systems LLP
|
|
*/
|
|
#pragma once
|
|
|
|
#include "td/utils/int_types.h"
|
|
#include "td/utils/buffer.h"
|
|
#include "adnl/utils.hpp"
|
|
#include "common/io.hpp"
|
|
|
|
#include "persistent-vector.h"
|
|
|
|
#include "validator-session-description.h"
|
|
|
|
#include "validator-session-types.h"
|
|
|
|
#include <limits>
|
|
|
|
namespace td {
|
|
|
|
td::StringBuilder& operator<<(td::StringBuilder& sb, const ton::ton_api::validatorSession_round_Message* message);
|
|
|
|
}
|
|
|
|
namespace ton {
|
|
|
|
namespace validatorsession {
|
|
|
|
using HashType = ValidatorSessionDescription::HashType;
|
|
|
|
constexpr int VERBOSITY_NAME(VALIDATOR_SESSION_WARNING) = verbosity_WARNING;
|
|
constexpr int VERBOSITY_NAME(VALIDATOR_SESSION_NOTICE) = verbosity_DEBUG;
|
|
constexpr int VERBOSITY_NAME(VALIDATOR_SESSION_INFO) = verbosity_DEBUG;
|
|
constexpr int VERBOSITY_NAME(VALIDATOR_SESSION_DEBUG) = verbosity_DEBUG;
|
|
constexpr int VERBOSITY_NAME(VALIDATOR_SESSION_EXTRA_DEBUG) = verbosity_DEBUG + 1;
|
|
|
|
struct SessionBlockCandidateSignature : public ValidatorSessionDescription::RootObject {
|
|
public:
|
|
static auto create_hash(ValidatorSessionDescription& desc, td::Slice data) {
|
|
auto obj = create_tl_object<ton_api::hashable_blockSignature>(desc.compute_hash(data));
|
|
return desc.compute_hash(serialize_tl_object(obj, true).as_slice());
|
|
}
|
|
|
|
static bool compare(const RootObject* r, td::Slice data, HashType hash) {
|
|
if (!r || r->get_size() < sizeof(SessionBlockCandidateSignature)) {
|
|
return false;
|
|
}
|
|
auto R = static_cast<const SessionBlockCandidateSignature*>(r);
|
|
return R->hash_ == hash && R->data_.ubegin() == data.ubegin() && R->data_.size() == data.size();
|
|
}
|
|
|
|
static auto lookup(ValidatorSessionDescription& desc, td::Slice data, HashType hash, bool temp) {
|
|
auto r = desc.get_by_hash(hash, temp);
|
|
if (compare(r, data, hash)) {
|
|
desc.on_reuse();
|
|
return static_cast<const SessionBlockCandidateSignature*>(r);
|
|
}
|
|
return static_cast<const SessionBlockCandidateSignature*>(nullptr);
|
|
}
|
|
static SessionBlockCandidateSignature* create(ValidatorSessionDescription& desc, td::BufferSlice value) {
|
|
auto hash = create_hash(desc, value.as_slice());
|
|
auto d = static_cast<td::uint8*>(desc.alloc(value.size(), 8, false));
|
|
td::MutableSlice s{d, value.size()};
|
|
s.copy_from(value.as_slice());
|
|
return new (desc, true) SessionBlockCandidateSignature{desc, s, hash};
|
|
}
|
|
static const SessionBlockCandidateSignature* move_to_persistent(ValidatorSessionDescription& desc,
|
|
const SessionBlockCandidateSignature* b) {
|
|
if (desc.is_persistent(b)) {
|
|
return b;
|
|
}
|
|
CHECK(desc.is_persistent(b->data_.ubegin()));
|
|
auto r = lookup(desc, b->data_, b->hash_, false);
|
|
if (r) {
|
|
return r;
|
|
}
|
|
return new (desc, false) SessionBlockCandidateSignature{desc, b->data_, b->hash_};
|
|
}
|
|
static const SessionBlockCandidateSignature* merge(ValidatorSessionDescription& desc,
|
|
const SessionBlockCandidateSignature* l,
|
|
const SessionBlockCandidateSignature* r);
|
|
SessionBlockCandidateSignature(ValidatorSessionDescription& desc, td::Slice data, HashType hash)
|
|
: RootObject(sizeof(SessionBlockCandidateSignature)), data_{data}, hash_(std::move(hash)) {
|
|
desc.update_hash(this, hash_);
|
|
}
|
|
td::BufferSlice value() const {
|
|
return td::BufferSlice{data_};
|
|
}
|
|
td::Slice as_slice() const {
|
|
return data_;
|
|
}
|
|
auto get_hash(ValidatorSessionDescription& desc) const {
|
|
return hash_;
|
|
}
|
|
|
|
private:
|
|
const td::Slice data_;
|
|
const HashType hash_;
|
|
};
|
|
|
|
using SessionBlockCandidateSignatureVector = CntVector<const SessionBlockCandidateSignature*>;
|
|
|
|
class SentBlock : public ValidatorSessionDescription::RootObject {
|
|
public:
|
|
static HashType create_hash(ValidatorSessionDescription& desc, td::uint32 src_idx, ValidatorSessionRootHash root_hash,
|
|
ValidatorSessionFileHash file_hash,
|
|
ValidatorSessionCollatedDataFileHash collated_data_file_hash) {
|
|
auto obj = create_tl_object<ton_api::hashable_sentBlock>(src_idx, get_vs_hash(desc, root_hash),
|
|
get_vs_hash(desc, file_hash),
|
|
get_vs_hash(desc, collated_data_file_hash));
|
|
return desc.compute_hash(serialize_tl_object(obj, true).as_slice());
|
|
}
|
|
static bool compare(const RootObject* root_object, td::uint32 src_idx, const ValidatorSessionRootHash& root_hash,
|
|
const ValidatorSessionFileHash& file_hash,
|
|
const ValidatorSessionCollatedDataFileHash& collated_data_file_hash, HashType hash) {
|
|
if (!root_object || root_object->get_size() < sizeof(SentBlock)) {
|
|
return false;
|
|
}
|
|
auto obj = static_cast<const SentBlock*>(root_object);
|
|
return obj->src_idx_ == src_idx && obj->root_hash_ == root_hash && obj->file_hash_ == file_hash &&
|
|
obj->collated_data_file_hash_ == collated_data_file_hash && obj->hash_ == hash;
|
|
}
|
|
static auto lookup(ValidatorSessionDescription& desc, td::uint32 src_idx, const ValidatorSessionRootHash& root_hash,
|
|
const ValidatorSessionFileHash& file_hash,
|
|
const ValidatorSessionCollatedDataFileHash& collated_data_file_hash, HashType hash, bool temp) {
|
|
auto r = desc.get_by_hash(hash, temp);
|
|
if (compare(r, src_idx, root_hash, file_hash, collated_data_file_hash, hash)) {
|
|
desc.on_reuse();
|
|
return static_cast<const SentBlock*>(r);
|
|
}
|
|
return static_cast<const SentBlock*>(nullptr);
|
|
}
|
|
static const SentBlock* create(ValidatorSessionDescription& desc, td::uint32 src_idx,
|
|
const ValidatorSessionRootHash& root_hash, const ValidatorSessionFileHash& file_hash,
|
|
const ValidatorSessionCollatedDataFileHash& collated_data_file_hash) {
|
|
auto hash = create_hash(desc, src_idx, root_hash, file_hash, collated_data_file_hash);
|
|
auto r = lookup(desc, src_idx, root_hash, file_hash, collated_data_file_hash, hash, true);
|
|
if (r) {
|
|
return r;
|
|
}
|
|
|
|
auto candidate_id = desc.candidate_id(src_idx, root_hash, file_hash, collated_data_file_hash);
|
|
|
|
return new (desc, true) SentBlock{desc, src_idx, root_hash, file_hash, collated_data_file_hash, candidate_id, hash};
|
|
}
|
|
static const SentBlock* create(ValidatorSessionDescription& desc, const ValidatorSessionCandidateId& zero) {
|
|
CHECK(zero.is_zero());
|
|
auto hash = create_hash(desc, 0, ValidatorSessionRootHash::zero(), ValidatorSessionFileHash::zero(),
|
|
ValidatorSessionCollatedDataFileHash::zero());
|
|
|
|
return new (desc, true) SentBlock{desc,
|
|
0,
|
|
ValidatorSessionRootHash::zero(),
|
|
ValidatorSessionFileHash::zero(),
|
|
ValidatorSessionCollatedDataFileHash::zero(),
|
|
zero,
|
|
hash};
|
|
}
|
|
static const SentBlock* move_to_persistent(ValidatorSessionDescription& desc, const SentBlock* b) {
|
|
if (desc.is_persistent(b)) {
|
|
return b;
|
|
}
|
|
auto r = lookup(desc, b->src_idx_, b->root_hash_, b->file_hash_, b->collated_data_file_hash_, b->hash_, false);
|
|
if (r) {
|
|
return r;
|
|
}
|
|
|
|
return new (desc, false) SentBlock{
|
|
desc, b->src_idx_, b->root_hash_, b->file_hash_, b->collated_data_file_hash_, b->candidate_id_, b->hash_};
|
|
}
|
|
SentBlock(ValidatorSessionDescription& desc, td::uint32 src_idx, ValidatorSessionRootHash root_hash,
|
|
ValidatorSessionFileHash file_hash, ValidatorSessionCollatedDataFileHash collated_data_file_hash,
|
|
ValidatorSessionCandidateId candidate_id, HashType hash)
|
|
: RootObject(sizeof(SentBlock))
|
|
, src_idx_(src_idx)
|
|
, root_hash_(std::move(root_hash))
|
|
, file_hash_(std::move(file_hash))
|
|
, collated_data_file_hash_(std::move(collated_data_file_hash))
|
|
, candidate_id_(candidate_id)
|
|
, hash_(std::move(hash)) {
|
|
desc.update_hash(this, hash_);
|
|
}
|
|
auto get_src_idx() const {
|
|
return src_idx_;
|
|
}
|
|
auto get_root_hash() const {
|
|
return root_hash_;
|
|
}
|
|
auto get_file_hash() const {
|
|
return file_hash_;
|
|
}
|
|
auto get_collated_data_file_hash() const {
|
|
return collated_data_file_hash_;
|
|
}
|
|
static ValidatorSessionCandidateId get_block_id(const SentBlock* block) {
|
|
return block ? block->candidate_id_ : skip_round_candidate_id();
|
|
}
|
|
HashType get_hash(ValidatorSessionDescription& desc) const {
|
|
return hash_;
|
|
}
|
|
bool operator<(const SentBlock& block) const {
|
|
if (src_idx_ < block.src_idx_) {
|
|
return true;
|
|
}
|
|
if (src_idx_ > block.src_idx_) {
|
|
return false;
|
|
}
|
|
if (candidate_id_ < block.candidate_id_) {
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
struct Compare {
|
|
bool operator()(const SentBlock* a, const SentBlock* b) const {
|
|
return *a < *b;
|
|
}
|
|
};
|
|
|
|
private:
|
|
const td::uint32 src_idx_;
|
|
const ValidatorSessionRootHash root_hash_;
|
|
const ValidatorSessionFileHash file_hash_;
|
|
const ValidatorSessionCollatedDataFileHash collated_data_file_hash_;
|
|
const ValidatorSessionCandidateId candidate_id_;
|
|
const HashType hash_;
|
|
};
|
|
|
|
class SessionBlockCandidate : public ValidatorSessionDescription::RootObject {
|
|
public:
|
|
static HashType create_hash(ValidatorSessionDescription& desc, HashType block, HashType approved) {
|
|
auto obj = create_tl_object<ton_api::hashable_blockCandidate>(block, approved);
|
|
return desc.compute_hash(serialize_tl_object(obj, true).as_slice());
|
|
}
|
|
static bool compare(const RootObject* r, const SentBlock* block, const SessionBlockCandidateSignatureVector* approved,
|
|
HashType hash) {
|
|
if (!r || r->get_size() < sizeof(SessionBlockCandidate)) {
|
|
return false;
|
|
}
|
|
auto R = static_cast<const SessionBlockCandidate*>(r);
|
|
return R->block_ == block && R->approved_by_ == approved && R->hash_ == hash;
|
|
}
|
|
static auto lookup(ValidatorSessionDescription& desc, const SentBlock* block,
|
|
const SessionBlockCandidateSignatureVector* approved, HashType hash, bool temp) {
|
|
auto r = desc.get_by_hash(hash, temp);
|
|
if (compare(r, block, approved, hash)) {
|
|
desc.on_reuse();
|
|
return static_cast<const SessionBlockCandidate*>(r);
|
|
}
|
|
return static_cast<const SessionBlockCandidate*>(nullptr);
|
|
}
|
|
static const SessionBlockCandidate* create(ValidatorSessionDescription& desc, const SentBlock* block,
|
|
const SessionBlockCandidateSignatureVector* approved) {
|
|
auto hash = create_hash(desc, get_vs_hash(desc, block), get_vs_hash(desc, approved));
|
|
|
|
auto r = lookup(desc, block, approved, hash, true);
|
|
if (r) {
|
|
return r;
|
|
}
|
|
|
|
return new (desc, true) SessionBlockCandidate(desc, block, approved, hash);
|
|
}
|
|
static const SessionBlockCandidate* create(ValidatorSessionDescription& desc, const SentBlock* block) {
|
|
std::vector<const SessionBlockCandidateSignature*> v;
|
|
v.resize(desc.get_total_nodes(), nullptr);
|
|
auto vec = SessionBlockCandidateSignatureVector::create(desc, std::move(v));
|
|
return create(desc, block, vec);
|
|
}
|
|
static const SessionBlockCandidate* move_to_persistent(ValidatorSessionDescription& desc,
|
|
const SessionBlockCandidate* b) {
|
|
if (desc.is_persistent(b)) {
|
|
return b;
|
|
}
|
|
auto block = SentBlock::move_to_persistent(desc, b->block_);
|
|
auto approved = SessionBlockCandidateSignatureVector::move_to_persistent(desc, b->approved_by_);
|
|
auto r = lookup(desc, block, approved, b->hash_, false);
|
|
if (r) {
|
|
return r;
|
|
}
|
|
|
|
return new (desc, false) SessionBlockCandidate{desc, block, approved, b->hash_};
|
|
}
|
|
SessionBlockCandidate(ValidatorSessionDescription& desc, const SentBlock* block,
|
|
const SessionBlockCandidateSignatureVector* approved, HashType hash)
|
|
: RootObject{sizeof(SessionBlockCandidate)}, block_(block), approved_by_(approved), hash_(hash) {
|
|
desc.update_hash(this, hash_);
|
|
}
|
|
static const SessionBlockCandidate* merge(ValidatorSessionDescription& desc, const SessionBlockCandidate* l,
|
|
const SessionBlockCandidate* r);
|
|
auto get_block() const {
|
|
return block_;
|
|
}
|
|
auto get_id() const {
|
|
return SentBlock::get_block_id(block_);
|
|
}
|
|
auto get_src_idx() const {
|
|
return block_ ? block_->get_src_idx() : std::numeric_limits<td::uint32>::max();
|
|
}
|
|
bool check_block_is_approved_by(td::uint32 src_idx) const {
|
|
return approved_by_->at(src_idx);
|
|
}
|
|
bool check_block_is_approved(ValidatorSessionDescription& desc) const;
|
|
auto get_hash(ValidatorSessionDescription& desc) const {
|
|
return hash_;
|
|
}
|
|
auto get_approvers_list() const {
|
|
return approved_by_;
|
|
}
|
|
static const SessionBlockCandidate* push(ValidatorSessionDescription& desc, const SessionBlockCandidate* state,
|
|
td::uint32 src_idx, const SessionBlockCandidateSignature* sig) {
|
|
CHECK(state);
|
|
if (state->approved_by_->at(src_idx)) {
|
|
return state;
|
|
}
|
|
return create(desc, state->block_,
|
|
SessionBlockCandidateSignatureVector::change(desc, state->approved_by_, src_idx, sig));
|
|
}
|
|
class Compare {
|
|
public:
|
|
bool operator()(const SessionBlockCandidate* l, const SessionBlockCandidate* r) {
|
|
return l->get_id() < r->get_id();
|
|
}
|
|
};
|
|
|
|
private:
|
|
const SentBlock* block_;
|
|
const SessionBlockCandidateSignatureVector* approved_by_;
|
|
const HashType hash_;
|
|
};
|
|
|
|
class SessionVoteCandidate : public ValidatorSessionDescription::RootObject {
|
|
public:
|
|
static HashType create_hash(ValidatorSessionDescription& desc, HashType block, HashType voted) {
|
|
auto obj = create_tl_object<ton_api::hashable_blockVoteCandidate>(block, voted);
|
|
return desc.compute_hash(serialize_tl_object(obj, true).as_slice());
|
|
}
|
|
static bool compare(const RootObject* r, const SentBlock* block, const CntVector<bool>* voted, HashType hash) {
|
|
if (!r || r->get_size() < sizeof(SessionVoteCandidate)) {
|
|
return false;
|
|
}
|
|
auto R = static_cast<const SessionVoteCandidate*>(r);
|
|
return R->block_ == block && R->voted_by_ == voted && R->hash_ == hash;
|
|
}
|
|
static auto lookup(ValidatorSessionDescription& desc, const SentBlock* block, const CntVector<bool>* voted,
|
|
HashType hash, bool temp) {
|
|
auto r = desc.get_by_hash(hash, temp);
|
|
if (compare(r, block, voted, hash)) {
|
|
desc.on_reuse();
|
|
return static_cast<const SessionVoteCandidate*>(r);
|
|
}
|
|
return static_cast<const SessionVoteCandidate*>(nullptr);
|
|
}
|
|
static const SessionVoteCandidate* create(ValidatorSessionDescription& desc, const SentBlock* block,
|
|
const CntVector<bool>* voted) {
|
|
auto hash = create_hash(desc, get_vs_hash(desc, block), get_vs_hash(desc, voted));
|
|
|
|
auto r = lookup(desc, block, voted, hash, true);
|
|
if (r) {
|
|
return r;
|
|
}
|
|
|
|
return new (desc, true) SessionVoteCandidate(desc, block, voted, hash);
|
|
}
|
|
static const SessionVoteCandidate* create(ValidatorSessionDescription& desc, const SentBlock* block) {
|
|
std::vector<bool> v;
|
|
v.resize(desc.get_total_nodes(), false);
|
|
auto vec = CntVector<bool>::create(desc, std::move(v));
|
|
return create(desc, block, vec);
|
|
}
|
|
static const SessionVoteCandidate* move_to_persistent(ValidatorSessionDescription& desc,
|
|
const SessionVoteCandidate* b) {
|
|
if (desc.is_persistent(b)) {
|
|
return b;
|
|
}
|
|
auto block = SentBlock::move_to_persistent(desc, b->block_);
|
|
auto voted = CntVector<bool>::move_to_persistent(desc, b->voted_by_);
|
|
auto r = lookup(desc, block, voted, b->hash_, false);
|
|
if (r) {
|
|
return r;
|
|
}
|
|
|
|
return new (desc, false) SessionVoteCandidate{desc, block, voted, b->hash_};
|
|
}
|
|
SessionVoteCandidate(ValidatorSessionDescription& desc, const SentBlock* block, const CntVector<bool>* voted,
|
|
HashType hash)
|
|
: RootObject{sizeof(SessionVoteCandidate)}, block_(block), voted_by_(voted), hash_(hash) {
|
|
desc.update_hash(this, hash_);
|
|
}
|
|
static const SessionVoteCandidate* merge(ValidatorSessionDescription& desc, const SessionVoteCandidate* l,
|
|
const SessionVoteCandidate* r);
|
|
auto get_block() const {
|
|
return block_;
|
|
}
|
|
auto get_id() const {
|
|
return SentBlock::get_block_id(block_);
|
|
}
|
|
auto get_src_idx() const {
|
|
return block_ ? block_->get_src_idx() : std::numeric_limits<td::uint32>::max();
|
|
}
|
|
bool check_block_is_voted_by(td::uint32 src_idx) const {
|
|
return voted_by_->at(src_idx);
|
|
}
|
|
bool check_block_is_voted(ValidatorSessionDescription& desc) const;
|
|
auto get_hash(ValidatorSessionDescription& desc) const {
|
|
return hash_;
|
|
}
|
|
auto get_voters_list() const {
|
|
return voted_by_;
|
|
}
|
|
static const SessionVoteCandidate* push(ValidatorSessionDescription& desc, const SessionVoteCandidate* state,
|
|
td::uint32 src_idx) {
|
|
CHECK(state);
|
|
if (state->voted_by_->at(src_idx)) {
|
|
return state;
|
|
}
|
|
return create(desc, state->block_, CntVector<bool>::change(desc, state->voted_by_, src_idx, true));
|
|
}
|
|
class Compare {
|
|
public:
|
|
bool operator()(const SessionVoteCandidate* l, const SessionVoteCandidate* r) {
|
|
return l->get_id() < r->get_id();
|
|
}
|
|
};
|
|
|
|
private:
|
|
const SentBlock* block_;
|
|
const CntVector<bool>* voted_by_;
|
|
const HashType hash_;
|
|
};
|
|
|
|
//using SentBlockVector = CntSortedVector<const SentBlock*, SentBlock::Compare>;
|
|
|
|
class ValidatorSessionRoundState;
|
|
class ValidatorSessionOldRoundState : public ValidatorSessionDescription::RootObject {
|
|
public:
|
|
static HashType create_hash(ValidatorSessionDescription& desc, td::uint32 seqno, HashType block, HashType signatures,
|
|
HashType approve_signatures) {
|
|
auto obj =
|
|
create_tl_object<ton_api::hashable_validatorSessionOldRound>(seqno, block, signatures, approve_signatures);
|
|
return desc.compute_hash(serialize_tl_object(obj, true).as_slice());
|
|
}
|
|
static bool compare(const RootObject* r, td::uint32 seqno, const SentBlock* block,
|
|
const SessionBlockCandidateSignatureVector* signatures,
|
|
const SessionBlockCandidateSignatureVector* approve_signatures, HashType hash) {
|
|
if (!r || r->get_size() < sizeof(ValidatorSessionOldRoundState)) {
|
|
return false;
|
|
}
|
|
auto R = static_cast<const ValidatorSessionOldRoundState*>(r);
|
|
return R->seqno_ == seqno && R->block_ == block && R->signatures_ == signatures &&
|
|
R->approve_signatures_ == approve_signatures && R->hash_ == hash;
|
|
}
|
|
static auto lookup(ValidatorSessionDescription& desc, td::uint32 seqno, const SentBlock* block,
|
|
const SessionBlockCandidateSignatureVector* signatures,
|
|
const SessionBlockCandidateSignatureVector* approve_signatures, HashType hash, bool temp) {
|
|
auto r = desc.get_by_hash(hash, temp);
|
|
if (compare(r, seqno, block, signatures, approve_signatures, hash)) {
|
|
desc.on_reuse();
|
|
return static_cast<const ValidatorSessionOldRoundState*>(r);
|
|
}
|
|
return static_cast<const ValidatorSessionOldRoundState*>(nullptr);
|
|
}
|
|
static const ValidatorSessionOldRoundState* move_to_persistent(ValidatorSessionDescription& desc,
|
|
const ValidatorSessionOldRoundState* b) {
|
|
if (desc.is_persistent(b)) {
|
|
return b;
|
|
}
|
|
auto signatures = SessionBlockCandidateSignatureVector::move_to_persistent(desc, b->signatures_);
|
|
auto approve_signatures = SessionBlockCandidateSignatureVector::move_to_persistent(desc, b->approve_signatures_);
|
|
auto block = SentBlock::move_to_persistent(desc, b->block_);
|
|
auto r = lookup(desc, b->seqno_, block, signatures, approve_signatures, b->hash_, false);
|
|
if (r) {
|
|
return r;
|
|
}
|
|
|
|
return new (desc, false)
|
|
ValidatorSessionOldRoundState{desc, b->seqno_, block, signatures, approve_signatures, b->hash_};
|
|
}
|
|
static const ValidatorSessionOldRoundState* create(ValidatorSessionDescription& desc, td::uint32 seqno,
|
|
const SentBlock* block,
|
|
const SessionBlockCandidateSignatureVector* signatures,
|
|
const SessionBlockCandidateSignatureVector* approve_signatures) {
|
|
auto hash = create_hash(desc, seqno, get_vs_hash(desc, block), get_vs_hash(desc, signatures),
|
|
get_vs_hash(desc, approve_signatures));
|
|
|
|
auto r = lookup(desc, seqno, block, signatures, approve_signatures, hash, true);
|
|
if (r) {
|
|
return r;
|
|
}
|
|
|
|
return new (desc, true) ValidatorSessionOldRoundState(desc, seqno, block, signatures, approve_signatures, hash);
|
|
}
|
|
static const ValidatorSessionOldRoundState* create(ValidatorSessionDescription& desc,
|
|
const ValidatorSessionRoundState* round);
|
|
ValidatorSessionOldRoundState(ValidatorSessionDescription& desc, td::uint32 seqno, const SentBlock* block,
|
|
const SessionBlockCandidateSignatureVector* signatures,
|
|
const SessionBlockCandidateSignatureVector* approve_signatures, HashType hash)
|
|
: RootObject(sizeof(ValidatorSessionOldRoundState))
|
|
, seqno_(seqno)
|
|
, block_(std::move(block))
|
|
, signatures_(std::move(signatures))
|
|
, approve_signatures_(std::move(approve_signatures))
|
|
, hash_(std::move(hash)) {
|
|
desc.update_hash(this, hash_);
|
|
}
|
|
static const ValidatorSessionOldRoundState* merge(ValidatorSessionDescription& desc,
|
|
const ValidatorSessionOldRoundState* left,
|
|
const ValidatorSessionOldRoundState* right);
|
|
static const ValidatorSessionOldRoundState* merge(ValidatorSessionDescription& desc,
|
|
const ValidatorSessionOldRoundState* left,
|
|
const ValidatorSessionRoundState* right);
|
|
static const ValidatorSessionOldRoundState* action(ValidatorSessionDescription& desc,
|
|
const ValidatorSessionOldRoundState* state, td::uint32 src_idx,
|
|
td::uint32 att,
|
|
const ton_api::validatorSession_message_approvedBlock& action);
|
|
static const ValidatorSessionOldRoundState* action(ValidatorSessionDescription& desc,
|
|
const ValidatorSessionOldRoundState* state, td::uint32 src_idx,
|
|
td::uint32 att,
|
|
const ton_api::validatorSession_message_commit& action);
|
|
template <class T>
|
|
static const ValidatorSessionOldRoundState* action(ValidatorSessionDescription& desc,
|
|
const ValidatorSessionOldRoundState* state, td::uint32 src_idx,
|
|
td::uint32 att, const T& action) {
|
|
VLOG(VALIDATOR_SESSION_WARNING) << "[validator session][node " << desc.get_source_id(src_idx) << "][" << action
|
|
<< "]: invalid message in old round: expected APPROVE/COMMIT";
|
|
return state;
|
|
}
|
|
static const ValidatorSessionOldRoundState* action(ValidatorSessionDescription& desc,
|
|
const ValidatorSessionOldRoundState* state, td::uint32 src_idx,
|
|
td::uint32 att,
|
|
const ton_api::validatorSession_round_Message* action);
|
|
|
|
auto get_hash(ValidatorSessionDescription& desc) const {
|
|
return hash_;
|
|
}
|
|
|
|
auto get_block() const {
|
|
return block_;
|
|
}
|
|
|
|
auto get_signatures() const {
|
|
return signatures_;
|
|
}
|
|
|
|
auto get_approve_signatures() const {
|
|
return approve_signatures_;
|
|
}
|
|
|
|
auto get_seqno() const {
|
|
return seqno_;
|
|
}
|
|
|
|
bool check_block_is_signed_by(td::uint32 src_idx) const {
|
|
return signatures_->at(src_idx) != nullptr;
|
|
}
|
|
bool check_block_is_approved_by(td::uint32 src_idx) const {
|
|
return approve_signatures_->at(src_idx) != nullptr;
|
|
}
|
|
|
|
ValidatorSessionCandidateId get_block_id() const {
|
|
return SentBlock::get_block_id(block_);
|
|
}
|
|
|
|
private:
|
|
const td::uint32 seqno_;
|
|
const SentBlock* block_;
|
|
const SessionBlockCandidateSignatureVector* signatures_;
|
|
const SessionBlockCandidateSignatureVector* approve_signatures_;
|
|
const HashType hash_;
|
|
};
|
|
|
|
using VoteVector = CntSortedVector<const SessionVoteCandidate*, SessionVoteCandidate::Compare>;
|
|
using ApproveVector = CntSortedVector<const SessionBlockCandidate*, SessionBlockCandidate::Compare>;
|
|
|
|
class ValidatorSessionRoundAttemptState : public ValidatorSessionDescription::RootObject {
|
|
public:
|
|
static HashType create_hash(ValidatorSessionDescription& desc, td::uint32 seqno, HashType votes,
|
|
HashType precommitted, bool vote_for_inited, HashType vote_for) {
|
|
auto obj = create_tl_object<ton_api::hashable_validatorSessionRoundAttempt>(seqno, votes, precommitted,
|
|
vote_for_inited, vote_for);
|
|
return desc.compute_hash(serialize_tl_object(obj, true).as_slice());
|
|
}
|
|
static bool compare(const RootObject* r, td::uint32 seqno, const VoteVector* votes,
|
|
const CntVector<bool>* precommitted, const SentBlock* vote_for, bool vote_for_inited,
|
|
HashType hash) {
|
|
if (!r || r->get_size() < sizeof(ValidatorSessionRoundAttemptState)) {
|
|
return false;
|
|
}
|
|
auto R = static_cast<const ValidatorSessionRoundAttemptState*>(r);
|
|
return R->seqno_ == seqno && R->votes_ == votes && R->precommitted_ == precommitted && R->vote_for_ == vote_for &&
|
|
R->vote_for_inited_ == vote_for_inited && R->hash_ == hash;
|
|
}
|
|
static auto lookup(ValidatorSessionDescription& desc, td::uint32 seqno, const VoteVector* votes,
|
|
const CntVector<bool>* precommitted, const SentBlock* vote_for, bool vote_for_inited,
|
|
HashType hash, bool temp) {
|
|
auto r = desc.get_by_hash(hash, temp);
|
|
if (compare(r, seqno, votes, precommitted, vote_for, vote_for_inited, hash)) {
|
|
desc.on_reuse();
|
|
return static_cast<const ValidatorSessionRoundAttemptState*>(r);
|
|
}
|
|
return static_cast<const ValidatorSessionRoundAttemptState*>(nullptr);
|
|
}
|
|
static const ValidatorSessionRoundAttemptState* move_to_persistent(ValidatorSessionDescription& desc,
|
|
const ValidatorSessionRoundAttemptState* b) {
|
|
if (desc.is_persistent(b)) {
|
|
return b;
|
|
}
|
|
auto votes = VoteVector::move_to_persistent(desc, b->votes_);
|
|
auto precommitted = CntVector<bool>::move_to_persistent(desc, b->precommitted_);
|
|
auto vote_for = SentBlock::move_to_persistent(desc, b->vote_for_);
|
|
|
|
auto r = lookup(desc, b->seqno_, votes, precommitted, vote_for, b->vote_for_inited_, b->hash_, false);
|
|
if (r) {
|
|
return r;
|
|
}
|
|
|
|
return new (desc, false) ValidatorSessionRoundAttemptState{desc, b->seqno_, votes, precommitted,
|
|
vote_for, b->vote_for_inited_, b->hash_};
|
|
}
|
|
|
|
static const ValidatorSessionRoundAttemptState* create(ValidatorSessionDescription& desc, td::uint32 seqno,
|
|
const VoteVector* votes, const CntVector<bool>* precommitted,
|
|
const SentBlock* vote_for, bool vote_for_inited) {
|
|
auto hash = create_hash(desc, seqno, get_vs_hash(desc, votes), get_vs_hash(desc, precommitted),
|
|
get_vs_hash(desc, vote_for), vote_for_inited);
|
|
|
|
auto r = lookup(desc, seqno, votes, precommitted, vote_for, vote_for_inited, hash, true);
|
|
if (r) {
|
|
return r;
|
|
}
|
|
|
|
return new (desc, true)
|
|
ValidatorSessionRoundAttemptState(desc, seqno, votes, precommitted, vote_for, vote_for_inited, hash);
|
|
}
|
|
static const ValidatorSessionRoundAttemptState* create(ValidatorSessionDescription& desc, td::uint32 seqno) {
|
|
std::vector<bool> x;
|
|
x.resize(desc.get_total_nodes(), false);
|
|
auto p = CntVector<bool>::create(desc, std::move(x));
|
|
|
|
return create(desc, seqno, nullptr, p, nullptr, false);
|
|
}
|
|
|
|
ValidatorSessionRoundAttemptState(ValidatorSessionDescription& desc, td::uint32 seqno, const VoteVector* votes,
|
|
const CntVector<bool>* precommitted, const SentBlock* vote_for,
|
|
bool vote_for_inited, HashType hash)
|
|
: RootObject{sizeof(ValidatorSessionRoundAttemptState)}
|
|
, seqno_(seqno)
|
|
, votes_(votes)
|
|
, precommitted_(precommitted)
|
|
, vote_for_(vote_for)
|
|
, vote_for_inited_(vote_for_inited)
|
|
, hash_(std::move(hash)) {
|
|
desc.update_hash(this, hash_);
|
|
}
|
|
auto get_hash(ValidatorSessionDescription& desc) const {
|
|
return hash_;
|
|
}
|
|
auto get_seqno() const {
|
|
return seqno_;
|
|
}
|
|
auto get_votes() const {
|
|
return votes_;
|
|
}
|
|
auto get_precommits() const {
|
|
return precommitted_;
|
|
}
|
|
const SentBlock* get_voted_block(ValidatorSessionDescription& desc, bool& f) const;
|
|
const SentBlock* get_vote_for_block(ValidatorSessionDescription& desc, bool& f) const {
|
|
f = vote_for_inited_;
|
|
return vote_for_;
|
|
}
|
|
bool check_attempt_is_precommitted(ValidatorSessionDescription& desc) const;
|
|
bool check_vote_received_from(td::uint32 src_idx) const;
|
|
bool check_precommit_received_from(td::uint32 src_idx) const;
|
|
|
|
bool operator<(const ValidatorSessionRoundAttemptState& right) const {
|
|
return seqno_ < right.seqno_;
|
|
}
|
|
struct Compare {
|
|
bool operator()(const ValidatorSessionRoundAttemptState* a, const ValidatorSessionRoundAttemptState* b) const {
|
|
return *a < *b;
|
|
}
|
|
};
|
|
|
|
static const ValidatorSessionRoundAttemptState* merge(ValidatorSessionDescription& desc,
|
|
const ValidatorSessionRoundAttemptState* left,
|
|
const ValidatorSessionRoundAttemptState* right);
|
|
static const ValidatorSessionRoundAttemptState* action(ValidatorSessionDescription& desc,
|
|
const ValidatorSessionRoundAttemptState* state,
|
|
td::uint32 src_idx, td::uint32 att,
|
|
const ton_api::validatorSession_message_voteFor& act,
|
|
const ValidatorSessionRoundState* round);
|
|
static const ValidatorSessionRoundAttemptState* action(ValidatorSessionDescription& desc,
|
|
const ValidatorSessionRoundAttemptState* state,
|
|
td::uint32 src_idx, td::uint32 att,
|
|
const ton_api::validatorSession_message_vote& act,
|
|
const ValidatorSessionRoundState* round);
|
|
static const ValidatorSessionRoundAttemptState* action(ValidatorSessionDescription& desc,
|
|
const ValidatorSessionRoundAttemptState* state,
|
|
td::uint32 src_idx, td::uint32 att,
|
|
const ton_api::validatorSession_message_precommit& act,
|
|
const ValidatorSessionRoundState* round);
|
|
static const ValidatorSessionRoundAttemptState* action(ValidatorSessionDescription& desc,
|
|
const ValidatorSessionRoundAttemptState* state,
|
|
td::uint32 src_idx, td::uint32 att,
|
|
const ton_api::validatorSession_message_empty& act,
|
|
const ValidatorSessionRoundState* round);
|
|
static const ValidatorSessionRoundAttemptState* action(ValidatorSessionDescription& desc,
|
|
const ValidatorSessionRoundAttemptState* state,
|
|
td::uint32 src_idx, td::uint32 att,
|
|
const ton_api::validatorSession_round_Message* action,
|
|
const ValidatorSessionRoundState* round);
|
|
template <class T>
|
|
static const ValidatorSessionRoundAttemptState* action(ValidatorSessionDescription& desc,
|
|
const ValidatorSessionRoundAttemptState* state,
|
|
td::uint32 src_idx, td::uint32 att, const T& action,
|
|
const ValidatorSessionRoundState* round) {
|
|
UNREACHABLE();
|
|
}
|
|
static const ValidatorSessionRoundAttemptState* try_vote(ValidatorSessionDescription& desc,
|
|
const ValidatorSessionRoundAttemptState* state,
|
|
td::uint32 src_idx, td::uint32 att,
|
|
const ValidatorSessionRoundState* round,
|
|
const ton_api::validatorSession_round_Message* cmp,
|
|
bool& made);
|
|
static const ValidatorSessionRoundAttemptState* try_precommit(ValidatorSessionDescription& desc,
|
|
const ValidatorSessionRoundAttemptState* state,
|
|
td::uint32 src_idx, td::uint32 att,
|
|
const ValidatorSessionRoundState* round,
|
|
const ton_api::validatorSession_round_Message* cmp,
|
|
bool& made);
|
|
static const ValidatorSessionRoundAttemptState* make_one(ValidatorSessionDescription& desc,
|
|
const ValidatorSessionRoundAttemptState* state,
|
|
td::uint32 src_idx, td::uint32 att,
|
|
const ValidatorSessionRoundState* round,
|
|
const ton_api::validatorSession_round_Message* cmp,
|
|
bool& made);
|
|
tl_object_ptr<ton_api::validatorSession_round_Message> create_action(ValidatorSessionDescription& desc,
|
|
const ValidatorSessionRoundState* round,
|
|
td::uint32 src_idx, td::uint32 att) const;
|
|
void dump(ValidatorSessionDescription& desc, td::StringBuilder& sb) const;
|
|
|
|
private:
|
|
const td::uint32 seqno_;
|
|
const VoteVector* votes_;
|
|
const CntVector<bool>* precommitted_;
|
|
const SentBlock* vote_for_;
|
|
const bool vote_for_inited_;
|
|
const HashType hash_;
|
|
};
|
|
|
|
using AttemptVector =
|
|
CntSortedVector<const ValidatorSessionRoundAttemptState*, ValidatorSessionRoundAttemptState::Compare>;
|
|
|
|
class ValidatorSessionRoundState : public ValidatorSessionDescription::RootObject {
|
|
public:
|
|
static HashType create_hash(ValidatorSessionDescription& desc, const SentBlock* precommitted_block, td::uint32 seqno,
|
|
bool precommitted, const CntVector<td::uint32>* first_attempt,
|
|
const CntVector<td::uint32>* last_precommit, const ApproveVector* sent,
|
|
const CntVector<const SessionBlockCandidateSignature*>* signatures,
|
|
const AttemptVector* attempts) {
|
|
auto obj = create_tl_object<ton_api::hashable_validatorSessionRound>(
|
|
get_vs_hash(desc, precommitted_block), seqno, precommitted, get_vs_hash(desc, first_attempt),
|
|
get_vs_hash(desc, last_precommit), get_vs_hash(desc, sent), get_vs_hash(desc, signatures),
|
|
get_vs_hash(desc, attempts));
|
|
return desc.compute_hash(serialize_tl_object(obj, true).as_slice());
|
|
}
|
|
static bool compare(const RootObject* root_object, const SentBlock* precommitted_block, td::uint32 seqno,
|
|
bool precommitted, const CntVector<td::uint32>* first_attempt,
|
|
const CntVector<td::uint32>* last_precommit, const ApproveVector* sent,
|
|
const CntVector<const SessionBlockCandidateSignature*>* signatures, const AttemptVector* attempts,
|
|
HashType hash) {
|
|
if (!root_object || root_object->get_size() < sizeof(ValidatorSessionRoundState)) {
|
|
return false;
|
|
}
|
|
auto obj = static_cast<const ValidatorSessionRoundState*>(root_object);
|
|
return obj->precommitted_block_ == precommitted_block && obj->seqno_ == seqno &&
|
|
obj->precommitted_ == precommitted && obj->first_attempt_ == first_attempt &&
|
|
obj->last_precommit_ == last_precommit && obj->sent_blocks_ == sent && obj->signatures_ == signatures &&
|
|
obj->attempts_ == attempts && obj->hash_ == hash;
|
|
}
|
|
static const ValidatorSessionRoundState* lookup(ValidatorSessionDescription& desc,
|
|
const SentBlock* precommitted_block, td::uint32 seqno,
|
|
bool precommitted, const CntVector<td::uint32>* first_attempt,
|
|
const CntVector<td::uint32>* last_precommit,
|
|
const ApproveVector* sent,
|
|
const CntVector<const SessionBlockCandidateSignature*>* signatures,
|
|
const AttemptVector* attempts, HashType hash, bool temp) {
|
|
auto r = desc.get_by_hash(hash, temp);
|
|
if (compare(r, precommitted_block, seqno, precommitted, first_attempt, last_precommit, sent, signatures, attempts,
|
|
hash)) {
|
|
desc.on_reuse();
|
|
return static_cast<const ValidatorSessionRoundState*>(r);
|
|
}
|
|
|
|
return nullptr;
|
|
}
|
|
static const ValidatorSessionRoundState* create(ValidatorSessionDescription& desc,
|
|
const SentBlock* precommitted_block, td::uint32 seqno,
|
|
bool precommitted, const CntVector<td::uint32>* first_attempt,
|
|
const CntVector<td::uint32>* last_precommit,
|
|
const ApproveVector* sent,
|
|
const CntVector<const SessionBlockCandidateSignature*>* signatures,
|
|
const AttemptVector* attempts) {
|
|
auto hash = create_hash(desc, precommitted_block, seqno, precommitted, first_attempt, last_precommit, sent,
|
|
signatures, attempts);
|
|
|
|
auto r = lookup(desc, precommitted_block, seqno, precommitted, first_attempt, last_precommit, sent, signatures,
|
|
attempts, hash, true);
|
|
if (r) {
|
|
return r;
|
|
}
|
|
|
|
return new (desc, true) ValidatorSessionRoundState(desc, precommitted_block, seqno, precommitted, first_attempt,
|
|
last_precommit, sent, signatures, attempts, hash);
|
|
}
|
|
static const ValidatorSessionRoundState* create(ValidatorSessionDescription& desc, td::uint32 seqno) {
|
|
std::vector<const SessionBlockCandidateSignature*> signs;
|
|
signs.resize(desc.get_total_nodes(), nullptr);
|
|
std::vector<td::uint32> v;
|
|
v.resize(desc.get_total_nodes(), 0);
|
|
auto first_attempt = CntVector<td::uint32>::create(desc, v);
|
|
auto last_precommit = CntVector<td::uint32>::create(desc, v);
|
|
auto signatures = CntVector<const SessionBlockCandidateSignature*>::create(desc, std::move(signs));
|
|
return create(desc, nullptr, seqno, false, first_attempt, last_precommit, nullptr, signatures, nullptr);
|
|
}
|
|
static const ValidatorSessionRoundState* move_to_persistent(ValidatorSessionDescription& desc,
|
|
const ValidatorSessionRoundState* b) {
|
|
if (desc.is_persistent(b)) {
|
|
return b;
|
|
}
|
|
auto precommitted_block = SentBlock::move_to_persistent(desc, b->precommitted_block_);
|
|
auto first_attempt = CntVector<td::uint32>::move_to_persistent(desc, b->first_attempt_);
|
|
auto last_precommit = CntVector<td::uint32>::move_to_persistent(desc, b->last_precommit_);
|
|
auto sent = ApproveVector::move_to_persistent(desc, b->sent_blocks_);
|
|
auto signatures = CntVector<const SessionBlockCandidateSignature*>::move_to_persistent(desc, b->signatures_);
|
|
auto attempts = AttemptVector::move_to_persistent(desc, b->attempts_);
|
|
auto r = lookup(desc, precommitted_block, b->seqno_, b->precommitted_, first_attempt, last_precommit, sent,
|
|
signatures, attempts, b->hash_, false);
|
|
if (r) {
|
|
return r;
|
|
}
|
|
return new (desc, false)
|
|
ValidatorSessionRoundState{desc, precommitted_block, b->seqno_, b->precommitted_, first_attempt, last_precommit,
|
|
sent, signatures, attempts, b->hash_};
|
|
}
|
|
ValidatorSessionRoundState(ValidatorSessionDescription& desc, const SentBlock* precommitted_block, td::uint32 seqno,
|
|
bool precommitted, const CntVector<td::uint32>* first_attempt,
|
|
const CntVector<td::uint32>* last_precommit, const ApproveVector* sent_blocks,
|
|
const CntVector<const SessionBlockCandidateSignature*>* signatures,
|
|
const AttemptVector* attempts, HashType hash)
|
|
: RootObject{sizeof(ValidatorSessionRoundState)}
|
|
, precommitted_block_(precommitted_block)
|
|
, seqno_(seqno)
|
|
, precommitted_(precommitted)
|
|
, first_attempt_(first_attempt)
|
|
, last_precommit_(last_precommit)
|
|
, sent_blocks_(std::move(sent_blocks))
|
|
, signatures_(std::move(signatures))
|
|
, attempts_(std::move(attempts))
|
|
, hash_(std::move(hash)) {
|
|
desc.update_hash(this, hash_);
|
|
}
|
|
auto get_hash(ValidatorSessionDescription& desc) const {
|
|
return hash_;
|
|
}
|
|
|
|
static const ValidatorSessionRoundState* merge(ValidatorSessionDescription& desc,
|
|
const ValidatorSessionRoundState* left,
|
|
const ValidatorSessionRoundState* right);
|
|
static const ValidatorSessionRoundState* action(ValidatorSessionDescription& desc,
|
|
const ValidatorSessionRoundState* state, td::uint32 src_idx,
|
|
td::uint32 att,
|
|
const ton_api::validatorSession_message_submittedBlock& action);
|
|
static const ValidatorSessionRoundState* action(ValidatorSessionDescription& desc,
|
|
const ValidatorSessionRoundState* state, td::uint32 src_idx,
|
|
td::uint32 att,
|
|
const ton_api::validatorSession_message_approvedBlock& action);
|
|
static const ValidatorSessionRoundState* action(ValidatorSessionDescription& desc,
|
|
const ValidatorSessionRoundState* state, td::uint32 src_idx,
|
|
td::uint32 att,
|
|
const ton_api::validatorSession_message_rejectedBlock& action);
|
|
static const ValidatorSessionRoundState* action(ValidatorSessionDescription& desc,
|
|
const ValidatorSessionRoundState* state, td::uint32 src_idx,
|
|
td::uint32 att,
|
|
const ton_api::validatorSession_message_commit& action);
|
|
static const ValidatorSessionRoundState* forward_action_to_attempt(
|
|
ValidatorSessionDescription& desc, const ValidatorSessionRoundState* state, td::uint32 src_idx, td::uint32 att,
|
|
const ton_api::validatorSession_round_Message* act);
|
|
template <class T>
|
|
static const ValidatorSessionRoundState* action(ValidatorSessionDescription& desc,
|
|
const ValidatorSessionRoundState* state, td::uint32 src_idx,
|
|
td::uint32 att, const T& act) {
|
|
return forward_action_to_attempt(desc, state, src_idx, att, &act);
|
|
}
|
|
static const ValidatorSessionRoundState* action(ValidatorSessionDescription& desc,
|
|
const ValidatorSessionRoundState* state, td::uint32 src_idx,
|
|
td::uint32 att,
|
|
const ton_api::validatorSession_round_Message* action);
|
|
static const ValidatorSessionRoundState* make_one(ValidatorSessionDescription& desc,
|
|
const ValidatorSessionRoundState* state, td::uint32 src_idx,
|
|
td::uint32 att, bool& made);
|
|
|
|
auto get_precommitted_block(bool& f) const {
|
|
f = precommitted_;
|
|
return precommitted_block_;
|
|
}
|
|
auto get_signatures() const {
|
|
return signatures_;
|
|
}
|
|
td::uint32 get_seqno() const {
|
|
return seqno_;
|
|
}
|
|
auto get_first_attempt(td::uint32 src_idx) const {
|
|
return first_attempt_->at(src_idx);
|
|
}
|
|
auto get_last_precommit(td::uint32 src_idx) const {
|
|
return last_precommit_->at(src_idx);
|
|
}
|
|
auto get_sent_blocks() const {
|
|
return sent_blocks_;
|
|
}
|
|
|
|
const SessionBlockCandidate* get_block(ValidatorSessionCandidateId block_hash) const;
|
|
std::vector<td::uint32> get_block_approvers(ValidatorSessionDescription& desc,
|
|
ValidatorSessionCandidateId block) const;
|
|
std::vector<const SentBlock*> get_blocks_approved_by(ValidatorSessionDescription& desc, td::uint32 src_idx) const;
|
|
|
|
bool check_block_is_signed(ValidatorSessionDescription& desc) const;
|
|
bool check_block_is_signed_by(td::uint32 src_idx) const {
|
|
return signatures_->at(src_idx) != nullptr;
|
|
}
|
|
bool check_block_is_approved(ValidatorSessionCandidateId block_id) const;
|
|
bool check_block_is_approved_by(td::uint32 src_idx, ValidatorSessionCandidateId block_id) const;
|
|
const SentBlock* check_block_is_sent(ValidatorSessionCandidateId block_id) const;
|
|
bool check_block_is_sent_by(td::uint32 src_idx) const;
|
|
|
|
bool check_need_generate_vote_for(ValidatorSessionDescription& desc, td::uint32 src_idx, td::uint32 att) const;
|
|
tl_object_ptr<ton_api::validatorSession_message_voteFor> generate_vote_for(ValidatorSessionDescription& desc,
|
|
td::uint32 src_idx, td::uint32 att) const;
|
|
|
|
const SentBlock* choose_block_to_sign(ValidatorSessionDescription& desc, td::uint32 src_idx, bool& found) const;
|
|
std::vector<const SentBlock*> choose_blocks_to_approve(ValidatorSessionDescription& desc, td::uint32 src_idx) const;
|
|
const SentBlock* choose_block_to_vote(ValidatorSessionDescription& desc, td::uint32 src_idx, td::uint32 att,
|
|
const SentBlock* vote_for, bool vote_for_inited, bool& found) const;
|
|
|
|
tl_object_ptr<ton_api::validatorSession_round_Message> create_action(ValidatorSessionDescription& desc,
|
|
td::uint32 src_idx, td::uint32 att) const;
|
|
void dump(ValidatorSessionDescription& desc, td::StringBuilder& sb, td::uint32 att) const;
|
|
void dump_cur_attempt(ValidatorSessionDescription& desc, td::StringBuilder& sb) const;
|
|
|
|
private:
|
|
const SentBlock* precommitted_block_;
|
|
const td::uint32 seqno_;
|
|
const bool precommitted_;
|
|
const CntVector<td::uint32>* first_attempt_;
|
|
const CntVector<td::uint32>* last_precommit_;
|
|
const ApproveVector* sent_blocks_;
|
|
const CntVector<const SessionBlockCandidateSignature*>* signatures_;
|
|
const AttemptVector* attempts_;
|
|
const HashType hash_;
|
|
};
|
|
|
|
class ValidatorSessionState : public ValidatorSessionDescription::RootObject {
|
|
public:
|
|
static HashType create_hash(ValidatorSessionDescription& desc, const CntVector<td::uint32>* att,
|
|
const CntVector<const ValidatorSessionOldRoundState*>* old_rounds,
|
|
const ValidatorSessionRoundState* cur_round) {
|
|
auto obj = create_tl_object<ton_api::hashable_validatorSession>(
|
|
get_vs_hash(desc, att), get_vs_hash(desc, old_rounds), get_vs_hash(desc, cur_round));
|
|
return desc.compute_hash(serialize_tl_object(obj, true).as_slice());
|
|
}
|
|
static bool compare(const RootObject* r, const CntVector<td::uint32>* att,
|
|
const CntVector<const ValidatorSessionOldRoundState*>* old_rounds,
|
|
const ValidatorSessionRoundState* cur_round, HashType hash) {
|
|
if (!r || r->get_size() < sizeof(ValidatorSessionState)) {
|
|
return false;
|
|
}
|
|
auto R = static_cast<const ValidatorSessionState*>(r);
|
|
return R->att_ == att && R->old_rounds_ == old_rounds && R->cur_round_ == cur_round && R->hash_ == hash;
|
|
}
|
|
static const ValidatorSessionState* lookup(ValidatorSessionDescription& desc, const CntVector<td::uint32>* att,
|
|
const CntVector<const ValidatorSessionOldRoundState*>* old_rounds,
|
|
const ValidatorSessionRoundState* cur_round, HashType hash, bool temp) {
|
|
auto r = desc.get_by_hash(hash, temp);
|
|
if (compare(r, att, old_rounds, cur_round, hash)) {
|
|
desc.on_reuse();
|
|
return static_cast<const ValidatorSessionState*>(r);
|
|
}
|
|
return nullptr;
|
|
}
|
|
static const ValidatorSessionState* create(ValidatorSessionDescription& desc, const CntVector<td::uint32>* att,
|
|
const CntVector<const ValidatorSessionOldRoundState*>* old_rounds,
|
|
const ValidatorSessionRoundState* cur_round) {
|
|
auto hash = create_hash(desc, att, old_rounds, cur_round);
|
|
auto r = lookup(desc, att, old_rounds, cur_round, hash, true);
|
|
if (r) {
|
|
return r;
|
|
}
|
|
|
|
return new (desc, true) ValidatorSessionState{desc, att, old_rounds, cur_round, hash};
|
|
}
|
|
static const ValidatorSessionState* create(ValidatorSessionDescription& desc) {
|
|
std::vector<td::uint32> vec;
|
|
vec.resize(desc.get_total_nodes(), 0);
|
|
auto ts = CntVector<td::uint32>::create(desc, std::move(vec));
|
|
auto cur_round = ValidatorSessionRoundState::create(desc, 0);
|
|
return create(desc, ts, nullptr, std::move(cur_round));
|
|
}
|
|
static const ValidatorSessionState* move_to_persistent(ValidatorSessionDescription& desc,
|
|
const ValidatorSessionState* b) {
|
|
if (desc.is_persistent(b)) {
|
|
return b;
|
|
}
|
|
auto ts = CntVector<td::uint32>::move_to_persistent(desc, b->att_);
|
|
auto old_rounds = CntVector<const ValidatorSessionOldRoundState*>::move_to_persistent(desc, b->old_rounds_);
|
|
auto cur_round = ValidatorSessionRoundState::move_to_persistent(desc, b->cur_round_);
|
|
auto r = lookup(desc, ts, old_rounds, cur_round, b->hash_, false);
|
|
if (r) {
|
|
return r;
|
|
}
|
|
return new (desc, false) ValidatorSessionState{desc, ts, old_rounds, cur_round, b->hash_};
|
|
}
|
|
ValidatorSessionState(ValidatorSessionDescription& desc, const CntVector<td::uint32>* att,
|
|
const CntVector<const ValidatorSessionOldRoundState*>* old_rounds,
|
|
const ValidatorSessionRoundState* cur_round, HashType hash)
|
|
: RootObject{sizeof(ValidatorSessionState)}
|
|
, att_(att)
|
|
, old_rounds_(std::move(old_rounds))
|
|
, cur_round_(std::move(cur_round))
|
|
, hash_(std::move(hash)) {
|
|
desc.update_hash(this, hash_);
|
|
}
|
|
|
|
auto get_hash(ValidatorSessionDescription& desc) const {
|
|
return hash_;
|
|
}
|
|
|
|
auto cur_round_seqno() const {
|
|
return cur_round_->get_seqno();
|
|
}
|
|
auto get_ts(td::uint32 src_idx) const {
|
|
return att_->at(src_idx);
|
|
}
|
|
|
|
const SentBlock* choose_block_to_sign(ValidatorSessionDescription& desc, td::uint32 src_idx, bool& found) const;
|
|
const SentBlock* get_committed_block(ValidatorSessionDescription& desc, td::uint32 seqno) const;
|
|
const SentBlock* get_block(ValidatorSessionDescription& desc, ValidatorSessionCandidateId id, bool& found) const;
|
|
std::vector<const SentBlock*> get_blocks_approved_by(ValidatorSessionDescription& desc, td::uint32 src_idx) const;
|
|
const CntVector<const SessionBlockCandidateSignature*>* get_committed_block_signatures(
|
|
ValidatorSessionDescription& desc, td::uint32 seqno) const;
|
|
const CntVector<const SessionBlockCandidateSignature*>* get_committed_block_approve_signatures(
|
|
ValidatorSessionDescription& desc, td::uint32 seqno) const;
|
|
|
|
std::vector<const SentBlock*> choose_blocks_to_approve(ValidatorSessionDescription& desc, td::uint32 src_idx) const {
|
|
return cur_round_->choose_blocks_to_approve(desc, src_idx);
|
|
}
|
|
bool check_block_is_sent_by(ValidatorSessionDescription& desc, td::uint32 src_idx) const {
|
|
return cur_round_->check_block_is_sent_by(src_idx);
|
|
}
|
|
bool check_block_is_signed_by(ValidatorSessionDescription& desc, td::uint32 src_idx) const {
|
|
return cur_round_->check_block_is_signed_by(src_idx);
|
|
}
|
|
bool check_block_is_approved_by(ValidatorSessionDescription& desc, td::uint32 src_idx,
|
|
ValidatorSessionCandidateId id) const {
|
|
auto B = cur_round_->get_block(id);
|
|
if (!B) {
|
|
return false;
|
|
}
|
|
return B->check_block_is_approved_by(src_idx);
|
|
}
|
|
bool check_need_generate_vote_for(ValidatorSessionDescription& desc, td::uint32 src_idx, td::uint32 att) const {
|
|
return cur_round_->check_need_generate_vote_for(desc, src_idx, att);
|
|
}
|
|
tl_object_ptr<ton_api::validatorSession_message_voteFor> generate_vote_for(ValidatorSessionDescription& desc,
|
|
td::uint32 src_idx, td::uint32 att) const {
|
|
return cur_round_->generate_vote_for(desc, src_idx, att);
|
|
}
|
|
std::vector<td::uint32> get_block_approvers(ValidatorSessionDescription& desc,
|
|
ValidatorSessionCandidateId block) const {
|
|
return cur_round_->get_block_approvers(desc, block);
|
|
}
|
|
|
|
tl_object_ptr<ton_api::validatorSession_round_Message> create_action(ValidatorSessionDescription& desc,
|
|
td::uint32 src_idx, td::uint32 att) const;
|
|
|
|
void dump(ValidatorSessionDescription& desc, td::StringBuilder& sb, td::uint32 att) const;
|
|
void dump_cur_attempt(ValidatorSessionDescription& desc, td::StringBuilder& sb) const {
|
|
cur_round_->dump_cur_attempt(desc, sb);
|
|
}
|
|
|
|
static const ValidatorSessionState* make_one(ValidatorSessionDescription& desc, const ValidatorSessionState* state,
|
|
td::uint32 src_idx, td::uint32 att, bool& made);
|
|
static const ValidatorSessionState* make_all(ValidatorSessionDescription& desc, const ValidatorSessionState* state,
|
|
td::uint32 src_idx, td::uint32 att) {
|
|
bool made;
|
|
do {
|
|
made = false;
|
|
state = make_one(desc, state, src_idx, att, made);
|
|
} while (made);
|
|
return state;
|
|
}
|
|
|
|
static const ValidatorSessionState* merge(ValidatorSessionDescription& desc, const ValidatorSessionState* left,
|
|
const ValidatorSessionState* right);
|
|
static const ValidatorSessionState* action(ValidatorSessionDescription& desc, const ValidatorSessionState* state,
|
|
td::uint32 src_idx, td::uint32 att,
|
|
const ton_api::validatorSession_round_Message* action);
|
|
|
|
private:
|
|
const CntVector<td::uint32>* att_;
|
|
const CntVector<const ValidatorSessionOldRoundState*>* old_rounds_;
|
|
const ValidatorSessionRoundState* cur_round_;
|
|
const HashType hash_;
|
|
};
|
|
|
|
} // namespace validatorsession
|
|
|
|
} // namespace ton
|