AdamISZ
7 years ago
1 changed files with 0 additions and 269 deletions
@ -1,269 +0,0 @@ |
|||||||
#!/usr/bin/env python2 |
|
||||||
from __future__ import print_function |
|
||||||
""" |
|
||||||
Tool to create or receive a single input, output pair |
|
||||||
signed single|acp to allow the receiver to add further |
|
||||||
inputs and outputs and broadcast a full transaction. |
|
||||||
Thus the original input can be tainted and the taint |
|
||||||
can spread to other outputs included. |
|
||||||
This is a tool for Joinmarket wallets specifically. |
|
||||||
""" |
|
||||||
import binascii |
|
||||||
from optparse import OptionParser |
|
||||||
from pprint import pformat |
|
||||||
import jmbitcoin as btc |
|
||||||
from jmclient import ( |
|
||||||
load_program_config, validate_address, jm_single, WalletError, sync_wallet, |
|
||||||
RegtestBitcoinCoreInterface, estimate_tx_fee, get_p2pk_vbyte, |
|
||||||
get_p2sh_vbyte, open_test_wallet_maybe, get_wallet_path) |
|
||||||
|
|
||||||
|
|
||||||
def get_parser(): |
|
||||||
parser = OptionParser( |
|
||||||
usage= |
|
||||||
'usage: %prog [options] walletfile make utxo\n' + \ |
|
||||||
'or: %prog [options] walletfile take amount-in-satoshis destination-addr txhex', |
|
||||||
description='Makes a single|acp signed in-out pair for giving to others ' |
|
||||||
+ |
|
||||||
'or receives such a pair and creates a full transaction with your utxos. ' |
|
||||||
+ |
|
||||||
'Primarily useful for spreading utxo taint.') |
|
||||||
parser.add_option( |
|
||||||
'-b', |
|
||||||
'--bump', |
|
||||||
type='int', |
|
||||||
dest='bump', |
|
||||||
default=10000, |
|
||||||
help= |
|
||||||
'How much bigger the output is than the input, when making a single|acp in-out pair.') |
|
||||||
parser.add_option( |
|
||||||
'-m', |
|
||||||
'--mixdepth', |
|
||||||
type='int', |
|
||||||
dest='mixdepth', |
|
||||||
help= |
|
||||||
'Mixing depth to source utxo from. default=0.', |
|
||||||
default=0) |
|
||||||
parser.add_option('-a', |
|
||||||
'--amtmixdepths', |
|
||||||
action='store', |
|
||||||
type='int', |
|
||||||
dest='amtmixdepths', |
|
||||||
help='number of mixdepths in wallet, default 5', |
|
||||||
default=5) |
|
||||||
parser.add_option('-g', |
|
||||||
'--gap-limit', |
|
||||||
type="int", |
|
||||||
action='store', |
|
||||||
dest='gaplimit', |
|
||||||
help='gap limit for wallet, default=6', |
|
||||||
default=6) |
|
||||||
parser.add_option('--fast', |
|
||||||
action='store_true', |
|
||||||
dest='fastsync', |
|
||||||
default=False, |
|
||||||
help=('choose to do fast wallet sync, only for Core and ' |
|
||||||
'only for previously synced wallet')) |
|
||||||
return parser |
|
||||||
|
|
||||||
def is_utxo(utxo): |
|
||||||
try: |
|
||||||
txid, N = utxo.split(":") |
|
||||||
assert len(txid) == 64 |
|
||||||
N = int(N) |
|
||||||
except: |
|
||||||
return False |
|
||||||
return True |
|
||||||
|
|
||||||
def cli_get_wallet(wallet_name, sync=True): |
|
||||||
wallet_path = get_wallet_path(wallet_name, None) |
|
||||||
wallet = open_test_wallet_maybe( |
|
||||||
wallet_path, wallet_name, options.amtmixdepths, gap_limit=options.gaplimit) |
|
||||||
|
|
||||||
if jm_single().config.get("BLOCKCHAIN", |
|
||||||
"blockchain_source") == "electrum-server": |
|
||||||
jm_single().bc_interface.synctype = "with-script" |
|
||||||
if sync: |
|
||||||
while not jm_single().bc_interface.wallet_synced: |
|
||||||
sync_wallet(wallet, fast=options.fastsync) |
|
||||||
return wallet |
|
||||||
|
|
||||||
#======Electrum specific utils========================= |
|
||||||
def rev_hex(s): |
|
||||||
return s.decode('hex')[::-1].encode('hex') |
|
||||||
|
|
||||||
def int_to_hex(i, length=1): |
|
||||||
s = hex(i)[2:].rstrip('L') |
|
||||||
s = "0"*(2*length - len(s)) + s |
|
||||||
return rev_hex(s) |
|
||||||
|
|
||||||
def serialize_derivation(roc, i): |
|
||||||
x = ''.join(map(lambda x: int_to_hex(x, 2), (roc, i))) |
|
||||||
print("returning: ", x) |
|
||||||
raw_input() |
|
||||||
return x |
|
||||||
#======================================================= |
|
||||||
|
|
||||||
|
|
||||||
def get_script_amount_from_utxo(wallet, utxo): |
|
||||||
"""Given a JM wallet and a utxo string, find |
|
||||||
the corresponding private key and amount controlled |
|
||||||
in satoshis. |
|
||||||
""" |
|
||||||
for md, utxos in wallet.get_utxos_by_mixdepth_().items(): |
|
||||||
for (txid, index), utxo in utxos.items(): |
|
||||||
txhex = binascii.hexlify(txid) + ':' + str(index) |
|
||||||
if txhex != utxo: |
|
||||||
continue |
|
||||||
script = wallet.get_script_path(utxo['path']) |
|
||||||
print("Found utxo, its value is: {}".format(utxo['value'])) |
|
||||||
return script, utxo['value'] |
|
||||||
return None, None |
|
||||||
|
|
||||||
|
|
||||||
def create_single_acp_pair(wallet, utxo_in, script, addr_out, amount, bump, segwit=False): |
|
||||||
"""Given a utxo and a signing key for it, and its amout in satoshis, |
|
||||||
sign a "transaction" consisting of only 1 input and one output, signed |
|
||||||
with single|acp sighash flags so it can be grafted into a bigger |
|
||||||
transaction. |
|
||||||
Also provide a destination address and a 'bump' value (so the creator |
|
||||||
can claim more output in the final transaction. |
|
||||||
Note that, for safety, bump *should* be positive if the recipient |
|
||||||
is untrusted, since otherwise they can waste your money by simply |
|
||||||
broadcasting this transaction without adding any inputs of their own. |
|
||||||
Returns the serialized 1 in, 1 out, and signed transaction. |
|
||||||
""" |
|
||||||
assert bump >= 0, "Output of single|acp pair must be bigger than input for safety." |
|
||||||
out = {"address": addr_out, "value": amount + bump} |
|
||||||
tx = btc.mktx([utxo_in], [out]) |
|
||||||
return wallet.sign_tx(tx, {0: (script, amount)}, |
|
||||||
hashcode=btc.SIGHASH_SINGLE|btc.SIGHASH_ANYONECANPAY) |
|
||||||
|
|
||||||
|
|
||||||
def graft_onto_single_acp(wallet, txhex, amount, destaddr): |
|
||||||
"""Given a serialized txhex which is checked to be of |
|
||||||
form single|acp (one in, one out), a destination address |
|
||||||
and an amount to spend, grafts in this in-out pair (at index zero) |
|
||||||
to our own transaction spending amount amount to destination destaddr, |
|
||||||
and uses a user-specified transaction fee (normal joinmarket |
|
||||||
configuration), and sanity checks that the bump value is not |
|
||||||
greater than user specified bump option. |
|
||||||
Returned: serialized txhex of fully signed transaction. |
|
||||||
""" |
|
||||||
d = btc.deserialize(txhex) |
|
||||||
if len(d['ins']) != 1 or len(d['outs']) != 1: |
|
||||||
return (False, "Proposed tx should have 1 in 1 out, has: " + ','.join( |
|
||||||
[str(len(d[x])) for x in ['ins', 'outs']])) |
|
||||||
#most important part: check provider hasn't bumped more than options.bump: |
|
||||||
other_utxo_in = d['ins'][0]['outpoint']['hash'] + ":" + str(d['ins'][0]['outpoint']['index']) |
|
||||||
res = jm_single().bc_interface.query_utxo_set(other_utxo_in) |
|
||||||
assert len(res) == 1 |
|
||||||
if not res[0]: |
|
||||||
return (False, "Utxo provided by counterparty not found.") |
|
||||||
excess = d['outs'][0]['value'] - res[0]["value"] |
|
||||||
if not excess <= options.bump: |
|
||||||
return (False, "Counterparty claims too much excess value: " + str(excess)) |
|
||||||
#Last sanity check - ensure that it's single|acp, else we're wasting our time |
|
||||||
try: |
|
||||||
if 'txinwitness' in d['ins'][0]: |
|
||||||
sig, pub = d['ins'][0]['txinwitness'] |
|
||||||
else: |
|
||||||
sig, pub = btc.deserialize_script(d['ins'][0]['script']) |
|
||||||
assert sig[-2:] == "83" |
|
||||||
except Exception as e: |
|
||||||
return (False, "The transaction's signature does not parse as signed with " |
|
||||||
"SIGHASH_SINGLE|SIGHASH_ANYONECANPAY, for p2pkh or p2sh-p2wpkh, or " |
|
||||||
"is otherwise invalid, and so is not valid for this function.\n" + repr(e)) |
|
||||||
#source inputs for our own chosen spending amount: |
|
||||||
try: |
|
||||||
input_utxos = wallet.select_utxos(options.mixdepth, amount) |
|
||||||
except Exception as e: |
|
||||||
return (False, "Unable to select sufficient coins from mixdepth: " + str(options.mixdepth)) |
|
||||||
total_selected = sum([x['value'] for x in input_utxos.values()]) |
|
||||||
fee = estimate_tx_fee(len(input_utxos)+1, 3, txtype='p2sh-p2wpkh') |
|
||||||
change_amount = total_selected - amount - excess - fee |
|
||||||
changeaddr = wallet.get_new_addr(options.mixdepth, 1) |
|
||||||
#Build new transaction and, graft in signature |
|
||||||
ins = [other_utxo_in] + input_utxos.keys() |
|
||||||
outs = [d['outs'][0], {'address': destaddr, 'value': amount}, |
|
||||||
{'address': changeaddr, 'value': change_amount}] |
|
||||||
fulltx = btc.mktx(ins, outs) |
|
||||||
df = btc.deserialize(fulltx) |
|
||||||
#put back in original signature |
|
||||||
df['ins'][0]['script'] = d['ins'][0]['script'] |
|
||||||
if 'txinwitness' in d['ins'][0]: |
|
||||||
df['ins'][0]['txinwitness'] = d['ins'][0]['txinwitness'] |
|
||||||
|
|
||||||
for i, iu in enumerate(input_utxos): |
|
||||||
script, inamt = get_script_amount_from_utxo(wallet, iu) |
|
||||||
print("Signing index: ", i+1, " with script: ", script, " and amount: ", inamt, " for utxo: ", iu) |
|
||||||
fulltx = wallet.sign_tx(df, {i: (script, inamt)}) |
|
||||||
|
|
||||||
return True, btc.serialize(fulltx) |
|
||||||
|
|
||||||
|
|
||||||
if __name__ == "__main__": |
|
||||||
parser = get_parser() |
|
||||||
(options, args) = parser.parse_args() |
|
||||||
load_program_config() |
|
||||||
if get_wallet_cls() != SegwitWallet: |
|
||||||
print("Only segwit wallets are supported; remove any setting of `segwit`" |
|
||||||
" in `POLICY` in joinmarket.cfg. Quitting.") |
|
||||||
exit(0) |
|
||||||
#default args causes wallet sync here: |
|
||||||
wallet = cli_get_wallet(args[0]) |
|
||||||
if args[1] not in ['make', 'take']: |
|
||||||
print("Second argument must be 'make' or 'take', see '--help'") |
|
||||||
exit(0) |
|
||||||
if args[1] == "make": |
|
||||||
if len(args) < 3 or not is_utxo(args[2]): |
|
||||||
print("You must provide a utxo as third argument; 64 character hex " |
|
||||||
"txid, followed by ':', followed by the output index. " |
|
||||||
"Use wallet-tool.py method 'showutxos' to select one") |
|
||||||
exit(0) |
|
||||||
utxo_in = args[2] |
|
||||||
script, amount = get_script_amount_from_utxo(wallet, utxo_in) |
|
||||||
if not script: |
|
||||||
print("Failed to find the utxo's private key from the wallet; check " |
|
||||||
"if this utxo is actually contained in the wallet using " |
|
||||||
"wallet-tool.py showutxos") |
|
||||||
exit(0) |
|
||||||
#destination sourced from wallet |
|
||||||
addr_out = wallet.get_new_addr((options.mixdepth+1)%options.amtmixdepths, 1) |
|
||||||
single_acp = create_single_acp_pair(wallet, utxo_in, script, addr_out, amount, |
|
||||||
options.bump, segwit=True) |
|
||||||
print("Created the following one-in, one-out transaction, which will not " |
|
||||||
"be valid to broadcast itself (negative fee). Pass it to your " |
|
||||||
"counterparty:") |
|
||||||
print(pformat(single_acp)) |
|
||||||
print("Pass the following raw hex to your counterparty:") |
|
||||||
print(btc.serialize(single_acp)) |
|
||||||
exit(0) |
|
||||||
elif args[1] == "take": |
|
||||||
try: |
|
||||||
amount, destaddr, txhex = args[2:5] |
|
||||||
#sanity check input |
|
||||||
amount = int(amount) |
|
||||||
assert amount > 0 |
|
||||||
assert validate_address(destaddr) |
|
||||||
binascii.unhexlify(txhex) |
|
||||||
except Exception as e: |
|
||||||
print("Syntax error, should be 5 arguments, see --help. ", repr(e)) |
|
||||||
exit(0) |
|
||||||
success, complete_tx = graft_onto_single_acp(wallet, txhex, amount, destaddr) |
|
||||||
if not success: |
|
||||||
print("Quitting, reason: " + complete_tx) |
|
||||||
exit(0) |
|
||||||
#allow user to decide whether to broadcast: |
|
||||||
print("The following transaction has been prepared:") |
|
||||||
print(pformat(btc.deserialize(complete_tx))) |
|
||||||
broadcast = raw_input("Do you want to broadcast now? (y/n): ") |
|
||||||
if broadcast == "y": |
|
||||||
success = jm_single().bc_interface.pushtx(complete_tx) |
|
||||||
if not success: |
|
||||||
print("Failed to broadcast.") |
|
||||||
exit(0) |
|
||||||
else: |
|
||||||
print("You chose not to broadcast.") |
|
||||||
exit(0) |
|
||||||
print('done') |
|
||||||
Loading…
Reference in new issue