handle missing witness_utxo

cryptoadvance · Jan 20, 2021 · 9250a1e · 9250a1e
1 parent b6ef806
commit 9250a1e
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/src/apps/wallets/manager.py b/src/apps/wallets/manager.py
@@ -229,6 +229,13 @@ async def sign_psbt(self, stream, show_screen, encoding=BASE64_STREAM):
         else:
             data = stream.read()
         psbt = PSBT.parse(data)
+        # check if all witness_utxos are there and fill if not
+        for i, inp in enumerate(psbt.inputs):
+            if inp.witness_utxo is None:
+                if inp.non_witness_utxo is None:
+                    raise WalletError("Invalid PSBT - missing previous transaction")
+                inp.witness_utxo = inp.non_witness_utxo.vout[psbt.tx.vin[i].vout]
+
         wallets, meta = self.parse_psbt(psbt=psbt)
         # there is an unknown wallet
         # wallet is a list of tuples: (wallet, amount)