ts_misc.py 5.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162
  1. #!/usr/bin/env python3
  2. #
  3. # mmgen = Multi-Mode GENerator, command-line Bitcoin cold storage solution
  4. # Copyright (C)2013-2019 The MMGen Project <mmgen@tuta.io>
  5. #
  6. # This program is free software: you can redistribute it and/or modify
  7. # it under the terms of the GNU General Public License as published by
  8. # the Free Software Foundation, either version 3 of the License, or
  9. # (at your option) any later version.
  10. #
  11. # This program is distributed in the hope that it will be useful,
  12. # but WITHOUT ANY WARRANTY; without even the implied warranty of
  13. # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  14. # GNU General Public License for more details.
  15. #
  16. # You should have received a copy of the GNU General Public License
  17. # along with this program. If not, see <http://www.gnu.org/licenses/>.
  18. """
  19. ts_misc.py: Miscellaneous test groups for the test.py test suite
  20. """
  21. from mmgen.globalvars import g
  22. from test.common import *
  23. from test.test_py_d.common import *
  24. from test.test_py_d.ts_base import *
  25. from test.test_py_d.ts_main import TestSuiteMain
  26. class TestSuiteHelp(TestSuiteBase):
  27. 'help and usage screens'
  28. tmpdir_nums = []
  29. passthru_opts = ('coin','testnet')
  30. cmd_group = (
  31. ('helpscreens', (1,'help screens', [])),
  32. ('longhelpscreens', (1,'help screens (--longhelp)',[])),
  33. ('tool_help', (1,"'mmgen-tool' usage screen",[])),
  34. ('test_help', (1,"'test.py' help screens",[])),
  35. )
  36. def helpscreens(self,arg='--help'):
  37. scripts = (
  38. 'walletgen','walletconv','walletchk','txcreate','txsign','txsend','txdo','txbump',
  39. 'addrgen','addrimport','keygen','passchg','tool','passgen','regtest','autosign')
  40. for s in scripts:
  41. t = self._run_cmd('mmgen-'+s,[arg],extra_desc='(mmgen-{})'.format(s),no_output=True)
  42. return t
  43. def longhelpscreens(self):
  44. return self.helpscreens(arg='--longhelp')
  45. def _run_cmd( self, cmd_name,
  46. cmd_args = [],
  47. no_msg = False,
  48. extra_desc = '',
  49. cmd_dir = 'cmds',
  50. no_output = False):
  51. t = self.spawn( cmd_name,
  52. args = cmd_args,
  53. no_msg = no_msg,
  54. extra_desc = extra_desc,
  55. cmd_dir = cmd_dir,
  56. no_output = no_output)
  57. t.read()
  58. ret = t.p.wait()
  59. if ret == 0:
  60. msg('OK')
  61. else:
  62. rdie(1,"\n'{}' returned {}".format(self.test_name,ret))
  63. t.skip_ok = True
  64. return t
  65. def tool_help(self):
  66. self._run_cmd('mmgen-tool',['help'],extra_desc="('mmgen-tool help')")
  67. return self._run_cmd('mmgen-tool',['usage'],extra_desc="('mmgen-tool usage')")
  68. def test_help(self):
  69. self._run_cmd('test.py',['-h'],cmd_dir='test')
  70. self._run_cmd('test.py',['-L'],cmd_dir='test',extra_desc='(cmd group list)')
  71. return self._run_cmd('test.py',['-l'],cmd_dir='test',extra_desc='(cmd list)')
  72. class TestSuiteTool(TestSuiteMain,TestSuiteBase):
  73. "tests for interactive 'mmgen-tool' commands"
  74. networks = ('btc',)
  75. segwit_opts_ok = False
  76. tmpdir_nums = [9]
  77. enc_infn = 'tool_encrypt.in'
  78. cmd_group = (
  79. ('tool_find_incog_data', (9,"'mmgen-tool find_incog_data'", [[[hincog_fn],1],[[incog_id_fn],1]])),
  80. ('tool_encrypt', (9,"'mmgen-tool encrypt' (random data)", [])),
  81. ('tool_decrypt', (9,"'mmgen-tool decrypt' (random data)", [[[enc_infn+'.mmenc'],9]])),
  82. # ('tool_encrypt_ref', (9,"'mmgen-tool encrypt' (reference text)", [])),
  83. )
  84. def tool_encrypt(self):
  85. infile = joinpath(self.tmpdir,self.enc_infn)
  86. write_to_file(infile,os.urandom(1033),binary=True)
  87. t = self.spawn('mmgen-tool',['-d',self.tmpdir,self.usr_rand_arg,'encrypt',infile])
  88. t.usr_rand(self.usr_rand_chars)
  89. t.hash_preset('user data','1')
  90. t.passphrase_new('user data',tool_enc_passwd)
  91. t.written_to_file('Encrypted data')
  92. return t
  93. def tool_decrypt(self,f1):
  94. out_fn = 'tool_encrypt.out'
  95. t = self.spawn('mmgen-tool',['-d',self.tmpdir,'decrypt',f1,'outfile='+out_fn,'hash_preset=1'])
  96. t.passphrase('user data',tool_enc_passwd)
  97. t.written_to_file('Decrypted data')
  98. d1 = self.read_from_tmpfile(self.enc_infn,binary=True)
  99. d2 = self.read_from_tmpfile(out_fn,binary=True)
  100. cmp_or_die(d1,d2)
  101. return t
  102. def tool_find_incog_data(self,f1,f2):
  103. i_id = read_from_file(f2).rstrip()
  104. vmsg('Incog ID: {}'.format(cyan(i_id)))
  105. t = self.spawn('mmgen-tool',['-d',self.tmpdir,'find_incog_data',f1,i_id])
  106. o = t.expect_getend('Incog data for ID {} found at offset '.format(i_id))
  107. os.unlink(f1)
  108. cmp_or_die(hincog_offset,int(o))
  109. return t
  110. class TestSuiteRefTX(TestSuiteMain,TestSuiteBase):
  111. 'create a reference transaction file (administrative command)'
  112. segwit_opts_ok = False
  113. passthru_opts = ('coin','testnet')
  114. tmpdir_nums = [31,32,33,34]
  115. cmd_group = (
  116. ('ref_tx_addrgen1', (31,'address generation (legacy)', [[[],1]])),
  117. ('ref_tx_addrgen2', (32,'address generation (compressed)', [[[],1]])),
  118. ('ref_tx_addrgen3', (33,'address generation (segwit)', [[[],1]])),
  119. ('ref_tx_addrgen4', (34,'address generation (bech32)', [[[],1]])),
  120. ('ref_tx_txcreate', (31,'transaction creation',
  121. ([['addrs'],31],[['addrs'],32],[['addrs'],33],[['addrs'],34]))),
  122. )
  123. def __init__(self,trunner,cfgs,spawn):
  124. for n in self.tmpdir_nums:
  125. cfgs[str(n)].update({ 'addr_idx_list': '1-2',
  126. 'segwit': n in (33,34),
  127. 'dep_generators': { 'addrs':'ref_tx_addrgen'+str(n)[-1] }})
  128. return TestSuiteMain.__init__(self,trunner,cfgs,spawn)
  129. def ref_tx_addrgen(self,atype):
  130. if atype not in g.proto.mmtypes: return
  131. t = self.spawn('mmgen-addrgen',['--outdir='+self.tmpdir,'--type='+atype,dfl_words_file,'1-2'])
  132. t.read()
  133. return t
  134. def ref_tx_addrgen1(self): return self.ref_tx_addrgen(atype='L')
  135. def ref_tx_addrgen2(self): return self.ref_tx_addrgen(atype='C')
  136. def ref_tx_addrgen3(self): return self.ref_tx_addrgen(atype='S')
  137. def ref_tx_addrgen4(self): return self.ref_tx_addrgen(atype='B')
  138. def ref_tx_txcreate(self,f1,f2,f3,f4):
  139. sources = ['31','32']
  140. if 'S' in g.proto.mmtypes: sources += ['33']
  141. if 'B' in g.proto.mmtypes: sources += ['34']
  142. return self.txcreate_common(
  143. addrs_per_wallet = 2,
  144. sources = sources,
  145. add_args = ['--locktime=1320969600'],
  146. do_label = True )