LCOV - code coverage report
Current view: top level - src/wallet/rpc - util.cpp (source / functions) Coverage Total Hit
Test: total_coverage.info Lines: 98.1 % 103 101
Test Date: 2024-08-28 05:13:07 Functions: 100.0 % 13 13
Branches: 61.3 % 155 95

             Branch data     Line data    Source code
       1                 :             : // Copyright (c) 2011-2022 The Bitcoin Core developers
       2                 :             : // Distributed under the MIT software license, see the accompanying
       3                 :             : // file COPYING or http://www.opensource.org/licenses/mit-license.php.
       4                 :             : 
       5                 :             : #include <wallet/rpc/util.h>
       6                 :             : 
       7                 :             : #include <common/url.h>
       8                 :             : #include <rpc/util.h>
       9                 :             : #include <util/any.h>
      10                 :             : #include <util/translation.h>
      11                 :             : #include <wallet/context.h>
      12                 :             : #include <wallet/wallet.h>
      13                 :             : 
      14                 :             : #include <string_view>
      15                 :             : #include <univalue.h>
      16                 :             : 
      17                 :             : #include <boost/date_time/posix_time/posix_time.hpp>
      18                 :             : 
      19                 :             : namespace wallet {
      20                 :             : static const std::string WALLET_ENDPOINT_BASE = "/wallet/";
      21                 :             : const std::string HELP_REQUIRING_PASSPHRASE{"\nRequires wallet passphrase to be set with walletpassphrase call if wallet is encrypted.\n"};
      22                 :             : 
      23                 :        1708 : int64_t ParseISO8601DateTime(const std::string& str)
      24                 :             : {
      25   [ +  +  +  -  :        1708 :     static const boost::posix_time::ptime epoch = boost::posix_time::from_time_t(0);
                   +  - ]
      26                 :        1708 :     static const std::locale loc(std::locale::classic(),
      27   [ +  +  +  -  :        2072 :         new boost::posix_time::time_input_facet("%Y-%m-%dT%H:%M:%SZ"));
          +  -  +  -  +  
          -  +  -  +  -  
                   -  - ]
      28                 :        1708 :     std::istringstream iss(str);
      29         [ +  - ]:        1708 :     iss.imbue(loc);
      30                 :        1708 :     boost::posix_time::ptime ptime(boost::date_time::not_a_date_time);
      31         [ +  - ]:        1708 :     iss >> ptime;
      32   [ +  +  +  + ]:        1708 :     if (ptime.is_not_a_date_time() || epoch > ptime)
      33                 :             :         return 0;
      34                 :         855 :     return (ptime - epoch).total_seconds();
      35                 :        1708 : }
      36                 :             : 
      37                 :        3101 : bool GetAvoidReuseFlag(const CWallet& wallet, const UniValue& param) {
      38                 :        3101 :     bool can_avoid_reuse = wallet.IsWalletFlagSet(WALLET_FLAG_AVOID_REUSE);
      39         [ +  + ]:        3101 :     bool avoid_reuse = param.isNull() ? can_avoid_reuse : param.get_bool();
      40                 :             : 
      41         [ -  + ]:        3101 :     if (avoid_reuse && !can_avoid_reuse) {
      42   [ #  #  #  # ]:           0 :         throw JSONRPCError(RPC_WALLET_ERROR, "wallet does not have the \"avoid reuse\" feature enabled");
      43                 :             :     }
      44                 :             : 
      45                 :        3101 :     return avoid_reuse;
      46                 :             : }
      47                 :             : 
      48                 :             : /** Used by RPC commands that have an include_watchonly parameter.
      49                 :             :  *  We default to true for watchonly wallets if include_watchonly isn't
      50                 :             :  *  explicitly set.
      51                 :             :  */
      52                 :        3697 : bool ParseIncludeWatchonly(const UniValue& include_watchonly, const CWallet& wallet)
      53                 :             : {
      54         [ +  + ]:        3697 :     if (include_watchonly.isNull()) {
      55                 :             :         // if include_watchonly isn't explicitly set, then check if we have a watchonly wallet
      56                 :        2859 :         return wallet.IsWalletFlagSet(WALLET_FLAG_DISABLE_PRIVATE_KEYS);
      57                 :             :     }
      58                 :             : 
      59                 :             :     // otherwise return whatever include_watchonly was set to
      60                 :         838 :     return include_watchonly.get_bool();
      61                 :             : }
      62                 :             : 
      63                 :       35926 : bool GetWalletNameFromJSONRPCRequest(const JSONRPCRequest& request, std::string& wallet_name)
      64                 :             : {
      65         [ +  + ]:       35926 :     if (request.URI.starts_with(WALLET_ENDPOINT_BASE)) {
      66                 :             :         // wallet endpoint was used
      67                 :       22882 :         wallet_name = UrlDecode(std::string_view{request.URI}.substr(WALLET_ENDPOINT_BASE.size()));
      68                 :       22882 :         return true;
      69                 :             :     }
      70                 :             :     return false;
      71                 :             : }
      72                 :             : 
      73                 :       35465 : std::shared_ptr<CWallet> GetWalletForJSONRPCRequest(const JSONRPCRequest& request)
      74                 :             : {
      75                 :       35465 :     CHECK_NONFATAL(request.mode == JSONRPCRequest::EXECUTE);
      76                 :       35465 :     WalletContext& context = EnsureWalletContext(request.context);
      77                 :             : 
      78         [ +  - ]:       35465 :     std::string wallet_name;
      79   [ +  -  +  + ]:       35465 :     if (GetWalletNameFromJSONRPCRequest(request, wallet_name)) {
      80         [ +  - ]:       22590 :         std::shared_ptr<CWallet> pwallet = GetWallet(context, wallet_name);
      81   [ +  +  +  -  :       22609 :         if (!pwallet) throw JSONRPCError(RPC_WALLET_NOT_FOUND, "Requested wallet does not exist or is not loaded");
                   +  - ]
      82                 :             :         return pwallet;
      83                 :          19 :     }
      84                 :             : 
      85                 :       12875 :     size_t count{0};
      86         [ +  - ]:       12875 :     auto wallet = GetDefaultWallet(context, count);
      87   [ +  +  -  + ]:       12875 :     if (wallet) return wallet;
      88                 :             : 
      89         [ +  + ]:          41 :     if (count == 0) {
      90         [ +  - ]:          19 :         throw JSONRPCError(
      91   [ +  -  +  - ]:          38 :             RPC_WALLET_NOT_FOUND, "No wallet is loaded. Load a wallet using loadwallet or create a new one with createwallet. (Note: A default wallet is no longer automatically created)");
      92                 :             :     }
      93         [ +  - ]:          22 :     throw JSONRPCError(RPC_WALLET_NOT_SPECIFIED,
      94   [ +  -  +  - ]:          44 :         "Wallet file not specified (must request wallet RPC through /wallet/<filename> uri-path).");
      95                 :       48280 : }
      96                 :             : 
      97                 :        7365 : void EnsureWalletIsUnlocked(const CWallet& wallet)
      98                 :             : {
      99         [ +  + ]:        7365 :     if (wallet.IsLocked()) {
     100   [ +  -  +  - ]:          52 :         throw JSONRPCError(RPC_WALLET_UNLOCK_NEEDED, "Error: Please enter the wallet passphrase with walletpassphrase first.");
     101                 :             :     }
     102                 :        7339 : }
     103                 :             : 
     104                 :       37116 : WalletContext& EnsureWalletContext(const std::any& context)
     105                 :             : {
     106                 :       37116 :     auto wallet_context = util::AnyPtr<WalletContext>(context);
     107         [ -  + ]:       37116 :     if (!wallet_context) {
     108   [ #  #  #  # ]:           0 :         throw JSONRPCError(RPC_INTERNAL_ERROR, "Wallet context not found");
     109                 :             :     }
     110                 :       37116 :     return *wallet_context;
     111                 :             : }
     112                 :             : 
     113                 :             : // also_create should only be set to true only when the RPC is expected to add things to a blank wallet and make it no longer blank
     114                 :         673 : LegacyScriptPubKeyMan& EnsureLegacyScriptPubKeyMan(CWallet& wallet, bool also_create)
     115                 :             : {
     116                 :         673 :     LegacyScriptPubKeyMan* spk_man = wallet.GetLegacyScriptPubKeyMan();
     117         [ +  + ]:         673 :     if (!spk_man && also_create) {
     118                 :           7 :         spk_man = wallet.GetOrCreateLegacyScriptPubKeyMan();
     119                 :             :     }
     120         [ +  + ]:         673 :     if (!spk_man) {
     121   [ +  -  +  - ]:          14 :         throw JSONRPCError(RPC_WALLET_ERROR, "Only legacy wallets are supported by this command");
     122                 :             :     }
     123                 :         666 :     return *spk_man;
     124                 :             : }
     125                 :             : 
     126                 :         292 : const LegacyScriptPubKeyMan& EnsureConstLegacyScriptPubKeyMan(const CWallet& wallet)
     127                 :             : {
     128                 :         292 :     const LegacyScriptPubKeyMan* spk_man = wallet.GetLegacyScriptPubKeyMan();
     129         [ +  + ]:         292 :     if (!spk_man) {
     130   [ +  -  +  - ]:           4 :         throw JSONRPCError(RPC_WALLET_ERROR, "Only legacy wallets are supported by this command");
     131                 :             :     }
     132                 :         290 :     return *spk_man;
     133                 :             : }
     134                 :             : 
     135                 :       22055 : std::string LabelFromValue(const UniValue& value)
     136                 :             : {
     137   [ +  +  +  - ]:       22419 :     static const std::string empty_string;
     138         [ +  + ]:       22055 :     if (value.isNull()) return empty_string;
     139                 :             : 
     140                 :        1527 :     const std::string& label{value.get_str()};
     141         [ +  + ]:        1527 :     if (label == "*")
     142   [ +  -  +  - ]:          36 :         throw JSONRPCError(RPC_WALLET_INVALID_LABEL_NAME, "Invalid label name");
     143                 :        1509 :     return label;
     144                 :             : }
     145                 :             : 
     146                 :       45322 : void PushParentDescriptors(const CWallet& wallet, const CScript& script_pubkey, UniValue& entry)
     147                 :             : {
     148                 :       45322 :     UniValue parent_descs(UniValue::VARR);
     149   [ +  -  +  + ]:       80540 :     for (const auto& desc: wallet.GetWalletDescriptors(script_pubkey)) {
     150   [ +  -  +  -  :       35218 :         parent_descs.push_back(desc.descriptor->ToString());
                   +  - ]
     151                 :       45322 :     }
     152   [ +  -  +  - ]:       90644 :     entry.pushKV("parent_descs", std::move(parent_descs));
     153                 :       45322 : }
     154                 :             : 
     155                 :         320 : void HandleWalletError(const std::shared_ptr<CWallet> wallet, DatabaseStatus& status, bilingual_str& error)
     156                 :             : {
     157         [ +  + ]:         320 :     if (!wallet) {
     158                 :             :         // Map bad format to not found, since bad format is returned when the
     159                 :             :         // wallet directory exists, but doesn't contain a data file.
     160                 :          41 :         RPCErrorCode code = RPC_WALLET_ERROR;
     161   [ +  +  +  +  :          41 :         switch (status) {
                      + ]
     162                 :          20 :             case DatabaseStatus::FAILED_NOT_FOUND:
     163                 :          20 :             case DatabaseStatus::FAILED_BAD_FORMAT:
     164                 :          20 :                 code = RPC_WALLET_NOT_FOUND;
     165                 :          20 :                 break;
     166                 :           1 :             case DatabaseStatus::FAILED_ALREADY_LOADED:
     167                 :           1 :                 code = RPC_WALLET_ALREADY_LOADED;
     168                 :           1 :                 break;
     169                 :           2 :             case DatabaseStatus::FAILED_ALREADY_EXISTS:
     170                 :           2 :                 code = RPC_WALLET_ALREADY_EXISTS;
     171                 :           2 :                 break;
     172                 :           2 :             case DatabaseStatus::FAILED_INVALID_BACKUP_FILE:
     173                 :           2 :                 code = RPC_INVALID_PARAMETER;
     174                 :           2 :                 break;
     175                 :             :             default: // RPC_WALLET_ERROR is returned for all other cases.
     176                 :             :                 break;
     177                 :             :         }
     178         [ +  - ]:          41 :         throw JSONRPCError(code, error.original);
     179                 :             :     }
     180                 :         279 : }
     181                 :             : 
     182                 :        3162 : void AppendLastProcessedBlock(UniValue& entry, const CWallet& wallet)
     183                 :             : {
     184                 :        3162 :     AssertLockHeld(wallet.cs_wallet);
     185                 :        3162 :     UniValue lastprocessedblock{UniValue::VOBJ};
     186   [ +  -  +  -  :        6324 :     lastprocessedblock.pushKV("hash", wallet.GetLastBlockHash().GetHex());
             +  -  +  - ]
     187   [ +  -  +  -  :        6324 :     lastprocessedblock.pushKV("height", wallet.GetLastBlockHeight());
                   +  - ]
     188   [ +  -  +  - ]:        6324 :     entry.pushKV("lastprocessedblock", std::move(lastprocessedblock));
     189                 :        3162 : }
     190                 :             : 
     191                 :             : } // namespace wallet
        

Generated by: LCOV version 2.0-1