privatesend-util.h

Go to the documentation of this file.

// Copyright (c) 2014-2019 The Dash Core developers
// Distributed under the MIT/X11 software license, see the accompanying
// file COPYING or http://www.opensource.org/licenses/mit-license.php.

#ifndef PRIVATESENDUTIL_H
#define PRIVATESENDUTIL_H

#include <wallet/wallet.h>

class CTransactionBuilder;

class CKeyHolder
{
private:
    CReserveKey reserveKey;
    CPubKey pubKey;

public:
    CKeyHolder(CWallet* pwalletIn);
    CKeyHolder(CKeyHolder&&) = delete;
    CKeyHolder& operator=(CKeyHolder&&) = delete;
    void KeepKey();
    void ReturnKey();

    CScript GetScriptForDestination() const;
};

class CKeyHolderStorage
{
private:
    std::vector<std::unique_ptr<CKeyHolder> > storage;
    mutable CCriticalSection cs_storage;

public:
    CScript AddKey(CWallet* pwalletIn);
    void KeepAll();
    void ReturnAll();
};

class CTransactionBuilderOutput
{
    CTransactionBuilder* pTxBuilder{nullptr};
    CReserveKey key;
    CAmount nAmount{0};
    CScript script;

public:
    CTransactionBuilderOutput(CTransactionBuilder* pTxBuilderIn, CWallet* pwalletIn, CAmount nAmountIn);
    CTransactionBuilderOutput(CTransactionBuilderOutput&&) = delete;
    CTransactionBuilderOutput& operator=(CTransactionBuilderOutput&&) = delete;
    CScript GetScript() const { return script; }
    CAmount GetAmount() const { return nAmount; }
    bool UpdateAmount(CAmount nAmount);
    void KeepKey() { key.KeepKey(); }
    void ReturnKey() { key.ReturnKey(); }
};

class CTransactionBuilder
{
    CWallet* pwallet{nullptr};
    CCoinControl coinControl;
    CReserveKey dummyReserveKey;
    CompactTallyItem tallyItem;
    int nBytesBase{0};
    int nBytesOutput{0};
    bool fKeepKeys{false};
    mutable CCriticalSection cs_outputs;
    std::vector<std::unique_ptr<CTransactionBuilderOutput>> vecOutputs;
    friend class CTransactionBuilderOutput;

public:
    CTransactionBuilder(CWallet* pwalletIn, const CompactTallyItem& tallyItemIn);
    ~CTransactionBuilder();
    bool CouldAddOutput(CAmount nAmountOutput) const;
    bool CouldAddOutputs(const std::vector<CAmount>& vecOutputAmounts) const;
    CTransactionBuilderOutput* AddOutput(CAmount nAmountOutput = 0);
    CAmount GetAmountInitial() const { return tallyItem.nAmount; }
    CAmount GetAmountLeft() const { return GetAmountInitial() - GetAmountUsed() - GetFee(GetBytesTotal()); }
    bool IsDust(CAmount nAmount) const;
    int CountOutputs() const { return vecOutputs.size(); }
    bool Commit(std::string& strResult);
    std::string ToString() const;

private:
    void Clear();
    unsigned int GetBytesTotal() const;
    static CAmount GetAmountLeft(CAmount nAmountInitial, CAmount nAmountUsed, CAmount nFee);
    CAmount GetAmountUsed() const;
    CAmount GetFee(unsigned int nBytes) const;
    int GetSizeOfCompactSizeDiff(size_t nAdd) const;
};

#endif //PRIVATESENDUTIL_H