path: root/include/hotstuff/client.h

#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 {

struct MsgReqCmd {
    static const opcode_t opcode = 0x4;
    DataStream serialized;
    command_t cmd;
    MsgReqCmd(const Command &cmd) { serialized << cmd; }
    MsgReqCmd(DataStream &&s): serialized(std::move(s)) {}
};

struct MsgRespCmd {
    static const opcode_t opcode = 0x5;
    DataStream serialized;
#if HOTSTUFF_CMD_RESPSIZE > 0
    uint8_t payload[HOTSTUFF_CMD_RESPSIZE];
#endif
    Finality fin;
    MsgRespCmd(const Finality &fin) {
        serialized << fin;
#if HOTSTUFF_CMD_RESPSIZE > 0
        serialized.put_data(payload, payload + HOTSTUFF_CMD_RESPSIZE);
#endif
    }
    MsgRespCmd(DataStream &&s) { s >> fin; }
};

class CommandDummy: public Command {
    uint32_t cid;
    uint32_t n;
    uint256_t hash;
#if HOTSTUFF_CMD_REQSIZE > 0
    uint8_t payload[HOTSTUFF_CMD_REQSIZE];
#endif
    uint256_t compute_hash() {
        DataStream s;
        s << cid << n;
#if HOTSTUFF_CMD_REQSIZE > 0
        s << payload;
#endif
        return s.get_hash();
    }

    public:
    CommandDummy() {}
    ~CommandDummy() override {}

    CommandDummy(uint32_t cid, uint32_t n):
        cid(cid), n(n), hash(compute_hash()) {}

    void serialize(DataStream &s) const override {
        s << cid << n;
#if HOTSTUFF_CMD_REQSIZE > 0
        s.put_data(payload, payload + HOTSTUFF_CMD_REQSIZE);
#endif
    }

    void unserialize(DataStream &s) override {
        s >> cid >> n;
        hash = compute_hash();
    }

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

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

}

#endif