|
|
|
#!/usr/bin/env python3
|
|
|
|
# -*- coding: utf-8 -*-
|
|
|
|
|
|
|
|
# Copyright (c) 2019 tecnovert
|
|
|
|
# Distributed under the MIT software license, see the accompanying
|
|
|
|
# file LICENSE.txt or http://www.opensource.org/licenses/mit-license.php.
|
|
|
|
|
|
|
|
"""
|
|
|
|
Atomic Swap Client - Proof of Concept
|
|
|
|
|
|
|
|
Dependencies:
|
|
|
|
$ pacman -S python-pyzmq python-plyvel protobuf
|
|
|
|
|
|
|
|
"""
|
|
|
|
|
|
|
|
import sys
|
|
|
|
import os
|
|
|
|
import time
|
|
|
|
import json
|
|
|
|
import traceback
|
|
|
|
import signal
|
|
|
|
import subprocess
|
|
|
|
import logging
|
|
|
|
|
|
|
|
import basicswap.config as cfg
|
|
|
|
from basicswap import __version__
|
|
|
|
from basicswap.basicswap import BasicSwap
|
|
|
|
from basicswap.http_server import HttpThread
|
|
|
|
|
|
|
|
|
|
|
|
logger = logging.getLogger()
|
|
|
|
logger.level = logging.DEBUG
|
|
|
|
if not len(logger.handlers):
|
|
|
|
logger.addHandler(logging.StreamHandler(sys.stdout))
|
|
|
|
|
|
|
|
ALLOW_CORS = False
|
|
|
|
swap_client = None
|
|
|
|
|
|
|
|
|
|
|
|
def signal_handler(sig, frame):
|
|
|
|
logger.info('Signal %d detected, ending program.' % (sig))
|
|
|
|
if swap_client is not None:
|
|
|
|
swap_client.stopRunning()
|
|
|
|
|
|
|
|
|
|
|
|
def startDaemon(node_dir, bin_dir, daemon_bin, opts=[]):
|
|
|
|
daemon_bin = os.path.expanduser(os.path.join(bin_dir, daemon_bin))
|
|
|
|
|
|
|
|
args = [daemon_bin, '-datadir=' + os.path.expanduser(node_dir)] + opts
|
|
|
|
logger.info('Starting node ' + daemon_bin + ' ' + '-datadir=' + node_dir)
|
|
|
|
return subprocess.Popen(args, stdin=subprocess.PIPE, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
|
|
|
|
|
|
|
|
|
|
|
|
def runClient(fp, dataDir, chain):
|
|
|
|
global swap_client
|
|
|
|
settings_path = os.path.join(dataDir, 'basicswap.json')
|
|
|
|
|
|
|
|
if not os.path.exists(settings_path):
|
|
|
|
raise ValueError('Settings file not found: ' + str(settings_path))
|
|
|
|
|
|
|
|
with open(settings_path) as fs:
|
|
|
|
settings = json.load(fs)
|
|
|
|
|
|
|
|
daemons = []
|
|
|
|
|
|
|
|
for c, v in settings['chainclients'].items():
|
|
|
|
if v['manage_daemon'] is True:
|
|
|
|
logger.info('Starting {} daemon'.format(c.capitalize()))
|
|
|
|
if c == 'particl':
|
|
|
|
daemons.append(startDaemon(v['datadir'], v['bindir'], cfg.PARTICLD))
|
|
|
|
logger.info('Started {} {}'.format(cfg.PARTICLD, daemons[-1].pid))
|
|
|
|
elif c == 'bitcoin':
|
|
|
|
daemons.append(startDaemon(v['datadir'], v['bindir'], cfg.BITCOIND))
|
|
|
|
logger.info('Started {} {}'.format(cfg.BITCOIND, daemons[-1].pid))
|
|
|
|
elif c == 'litecoin':
|
|
|
|
daemons.append(startDaemon(v['datadir'], v['bindir'], cfg.LITECOIND))
|
|
|
|
logger.info('Started {} {}'.format(cfg.LITECOIND, daemons[-1].pid))
|
|
|
|
else:
|
|
|
|
logger.warning('Unknown chain', c)
|
|
|
|
|
|
|
|
swap_client = BasicSwap(fp, dataDir, settings, chain)
|
|
|
|
|
|
|
|
signal.signal(signal.SIGINT, signal_handler)
|
|
|
|
signal.signal(signal.SIGTERM, signal_handler)
|
|
|
|
swap_client.start()
|
|
|
|
|
|
|
|
threads = []
|
|
|
|
if 'htmlhost' in settings:
|
|
|
|
swap_client.log.info('Starting server at %s:%d.' % (settings['htmlhost'], settings['htmlport']))
|
|
|
|
allow_cors = settings['allowcors'] if 'allowcors' in settings else ALLOW_CORS
|
|
|
|
tS1 = HttpThread(fp, settings['htmlhost'], settings['htmlport'], allow_cors, swap_client)
|
|
|
|
threads.append(tS1)
|
|
|
|
tS1.start()
|
|
|
|
|
|
|
|
try:
|
|
|
|
logger.info('Exit with Ctrl + c.')
|
|
|
|
while swap_client.is_running:
|
|
|
|
time.sleep(0.5)
|
|
|
|
swap_client.update()
|
|
|
|
except Exception:
|
|
|
|
traceback.print_exc()
|
|
|
|
|
|
|
|
swap_client.log.info('Stopping threads.')
|
|
|
|
for t in threads:
|
|
|
|
t.stop()
|
|
|
|
t.join()
|
|
|
|
|
|
|
|
for d in daemons:
|
|
|
|
logger.info('Terminating {}'.format(d.pid))
|
|
|
|
d.terminate()
|
|
|
|
d.wait(timeout=120)
|
|
|
|
if d.stdout:
|
|
|
|
d.stdout.close()
|
|
|
|
if d.stderr:
|
|
|
|
d.stderr.close()
|
|
|
|
if d.stdin:
|
|
|
|
d.stdin.close()
|
|
|
|
|
|
|
|
|
|
|
|
def printVersion():
|
|
|
|
logger.info('Basicswap version:', __version__)
|
|
|
|
|
|
|
|
|
|
|
|
def printHelp():
|
|
|
|
logger.info('basicswap-run --datadir=path -testnet')
|
|
|
|
|
|
|
|
|
|
|
|
def main():
|
|
|
|
data_dir = None
|
|
|
|
chain = 'mainnet'
|
|
|
|
|
|
|
|
for v in sys.argv[1:]:
|
|
|
|
if len(v) < 2 or v[0] != '-':
|
|
|
|
logger.warning('Unknown argument %s', v)
|
|
|
|
continue
|
|
|
|
|
|
|
|
s = v.split('=')
|
|
|
|
name = s[0].strip()
|
|
|
|
|
|
|
|
for i in range(2):
|
|
|
|
if name[0] == '-':
|
|
|
|
name = name[1:]
|
|
|
|
|
|
|
|
if name == 'v' or name == 'version':
|
|
|
|
printVersion()
|
|
|
|
return 0
|
|
|
|
if name == 'h' or name == 'help':
|
|
|
|
printHelp()
|
|
|
|
return 0
|
|
|
|
if name == 'testnet':
|
|
|
|
chain = 'testnet'
|
|
|
|
continue
|
|
|
|
if name == 'regtest':
|
|
|
|
chain = 'regtest'
|
|
|
|
continue
|
|
|
|
|
|
|
|
if len(s) == 2:
|
|
|
|
if name == 'datadir':
|
|
|
|
data_dir = os.path.expanduser(s[1])
|
|
|
|
continue
|
|
|
|
|
|
|
|
logger.warning('Unknown argument %s', v)
|
|
|
|
|
|
|
|
if data_dir is None:
|
|
|
|
default_datadir = '~/.basicswap'
|
|
|
|
data_dir = os.path.join(os.path.expanduser(default_datadir))
|
|
|
|
logger.info('Using datadir: %s', data_dir)
|
|
|
|
logger.info('Chain: %s', chain)
|
|
|
|
|
|
|
|
if not os.path.exists(data_dir):
|
|
|
|
os.makedirs(data_dir)
|
|
|
|
|
|
|
|
with open(os.path.join(data_dir, 'basicswap.log'), 'a') as fp:
|
|
|
|
logger.info(os.path.basename(sys.argv[0]) + ', version: ' + __version__ + '\n\n')
|
|
|
|
runClient(fp, data_dir, chain)
|
|
|
|
|
|
|
|
logger.info('Done.')
|
|
|
|
return swap_client.fail_code if swap_client is not None else 0
|
|
|
|
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
|
|
main()
|