-
Notifications
You must be signed in to change notification settings - Fork 377
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
a244777
commit 840478f
Showing
1 changed file
with
213 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,213 @@ | ||
#!/usr/bin/env python2 | ||
|
||
from test_framework.authproxy import AuthServiceProxy, JSONRPCException | ||
import os | ||
import random | ||
import sys | ||
import time | ||
import subprocess | ||
import shutil | ||
|
||
if len(sys.argv) != 3: | ||
print("paths to bitcoind and sidechain daemon must be included as arguments") | ||
sys.exit(0) | ||
print(sys.argv[1]) | ||
print(sys.argv[2]) | ||
|
||
def sync_all(sidechain, sidechain2): | ||
timeout = 20 | ||
while len(sidechain.getrawmempool()) != len(sidechain2.getrawmempool()): | ||
time.sleep(1) | ||
timeout -= 1 | ||
if timeout == 0: | ||
raise Exception("Peg-in has failed to propagate.") | ||
sidechain2.generate(1) | ||
while sidechain.getblockcount() != sidechain2.getblockcount(): | ||
time.sleep(1) | ||
timeout -= 1 | ||
if timeout == 0: | ||
raise Exception("Blocks are not propagating.") | ||
|
||
fedpeg_key="cPxqWyf1HDGpGFH1dnfjz8HbiWxvwG8WXyetbuAiw4thKXUdXLpR" | ||
fedpeg_pubkey="512103dff4923d778550cc13ce0d887d737553b4b58f4e8e886507fc39f5e447b2186451ae" | ||
|
||
bitcoin_datadir="/tmp/"+''.join(random.choice('0123456789ABCDEF') for i in range(5)) | ||
bitcoin_pass=''.join(random.choice('0123456789ABCDEF') for i in range(10)) | ||
sidechain_datadir="/tmp/"+''.join(random.choice('0123456789ABCDEF') for i in range(5)) | ||
sidechain_pass=''.join(random.choice('0123456789ABCDEF') for i in range(10)) | ||
sidechain2_datadir="/tmp/"+''.join(random.choice('0123456789ABCDEF') for i in range(5)) | ||
sidechain2_pass=''.join(random.choice('0123456789ABCDEF') for i in range(10)) | ||
|
||
bitcoin_port = 8000 + os.getpid()%999 | ||
sidechain_port = bitcoin_port + 1 | ||
sidechain2_port = bitcoin_port + 2 | ||
sidechain1_p2p_port = bitcoin_port + 3 | ||
sidechain2_p2p_port = bitcoin_port + 4 | ||
|
||
os.makedirs(bitcoin_datadir) | ||
os.makedirs(sidechain_datadir) | ||
os.makedirs(sidechain2_datadir) | ||
|
||
with open(os.path.join(bitcoin_datadir, "bitcoin.conf"), 'w') as f: | ||
f.write("regtest=1\n") | ||
f.write("rpcuser=bitcoinrpc\n") | ||
f.write("rpcpassword="+bitcoin_pass+"\n") | ||
f.write("rpcport="+str(bitcoin_port)+"\n") | ||
f.write("discover=0\n") | ||
f.write("listen=0\n") | ||
f.write("testnet=0\n") | ||
f.write("txindex=1\n") | ||
f.write("daemon=1\n") | ||
f.write("listen=0\n") | ||
|
||
with open(os.path.join(sidechain_datadir, "elements.conf"), 'w') as f: | ||
f.write("regtest=1\n") | ||
f.write("rpcuser=sidechainrpc\n") | ||
f.write("rpcpassword="+sidechain_pass+"\n") | ||
f.write("rpcport="+str(sidechain_port)+"\n") | ||
f.write("discover=0\n") | ||
f.write("testnet=0\n") | ||
f.write("txindex=1\n") | ||
f.write("fedpegscript="+fedpeg_pubkey+"\n") | ||
f.write("daemon=1\n") | ||
f.write("mainchainrpchost=127.0.0.1\n") | ||
f.write("mainchainrpcport="+str(bitcoin_port)+"\n") | ||
f.write("mainchainrpcuser=bitcoinrpc\n") | ||
f.write("mainchainrpcpassword="+bitcoin_pass+"\n") | ||
f.write("validatepegin=1\n") | ||
f.write("validatepegout=0\n") | ||
f.write("port="+str(sidechain1_p2p_port)+"\n") | ||
f.write("connect=localhost:"+str(sidechain2_p2p_port)+"\n") | ||
f.write("listen=1\n") | ||
|
||
with open(os.path.join(sidechain2_datadir, "elements.conf"), 'w') as f: | ||
f.write("regtest=1\n") | ||
f.write("rpcuser=sidechainrpc2\n") | ||
f.write("rpcpassword="+sidechain2_pass+"\n") | ||
f.write("rpcport="+str(sidechain2_port)+"\n") | ||
f.write("discover=0\n") | ||
f.write("testnet=0\n") | ||
f.write("txindex=1\n") | ||
f.write("fedpegscript="+fedpeg_pubkey+"\n") | ||
f.write("daemon=1\n") | ||
f.write("mainchainrpchost=127.0.0.1\n") | ||
f.write("mainchainrpcport="+str(bitcoin_port)+"\n") | ||
f.write("mainchainrpcuser=bitcoinrpc\n") | ||
f.write("mainchainrpcpassword="+bitcoin_pass+"\n") | ||
f.write("validatepegin=1\n") | ||
f.write("validatepegout=0\n") | ||
f.write("port="+str(sidechain2_p2p_port)+"\n") | ||
f.write("connect=localhost:"+str(sidechain1_p2p_port)+"\n") | ||
f.write("listen=1\n") | ||
|
||
try: | ||
|
||
# Default is 8, meaning 8+2 confirms for wallet acceptance normally | ||
# this will require 10+2. | ||
sidechain_args = " -peginconfirmationdepth=10 " | ||
|
||
# Start daemons | ||
print("Starting daemons at "+bitcoin_datadir+", "+sidechain_datadir+" and "+sidechain2_datadir) | ||
bitcoindstart = sys.argv[1]+"/bitcoind -datadir="+bitcoin_datadir | ||
subprocess.Popen(bitcoindstart.split(), stdout=subprocess.PIPE) | ||
|
||
sidechainstart = sys.argv[2]+"/elementsd -datadir="+sidechain_datadir + sidechain_args | ||
subprocess.Popen(sidechainstart.split(), stdout=subprocess.PIPE) | ||
|
||
sidechain2start = sys.argv[2]+"/elementsd -datadir="+sidechain2_datadir + sidechain_args | ||
subprocess.Popen(sidechain2start.split(), stdout=subprocess.PIPE) | ||
|
||
print("Daemons started") | ||
time.sleep(3) | ||
|
||
bitcoin = AuthServiceProxy("http://bitcoinrpc:"+bitcoin_pass+"@127.0.0.1:"+str(bitcoin_port)) | ||
sidechain = AuthServiceProxy("http://sidechainrpc:"+sidechain_pass+"@127.0.0.1:"+str(sidechain_port)) | ||
sidechain2 = AuthServiceProxy("http://sidechainrpc2:"+sidechain2_pass+"@127.0.0.1:"+str(sidechain2_port)) | ||
print("Daemons started, making blocks to get funds") | ||
|
||
bitcoin.generate(101) | ||
sidechain.generate(101) | ||
|
||
addr = bitcoin.getnewaddress() | ||
|
||
addrs = sidechain.getpeginaddress() | ||
txid1 = bitcoin.sendtoaddress(addrs["mainchain_address"], 24) | ||
# 10+2 confirms required to get into mempool and confirm | ||
bitcoin.generate(11) | ||
time.sleep(2) | ||
proof = bitcoin.gettxoutproof([txid1]) | ||
raw = bitcoin.getrawtransaction(txid1) | ||
|
||
print("Attempting peg-in") | ||
try: | ||
pegtxid = sidechain.claimpegin(raw, proof) | ||
raise Exception("Peg-in should not mature enough yet, need another block.") | ||
except JSONRPCException as e: | ||
assert("Peg-in Bitcoin transaction needs more confirmations to be sent." in e.error["message"]) | ||
pass | ||
|
||
# Should fail due to non-matching wallet address | ||
try: | ||
pegtxid = sidechain.claimpegin(raw, proof, sidechain.getnewaddress()) | ||
raise Exception("Peg-in with non-matching witness_program should fail.") | ||
except JSONRPCException as e: | ||
assert("Given witness_program is not a valid v0 witness program" in e.error["message"]) | ||
pass | ||
|
||
# 12 confirms allows in mempool | ||
bitcoin.generate(1) | ||
|
||
# Should succeed via wallet lookup for address match, and when given | ||
pegtxid1 = sidechain.claimpegin(raw, proof) | ||
|
||
sync_all(sidechain, sidechain2) | ||
|
||
tx1 = sidechain.gettransaction(pegtxid1) | ||
|
||
if "confirmations" in tx1 and tx1["confirmations"] > 0: | ||
print("Peg-in is confirmed: Success!") | ||
else: | ||
raise Exception("Peg-in confirmation has failed.") | ||
|
||
# Do many claims in mempool | ||
n_claims = 100 | ||
|
||
print("Flooding mempool with many small claims") | ||
pegtxs = [] | ||
sidechain.generate(101) | ||
|
||
for i in range(n_claims): | ||
addrs = sidechain.getpeginaddress() | ||
txid = bitcoin.sendtoaddress(addrs["mainchain_address"], 1) | ||
bitcoin.generate(12) | ||
proof = bitcoin.gettxoutproof([txid]) | ||
raw = bitcoin.getrawtransaction(txid) | ||
pegtxs += [sidechain.claimpegin(raw, proof)] | ||
|
||
sync_all(sidechain, sidechain2) | ||
|
||
sidechain2.generate(1) | ||
for pegtxid in pegtxs: | ||
tx = sidechain.gettransaction(pegtxid) | ||
if "confirmations" not in tx or tx["confirmations"] == 0: | ||
raise Exception("Peg-in confirmation has failed.") | ||
|
||
# TODO test reorgs, conflicting peg-ins | ||
|
||
print("Success!") | ||
|
||
except JSONRPCException as e: | ||
print("Pegging testing failed, aborting:") | ||
print(e.error) | ||
except Exception as e: | ||
print("Pegging testing failed, aborting:") | ||
print(e) | ||
|
||
print("Stopping daemons and cleaning up") | ||
bitcoin.stop() | ||
sidechain.stop() | ||
|
||
time.sleep(5) | ||
|
||
shutil.rmtree(sidechain_datadir) | ||
shutil.rmtree(bitcoin_datadir) |