stakecontendercache.h

Go to the documentation of this file.

// Copyright (c) 2024 The Bitcoin developers
// Distributed under the MIT software license, see the accompanying
// file COPYING or http://www.opensource.org/licenses/mit-license.php.
 
#ifndef BITCOIN_AVALANCHE_STAKECONTENDERCACHE_H
#define BITCOIN_AVALANCHE_STAKECONTENDERCACHE_H
 
#include <avalanche/proof.h>
#include <avalanche/stakecontender.h>
#include <primitives/blockhash.h>
#include <script/script.h>
#include <util/hasher.h>
 
#include <boost/multi_index/hashed_index.hpp>
#include <boost/multi_index/member.hpp>
#include <boost/multi_index/ordered_index.hpp>
#include <boost/multi_index_container.hpp>
 
#include <cstdint>
#include <functional>
#include <optional>
#include <utility>
#include <vector>
 
class CBlockIndex;
 
namespace avalanche {
 
enum StakeContenderStatus : uint8_t {
    UNKNOWN = 0,
    // Set according to avalanche acceptance
    ACCEPTED = (1 << 0),
    // If set, this contender should be in the stake winner set
    IN_WINNER_SET = (1 << 1),
};
 
struct StakeContenderCacheEntry {
    BlockHash prevblockhash;
    int blockheight;
    int64_t prevblocktime;
    ProofId proofid;
    uint8_t status;
    // Cache payout script and proof score because the peer manager does not
    // track past-valid proofs.
    CScript payoutScriptPubkey;
    uint32_t score;
 
    StakeContenderCacheEntry(const BlockHash &_prevblockhash, int _blockheight,
                             int64_t _prevblocktime, const ProofId &_proofid,
                             uint8_t _status,
                             const CScript &_payoutScriptPubkey,
                             uint32_t _score)
        : prevblockhash(_prevblockhash), blockheight(_blockheight),
          prevblocktime(_prevblocktime), proofid(_proofid), status(_status),
          payoutScriptPubkey(_payoutScriptPubkey), score(_score) {}
 
    double computeRewardRank() const {
        return getStakeContenderId().ComputeProofRewardRank(score);
    }
    StakeContenderId getStakeContenderId() const {
        return StakeContenderId{prevblockhash, proofid};
    }
    bool isAccepted() const { return status & StakeContenderStatus::ACCEPTED; }
    bool isInWinnerSet() const {
        return status & StakeContenderStatus::IN_WINNER_SET;
    }
};
 
struct ManualWinners {
    BlockHash prevblockhash;
    int blockheight;
    std::vector<CScript> payoutScripts;
 
    ManualWinners(const BlockHash &_prevblockhash, int _blockheight,
                  const std::vector<CScript> &_payoutScripts)
        : prevblockhash(_prevblockhash), blockheight(_blockheight),
          payoutScripts(_payoutScripts) {}
};
 
struct stakecontenderid_index {
    using result_type = StakeContenderId;
    result_type operator()(const StakeContenderCacheEntry &entry) const {
        return entry.getStakeContenderId();
    }
};
 
struct by_stakecontenderid;
struct by_prevblockhash;
struct by_blockheight;
 
namespace bmi = boost::multi_index;
 
struct StakeContenderCacheInfo {
    BlockHash prevblockhash;
    int64_t prevblocktime;
    ProofId proofid;
    int voteStatus;
};
 
class StakeContenderCache {
    int lastPromotedHeight{0};
 
    using ContenderSet = boost::multi_index_container<
        StakeContenderCacheEntry,
        bmi::indexed_by<
            // index by stake contender id
            bmi::hashed_unique<bmi::tag<by_stakecontenderid>,
                               stakecontenderid_index, SaltedUint256Hasher>,
            // index by prevblockhash
            bmi::hashed_non_unique<
                bmi::tag<by_prevblockhash>,
                bmi::member<StakeContenderCacheEntry, BlockHash,
                            &StakeContenderCacheEntry::prevblockhash>,
                SaltedUint256Hasher>,
            // index by block height
            bmi::ordered_non_unique<
                bmi::tag<by_blockheight>,
                bmi::member<StakeContenderCacheEntry, int,
                            &StakeContenderCacheEntry::blockheight>>>>;
 
    ContenderSet contenders;
 
    using ManualWinnersSet = boost::multi_index_container<
        ManualWinners,
        bmi::indexed_by<
            // index by prevblockhash
            bmi::hashed_unique<bmi::tag<by_prevblockhash>,
                               bmi::member<ManualWinners, BlockHash,
                                           &ManualWinners::prevblockhash>,
                               SaltedUint256Hasher>,
            // index by block height
            bmi::ordered_unique<
                bmi::tag<by_blockheight>,
                bmi::member<ManualWinners, int, &ManualWinners::blockheight>>>>;
 
    ManualWinnersSet manualWinners;
 
public:
    StakeContenderCache() {}
 
    void cleanup(const int requestedMinHeight);
 
    bool isEmpty() const {
        return (contenders.size() + manualWinners.size()) == 0;
    }
 
    bool add(const CBlockIndex *pindex, const ProofRef &proof,
             uint8_t status = StakeContenderStatus::UNKNOWN);
 
    void promoteToBlock(
        const CBlockIndex *activeTip,
        std::function<bool(const ProofId &proofid)> const &shouldPromote);
 
    bool setWinners(const CBlockIndex *pindex,
                    const std::vector<CScript> &payoutScripts);
 
    bool accept(const StakeContenderId &contenderId);
    bool finalize(const StakeContenderId &contenderId);
    bool reject(const StakeContenderId &contenderId);
 
    std::optional<StakeContenderCacheInfo>
    getContenderInfo(const StakeContenderId &contenderId) const;
 
    int getVoteStatus(const StakeContenderId &contenderId,
                      BlockHash &prevblockhashout) const;
 
    size_t getPollableContenders(
        const BlockHash &prevblockhash, size_t maxPollable,
        std::vector<StakeContenderId> &pollableContenders) const;
 
    bool getWinners(const BlockHash &prevblockhash,
                    std::vector<std::pair<ProofId, CScript>> &winners) const;
};
 
} // namespace avalanche
 
#endif // BITCOIN_AVALANCHE_STAKECONTENDERCACHE_H