/root/bitcoin/src/wallet/db.h

Source (jump to first uncovered line)
// Copyright (c) 2009-2010 Satoshi Nakamoto
// Copyright (c) 2009-2021 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_DB_H
#define BITCOIN_WALLET_DB_H

#include <clientversion.h>
#include <streams.h>
#include <support/allocators/secure.h>
#include <util/fs.h>

#include <atomic>
#include <memory>
#include <optional>
#include <string>

class ArgsManager;
struct bilingual_str;

namespace wallet {
// BytePrefix compares equality with other byte spans that begin with the same prefix.
struct BytePrefix {
    Span<const std::byte> prefix;
};
bool operator<(BytePrefix a, Span<const std::byte> b);
bool operator<(Span<const std::byte> a, BytePrefix b);

class DatabaseCursor
{
public:
    explicit DatabaseCursor() = default;
    virtual ~DatabaseCursor() = default;

    DatabaseCursor(const DatabaseCursor&) = delete;
    DatabaseCursor& operator=(const DatabaseCursor&) = delete;

    enum class Status
    {
        FAIL,
        MORE,
        DONE,
    };

    virtual Status Next(DataStream& key, DataStream& value) { return Status::FAIL; }
};

/** RAII class that provides access to a WalletDatabase */
class DatabaseBatch
{
private:
    virtual bool ReadKey(DataStream&& key, DataStream& value) = 0;
    virtual bool WriteKey(DataStream&& key, DataStream&& value, bool overwrite = true) = 0;
    virtual bool EraseKey(DataStream&& key) = 0;
    virtual bool HasKey(DataStream&& key) = 0;

public:
    explicit DatabaseBatch() = default;
    virtual ~DatabaseBatch() = default;

    DatabaseBatch(const DatabaseBatch&) = delete;
    DatabaseBatch& operator=(const DatabaseBatch&) = delete;

    virtual void Flush() = 0;
    virtual void Close() = 0;

    template <typename K, typename T>
    bool Read(const K& key, T& value)
    {
        DataStream ssKey{};
        ssKey.reserve(1000);
        ssKey << key;

        DataStream ssValue{};
        if (!ReadKey(std::move(ssKey), ssValue)) return false;
        try {
            ssValue >> value;
            return true;
        } catch (const std::exception&) {
            return false;
        }
    }

    template <typename K, typename T>
    bool Write(const K& key, const T& value, bool fOverwrite = true)
    {
        DataStream ssKey{};
        ssKey.reserve(1000);
        ssKey << key;

        DataStream ssValue{};
        ssValue.reserve(10000);
        ssValue << value;

        return WriteKey(std::move(ssKey), std::move(ssValue), fOverwrite);
    }

    template <typename K>
    bool Erase(const K& key)
    {
        DataStream ssKey{};
        ssKey.reserve(1000);
        ssKey << key;

        return EraseKey(std::move(ssKey));
    }

    template <typename K>
    bool Exists(const K& key)
    {
        DataStream ssKey{};
        ssKey.reserve(1000);
        ssKey << key;

        return HasKey(std::move(ssKey));
    }
    virtual bool ErasePrefix(Span<const std::byte> prefix) = 0;

    virtual std::unique_ptr<DatabaseCursor> GetNewCursor() = 0;
    virtual std::unique_ptr<DatabaseCursor> GetNewPrefixCursor(Span<const std::byte> prefix) = 0;
    virtual bool TxnBegin() = 0;
    virtual bool TxnCommit() = 0;
    virtual bool TxnAbort() = 0;
    virtual bool HasActiveTxn() = 0;
};

/** An instance of this class represents one database.
 **/
class WalletDatabase
{
public:
    /** Create dummy DB handle */
    WalletDatabase() : nUpdateCounter(0) {}
    virtual ~WalletDatabase() = default;

    /** Open the database if it is not already opened. */
    virtual void Open() = 0;

    //! Counts the number of active database users to be sure that the database is not closed while someone is using it
    std::atomic<int> m_refcount{0};
    /** Indicate the a new database user has began using the database. Increments m_refcount */
    virtual void AddRef() = 0;
    /** Indicate that database user has stopped using the database and that it could be flushed or closed. Decrement m_refcount */
    virtual void RemoveRef() = 0;

    /** Rewrite the entire database on disk, with the exception of key pszSkip if non-zero
     */
    virtual bool Rewrite(const char* pszSkip=nullptr) = 0;

    /** Back up the entire database to a file.
     */
    virtual bool Backup(const std::string& strDest) const = 0;

    /** Make sure all changes are flushed to database file.
     */
    virtual void Flush() = 0;
    /** Flush to the database file and close the database.
     *  Also close the environment if no other databases are open in it.
     */
    virtual void Close() = 0;
    /* flush the wallet passively (TRY_LOCK)
       ideal to be called periodically */
    virtual bool PeriodicFlush() = 0;

    virtual void IncrementUpdateCounter() = 0;

    virtual void ReloadDbEnv() = 0;

    /** Return path to main database file for logs and error messages. */
    virtual std::string Filename() = 0;

    virtual std::string Format() = 0;

    std::atomic<unsigned int> nUpdateCounter;
    unsigned int nLastSeen{0};
    unsigned int nLastFlushed{0};
    int64_t nLastWalletUpdate{0};

    /** Make a DatabaseBatch connected to this database */
    virtual std::unique_ptr<DatabaseBatch> MakeBatch(bool flush_on_close = true) = 0;
};

enum class DatabaseFormat {
    BERKELEY,
    SQLITE,
    BERKELEY_RO,
    BERKELEY_SWAP,
};

struct DatabaseOptions {
    bool require_existing = false;
    bool require_create = false;
    std::optional<DatabaseFormat> require_format;
    uint64_t create_flags = 0;
    SecureString create_passphrase;

    // Specialized options. Not every option is supported by every backend.
    bool verify = true;             //!< Check data integrity on load.
    bool use_unsafe_sync = false;   //!< Disable file sync for faster performance.
    bool use_shared_memory = false; //!< Let other processes access the database.
    int64_t max_log_mb = 100;       //!< Max log size to allow before consolidating.
};

enum class DatabaseStatus {
    SUCCESS,
    FAILED_BAD_PATH,
    FAILED_BAD_FORMAT,
    FAILED_ALREADY_LOADED,
    FAILED_ALREADY_EXISTS,
    FAILED_NOT_FOUND,
    FAILED_CREATE,
    FAILED_LOAD,
    FAILED_VERIFY,
    FAILED_ENCRYPT,
    FAILED_INVALID_BACKUP_FILE,
};

/** Recursively list database paths in directory. */
std::vector<std::pair<fs::path, std::string>> ListDatabases(const fs::path& path);

void ReadDatabaseArgs(const ArgsManager& args, DatabaseOptions& options);
std::unique_ptr<WalletDatabase> MakeDatabase(const fs::path& path, const DatabaseOptions& options, DatabaseStatus& status, bilingual_str& error);

fs::path BDBDataFile(const fs::path& path);
fs::path SQLiteDataFile(const fs::path& path);
bool IsBDBFile(const fs::path& path);
bool IsSQLiteFile(const fs::path& path);
} // namespace wallet

#endif // BITCOIN_WALLET_DB_H

Coverage Report

Created: 2025-02-21 14:36

Line	Count	Source (jump to first uncovered line)
1		// Copyright (c) 2009-2010 Satoshi Nakamoto
2		// Copyright (c) 2009-2021 The Bitcoin Core developers
3		// Distributed under the MIT software license, see the accompanying
4		// file COPYING or http://www.opensource.org/licenses/mit-license.php.
5
6		#ifndef BITCOIN_WALLET_DB_H
7		#define BITCOIN_WALLET_DB_H
8
9		#include <clientversion.h>
10		#include <streams.h>
11		#include <support/allocators/secure.h>
12		#include <util/fs.h>
13
14		#include <atomic>
15		#include <memory>
16		#include <optional>
17		#include <string>
18
19		class ArgsManager;
20		struct bilingual_str;
21
22		namespace wallet {
23		// BytePrefix compares equality with other byte spans that begin with the same prefix.
24		struct BytePrefix {
25		Span<const std::byte> prefix;
26		};
27		bool operator<(BytePrefix a, Span<const std::byte> b);
28		bool operator<(Span<const std::byte> a, BytePrefix b);
29
30		class DatabaseCursor
31		{
32		public:
33	0	explicit DatabaseCursor() = default;
34	0	virtual ~DatabaseCursor() = default;
35
36		DatabaseCursor(const DatabaseCursor&) = delete;
37		DatabaseCursor& operator=(const DatabaseCursor&) = delete;
38
39		enum class Status
40		{
41		FAIL,
42		MORE,
43		DONE,
44		};
45
46	0	virtual Status Next(DataStream& key, DataStream& value) { return Status::FAIL; }
47		};
48
49		/** RAII class that provides access to a WalletDatabase */
50		class DatabaseBatch
51		{
52		private:
53		virtual bool ReadKey(DataStream&& key, DataStream& value) = 0;
54		virtual bool WriteKey(DataStream&& key, DataStream&& value, bool overwrite = true) = 0;
55		virtual bool EraseKey(DataStream&& key) = 0;
56		virtual bool HasKey(DataStream&& key) = 0;
57
58		public:
59	0	explicit DatabaseBatch() = default;
60	0	virtual ~DatabaseBatch() = default;
61
62		DatabaseBatch(const DatabaseBatch&) = delete;
63		DatabaseBatch& operator=(const DatabaseBatch&) = delete;
64
65		virtual void Flush() = 0;
66		virtual void Close() = 0;
67
68		template <typename K, typename T>
69		bool Read(const K& key, T& value)
70	0	{
71	0	DataStream ssKey{};
72	0	ssKey.reserve(1000);
73	0	ssKey << key;
74
75	0	DataStream ssValue{};
76	0	if (!ReadKey(std::move(ssKey), ssValue)) return false;
77	0	try {
78	0	ssValue >> value;
79	0	return true;
80	0	} catch (const std::exception&) {
81	0	return false;
82	0	}
83	0	} Unexecuted instantiation: _ZN6wallet13DatabaseBatch4ReadINSt7__cxx1112basic_stringIcSt11char_traitsIcESaIcEEEmEEbRKT_RT0_ Unexecuted instantiation: _ZN6wallet13DatabaseBatch4ReadISt4pairINSt7__cxx1112basic_stringIcSt11char_traitsIcESaIcEEE7CPubKeyESt6vectorIhSaIhEEEEbRKT_RT0_ Unexecuted instantiation: _ZN6wallet13DatabaseBatch4ReadINSt7__cxx1112basic_stringIcSt11char_traitsIcESaIcEEE13CBlockLocatorEEbRKT_RT0_ Unexecuted instantiation: _ZN6wallet13DatabaseBatch4ReadISt4pairINSt7__cxx1112basic_stringIcSt11char_traitsIcESaIcEEElENS_8CKeyPoolEEEbRKT_RT0_ Unexecuted instantiation: _ZN6wallet13DatabaseBatch4ReadINSt7__cxx1112basic_stringIcSt11char_traitsIcESaIcEEEiEEbRKT_RT0_
84
85		template <typename K, typename T>
86		bool Write(const K& key, const T& value, bool fOverwrite = true)
87	0	{
88	0	DataStream ssKey{};
89	0	ssKey.reserve(1000);
90	0	ssKey << key;
91
92	0	DataStream ssValue{};
93	0	ssValue.reserve(10000);
94	0	ssValue << value;
95
96	0	return WriteKey(std::move(ssKey), std::move(ssValue), fOverwrite);
97	0	} Unexecuted instantiation: _ZN6wallet13DatabaseBatch5WriteI4SpanIhES3_EEbRKT_RKT0_b Unexecuted instantiation: _ZN6wallet13DatabaseBatch5WriteI4SpanIKSt4byteES5_EEbRKT_RKT0_b Unexecuted instantiation: _ZN6wallet13DatabaseBatch5WriteISt4pairINSt7__cxx1112basic_stringIcSt11char_traitsIcESaIcEEES8_ES8_EEbRKT_RKT0_b Unexecuted instantiation: _ZN6wallet13DatabaseBatch5WriteISt4pairINSt7__cxx1112basic_stringIcSt11char_traitsIcESaIcEEE22transaction_identifierILb0EEENS_9CWalletTxEEEbRKT_RKT0_b Unexecuted instantiation: _ZN6wallet13DatabaseBatch5WriteISt4pairINSt7__cxx1112basic_stringIcSt11char_traitsIcESaIcEEE7CPubKeyENS_12CKeyMetadataEEEbRKT_RKT0_b Unexecuted instantiation: _ZN6wallet13DatabaseBatch5WriteISt4pairINSt7__cxx1112basic_stringIcSt11char_traitsIcESaIcEEE7CPubKeyES2_ISt6vectorIh16secure_allocatorIhEE7uint256EEEbRKT_RKT0_b Unexecuted instantiation: _ZN6wallet13DatabaseBatch5WriteISt4pairINSt7__cxx1112basic_stringIcSt11char_traitsIcESaIcEEE7CPubKeyES2_ISt6vectorIhSaIhEE7uint256EEEbRKT_RKT0_b Unexecuted instantiation: _ZN6wallet13DatabaseBatch5WriteISt4pairINSt7__cxx1112basic_stringIcSt11char_traitsIcESaIcEEEjENS_10CMasterKeyEEEbRKT_RKT0_b Unexecuted instantiation: _ZN6wallet13DatabaseBatch5WriteISt4pairINSt7__cxx1112basic_stringIcSt11char_traitsIcESaIcEEE7uint160E7CScriptEEbRKT_RKT0_b Unexecuted instantiation: _ZN6wallet13DatabaseBatch5WriteISt4pairINSt7__cxx1112basic_stringIcSt11char_traitsIcESaIcEEE7CScriptENS_12CKeyMetadataEEEbRKT_RKT0_b Unexecuted instantiation: _ZN6wallet13DatabaseBatch5WriteISt4pairINSt7__cxx1112basic_stringIcSt11char_traitsIcESaIcEEE7CScriptEhEEbRKT_RKT0_b Unexecuted instantiation: _ZN6wallet13DatabaseBatch5WriteINSt7__cxx1112basic_stringIcSt11char_traitsIcESaIcEEE13CBlockLocatorEEbRKT_RKT0_b Unexecuted instantiation: _ZN6wallet13DatabaseBatch5WriteINSt7__cxx1112basic_stringIcSt11char_traitsIcESaIcEEElEEbRKT_RKT0_b Unexecuted instantiation: _ZN6wallet13DatabaseBatch5WriteISt4pairINSt7__cxx1112basic_stringIcSt11char_traitsIcESaIcEEElENS_8CKeyPoolEEEbRKT_RKT0_b Unexecuted instantiation: _ZN6wallet13DatabaseBatch5WriteISt4pairINSt7__cxx1112basic_stringIcSt11char_traitsIcESaIcEEEhE7uint256EEbRKT_RKT0_b Unexecuted instantiation: _ZN6wallet13DatabaseBatch5WriteISt4pairINSt7__cxx1112basic_stringIcSt11char_traitsIcESaIcEEES2_I7uint2567CPubKeyEES2_ISt6vectorIh16secure_allocatorIhEES9_EEEbRKT_RKT0_b Unexecuted instantiation: _ZN6wallet13DatabaseBatch5WriteISt4pairINSt7__cxx1112basic_stringIcSt11char_traitsIcESaIcEEES2_I7uint2567CPubKeyEESt6vectorIhSaIhEEEEbRKT_RKT0_b Unexecuted instantiation: _ZN6wallet13DatabaseBatch5WriteISt4pairINSt7__cxx1112basic_stringIcSt11char_traitsIcESaIcEEE7uint256ENS_16WalletDescriptorEEEbRKT_RKT0_b Unexecuted instantiation: _ZN6wallet13DatabaseBatch5WriteISt4pairIS2_INSt7__cxx1112basic_stringIcSt11char_traitsIcESaIcEEE7uint256ES2_IjjEESt6vectorIhSaIhEEEEbRKT_RKT0_b Unexecuted instantiation: _ZN6wallet13DatabaseBatch5WriteISt4pairIS2_INSt7__cxx1112basic_stringIcSt11char_traitsIcESaIcEEE7uint256EjESt6vectorIhSaIhEEEEbRKT_RKT0_b Unexecuted instantiation: _ZN6wallet13DatabaseBatch5WriteISt4pairINSt7__cxx1112basic_stringIcSt11char_traitsIcESaIcEEES2_I22transaction_identifierILb0EEjEEhEEbRKT_RKT0_b Unexecuted instantiation: _ZN6wallet13DatabaseBatch5WriteINSt7__cxx1112basic_stringIcSt11char_traitsIcESaIcEEEiEEbRKT_RKT0_b Unexecuted instantiation: _ZN6wallet13DatabaseBatch5WriteISt4pairINSt7__cxx1112basic_stringIcSt11char_traitsIcESaIcEEES2_IS8_S8_EES8_EEbRKT_RKT0_b Unexecuted instantiation: _ZN6wallet13DatabaseBatch5WriteINSt7__cxx1112basic_stringIcSt11char_traitsIcESaIcEEENS_8CHDChainEEEbRKT_RKT0_b Unexecuted instantiation: _ZN6wallet13DatabaseBatch5WriteINSt7__cxx1112basic_stringIcSt11char_traitsIcESaIcEEEmEEbRKT_RKT0_b
98
99		template <typename K>
100		bool Erase(const K& key)
101	0	{
102	0	DataStream ssKey{};
103	0	ssKey.reserve(1000);
104	0	ssKey << key;
105
106	0	return EraseKey(std::move(ssKey));
107	0	} Unexecuted instantiation: _ZN6wallet13DatabaseBatch5EraseISt4pairINSt7__cxx1112basic_stringIcSt11char_traitsIcESaIcEEES8_EEEbRKT_ Unexecuted instantiation: _ZN6wallet13DatabaseBatch5EraseISt4pairINSt7__cxx1112basic_stringIcSt11char_traitsIcESaIcEEE7uint256EEEbRKT_ Unexecuted instantiation: _ZN6wallet13DatabaseBatch5EraseISt4pairINSt7__cxx1112basic_stringIcSt11char_traitsIcESaIcEEE7CPubKeyEEEbRKT_ Unexecuted instantiation: _ZN6wallet13DatabaseBatch5EraseISt4pairINSt7__cxx1112basic_stringIcSt11char_traitsIcESaIcEEEjEEEbRKT_ Unexecuted instantiation: _ZN6wallet13DatabaseBatch5EraseISt4pairINSt7__cxx1112basic_stringIcSt11char_traitsIcESaIcEEE7CScriptEEEbRKT_ Unexecuted instantiation: _ZN6wallet13DatabaseBatch5EraseISt4pairINSt7__cxx1112basic_stringIcSt11char_traitsIcESaIcEEElEEEbRKT_ Unexecuted instantiation: _ZN6wallet13DatabaseBatch5EraseISt4pairINSt7__cxx1112basic_stringIcSt11char_traitsIcESaIcEEEhEEEbRKT_ Unexecuted instantiation: _ZN6wallet13DatabaseBatch5EraseISt4pairINSt7__cxx1112basic_stringIcSt11char_traitsIcESaIcEEES2_I7uint2567CPubKeyEEEEbRKT_ Unexecuted instantiation: _ZN6wallet13DatabaseBatch5EraseISt4pairINSt7__cxx1112basic_stringIcSt11char_traitsIcESaIcEEES2_I22transaction_identifierILb0EEjEEEEbRKT_ Unexecuted instantiation: _ZN6wallet13DatabaseBatch5EraseISt4pairINSt7__cxx1112basic_stringIcSt11char_traitsIcESaIcEEES2_IS8_S8_EEEEbRKT_
108
109		template <typename K>
110		bool Exists(const K& key)
111		{
112		DataStream ssKey{};
113		ssKey.reserve(1000);
114		ssKey << key;
115
116		return HasKey(std::move(ssKey));
117		}
118		virtual bool ErasePrefix(Span<const std::byte> prefix) = 0;
119
120		virtual std::unique_ptr<DatabaseCursor> GetNewCursor() = 0;
121		virtual std::unique_ptr<DatabaseCursor> GetNewPrefixCursor(Span<const std::byte> prefix) = 0;
122		virtual bool TxnBegin() = 0;
123		virtual bool TxnCommit() = 0;
124		virtual bool TxnAbort() = 0;
125		virtual bool HasActiveTxn() = 0;
126		};
127
128		/** An instance of this class represents one database.
129		**/
130		class WalletDatabase
131		{
132		public:
133		/** Create dummy DB handle */
134	0	WalletDatabase() : nUpdateCounter(0) {}
135	0	virtual ~WalletDatabase() = default;
136
137		/** Open the database if it is not already opened. */
138		virtual void Open() = 0;
139
140		//! Counts the number of active database users to be sure that the database is not closed while someone is using it
141		std::atomic<int> m_refcount{0};
142		/** Indicate the a new database user has began using the database. Increments m_refcount */
143		virtual void AddRef() = 0;
144		/** Indicate that database user has stopped using the database and that it could be flushed or closed. Decrement m_refcount */
145		virtual void RemoveRef() = 0;
146
147		/** Rewrite the entire database on disk, with the exception of key pszSkip if non-zero
148		*/
149		virtual bool Rewrite(const char* pszSkip=nullptr) = 0;
150
151		/** Back up the entire database to a file.
152		*/
153		virtual bool Backup(const std::string& strDest) const = 0;
154
155		/** Make sure all changes are flushed to database file.
156		*/
157		virtual void Flush() = 0;
158		/** Flush to the database file and close the database.
159		* Also close the environment if no other databases are open in it.
160		*/
161		virtual void Close() = 0;
162		/* flush the wallet passively (TRY_LOCK)
163		ideal to be called periodically */
164		virtual bool PeriodicFlush() = 0;
165
166		virtual void IncrementUpdateCounter() = 0;
167
168		virtual void ReloadDbEnv() = 0;
169
170		/** Return path to main database file for logs and error messages. */
171		virtual std::string Filename() = 0;
172
173		virtual std::string Format() = 0;
174
175		std::atomic<unsigned int> nUpdateCounter;
176		unsigned int nLastSeen{0};
177		unsigned int nLastFlushed{0};
178		int64_t nLastWalletUpdate{0};
179
180		/** Make a DatabaseBatch connected to this database */
181		virtual std::unique_ptr<DatabaseBatch> MakeBatch(bool flush_on_close = true) = 0;
182		};
183
184		enum class DatabaseFormat {
185		BERKELEY,
186		SQLITE,
187		BERKELEY_RO,
188		BERKELEY_SWAP,
189		};
190
191		struct DatabaseOptions {
192		bool require_existing = false;
193		bool require_create = false;
194		std::optional<DatabaseFormat> require_format;
195		uint64_t create_flags = 0;
196		SecureString create_passphrase;
197
198		// Specialized options. Not every option is supported by every backend.
199		bool verify = true; //!< Check data integrity on load.
200		bool use_unsafe_sync = false; //!< Disable file sync for faster performance.
201		bool use_shared_memory = false; //!< Let other processes access the database.
202		int64_t max_log_mb = 100; //!< Max log size to allow before consolidating.
203		};
204
205		enum class DatabaseStatus {
206		SUCCESS,
207		FAILED_BAD_PATH,
208		FAILED_BAD_FORMAT,
209		FAILED_ALREADY_LOADED,
210		FAILED_ALREADY_EXISTS,
211		FAILED_NOT_FOUND,
212		FAILED_CREATE,
213		FAILED_LOAD,
214		FAILED_VERIFY,
215		FAILED_ENCRYPT,
216		FAILED_INVALID_BACKUP_FILE,
217		};
218
219		/** Recursively list database paths in directory. */
220		std::vector<std::pair<fs::path, std::string>> ListDatabases(const fs::path& path);
221
222		void ReadDatabaseArgs(const ArgsManager& args, DatabaseOptions& options);
223		std::unique_ptr<WalletDatabase> MakeDatabase(const fs::path& path, const DatabaseOptions& options, DatabaseStatus& status, bilingual_str& error);
224
225		fs::path BDBDataFile(const fs::path& path);
226		fs::path SQLiteDataFile(const fs::path& path);
227		bool IsBDBFile(const fs::path& path);
228		bool IsSQLiteFile(const fs::path& path);
229		} // namespace wallet
230
231		#endif // BITCOIN_WALLET_DB_H