151 lines
8.1 KiB
Python
Executable file
151 lines
8.1 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
#
|
|
# mmgen = Multi-Mode GENerator, command-line Bitcoin cold storage solution
|
|
# Copyright (C)2013-2020 The MMGen Project <mmgen@tuta.io>
|
|
#
|
|
# This program is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
"""
|
|
test/scrambletest.py: seed scrambling and addrlist data generation tests for all
|
|
supported coins + passwords
|
|
"""
|
|
|
|
import sys,os
|
|
from subprocess import run,PIPE
|
|
|
|
from include.tests_header import repo_root
|
|
from mmgen.common import *
|
|
from test.include.common import *
|
|
|
|
opts_data = {
|
|
'text': {
|
|
'desc': 'Test seed scrambling and addrlist data generation for all supported altcoins',
|
|
'usage':'[options] [command]',
|
|
'options': """
|
|
-h, --help Print this help message
|
|
--, --longhelp Print help message for long options (common options)
|
|
-C, --coverage Produce code coverage info using trace module
|
|
-l, --list-cmds List and describe the tests and commands in this test suite
|
|
-s, --system Test scripts and modules installed on system rather than
|
|
those in the repo root
|
|
-v, --verbose Produce more verbose output
|
|
""",
|
|
'notes': """
|
|
Valid commands: 'coin','pw'
|
|
If no command is given, the whole suite of tests is run.
|
|
"""
|
|
}
|
|
}
|
|
|
|
cmd_args = opts.init(opts_data)
|
|
|
|
os.environ['MMGEN_DEBUG_ADDRLIST'] = '1'
|
|
if not opt.system:
|
|
os.environ['PYTHONPATH'] = repo_root
|
|
|
|
from collections import namedtuple
|
|
td = namedtuple('scrambletest_entry',['seed','str','id_str','lbl','addr'])
|
|
|
|
coin_data = {
|
|
# SCRAMBLED_SEED[:8] SCRAMBLE_KEY ID_STR LBL FIRST ADDR
|
|
'btc': td('456d7f5f1c4bfe3b','(none)', '', '', '1MU7EdgqYy9JX35L25hR6CmXXcSEBDAwyv'),
|
|
'btc_compressed':td('bf98a4af5464a4ef','compressed', '-C', 'COMPRESSED', '1F97Jd89wwmu4ELadesAdGDzg3d8Y6j5iP'),
|
|
'btc_segwit': td('b56962d829ffc678','segwit', '-S', 'SEGWIT', '36TvVzU5mxSjJ3D9qKAmYzCV7iUqtTDezF'),
|
|
'btc_bech32': td('d09eea818f9ad17f','bech32', '-B', 'BECH32', 'bc1q8snv94j6959y3gkqv4gku0cm5mersnpucsvw5z'),
|
|
'bch': td('456d7f5f1c4bfe3b','(none)', '', '', '1MU7EdgqYy9JX35L25hR6CmXXcSEBDAwyv'),
|
|
'bch_compressed':td('bf98a4af5464a4ef','compressed', '-C', 'COMPRESSED', '1F97Jd89wwmu4ELadesAdGDzg3d8Y6j5iP'),
|
|
'ltc': td('b11f16632e63ba92','ltc:legacy', '-LTC', 'LTC', 'LMxB474SVfxeYdqxNrM1WZDZMnifteSMv1'),
|
|
'ltc_compressed':td('7ccf465d466ee7d3','ltc:compressed','-LTC-C','LTC:COMPRESSED','LdkebBKVXSs6NNoPJWGM8KciDnL8LhXXjb'),
|
|
'ltc_segwit': td('9460f5ba15e82768','ltc:segwit', '-LTC-S','LTC:SEGWIT', 'MQrY3vEbqKMBgegXrSaR93R2HoTDE5bKrY'),
|
|
'ltc_bech32': td('dbdbff2e196e27d3','ltc:bech32', '-LTC-B','LTC:BECH32', 'ltc1qdvgqsz94ht20lr8fyk5v7n884hu9p7d8k9easu'),
|
|
'eth': td('213ed116869b19f2','eth', '-ETH', 'ETH', 'e704b6cfd9f0edb2e6cfbd0c913438d37ede7b35'),
|
|
'etc': td('909def37096f5ab8','etc', '-ETC', 'ETC', '1a6acbef8c38f52f20d04ecded2992b04d8608d7'),
|
|
'dash': td('1319d347b021f952','dash:legacy', '-DASH', 'DASH', 'XoK491fppGNZQUUS9uEFkT6L9u8xxVFJNJ'),
|
|
'emc': td('7e1a29853d2db875','emc:legacy', '-EMC', 'EMC', 'EU4L6x2b5QUb2gRQsBAAuB8TuPEwUxCNZU'),
|
|
'zec': td('0bf9b5b20af7b5a0','zec:legacy', '-ZEC', 'ZEC', 't1URz8BHxV38v3gsaN6oHQNKC16s35R9WkY'),
|
|
'zec_zcash_z': td('b15570d033df9b1a','zec:zcash_z', '-ZEC-Z','ZEC:ZCASH_Z', 'zcLMMsnzfFYZWU4avRBnuc83yh4jTtJXbtP32uWrs3ickzu1krMU4ppZCQPTwwfE9hLnRuFDSYF8VFW13aT9eeQK8aov3Ge'),
|
|
'xmr': td('c76af3b088da3364','xmr:monero', '-XMR-M','XMR:MONERO', '41tmwZd2CdXEGtWqGY9fH9FVtQM8VxZASYPQ3VJQhFjtGWYzQFuidD21vJYTi2yy3tXRYXTNXBTaYVLav62rwUUpFFyicZU'),
|
|
}
|
|
|
|
passwd_data = {
|
|
'dfl_dfl_αω': td('b78c363f3d6714f6', 'b58:20:αω', '-αω-b58-20', 'αω b58:20', 'L3JcAZze6LQS2TFodoW9'),
|
|
'dfl_H_αω': td('2cba1ba1a73d5497', 'b58:10:αω', '-αω-b58-10', 'αω b58:10', 'KgSBfzPBWj'),
|
|
'b32_dfl_αω': td('ad71ff8d0512660d', 'b32:24:αω', '-αω-b32-24', 'αω b32:24', 'VQFESDGWAQCOCSG6GDLIG5OQ'),
|
|
'b32_H_αω': td('b050a1613dd4d3ad', 'b32:12:αω', '-αω-b32-12', 'αω b32:12', 'QTI7HTVN3JOE'),
|
|
'hex_dfl_αω': td('8322b26abd931d55', 'hex:64:αω', '-αω-hex-64', 'αω hex:64', '6e9f8d5d2ec6c4f2a079b163e37f5dc2a7e6adb221c4de315078ffad971ba260'),
|
|
'hex_H_αω': td('46beb852d38ed5c4', 'hex:32:αω', '-αω-hex-32', 'αω hex:32', '86143b36b649728cc8f3677872ed37d3'),
|
|
'bip39_dfl_αω':td('95b383d5092a55df', 'bip39:24:αω','-αω-bip39-24','αω bip39:24','treat athlete brand top beauty poverty senior unhappy vacant domain yellow scale fossil aim lonely fatal sun nuclear such ancient stage require stool similar'),
|
|
'bip39_18_αω': td('29e5a605ffa36142', 'bip39:18:αω','-αω-bip39-18','αω bip39:18','better legal various ketchup then range festival either tomato cradle say absorb solar earth alter pattern canyon liar'),
|
|
'bip39_12_αω': td('efa13cb309d7fc1d', 'bip39:12:αω','-αω-bip39-12','αω bip39:12','lady oppose theme fit position merry reopen acquire tuna dentist young chunk'),
|
|
'xmrseed_dfl_αω':td('62f5b72a5ca89cab', 'xmrseed:25:αω','-αω-xmrseed-25','αω xmrseed:25','tequila eden skulls giving jester hospital dreams bakery adjust nanny cactus inwardly films amply nanny soggy vials muppet yellow woken ashtray organs exhale foes eden'),
|
|
}
|
|
|
|
cvr_opts = ' -m trace --count --coverdir={} --file={}'.format(*init_coverage()) if opt.coverage else ''
|
|
cmd_base = 'python3{} cmds/mmgen-{{}}gen -qS'.format(cvr_opts)
|
|
|
|
def get_cmd_output(cmd):
|
|
cp = run(cmd.split(),stdout=PIPE,stderr=PIPE)
|
|
if cp.returncode != 0:
|
|
ydie(2,'\nSpawned program exited with error code {}:\n{}'.format(cp.returncode,cp.stderr.decode()))
|
|
return cp.stdout.decode().splitlines()
|
|
|
|
def do_test(cmd,tdata,msg_str,addr_desc):
|
|
vmsg(green('Executing: {}'.format(cmd)))
|
|
msg_r('Testing: ' + msg_str)
|
|
|
|
lines = get_cmd_output(cmd)
|
|
cmd_out = dict([e[9:].split(': ') for e in lines if e.startswith('sc_debug_')])
|
|
cmd_out['addr'] = lines[-2].split(None,1)[-1]
|
|
|
|
ref_data = tdata._asdict()
|
|
vmsg('')
|
|
for k in ref_data:
|
|
if cmd_out[k] == ref_data[k]:
|
|
s = k.replace('seed','seed[:8]').replace('addr',addr_desc)
|
|
vmsg(' {:9}: {}'.format(s,cmd_out[k]))
|
|
else:
|
|
rdie(1,'\nError: sc_{} value {} does not match reference value {}'.format(k,cmd_out[k],ref_data[k]))
|
|
msg('OK')
|
|
|
|
def do_coin_tests():
|
|
bmsg('Testing address scramble strings and list IDs')
|
|
for tname,tdata in coin_data.items():
|
|
if tname == 'zec_zcash_z' and g.platform == 'win':
|
|
msg("Skipping 'zec_zcash_z' test for Windows platform")
|
|
continue
|
|
coin,mmtype = tname.split('_',1) if '_' in tname else (tname,None)
|
|
type_arg = ' --type='+mmtype if mmtype else ''
|
|
cmd = cmd_base.format('addr') + ' --coin={}{} test/ref/98831F3A.mmwords 1'.format(coin,type_arg)
|
|
do_test(cmd,tdata,'--coin {:4} {:22}'.format(coin.upper(),type_arg),'address')
|
|
|
|
def do_passwd_tests():
|
|
bmsg('Testing password scramble strings and list IDs')
|
|
for tname,tdata in passwd_data.items():
|
|
a,b,pwid = tname.split('_')
|
|
fmt_arg = '' if a == 'dfl' else '--passwd-fmt={} '.format(a)
|
|
len_arg = '' if b == 'dfl' else '--passwd-len={} '.format(b)
|
|
fs = '{}' + fmt_arg + len_arg + '{}' + pwid + ' 1'
|
|
cmd = cmd_base.format('pass') + ' ' + fs.format('--accept-defaults ','test/ref/98831F3A.mmwords ')
|
|
s = fs.format('','')
|
|
do_test(cmd,tdata,s+' '*(40-len(s)),'password')
|
|
|
|
start_time = int(time.time())
|
|
|
|
cmds = cmd_args or ('coin','pw')
|
|
for cmd in cmds:
|
|
{'coin': do_coin_tests, 'pw': do_passwd_tests }[cmd]()
|
|
|
|
t = int(time.time()) - start_time
|
|
m = 'All requested tests finished OK, elapsed time: {:02}:{:02}'
|
|
gmsg(m.format(t//60,t%60))
|