LCOV - code coverage report
Current view: top level - src/wallet - db.h (source / functions) Hit Total Coverage
Test: fuzz_coverage.info Lines: 0 46 0.0 %
Date: 2023-11-12 01:39:15 Functions: 0 52 0.0 %
Branches: 0 416 0.0 %

           Branch data     Line data    Source code
       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                 :            : void SplitWalletPath(const fs::path& wallet_path, fs::path& env_directory, std::string& database_filename);
      24                 :            : 
      25                 :            : class DatabaseCursor
      26                 :            : {
      27                 :            : public:
      28                 :          0 :     explicit DatabaseCursor() {}
      29                 :          0 :     virtual ~DatabaseCursor() {}
      30                 :            : 
      31                 :            :     DatabaseCursor(const DatabaseCursor&) = delete;
      32                 :            :     DatabaseCursor& operator=(const DatabaseCursor&) = delete;
      33                 :            : 
      34                 :            :     enum class Status
      35                 :            :     {
      36                 :            :         FAIL,
      37                 :            :         MORE,
      38                 :            :         DONE,
      39                 :            :     };
      40                 :            : 
      41                 :          0 :     virtual Status Next(DataStream& key, DataStream& value) { return Status::FAIL; }
      42                 :            : };
      43                 :            : 
      44                 :            : /** RAII class that provides access to a WalletDatabase */
      45                 :            : class DatabaseBatch
      46                 :            : {
      47                 :            : private:
      48                 :            :     virtual bool ReadKey(DataStream&& key, DataStream& value) = 0;
      49                 :            :     virtual bool WriteKey(DataStream&& key, DataStream&& value, bool overwrite = true) = 0;
      50                 :            :     virtual bool EraseKey(DataStream&& key) = 0;
      51                 :            :     virtual bool HasKey(DataStream&& key) = 0;
      52                 :            : 
      53                 :            : public:
      54                 :          0 :     explicit DatabaseBatch() {}
      55                 :          0 :     virtual ~DatabaseBatch() {}
      56                 :            : 
      57                 :            :     DatabaseBatch(const DatabaseBatch&) = delete;
      58                 :            :     DatabaseBatch& operator=(const DatabaseBatch&) = delete;
      59                 :            : 
      60                 :            :     virtual void Flush() = 0;
      61                 :            :     virtual void Close() = 0;
      62                 :            : 
      63                 :            :     template <typename K, typename T>
      64                 :          0 :     bool Read(const K& key, T& value)
      65                 :            :     {
      66                 :          0 :         DataStream ssKey{};
      67 [ #  # ][ #  # ]:          0 :         ssKey.reserve(1000);
         [ #  # ][ #  # ]
                 [ #  # ]
      68 [ #  # ][ #  # ]:          0 :         ssKey << key;
         [ #  # ][ #  # ]
                 [ #  # ]
      69                 :            : 
      70 [ #  # ][ #  # ]:          0 :         CDataStream ssValue(SER_DISK, CLIENT_VERSION);
         [ #  # ][ #  # ]
                 [ #  # ]
      71 [ #  # ][ #  # ]:          0 :         if (!ReadKey(std::move(ssKey), ssValue)) return false;
         [ #  # ][ #  # ]
         [ #  # ][ #  # ]
         [ #  # ][ #  # ]
         [ #  # ][ #  # ]
      72                 :            :         try {
      73 [ #  # ][ #  # ]:          0 :             ssValue >> value;
         [ #  # ][ #  # ]
                 [ #  # ]
      74                 :          0 :             return true;
      75 [ #  # ][ #  # ]:          0 :         } catch (const std::exception&) {
         [ #  # ][ #  # ]
                 [ #  # ]
      76                 :          0 :             return false;
      77 [ #  # ][ #  # ]:          0 :         }
         [ #  # ][ #  # ]
                 [ #  # ]
      78                 :          0 :     }
      79                 :            : 
      80                 :            :     template <typename K, typename T>
      81                 :          0 :     bool Write(const K& key, const T& value, bool fOverwrite = true)
      82                 :            :     {
      83                 :          0 :         DataStream ssKey{};
      84 [ #  # ][ #  # ]:          0 :         ssKey.reserve(1000);
         [ #  # ][ #  # ]
         [ #  # ][ #  # ]
         [ #  # ][ #  # ]
         [ #  # ][ #  # ]
         [ #  # ][ #  # ]
         [ #  # ][ #  # ]
         [ #  # ][ #  # ]
         [ #  # ][ #  # ]
         [ #  # ][ #  # ]
         [ #  # ][ #  # ]
                 [ #  # ]
      85 [ #  # ][ #  # ]:          0 :         ssKey << key;
         [ #  # ][ #  # ]
         [ #  # ][ #  # ]
         [ #  # ][ #  # ]
         [ #  # ][ #  # ]
         [ #  # ][ #  # ]
         [ #  # ][ #  # ]
         [ #  # ][ #  # ]
         [ #  # ][ #  # ]
         [ #  # ][ #  # ]
         [ #  # ][ #  # ]
                 [ #  # ]
      86                 :            : 
      87 [ #  # ][ #  # ]:          0 :         CDataStream ssValue(SER_DISK, CLIENT_VERSION);
         [ #  # ][ #  # ]
         [ #  # ][ #  # ]
         [ #  # ][ #  # ]
         [ #  # ][ #  # ]
         [ #  # ][ #  # ]
         [ #  # ][ #  # ]
         [ #  # ][ #  # ]
         [ #  # ][ #  # ]
         [ #  # ][ #  # ]
         [ #  # ][ #  # ]
                 [ #  # ]
      88 [ #  # ][ #  # ]:          0 :         ssValue.reserve(10000);
         [ #  # ][ #  # ]
         [ #  # ][ #  # ]
         [ #  # ][ #  # ]
         [ #  # ][ #  # ]
         [ #  # ][ #  # ]
         [ #  # ][ #  # ]
         [ #  # ][ #  # ]
         [ #  # ][ #  # ]
         [ #  # ][ #  # ]
         [ #  # ][ #  # ]
                 [ #  # ]
      89 [ #  # ][ #  # ]:          0 :         ssValue << value;
         [ #  # ][ #  # ]
         [ #  # ][ #  # ]
         [ #  # ][ #  # ]
         [ #  # ][ #  # ]
         [ #  # ][ #  # ]
         [ #  # ][ #  # ]
         [ #  # ][ #  # ]
         [ #  # ][ #  # ]
         [ #  # ][ #  # ]
         [ #  # ][ #  # ]
                 [ #  # ]
      90                 :            : 
      91 [ #  # ][ #  # ]:          0 :         return WriteKey(std::move(ssKey), std::move(ssValue), fOverwrite);
         [ #  # ][ #  # ]
         [ #  # ][ #  # ]
         [ #  # ][ #  # ]
         [ #  # ][ #  # ]
         [ #  # ][ #  # ]
         [ #  # ][ #  # ]
         [ #  # ][ #  # ]
         [ #  # ][ #  # ]
         [ #  # ][ #  # ]
         [ #  # ][ #  # ]
                 [ #  # ]
      92                 :          0 :     }
      93                 :            : 
      94                 :            :     template <typename K>
      95                 :          0 :     bool Erase(const K& key)
      96                 :            :     {
      97                 :          0 :         DataStream ssKey{};
      98 [ #  # ][ #  # ]:          0 :         ssKey.reserve(1000);
         [ #  # ][ #  # ]
         [ #  # ][ #  # ]
         [ #  # ][ #  # ]
         [ #  # ][ #  # ]
      99 [ #  # ][ #  # ]:          0 :         ssKey << key;
         [ #  # ][ #  # ]
         [ #  # ][ #  # ]
         [ #  # ][ #  # ]
         [ #  # ][ #  # ]
     100                 :            : 
     101 [ #  # ][ #  # ]:          0 :         return EraseKey(std::move(ssKey));
         [ #  # ][ #  # ]
         [ #  # ][ #  # ]
         [ #  # ][ #  # ]
         [ #  # ][ #  # ]
     102                 :          0 :     }
     103                 :            : 
     104                 :            :     template <typename K>
     105                 :            :     bool Exists(const K& key)
     106                 :            :     {
     107                 :            :         DataStream ssKey{};
     108                 :            :         ssKey.reserve(1000);
     109                 :            :         ssKey << key;
     110                 :            : 
     111                 :            :         return HasKey(std::move(ssKey));
     112                 :            :     }
     113                 :            :     virtual bool ErasePrefix(Span<const std::byte> prefix) = 0;
     114                 :            : 
     115                 :            :     virtual std::unique_ptr<DatabaseCursor> GetNewCursor() = 0;
     116                 :            :     virtual std::unique_ptr<DatabaseCursor> GetNewPrefixCursor(Span<const std::byte> prefix) = 0;
     117                 :            :     virtual bool TxnBegin() = 0;
     118                 :            :     virtual bool TxnCommit() = 0;
     119                 :            :     virtual bool TxnAbort() = 0;
     120                 :            : };
     121                 :            : 
     122                 :            : /** An instance of this class represents one database.
     123                 :            :  **/
     124                 :            : class WalletDatabase
     125                 :            : {
     126                 :            : public:
     127                 :            :     /** Create dummy DB handle */
     128                 :          0 :     WalletDatabase() : nUpdateCounter(0) {}
     129                 :          0 :     virtual ~WalletDatabase() {};
     130                 :            : 
     131                 :            :     /** Open the database if it is not already opened. */
     132                 :            :     virtual void Open() = 0;
     133                 :            : 
     134                 :            :     //! Counts the number of active database users to be sure that the database is not closed while someone is using it
     135                 :          0 :     std::atomic<int> m_refcount{0};
     136                 :            :     /** Indicate the a new database user has began using the database. Increments m_refcount */
     137                 :            :     virtual void AddRef() = 0;
     138                 :            :     /** Indicate that database user has stopped using the database and that it could be flushed or closed. Decrement m_refcount */
     139                 :            :     virtual void RemoveRef() = 0;
     140                 :            : 
     141                 :            :     /** Rewrite the entire database on disk, with the exception of key pszSkip if non-zero
     142                 :            :      */
     143                 :            :     virtual bool Rewrite(const char* pszSkip=nullptr) = 0;
     144                 :            : 
     145                 :            :     /** Back up the entire database to a file.
     146                 :            :      */
     147                 :            :     virtual bool Backup(const std::string& strDest) const = 0;
     148                 :            : 
     149                 :            :     /** Make sure all changes are flushed to database file.
     150                 :            :      */
     151                 :            :     virtual void Flush() = 0;
     152                 :            :     /** Flush to the database file and close the database.
     153                 :            :      *  Also close the environment if no other databases are open in it.
     154                 :            :      */
     155                 :            :     virtual void Close() = 0;
     156                 :            :     /* flush the wallet passively (TRY_LOCK)
     157                 :            :        ideal to be called periodically */
     158                 :            :     virtual bool PeriodicFlush() = 0;
     159                 :            : 
     160                 :            :     virtual void IncrementUpdateCounter() = 0;
     161                 :            : 
     162                 :            :     virtual void ReloadDbEnv() = 0;
     163                 :            : 
     164                 :            :     /** Return path to main database file for logs and error messages. */
     165                 :            :     virtual std::string Filename() = 0;
     166                 :            : 
     167                 :            :     virtual std::string Format() = 0;
     168                 :            : 
     169                 :            :     std::atomic<unsigned int> nUpdateCounter;
     170                 :          0 :     unsigned int nLastSeen{0};
     171                 :          0 :     unsigned int nLastFlushed{0};
     172                 :          0 :     int64_t nLastWalletUpdate{0};
     173                 :            : 
     174                 :            :     /** Make a DatabaseBatch connected to this database */
     175                 :            :     virtual std::unique_ptr<DatabaseBatch> MakeBatch(bool flush_on_close = true) = 0;
     176                 :            : };
     177                 :            : 
     178                 :            : enum class DatabaseFormat {
     179                 :            :     BERKELEY,
     180                 :            :     SQLITE,
     181                 :            : };
     182                 :            : 
     183                 :          0 : struct DatabaseOptions {
     184                 :          0 :     bool require_existing = false;
     185                 :          0 :     bool require_create = false;
     186                 :            :     std::optional<DatabaseFormat> require_format;
     187                 :          0 :     uint64_t create_flags = 0;
     188                 :            :     SecureString create_passphrase;
     189                 :            : 
     190                 :            :     // Specialized options. Not every option is supported by every backend.
     191                 :          0 :     bool verify = true;             //!< Check data integrity on load.
     192                 :          0 :     bool use_unsafe_sync = false;   //!< Disable file sync for faster performance.
     193                 :          0 :     bool use_shared_memory = false; //!< Let other processes access the database.
     194                 :          0 :     int64_t max_log_mb = 100;       //!< Max log size to allow before consolidating.
     195                 :            : };
     196                 :            : 
     197                 :            : enum class DatabaseStatus {
     198                 :            :     SUCCESS,
     199                 :            :     FAILED_BAD_PATH,
     200                 :            :     FAILED_BAD_FORMAT,
     201                 :            :     FAILED_ALREADY_LOADED,
     202                 :            :     FAILED_ALREADY_EXISTS,
     203                 :            :     FAILED_NOT_FOUND,
     204                 :            :     FAILED_CREATE,
     205                 :            :     FAILED_LOAD,
     206                 :            :     FAILED_VERIFY,
     207                 :            :     FAILED_ENCRYPT,
     208                 :            :     FAILED_INVALID_BACKUP_FILE,
     209                 :            : };
     210                 :            : 
     211                 :            : /** Recursively list database paths in directory. */
     212                 :            : std::vector<fs::path> ListDatabases(const fs::path& path);
     213                 :            : 
     214                 :            : void ReadDatabaseArgs(const ArgsManager& args, DatabaseOptions& options);
     215                 :            : std::unique_ptr<WalletDatabase> MakeDatabase(const fs::path& path, const DatabaseOptions& options, DatabaseStatus& status, bilingual_str& error);
     216                 :            : 
     217                 :            : fs::path BDBDataFile(const fs::path& path);
     218                 :            : fs::path SQLiteDataFile(const fs::path& path);
     219                 :            : bool IsBDBFile(const fs::path& path);
     220                 :            : bool IsSQLiteFile(const fs::path& path);
     221                 :            : } // namespace wallet
     222                 :            : 
     223                 :            : #endif // BITCOIN_WALLET_DB_H

Generated by: LCOV version 1.14