include/hotstuff/client.h


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88

#ifndef _HOTSTUFF_CLIENT_H
#define _HOTSTUFF_CLIENT_H

#include "salticidae/msg.h"
#include "hotstuff/type.h"
#include "hotstuff/entity.h"
#include "hotstuff/consensus.h"

namespace hotstuff {

enum {
    REQ_CMD = 0x4,
    RESP_CMD = 0x5,
    CHK_CMD = 0x6
};

struct Finality: public Serializable {
    ReplicaID rid;
    int8_t decision;
    uint256_t blk_hash;
    
    public:
    Finality() = default;
    Finality(ReplicaID rid, int8_t decision, uint256_t blk_hash):
        rid(rid), decision(decision), blk_hash(blk_hash) {}

    void serialize(DataStream &s) const override {
        s << rid << decision;
        if (decision == 1) s << blk_hash;
    }

    void unserialize(DataStream &s) override {
        s >> rid >> decision;
        if (decision == 1) s >> blk_hash;
    }
};

struct MsgClient: public salticidae::MsgBase<> {
    using MsgBase::MsgBase;
    void gen_reqcmd(const Command &cmd);
    void parse_reqcmd(command_t &cmd, HotStuffCore *hsc) const;

    void gen_respcmd(const uint256_t &cmd_hash, const Finality &fin);
    void parse_respcmd(uint256_t &cmd_hash, Finality &fin) const;

    void gen_chkcmd(const uint256_t &cmd_hash);
    void parse_chkcmd(uint256_t &cmd_hash) const;
};

class CommandDummy: public Command {
    static uint64_t cnt;
    uint64_t n;
    uint256_t hash;

    public:

    CommandDummy() {}

    ~CommandDummy() override {}

    CommandDummy(uint64_t n):
        n(n), hash(salticidae::get_hash(*this)) {}

    static command_t make_cmd() {
        return new CommandDummy(cnt++);
    }

    void serialize(DataStream &s) const override {
        s << n;
    }

    void unserialize(DataStream &s) override {
        s >> n;
        hash = salticidae::get_hash(*this);
    }

    const uint256_t &get_hash() const override {
        return hash;
    }

    bool verify() const override {
        return true;
    }
};

}

#endif