Branch data     Line data    Source code

       1                 :             : // Copyright (c) 2017-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/walletutil.h>
       6                 :             : 
       7                 :             : #include <chainparams.h>
       8                 :             : #include <common/args.h>
       9                 :             : #include <key_io.h>
      10                 :             : #include <logging.h>
      11                 :             : 
      12                 :             : namespace wallet {
      13                 :        2065 : fs::path GetWalletDir()
      14                 :             : {
      15                 :        2065 :     fs::path path;
      16                 :             : 
      17   [ +  -  +  -  :        2065 :     if (gArgs.IsArgSet("-walletdir")) {
                   +  + ]
      18   [ +  -  +  - ]:          57 :         path = gArgs.GetPathArg("-walletdir");
      19   [ +  -  -  + ]:          19 :         if (!fs::is_directory(path)) {
      20                 :             :             // If the path specified doesn't exist, we return the deliberately
      21                 :             :             // invalid empty string.
      22         [ #  # ]:           0 :             path = "";
      23                 :             :         }
      24                 :             :     } else {
      25         [ +  - ]:        2046 :         path = gArgs.GetDataDirNet();
      26                 :             :         // If a wallets directory exists, use that, otherwise default to GetDataDir
      27   [ +  -  +  -  :        8184 :         if (fs::is_directory(path / "wallets")) {
             +  -  +  + ]
      28         [ +  - ]:        1460 :             path /= "wallets";
      29                 :             :         }
      30                 :             :     }
      31                 :             : 
      32                 :        2065 :     return path;
      33                 :           0 : }
      34                 :             : 
      35                 :        3086 : WalletDescriptor GenerateWalletDescriptor(const CExtPubKey& master_key, const OutputType& addr_type, bool internal)
      36                 :             : {
      37                 :        3086 :     int64_t creation_time = GetTime();
      38                 :             : 
      39                 :        3086 :     std::string xpub = EncodeExtPubKey(master_key);
      40                 :             : 
      41                 :             :     // Build descriptor string
      42         [ +  - ]:        3086 :     std::string desc_prefix;
      43         [ +  - ]:        3086 :     std::string desc_suffix = "/*)";
      44   [ +  +  +  +  :        3086 :     switch (addr_type) {
                   -  - ]
      45                 :         766 :     case OutputType::LEGACY: {
      46         [ +  - ]:        1532 :         desc_prefix = "pkh(" + xpub + "/44h";
      47                 :         766 :         break;
      48                 :             :     }
      49                 :         766 :     case OutputType::P2SH_SEGWIT: {
      50         [ +  - ]:        1532 :         desc_prefix = "sh(wpkh(" + xpub + "/49h";
      51         [ +  - ]:         766 :         desc_suffix += ")";
      52                 :             :         break;
      53                 :             :     }
      54                 :         770 :     case OutputType::BECH32: {
      55         [ +  - ]:        1540 :         desc_prefix = "wpkh(" + xpub + "/84h";
      56                 :         770 :         break;
      57                 :             :     }
      58                 :         784 :     case OutputType::BECH32M: {
      59         [ +  - ]:        1568 :         desc_prefix = "tr(" + xpub + "/86h";
      60                 :         784 :         break;
      61                 :             :     }
      62                 :           0 :     case OutputType::UNKNOWN: {
      63                 :             :         // We should never have a DescriptorScriptPubKeyMan for an UNKNOWN OutputType,
      64                 :             :         // so if we get to this point something is wrong
      65                 :           0 :         assert(false);
      66                 :             :     }
      67                 :             :     } // no default case, so the compiler can warn about missing cases
      68         [ -  + ]:        3086 :     assert(!desc_prefix.empty());
      69                 :             : 
      70                 :             :     // Mainnet derives at 0', testnet and regtest derive at 1'
      71   [ +  -  +  + ]:        3086 :     if (Params().IsTestChain()) {
      72         [ +  - ]:        2878 :         desc_prefix += "/1h";
      73                 :             :     } else {
      74         [ +  - ]:         208 :         desc_prefix += "/0h";
      75                 :             :     }
      76                 :             : 
      77   [ +  +  +  - ]:        7715 :     std::string internal_path = internal ? "/1" : "/0";
      78   [ +  -  +  -  :        9258 :     std::string desc_str = desc_prefix + "/0h" + internal_path + desc_suffix;
                   +  - ]
      79                 :             : 
      80                 :             :     // Make the descriptor
      81                 :        6172 :     FlatSigningProvider keys;
      82         [ +  - ]:        6172 :     std::string error;
      83         [ +  - ]:        6172 :     std::vector<std::unique_ptr<Descriptor>> desc = Parse(desc_str, keys, error, false);
      84   [ +  -  +  -  :        3086 :     WalletDescriptor w_desc(std::move(desc.at(0)), creation_time, 0, 0, 0);
                   +  - ]
      85                 :        3086 :     return w_desc;
      86                 :        3086 : }
      87                 :             : 
      88                 :             : } // namespace wallet