81d60bc283
* fix issue with trx amounts precision in autoix-mempool.py * fix issue with trx amounts precision in p2p-autoinstantsend.py * fix issue with trx amounts precision in p2p-instantsend.py * Rallback adding empty line * Rallback adding empty line
307 lines
11 KiB
Python
Executable File
307 lines
11 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
# Copyright (c) 2018 The Dash Core developers
|
|
# Distributed under the MIT software license, see the accompanying
|
|
# file COPYING or http://www.opensource.org/licenses/mit-license.php.
|
|
|
|
from test_framework.mininode import *
|
|
from test_framework.test_framework import BitcoinTestFramework
|
|
from test_framework.util import *
|
|
from time import *
|
|
|
|
'''
|
|
autoix-mempool.py
|
|
|
|
Checks if automatic InstantSend locks stop working when transaction mempool
|
|
is full (more than 0.1 part from max value).
|
|
|
|
'''
|
|
|
|
MASTERNODE_COLLATERAL = 1000
|
|
MAX_MEMPOOL_SIZE = 5 # max node mempool in MBs
|
|
MB_SIZE = 1000000 # C++ code use this coefficient to calc MB in mempool
|
|
AUTO_IX_MEM_THRESHOLD = 0.1
|
|
|
|
|
|
class MasternodeInfo:
|
|
def __init__(self, key, collateral_id, collateral_out):
|
|
self.key = key
|
|
self.collateral_id = collateral_id
|
|
self.collateral_out = collateral_out
|
|
|
|
|
|
class AutoIXMempoolTest(BitcoinTestFramework):
|
|
def __init__(self):
|
|
super().__init__()
|
|
self.mn_count = 10
|
|
self.num_nodes = self.mn_count + 3
|
|
self.mninfo = []
|
|
self.setup_clean_chain = True
|
|
self.is_network_split = False
|
|
# set sender, receiver, isolated nodes
|
|
self.receiver_idx = self.num_nodes - 2
|
|
self.sender_idx = self.num_nodes - 3
|
|
# additional args
|
|
self.extra_args = ["-maxmempool=%d" % MAX_MEMPOOL_SIZE]
|
|
|
|
def create_simple_node(self):
|
|
idx = len(self.nodes)
|
|
args = ["-debug"] + self.extra_args
|
|
self.nodes.append(start_node(idx, self.options.tmpdir,
|
|
args))
|
|
for i in range(0, idx):
|
|
connect_nodes(self.nodes[i], idx)
|
|
|
|
def get_mnconf_file(self):
|
|
return os.path.join(self.options.tmpdir, "node0/regtest/masternode.conf")
|
|
|
|
def prepare_masternodes(self):
|
|
for idx in range(0, self.mn_count):
|
|
key = self.nodes[0].masternode("genkey")
|
|
address = self.nodes[0].getnewaddress()
|
|
txid = self.nodes[0].sendtoaddress(address, MASTERNODE_COLLATERAL)
|
|
txrow = self.nodes[0].getrawtransaction(txid, True)
|
|
collateral_vout = 0
|
|
for vout_idx in range(0, len(txrow["vout"])):
|
|
vout = txrow["vout"][vout_idx]
|
|
if vout["value"] == MASTERNODE_COLLATERAL:
|
|
collateral_vout = vout_idx
|
|
self.nodes[0].lockunspent(False,
|
|
[{"txid": txid, "vout": collateral_vout}])
|
|
self.mninfo.append(MasternodeInfo(key, txid, collateral_vout))
|
|
|
|
def write_mn_config(self):
|
|
conf = self.get_mnconf_file()
|
|
f = open(conf, 'a')
|
|
for idx in range(0, self.mn_count):
|
|
f.write("mn%d 127.0.0.1:%d %s %s %d\n" % (idx + 1, p2p_port(idx + 1),
|
|
self.mninfo[idx].key,
|
|
self.mninfo[idx].collateral_id,
|
|
self.mninfo[idx].collateral_out))
|
|
f.close()
|
|
|
|
def create_masternodes(self):
|
|
for idx in range(0, self.mn_count):
|
|
args = ['-debug=masternode', '-externalip=127.0.0.1', '-masternode=1',
|
|
'-masternodeprivkey=%s' % self.mninfo[idx].key] + self.extra_args
|
|
self.nodes.append(start_node(idx + 1, self.options.tmpdir, args))
|
|
for i in range(0, idx + 1):
|
|
connect_nodes(self.nodes[i], idx + 1)
|
|
|
|
def setup_network(self):
|
|
self.nodes = []
|
|
# create faucet node for collateral and transactions
|
|
args = ["-debug"] + self.extra_args
|
|
self.nodes.append(start_node(0, self.options.tmpdir, args))
|
|
required_balance = MASTERNODE_COLLATERAL * self.mn_count + 1
|
|
while self.nodes[0].getbalance() < required_balance:
|
|
set_mocktime(get_mocktime() + 1)
|
|
set_node_times(self.nodes, get_mocktime())
|
|
self.nodes[0].generate(1)
|
|
# create masternodes
|
|
self.prepare_masternodes()
|
|
self.write_mn_config()
|
|
stop_node(self.nodes[0], 0)
|
|
args = ["-debug",
|
|
"-sporkkey=cP4EKFyJsHT39LDqgdcB43Y3YXjNyjb5Fuas1GQSeAtjnZWmZEQK"] + \
|
|
self.extra_args
|
|
self.nodes[0] = start_node(0, self.options.tmpdir,
|
|
args)
|
|
self.create_masternodes()
|
|
# create connected simple nodes
|
|
for i in range(0, self.num_nodes - self.mn_count - 1):
|
|
self.create_simple_node()
|
|
set_mocktime(get_mocktime() + 1)
|
|
set_node_times(self.nodes, get_mocktime())
|
|
self.nodes[0].generate(1)
|
|
# sync nodes
|
|
self.sync_all()
|
|
set_mocktime(get_mocktime() + 1)
|
|
set_node_times(self.nodes, get_mocktime())
|
|
sync_masternodes(self.nodes)
|
|
for i in range(1, self.mn_count + 1):
|
|
res = self.nodes[0].masternode("start-alias", "mn%d" % i)
|
|
assert(res["result"] == 'successful')
|
|
sync_masternodes(self.nodes)
|
|
mn_info = self.nodes[0].masternodelist("status")
|
|
assert(len(mn_info) == self.mn_count)
|
|
for status in mn_info.values():
|
|
assert(status == 'ENABLED')
|
|
|
|
def get_autoix_bip9_status(self):
|
|
info = self.nodes[0].getblockchaininfo()
|
|
return info['bip9_softforks']['autoix']['status']
|
|
|
|
def activate_autoix_bip9(self):
|
|
# sync nodes periodically
|
|
# if we sync them too often, activation takes too many time
|
|
# if we sync them too rarely, nodes failed to update its state and
|
|
# bip9 status is not updated
|
|
# so, in this code nodes are synced once per 20 blocks
|
|
counter = 0
|
|
sync_period = 10
|
|
|
|
while self.get_autoix_bip9_status() == 'defined':
|
|
set_mocktime(get_mocktime() + 1)
|
|
set_node_times(self.nodes, get_mocktime())
|
|
self.nodes[0].generate(1)
|
|
counter += 1
|
|
if counter % sync_period == 0:
|
|
# sync nodes
|
|
self.sync_all()
|
|
sync_masternodes(self.nodes)
|
|
|
|
while self.get_autoix_bip9_status() == 'started':
|
|
set_mocktime(get_mocktime() + 1)
|
|
set_node_times(self.nodes, get_mocktime())
|
|
self.nodes[0].generate(1)
|
|
counter += 1
|
|
if counter % sync_period == 0:
|
|
# sync nodes
|
|
self.sync_all()
|
|
sync_masternodes(self.nodes)
|
|
|
|
while self.get_autoix_bip9_status() == 'locked_in':
|
|
set_mocktime(get_mocktime() + 1)
|
|
set_node_times(self.nodes, get_mocktime())
|
|
self.nodes[0].generate(1)
|
|
counter += 1
|
|
if counter % sync_period == 0:
|
|
# sync nodes
|
|
self.sync_all()
|
|
sync_masternodes(self.nodes)
|
|
|
|
# sync nodes
|
|
self.sync_all()
|
|
sync_masternodes(self.nodes)
|
|
|
|
assert(self.get_autoix_bip9_status() == 'active')
|
|
|
|
def get_autoix_spork_state(self):
|
|
info = self.nodes[0].spork('active')
|
|
return info['SPORK_16_INSTANTSEND_AUTOLOCKS']
|
|
|
|
def set_autoix_spork_state(self, state):
|
|
if state:
|
|
value = 0
|
|
else:
|
|
value = 4070908800
|
|
self.nodes[0].spork('SPORK_16_INSTANTSEND_AUTOLOCKS', value)
|
|
|
|
def enforce_masternode_payments(self):
|
|
self.nodes[0].spork('SPORK_8_MASTERNODE_PAYMENT_ENFORCEMENT', 0)
|
|
|
|
def create_raw_trx(self, node_from, node_to, amount, min_inputs, max_inputs):
|
|
assert(min_inputs <= max_inputs)
|
|
# fill inputs
|
|
inputs=[]
|
|
balances = node_from.listunspent()
|
|
in_amount = 0.0
|
|
last_amount = 0.0
|
|
for tx in balances:
|
|
if len(inputs) < min_inputs:
|
|
input = {}
|
|
input["txid"] = tx['txid']
|
|
input['vout'] = tx['vout']
|
|
in_amount += float(tx['amount'])
|
|
inputs.append(input)
|
|
elif in_amount > amount:
|
|
break
|
|
elif len(inputs) < max_inputs:
|
|
input = {}
|
|
input["txid"] = tx['txid']
|
|
input['vout'] = tx['vout']
|
|
in_amount += float(tx['amount'])
|
|
inputs.append(input)
|
|
else:
|
|
input = {}
|
|
input["txid"] = tx['txid']
|
|
input['vout'] = tx['vout']
|
|
in_amount -= last_amount
|
|
in_amount += float(tx['amount'])
|
|
inputs[-1] = input
|
|
last_amount = float(tx['amount'])
|
|
|
|
assert(len(inputs) > 0)
|
|
assert(in_amount > amount)
|
|
# fill outputs
|
|
receiver_address = node_to.getnewaddress()
|
|
change_address = node_from.getnewaddress()
|
|
fee = 0.001
|
|
outputs={}
|
|
outputs[receiver_address] = satoshi_round(amount)
|
|
outputs[change_address] = satoshi_round(in_amount - amount - fee)
|
|
rawtx = node_from.createrawtransaction(inputs, outputs)
|
|
return node_from.signrawtransaction(rawtx)
|
|
|
|
def check_IX_lock(self, txid, node):
|
|
# wait for instantsend locks
|
|
start = time()
|
|
locked = False
|
|
while True:
|
|
is_trx = node.gettransaction(txid)
|
|
if is_trx['instantlock']:
|
|
locked = True
|
|
break
|
|
if time() > start + 10:
|
|
break
|
|
sleep(0.1)
|
|
return locked
|
|
|
|
# sends regular IX with high fee and may inputs (not-simple transaction)
|
|
def send_regular_IX(self, sender, receiver):
|
|
receiver_addr = receiver.getnewaddress()
|
|
txid = sender.instantsendtoaddress(receiver_addr, 1.0)
|
|
return self.check_IX_lock(txid, sender)
|
|
|
|
# sends simple trx, it should become IX if autolocks are allowed
|
|
def send_simple_tx(self, sender, receiver):
|
|
raw_tx = self.create_raw_trx(sender, receiver, 1.0, 1, 4)
|
|
txid = self.nodes[0].sendrawtransaction(raw_tx['hex'])
|
|
self.sync_all()
|
|
return self.check_IX_lock(txid, sender)
|
|
|
|
def get_mempool_size(self, node):
|
|
info = node.getmempoolinfo()
|
|
return info['usage']
|
|
|
|
def fill_mempool(self):
|
|
node = self.nodes[0]
|
|
rec_address = node.getnewaddress()
|
|
while self.get_mempool_size(node) < MAX_MEMPOOL_SIZE * MB_SIZE * AUTO_IX_MEM_THRESHOLD + 10000:
|
|
node.sendtoaddress(rec_address, 1.0)
|
|
sleep(0.1)
|
|
self.sync_all()
|
|
|
|
def run_test(self):
|
|
self.enforce_masternode_payments() # required for bip9 activation
|
|
self.activate_autoix_bip9()
|
|
self.set_autoix_spork_state(True)
|
|
|
|
# check pre-conditions for autoIX
|
|
assert(self.get_autoix_bip9_status() == 'active')
|
|
assert(self.get_autoix_spork_state())
|
|
|
|
# autoIX is working
|
|
assert(self.send_simple_tx(self.nodes[0], self.nodes[self.receiver_idx]))
|
|
|
|
# send funds for InstantSend after filling mempool and give them 6 confirmations
|
|
rec_address = self.nodes[self.receiver_idx].getnewaddress()
|
|
self.nodes[0].sendtoaddress(rec_address, 500.0)
|
|
self.nodes[0].sendtoaddress(rec_address, 500.0)
|
|
self.sync_all()
|
|
for i in range(0, 2):
|
|
self.nodes[self.receiver_idx].generate(1)
|
|
self.sync_all()
|
|
|
|
# fill mempool with transactions
|
|
self.fill_mempool()
|
|
|
|
# autoIX is not working now
|
|
assert(not self.send_simple_tx(self.nodes[self.receiver_idx], self.nodes[0]))
|
|
# regular IX is still working
|
|
assert(self.send_regular_IX(self.nodes[self.receiver_idx], self.nodes[0]))
|
|
|
|
|
|
if __name__ == '__main__':
|
|
AutoIXMempoolTest().main()
|