ts_input.py 9.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268
  1. #!/usr/bin/env python3
  2. #
  3. # mmgen = Multi-Mode GENerator, command-line Bitcoin cold storage solution
  4. # Copyright (C)2013-2020 The MMGen Project <mmgen@tuta.io>
  5. #
  6. # Project source code repository: https://github.com/mmgen/mmgen
  7. # Licensed according to the terms of GPL Version 3. See LICENSE for details.
  8. """
  9. ts_input.py: user input tests for the MMGen test.py test suite
  10. """
  11. from ..include.common import *
  12. from .ts_base import *
  13. from .input import *
  14. from mmgen.wallet import Wallet
  15. class TestSuiteInput(TestSuiteBase):
  16. 'user input'
  17. networks = ('btc',)
  18. tmpdir_nums = []
  19. cmd_group = (
  20. ('get_passphrase_ui', (1,"hash preset, password and label (wallet.py)", [])),
  21. ('get_passphrase_cmdline', (1,"hash preset, password and label (wallet.py - from cmdline)", [])),
  22. ('get_passphrase_crypto', (1,"hash preset, password and label (crypto.py)", [])),
  23. ('password_entry_noecho', (1,"utf8 password entry", [])),
  24. ('password_entry_echo', (1,"utf8 password entry (echoed)", [])),
  25. ('mnemonic_entry_mmgen', (1,"stealth mnemonic entry (mmgen)", [])),
  26. ('mnemonic_entry_mmgen_minimal', (1,"stealth mnemonic entry (mmgen - minimal entry mode)", [])),
  27. ('mnemonic_entry_bip39', (1,"stealth mnemonic entry (bip39)", [])),
  28. ('mnemonic_entry_bip39_short', (1,"stealth mnemonic entry (bip39 - short entry mode)", [])),
  29. ('mn2hex_interactive_mmgen', (1,"mn2hex_interactive (mmgen)", [])),
  30. ('mn2hex_interactive_mmgen_fixed',(1,"mn2hex_interactive (mmgen - fixed (10-letter) entry mode)", [])),
  31. ('mn2hex_interactive_bip39', (1,"mn2hex_interactive (bip39)", [])),
  32. ('mn2hex_interactive_bip39_short',(1,"mn2hex_interactive (bip39 - short entry mode (+pad entry))", [])),
  33. ('mn2hex_interactive_bip39_fixed',(1,"mn2hex_interactive (bip39 - fixed (4-letter) entry mode)", [])),
  34. ('mn2hex_interactive_xmr', (1,"mn2hex_interactive (xmrseed)", [])),
  35. ('mn2hex_interactive_xmr_short', (1,"mn2hex_interactive (xmrseed - short entry mode)", [])),
  36. ('dieroll_entry', (1,"dieroll entry (base6d)", [])),
  37. ('dieroll_entry_usrrand', (1,"dieroll entry (base6d) with added user entropy", [])),
  38. )
  39. def get_passphrase_ui(self):
  40. t = self.spawn('test/misc/get_passphrase.py',['--usr-randchars=0','seed'],cmd_dir='.')
  41. # 1 - new wallet, default hp,label;empty pw
  42. t.expect('accept the default.*: ','\n',regex=True)
  43. # bad repeat
  44. t.expect('new MMGen wallet: ','pass1\n')
  45. t.expect('peat passphrase: ','pass2\n')
  46. # good repeat
  47. t.expect('new MMGen wallet: ','\n')
  48. t.expect('peat passphrase: ','\n')
  49. t.expect('mpty pass')
  50. t.expect('no label: ','\n')
  51. t.expect('[][3][No Label]')
  52. # 2 - new wallet, user-selected hp,pw,label
  53. t.expect('accept the default.*: ', '1\n', regex=True)
  54. t.expect('new MMGen wallet: ','pass1\n')
  55. t.expect('peat passphrase: ','pass1\n')
  56. t.expect('no label: ','lbl1\n')
  57. t.expect('[pass1][1][lbl1]')
  58. # 3 - passchg, nothing changes
  59. t.expect('new hash preset')
  60. t.expect('reuse the old value.*: ','\n',regex=True)
  61. t.expect('unchanged')
  62. t.expect('new passphrase.*: ','pass1\n',regex=True)
  63. t.expect('peat passphrase: ','pass1\n')
  64. t.expect('unchanged')
  65. t.expect('reuse the label .*: ','\n',regex=True)
  66. t.expect('unchanged')
  67. t.expect('[pass1][1][lbl1]')
  68. # 4 - passchg, everything changes
  69. t.expect('new hash preset')
  70. t.expect('reuse the old value.*: ','2\n',regex=True)
  71. t.expect(' changed to')
  72. t.expect('new passphrase.*: ','pass2\n',regex=True)
  73. t.expect('peat passphrase: ','pass2\n')
  74. t.expect(' changed')
  75. t.expect('reuse the label .*: ','lbl2\n',regex=True)
  76. t.expect(' changed to')
  77. t.expect('[pass2][2][lbl2]')
  78. # 5 - wallet from file
  79. t.expect('from file')
  80. # bad passphrase
  81. t.expect('passphrase for MMGen wallet: ','bad\n')
  82. t.expect('Trying again')
  83. # good passphrase
  84. t.expect('passphrase for MMGen wallet: ','reference password\n')
  85. t.expect('[reference password][1][No Label]')
  86. t.read()
  87. return t
  88. def get_passphrase_cmdline(self):
  89. open('test/trash/pwfile','w').write('reference password\n')
  90. t = self.spawn('test/misc/get_passphrase.py', [
  91. '--usr-randchars=0',
  92. '--label=MyLabel',
  93. '--passwd-file=test/trash/pwfile',
  94. '--hash-preset=1',
  95. 'seed' ],
  96. cmd_dir = '.' )
  97. for foo in range(4):
  98. t.expect('[reference password][1][MyLabel]')
  99. t.read()
  100. return t
  101. def get_passphrase_crypto(self):
  102. t = self.spawn('test/misc/get_passphrase.py',['--usr-randchars=0','crypto'],cmd_dir='.')
  103. # new passwd
  104. t.expect('passphrase for .*: ', 'x\n', regex=True)
  105. t.expect('peat passphrase: ', '\n')
  106. t.expect('passphrase for .*: ', 'pass1\n', regex=True)
  107. t.expect('peat passphrase: ', 'pass1\n')
  108. t.expect('[pass1]')
  109. # existing passwd
  110. t.expect('passphrase for .*: ', 'pass2\n', regex=True)
  111. t.expect('[pass2]')
  112. # hash preset
  113. t.expect('accept the default .*: ', '0\n', regex=True)
  114. t.expect('nvalid')
  115. t.expect('accept the default .*: ', '8\n', regex=True)
  116. t.expect('nvalid')
  117. t.expect('accept the default .*: ', '7\n', regex=True)
  118. t.expect('[7]')
  119. # hash preset (default)
  120. t.expect('accept the default .*: ', '\n', regex=True)
  121. t.expect(f'[{g.dfl_hash_preset}]')
  122. t.read()
  123. return t
  124. def password_entry(self,prompt,cmd_args):
  125. t = self.spawn('test/misc/password_entry.py',cmd_args,cmd_dir='.')
  126. pw = 'abc-α'
  127. t.expect(prompt,pw)
  128. ret = t.expect_getend('Entered: ')
  129. assert ret == pw,'Password mismatch! {} != {}'.format(ret,pw)
  130. return t
  131. def password_entry_noecho(self):
  132. if self.skip_for_win():
  133. m = "getpass() doesn't work with pexpect.popen_spawn!\n"
  134. m += 'Perform the following test by hand with non-ASCII password abc-α:\n'
  135. m += ' test/misc/password_entry.py'
  136. return ('skip_warn',m)
  137. return self.password_entry('Enter passphrase: ',[])
  138. def password_entry_echo(self):
  139. if self.skip_for_win():
  140. m = "getpass() doesn't work with pexpect.popen_spawn!\n"
  141. m += 'Perform the following test by hand with non-ASCII password abc-α:\n'
  142. m += ' test/misc/password_entry.py --echo-passphrase'
  143. return ('skip_warn',m)
  144. return self.password_entry('Enter passphrase (echoed): ',['--echo-passphrase'])
  145. def _mn2hex(self,fmt,entry_mode='full',mn=None,pad_entry=False):
  146. mn = mn or sample_mn[fmt]['mn'].split()
  147. t = self.spawn('mmgen-tool',['mn2hex_interactive','fmt='+fmt,'mn_len=12','print_mn=1'])
  148. from mmgen.mn_entry import mn_entry
  149. mne = mn_entry(fmt,entry_mode)
  150. t.expect('Entry mode: ',str(mne.entry_modes.index(entry_mode)+1))
  151. t.expect('Using (.+) entry mode',regex=True)
  152. mode = t.p.match.group(1).lower()
  153. assert mode == mne.em.name.lower(), '{} != {}'.format(mode,mne.em.name.lower())
  154. stealth_mnemonic_entry(t,mne,mn,entry_mode=entry_mode,pad_entry=pad_entry)
  155. t.expect(sample_mn[fmt]['hex'])
  156. t.read()
  157. return t
  158. def _user_seed_entry(self,fmt,usr_rand=False,out_fmt=None,entry_mode='full',mn=None):
  159. wcls = Wallet.fmt_code_to_type(fmt)
  160. wf = os.path.join(ref_dir,'FE3C6545.{}'.format(wcls.ext))
  161. if wcls.wclass == 'mnemonic':
  162. mn = mn or read_from_file(wf).strip().split()
  163. elif wcls.wclass == 'dieroll':
  164. mn = mn or list(read_from_file(wf).strip().translate(dict((ord(ws),None) for ws in '\t\n ')))
  165. for idx,val in ((5,'x'),(18,'0'),(30,'7'),(44,'9')):
  166. mn.insert(idx,val)
  167. t = self.spawn('mmgen-walletconv',['-r10','-S','-i',fmt,'-o',out_fmt or fmt])
  168. t.expect('{} type: {}'.format(capfirst(wcls.wclass),wcls.mn_type))
  169. t.expect(wcls.choose_seedlen_prompt,'1')
  170. t.expect('(Y/n): ','y')
  171. if wcls.wclass == 'mnemonic':
  172. t.expect('Entry mode: ','6')
  173. t.expect('invalid')
  174. from mmgen.mn_entry import mn_entry
  175. mne = mn_entry(fmt,entry_mode)
  176. t.expect('Entry mode: ',str(mne.entry_modes.index(entry_mode)+1))
  177. t.expect('Using (.+) entry mode',regex=True)
  178. mode = t.p.match.group(1).lower()
  179. assert mode == mne.em.name.lower(), '{} != {}'.format(mode,mne.em.name.lower())
  180. stealth_mnemonic_entry(t,mne,mn,entry_mode=entry_mode)
  181. elif wcls.wclass == 'dieroll':
  182. user_dieroll_entry(t,mn)
  183. if usr_rand:
  184. t.expect(wcls.user_entropy_prompt,'y')
  185. t.usr_rand(10)
  186. else:
  187. t.expect(wcls.user_entropy_prompt,'n')
  188. if not usr_rand:
  189. sid_chk = 'FE3C6545'
  190. sid = t.expect_getend('Valid {} for Seed ID '.format(wcls.desc))[:8]
  191. assert sid == sid_chk,'Seed ID mismatch! {} != {}'.format(sid,sid_chk)
  192. t.expect('to confirm: ','YES\n')
  193. t.read()
  194. return t
  195. def mnemonic_entry_mmgen_minimal(self):
  196. from mmgen.mn_entry import mn_entry
  197. # erase_chars: '\b\x7f'
  198. m = mn_entry('mmgen','minimal')
  199. np = 2
  200. mn = (
  201. 'z',
  202. 'aa',
  203. '1d2ud',
  204. 'fo{}ot{}#'.format('1' * np, '2' * (m.em.pad_max - np)), # substring of 'football'
  205. 'des1p)%erate\n', # substring of 'desperately'
  206. '#t!(ie',
  207. '!)sto8o',
  208. 'the123m8!%s',
  209. '349t(5)rip',
  210. 'di\b\bdesce',
  211. 'cea',
  212. 'bu\x7f\x7fsuic',
  213. 'app\bpl',
  214. 'wd',
  215. 'busy')
  216. return self._user_seed_entry('words',entry_mode='minimal',mn=mn)
  217. def mnemonic_entry_mmgen(self): return self._user_seed_entry('words',entry_mode='full')
  218. def mnemonic_entry_bip39(self): return self._user_seed_entry('bip39',entry_mode='full')
  219. def mnemonic_entry_bip39_short(self): return self._user_seed_entry('bip39',entry_mode='short')
  220. def mn2hex_interactive_mmgen(self): return self._mn2hex('mmgen',entry_mode='full')
  221. def mn2hex_interactive_mmgen_fixed(self): return self._mn2hex('mmgen',entry_mode='fixed')
  222. def mn2hex_interactive_bip39(self): return self._mn2hex('bip39',entry_mode='full')
  223. def mn2hex_interactive_bip39_short(self): return self._mn2hex('bip39',entry_mode='short',pad_entry=True)
  224. def mn2hex_interactive_bip39_fixed(self): return self._mn2hex('bip39',entry_mode='fixed')
  225. def mn2hex_interactive_xmr(self): return self._mn2hex('xmrseed',entry_mode='full')
  226. def mn2hex_interactive_xmr_short(self): return self._mn2hex('xmrseed',entry_mode='short')
  227. def dieroll_entry(self): return self._user_seed_entry('dieroll')
  228. def dieroll_entry_usrrand(self): return self._user_seed_entry('dieroll',usr_rand=True,out_fmt='bip39')