aboutsummaryrefslogtreecommitdiff
path: root/src/entity.h
diff options
context:
space:
mode:
authorDeterminant <[email protected]>2018-07-16 05:00:14 -0400
committerDeterminant <[email protected]>2018-07-16 05:00:14 -0400
commita7cfb274d651e858ab06eff5b28a6f77e0178cf1 (patch)
tree7d76e5b5be935ef63a47dd1ed43b391f9a7e513c /src/entity.h
parentaac40104573f8aabca86410cc08584acaaa74e26 (diff)
move header files to include/hotstuff
Diffstat (limited to 'src/entity.h')
-rw-r--r--src/entity.h309
1 files changed, 0 insertions, 309 deletions
diff --git a/src/entity.h b/src/entity.h
deleted file mode 100644
index b3a0df4..0000000
--- a/src/entity.h
+++ /dev/null
@@ -1,309 +0,0 @@
-#ifndef _HOTSTUFF_ENT_H
-#define _HOTSTUFF_ENT_H
-
-#include <vector>
-#include <unordered_map>
-#include <unordered_set>
-#include <string>
-#include <cstddef>
-#include <ios>
-
-#include "salticidae/netaddr.h"
-#include "salticidae/ref.h"
-#include "type.h"
-#include "util.h"
-#include "crypto.h"
-
-namespace hotstuff {
-
-enum EntityType {
- ENT_TYPE_CMD = 0x0,
- ENT_TYPE_BLK = 0x1
-};
-
-struct ReplicaInfo {
- ReplicaID id;
- salticidae::NetAddr addr;
- pubkey_bt pubkey;
-
- ReplicaInfo(ReplicaID id,
- const salticidae::NetAddr &addr,
- pubkey_bt &&pubkey):
- id(id), addr(addr), pubkey(std::move(pubkey)) {}
-
- ReplicaInfo(const ReplicaInfo &other):
- id(other.id), addr(other.addr),
- pubkey(other.pubkey->clone()) {}
-
- ReplicaInfo(ReplicaInfo &&other):
- id(other.id), addr(other.addr),
- pubkey(std::move(other.pubkey)) {}
-};
-
-class ReplicaConfig {
- std::unordered_map<ReplicaID, ReplicaInfo> replica_map;
-
- public:
- size_t nmajority;
-
- void add_replica(ReplicaID rid, const ReplicaInfo &info) {
- replica_map.insert(std::make_pair(rid, info));
- }
-
- const ReplicaInfo &get_info(ReplicaID rid) const {
- auto it = replica_map.find(rid);
- if (it == replica_map.end())
- throw HotStuffError("rid %s not found",
- get_hex(rid).c_str());
- return it->second;
- }
-
- const PubKey &get_pubkey(ReplicaID rid) const {
- return *(get_info(rid).pubkey);
- }
-
- const salticidae::NetAddr &get_addr(ReplicaID rid) const {
- return get_info(rid).addr;
- }
-};
-
-class Block;
-class HotStuffCore;
-
-using block_t = salticidae::RcObj<Block>;
-using block_weak_t = salticidae::WeakObj<Block>;
-
-struct Finality: public Serializable {
- int8_t decision;
- uint256_t blk_hash;
-
- public:
- Finality(): decision(0) {}
- Finality(int8_t decision, uint256_t blk_hash):
- decision(decision), blk_hash(blk_hash) {}
-
- void serialize(DataStream &s) const override {
- s << decision;
- if (decision == 1) s << blk_hash;
- }
-
- void unserialize(DataStream &s) override {
- s >> decision;
- if (decision == 1) s >> blk_hash;
- }
-};
-
-class Command: public Serializable {
- friend HotStuffCore;
- block_weak_t container;
- public:
- virtual ~Command() = default;
- virtual const uint256_t &get_hash() const = 0;
- virtual bool verify() const = 0;
- inline int8_t get_decision() const;
- inline Finality get_finality() const;
- block_t get_container() const {
- return container;
- }
-};
-
-using command_t = RcObj<Command>;
-
-template<typename Hashable>
-inline static std::vector<uint256_t>
-get_hashes(const std::vector<Hashable> &plist) {
- std::vector<uint256_t> hashes;
- for (const auto &p: plist)
- hashes.push_back(p->get_hash());
- return std::move(hashes);
-}
-
-class Block {
- friend HotStuffCore;
- std::vector<uint256_t> parent_hashes;
- std::vector<command_t> cmds;
- quorum_cert_bt qc;
- uint256_t hash;
-
- std::vector<block_t> parents;
- block_t qc_ref;
- quorum_cert_bt self_qc;
- uint32_t height;
- bool delivered;
- int8_t decision;
-
- std::unordered_set<ReplicaID> voted;
-
- public:
- Block():
- qc(nullptr),
- qc_ref(nullptr),
- self_qc(nullptr), height(0),
- delivered(false), decision(0) {}
-
- Block(bool delivered, int8_t decision):
- qc(nullptr),
- hash(salticidae::get_hash(*this)),
- qc_ref(nullptr),
- self_qc(nullptr), height(0),
- delivered(delivered), decision(decision) {}
-
- Block(const std::vector<block_t> &parents,
- const std::vector<command_t> &cmds,
- uint32_t height,
- quorum_cert_bt &&qc,
- const block_t &qc_ref,
- quorum_cert_bt &&self_qc,
- int8_t decision = 0):
- parent_hashes(get_hashes(parents)),
- cmds(cmds),
- qc(std::move(qc)),
- hash(salticidae::get_hash(*this)),
- parents(parents),
- qc_ref(qc_ref),
- self_qc(std::move(self_qc)),
- height(height),
- delivered(0),
- decision(decision) {}
-
- void serialize(DataStream &s) const;
-
- void unserialize(DataStream &s, HotStuffCore *hsc);
-
- const std::vector<command_t> &get_cmds() const {
- return cmds;
- }
-
- const std::vector<block_t> &get_parents() const {
- return parents;
- }
-
- const std::vector<uint256_t> &get_parent_hashes() const {
- return parent_hashes;
- }
-
- const uint256_t &get_hash() const { return hash; }
-
- bool verify(const ReplicaConfig &config) const {
- if (qc && !qc->verify(config)) return false;
- for (auto cmd: cmds)
- if (!cmd->verify()) return false;
- return true;
- }
-
- int8_t get_decision() const { return decision; }
-
- bool is_delivered() const { return delivered; }
-
- uint32_t get_height() const { return height; }
-
- const quorum_cert_bt &get_qc() const { return qc; }
-
- operator std::string () const {
- DataStream s;
- s << "<block "
- << "id=" << get_hex10(hash) << " "
- << "height=" << std::to_string(height) << " "
- << "parent=" << get_hex10(parent_hashes[0]) << ">";
- return std::string(std::move(s));
- }
-};
-
-struct BlockHeightCmp {
- bool operator()(const block_t &a, const block_t &b) const {
- return a->get_height() < b->get_height();
- }
-};
-
-int8_t Command::get_decision() const {
- block_t cptr = container;
- return cptr ? cptr->get_decision() : 0;
-}
-
-Finality Command::get_finality() const {
- block_t blk = get_container();
- return Finality(get_decision(),
- blk ? blk->get_hash() : uint256_t());
-}
-
-class EntityStorage {
- std::unordered_map<const uint256_t, block_t> blk_cache;
- std::unordered_map<const uint256_t, command_t> cmd_cache;
- public:
- bool is_blk_delivered(const uint256_t &blk_hash) {
- auto it = blk_cache.find(blk_hash);
- if (it == blk_cache.end()) return false;
- return it->second->is_delivered();
- }
-
- bool is_blk_fetched(const uint256_t &blk_hash) {
- return blk_cache.count(blk_hash);
- }
-
- const block_t &add_blk(Block &&_blk) {
- block_t blk = new Block(std::move(_blk));
- return blk_cache.insert(std::make_pair(blk->get_hash(), blk)).first->second;
- }
-
- const block_t &add_blk(const block_t &blk) {
- return blk_cache.insert(std::make_pair(blk->get_hash(), blk)).first->second;
- }
-
- block_t find_blk(const uint256_t &blk_hash) {
- auto it = blk_cache.find(blk_hash);
- return it == blk_cache.end() ? nullptr : it->second;
- }
-
- bool is_cmd_fetched(const uint256_t &cmd_hash) {
- return cmd_cache.count(cmd_hash);
- }
-
- const command_t &add_cmd(const command_t &cmd) {
- return cmd_cache.insert(std::make_pair(cmd->get_hash(), cmd)).first->second;
- }
-
- command_t find_cmd(const uint256_t &cmd_hash) {
- auto it = cmd_cache.find(cmd_hash);
- return it == cmd_cache.end() ? nullptr: it->second;
- }
-
- size_t get_cmd_cache_size() {
- return cmd_cache.size();
- }
- size_t get_blk_cache_size() {
- return blk_cache.size();
- }
-
- bool try_release_cmd(const command_t &cmd) {
- if (cmd.get_cnt() == 2) /* only referred by cmd and the storage */
- {
- const auto &cmd_hash = cmd->get_hash();
- cmd_cache.erase(cmd_hash);
- return true;
- }
- return false;
- }
-
- bool try_release_blk(const block_t &blk) {
- if (blk.get_cnt() == 2) /* only referred by blk and the storage */
- {
- const auto &blk_hash = blk->get_hash();
-#ifdef HOTSTUFF_ENABLE_LOG_PROTO
- HOTSTUFF_LOG_INFO("releasing blk %.10s", get_hex(blk_hash).c_str());
-#endif
- for (const auto &cmd: blk->get_cmds())
- try_release_cmd(cmd);
- blk_cache.erase(blk_hash);
- return true;
- }
-#ifdef HOTSTUFF_ENABLE_LOG_PROTO
- else
- HOTSTUFF_LOG_INFO("cannot release (%lu)", blk.get_cnt());
-#endif
- return false;
- }
-};
-
-}
-
-#endif