ts_ref_altcoin.py 9.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241
  1. #!/usr/bin/env python3
  2. #
  3. # mmgen = Multi-Mode GENerator, command-line Bitcoin cold storage solution
  4. # Copyright (C)2013-2023 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. test.test_py_d.ts_ref_altcoin: Altcoin reference file tests for the test.py test suite
  20. """
  21. import os
  22. from mmgen.color import set_vt100
  23. from .common import pwfile,dfl_wpasswd,ref_dir,dfl_words_file,dfl_addr_idx_list
  24. from ..include.common import cfg,joinpath,start_test_daemons,stop_test_daemons,cmp_or_die
  25. from .ts_ref import TestSuiteRef
  26. from .ts_base import TestSuiteBase
  27. class TestSuiteRefAltcoin(TestSuiteRef,TestSuiteBase):
  28. 'saved and generated altcoin reference files'
  29. tmpdir_nums = [8]
  30. networks = ('btc',)
  31. chk_data = {
  32. 'ref_addrfile_chksum_zec': '903E 7225 DD86 6E01',
  33. 'ref_addrfile_chksum_zec_z': '9C7A 72DC 3D4A B3AF',
  34. 'ref_addrfile_chksum_xmr': '4369 0253 AC2C 0E38',
  35. 'ref_addrfile_chksum_dash':'FBC1 6B6A 0988 4403',
  36. 'ref_addrfile_chksum_eth': 'E554 076E 7AF6 66A3',
  37. 'ref_addrfile_chksum_etc': 'E97A D796 B495 E8BC',
  38. 'ref_keyaddrfile_chksum_zec': 'F05A 5A5C 0C8E 2617',
  39. 'ref_keyaddrfile_chksum_zec_z': '6B87 9B2D 0D8D 8D1E',
  40. 'ref_keyaddrfile_chksum_xmr': 'E0D7 9612 3D67 404A',
  41. 'ref_viewkeyaddrfile_chksum_xmr': '40C9 0E61 B743 229C',
  42. 'ref_keyaddrfile_chksum_dash': 'E83D 2C63 FEA2 4142',
  43. 'ref_keyaddrfile_chksum_eth': 'E400 70D9 0AE3 C7C2',
  44. 'ref_keyaddrfile_chksum_etc': 'EF49 967D BD6C FE45',
  45. }
  46. cmd_group = (
  47. ('ref_altcoin_tx_chk', 'signing saved reference tx files'),
  48. ('ref_addrfile_gen_eth', 'generate address file (ETH)'),
  49. ('ref_addrfile_gen_etc', 'generate address file (ETC)'),
  50. ('ref_addrfile_gen_dash', 'generate address file (DASH)'),
  51. ('ref_addrfile_gen_zec', 'generate address file (ZEC-T)'),
  52. ('ref_addrfile_gen_zec_z','generate address file (ZEC-Z)'),
  53. ('ref_addrfile_gen_xmr', 'generate address file (XMR)'),
  54. # we test the unoptimized ed25519 mod in unit_tests.py, so skip this
  55. # ('ref_addrfile_gen_xmr_slow','generate address file (XMR - unoptimized ed25519 module)'),
  56. ('ref_keyaddrfile_gen_eth', 'generate key-address file (ETH)'),
  57. ('ref_keyaddrfile_gen_etc', 'generate key-address file (ETC)'),
  58. ('ref_keyaddrfile_gen_dash', 'generate key-address file (DASH)'),
  59. ('ref_keyaddrfile_gen_zec', 'generate key-address file (ZEC-T)'),
  60. ('ref_keyaddrfile_gen_zec_z','generate key-address file (ZEC-Z)'),
  61. ('ref_keyaddrfile_gen_xmr', 'generate key-address file (XMR)'),
  62. ('ref_viewkeyaddrfile_gen_xmr','generate viewkey-address file (XMR)'),
  63. ('ref_addrfile_chk_eth', 'reference address file (ETH)'),
  64. ('ref_addrfile_chk_etc', 'reference address file (ETC)'),
  65. ('ref_addrfile_chk_dash','reference address file (DASH)'),
  66. ('ref_addrfile_chk_zec', 'reference address file (ZEC-T)'),
  67. ('ref_addrfile_chk_zec_z','reference address file (ZEC-Z)'),
  68. ('ref_addrfile_chk_xmr', 'reference address file (XMR)'),
  69. ('ref_keyaddrfile_chk_eth', 'reference key-address file (ETH)'),
  70. ('ref_keyaddrfile_chk_etc', 'reference key-address file (ETC)'),
  71. ('ref_keyaddrfile_chk_dash','reference key-address file (DASH)'),
  72. ('ref_keyaddrfile_chk_zec', 'reference key-address file (ZEC-T)'),
  73. ('ref_keyaddrfile_chk_zec_z','reference key-address file (ZEC-Z)'),
  74. ('ref_keyaddrfile_chk_xmr', 'reference key-address file (XMR)'),
  75. ('ref_viewkeyaddrfile_chk_xmr', 'reference viewkey-address file (XMR)'),
  76. )
  77. def ref_altcoin_tx_chk(self):
  78. """
  79. Check saved transaction files for *all* configured altcoins
  80. Though this basically duplicates the autosign test, here we do everything
  81. via the command line, so it's worth doing
  82. """
  83. self.write_to_tmpfile(pwfile,dfl_wpasswd)
  84. passfile = joinpath(self.tmpdir,pwfile)
  85. from mmgen.tx.file import MMGenTxFile
  86. src = TestSuiteRef.sources['ref_tx_file']
  87. for coin,files in src.items():
  88. if coin == 'mm1':
  89. coin = 'eth'
  90. token_desc = ':MM1'
  91. else:
  92. token_desc = ''
  93. for fn in files:
  94. if not fn: # no etc testnet TX file
  95. continue
  96. txfile = joinpath(
  97. ref_dir,
  98. self._get_ref_subdir_by_coin(coin),
  99. fn )
  100. proto = MMGenTxFile.get_proto(cfg,txfile,quiet_open=True)
  101. if proto.sign_mode == 'daemon':
  102. start_test_daemons(proto.network_id)
  103. set_vt100()
  104. t = self.spawn(
  105. 'mmgen-txsign',
  106. ['--outdir=test/trash','--yes', f'--passwd-file={passfile}', dfl_words_file, txfile],
  107. extra_desc = f'{proto.coin}{token_desc} {proto.network}')
  108. t.read()
  109. t.ok()
  110. if proto.sign_mode == 'daemon' and not proto.network_id == 'btc':
  111. stop_test_daemons(proto.network_id)
  112. return 'ok'
  113. def ref_altcoin_addrgen(self,coin,mmtype,gen_what='addr',coin_suf='',add_args=[],addr_idx_list=None):
  114. wf = dfl_words_file
  115. t = self.spawn(
  116. 'mmgen-keygen' if 'key' in gen_what else 'mmgen-addrgen',
  117. ['-Sq','--coin='+coin] +
  118. (['--type='+mmtype] if mmtype else []) +
  119. add_args +
  120. [wf,addr_idx_list or dfl_addr_idx_list])
  121. if 'key' in gen_what:
  122. t.expect('Encrypt key list? (y/N): ','N')
  123. chk = t.expect_getend(r'.* data checksum for \S*: ',regex=True)
  124. chk_ref = self.chk_data[
  125. 'ref_{}addrfile_chksum_{}{}'.format(
  126. (gen_what if 'key' in gen_what else ''),
  127. coin.lower(),
  128. coin_suf )
  129. ]
  130. cmp_or_die(chk,chk_ref,desc=f'{gen_what}list data checksum')
  131. return t
  132. def ref_addrfile_gen_eth(self):
  133. return self.ref_altcoin_addrgen(coin='ETH',mmtype='ethereum')
  134. def ref_addrfile_gen_etc(self):
  135. return self.ref_altcoin_addrgen(coin='ETC',mmtype='ethereum')
  136. def ref_addrfile_gen_dash(self):
  137. return self.ref_altcoin_addrgen(coin='DASH',mmtype='compressed')
  138. def ref_addrfile_gen_zec(self):
  139. return self.ref_altcoin_addrgen(coin='ZEC',mmtype='compressed')
  140. def ref_addrfile_gen_zec_z(self):
  141. return self.ref_altcoin_addrgen(coin='ZEC',mmtype='zcash_z',coin_suf='_z')
  142. def ref_addrfile_gen_xmr(self):
  143. return self.ref_altcoin_addrgen(coin='XMR',mmtype='monero')
  144. def ref_addrfile_gen_xmr_slow(self):
  145. return self.ref_altcoin_addrgen(coin='XMR',mmtype='monero',add_args=['--keygen-backend=2'])
  146. def ref_keyaddrfile_gen_eth(self):
  147. return self.ref_altcoin_addrgen(coin='ETH',mmtype='ethereum',gen_what='key')
  148. def ref_keyaddrfile_gen_etc(self):
  149. return self.ref_altcoin_addrgen(coin='ETC',mmtype='ethereum',gen_what='key')
  150. def ref_keyaddrfile_gen_dash(self):
  151. return self.ref_altcoin_addrgen(coin='DASH',mmtype='compressed',gen_what='key')
  152. def ref_keyaddrfile_gen_zec(self):
  153. return self.ref_altcoin_addrgen(coin='ZEC',mmtype='compressed',gen_what='key')
  154. def ref_keyaddrfile_gen_zec_z(self):
  155. return self.ref_altcoin_addrgen(coin='ZEC',mmtype='zcash_z',coin_suf='_z',gen_what='key')
  156. def ref_keyaddrfile_gen_xmr(self):
  157. return self.ref_altcoin_addrgen(coin='XMR',mmtype='monero',gen_what='key')
  158. def ref_viewkeyaddrfile_gen_xmr(self):
  159. return self.ref_altcoin_addrgen(
  160. coin = 'XMR',
  161. mmtype = 'monero',
  162. gen_what = 'viewkey',
  163. add_args = ['--viewkeys'],
  164. addr_idx_list = '1-3' )
  165. def ref_addrfile_chk_eth(self):
  166. return self.ref_addrfile_chk(ftype='addr',coin='ETH',subdir='ethereum',pfx='-ETH',
  167. pat='ETH Mainnet.*Ethereum')
  168. def ref_addrfile_chk_etc(self):
  169. return self.ref_addrfile_chk(ftype='addr',coin='ETC',subdir='ethereum_classic',pfx='-ETC',
  170. pat='ETH Mainnet.*Ethereum')
  171. def ref_addrfile_chk_dash(self):
  172. return self.ref_addrfile_chk(ftype='addr',coin='DASH',subdir='dash',pfx='-DASH-C',
  173. pat='DASH Mainnet.*Compressed')
  174. def ref_addrfile_chk_zec(self):
  175. return self.ref_addrfile_chk(ftype='addr',coin='ZEC',subdir='zcash',pfx='-ZEC-C',
  176. pat='ZEC Mainnet.*Compressed')
  177. def ref_addrfile_chk_zec_z(self):
  178. return self.ref_addrfile_chk(ftype='addr',coin='ZEC',subdir='zcash',pfx='-ZEC-Z',mmtype='z',
  179. pat='ZEC Mainnet.*Zcash_z')
  180. def ref_addrfile_chk_xmr(self):
  181. return self.ref_addrfile_chk(ftype='addr',coin='XMR',subdir='monero',pfx='-XMR-M',
  182. pat='XMR Mainnet.*Monero')
  183. def ref_keyaddrfile_chk_eth(self):
  184. return self.ref_addrfile_chk(ftype='keyaddr',coin='ETH',subdir='ethereum',pfx='-ETH',
  185. pat='ETH Mainnet.*Ethereum')
  186. def ref_keyaddrfile_chk_etc(self):
  187. return self.ref_addrfile_chk(ftype='keyaddr',coin='ETC',subdir='ethereum_classic',pfx='-ETC',
  188. pat='ETH Mainnet.*Ethereum')
  189. def ref_keyaddrfile_chk_dash(self):
  190. return self.ref_addrfile_chk(ftype='keyaddr',coin='DASH',subdir='dash',pfx='-DASH-C',
  191. pat='DASH Mainnet.*Compressed')
  192. def ref_keyaddrfile_chk_zec(self):
  193. return self.ref_addrfile_chk(ftype='keyaddr',coin='ZEC',subdir='zcash',pfx='-ZEC-C',
  194. pat='ZEC Mainnet.*Compressed')
  195. def ref_keyaddrfile_chk_zec_z(self):
  196. return self.ref_addrfile_chk(ftype='keyaddr',coin='ZEC',subdir='zcash',pfx='-ZEC-Z',mmtype='z',
  197. pat='ZEC Mainnet.*Zcash_z')
  198. def ref_keyaddrfile_chk_xmr(self):
  199. return self.ref_addrfile_chk(ftype='keyaddr',coin='XMR',subdir='monero',pfx='-XMR-M',
  200. pat='XMR Mainnet.*Monero')
  201. def ref_viewkeyaddrfile_chk_xmr(self):
  202. return self.ref_addrfile_chk(ftype='viewkeyaddr',coin='XMR',subdir='monero',pfx='-XMR-M',
  203. pat='XMR Mainnet.*Monero')