initial commit

validator/get-next-key-blocks.cpp

@@ -0,0 +1,91 @@
+/*
+    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 "get-next-key-blocks.h"
+
+namespace ton {
+
+namespace validator {
+
+void GetNextKeyBlocks::finish_query() {
+  if (promise_) {
+    promise_.set_value(std::move(res_));
+  }
+  stop();
+}
+
+void GetNextKeyBlocks::abort_query(td::Status error) {
+  if (promise_) {
+    promise_.set_error(std::move(error));
+  }
+  stop();
+}
+
+void GetNextKeyBlocks::alarm() {
+  abort_query(td::Status::Error(ErrorCode::timeout, "timeout"));
+}
+
+void GetNextKeyBlocks::start_up() {
+  if (!block_id_.is_masterchain()) {
+    abort_query(td::Status::Error(ErrorCode::notready, "block is not from masterchain"));
+    return;
+  }
+
+  if (block_id_.id.seqno > last_known_key_block_->id().seqno()) {
+    finish_query();
+    return;
+  }
+
+  if (block_id_.id.seqno == last_known_key_block_->id().seqno()) {
+    if (block_id_ == last_known_key_block_->id()) {
+      finish_query();
+    } else {
+      abort_query(td::Status::Error(ErrorCode::protoviolation, "bad block id"));
+    }
+    return;
+  }
+
+  if (block_id_.seqno() > 0) {
+    BlockIdExt block_id;
+    auto f = masterchain_state_->prev_key_block_id(block_id_.seqno());
+    if (!f.is_valid()) {
+      abort_query(td::Status::Error(ErrorCode::protoviolation, "bad block id (not key?)"));
+      return;
+    } else if (f != block_id_) {
+      abort_query(td::Status::Error(ErrorCode::protoviolation, "bad block id (not key?) 2"));
+      return;
+    }
+  }
+
+  while (res_.size() < limit_) {
+    auto b = masterchain_state_->next_key_block_id(block_id_.seqno() + 1);
+    if (b.is_valid()) {
+      res_.push_back(b);
+      block_id_ = b;
+    } else {
+      finish_query();
+      return;
+    }
+  }
+
+  finish_query();
+}
+
+}  // namespace validator
+
+}  // namespace ton