Host-customized fork of https://github.com/tecnovert/basicswap/
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
736 lines
29 KiB
736 lines
29 KiB
#!/usr/bin/env python |
|
# -*- coding: utf-8 -*- |
|
|
|
# Copyright (c) 2023 tecnovert |
|
# Distributed under the MIT software license, see the accompanying |
|
# file LICENSE or http://www.opensource.org/licenses/mit-license.php. |
|
|
|
import random |
|
import hashlib |
|
|
|
from io import BytesIO |
|
from coincurve.keys import ( |
|
PublicKey, |
|
PrivateKey, |
|
) |
|
from .btc import BTCInterface, find_vout_for_address_from_txobj, findOutput |
|
from basicswap.rpc import make_rpc_func |
|
from basicswap.chainparams import Coins |
|
from basicswap.interface.contrib.nav_test_framework.mininode import ( |
|
CTxIn, |
|
CTxOut, |
|
CBlock, |
|
COutPoint, |
|
CTransaction, |
|
CTxInWitness, |
|
FromHex, |
|
uint256_from_str, |
|
) |
|
from basicswap.util.crypto import hash160 |
|
from basicswap.util.address import ( |
|
decodeWif, |
|
pubkeyToAddress, |
|
encodeAddress, |
|
) |
|
from basicswap.util import ( |
|
i2b, i2h, |
|
ensure, |
|
) |
|
from basicswap.basicswap_util import ( |
|
getVoutByScriptPubKey, |
|
) |
|
|
|
from basicswap.interface.contrib.nav_test_framework.script import ( |
|
CScript, |
|
OP_0, |
|
OP_EQUAL, |
|
OP_DUP, OP_HASH160, OP_EQUALVERIFY, OP_CHECKSIG, |
|
SIGHASH_ALL, |
|
SegwitVersion1SignatureHash, |
|
) |
|
from mnemonic import Mnemonic |
|
|
|
|
|
class NAVInterface(BTCInterface): |
|
@staticmethod |
|
def coin_type(): |
|
return Coins.NAV |
|
|
|
@staticmethod |
|
def txVersion() -> int: |
|
return 3 |
|
|
|
@staticmethod |
|
def txoType(): |
|
return CTxOut |
|
|
|
def __init__(self, coin_settings, network, swap_client=None): |
|
super(NAVInterface, self).__init__(coin_settings, network, swap_client) |
|
# No multiwallet support |
|
self.rpc_wallet = make_rpc_func(self._rpcport, self._rpcauth, host=self._rpc_host) |
|
|
|
def checkWallets(self) -> int: |
|
return 1 |
|
|
|
def use_p2shp2wsh(self) -> bool: |
|
# p2sh-p2wsh |
|
return True |
|
|
|
def seedToMnemonic(self, key: bytes) -> None: |
|
return Mnemonic('english').to_mnemonic(key) |
|
|
|
def initialiseWallet(self, key): |
|
# Load with -importmnemonic= parameter |
|
pass |
|
|
|
def getWalletSeedID(self): |
|
return self.rpc('getwalletinfo')['hdmasterkeyid'] |
|
|
|
def withdrawCoin(self, value, addr_to: str, subfee: bool): |
|
strdzeel = '' |
|
params = [addr_to, value, '', '', strdzeel, subfee] |
|
return self.rpc('sendtoaddress', params) |
|
|
|
def getSpendableBalance(self) -> int: |
|
return self.make_int(self.rpc('getwalletinfo')['balance']) |
|
|
|
def signTxWithWallet(self, tx: bytes) -> bytes: |
|
rv = self.rpc('signrawtransaction', [tx.hex()]) |
|
|
|
return bytes.fromhex(rv['hex']) |
|
|
|
def checkExpectedSeed(self, key_hash: str): |
|
try: |
|
rv = self.rpc('dumpmnemonic') |
|
entropy = Mnemonic('english').to_entropy(rv.split(' ')) |
|
|
|
entropy_hash = self.getAddressHashFromKey(entropy)[::-1].hex() |
|
self._have_checked_seed = True |
|
return entropy_hash == key_hash |
|
except Exception as e: |
|
self._log.warning('checkExpectedSeed failed: {}'.format(str(e))) |
|
return False |
|
|
|
def getScriptForP2PKH(self, pkh: bytes) -> bytearray: |
|
# Return P2PKH |
|
return CScript([OP_DUP, OP_HASH160, pkh, OP_EQUALVERIFY, OP_CHECKSIG]) |
|
|
|
def getScriptForPubkeyHash(self, pkh: bytes) -> bytearray: |
|
# Return P2SH-p2wpkh |
|
|
|
script = CScript([OP_0, pkh]) |
|
script_hash = hash160(script) |
|
assert len(script_hash) == 20 |
|
|
|
return CScript([OP_HASH160, script_hash, OP_EQUAL]) |
|
|
|
def getInputScriptForPubkeyHash(self, pkh: bytes) -> bytearray: |
|
script = CScript([OP_0, pkh]) |
|
return bytes((len(script),)) + script |
|
|
|
def encodeSegwitAddress(self, pkh: bytes) -> str: |
|
# P2SH-p2wpkh |
|
script = CScript([OP_0, pkh]) |
|
script_hash = hash160(script) |
|
assert len(script_hash) == 20 |
|
return encodeAddress(bytes((self.chainparams_network()['script_address'],)) + script_hash) |
|
|
|
def encodeSegwitAddressScript(self, script: bytes) -> str: |
|
if len(script) == 23 and script[0] == OP_HASH160 and script[1] == 20 and script[22] == OP_EQUAL: |
|
script_hash = script[2:22] |
|
return encodeAddress(bytes((self.chainparams_network()['script_address'],)) + script_hash) |
|
raise ValueError('Unknown Script') |
|
|
|
def loadTx(self, tx_bytes: bytes) -> CTransaction: |
|
# Load tx from bytes to internal representation |
|
tx = CTransaction() |
|
tx.deserialize(BytesIO(tx_bytes)) |
|
return tx |
|
|
|
def signTx(self, key_bytes: bytes, tx_bytes: bytes, input_n: int, prevout_script, prevout_value: int): |
|
tx = self.loadTx(tx_bytes) |
|
sig_hash = SegwitVersion1SignatureHash(prevout_script, tx, input_n, SIGHASH_ALL, prevout_value) |
|
eck = PrivateKey(key_bytes) |
|
return eck.sign(sig_hash, hasher=None) + bytes((SIGHASH_ALL,)) |
|
|
|
def setTxSignature(self, tx_bytes: bytes, stack) -> bytes: |
|
tx = self.loadTx(tx_bytes) |
|
tx.wit.vtxinwit.clear() |
|
tx.wit.vtxinwit.append(CTxInWitness()) |
|
tx.wit.vtxinwit[0].scriptWitness.stack = stack |
|
return tx.serialize_with_witness() |
|
|
|
def getProofOfFunds(self, amount_for, extra_commit_bytes): |
|
# TODO: Lock unspent and use same output/s to fund bid |
|
|
|
unspents_by_addr = dict() |
|
unspents = self.rpc('listunspent') |
|
for u in unspents: |
|
if u['spendable'] is not True: |
|
continue |
|
if u['address'] not in unspents_by_addr: |
|
unspents_by_addr[u['address']] = {'total': 0, 'utxos': []} |
|
utxo_amount: int = self.make_int(u['amount'], r=1) |
|
unspents_by_addr[u['address']]['total'] += utxo_amount |
|
unspents_by_addr[u['address']]['utxos'].append((utxo_amount, u['txid'], u['vout'])) |
|
|
|
max_utxos: int = 4 |
|
|
|
viable_addrs = [] |
|
for addr, data in unspents_by_addr.items(): |
|
if data['total'] >= amount_for: |
|
# Sort from largest to smallest amount |
|
sorted_utxos = sorted(data['utxos'], key=lambda x: x[0]) |
|
|
|
# Max outputs required to reach amount_for |
|
utxos_req: int = 0 |
|
sum_value: int = 0 |
|
for utxo in sorted_utxos: |
|
sum_value += utxo[0] |
|
utxos_req += 1 |
|
if sum_value >= amount_for: |
|
break |
|
|
|
if utxos_req <= max_utxos: |
|
viable_addrs.append(addr) |
|
continue |
|
|
|
ensure(len(viable_addrs) > 0, 'Could not find address with enough funds for proof') |
|
|
|
sign_for_addr: str = random.choice(viable_addrs) |
|
self._log.debug('sign_for_addr %s', sign_for_addr) |
|
|
|
prove_utxos = [] |
|
sorted_utxos = sorted(unspents_by_addr[sign_for_addr]['utxos'], key=lambda x: x[0]) |
|
|
|
hasher = hashlib.sha256() |
|
|
|
sum_value: int = 0 |
|
for utxo in sorted_utxos: |
|
sum_value += utxo[0] |
|
outpoint = (bytes.fromhex(utxo[1]), utxo[2]) |
|
prove_utxos.append(outpoint) |
|
hasher.update(outpoint[0]) |
|
hasher.update(outpoint[1].to_bytes(2, 'big')) |
|
if sum_value >= amount_for: |
|
break |
|
utxos_hash = hasher.digest() |
|
|
|
if self.using_segwit(): # TODO: Use isSegwitAddress when scantxoutset can use combo |
|
# 'Address does not refer to key' for non p2pkh |
|
addr_info = self.rpc('validateaddress', [addr, ]) |
|
if 'isscript' in addr_info and addr_info['isscript'] and 'hex' in addr_info: |
|
pkh = bytes.fromhex(addr_info['hex'])[2:] |
|
sign_for_addr = self.pkh_to_address(pkh) |
|
self._log.debug('sign_for_addr converted %s', sign_for_addr) |
|
|
|
signature = self.rpc('signmessage', [sign_for_addr, sign_for_addr + '_swap_proof_' + utxos_hash.hex() + extra_commit_bytes.hex()]) |
|
|
|
return (sign_for_addr, signature, prove_utxos) |
|
|
|
def verifyProofOfFunds(self, address, signature, utxos, extra_commit_bytes): |
|
hasher = hashlib.sha256() |
|
sum_value: int = 0 |
|
for outpoint in utxos: |
|
hasher.update(outpoint[0]) |
|
hasher.update(outpoint[1].to_bytes(2, 'big')) |
|
utxos_hash = hasher.digest() |
|
|
|
passed = self.verifyMessage(address, address + '_swap_proof_' + utxos_hash.hex() + extra_commit_bytes.hex(), signature) |
|
ensure(passed is True, 'Proof of funds signature invalid') |
|
|
|
if self.using_segwit(): |
|
address = self.encodeSegwitAddress(self.decodeAddress(address)[1:]) |
|
|
|
sum_value: int = 0 |
|
for outpoint in utxos: |
|
txout = self.rpc('gettxout', [outpoint[0].hex(), outpoint[1]]) |
|
sum_value += self.make_int(txout['value']) |
|
|
|
return sum_value |
|
|
|
def createRawFundedTransaction(self, addr_to: str, amount: int, sub_fee: bool = False, lock_unspents: bool = True) -> str: |
|
txn = self.rpc('createrawtransaction', [[], {addr_to: self.format_amount(amount)}]) |
|
fee_rate, fee_src = self.get_fee_rate(self._conf_target) |
|
self._log.debug(f'Fee rate: {fee_rate}, source: {fee_src}, block target: {self._conf_target}') |
|
if sub_fee: |
|
raise ValueError('Navcoin fundrawtransaction is missing the subtractFeeFromOutputs parameter') |
|
# options['subtractFeeFromOutputs'] = [0,] |
|
|
|
fee_rate = self.make_int(fee_rate, r=1) |
|
return self.fundTx(txn, fee_rate, lock_unspents).hex() |
|
|
|
def isAddressMine(self, address: str, or_watch_only: bool = False) -> bool: |
|
addr_info = self.rpc('validateaddress', [address]) |
|
if not or_watch_only: |
|
return addr_info['ismine'] |
|
return addr_info['ismine'] or addr_info['iswatchonly'] |
|
|
|
def createRawSignedTransaction(self, addr_to, amount) -> str: |
|
txn_funded = self.createRawFundedTransaction(addr_to, amount) |
|
return self.rpc('signrawtransaction', [txn_funded])['hex'] |
|
|
|
def getBlockchainInfo(self): |
|
rv = self.rpc('getblockchaininfo') |
|
synced = round(rv['verificationprogress'], 3) |
|
if synced >= 0.997: |
|
rv['verificationprogress'] = 1.0 |
|
return rv |
|
|
|
def encodeScriptDest(self, script_dest: bytes) -> str: |
|
script_hash = script_dest[2:-1] # Extract hash from script |
|
return self.sh_to_address(script_hash) |
|
|
|
def encode_p2wsh(self, script: bytes) -> str: |
|
return pubkeyToAddress(self.chainparams_network()['script_address'], script) |
|
|
|
def find_prevout_info(self, txn_hex: str, txn_script: bytes): |
|
txjs = self.rpc('decoderawtransaction', [txn_hex]) |
|
n = getVoutByScriptPubKey(txjs, self.getScriptDest(txn_script).hex()) |
|
|
|
return { |
|
'txid': txjs['txid'], |
|
'vout': n, |
|
'scriptPubKey': txjs['vout'][n]['scriptPubKey']['hex'], |
|
'redeemScript': txn_script.hex(), |
|
'amount': txjs['vout'][n]['value'] |
|
} |
|
|
|
def getNewAddress(self, use_segwit: bool, label: str = 'swap_receive') -> str: |
|
address: str = self.rpc('getnewaddress', [label,]) |
|
if use_segwit: |
|
return self.rpc('addwitnessaddress', [address,]) |
|
return address |
|
|
|
def createRedeemTxn(self, prevout, output_addr: str, output_value: int, txn_script: bytes) -> str: |
|
tx = CTransaction() |
|
tx.nVersion = self.txVersion() |
|
prev_txid = uint256_from_str(bytes.fromhex(prevout['txid'])[::-1]) |
|
|
|
tx.vin.append(CTxIn(COutPoint(prev_txid, prevout['vout']), |
|
scriptSig=self.getScriptScriptSig(txn_script))) |
|
pkh = self.decodeAddress(output_addr) |
|
script = self.getScriptForPubkeyHash(pkh) |
|
tx.vout.append(self.txoType()(output_value, script)) |
|
tx.rehash() |
|
return tx.serialize().hex() |
|
|
|
def createRefundTxn(self, prevout, output_addr: str, output_value: int, locktime: int, sequence: int, txn_script: bytes) -> str: |
|
tx = CTransaction() |
|
tx.nVersion = self.txVersion() |
|
tx.nLockTime = locktime |
|
prev_txid = uint256_from_str(bytes.fromhex(prevout['txid'])[::-1]) |
|
tx.vin.append(CTxIn(COutPoint(prev_txid, prevout['vout']), |
|
nSequence=sequence, |
|
scriptSig=self.getScriptScriptSig(txn_script))) |
|
pkh = self.decodeAddress(output_addr) |
|
script = self.getScriptForPubkeyHash(pkh) |
|
tx.vout.append(self.txoType()(output_value, script)) |
|
tx.rehash() |
|
return tx.serialize().hex() |
|
|
|
def getTxSignature(self, tx_hex: str, prevout_data, key_wif: str) -> str: |
|
key = decodeWif(key_wif) |
|
redeem_script = bytes.fromhex(prevout_data['redeemScript']) |
|
sig = self.signTx(key, bytes.fromhex(tx_hex), 0, redeem_script, self.make_int(prevout_data['amount'])) |
|
|
|
return sig.hex() |
|
|
|
def verifyTxSig(self, tx_bytes: bytes, sig: bytes, K: bytes, input_n: int, prevout_script: bytes, prevout_value: int) -> bool: |
|
tx = self.loadTx(tx_bytes) |
|
sig_hash = SegwitVersion1SignatureHash(prevout_script, tx, input_n, SIGHASH_ALL, prevout_value) |
|
|
|
pubkey = PublicKey(K) |
|
return pubkey.verify(sig[: -1], sig_hash, hasher=None) # Pop the hashtype byte |
|
|
|
def verifyRawTransaction(self, tx_hex: str, prevouts): |
|
# Only checks signature |
|
# verifyrawtransaction |
|
self._log.warning('NAV verifyRawTransaction only checks signature') |
|
inputs_valid: bool = False |
|
validscripts: int = 0 |
|
|
|
tx_bytes = bytes.fromhex(tx_hex) |
|
tx = self.loadTx(bytes.fromhex(tx_hex)) |
|
|
|
signature = tx.wit.vtxinwit[0].scriptWitness.stack[0] |
|
pubkey = tx.wit.vtxinwit[0].scriptWitness.stack[1] |
|
|
|
input_n: int = 0 |
|
prevout_data = prevouts[input_n] |
|
redeem_script = bytes.fromhex(prevout_data['redeemScript']) |
|
prevout_value = self.make_int(prevout_data['amount']) |
|
|
|
if self.verifyTxSig(tx_bytes, signature, pubkey, input_n, redeem_script, prevout_value): |
|
validscripts += 1 |
|
|
|
# TODO: validate inputs |
|
inputs_valid = True |
|
|
|
return { |
|
'inputs_valid': inputs_valid, |
|
'validscripts': validscripts, |
|
} |
|
|
|
def getHTLCSpendTxVSize(self, redeem: bool = True) -> int: |
|
tx_vsize = 5 # Add a few bytes, sequence in script takes variable amount of bytes |
|
|
|
tx_vsize += 184 if redeem else 187 |
|
return tx_vsize |
|
|
|
def getTxid(self, tx) -> bytes: |
|
if isinstance(tx, str): |
|
tx = bytes.fromhex(tx) |
|
if isinstance(tx, bytes): |
|
tx = self.loadTx(tx) |
|
tx.rehash() |
|
return i2b(tx.sha256) |
|
|
|
def rescanBlockchainForAddress(self, height_start: int, addr_find: str): |
|
# Very ugly workaround for missing `rescanblockchain` rpc command |
|
|
|
chain_blocks: int = self.getChainHeight() |
|
|
|
current_height: int = chain_blocks |
|
block_hash = self.rpc('getblockhash', [current_height]) |
|
|
|
script_hash: bytes = self.decodeAddress(addr_find) |
|
find_scriptPubKey = self.getDestForScriptHash(script_hash) |
|
|
|
while current_height > height_start: |
|
block_hash = self.rpc('getblockhash', [current_height]) |
|
|
|
block = self.rpc('getblock', [block_hash, False]) |
|
decoded_block = CBlock() |
|
decoded_block = FromHex(decoded_block, block) |
|
for tx in decoded_block.vtx: |
|
for txo in tx.vout: |
|
if txo.scriptPubKey == find_scriptPubKey: |
|
tx.rehash() |
|
txid = i2b(tx.sha256) |
|
self._log.info('Found output to addr: {} in tx {} in block {}'.format(addr_find, txid.hex(), block_hash)) |
|
self._log.info('rescanblockchain hack invalidateblock {}'.format(block_hash)) |
|
self.rpc('invalidateblock', [block_hash]) |
|
self.rpc('reconsiderblock', [block_hash]) |
|
return |
|
current_height -= 1 |
|
|
|
def getLockTxHeight(self, txid, dest_address, bid_amount, rescan_from, find_index: bool = False): |
|
# Add watchonly address and rescan if required |
|
|
|
if not self.isAddressMine(dest_address, or_watch_only=True): |
|
self.importWatchOnlyAddress(dest_address, 'bid') |
|
self._log.info('Imported watch-only addr: {}'.format(dest_address)) |
|
self._log.info('Rescanning {} chain from height: {}'.format(self.coin_name(), rescan_from)) |
|
self.rescanBlockchainForAddress(rescan_from, dest_address) |
|
|
|
return_txid = True if txid is None else False |
|
if txid is None: |
|
txns = self.rpc('listunspent', [0, 9999999, [dest_address, ]]) |
|
|
|
for tx in txns: |
|
if self.make_int(tx['amount']) == bid_amount: |
|
txid = bytes.fromhex(tx['txid']) |
|
break |
|
|
|
if txid is None: |
|
return None |
|
|
|
try: |
|
tx = self.rpc('gettransaction', [txid.hex()]) |
|
|
|
block_height = 0 |
|
if 'blockhash' in tx: |
|
block_header = self.rpc('getblockheader', [tx['blockhash']]) |
|
block_height = block_header['height'] |
|
|
|
rv = { |
|
'depth': 0 if 'confirmations' not in tx else tx['confirmations'], |
|
'height': block_height} |
|
|
|
except Exception as e: |
|
self._log.debug('getLockTxHeight gettransaction failed: %s, %s', txid.hex(), str(e)) |
|
return None |
|
|
|
if find_index: |
|
tx_obj = self.rpc('decoderawtransaction', [tx['hex']]) |
|
rv['index'] = find_vout_for_address_from_txobj(tx_obj, dest_address) |
|
|
|
if return_txid: |
|
rv['txid'] = txid.hex() |
|
|
|
return rv |
|
|
|
def getBlockWithTxns(self, block_hash): |
|
# TODO: Bypass decoderawtransaction and getblockheader |
|
block = self.rpc('getblock', [block_hash, False]) |
|
block_header = self.rpc('getblockheader', [block_hash]) |
|
decoded_block = CBlock() |
|
decoded_block = FromHex(decoded_block, block) |
|
|
|
tx_rv = [] |
|
for tx in decoded_block.vtx: |
|
tx_hex = tx.serialize_with_witness().hex() |
|
tx_dec = self.rpc('decoderawtransaction', [tx_hex]) |
|
if 'hex' not in tx_dec: |
|
tx_dec['hex'] = tx_hex |
|
|
|
tx_rv.append(tx_dec) |
|
|
|
block_rv = { |
|
'hash': block_hash, |
|
'tx': tx_rv, |
|
'confirmations': block_header['confirmations'], |
|
'height': block_header['height'], |
|
'version': block_header['version'], |
|
'merkleroot': block_header['merkleroot'], |
|
} |
|
|
|
return block_rv |
|
|
|
def getScriptScriptSig(self, script: bytes) -> bytearray: |
|
return self.getP2SHP2WSHScriptSig(script) |
|
|
|
def getScriptDest(self, script): |
|
return self.getP2SHP2WSHDest(script) |
|
|
|
def getDestForScriptHash(self, script_hash): |
|
assert len(script_hash) == 20 |
|
return CScript([OP_HASH160, script_hash, OP_EQUAL]) |
|
|
|
def pubkey_to_segwit_address(self, pk: bytes) -> str: |
|
pkh = hash160(pk) |
|
script_out = self.getScriptForPubkeyHash(pkh) |
|
return self.encodeSegwitAddressScript(script_out) |
|
|
|
def createBLockTx(self, Kbs: bytes, output_amount: int, vkbv=None) -> bytes: |
|
tx = CTransaction() |
|
tx.nVersion = self.txVersion() |
|
script_pk = self.getPkDest(Kbs) |
|
tx.vout.append(self.txoType()(output_amount, script_pk)) |
|
return tx.serialize() |
|
|
|
def spendBLockTx(self, chain_b_lock_txid: bytes, address_to: str, kbv: bytes, kbs: bytes, cb_swap_value: int, b_fee: int, restore_height: int) -> bytes: |
|
self._log.info('spendBLockTx %s:\n', chain_b_lock_txid.hex()) |
|
wtx = self.rpc('gettransaction', [chain_b_lock_txid.hex(), ]) |
|
lock_tx = self.loadTx(bytes.fromhex(wtx['hex'])) |
|
|
|
Kbs = self.getPubkey(kbs) |
|
script_pk = self.getPkDest(Kbs) |
|
locked_n = findOutput(lock_tx, script_pk) |
|
ensure(locked_n is not None, 'Output not found in tx') |
|
pkh_to = self.decodeAddress(address_to) |
|
|
|
tx = CTransaction() |
|
tx.nVersion = self.txVersion() |
|
|
|
chain_b_lock_txid_int = uint256_from_str(chain_b_lock_txid[::-1]) |
|
|
|
script_sig = self.getInputScriptForPubkeyHash(self.getPubkeyHash(Kbs)) |
|
|
|
tx.vin.append(CTxIn(COutPoint(chain_b_lock_txid_int, locked_n), |
|
nSequence=0, |
|
scriptSig=script_sig)) |
|
tx.vout.append(self.txoType()(cb_swap_value, self.getScriptForPubkeyHash(pkh_to))) |
|
|
|
pay_fee = self.getBLockSpendTxFee(tx, b_fee) |
|
tx.vout[0].nValue = cb_swap_value - pay_fee |
|
|
|
b_lock_spend_tx = tx.serialize() |
|
b_lock_spend_tx = self.signTxWithKey(b_lock_spend_tx, kbs, cb_swap_value) |
|
|
|
return bytes.fromhex(self.publishTx(b_lock_spend_tx)) |
|
|
|
def signTxWithKey(self, tx: bytes, key: bytes, prev_amount: int) -> bytes: |
|
Key = self.getPubkey(key) |
|
pkh = self.getPubkeyHash(Key) |
|
script = self.getScriptForP2PKH(pkh) |
|
|
|
sig = self.signTx(key, tx, 0, script, prev_amount) |
|
|
|
stack = [ |
|
sig, |
|
Key, |
|
] |
|
return self.setTxSignature(tx, stack) |
|
|
|
def findTxnByHash(self, txid_hex: str): |
|
# Only works for wallet txns |
|
try: |
|
rv = self.rpc('gettransaction', [txid_hex]) |
|
except Exception as ex: |
|
self._log.debug('findTxnByHash getrawtransaction failed: {}'.format(txid_hex)) |
|
return None |
|
if 'confirmations' in rv and rv['confirmations'] >= self.blocks_confirmed: |
|
block_height = self.getBlockHeader(rv['blockhash'])['height'] |
|
return {'txid': txid_hex, 'amount': 0, 'height': block_height} |
|
return None |
|
|
|
def createSCLockTx(self, value: int, script: bytearray, vkbv: bytes = None) -> bytes: |
|
tx = CTransaction() |
|
tx.nVersion = self.txVersion() |
|
tx.vout.append(self.txoType()(value, self.getScriptDest(script))) |
|
|
|
return tx.serialize() |
|
|
|
def fundTx(self, tx_hex: str, feerate: int, lock_unspents: bool = True): |
|
feerate_str = self.format_amount(feerate) |
|
# TODO: unlock unspents if bid cancelled |
|
options = { |
|
'lockUnspents': lock_unspents, |
|
'feeRate': feerate_str, |
|
} |
|
rv = self.rpc('fundrawtransaction', [tx_hex, options]) |
|
|
|
# Sign transaction then strip witness data to fill scriptsig |
|
rv = self.rpc('signrawtransaction', [rv['hex']]) |
|
|
|
tx_signed = self.loadTx(bytes.fromhex(rv['hex'])) |
|
if len(tx_signed.vin) != len(tx_signed.wit.vtxinwit): |
|
raise ValueError('txn has non segwit input') |
|
for witness_data in tx_signed.wit.vtxinwit: |
|
if len(witness_data.scriptWitness.stack) < 2: |
|
raise ValueError('txn has non segwit input') |
|
|
|
return tx_signed.serialize_without_witness() |
|
|
|
def fundSCLockTx(self, tx_bytes: bytes, feerate, vkbv=None) -> bytes: |
|
tx_funded = self.fundTx(tx_bytes.hex(), feerate) |
|
return tx_funded |
|
|
|
def createSCLockRefundTx(self, tx_lock_bytes, script_lock, Kal, Kaf, lock1_value, csv_val, tx_fee_rate, vkbv=None): |
|
tx_lock = CTransaction() |
|
tx_lock = self.loadTx(tx_lock_bytes) |
|
|
|
output_script = self.getScriptDest(script_lock) |
|
locked_n = findOutput(tx_lock, output_script) |
|
ensure(locked_n is not None, 'Output not found in tx') |
|
locked_coin = tx_lock.vout[locked_n].nValue |
|
|
|
tx_lock.rehash() |
|
tx_lock_id_int = tx_lock.sha256 |
|
|
|
refund_script = self.genScriptLockRefundTxScript(Kal, Kaf, csv_val) |
|
tx = CTransaction() |
|
tx.nVersion = self.txVersion() |
|
tx.vin.append(CTxIn(COutPoint(tx_lock_id_int, locked_n), |
|
nSequence=lock1_value, |
|
scriptSig=self.getScriptScriptSig(script_lock))) |
|
tx.vout.append(self.txoType()(locked_coin, self.getScriptDest(refund_script))) |
|
|
|
dummy_witness_stack = self.getScriptLockTxDummyWitness(script_lock) |
|
witness_bytes = self.getWitnessStackSerialisedLength(dummy_witness_stack) |
|
vsize = self.getTxVSize(tx, add_witness_bytes=witness_bytes) |
|
pay_fee = round(tx_fee_rate * vsize / 1000) |
|
tx.vout[0].nValue = locked_coin - pay_fee |
|
|
|
tx.rehash() |
|
self._log.info('createSCLockRefundTx %s:\n fee_rate, vsize, fee: %ld, %ld, %ld.', |
|
i2h(tx.sha256), tx_fee_rate, vsize, pay_fee) |
|
|
|
return tx.serialize(), refund_script, tx.vout[0].nValue |
|
|
|
def createSCLockRefundSpendTx(self, tx_lock_refund_bytes, script_lock_refund, pkh_refund_to, tx_fee_rate, vkbv=None): |
|
# Returns the coinA locked coin to the leader |
|
# The follower will sign the multisig path with a signature encumbered by the leader's coinB spend pubkey |
|
# If the leader publishes the decrypted signature the leader's coinB spend privatekey will be revealed to the follower |
|
|
|
tx_lock_refund = self.loadTx(tx_lock_refund_bytes) |
|
|
|
output_script = self.getScriptDest(script_lock_refund) |
|
locked_n = findOutput(tx_lock_refund, output_script) |
|
ensure(locked_n is not None, 'Output not found in tx') |
|
locked_coin = tx_lock_refund.vout[locked_n].nValue |
|
|
|
tx_lock_refund.rehash() |
|
tx_lock_refund_hash_int = tx_lock_refund.sha256 |
|
|
|
tx = CTransaction() |
|
tx.nVersion = self.txVersion() |
|
tx.vin.append(CTxIn(COutPoint(tx_lock_refund_hash_int, locked_n), |
|
nSequence=0, |
|
scriptSig=self.getScriptScriptSig(script_lock_refund))) |
|
|
|
tx.vout.append(self.txoType()(locked_coin, self.getScriptForPubkeyHash(pkh_refund_to))) |
|
|
|
dummy_witness_stack = self.getScriptLockRefundSpendTxDummyWitness(script_lock_refund) |
|
witness_bytes = self.getWitnessStackSerialisedLength(dummy_witness_stack) |
|
vsize = self.getTxVSize(tx, add_witness_bytes=witness_bytes) |
|
pay_fee = round(tx_fee_rate * vsize / 1000) |
|
tx.vout[0].nValue = locked_coin - pay_fee |
|
|
|
tx.rehash() |
|
self._log.info('createSCLockRefundSpendTx %s:\n fee_rate, vsize, fee: %ld, %ld, %ld.', |
|
i2h(tx.sha256), tx_fee_rate, vsize, pay_fee) |
|
|
|
return tx.serialize() |
|
|
|
def createSCLockRefundSpendToFTx(self, tx_lock_refund_bytes, script_lock_refund, pkh_dest, tx_fee_rate, vkbv=None): |
|
# lock refund swipe tx |
|
# Sends the coinA locked coin to the follower |
|
|
|
tx_lock_refund = self.loadTx(tx_lock_refund_bytes) |
|
|
|
output_script = self.getScriptDest(script_lock_refund) |
|
locked_n = findOutput(tx_lock_refund, output_script) |
|
ensure(locked_n is not None, 'Output not found in tx') |
|
locked_coin = tx_lock_refund.vout[locked_n].nValue |
|
|
|
A, B, lock2_value, C = self.extractScriptLockRefundScriptValues(script_lock_refund) |
|
|
|
tx_lock_refund.rehash() |
|
tx_lock_refund_hash_int = tx_lock_refund.sha256 |
|
|
|
tx = CTransaction() |
|
tx.nVersion = self.txVersion() |
|
tx.vin.append(CTxIn(COutPoint(tx_lock_refund_hash_int, locked_n), |
|
nSequence=lock2_value, |
|
scriptSig=self.getScriptScriptSig(script_lock_refund))) |
|
|
|
tx.vout.append(self.txoType()(locked_coin, self.getScriptForPubkeyHash(pkh_dest))) |
|
|
|
dummy_witness_stack = self.getScriptLockRefundSwipeTxDummyWitness(script_lock_refund) |
|
witness_bytes = self.getWitnessStackSerialisedLength(dummy_witness_stack) |
|
vsize = self.getTxVSize(tx, add_witness_bytes=witness_bytes) |
|
pay_fee = round(tx_fee_rate * vsize / 1000) |
|
tx.vout[0].nValue = locked_coin - pay_fee |
|
|
|
tx.rehash() |
|
self._log.info('createSCLockRefundSpendToFTx %s:\n fee_rate, vsize, fee: %ld, %ld, %ld.', |
|
i2h(tx.sha256), tx_fee_rate, vsize, pay_fee) |
|
|
|
return tx.serialize() |
|
|
|
def createSCLockSpendTx(self, tx_lock_bytes, script_lock, pkh_dest, tx_fee_rate, vkbv=None, fee_info={}): |
|
tx_lock = self.loadTx(tx_lock_bytes) |
|
output_script = self.getScriptDest(script_lock) |
|
locked_n = findOutput(tx_lock, output_script) |
|
ensure(locked_n is not None, 'Output not found in tx') |
|
locked_coin = tx_lock.vout[locked_n].nValue |
|
|
|
tx_lock.rehash() |
|
tx_lock_id_int = tx_lock.sha256 |
|
|
|
tx = CTransaction() |
|
tx.nVersion = self.txVersion() |
|
tx.vin.append(CTxIn(COutPoint(tx_lock_id_int, locked_n), |
|
scriptSig=self.getScriptScriptSig(script_lock))) |
|
|
|
tx.vout.append(self.txoType()(locked_coin, self.getScriptForPubkeyHash(pkh_dest))) |
|
|
|
dummy_witness_stack = self.getScriptLockTxDummyWitness(script_lock) |
|
witness_bytes = self.getWitnessStackSerialisedLength(dummy_witness_stack) |
|
vsize = self.getTxVSize(tx, add_witness_bytes=witness_bytes) |
|
pay_fee = round(tx_fee_rate * vsize / 1000) |
|
tx.vout[0].nValue = locked_coin - pay_fee |
|
|
|
fee_info['fee_paid'] = pay_fee |
|
fee_info['rate_used'] = tx_fee_rate |
|
fee_info['witness_bytes'] = witness_bytes |
|
fee_info['vsize'] = vsize |
|
|
|
tx.rehash() |
|
self._log.info('createSCLockSpendTx %s:\n fee_rate, vsize, fee: %ld, %ld, %ld.', |
|
i2h(tx.sha256), tx_fee_rate, vsize, pay_fee) |
|
|
|
return tx.serialize()
|
|
|