walletutil.h

Go to the documentation of this file.

// Copyright (c) 2017-present The Bitcoin Core developers
// Distributed under the MIT software license, see the accompanying
// file COPYING or http://www.opensource.org/licenses/mit-license.php.
 
#ifndef BITCOIN_WALLET_WALLETUTIL_H
#define BITCOIN_WALLET_WALLETUTIL_H
 
#include <script/descriptor.h>
#include <util/fs.h>
 
#include <vector>
 
namespace wallet {
 
enum WalletFlags : uint64_t {
    // wallet flags in the upper section (> 1 << 31) will lead to not opening the wallet if flag is unknown
    // unknown wallet flags in the lower section <= (1 << 31) will be tolerated
 
    // will categorize coins as clean (not reused) and dirty (reused), and handle
    // them with privacy considerations in mind
    WALLET_FLAG_AVOID_REUSE = (1ULL << 0),
 
    // Indicates that the metadata has already been upgraded to contain key origins
    WALLET_FLAG_KEY_ORIGIN_METADATA = (1ULL << 1),
 
    // Indicates that the descriptor cache has been upgraded to cache last hardened xpubs
    WALLET_FLAG_LAST_HARDENED_XPUB_CACHED = (1ULL << 2),
 
    // will enforce the rule that the wallet can't contain any private keys (only watch-only/pubkeys)
    WALLET_FLAG_DISABLE_PRIVATE_KEYS = (1ULL << 32),
 
    WALLET_FLAG_BLANK_WALLET = (1ULL << 33),
 
    WALLET_FLAG_DESCRIPTORS = (1ULL << 34),
 
    WALLET_FLAG_EXTERNAL_SIGNER = (1ULL << 35),
};
 
fs::path GetWalletDir();
 
class WalletDescriptor
{
private:
    int32_t range_start = 0; // First item in range; start of range, inclusive, i.e. [range_start, range_end). This never changes.
    int32_t next_index = 0; // Position of the next item to generate
    int32_t range_end = 0; // Item after the last; end of range, exclusive, i.e. [range_start, range_end). This will increment with each TopUp()
 
    mutable std::optional<uint256> m_canonical_hash; // Hash of the canonical string, used as a shortcut for comparing canonical strings
 
    uint256 GetCanonicalHash() const;
 
public:
    const std::shared_ptr<const Descriptor> descriptor;
    uint64_t creation_time = 0;
    DescriptorCache cache;
 
    int32_t GetStart() const { return range_start; }
    int32_t GetNext() const { return next_index; }
    int32_t GetEnd() const { return range_end; }
 
    void IncNext()
    {
        next_index++;
    }
 
    void DecNext()
    {
        next_index--;
    }
 
    void SetEnd(int32_t end)
    {
        if (!descriptor->IsRange()) {
            CHECK_NONFATAL(end == 1);
        }
        range_end = end;
    }
 
    template <typename Stream>
    void Serialize(Stream& s) const
    {
        std::string descriptor_str = descriptor->ToString();
        s << descriptor_str << creation_time << next_index << range_start << range_end;
    }
 
    template <typename Stream>
    static WalletDescriptor FromStream(deserialize_type, Stream& s)
    {
        std::string descriptor_str;
        uint64_t creation_time;
        int32_t next_index, range_start, range_end;
        s >> descriptor_str >> creation_time >> next_index >> range_start >> range_end;
 
        std::string error;
        FlatSigningProvider keys;
        auto descs = Parse(descriptor_str, keys, error, true);
        if (descs.empty()) {
            throw std::ios_base::failure("Invalid descriptor: " + error);
        }
        if (descs.size() > 1) {
            throw std::ios_base::failure("Can't load a multipath descriptor from databases");
        }
        return WalletDescriptor(std::move(descs.at(0)), creation_time, range_start, range_end, next_index);
    }
 
    WalletDescriptor() = delete;
    WalletDescriptor(std::shared_ptr<Descriptor> descriptor, uint64_t creation_time, int32_t range_start, int32_t range_end, int32_t next_index)
    : range_start(descriptor->IsRange() ? range_start : 0),
      next_index(next_index),
      range_end(descriptor->IsRange() ? range_end : 1),
      descriptor(descriptor),
      creation_time(creation_time)
    {}
 
    void UpdateFrom(const WalletDescriptor& other);
 
    // Compare by using the canonical string to make the hardened indicators consistent for comparison
    bool IsCanonicallyEquivalent(const WalletDescriptor& other) const;
};
 
WalletDescriptor GenerateWalletDescriptor(const CExtPubKey& master_key, const OutputType& output_type, bool internal);
} // namespace wallet
 
#endif // BITCOIN_WALLET_WALLETUTIL_H