[HW] Sign P2CS inputs on Ledger (and trigger warning on Trezor)

Author: random-zebra

src/ledgerClient.py

@@ -148,7 +148,8 @@ def append_inputs_to_TX(self, utxo, bip32_path):
             'pubkey': curr_pubkey,
             'bip32_path': bip32_path,
             'outputIndex': utxo['vout'],
-            'txid': utxo['txid']
+            'txid': utxo['txid'],
+            'p2cs': (utxo['staker'] != "")
         })
 
 
@@ -370,7 +371,10 @@ def signTxSign(self, ctrl):
                 inputTx.prevOut = bytearray.fromhex(new_input['txid'])[::-1] + int.to_bytes(new_input['outputIndex'], 4,
                                                                                             byteorder='little')
 
-                inputTx.script = bytearray([len(sig)]) + sig + bytearray([0x21]) + new_input['pubkey']
+                inputTx.script = bytearray([len(sig)]) + sig
+                if new_input['p2cs']:
+                    inputTx.script += bytearray([0x00])
+                inputTx.script += bytearray([0x21]) + new_input['pubkey']
 
                 inputTx.sequence = bytearray([0xFF, 0xFF, 0xFF, 0xFF])
 

src/trezorClient.py

@@ -84,6 +84,9 @@ def __init__(self, model, main_wnd, *args, **kwargs):
 
     @process_trezor_exceptions
     def append_inputs_to_TX(self, utxo, bip32_path, inputs):
+        if utxo['staker'] != "":
+            printException(getCallerName(), getFunctionName(), "Unable to sing P2CS on Trezor", "")
+            return
         # Update amount
         self.amount += int(utxo['satoshis'])
         # Add input