mirror of
https://github.com/JoinMarket-Org/joinmarket-clientserver.git
synced 2026-08-17 13:07:51 +02:00
54618e61da
Previous to this commit, the jmbitcoin package was accessed via a file in jmclient `btc.py` which was originally added as an interface to allow the client to use a non-jmbitcoin package to provide the implementation; while this idea is useful, the way it was implemented was not, moreover it is not currently used and contained duplicated code that was unmanaged. Also, the original usage of this was only by the electrum plugin, which has currently been abandoned. This simplifies the code and avoids spurious error messages. Note that most of the changes are a result of pulling the logging function directly from the jmbase package instead of indirectly via this interface (which was unnecessary and not connected with jmbitcoin).
119 lines
4.6 KiB
Python
119 lines
4.6 KiB
Python
#! /usr/bin/env python
|
|
from __future__ import (absolute_import, division,
|
|
print_function, unicode_literals)
|
|
from builtins import * # noqa: F401
|
|
'''Some helper functions for testing'''
|
|
|
|
import sys
|
|
import os
|
|
import binascii
|
|
import random
|
|
from decimal import Decimal
|
|
|
|
data_dir = os.path.dirname(os.path.dirname(os.path.realpath(__file__)))
|
|
sys.path.insert(0, os.path.join(data_dir))
|
|
|
|
from jmbase import get_log
|
|
from jmclient import open_test_wallet_maybe, BIP32Wallet, SegwitLegacyWallet, \
|
|
estimate_tx_fee, jm_single
|
|
import jmbitcoin as btc
|
|
from jmbase import chunks
|
|
|
|
log = get_log()
|
|
|
|
def make_sign_and_push(ins_full,
|
|
wallet,
|
|
amount,
|
|
output_addr=None,
|
|
change_addr=None,
|
|
hashcode=btc.SIGHASH_ALL,
|
|
estimate_fee = False):
|
|
"""Utility function for easily building transactions
|
|
from wallets
|
|
"""
|
|
total = sum(x['value'] for x in ins_full.values())
|
|
ins = ins_full.keys()
|
|
#random output address and change addr
|
|
output_addr = wallet.get_new_addr(1, 1) if not output_addr else output_addr
|
|
change_addr = wallet.get_new_addr(1, 0) if not change_addr else change_addr
|
|
fee_est = estimate_tx_fee(len(ins), 2) if estimate_fee else 10000
|
|
outs = [{'value': amount,
|
|
'address': output_addr}, {'value': total - amount - fee_est,
|
|
'address': change_addr}]
|
|
|
|
tx = btc.mktx(ins, outs)
|
|
de_tx = btc.deserialize(tx)
|
|
for index, ins in enumerate(de_tx['ins']):
|
|
utxo = ins['outpoint']['hash'] + ':' + str(ins['outpoint']['index'])
|
|
addr = ins_full[utxo]['address']
|
|
priv = wallet.get_key_from_addr(addr)
|
|
if index % 2:
|
|
priv = binascii.unhexlify(priv)
|
|
tx = btc.sign(tx, index, priv, hashcode=hashcode)
|
|
#pushtx returns False on any error
|
|
print(btc.deserialize(tx))
|
|
push_succeed = jm_single().bc_interface.pushtx(tx)
|
|
if push_succeed:
|
|
return btc.txhash(tx)
|
|
else:
|
|
return False
|
|
|
|
def make_wallets(n,
|
|
wallet_structures=None,
|
|
mean_amt=1,
|
|
sdev_amt=0,
|
|
start_index=0,
|
|
fixed_seeds=None,
|
|
test_wallet=False,
|
|
passwords=None,
|
|
walletclass=SegwitLegacyWallet,
|
|
mixdepths=5):
|
|
'''n: number of wallets to be created
|
|
wallet_structure: array of n arrays , each subarray
|
|
specifying the number of addresses to be populated with coins
|
|
at each depth (for now, this will only populate coins into 'receive' addresses)
|
|
mean_amt: the number of coins (in btc units) in each address as above
|
|
sdev_amt: if randomness in amouts is desired, specify here.
|
|
Returns: a dict of dicts of form {0:{'seed':seed,'wallet':Wallet object},1:..,}
|
|
Default Wallet constructor is joinmarket.Wallet, else use TestWallet,
|
|
which takes a password parameter as in the list passwords.
|
|
'''
|
|
# FIXME: this is basically the same code as jmclient/test/commontest.py
|
|
if len(wallet_structures) != n:
|
|
raise Exception("Number of wallets doesn't match wallet structures")
|
|
if not fixed_seeds:
|
|
seeds = chunks(binascii.hexlify(os.urandom(BIP32Wallet.ENTROPY_BYTES * n)),
|
|
BIP32Wallet.ENTROPY_BYTES * 2)
|
|
else:
|
|
seeds = fixed_seeds
|
|
wallets = {}
|
|
for i in range(n):
|
|
assert len(seeds[i]) == BIP32Wallet.ENTROPY_BYTES * 2
|
|
|
|
# FIXME: pwd is ignored (but do we really need this anyway?)
|
|
if test_wallet and passwords and i < len(passwords):
|
|
pwd = passwords[i]
|
|
else:
|
|
pwd = None
|
|
|
|
w = open_test_wallet_maybe(seeds[i], seeds[i], mixdepths - 1,
|
|
test_wallet_cls=walletclass)
|
|
|
|
wallets[i + start_index] = {'seed': seeds[i].decode('ascii'),
|
|
'wallet': w}
|
|
for j in range(mixdepths):
|
|
for k in range(wallet_structures[i][j]):
|
|
deviation = sdev_amt * random.random()
|
|
amt = mean_amt - sdev_amt / 2.0 + deviation
|
|
if amt < 0: amt = 0.001
|
|
amt = float(Decimal(amt).quantize(Decimal(10)**-8))
|
|
jm_single().bc_interface.grab_coins(w.get_external_addr(j), amt)
|
|
return wallets
|
|
|
|
|
|
def interact(process, inputs, expected):
|
|
if len(inputs) != len(expected):
|
|
raise Exception("Invalid inputs to interact()")
|
|
for i, inp in enumerate(inputs):
|
|
process.expect(expected[i])
|
|
process.sendline(inp)
|