From 8c1df23332dd70e443f2c53ca820af931bb9e6cc Mon Sep 17 00:00:00 2001 From: MMGen Date: Tue, 12 Feb 2019 10:41:27 +0000 Subject: [PATCH] py3port: 'test.py autosign_live' - test device insertion/removal + LED --- mmgen/main_autosign.py | 3 +- scripts/test-release.sh | 9 ++- test/test.py | 143 +++++++++++++++++++++++++++++++++++----- 3 files changed, 136 insertions(+), 19 deletions(-) diff --git a/mmgen/main_autosign.py b/mmgen/main_autosign.py index 8f2b2d25..6be23403 100755 --- a/mmgen/main_autosign.py +++ b/mmgen/main_autosign.py @@ -38,6 +38,7 @@ opts_data = lambda: { -h, --help Print this help message --, --longhelp Print help message for long options (common options) -c, --coins=c Coins to sign for (comma-separated list) +-I, --no-insert-check Don't check for device insertion -l, --led Use status LED to signal standby, busy and error -m, --mountpoint=m Specify an alternate mountpoint (default: '{mp}') -s, --stealth-led Stealth LED mode - signal busy and error only, and only @@ -349,7 +350,7 @@ def set_led(cmd): led_thread.start() def get_insert_status(): - if os.getenv('MMGEN_TEST_SUITE'): return True + if opt.no_insert_check: return True try: os.stat(os.path.join('/dev/disk/by-label',part_label)) except: return False else: return True diff --git a/scripts/test-release.sh b/scripts/test-release.sh index 67e3a643..4d625847 100755 --- a/scripts/test-release.sh +++ b/scripts/test-release.sh @@ -19,7 +19,7 @@ rounds=100 rounds_low=20 rounds_spec=500 gen_rounds=10 monero_addrs='3,99,2,22-24,101-104' dfl_tests='obj sha256 alts monero eth autosign btc btc_tn btc_rt bch bch_rt ltc ltc_tn ltc_rt tool gen' -add_tests='autosign_minimal' +add_tests='autosign_minimal autosign_live' PROGNAME=$(basename $0) while getopts hbCfilnPRtvV OPT @@ -244,6 +244,13 @@ s_autosign_minimal='The bitcoin mainnet and testnet daemons must be running for t_autosign_minimal=("$test_py autosign_minimal") f_autosign_minimal='Autosign Minimal test complete' +i_autosign_live='Autosign Live' +s_autosign_live="The bitcoin mainnet and testnet daemons must be running for the following test\n" +s_autosign_live+="${YELLOW}Mountpoint, '/etc/fstab' and removable device must be configured " +s_autosign_live+="as described in 'mmgen-autosign --help'${RESET}" +t_autosign_live=("$test_py autosign_live") +f_autosign_live='Autosign Live test complete' + i_btc='Bitcoin mainnet' s_btc='The bitcoin (mainnet) daemon must both be running for the following tests' t_btc=( diff --git a/test/test.py b/test/test.py index b65fdc6c..cab8b979 100755 --- a/test/test.py +++ b/test/test.py @@ -1093,6 +1093,9 @@ cmd_group['autosign'] = ( cmd_group['autosign_minimal'] = ( ('autosign_minimal', 'transaction autosigning (BTC,ETH,ETC)'), ) +cmd_group['autosign_live'] = ( + ('autosign_live', 'transaction autosigning (BTC,ETH,ETC - test device insertion/removal + LED)'), +) cmd_group['ref_alt'] = ( ('ref_addrfile_gen_eth', 'generate address file (ETH)'), @@ -1208,6 +1211,11 @@ for a,b in cmd_group['autosign_minimal']: cmd_list['autosign_minimal'].append(a) cmd_data[a] = (18,b,[[[],18]]) +cmd_data['info_autosign_live'] = 'autosign_live',[18] +for a,b in cmd_group['autosign_live']: + cmd_list['autosign_live'].append(a) + cmd_data[a] = (18,b,[[[],18]]) + cmd_data['info_ref_alt'] = 'altcoin reference files',[8] for a,b in cmd_group['ref_alt']: cmd_list['ref_alt'].append(a) @@ -1271,6 +1279,9 @@ os.environ['MMGEN_TEST_SUITE'] = '1' def get_segwit_arg(cfg): return ['--type='+('segwit','bech32')[bool(opt.bech32)]] if cfg['segwit'] else [] +def omsg(s): sys.stderr.write(s+'\n') +def omsg_r(s): sys.stderr.write(s) + if opt.exact_output: def imsg(s): os.write(2,s.encode() + b'\n') def imsg_r(s): os.write(2,s.encode()) @@ -2439,26 +2450,26 @@ class MMGenTestSuite(object): os.unlink(f1) cmp_or_die(hincog_offset,int(o)) - def autosign_minimal(self,name): + def autosign_live(self,name): + return self.autosign_minimal(name,live=True) + + def autosign_minimal(self,name,live=False): return self.autosign(name, coins=['btc','eth'], txfiles=['btc','eth','erc20','etc'], - txcount=7) + txcount=7, + live=live) # tests everything except device detection, mount/unmount def autosign( self,name, coins=['btc','bch','ltc','eth'], txfiles=['btc','bch','ltc','eth','erc20','etc'], - txcount=11): - - opts = ['--mountpoint='+cfg['tmpdir'],'--coins='+','.join(coins)] + txcount=11, + live=False): if skip_for_win(): return - try: os.mkdir(os.path.join(cfg['tmpdir'],'tx')) - except: pass - - def make_wallet(): + def make_wallet(opts): t = MMGenExpect(name,'mmgen-autosign',opts+['gen_key'],extra_desc='(gen_key)') t.expect_getend('Wrote key file ') t.ok() @@ -2495,7 +2506,7 @@ class MMGenTestSuite(object): wf = t.written_to_file('Autosign wallet') t.ok() - def copy_files(): + def copy_files(mountpoint,remove_signed_only=False,include_bad_tx=True): fdata_in = (('btc',''), ('bch',''), ('ltc','litecoin'), @@ -2509,13 +2520,71 @@ class MMGenTestSuite(object): for f,fn in zip(tfs,tfns): if fn: # use empty fn to skip file - shutil.copyfile(f,os.path.join(cfg['tmpdir'],'tx',fn)) + target = os.path.join(mountpoint,'tx',fn) + remove_signed_only or shutil.copyfile(f,target) + try: os.unlink(target.replace('.rawtx','.sigtx')) + except: pass # make a bad tx file - with open(os.path.join(cfg['tmpdir'],'tx','bad.rawtx'),'w') as f: - f.write('bad tx data') + bad_tx = os.path.join(mountpoint,'tx','bad.rawtx') + if include_bad_tx and not remove_signed_only: + with open(bad_tx,'w') as f: + f.write('bad tx data') + if not include_bad_tx: + try: os.unlink(bad_tx) + except: pass - def do_autosign(): + def do_autosign_live(opts,mountpoint,led_opts=[],gen_wallet=True): + + def do_mount(): + try: subprocess.check_call(['mount',mountpoint]) + except: pass + + def do_unmount(): + try: subprocess.check_call(['umount',mountpoint]) + except: pass + omsg_r(blue('\nRemove removable device and then hit ENTER ')) + input() + + if gen_wallet: make_wallet(opts) + else: do_mount() + + copy_files(mountpoint,include_bad_tx=not led_opts) + + desc = '(sign)' + m1 = "Running 'mmgen-autosign wait'" + m2 = 'Insert removable device ' + + if led_opts: + if led_opts == ['--led']: + m1 = "Running 'mmgen-autosign wait' with --led. The LED should start blinking slowly now" + elif led_opts == ['--stealth-led']: + m1 = "Running 'mmgen-autosign wait' with --stealth-led. You should see no LED activity now" + m2 = 'Insert removable device and watch for fast LED activity during signing' + desc = '(sign - {})'.format(led_opts[0]) + + def do_loop(): + omsg(blue(m2)) + t.expect('{} transactions signed'.format(txcount)) + if not led_opts: + t.expect('1 transaction failed to sign') + t.expect('Waiting') + + do_unmount() + omsg(green(m1)) + t = MMGenExpect(name,'mmgen-autosign',opts+led_opts+['wait'],extra_desc=desc) + if not opt.exact_output: omsg('') + do_loop() + do_mount() # race condition due to device insertion detection + copy_files(mountpoint,remove_signed_only=True,include_bad_tx=not led_opts) + do_unmount() + do_loop() + t.kill(2) + t.ok(exit_val=1) + + def do_autosign(opts,mountpoint): + make_wallet(opts) + copy_files(mountpoint,include_bad_tx=True) t = MMGenExpect(name,'mmgen-autosign',opts+['wait'],extra_desc='(sign)') t.expect('{} transactions signed'.format(txcount)) t.expect('1 transaction failed to sign') @@ -2523,9 +2592,49 @@ class MMGenTestSuite(object): t.kill(2) t.ok(exit_val=1) - make_wallet() - copy_files() - do_autosign() + if live: + mountpoint = '/mnt/tx' + if not os.path.ismount(mountpoint): + try: + subprocess.check_call(['mount',mountpoint]) + imsg("Mounted '{}'".format(mountpoint)) + except: + imsg("Could not mount '{}'! Exiting".format(mountpoint)) + return + + txdir = os.path.join(mountpoint,'tx') + if not os.path.isdir(txdir): + imsg("Directory '{}' does not exist! Exiting".format(mountpoint)) + return + + opts = ['--coins='+','.join(coins)] + led_files = { 'opi': ('/sys/class/leds/orangepi:red:status/brightness',), + 'rpi': ('/sys/class/leds/led0/brightness','/sys/class/leds/led0/trigger') } + for k in ('opi','rpi'): + if os.path.exists(led_files[k][0]): + led_support = k + break + else: + led_support = None + + if led_support: + for fn in (led_files[led_support]): + subprocess.check_call(['sudo','chmod','0666',fn]) + omsg(purple('Running autosign test with no LED')) + do_autosign_live(opts,mountpoint) + omsg(purple("Running autosign test with '--led'")) + do_autosign_live(opts,mountpoint,led_opts=['--led'],gen_wallet=False) + omsg(purple("Running autosign test with '--stealth-led'")) + do_autosign_live(opts,mountpoint,led_opts=['--stealth-led'],gen_wallet=False) + else: + do_autosign_live(opts,mountpoint) + else: + mountpoint = cfg['tmpdir'] + opts = ['--no-insert-check','--mountpoint='+mountpoint,'--coins='+','.join(coins)] + try: os.mkdir(os.path.join(mountpoint,'tx')) + except: pass + do_autosign(opts,mountpoint) + # Saved reference file tests def ref_wallet_conv(self,name):