Branch data     Line data    Source code

       1                 :             : // Copyright (c) 2020-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 <chainparams.h>
       6                 :             : #include <common/args.h>
       7                 :             : #include <common/system.h>
       8                 :             : #include <external_signer.h>
       9                 :             : #include <node/types.h>
      10                 :             : #include <wallet/external_signer_scriptpubkeyman.h>
      11                 :             : 
      12                 :             : #include <iostream>
      13                 :             : #include <key_io.h>
      14                 :             : #include <memory>
      15                 :             : #include <stdexcept>
      16                 :             : #include <string>
      17                 :             : #include <univalue.h>
      18                 :             : #include <utility>
      19                 :             : #include <vector>
      20                 :             : 
      21                 :             : using common::PSBTError;
      22                 :             : 
      23                 :             : namespace wallet {
      24                 :           8 : bool ExternalSignerScriptPubKeyMan::SetupDescriptor(WalletBatch& batch, std::unique_ptr<Descriptor> desc)
      25                 :             : {
      26                 :           8 :     LOCK(cs_desc_man);
      27   [ +  -  -  + ]:           8 :     assert(m_storage.IsWalletFlagSet(WALLET_FLAG_DESCRIPTORS));
      28   [ +  -  -  + ]:           8 :     assert(m_storage.IsWalletFlagSet(WALLET_FLAG_EXTERNAL_SIGNER));
      29                 :             : 
      30         [ +  - ]:           8 :     int64_t creation_time = GetTime();
      31                 :             : 
      32                 :             :     // Make the descriptor
      33   [ +  -  +  - ]:           8 :     WalletDescriptor w_desc(std::move(desc), creation_time, 0, 0, 0);
      34         [ +  - ]:           8 :     m_wallet_descriptor = w_desc;
      35                 :             : 
      36                 :             :     // Store the descriptor
      37   [ +  -  +  -  :           8 :     if (!batch.WriteDescriptor(GetID(), m_wallet_descriptor)) {
                   -  + ]
      38   [ #  #  #  # ]:           0 :         throw std::runtime_error(std::string(__func__) + ": writing descriptor failed");
      39                 :             :     }
      40                 :             : 
      41                 :             :     // TopUp
      42         [ +  - ]:           8 :     TopUpWithDB(batch);
      43                 :             : 
      44         [ +  - ]:           8 :     m_storage.UnsetBlankWalletFlag(batch);
      45                 :          16 :     return true;
      46         [ +  - ]:          16 : }
      47                 :             : 
      48                 :          11 : ExternalSigner ExternalSignerScriptPubKeyMan::GetExternalSigner() {
      49   [ +  -  +  - ]:          22 :     const std::string command = gArgs.GetArg("-signer", "");
      50   [ -  +  -  -  :          11 :     if (command == "") throw std::runtime_error(std::string(__func__) + ": restart bitcoind with -signer=<cmd>");
                   -  - ]
      51                 :          11 :     std::vector<ExternalSigner> signers;
      52   [ +  -  +  -  :          12 :     ExternalSigner::Enumerate(command, signers, Params().GetChainTypeString());
                   +  + ]
      53   [ -  +  -  -  :          10 :     if (signers.empty()) throw std::runtime_error(std::string(__func__) + ": No external signers found");
                   -  - ]
      54                 :             :     // TODO: add fingerprint argument instead of failing in case of multiple signers.
      55   [ +  +  +  -  :          12 :     if (signers.size() > 1) throw std::runtime_error(std::string(__func__) + ": More than one external signer found. Please connect only one at a time.");
                   +  - ]
      56         [ +  - ]:           9 :     return signers[0];
      57                 :          11 : }
      58                 :             : 
      59                 :           4 : util::Result<void> ExternalSignerScriptPubKeyMan::DisplayAddress(const CTxDestination& dest, const ExternalSigner &signer) const
      60                 :             : {
      61                 :             :     // TODO: avoid the need to infer a descriptor from inside a descriptor wallet
      62                 :           4 :     const CScript& scriptPubKey = GetScriptForDestination(dest);
      63         [ +  - ]:           4 :     auto provider = GetSolvingProvider(scriptPubKey);
      64         [ +  - ]:           4 :     auto descriptor = InferDescriptor(scriptPubKey, *provider);
      65                 :             : 
      66   [ +  -  +  - ]:           4 :     const UniValue& result = signer.DisplayAddress(descriptor->ToString());
      67                 :             : 
      68         [ +  - ]:           4 :     const UniValue& error = result.find_value("error");
      69   [ -  +  -  -  :           4 :     if (error.isStr()) return util::Error{strprintf(_("Signer returned error: %s"), error.getValStr())};
                   -  - ]
      70                 :             : 
      71         [ +  - ]:           4 :     const UniValue& ret_address = result.find_value("address");
      72   [ -  +  -  - ]:           4 :     if (!ret_address.isStr()) return util::Error{_("Signer did not echo address")};
      73                 :             : 
      74   [ +  -  +  + ]:           4 :     if (ret_address.getValStr() != EncodeDestination(dest)) {
      75   [ +  -  +  - ]:           4 :         return util::Error{strprintf(_("Signer echoed unexpected address %s"), ret_address.getValStr())};
      76                 :             :     }
      77                 :             : 
      78                 :           3 :     return util::Result<void>();
      79                 :           4 : }
      80                 :             : 
      81                 :             : // If sign is true, transaction must previously have been filled
      82                 :          48 : std::optional<PSBTError> ExternalSignerScriptPubKeyMan::FillPSBT(PartiallySignedTransaction& psbt, const PrecomputedTransactionData& txdata, int sighash_type, bool sign, bool bip32derivs, int* n_signed, bool finalize) const
      83                 :             : {
      84         [ +  + ]:          48 :     if (!sign) {
      85                 :          24 :         return DescriptorScriptPubKeyMan::FillPSBT(psbt, txdata, sighash_type, false, bip32derivs, n_signed, finalize);
      86                 :             :     }
      87                 :             : 
      88                 :             :     // Already complete if every input is now signed
      89                 :          24 :     bool complete = true;
      90         [ +  + ]:          48 :     for (const auto& input : psbt.inputs) {
      91                 :             :         // TODO: for multisig wallets, we should only care if all _our_ inputs are signed
      92                 :          24 :         complete &= PSBTInputSigned(input);
      93                 :             :     }
      94         [ +  + ]:          24 :     if (complete) return {};
      95                 :             : 
      96         [ +  - ]:           3 :     std::string strFailReason;
      97   [ +  -  +  -  :           3 :     if(!GetExternalSigner().SignTransaction(psbt, strFailReason)) {
                   -  + ]
      98         [ #  # ]:           0 :         tfm::format(std::cerr, "Failed to sign: %s\n", strFailReason);
      99                 :           0 :         return PSBTError::EXTERNAL_SIGNER_FAILED;
     100                 :             :     }
     101   [ +  -  +  - ]:           3 :     if (finalize) FinalizePSBT(psbt); // This won't work in a multisig setup
     102                 :           3 :     return {};
     103                 :           3 : }
     104                 :             : } // namespace wallet