test.py 109 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358235923602361236223632364236523662367236823692370237123722373237423752376237723782379238023812382238323842385238623872388238923902391239223932394239523962397239823992400240124022403240424052406240724082409241024112412241324142415241624172418241924202421242224232424242524262427242824292430243124322433243424352436243724382439244024412442244324442445244624472448244924502451245224532454245524562457245824592460246124622463246424652466246724682469247024712472247324742475247624772478247924802481248224832484248524862487248824892490249124922493249424952496249724982499250025012502250325042505250625072508250925102511251225132514251525162517251825192520252125222523252425252526252725282529253025312532253325342535253625372538253925402541254225432544254525462547254825492550255125522553255425552556255725582559256025612562256325642565256625672568256925702571257225732574257525762577257825792580258125822583258425852586258725882589259025912592259325942595259625972598259926002601260226032604260526062607260826092610261126122613261426152616261726182619262026212622262326242625262626272628262926302631263226332634263526362637263826392640264126422643264426452646264726482649265026512652265326542655265626572658265926602661266226632664266526662667266826692670267126722673267426752676267726782679268026812682268326842685268626872688268926902691269226932694269526962697269826992700270127022703270427052706270727082709271027112712271327142715271627172718271927202721272227232724272527262727272827292730273127322733273427352736273727382739274027412742274327442745274627472748274927502751275227532754275527562757275827592760276127622763276427652766276727682769277027712772277327742775277627772778277927802781278227832784278527862787278827892790279127922793279427952796279727982799280028012802280328042805280628072808280928102811281228132814281528162817281828192820282128222823282428252826282728282829283028312832283328342835283628372838283928402841284228432844284528462847284828492850285128522853285428552856285728582859286028612862286328642865286628672868286928702871287228732874287528762877287828792880288128822883288428852886288728882889289028912892289328942895289628972898289929002901290229032904290529062907290829092910291129122913291429152916291729182919292029212922292329242925292629272928292929302931293229332934293529362937293829392940294129422943294429452946294729482949295029512952
  1. #!/usr/bin/env python
  2. # -*- coding: UTF-8 -*-
  3. # mmgen = Multi-Mode GENerator, command-line Bitcoin cold storage solution
  4. # Copyright (C)2013-2018 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: Test suite for the MMGen suite
  20. """
  21. import sys,os,subprocess,shutil,time,re
  22. repo_root = os.path.normpath(os.path.abspath(os.path.join(os.path.dirname(sys.argv[0]),os.pardir)))
  23. os.chdir(repo_root)
  24. sys.path.__setitem__(0,repo_root)
  25. # Import these _after_ local path's been added to sys.path
  26. from mmgen.common import *
  27. from mmgen.test import *
  28. from mmgen.protocol import CoinProtocol
  29. g.quiet = False # if 'quiet' was set in config file, disable here
  30. os.environ['MMGEN_QUIET'] = '0' # and for the spawned scripts
  31. log_file = 'test.py_log'
  32. hincog_fn = 'rand_data'
  33. hincog_bytes = 1024*1024
  34. hincog_offset = 98765
  35. hincog_seedlen = 256
  36. incog_id_fn = 'incog_id'
  37. non_mmgen_fn = 'coinkey'
  38. pwfile = 'passwd_file'
  39. ref_dir = os.path.join('test','ref')
  40. ref_wallet_brainpass = 'abc'
  41. ref_wallet_hash_preset = '1'
  42. ref_wallet_incog_offset = 123
  43. from mmgen.obj import MMGenTXLabel,PrivKey
  44. from mmgen.addr import AddrGenerator,KeyGenerator,AddrList,AddrData,AddrIdxList
  45. ref_tx_label = ''.join([unichr(i) for i in range(65,91) +
  46. range(1040,1072) + # cyrillic
  47. range(913,939) + # greek
  48. range(97,123)])[:MMGenTXLabel.max_len]
  49. ref_bw_hash_preset = '1'
  50. ref_bw_file = 'wallet.mmbrain'
  51. ref_bw_file_spc = 'wallet-spaced.mmbrain'
  52. ref_kafile_pass = 'kafile password'
  53. ref_kafile_hash_preset = '1'
  54. ref_enc_fn = 'sample-text.mmenc'
  55. tool_enc_passwd = "Scrypt it, don't hash it!"
  56. sample_text = \
  57. 'The Times 03/Jan/2009 Chancellor on brink of second bailout for banks\n'
  58. # Laggy flash media cause pexpect to crash, so create a temporary directory
  59. # under '/dev/shm' and put datadir and temp files here.
  60. shortopts = ''.join([e[1:] for e in sys.argv if len(e) > 1 and e[0] == '-' and e[1] != '-'])
  61. shortopts = ['-'+e for e in list(shortopts)]
  62. data_dir = os.path.join('test','data_dir')
  63. if not any(e in ('--skip-deps','--resume','-S','-r') for e in sys.argv+shortopts):
  64. if g.platform == 'win':
  65. try: os.listdir(data_dir)
  66. except: pass
  67. else:
  68. try: shutil.rmtree(data_dir)
  69. except: # we couldn't remove data dir - perhaps regtest daemon is running
  70. try: subprocess.call(['python',os.path.join('cmds','mmgen-regtest'),'stop'])
  71. except: rdie(1,'Unable to remove data dir!')
  72. else:
  73. time.sleep(2)
  74. shutil.rmtree(data_dir)
  75. os.mkdir(data_dir,0755)
  76. else:
  77. d,pfx = '/dev/shm','mmgen-test-'
  78. try:
  79. subprocess.call('rm -rf %s/%s*'%(d,pfx),shell=True)
  80. except Exception as e:
  81. die(2,'Unable to delete directory tree %s/%s* (%s)'%(d,pfx,e))
  82. try:
  83. import tempfile
  84. shm_dir = tempfile.mkdtemp('',pfx,d)
  85. except Exception as e:
  86. die(2,'Unable to create temporary directory in %s (%s)'%(d,e))
  87. dd = os.path.join(shm_dir,'data_dir')
  88. os.mkdir(dd,0755)
  89. try: os.unlink(data_dir)
  90. except: pass
  91. os.symlink(dd,data_dir)
  92. opts_data = lambda: {
  93. 'desc': 'Test suite for the MMGen suite',
  94. 'usage':'[options] [command(s) or metacommand(s)]',
  95. 'options': """
  96. -h, --help Print this help message
  97. --, --longhelp Print help message for long options (common options)
  98. -B, --bech32 Generate and use Bech32 addresses
  99. -b, --buf-keypress Use buffered keypresses as with real human input
  100. -c, --print-cmdline Print the command line of each spawned command
  101. -C, --coverage Produce code coverage info using trace module
  102. -d, --debug-scripts Turn on debugging output in executed scripts
  103. -x, --debug-pexpect Produce debugging output for pexpect calls
  104. -D, --direct-exec Bypass pexpect and execute a command directly (for
  105. debugging only)
  106. -e, --exact-output Show the exact output of the MMGen script(s) being run
  107. -g, --segwit Generate and use Segwit addresses
  108. -G, --segwit-random Generate and use a random mix of Segwit and Legacy addrs
  109. -l, --list-cmds List and describe the commands in the test suite
  110. -L, --log Log commands to file {lf}
  111. -n, --names Display command names instead of descriptions
  112. -O, --popen-spawn Use pexpect's popen_spawn instead of popen (always true, so ignored)
  113. -p, --pause Pause between tests, resuming on keypress
  114. -P, --profile Record the execution time of each script
  115. -q, --quiet Produce minimal output. Suppress dependency info
  116. -r, --resume=c Resume at command 'c' after interrupted run
  117. -s, --system Test scripts and modules installed on system rather
  118. than those in the repo root
  119. -S, --skip-deps Skip dependency checking for command
  120. -u, --usr-random Get random data interactively from user
  121. -t, --traceback Run the command inside the '{tbc}' script
  122. -v, --verbose Produce more verbose output
  123. -W, --no-dw-delete Don't remove default wallet from data dir after dw tests are done
  124. """.format(tbc=g.traceback_cmd,lf=log_file),
  125. 'notes': """
  126. If no command is given, the whole suite of tests is run.
  127. """
  128. }
  129. sys.argv = [sys.argv[0]] + ['--data-dir',data_dir] + sys.argv[1:]
  130. cmd_args = opts.init(opts_data)
  131. opt.popen_spawn = True # popen has issues, so use popen_spawn always
  132. ref_subdir = '' if g.proto.base_coin == 'BTC' else g.proto.name
  133. altcoin_pfx = '' if g.proto.base_coin == 'BTC' else '-'+g.proto.base_coin
  134. tn_ext = ('','.testnet')[g.testnet]
  135. coin_sel = g.coin.lower()
  136. if g.coin == 'B2X': coin_sel = 'btc'
  137. fork = {'bch':'btc','btc':'btc','ltc':'ltc'}[coin_sel]
  138. tx_fee = {'btc':'0.0001','bch':'0.001','ltc':'0.01'}[coin_sel]
  139. txbump_fee = {'btc':'123s','bch':'567s','ltc':'12345s'}[coin_sel]
  140. rtFundAmt = {'btc':'500','bch':'500','ltc':'5500'}[coin_sel]
  141. rtFee = {
  142. 'btc': ('20s','10s','60s','31s','10s','20s'),
  143. 'bch': ('20s','10s','60s','0.0001','10s','20s'),
  144. 'ltc': ('1000s','500s','1500s','0.05','400s','1000s')
  145. }[coin_sel]
  146. rtBals = {
  147. 'btc': ('499.9999488','399.9998282','399.9998147','399.9996877','13.00000000','986.99933647','999.99933647'),
  148. 'bch': ('499.9999416','399.9999124','399.99989','399.9997616','276.22339397','723.77626763','999.99966160'),
  149. 'ltc': ('5499.9971','5399.994085','5399.993545','5399.987145','13.00000000','10986.93714500','10999.93714500'),
  150. }[coin_sel]
  151. rtBobOp3 = {'btc':'S:2','bch':'L:3','ltc':'S:2'}[coin_sel]
  152. if opt.segwit and 'S' not in g.proto.mmtypes:
  153. die(1,'--segwit option incompatible with {}'.format(g.proto.__name__))
  154. if opt.bech32 and 'B' not in g.proto.mmtypes:
  155. die(1,'--bech32 option incompatible with {}'.format(g.proto.__name__))
  156. def randbool():
  157. return hexlify(os.urandom(1))[1] in '12345678'
  158. def get_segwit_bool():
  159. return randbool() if opt.segwit_random else True if opt.segwit or opt.bech32 else False
  160. def disable_debug():
  161. ds = os.getenv('MMGEN_DEBUG')
  162. if ds is not None: os.environ['MMGEN_DEBUG'] = ''
  163. return ds
  164. def restore_debug(ds):
  165. if ds is not None: os.environ['MMGEN_DEBUG'] = ds
  166. cfgs = {
  167. '15': {
  168. 'tmpdir': os.path.join('test','tmp15'),
  169. 'wpasswd': 'Dorian',
  170. 'kapasswd': 'Grok the blockchain',
  171. 'addr_idx_list': '12,99,5-10,5,12', # 8 addresses
  172. 'dep_generators': {
  173. pwfile: 'walletgen_dfl_wallet',
  174. 'addrs': 'addrgen_dfl_wallet',
  175. 'rawtx': 'txcreate_dfl_wallet',
  176. 'sigtx': 'txsign_dfl_wallet',
  177. 'mmseed': 'export_seed_dfl_wallet',
  178. 'del_dw_run': 'delete_dfl_wallet',
  179. },
  180. 'segwit': get_segwit_bool()
  181. },
  182. '16': {
  183. 'tmpdir': os.path.join('test','tmp16'),
  184. 'wpasswd': 'My changed password',
  185. 'hash_preset': '2',
  186. 'dep_generators': {
  187. pwfile: 'passchg_dfl_wallet',
  188. },
  189. 'segwit': get_segwit_bool()
  190. },
  191. '17': { 'tmpdir': os.path.join('test','tmp17') },
  192. '18': { 'tmpdir': os.path.join('test','tmp18') },
  193. '19': { 'tmpdir': os.path.join('test','tmp19'), 'wpasswd':'abc' },
  194. '1': {
  195. 'tmpdir': os.path.join('test','tmp1'),
  196. 'wpasswd': 'Dorian',
  197. 'kapasswd': 'Grok the blockchain',
  198. 'addr_idx_list': '12,99,5-10,5,12', # 8 addresses
  199. 'dep_generators': {
  200. pwfile: 'walletgen',
  201. 'mmdat': 'walletgen',
  202. 'addrs': 'addrgen',
  203. 'rawtx': 'txcreate',
  204. 'txbump': 'txbump',
  205. 'sigtx': 'txsign',
  206. 'mmwords': 'export_mnemonic',
  207. 'mmseed': 'export_seed',
  208. 'mmhex': 'export_hex',
  209. 'mmincog': 'export_incog',
  210. 'mmincox': 'export_incog_hex',
  211. hincog_fn: 'export_incog_hidden',
  212. incog_id_fn: 'export_incog_hidden',
  213. 'akeys.mmenc': 'keyaddrgen'
  214. },
  215. 'segwit': get_segwit_bool()
  216. },
  217. '2': {
  218. 'tmpdir': os.path.join('test','tmp2'),
  219. 'wpasswd': 'Hodling away',
  220. 'addr_idx_list': '37,45,3-6,22-23', # 8 addresses
  221. 'seed_len': 128,
  222. 'dep_generators': {
  223. 'mmdat': 'walletgen2',
  224. 'addrs': 'addrgen2',
  225. 'rawtx': 'txcreate2',
  226. 'sigtx': 'txsign2',
  227. 'mmwords': 'export_mnemonic2',
  228. },
  229. 'segwit': get_segwit_bool()
  230. },
  231. '3': {
  232. 'tmpdir': os.path.join('test','tmp3'),
  233. 'wpasswd': 'Major miner',
  234. 'addr_idx_list': '73,54,1022-1023,2-5', # 8 addresses
  235. 'dep_generators': {
  236. 'mmdat': 'walletgen3',
  237. 'addrs': 'addrgen3',
  238. 'rawtx': 'txcreate3',
  239. 'sigtx': 'txsign3'
  240. },
  241. 'segwit': get_segwit_bool()
  242. },
  243. '4': {
  244. 'tmpdir': os.path.join('test','tmp4'),
  245. 'wpasswd': 'Hashrate good',
  246. 'addr_idx_list': '63,1004,542-544,7-9', # 8 addresses
  247. 'seed_len': 192,
  248. 'dep_generators': {
  249. 'mmdat': 'walletgen4',
  250. 'mmbrain': 'walletgen4',
  251. 'addrs': 'addrgen4',
  252. 'rawtx': 'txcreate4',
  253. 'sigtx': 'txsign4',
  254. 'txdo': 'txdo4',
  255. },
  256. 'bw_filename': 'brainwallet.mmbrain',
  257. 'bw_params': '192,1',
  258. 'segwit': get_segwit_bool()
  259. },
  260. '14': {
  261. 'kapasswd': 'Maxwell',
  262. 'tmpdir': os.path.join('test','tmp14'),
  263. 'wpasswd': 'The Halving',
  264. 'addr_idx_list': '61,998,502-504,7-9', # 8 addresses
  265. 'seed_len': 256,
  266. 'dep_generators': {
  267. 'mmdat': 'walletgen14',
  268. 'addrs': 'addrgen14',
  269. 'akeys.mmenc': 'keyaddrgen14',
  270. },
  271. 'segwit': get_segwit_bool()
  272. },
  273. '5': {
  274. 'tmpdir': os.path.join('test','tmp5'),
  275. 'wpasswd': 'My changed password',
  276. 'hash_preset': '2',
  277. 'dep_generators': {
  278. 'mmdat': 'passchg',
  279. pwfile: 'passchg',
  280. },
  281. 'segwit': get_segwit_bool()
  282. },
  283. '6': {
  284. 'name': 'reference wallet check (128-bit)',
  285. 'seed_len': 128,
  286. 'seed_id': 'FE3C6545',
  287. 'ref_bw_seed_id': '33F10310',
  288. 'addrfile_chk': {
  289. 'btc': ('B230 7526 638F 38CB','B64D 7327 EF2A 60FE'),
  290. 'ltc': ('2B23 5E97 848A B961','928D 3CB6 78FF 9829'),
  291. },
  292. 'addrfile_segwit_chk': {
  293. 'btc': ('9914 6D10 2307 F348','7DBF 441F E188 8B37'),
  294. 'ltc': ('CC09 A190 B7DF B7CD','3676 4C49 14F8 1AD0'),
  295. },
  296. 'addrfile_bech32_chk': {
  297. 'btc': ('C529 D686 31AA ACD4','13C5 493E 1CD7 2852'),
  298. 'ltc': ('CC09 A190 B7DF B7CD','3676 4C49 14F8 1AD0'),
  299. },
  300. 'addrfile_compressed_chk': {
  301. 'btc': ('95EB 8CC0 7B3B 7856','629D FDE4 CDC0 F276'),
  302. 'ltc': ('35D5 8ECA 9A42 46C3','37E9 A36E 94A2 010F'),
  303. },
  304. 'keyaddrfile_chk': {
  305. 'btc': ('CF83 32FB 8A8B 08E2','FEBF 7878 97BB CC35'),
  306. 'ltc': ('1896 A26C 7F14 2D01','B41D BA63 0605 DD66'),
  307. },
  308. 'keyaddrfile_segwit_chk': {
  309. 'btc': ('C13B F717 D4E8 CF59','4DB5 BAF0 45B7 6E81'),
  310. 'ltc': ('054B 9794 55B4 5D82','C373 0074 DEE6 B70A'),
  311. },
  312. 'keyaddrfile_bech32_chk': {
  313. 'btc': ('934F 1C33 6C06 B18C','D994 DF67 6E53 6F5F'),
  314. 'ltc': ('054B 9794 55B4 5D82','C373 0074 DEE6 B70A'),
  315. },
  316. 'keyaddrfile_compressed_chk': {
  317. 'btc': ('E43A FA46 5751 720A','B995 A6CF D1CD FAD0'),
  318. 'ltc': ('7603 2FE3 2145 FFAD','3248 356A C707 4A41'),
  319. },
  320. 'passfile_chk': 'EB29 DC4F 924B 289F',
  321. 'passfile32_chk': '37B6 C218 2ABC 7508',
  322. 'passfilehex_chk': '523A F547 0E69 8323',
  323. 'wpasswd': 'reference password',
  324. 'ref_wallet': 'FE3C6545-D782B529[128,1].mmdat',
  325. 'ic_wallet': 'FE3C6545-E29303EA-5E229E30[128,1].mmincog',
  326. 'ic_wallet_hex': 'FE3C6545-BC4BE3F2-32586837[128,1].mmincox',
  327. 'hic_wallet': 'FE3C6545-161E495F-BEB7548E[128,1].incog-offset123',
  328. 'hic_wallet_old': 'FE3C6545-161E495F-9860A85B[128,1].incog-old.offset123',
  329. 'tmpdir': os.path.join('test','tmp6'),
  330. 'kapasswd': '',
  331. 'addr_idx_list': '1010,500-501,31-33,1,33,500,1011', # 8 addresses
  332. 'pass_idx_list': '1,4,9-11,1100',
  333. 'dep_generators': {
  334. 'mmdat': 'refwalletgen1',
  335. pwfile: 'refwalletgen1',
  336. 'addrs': 'refaddrgen1',
  337. 'akeys.mmenc': 'refkeyaddrgen1'
  338. },
  339. 'segwit': get_segwit_bool()
  340. },
  341. '7': {
  342. 'name': 'reference wallet check (192-bit)',
  343. 'seed_len': 192,
  344. 'seed_id': '1378FC64',
  345. 'ref_bw_seed_id': 'CE918388',
  346. 'addrfile_chk': {
  347. 'btc': ('8C17 A5FA 0470 6E89','0A59 C8CD 9439 8B81'),
  348. 'ltc': ('2B77 A009 D5D0 22AD','FCEC 0032 9EF9 B201'),
  349. },
  350. 'addrfile_segwit_chk': {
  351. 'btc': ('91C4 0414 89E4 2089','3BA6 7494 8E2B 858D'),
  352. 'ltc': ('8F12 FA7B 9F12 594C','E79E F55B 1536 56F2'),
  353. },
  354. 'addrfile_bech32_chk': {
  355. 'btc': ('2AA3 78DF B965 82EB','5B6A 4D12 820D BC3C'),
  356. },
  357. 'addrfile_compressed_chk': {
  358. 'btc': ('2615 8401 2E98 7ECA','DF38 22AB AAB0 124E'),
  359. 'ltc': ('197C C48C 3C37 AB0F','5072 15DA 1A90 5E99'),
  360. },
  361. 'keyaddrfile_chk': {
  362. 'btc': ('9648 5132 B98E 3AD9','2F72 C83F 44C5 0FAC'),
  363. 'ltc': ('DBD4 FAB6 7E46 CD07','1DA9 C245 F669 670C'),
  364. },
  365. 'keyaddrfile_segwit_chk': {
  366. 'btc': ('C98B DF08 A3D5 204B','25F2 AEB6 AAAC 8BBE'),
  367. 'ltc': ('1829 7FE7 2567 CB91','1305 9007 E515 B66A'),
  368. },
  369. 'keyaddrfile_bech32_chk': {
  370. 'btc': ('4A6B 3762 DF30 9368','A68A 53D4 874E 923C'),
  371. },
  372. 'keyaddrfile_compressed_chk': {
  373. 'btc': ('6D6D 3D35 04FD B9C3','B345 9CD8 9EAE 5489'),
  374. 'ltc': ('F5DA 9D60 6798 C4E9','F928 113B C9D7 9DF5'),
  375. },
  376. 'passfile_chk': 'ADEA 0083 094D 489A',
  377. 'passfile32_chk': '2A28 C5C7 36EC 217A',
  378. 'passfilehex_chk': 'B11C AC6A 1464 608D',
  379. 'wpasswd': 'reference password',
  380. 'ref_wallet': '1378FC64-6F0F9BB4[192,1].mmdat',
  381. 'ic_wallet': '1378FC64-2907DE97-F980D21F[192,1].mmincog',
  382. 'ic_wallet_hex': '1378FC64-4DCB5174-872806A7[192,1].mmincox',
  383. 'hic_wallet': '1378FC64-B55E9958-77256FC1[192,1].incog.offset123',
  384. 'hic_wallet_old': '1378FC64-B55E9958-D85FF20C[192,1].incog-old.offset123',
  385. 'tmpdir': os.path.join('test','tmp7'),
  386. 'kapasswd': '',
  387. 'addr_idx_list': '1010,500-501,31-33,1,33,500,1011', # 8 addresses
  388. 'pass_idx_list': '1,4,9-11,1100',
  389. 'dep_generators': {
  390. 'mmdat': 'refwalletgen2',
  391. pwfile: 'refwalletgen2',
  392. 'addrs': 'refaddrgen2',
  393. 'akeys.mmenc': 'refkeyaddrgen2'
  394. },
  395. 'segwit': get_segwit_bool()
  396. },
  397. '8': {
  398. 'name': 'reference wallet check (256-bit)',
  399. 'seed_len': 256,
  400. 'seed_id': '98831F3A',
  401. 'ref_bw_seed_id': 'B48CD7FC',
  402. 'addrfile_chk': {
  403. 'btc': ('6FEF 6FB9 7B13 5D91','3C2C 8558 BB54 079E'),
  404. 'ltc': ('AD52 C3FE 8924 AAF0','5738 5C4F 167C F9AE'),
  405. },
  406. 'addrfile_segwit_chk': {
  407. 'btc': ('06C1 9C87 F25C 4EE6','58D1 7B6C E9F9 9C14'),
  408. 'ltc': ('63DF E42A 0827 21C3','1A3F 3016 2E2B F33A'),
  409. },
  410. 'addrfile_bech32_chk': {
  411. 'btc': ('9D2A D4B6 5117 F02E','BA07 0DD8 E2A6 2C5A'),
  412. },
  413. 'addrfile_compressed_chk': {
  414. 'btc': ('A33C 4FDE F515 F5BC','5186 02C2 535E B7D5'),
  415. 'ltc': ('3FC0 8F03 C2D6 BD19','535E 5CDC 1CA7 08D5'),
  416. },
  417. 'keyaddrfile_chk': {
  418. 'btc': ('9F2D D781 1812 8BAD','7410 8F95 4B33 B4B2'),
  419. 'ltc': ('B804 978A 8796 3ED4','93A6 844C 8ECC BEF4'),
  420. },
  421. 'keyaddrfile_segwit_chk': {
  422. 'btc': ('A447 12C2 DD14 5A9B','0690 460D A600 D315'),
  423. 'ltc': ('E8A3 9F6E E164 A521','70ED 8557 5882 08A5'),
  424. },
  425. 'keyaddrfile_bech32_chk': {
  426. 'btc': ('D0DD BDE3 87BE 15AE','2C7B 70E5 5F96 9B09'),
  427. },
  428. 'keyaddrfile_compressed_chk': {
  429. 'btc': ('420A 8EB5 A9E2 7814','3243 DD92 809E FE8D'),
  430. 'ltc': ('8D1C 781F EB7F 44BC','678E 8EF9 1396 B140'),
  431. },
  432. 'passfile_chk': '2D6D 8FBA 422E 1315',
  433. 'passfile32_chk': 'F6C1 CDFB 97D9 FCAE',
  434. 'passfilehex_chk': 'BD4F A0AC 8628 4BE4',
  435. 'wpasswd': 'reference password',
  436. 'ref_wallet': '98831F3A-{}[256,1].mmdat'.format(('27F2BF93','E2687906')[g.testnet]),
  437. 'ref_addrfile': '98831F3A{}[1,31-33,500-501,1010-1011]{}.addrs',
  438. 'ref_segwitaddrfile':'98831F3A{}-S[1,31-33,500-501,1010-1011]{}.addrs',
  439. 'ref_bech32addrfile':'98831F3A{}-B[1,31-33,500-501,1010-1011]{}.addrs',
  440. 'ref_keyaddrfile': '98831F3A{}[1,31-33,500-501,1010-1011]{}.akeys.mmenc',
  441. 'ref_passwdfile': '98831F3A-фубар@crypto.org-b58-20[1,4,9-11,1100].pws',
  442. 'ref_addrfile_chksum': {
  443. 'btc': ('6FEF 6FB9 7B13 5D91','3C2C 8558 BB54 079E'),
  444. 'ltc': ('AD52 C3FE 8924 AAF0','5738 5C4F 167C F9AE'),
  445. },
  446. 'ref_segwitaddrfile_chksum': {
  447. 'btc': ('06C1 9C87 F25C 4EE6','58D1 7B6C E9F9 9C14'),
  448. 'ltc': ('63DF E42A 0827 21C3','1A3F 3016 2E2B F33A'),
  449. },
  450. 'ref_bech32addrfile_chksum': {
  451. 'btc': ('9D2A D4B6 5117 F02E','BA07 0DD8 E2A6 2C5A'),
  452. },
  453. 'ref_keyaddrfile_chksum': {
  454. 'btc': ('9F2D D781 1812 8BAD','7410 8F95 4B33 B4B2'),
  455. 'ltc': ('B804 978A 8796 3ED4','93A6 844C 8ECC BEF4'),
  456. },
  457. 'ref_addrfile_chksum_zec': '903E 7225 DD86 6E01',
  458. 'ref_addrfile_chksum_zec_z': '9C7A 72DC 3D4A B3AF',
  459. 'ref_addrfile_chksum_xmr': '4369 0253 AC2C 0E38',
  460. 'ref_addrfile_chksum_dash':'FBC1 6B6A 0988 4403',
  461. 'ref_addrfile_chksum_eth': 'E554 076E 7AF6 66A3',
  462. 'ref_addrfile_chksum_etc': 'E97A D796 B495 E8BC',
  463. 'ref_keyaddrfile_chksum_zec': 'F05A 5A5C 0C8E 2617',
  464. 'ref_keyaddrfile_chksum_zec_z': '4ADB 5AA4 4590 B60A',
  465. 'ref_keyaddrfile_chksum_xmr': 'E0D7 9612 3D67 404A',
  466. 'ref_keyaddrfile_chksum_dash': 'E83D 2C63 FEA2 4142',
  467. 'ref_keyaddrfile_chksum_eth': '3635 4DCF B752 8772',
  468. 'ref_keyaddrfile_chksum_etc': '9BAC 38E7 5C8E 42E0',
  469. 'ref_passwdfile_chksum': 'A983 DAB9 5514 27FB',
  470. # 'ref_fake_unspent_data':'98831F3A_unspent.json',
  471. 'ref_tx_file': {
  472. 'btc': 'FFB367[1.234]{}.rawtx',
  473. 'bch': '99BE60-BCH[106.6789]{}.rawtx',
  474. 'b2x': '6A52BC-B2X[106.6789,tl=1320969600]{}.rawtx',
  475. 'ltc': '75F455-LTC[106.6789]{}.rawtx',
  476. },
  477. 'ic_wallet': '98831F3A-5482381C-18460FB1[256,1].mmincog',
  478. 'ic_wallet_hex': '98831F3A-1630A9F2-870376A9[256,1].mmincox',
  479. 'hic_wallet': '98831F3A-F59B07A0-559CEF19[256,1].incog.offset123',
  480. 'hic_wallet_old': '98831F3A-F59B07A0-848535F3[256,1].incog-old.offset123',
  481. 'tmpdir': os.path.join('test','tmp8'),
  482. 'kapasswd': '',
  483. 'addr_idx_list': '1010,500-501,31-33,1,33,500,1011', # 8 addresses
  484. 'pass_idx_list': '1,4,9-11,1100',
  485. 'dep_generators': {
  486. 'mmdat': 'refwalletgen3',
  487. pwfile: 'refwalletgen3',
  488. 'addrs': 'refaddrgen3',
  489. 'akeys.mmenc': 'refkeyaddrgen3'
  490. },
  491. 'segwit': get_segwit_bool()
  492. },
  493. '9': {
  494. 'tmpdir': os.path.join('test','tmp9'),
  495. 'tool_enc_infn': 'tool_encrypt.in',
  496. # 'tool_enc_ref_infn': 'tool_encrypt_ref.in',
  497. 'wpasswd': 'reference password',
  498. 'dep_generators': {
  499. 'tool_encrypt.in': 'tool_encrypt',
  500. 'tool_encrypt.in.mmenc': 'tool_encrypt',
  501. # 'tool_encrypt_ref.in': 'tool_encrypt_ref',
  502. # 'tool_encrypt_ref.in.mmenc': 'tool_encrypt_ref',
  503. },
  504. },
  505. }
  506. from copy import deepcopy
  507. for a,b in (('6','11'),('7','12'),('8','13')):
  508. cfgs[b] = deepcopy(cfgs[a])
  509. cfgs[b]['tmpdir'] = os.path.join('test','tmp'+b)
  510. from collections import OrderedDict
  511. cmd_group = OrderedDict()
  512. cmd_group['help'] = OrderedDict([
  513. # test description depends
  514. ['helpscreens', (1,'help screens', [],1)],
  515. ['longhelpscreens', (1,'help screens (--longhelp)',[],1)],
  516. ])
  517. cmd_group['dfl_wallet'] = OrderedDict([
  518. ['walletgen_dfl_wallet', (15,'wallet generation (default wallet)',[[[],15]],1)],
  519. ['export_seed_dfl_wallet',(15,'seed export to mmseed format (default wallet)',[[[pwfile],15]],1)],
  520. ['addrgen_dfl_wallet',(15,'address generation (default wallet)',[[[pwfile],15]],1)],
  521. ['txcreate_dfl_wallet',(15,'transaction creation (default wallet)',[[['addrs'],15]],1)],
  522. ['txsign_dfl_wallet',(15,'transaction signing (default wallet)',[[['rawtx',pwfile],15]],1)],
  523. ['passchg_dfl_wallet',(16,'password, label and hash preset change (default wallet)',[[[pwfile],15]],1)],
  524. ['walletchk_newpass_dfl_wallet',(16,'wallet check with new pw, label and hash preset',[[[pwfile],16]],1)],
  525. ['delete_dfl_wallet',(15,'delete default wallet',[[[pwfile],15]],1)],
  526. ])
  527. cmd_group['main'] = OrderedDict([
  528. ['walletgen', (1,'wallet generation', [[['del_dw_run'],15]],1)],
  529. # ['walletchk', (1,'wallet check', [[['mmdat'],1]])],
  530. ['passchg', (5,'password, label and hash preset change',[[['mmdat',pwfile],1]],1)],
  531. ['walletchk_newpass',(5,'wallet check with new pw, label and hash preset',[[['mmdat',pwfile],5]],1)],
  532. ['addrgen', (1,'address generation', [[['mmdat',pwfile],1]],1)],
  533. ['addrimport', (1,'address import', [[['addrs'],1]],1)],
  534. ['txcreate', (1,'transaction creation', [[['addrs'],1]],1)],
  535. ['txbump', (1,'transaction fee bumping (no send)',[[['rawtx'],1]],1)],
  536. ['txsign', (1,'transaction signing', [[['mmdat','rawtx',pwfile,'txbump'],1]],1)],
  537. ['txsend', (1,'transaction sending', [[['sigtx'],1]])],
  538. # txdo must go after txsign
  539. ['txdo', (1,'online transaction', [[['sigtx','mmdat'],1]])],
  540. ['export_hex', (1,'seed export to hexadecimal format', [[['mmdat'],1]])],
  541. ['export_seed', (1,'seed export to mmseed format', [[['mmdat'],1]])],
  542. ['export_mnemonic', (1,'seed export to mmwords format', [[['mmdat'],1]])],
  543. ['export_incog', (1,'seed export to mmincog format', [[['mmdat'],1]])],
  544. ['export_incog_hex',(1,'seed export to mmincog hex format', [[['mmdat'],1]])],
  545. ['export_incog_hidden',(1,'seed export to hidden mmincog format', [[['mmdat'],1]])],
  546. ['addrgen_hex', (1,'address generation from mmhex file', [[['mmhex','addrs'],1]])],
  547. ['addrgen_seed', (1,'address generation from mmseed file', [[['mmseed','addrs'],1]])],
  548. ['addrgen_mnemonic',(1,'address generation from mmwords file',[[['mmwords','addrs'],1]])],
  549. ['addrgen_incog', (1,'address generation from mmincog file',[[['mmincog','addrs'],1]])],
  550. ['addrgen_incog_hex',(1,'address generation from mmincog hex file',[[['mmincox','addrs'],1]])],
  551. ['addrgen_incog_hidden',(1,'address generation from hidden mmincog file', [[[hincog_fn,'addrs'],1]])],
  552. ['keyaddrgen', (1,'key-address file generation', [[['mmdat',pwfile],1]])],
  553. ['txsign_keyaddr',(1,'transaction signing with key-address file', [[['akeys.mmenc','rawtx'],1]])],
  554. ['walletgen2',(2,'wallet generation (2), 128-bit seed', [[['del_dw_run'],15]])],
  555. ['addrgen2', (2,'address generation (2)', [[['mmdat'],2]])],
  556. ['txcreate2', (2,'transaction creation (2)', [[['addrs'],2]])],
  557. ['txsign2', (2,'transaction signing, two transactions',[[['mmdat','rawtx'],1],[['mmdat','rawtx'],2]])],
  558. ['export_mnemonic2', (2,'seed export to mmwords format (2)',[[['mmdat'],2]])],
  559. ['walletgen3',(3,'wallet generation (3)', [[['del_dw_run'],15]])],
  560. ['addrgen3', (3,'address generation (3)', [[['mmdat'],3]])],
  561. ['txcreate3', (3,'tx creation with inputs and outputs from two wallets', [[['addrs'],1],[['addrs'],3]])],
  562. ['txsign3', (3,'tx signing with inputs and outputs from two wallets',[[['mmdat'],1],[['mmdat','rawtx'],3]])],
  563. ['walletgen14', (14,'wallet generation (14)', [[['del_dw_run'],15]],14)],
  564. ['addrgen14', (14,'address generation (14)', [[['mmdat'],14]])],
  565. ['keyaddrgen14',(14,'key-address file generation (14)', [[['mmdat'],14]],14)],
  566. ['walletgen4',(4,'wallet generation (4) (brainwallet)', [[['del_dw_run'],15]])],
  567. ['addrgen4', (4,'address generation (4)', [[['mmdat'],4]])],
  568. ['txcreate4', (4,'tx creation with inputs and outputs from four seed sources, key-address file and non-MMGen inputs and outputs', [[['addrs'],1],[['addrs'],2],[['addrs'],3],[['addrs'],4],[['addrs','akeys.mmenc'],14]])],
  569. ['txsign4', (4,'tx signing with inputs and outputs from incog file, mnemonic file, wallet, brainwallet, key-address file and non-MMGen inputs and outputs', [[['mmincog'],1],[['mmwords'],2],[['mmdat'],3],[['mmbrain','rawtx'],4],[['akeys.mmenc'],14]])],
  570. ['txdo4', (4,'tx creation,signing and sending with inputs and outputs from four seed sources, key-address file and non-MMGen inputs and outputs', [[['addrs'],1],[['addrs'],2],[['addrs'],3],[['addrs'],4],[['addrs','akeys.mmenc'],14],[['mmincog'],1],[['mmwords'],2],[['mmdat'],3],[['mmbrain','rawtx'],4],[['akeys.mmenc'],14]])], # must go after txsign4
  571. ['txbump4', (4,'tx fee bump + send with inputs and outputs from four seed sources, key-address file and non-MMGen inputs and outputs', [[['akeys.mmenc'],14],[['mmincog'],1],[['mmwords'],2],[['mmdat'],3],[['akeys.mmenc'],14],[['mmbrain','sigtx','mmdat','txdo'],4]])], # must go after txsign4
  572. ])
  573. cmd_group['tool'] = OrderedDict([
  574. ['tool_encrypt', (9,"'mmgen-tool encrypt' (random data)", [],1)],
  575. ['tool_decrypt', (9,"'mmgen-tool decrypt' (random data)", [[[cfgs['9']['tool_enc_infn'],cfgs['9']['tool_enc_infn']+'.mmenc'],9]],1)],
  576. # ['tool_encrypt_ref', (9,"'mmgen-tool encrypt' (reference text)", [])],
  577. ['tool_find_incog_data', (9,"'mmgen-tool find_incog_data'", [[[hincog_fn],1],[[incog_id_fn],1]])],
  578. # ['pywallet', (9,"'mmgen-pywallet'", [],1)],
  579. ])
  580. # saved reference data
  581. cmd_group['ref'] = (
  582. # reading
  583. ('ref_wallet_chk', ([],'saved reference wallet')),
  584. ('ref_seed_chk', ([],'saved seed file')),
  585. ('ref_hex_chk', ([],'saved mmhex file')),
  586. ('ref_mn_chk', ([],'saved mnemonic file')),
  587. ('ref_hincog_chk', ([],'saved hidden incog reference wallet')),
  588. ('ref_brain_chk', ([],'saved brainwallet')),
  589. # generating new reference ('abc' brainwallet) files:
  590. ('refwalletgen', ([],'gen new refwallet')),
  591. ('refaddrgen', (['mmdat',pwfile],'new refwallet addr chksum')),
  592. ('refkeyaddrgen', (['mmdat',pwfile],'new refwallet key-addr chksum')),
  593. ('refaddrgen_compressed', (['mmdat',pwfile],'new refwallet addr chksum (compressed)')),
  594. ('refkeyaddrgen_compressed', (['mmdat',pwfile],'new refwallet key-addr chksum (compressed)')),
  595. ('refpasswdgen', (['mmdat',pwfile],'new refwallet passwd file chksum')),
  596. ('ref_b32passwdgen',(['mmdat',pwfile],'new refwallet passwd file chksum (base32)')),
  597. ('ref_hexpasswdgen',(['mmdat',pwfile],'new refwallet passwd file chksum (base32)')),
  598. )
  599. # misc. saved reference data
  600. cmd_group['ref_other'] = (
  601. ('ref_addrfile_chk', 'saved reference address file'),
  602. ('ref_segwitaddrfile_chk','saved reference address file (segwit)'),
  603. ('ref_bech32addrfile_chk','saved reference address file (bech32)'),
  604. ('ref_keyaddrfile_chk','saved reference key-address file'),
  605. ('ref_passwdfile_chk', 'saved reference password file'),
  606. # Create the fake inputs:
  607. # ('txcreate8', 'transaction creation (8)'),
  608. ('ref_tx_chk', 'saved reference tx file'),
  609. ('ref_brain_chk_spc3', 'saved brainwallet (non-standard spacing)'),
  610. ('ref_tool_decrypt', 'decryption of saved MMGen-encrypted file'),
  611. )
  612. # mmgen-walletconv:
  613. cmd_group['conv_in'] = ( # reading
  614. ('ref_wallet_conv', 'conversion of saved reference wallet'),
  615. ('ref_mn_conv', 'conversion of saved mnemonic'),
  616. ('ref_seed_conv', 'conversion of saved seed file'),
  617. ('ref_hex_conv', 'conversion of saved hexadecimal seed file'),
  618. ('ref_brain_conv', 'conversion of ref brainwallet'),
  619. ('ref_incog_conv', 'conversion of saved incog wallet'),
  620. ('ref_incox_conv', 'conversion of saved hex incog wallet'),
  621. ('ref_hincog_conv', 'conversion of saved hidden incog wallet'),
  622. ('ref_hincog_conv_old','conversion of saved hidden incog wallet (old format)')
  623. )
  624. cmd_group['conv_out'] = ( # writing
  625. ('ref_wallet_conv_out', 'ref seed conversion to wallet'),
  626. ('ref_mn_conv_out', 'ref seed conversion to mnemonic'),
  627. ('ref_hex_conv_out', 'ref seed conversion to hex seed'),
  628. ('ref_seed_conv_out', 'ref seed conversion to seed'),
  629. ('ref_incog_conv_out', 'ref seed conversion to incog data'),
  630. ('ref_incox_conv_out', 'ref seed conversion to hex incog data'),
  631. ('ref_hincog_conv_out', 'ref seed conversion to hidden incog data')
  632. )
  633. cmd_group['regtest'] = (
  634. ('regtest_setup', 'regtest (Bob and Alice) mode setup'),
  635. ('regtest_walletgen_bob', 'wallet generation (Bob)'),
  636. ('regtest_walletgen_alice', 'wallet generation (Alice)'),
  637. ('regtest_addrgen_bob', 'address generation (Bob)'),
  638. ('regtest_addrgen_alice', 'address generation (Alice)'),
  639. ('regtest_addrimport_bob', "importing Bob's addresses"),
  640. ('regtest_addrimport_alice', "importing Alice's addresses"),
  641. ('regtest_fund_bob', "funding Bob's wallet"),
  642. ('regtest_fund_alice', "funding Alice's wallet"),
  643. ('regtest_bob_bal1', "Bob's balance"),
  644. ('regtest_bob_split1', "splitting Bob's funds"),
  645. ('regtest_generate', 'mining a block'),
  646. ('regtest_bob_bal2', "Bob's balance"),
  647. ('regtest_bob_rbf_send', 'sending funds to Alice (RBF)'),
  648. ('regtest_get_mempool1', 'mempool (before RBF bump)'),
  649. ('regtest_bob_rbf_bump', 'bumping RBF transaction'),
  650. ('regtest_get_mempool2', 'mempool (after RBF bump)'),
  651. ('regtest_generate', 'mining a block'),
  652. ('regtest_bob_bal3', "Bob's balance"),
  653. ('regtest_bob_pre_import', 'sending to non-imported address'),
  654. ('regtest_generate', 'mining a block'),
  655. ('regtest_bob_import_addr', 'importing non-MMGen address with --rescan'),
  656. ('regtest_bob_bal4', "Bob's balance (after import with rescan)"),
  657. ('regtest_bob_import_list', 'importing flat address list'),
  658. ('regtest_bob_split2', "splitting Bob's funds"),
  659. ('regtest_generate', 'mining a block'),
  660. ('regtest_bob_bal5', "Bob's balance"),
  661. ('regtest_bob_send_non_mmgen', 'sending funds to Alice (from non-MMGen addrs)'),
  662. ('regtest_generate', 'mining a block'),
  663. ('regtest_bob_alice_bal', "Bob and Alice's balances"),
  664. ('regtest_alice_add_label1', 'adding a label'),
  665. ('regtest_alice_chk_label1', 'the label'),
  666. ('regtest_alice_add_label2', 'adding a label'),
  667. ('regtest_alice_chk_label2', 'the label'),
  668. ('regtest_alice_edit_label1', 'editing a label'),
  669. ('regtest_alice_chk_label3', 'the label'),
  670. ('regtest_alice_remove_label1','removing a label'),
  671. ('regtest_alice_chk_label4', 'the label'),
  672. ('regtest_alice_send_estimatefee','tx creation with no fee on command line'),
  673. ('regtest_alice_add_label_coinaddr','adding a label using the coin address'),
  674. ('regtest_alice_chk_label_coinaddr','the label'),
  675. ('regtest_alice_add_label_badaddr1','adding a label with invalid address'),
  676. ('regtest_alice_add_label_badaddr2','adding a label with invalid address for this chain'),
  677. ('regtest_alice_add_label_badaddr3','adding a label with wrong MMGen address'),
  678. ('regtest_alice_add_label_badaddr4','adding a label with wrong coin address'),
  679. ('regtest_alice_add_label_rpcfail','RPC failure code'),
  680. ('regtest_stop', 'stopping regtest daemon'),
  681. )
  682. cmd_group['regtest_split'] = (
  683. ('regtest_split_setup', 'regtest forking scenario setup'),
  684. ('regtest_walletgen_bob', "generating Bob's wallet"),
  685. ('regtest_addrgen_bob', "generating Bob's addresses"),
  686. ('regtest_addrimport_bob', "importing Bob's addresses"),
  687. ('regtest_fund_bob', "funding Bob's wallet"),
  688. ('regtest_split_fork', 'regtest split fork'),
  689. ('regtest_split_start_btc', 'start regtest daemon (BTC)'),
  690. ('regtest_split_start_b2x', 'start regtest daemon (B2X)'),
  691. ('regtest_split_gen_btc', 'mining a block (BTC)'),
  692. ('regtest_split_gen_b2x', 'mining 100 blocks (B2X)'),
  693. ('regtest_split_do_split', 'creating coin splitting transactions'),
  694. ('regtest_split_sign_b2x', 'signing B2X split transaction'),
  695. ('regtest_split_sign_btc', 'signing BTC split transaction'),
  696. ('regtest_split_send_b2x', 'sending B2X split transaction'),
  697. ('regtest_split_send_btc', 'sending BTC split transaction'),
  698. ('regtest_split_gen_btc', 'mining a block (BTC)'),
  699. ('regtest_split_gen_b2x2', 'mining a block (B2X)'),
  700. ('regtest_split_txdo_timelock_bad_btc', 'sending transaction with bad locktime (BTC)'),
  701. ('regtest_split_txdo_timelock_good_btc','sending transaction with good locktime (BTC)'),
  702. ('regtest_split_txdo_timelock_bad_b2x', 'sending transaction with bad locktime (B2X)'),
  703. ('regtest_split_txdo_timelock_good_b2x','sending transaction with good locktime (B2X)'),
  704. )
  705. cmd_group['misc'] = (
  706. ('autosign', 'transaction autosigning (BTC,BCH,LTC)'),
  707. )
  708. cmd_group['altcoin_ref'] = (
  709. ('ref_addrfile_chk_eth', 'reference address file (ETH)'),
  710. ('ref_addrfile_chk_etc', 'reference address file (ETC)'),
  711. ('ref_addrfile_chk_dash','reference address file (DASH)'),
  712. ('ref_addrfile_chk_zec', 'reference address file (ZEC-T)'),
  713. ('ref_addrfile_chk_xmr', 'reference address file (XMR)'),
  714. ('ref_addrfile_chk_zec_z','reference address file (ZEC-Z)'),
  715. ('ref_keyaddrfile_chk_eth', 'reference key-address file (ETH)'),
  716. ('ref_keyaddrfile_chk_etc', 'reference key-address file (ETC)'),
  717. ('ref_keyaddrfile_chk_dash','reference key-address file (DASH)'),
  718. ('ref_keyaddrfile_chk_zec', 'reference key-address file (ZEC-T)'),
  719. ('ref_keyaddrfile_chk_zec_z','reference key-address file (ZEC-Z)'),
  720. ('ref_keyaddrfile_chk_xmr', 'reference key-address file (XMR)'),
  721. )
  722. # undocumented admin cmds - precede with 'admin'
  723. cmd_group_admin = OrderedDict()
  724. cmd_group_admin['create_ref_tx'] = (
  725. ('ref_tx_setup', 'regtest (Bob and Alice) mode setup'),
  726. ('ref_tx_addrgen_bob_ref_wallet', 'address generation (Bob - reference wallet)'),
  727. ('ref_tx_addrimport_bob_ref_wallet', "importing Bob's addresses (reference wallet)"),
  728. ('ref_tx_fund_bob', "funding Bob's wallet (reference wallet)"),
  729. ('ref_tx_bob_split', "splitting Bob's funds (reference wallet)"),
  730. ('ref_tx_generate', 'mining a block'),
  731. ('ref_tx_bob_create_tx', "creating reference transaction"),
  732. ('ref_tx_bob_modify_tx', "modifying reference transaction (testnet+mainnet)"),
  733. )
  734. cmd_list_admin = OrderedDict()
  735. cmd_data_admin = OrderedDict()
  736. for k in cmd_group_admin: cmd_list_admin[k] = []
  737. cmd_data_admin['info_create_ref_tx'] = 'create reference tx',[8]
  738. for a,b in cmd_group_admin['create_ref_tx']:
  739. cmd_list_admin['create_ref_tx'].append(a)
  740. cmd_data_admin[a] = (8,b,[[[],8]])
  741. # end undocumented admin commands
  742. cmd_list = OrderedDict()
  743. for k in cmd_group: cmd_list[k] = []
  744. cmd_data = OrderedDict()
  745. for k,v in (
  746. ('help', ('help screens',[])),
  747. ('dfl_wallet', ('basic operations with default wallet',[15,16])),
  748. ('main', ('basic operations',[1,2,3,4,5,15,16])),
  749. ('tool', ('tools',[9]))
  750. ):
  751. cmd_data['info_'+k] = v
  752. for i in cmd_group[k]:
  753. cmd_list[k].append(i)
  754. cmd_data[i] = cmd_group[k][i]
  755. cmd_data['info_ref'] = 'reference data',[6,7,8]
  756. for a,b in cmd_group['ref']:
  757. for i,j in ((1,128),(2,192),(3,256)):
  758. k = a+str(i)
  759. cmd_list['ref'].append(k)
  760. cmd_data[k] = (5+i,'%s (%s-bit)' % (b[1],j),[[b[0],5+i]])
  761. cmd_data['info_ref_other'] = 'other reference data',[8]
  762. for a,b in cmd_group['ref_other']:
  763. cmd_list['ref_other'].append(a)
  764. cmd_data[a] = (8,b,[[[],8]])
  765. cmd_data['info_conv_in'] = 'wallet conversion from reference data',[11,12,13]
  766. for a,b in cmd_group['conv_in']:
  767. for i,j in ((1,128),(2,192),(3,256)):
  768. k = a+str(i)
  769. cmd_list['conv_in'].append(k)
  770. cmd_data[k] = (10+i,'%s (%s-bit)' % (b,j),[[[],10+i]])
  771. cmd_data['info_conv_out'] = 'wallet conversion to reference data',[11,12,13]
  772. for a,b in cmd_group['conv_out']:
  773. for i,j in ((1,128),(2,192),(3,256)):
  774. k = a+str(i)
  775. cmd_list['conv_out'].append(k)
  776. cmd_data[k] = (10+i,'%s (%s-bit)' % (b,j),[[[],10+i]])
  777. cmd_data['info_regtest'] = 'regtest mode',[17]
  778. for a,b in cmd_group['regtest']:
  779. cmd_list['regtest'].append(a)
  780. cmd_data[a] = (17,b,[[[],17]])
  781. # disable until B2X officially supported
  782. # cmd_data['info_regtest_split'] = 'regtest mode with fork and coin split',[17]
  783. # for a,b in cmd_group['regtest_split']:
  784. # cmd_list['regtest_split'].append(a)
  785. # cmd_data[a] = (19,b,[[[],19]])
  786. #
  787. cmd_data['info_misc'] = 'miscellaneous operations',[18]
  788. for a,b in cmd_group['misc']:
  789. cmd_list['misc'].append(a)
  790. cmd_data[a] = (18,b,[[[],18]])
  791. cmd_data['info_altcoin_ref'] = 'altcoin reference files',[8]
  792. for a,b in cmd_group['altcoin_ref']:
  793. cmd_list['altcoin_ref'].append(a)
  794. cmd_data[a] = (8,b,[[[],8]])
  795. utils = {
  796. 'check_deps': 'check dependencies for specified command',
  797. 'clean': 'clean specified tmp dir(s) 1,2,3,4,5 or 6 (no arg = all dirs)',
  798. }
  799. addrs_per_wallet = 8
  800. meta_cmds = OrderedDict([
  801. ['ref1', ('refwalletgen1','refaddrgen1','refkeyaddrgen1')],
  802. ['ref2', ('refwalletgen2','refaddrgen2','refkeyaddrgen2')],
  803. ['ref3', ('refwalletgen3','refaddrgen3','refkeyaddrgen3')],
  804. ['gen', ('walletgen','addrgen')],
  805. ['pass', ('passchg','walletchk_newpass')],
  806. ['tx', ('addrimport','txcreate','txsign','txsend')],
  807. ['export', [k for k in cmd_data if k[:7] == 'export_' and cmd_data[k][0] == 1]],
  808. ['gen_sp', [k for k in cmd_data if k[:8] == 'addrgen_' and cmd_data[k][0] == 1]],
  809. ['online', ('keyaddrgen','txsign_keyaddr')],
  810. ['2', [k for k in cmd_data if cmd_data[k][0] == 2]],
  811. ['3', [k for k in cmd_data if cmd_data[k][0] == 3]],
  812. ['4', [k for k in cmd_data if cmd_data[k][0] == 4]],
  813. ['saved_ref1', [c[0]+'1' for c in cmd_group['ref']]],
  814. ['saved_ref2', [c[0]+'2' for c in cmd_group['ref']]],
  815. ['saved_ref3', [c[0]+'3' for c in cmd_group['ref']]],
  816. ['saved_ref_other', [c[0] for c in cmd_group['ref_other']]],
  817. ['saved_ref_conv_in1', [c[0]+'1' for c in cmd_group['conv_in']]],
  818. ['saved_ref_conv_in2', [c[0]+'2' for c in cmd_group['conv_in']]],
  819. ['saved_ref_conv_in3', [c[0]+'3' for c in cmd_group['conv_in']]],
  820. ['saved_ref_conv_out1', [c[0]+'1' for c in cmd_group['conv_out']]],
  821. ['saved_ref_conv_out2', [c[0]+'2' for c in cmd_group['conv_out']]],
  822. ['saved_ref_conv_out3', [c[0]+'3' for c in cmd_group['conv_out']]],
  823. ['regtest', dict(cmd_group['regtest']).keys()],
  824. ])
  825. del cmd_group
  826. if opt.profile: opt.names = True
  827. if opt.resume: opt.skip_deps = True
  828. if opt.log:
  829. log_fd = open(log_file,'a')
  830. log_fd.write('\nLog started: %s\n' % make_timestr())
  831. usr_rand_chars = (5,30)[bool(opt.usr_random)]
  832. usr_rand_arg = '-r%s' % usr_rand_chars
  833. cmd_total = 0
  834. # Disable color in spawned scripts so we can parse their output
  835. os.environ['MMGEN_DISABLE_COLOR'] = '1'
  836. os.environ['MMGEN_NO_LICENSE'] = '1'
  837. os.environ['MMGEN_MIN_URANDCHARS'] = '3'
  838. os.environ['MMGEN_BOGUS_SEND'] = '1'
  839. def get_segwit_arg(cfg):
  840. return ['--type='+('segwit','bech32')[bool(opt.bech32)]] if cfg['segwit'] else []
  841. # Tell spawned programs they're running in the test suite
  842. os.environ['MMGEN_TEST_SUITE'] = '1'
  843. if opt.debug_scripts: os.environ['MMGEN_DEBUG'] = '1'
  844. if opt.exact_output:
  845. def msg(s): pass
  846. vmsg = vmsg_r = msg_r = msg
  847. else:
  848. def msg(s): sys.stderr.write(s+'\n')
  849. def vmsg(s):
  850. if opt.verbose: sys.stderr.write(s+'\n')
  851. def msg_r(s): sys.stderr.write(s)
  852. def vmsg_r(s):
  853. if opt.verbose: sys.stderr.write(s)
  854. stderr_save = sys.stderr
  855. def silence():
  856. if not (opt.verbose or opt.exact_output):
  857. f = ('/dev/null','stderr.out')[g.platform=='win']
  858. sys.stderr = open(f,'a')
  859. def end_silence():
  860. if not (opt.verbose or opt.exact_output):
  861. sys.stderr = stderr_save
  862. def errmsg(s): stderr_save.write(s+'\n')
  863. def errmsg_r(s): stderr_save.write(s)
  864. if opt.list_cmds:
  865. from mmgen.term import get_terminal_size
  866. tw = get_terminal_size()[0]
  867. fs = ' {:<{w}} - {}'
  868. Msg(green('AVAILABLE COMMANDS:'))
  869. w = max([len(i) for i in cmd_data])
  870. for cmd in cmd_data:
  871. if cmd[:5] == 'info_':
  872. m = capfirst(cmd_data[cmd][0])
  873. Msg(green(' %s:' % m))
  874. continue
  875. Msg(' '+fs.format(cmd,cmd_data[cmd][1],w=w))
  876. for cl,lbl in ((meta_cmds,'METACOMMANDS'),(cmd_list,'COMMAND GROUPS')):
  877. w = max([len(i) for i in cl])
  878. Msg('\n'+green('AVAILABLE {}:'.format(lbl)))
  879. for cmd in cl:
  880. ft = format_par(' '.join(cl[cmd]),width=tw,indent=4,as_list=True)
  881. sep = '' if not ft else ' ' if len(ft[0]) + len(cmd) < tw - 4 else '\n '
  882. Msg(' {}{}{}'.format(yellow(cmd+':'),sep,'\n'.join(ft).lstrip()))
  883. Msg('\n'+green('AVAILABLE UTILITIES:'))
  884. w = max([len(i) for i in utils])
  885. for cmd in sorted(utils):
  886. Msg(fs.format(cmd,utils[cmd],w=w))
  887. sys.exit(0)
  888. NL = ('\r\n','\n')[g.platform=='linux' and bool(opt.popen_spawn)]
  889. def get_file_with_ext(ext,mydir,delete=True,no_dot=False,return_list=False):
  890. dot = ('.','')[bool(no_dot)]
  891. flist = [os.path.join(mydir,f) for f in os.listdir(mydir)
  892. if f == ext or f[-len(dot+ext):] == dot+ext]
  893. if not flist: return False
  894. if return_list: return flist
  895. if len(flist) > 1:
  896. if delete:
  897. if not opt.quiet:
  898. msg("Multiple *.{} files in '{}' - deleting".format(ext,mydir))
  899. for f in flist:
  900. msg(f)
  901. os.unlink(f)
  902. return False
  903. else:
  904. return flist[0]
  905. def find_generated_exts(cmd):
  906. out = []
  907. for k in cfgs:
  908. for ext,prog in cfgs[k]['dep_generators'].items():
  909. if prog == cmd:
  910. out.append((ext,cfgs[k]['tmpdir']))
  911. return out
  912. def get_addrfile_checksum(display=False):
  913. addrfile = get_file_with_ext('addrs',cfg['tmpdir'])
  914. silence()
  915. chk = AddrList(addrfile).chksum
  916. if opt.verbose and display: msg('Checksum: %s' % cyan(chk))
  917. end_silence()
  918. return chk
  919. def verify_checksum_or_exit(checksum,chk):
  920. if checksum != chk:
  921. errmsg(red('Checksum error: %s' % chk))
  922. sys.exit(1)
  923. vmsg(green('Checksums match: %s') % (cyan(chk)))
  924. from test.mmgen_pexpect import MMGenPexpect
  925. class MMGenExpect(MMGenPexpect):
  926. def __init__(self,name,mmgen_cmd,cmd_args=[],extra_desc='',no_output=False,msg_only=False,no_msg=False):
  927. desc = ((cmd_data[name][1],name)[bool(opt.names)] + (' ' + extra_desc)).strip()
  928. passthru_args = ['testnet','rpc_host','rpc_port','regtest','coin']
  929. if not opt.system:
  930. os.environ['PYTHONPATH'] = repo_root
  931. mmgen_cmd = os.path.relpath(os.path.join(repo_root,'cmds',mmgen_cmd))
  932. elif g.platform == 'win':
  933. mmgen_cmd = os.path.join('/mingw64','opt','bin',mmgen_cmd)
  934. return MMGenPexpect.__init__(
  935. self,
  936. name,
  937. mmgen_cmd,
  938. cmd_args,
  939. desc,
  940. no_output=no_output,
  941. passthru_args=passthru_args,
  942. msg_only=msg_only,
  943. no_msg=no_msg)
  944. def create_fake_unspent_entry(coinaddr,al_id=None,idx=None,lbl=None,non_mmgen=False,segwit=False):
  945. if 'S' not in g.proto.mmtypes: segwit = False
  946. if lbl: lbl = ' ' + lbl
  947. spk_beg,spk_end = (
  948. ('76a914','88ac'),
  949. ('a914','87'),
  950. (g.proto.witness_vernum_hex+'14','')
  951. )[segwit and (coinaddr.addr_fmt=='p2sh') + 2*(coinaddr.addr_fmt=='bech32')]
  952. amt1,amt2 = {'btc':(10,40),'bch':(10,40),'ltc':(1000,4000)}[coin_sel]
  953. return {
  954. 'account': '{}:{}'.format(g.proto.base_coin.lower(),coinaddr) if non_mmgen \
  955. else (u'{}:{}{}'.format(al_id,idx,lbl.decode('utf8'))),
  956. 'vout': int(getrandnum(4) % 8),
  957. 'txid': hexlify(os.urandom(32)).decode('utf8'),
  958. 'amount': g.proto.coin_amt('%s.%s' % (amt1+(getrandnum(4) % amt2), getrandnum(4) % 100000000)),
  959. 'address': coinaddr,
  960. 'spendable': False,
  961. 'scriptPubKey': '{}{}{}'.format(spk_beg,coinaddr.hex,spk_end),
  962. 'confirmations': getrandnum(4) % 50000
  963. }
  964. labels = [
  965. "Automotive",
  966. "Travel expenses",
  967. "Healthcare",
  968. "Freelancing 1",
  969. "Freelancing 2",
  970. "Alice's allowance",
  971. "Bob's bequest",
  972. "House purchase",
  973. "Real estate fund",
  974. "Job 1",
  975. "XYZ Corp.",
  976. "Eddie's endowment",
  977. "Emergency fund",
  978. "Real estate fund",
  979. "Ian's inheritance",
  980. "",
  981. "Rainy day",
  982. "Fred's funds",
  983. "Job 2",
  984. "Carl's capital",
  985. ]
  986. label_iter = None
  987. def create_fake_unspent_data(adata,tx_data,non_mmgen_input=''):
  988. out = []
  989. for d in tx_data.values():
  990. al = adata.addrlist(d['al_id'])
  991. for n,(idx,coinaddr) in enumerate(al.addrpairs()):
  992. while True:
  993. try: lbl = next(label_iter)
  994. except: label_iter = iter(labels)
  995. else: break
  996. out.append(create_fake_unspent_entry(coinaddr,d['al_id'],idx,lbl,segwit=d['segwit']))
  997. if n == 0: # create a duplicate address. This means addrs_per_wallet += 1
  998. out.append(create_fake_unspent_entry(coinaddr,d['al_id'],idx,lbl,segwit=d['segwit']))
  999. if non_mmgen_input:
  1000. privkey = PrivKey(os.urandom(32),compressed=True,pubkey_type='std')
  1001. rand_coinaddr = AddrGenerator('p2pkh').to_addr(KeyGenerator('std').to_pubhex(privkey))
  1002. of = os.path.join(cfgs[non_mmgen_input]['tmpdir'],non_mmgen_fn)
  1003. write_data_to_file(of,privkey.wif+'\n','compressed {} key'.format(g.proto.name),silent=True)
  1004. out.append(create_fake_unspent_entry(rand_coinaddr,non_mmgen=True,segwit=False))
  1005. # msg('\n'.join([repr(o) for o in out])); sys.exit(0)
  1006. return out
  1007. def write_fake_data_to_file(d):
  1008. unspent_data_file = os.path.join(cfg['tmpdir'],'unspent.json')
  1009. write_data_to_file(unspent_data_file,d,'Unspent outputs',silent=True)
  1010. os.environ['MMGEN_BOGUS_WALLET_DATA'] = unspent_data_file
  1011. bwd_msg = 'MMGEN_BOGUS_WALLET_DATA=%s' % unspent_data_file
  1012. if opt.print_cmdline: msg(bwd_msg)
  1013. if opt.log: log_fd.write(bwd_msg + ' ')
  1014. if opt.verbose or opt.exact_output:
  1015. sys.stderr.write("Fake transaction wallet data written to file '%s'\n" % unspent_data_file)
  1016. def create_tx_data(sources):
  1017. tx_data,ad = {},AddrData()
  1018. for s in sources:
  1019. afile = get_file_with_ext('addrs',cfgs[s]['tmpdir'])
  1020. al = AddrList(afile)
  1021. ad.add(al)
  1022. aix = AddrIdxList(fmt_str=cfgs[s]['addr_idx_list'])
  1023. if len(aix) != addrs_per_wallet:
  1024. errmsg(red('Address index list length != %s: %s' %
  1025. (addrs_per_wallet,repr(aix))))
  1026. sys.exit(0)
  1027. tx_data[s] = {
  1028. 'addrfile': afile,
  1029. 'chk': al.chksum,
  1030. 'al_id': al.al_id,
  1031. 'addr_idxs': aix[-2:],
  1032. 'segwit': cfgs[s]['segwit']
  1033. }
  1034. return ad,tx_data
  1035. def make_txcreate_cmdline(tx_data):
  1036. privkey = PrivKey(os.urandom(32),compressed=True,pubkey_type='std')
  1037. t = ('p2pkh','segwit')['S' in g.proto.mmtypes]
  1038. rand_coinaddr = AddrGenerator(t).to_addr(KeyGenerator('std').to_pubhex(privkey))
  1039. # total of two outputs must be < 10 BTC (<1000 LTC)
  1040. mods = {'btc':(6,4),'bch':(6,4),'ltc':(600,400)}[coin_sel]
  1041. for k in cfgs:
  1042. cfgs[k]['amts'] = [None,None]
  1043. for idx,mod in enumerate(mods):
  1044. cfgs[k]['amts'][idx] = '%s.%s' % ((getrandnum(4) % mod), str(getrandnum(4))[:5])
  1045. cmd_args = ['-d',cfg['tmpdir']]
  1046. for num in tx_data:
  1047. s = tx_data[num]
  1048. cmd_args += [
  1049. '{}:{},{}'.format(s['al_id'],s['addr_idxs'][0],cfgs[num]['amts'][0]),
  1050. ]
  1051. # + one change address and one BTC address
  1052. if num is tx_data.keys()[-1]:
  1053. cmd_args += ['{}:{}'.format(s['al_id'],s['addr_idxs'][1])]
  1054. cmd_args += ['{},{}'.format(rand_coinaddr,cfgs[num]['amts'][1])]
  1055. return cmd_args + [tx_data[num]['addrfile'] for num in tx_data]
  1056. def add_comments_to_addr_file(addrfile,outfile):
  1057. silence()
  1058. msg(green("Adding comments to address file '%s'" % addrfile))
  1059. a = AddrList(addrfile)
  1060. for n,idx in enumerate(a.idxs(),1):
  1061. if n % 2: a.set_comment(idx,'Test address %s' % n)
  1062. a.format(enable_comments=True)
  1063. write_data_to_file(outfile,a.fmt_data,silent=True)
  1064. end_silence()
  1065. def make_brainwallet_file(fn):
  1066. # Print random words with random whitespace in between
  1067. from mmgen.mn_tirosh import words
  1068. wl = words.split()
  1069. nwords,ws_list,max_spaces = 10,' \n',5
  1070. def rand_ws_seq():
  1071. nchars = getrandnum(1) % max_spaces + 1
  1072. return ''.join([ws_list[getrandnum(1)%len(ws_list)] for i in range(nchars)])
  1073. rand_pairs = [wl[getrandnum(4) % len(wl)] + rand_ws_seq() for i in range(nwords)]
  1074. d = ''.join(rand_pairs).rstrip() + '\n'
  1075. if opt.verbose: msg_r('Brainwallet password:\n%s' % cyan(d))
  1076. write_data_to_file(fn,d,'brainwallet password',silent=True)
  1077. def do_between():
  1078. if opt.pause:
  1079. if keypress_confirm(green('Continue?'),default_yes=True):
  1080. if opt.verbose or opt.exact_output: sys.stderr.write('\n')
  1081. else:
  1082. errmsg('Exiting at user request')
  1083. sys.exit(0)
  1084. elif opt.verbose or opt.exact_output:
  1085. sys.stderr.write('\n')
  1086. rebuild_list = OrderedDict()
  1087. def check_needs_rerun(
  1088. ts,
  1089. cmd,
  1090. build=False,
  1091. root=True,
  1092. force_delete=False,
  1093. dpy=False
  1094. ):
  1095. rerun = (False,True)[root] # force_delete is not passed to recursive call
  1096. fns = []
  1097. if force_delete or not root:
  1098. # does cmd produce a needed dependency(ies)?
  1099. ret = ts.get_num_exts_for_cmd(cmd,dpy)
  1100. if ret:
  1101. for ext in ret[1]:
  1102. fn = get_file_with_ext(ext,cfgs[ret[0]]['tmpdir'],delete=build)
  1103. if fn:
  1104. if force_delete: os.unlink(fn)
  1105. else: fns.append(fn)
  1106. else: rerun = True
  1107. fdeps = ts.generate_file_deps(cmd)
  1108. cdeps = ts.generate_cmd_deps(fdeps)
  1109. # print 'cmd,fdeps,cdeps,fns: ',cmd,fdeps,cdeps,fns # DEBUG
  1110. for fn in fns:
  1111. my_age = os.stat(fn).st_mtime
  1112. for num,ext in fdeps:
  1113. f = get_file_with_ext(ext,cfgs[num]['tmpdir'],delete=build)
  1114. if f and os.stat(f).st_mtime > my_age:
  1115. rerun = True
  1116. for cdep in cdeps:
  1117. if check_needs_rerun(ts,cdep,build=build,root=False,dpy=cmd):
  1118. rerun = True
  1119. if build:
  1120. if rerun:
  1121. for fn in fns:
  1122. if not root: os.unlink(fn)
  1123. if not (dpy and opt.skip_deps):
  1124. ts.do_cmd(cmd)
  1125. if not root: do_between()
  1126. else:
  1127. # If prog produces multiple files:
  1128. if cmd not in rebuild_list or rerun == True:
  1129. rebuild_list[cmd] = (rerun,fns[0] if fns else '') # FIX
  1130. return rerun
  1131. def refcheck(desc,chk,refchk):
  1132. vmsg("Comparing %s '%s' to stored reference" % (desc,chk))
  1133. if chk == refchk:
  1134. ok()
  1135. else:
  1136. if not opt.verbose: errmsg('')
  1137. errmsg(red("""
  1138. Fatal error - %s '%s' does not match reference value '%s'. Aborting test
  1139. """.strip() % (desc,chk,refchk)))
  1140. sys.exit(3)
  1141. def check_deps(cmds):
  1142. if len(cmds) != 1:
  1143. die(1,'Usage: %s check_deps <command>' % g.prog_name)
  1144. cmd = cmds[0]
  1145. if cmd not in cmd_data:
  1146. die(1,"'%s': unrecognized command" % cmd)
  1147. if not opt.quiet:
  1148. msg("Checking dependencies for '%s'" % (cmd))
  1149. check_needs_rerun(ts,cmd,build=False)
  1150. w = max(len(i) for i in rebuild_list) + 1
  1151. for cmd in rebuild_list:
  1152. c = rebuild_list[cmd]
  1153. m = 'Rebuild' if (c[0] and c[1]) else 'Build' if c[0] else 'OK'
  1154. msg('cmd {:<{w}} {}'.format(cmd+':', m, w=w))
  1155. # mmsg(cmd,c)
  1156. def clean(usr_dirs=[]):
  1157. if opt.skip_deps: return
  1158. all_dirs = MMGenTestSuite().list_tmp_dirs()
  1159. dirs = (usr_dirs or all_dirs)
  1160. for d in sorted(dirs):
  1161. if str(d) in all_dirs:
  1162. cleandir(all_dirs[str(d)])
  1163. else:
  1164. die(1,'%s: invalid directory number' % d)
  1165. cleandir(os.path.join('test','data_dir'))
  1166. def skip_for_win():
  1167. if g.platform == 'win':
  1168. import traceback
  1169. f = traceback.extract_stack()[-2][-2]
  1170. msg("Skipping test '{}': not supported on Windows platform".format(f))
  1171. return True
  1172. else:
  1173. return False
  1174. class MMGenTestSuite(object):
  1175. def __init__(self):
  1176. pass
  1177. def list_tmp_dirs(self):
  1178. d = {}
  1179. for k in cfgs: d[k] = cfgs[k]['tmpdir']
  1180. return d
  1181. def get_num_exts_for_cmd(self,cmd,dpy=False): # dpy ignored here
  1182. num = str(cmd_data[cmd][0])
  1183. dgl = cfgs[num]['dep_generators']
  1184. # mmsg(num,cmd,dgl)
  1185. if cmd in dgl.values():
  1186. exts = [k for k in dgl if dgl[k] == cmd]
  1187. return (num,exts)
  1188. else:
  1189. return None
  1190. def do_cmd(self,cmd):
  1191. # delete files produced by this cmd
  1192. # for ext,tmpdir in find_generated_exts(cmd):
  1193. # print cmd, get_file_with_ext(ext,tmpdir)
  1194. d = [(str(num),ext) for exts,num in cmd_data[cmd][2] for ext in exts]
  1195. # delete files depended on by this cmd
  1196. al = [get_file_with_ext(ext,cfgs[num]['tmpdir']) for num,ext in d]
  1197. global cfg
  1198. cfg = cfgs[str(cmd_data[cmd][0])]
  1199. if opt.resume:
  1200. if cmd == opt.resume:
  1201. msg(yellow("Resuming at '%s'" % cmd))
  1202. opt.resume = False
  1203. opt.skip_deps = False
  1204. else:
  1205. return
  1206. if opt.profile: start = time.time()
  1207. self.__class__.__dict__[cmd](*([self,cmd] + al))
  1208. if opt.profile:
  1209. msg('\r\033[50C{:.4f}'.format(time.time() - start))
  1210. global cmd_total
  1211. cmd_total += 1
  1212. def generate_file_deps(self,cmd):
  1213. return [(str(n),e) for exts,n in cmd_data[cmd][2] for e in exts]
  1214. def generate_cmd_deps(self,fdeps):
  1215. return [cfgs[str(n)]['dep_generators'][ext] for n,ext in fdeps]
  1216. def helpscreens(self,name,arg='--help'):
  1217. scripts = (
  1218. 'walletgen','walletconv','walletchk','txcreate','txsign','txsend','txdo','txbump',
  1219. 'addrgen','addrimport','keygen','passchg','tool','passgen','regtest','autosign')
  1220. for s in scripts:
  1221. t = MMGenExpect(name,('mmgen-'+s),[arg],extra_desc='(mmgen-%s)'%s,no_output=True)
  1222. t.read()
  1223. t.ok()
  1224. def longhelpscreens(self,name): self.helpscreens(name,arg='--longhelp')
  1225. def walletgen(self,name,del_dw_run='dummy',seed_len=None,gen_dfl_wallet=False):
  1226. write_to_tmpfile(cfg,pwfile,cfg['wpasswd']+'\n')
  1227. args = ['-d',cfg['tmpdir'],'-p1']
  1228. if seed_len: args += ['-l',str(seed_len)]
  1229. t = MMGenExpect(name,'mmgen-walletgen', args + [usr_rand_arg])
  1230. t.license()
  1231. t.usr_rand(usr_rand_chars)
  1232. t.passphrase_new('new MMGen wallet',cfg['wpasswd'])
  1233. t.label()
  1234. global have_dfl_wallet
  1235. if not have_dfl_wallet:
  1236. t.expect('move it to the data directory? (Y/n): ',('n','y')[gen_dfl_wallet])
  1237. if gen_dfl_wallet: have_dfl_wallet = True
  1238. t.written_to_file('MMGen wallet')
  1239. t.ok()
  1240. def walletgen_dfl_wallet(self,name,seed_len=None):
  1241. self.walletgen(name,seed_len=seed_len,gen_dfl_wallet=True)
  1242. def brainwalletgen_ref(self,name):
  1243. sl_arg = '-l%s' % cfg['seed_len']
  1244. hp_arg = '-p%s' % ref_wallet_hash_preset
  1245. label = "test.py ref. wallet (pw '%s', seed len %s)" \
  1246. % (ref_wallet_brainpass,cfg['seed_len'])
  1247. bf = 'ref.mmbrain'
  1248. args = ['-d',cfg['tmpdir'],hp_arg,sl_arg,'-ib','-L',label]
  1249. write_to_tmpfile(cfg,bf,ref_wallet_brainpass)
  1250. write_to_tmpfile(cfg,pwfile,cfg['wpasswd'])
  1251. t = MMGenExpect(name,'mmgen-walletconv', args + [usr_rand_arg])
  1252. t.license()
  1253. t.expect('Enter brainwallet: ', ref_wallet_brainpass+'\n')
  1254. t.passphrase_new('new MMGen wallet',cfg['wpasswd'])
  1255. t.usr_rand(usr_rand_chars)
  1256. sid = os.path.basename(t.written_to_file('MMGen wallet').split('-')[0])
  1257. refcheck('Seed ID',sid,cfg['seed_id'])
  1258. def refwalletgen(self,name): self.brainwalletgen_ref(name)
  1259. def passchg(self,name,wf,pf):
  1260. silence()
  1261. write_to_tmpfile(cfg,pwfile,get_data_from_file(pf))
  1262. end_silence()
  1263. t = MMGenExpect(name,'mmgen-passchg', [usr_rand_arg] +
  1264. ['-d',cfg['tmpdir'],'-p','2','-L','Changed label'] + ([],[wf])[bool(wf)])
  1265. t.license()
  1266. t.passphrase('MMGen wallet',cfgs['1']['wpasswd'],pwtype='old')
  1267. t.expect_getend('Hash preset changed to ')
  1268. t.passphrase('MMGen wallet',cfg['wpasswd'],pwtype='new') # reuse passphrase?
  1269. t.expect('Repeat passphrase: ',cfg['wpasswd']+'\n')
  1270. t.usr_rand(usr_rand_chars)
  1271. # t.expect('Enter a wallet label.*: ','Changed Label\n',regex=True)
  1272. t.expect_getend('Label changed to ')
  1273. # t.expect_getend('Key ID changed: ')
  1274. if not wf:
  1275. t.expect("Type uppercase 'YES' to confirm: ",'YES\n')
  1276. t.written_to_file('New wallet')
  1277. t.expect('Securely deleting old wallet')
  1278. # t.expect('Okay to WIPE 1 regular file ? (Yes/No)','Yes\n')
  1279. t.expect('Wallet passphrase has changed')
  1280. t.expect_getend('has been changed to ')
  1281. else:
  1282. t.written_to_file('MMGen wallet')
  1283. t.ok()
  1284. def passchg_dfl_wallet(self,name,pf):
  1285. return self.passchg(name=name,wf=None,pf=pf)
  1286. def walletchk(self,name,wf,pf,desc='MMGen wallet',add_args=[],sid=None,pw=False,extra_desc=''):
  1287. args = []
  1288. hp = cfg['hash_preset'] if 'hash_preset' in cfg else '1'
  1289. wf_arg = ([],[wf])[bool(wf)]
  1290. t = MMGenExpect(name,'mmgen-walletchk',
  1291. add_args+args+['-p',hp]+wf_arg,
  1292. extra_desc=extra_desc)
  1293. if desc != 'hidden incognito data':
  1294. t.expect("Getting %s from file '" % (desc))
  1295. if pw:
  1296. t.passphrase(desc,cfg['wpasswd'])
  1297. t.expect(
  1298. ['Passphrase is OK', 'Passphrase.* are correct'],
  1299. regex=True
  1300. )
  1301. chk = t.expect_getend('Valid %s for Seed ID ' % desc)[:8]
  1302. if sid: t.cmp_or_die(chk,sid)
  1303. else: t.ok()
  1304. def walletchk_newpass(self,name,wf,pf):
  1305. return self.walletchk(name,wf,pf,pw=True)
  1306. def walletchk_newpass_dfl_wallet(self,name,pf):
  1307. return self.walletchk_newpass(name,wf=None,pf=pf)
  1308. def delete_dfl_wallet(self,name,pf):
  1309. with open(os.path.join(cfg['tmpdir'],'del_dw_run'),'w') as f: pass
  1310. if opt.no_dw_delete: return True
  1311. for wf in [f for f in os.listdir(g.data_dir) if f[-6:]=='.mmdat']:
  1312. os.unlink(os.path.join(g.data_dir,wf))
  1313. MMGenExpect(name,'',msg_only=True)
  1314. global have_dfl_wallet
  1315. have_dfl_wallet = False
  1316. ok()
  1317. def addrgen(self,name,wf,pf=None,check_ref=False,ftype='addr',id_str=None,extra_args=[],mmtype=None):
  1318. if ftype[:4] != 'pass' and not mmtype:
  1319. if cfg['segwit']: mmtype = ('segwit','bech32')[bool(opt.bech32)]
  1320. cmd_pfx = (ftype,'pass')[ftype[:4]=='pass']
  1321. t = MMGenExpect(name,'mmgen-{}gen'.format(cmd_pfx),
  1322. ['-d',cfg['tmpdir']] +
  1323. extra_args +
  1324. ([],['--type='+str(mmtype)])[bool(mmtype)] +
  1325. ([],[wf])[bool(wf)] +
  1326. ([],[id_str])[bool(id_str)] +
  1327. [cfg['{}_idx_list'.format(cmd_pfx)]],
  1328. extra_desc='({})'.format(mmtype) if mmtype in ('segwit','bech32') else '')
  1329. t.license()
  1330. t.passphrase('MMGen wallet',cfg['wpasswd'])
  1331. t.expect('Passphrase is OK')
  1332. desc = ('address','password')[ftype[:4]=='pass']
  1333. chk = t.expect_getend(r'Checksum for {} data .*?: '.format(desc),regex=True)
  1334. if ftype[:4] == 'pass':
  1335. t.expect('Encrypt password list? (y/N): ','\n')
  1336. t.written_to_file('Password list',oo=True)
  1337. else:
  1338. t.written_to_file('Addresses',oo=True)
  1339. if check_ref:
  1340. k = 'passfile32_chk' if ftype == 'pass32' \
  1341. else 'passfilehex_chk' if ftype == 'passhex' \
  1342. else 'passfile_chk' if ftype == 'pass' \
  1343. else '{}file{}_chk'.format(ftype,'_'+mmtype if mmtype else '')
  1344. chk_ref = cfg[k] if ftype[:4] == 'pass' else cfg[k][fork][g.testnet]
  1345. refcheck('{}list data checksum'.format(ftype),chk,chk_ref)
  1346. else:
  1347. t.ok()
  1348. def addrgen_dfl_wallet(self,name,pf=None,check_ref=False):
  1349. return self.addrgen(name,wf=None,pf=pf,check_ref=check_ref)
  1350. def refaddrgen(self,name,wf,pf):
  1351. self.addrgen(name,wf,pf=pf,check_ref=True)
  1352. def refaddrgen_compressed(self,name,wf,pf):
  1353. if opt.segwit or opt.bech32:
  1354. msg('Skipping non-Segwit address generation'); return True
  1355. self.addrgen(name,wf,pf=pf,check_ref=True,mmtype='compressed')
  1356. def addrimport(self,name,addrfile):
  1357. outfile = os.path.join(cfg['tmpdir'],'addrfile_w_comments')
  1358. add_comments_to_addr_file(addrfile,outfile)
  1359. t = MMGenExpect(name,'mmgen-addrimport', [outfile])
  1360. t.expect_getend(r'Checksum for address data .*\[.*\]: ',regex=True)
  1361. t.expect("Type uppercase 'YES' to confirm: ",'\n')
  1362. vmsg('This is a simulation, so no addresses were actually imported into the tracking\nwallet')
  1363. t.ok(exit_val=1)
  1364. def txcreate_common(self,name,sources=['1'],non_mmgen_input='',do_label=False,txdo_args=[],add_args=[],view=None):
  1365. if opt.verbose or opt.exact_output:
  1366. sys.stderr.write(green('Generating fake tracking wallet info\n'))
  1367. silence()
  1368. ad,tx_data = create_tx_data(sources)
  1369. dfake = create_fake_unspent_data(ad,tx_data,non_mmgen_input)
  1370. write_fake_data_to_file(repr(dfake))
  1371. cmd_args = make_txcreate_cmdline(tx_data)
  1372. end_silence()
  1373. if opt.verbose or opt.exact_output: sys.stderr.write('\n')
  1374. t = MMGenExpect(name,
  1375. 'mmgen-'+('txcreate','txdo')[bool(txdo_args)],
  1376. ([],['--rbf'])[g.proto.cap('rbf')] +
  1377. ['-f',tx_fee,'-B'] + add_args + cmd_args + txdo_args)
  1378. t.license()
  1379. if txdo_args and add_args: # txdo4
  1380. t.hash_preset('key-address data','1')
  1381. t.passphrase('key-address data',cfgs['14']['kapasswd'])
  1382. t.expect('Check key-to-address validity? (y/N): ','y')
  1383. for num in tx_data:
  1384. t.expect_getend('Getting address data from file ')
  1385. chk=t.expect_getend(r'Checksum for address data .*?: ',regex=True)
  1386. verify_checksum_or_exit(tx_data[num]['chk'],chk)
  1387. # not in tracking wallet warning, (1 + num sources) times
  1388. if t.expect(['Continue anyway? (y/N): ',
  1389. 'Unable to connect to {}'.format(g.proto.daemon_name)]) == 0:
  1390. t.send('y')
  1391. else:
  1392. errmsg(red('Error: unable to connect to {}. Exiting'.format(g.proto.daemon_name)))
  1393. sys.exit(1)
  1394. for num in tx_data:
  1395. t.expect('Continue anyway? (y/N): ','y')
  1396. t.expect(r"'q'=quit view, .*?:.",'M', regex=True)
  1397. t.expect(r"'q'=quit view, .*?:.",'q', regex=True)
  1398. outputs_list = [(addrs_per_wallet+1)*i + 1 for i in range(len(tx_data))]
  1399. if non_mmgen_input: outputs_list.append(len(tx_data)*(addrs_per_wallet+1) + 1)
  1400. t.expect('outputs to spend: ',' '.join([str(i) for i in outputs_list])+'\n')
  1401. if non_mmgen_input and not txdo_args: t.expect('Accept? (y/N): ','y')
  1402. t.expect('OK? (Y/n): ','y') # fee OK?
  1403. t.expect('OK? (Y/n): ','y') # change OK?
  1404. if do_label:
  1405. t.expect('Add a comment to transaction? (y/N): ','y')
  1406. t.expect('Comment: ',ref_tx_label.encode('utf8')+'\n')
  1407. else:
  1408. t.expect('Add a comment to transaction? (y/N): ','\n')
  1409. t.tx_view(view=view)
  1410. if txdo_args: return t
  1411. t.expect('Save transaction? (y/N): ','y')
  1412. t.written_to_file('Transaction')
  1413. t.ok()
  1414. def txcreate(self,name,addrfile):
  1415. self.txcreate_common(name,sources=['1'],add_args=['--vsize-adj=1.01'])
  1416. def txbump(self,name,txfile,prepend_args=[],seed_args=[]):
  1417. if not g.proto.cap('rbf'):
  1418. msg('Skipping RBF'); return True
  1419. args = prepend_args + ['-q','-d',cfg['tmpdir'],txfile] + seed_args
  1420. t = MMGenExpect(name,'mmgen-txbump',args)
  1421. if seed_args:
  1422. t.hash_preset('key-address data','1')
  1423. t.passphrase('key-address data',cfgs['14']['kapasswd'])
  1424. t.expect('Check key-to-address validity? (y/N): ','y')
  1425. t.expect('deduct the fee from (Hit ENTER for the change output): ','1\n')
  1426. # Fee must be > tx_fee + network relay fee (currently 0.00001)
  1427. t.expect('OK? (Y/n): ','\n')
  1428. t.expect('Enter transaction fee: ',txbump_fee+'\n')
  1429. t.expect('OK? (Y/n): ','\n')
  1430. if seed_args: # sign and send
  1431. t.expect('Edit transaction comment? (y/N): ','\n')
  1432. for cnum,desc in (('1','incognito data'),('3','MMGen wallet'),('4','MMGen wallet')):
  1433. t.passphrase(('%s' % desc),cfgs[cnum]['wpasswd'])
  1434. t.expect("Type uppercase 'YES' to confirm: ",'YES\n')
  1435. else:
  1436. t.expect('Add a comment to transaction? (y/N): ','\n')
  1437. t.expect('Save transaction? (y/N): ','y')
  1438. t.written_to_file('Transaction')
  1439. os.unlink(txfile) # our tx file replaces the original
  1440. os.system('touch ' + os.path.join(cfg['tmpdir'],'txbump'))
  1441. t.ok()
  1442. def txdo(self,name,addrfile,wallet):
  1443. t = self.txcreate_common(name,sources=['1'],txdo_args=[wallet])
  1444. self.txsign(name,'','',pf='',save=True,has_label=False,txdo_handle=t)
  1445. self.txsend(name,'',txdo_handle=t)
  1446. def txcreate_dfl_wallet(self,name,addrfile):
  1447. self.txcreate_common(name,sources=['15'])
  1448. def txsign_end(self,t,tnum=None,has_label=False):
  1449. t.expect('Signing transaction')
  1450. cprompt = ('Add a comment to transaction','Edit transaction comment')[has_label]
  1451. t.expect('%s? (y/N): ' % cprompt,'\n')
  1452. t.expect('Save signed transaction.*?\? \(Y/n\): ','y',regex=True)
  1453. add = ' #' + tnum if tnum else ''
  1454. t.written_to_file('Signed transaction' + add, oo=True)
  1455. def txsign(self,name,txfile,wf,pf='',bumpf='',save=True,has_label=False,txdo_handle=None,extra_opts=[]):
  1456. if txdo_handle:
  1457. t = txdo_handle
  1458. else:
  1459. t = MMGenExpect(name,'mmgen-txsign', extra_opts + ['-d',cfg['tmpdir'],txfile]+([],[wf])[bool(wf)])
  1460. t.license()
  1461. t.tx_view()
  1462. t.passphrase('MMGen wallet',cfg['wpasswd'])
  1463. if txdo_handle: return
  1464. if save:
  1465. self.txsign_end(t,has_label=has_label)
  1466. t.ok()
  1467. else:
  1468. cprompt = ('Add a comment to transaction','Edit transaction comment')[has_label]
  1469. t.expect('%s? (y/N): ' % cprompt,'\n')
  1470. t.expect('Save signed transaction? (Y/n): ','n')
  1471. t.ok(exit_val=1)
  1472. def txsign_dfl_wallet(self,name,txfile,pf='',save=True,has_label=False):
  1473. return self.txsign(name,txfile,wf=None,pf=pf,save=save,has_label=has_label)
  1474. def txsend(self,name,sigfile,txdo_handle=None,really_send=False,extra_opts=[]):
  1475. if txdo_handle:
  1476. t = txdo_handle
  1477. else:
  1478. if really_send: os.environ['MMGEN_BOGUS_SEND'] = ''
  1479. t = MMGenExpect(name,'mmgen-txsend', extra_opts + ['-d',cfg['tmpdir'],sigfile])
  1480. if really_send: os.environ['MMGEN_BOGUS_SEND'] = '1'
  1481. t.license()
  1482. t.tx_view(view='terse')
  1483. t.expect('Add a comment to transaction? (y/N): ','\n')
  1484. t.expect('Are you sure you want to broadcast this')
  1485. m = 'YES, I REALLY WANT TO DO THIS'
  1486. t.expect("'%s' to confirm: " % m,m+'\n')
  1487. if really_send:
  1488. txid = t.expect_getend('Transaction sent: ')
  1489. assert len(txid) == 64
  1490. else:
  1491. t.expect('BOGUS transaction NOT sent')
  1492. t.written_to_file('Sent transaction')
  1493. t.ok()
  1494. def walletconv_export(self,name,wf,desc,uargs=[],out_fmt='w',pf=None,out_pw=False):
  1495. opts = ['-d',cfg['tmpdir'],'-o',out_fmt] + uargs + \
  1496. ([],[wf])[bool(wf)] + ([],['-P',pf])[bool(pf)]
  1497. t = MMGenExpect(name,'mmgen-walletconv',opts)
  1498. t.license()
  1499. if not pf:
  1500. t.passphrase('MMGen wallet',cfg['wpasswd'])
  1501. if out_pw:
  1502. t.passphrase_new('new '+desc,cfg['wpasswd'])
  1503. t.usr_rand(usr_rand_chars)
  1504. if ' '.join(desc.split()[-2:]) == 'incognito data':
  1505. t.expect('Generating encryption key from OS random data ')
  1506. t.expect('Generating encryption key from OS random data ')
  1507. ic_id = t.expect_getend('New Incog Wallet ID: ')
  1508. t.expect('Generating encryption key from OS random data ')
  1509. if desc == 'hidden incognito data':
  1510. write_to_tmpfile(cfg,incog_id_fn,ic_id)
  1511. ret = t.expect(['Create? (Y/n): ',"'YES' to confirm: "])
  1512. if ret == 0:
  1513. t.send('\n')
  1514. t.expect('Enter file size: ',str(hincog_bytes)+'\n')
  1515. else:
  1516. t.send('YES\n')
  1517. if out_fmt == 'w': t.label()
  1518. return t.written_to_file(capfirst(desc),oo=True),t
  1519. def export_seed(self,name,wf,desc='seed data',out_fmt='seed',pf=None):
  1520. f,t = self.walletconv_export(name,wf,desc=desc,out_fmt=out_fmt,pf=pf)
  1521. silence()
  1522. msg('%s: %s' % (capfirst(desc),cyan(get_data_from_file(f,desc))))
  1523. end_silence()
  1524. t.ok()
  1525. def export_hex(self,name,wf,desc='hexadecimal seed data',out_fmt='hex',pf=None):
  1526. self.export_seed(name,wf,desc=desc,out_fmt=out_fmt,pf=pf)
  1527. def export_seed_dfl_wallet(self,name,pf,desc='seed data',out_fmt='seed'):
  1528. self.export_seed(name,wf=None,desc=desc,out_fmt=out_fmt,pf=pf)
  1529. def export_mnemonic(self,name,wf):
  1530. self.export_seed(name,wf,desc='mnemonic data',out_fmt='words')
  1531. def export_incog(self,name,wf,desc='incognito data',out_fmt='i',add_args=[]):
  1532. uargs = ['-p1',usr_rand_arg] + add_args
  1533. f,t = self.walletconv_export(name,wf,desc=desc,out_fmt=out_fmt,uargs=uargs,out_pw=True)
  1534. t.ok()
  1535. def export_incog_hex(self,name,wf):
  1536. self.export_incog(name,wf,desc='hex incognito data',out_fmt='xi')
  1537. # TODO: make outdir and hidden incog compatible (ignore --outdir and warn user?)
  1538. def export_incog_hidden(self,name,wf):
  1539. rf = os.path.join(cfg['tmpdir'],hincog_fn)
  1540. add_args = ['-J','%s,%s'%(rf,hincog_offset)]
  1541. self.export_incog(
  1542. name,wf,desc='hidden incognito data',out_fmt='hi',add_args=add_args)
  1543. def addrgen_seed(self,name,wf,foo,desc='seed data',in_fmt='seed'):
  1544. stdout = (False,True)[desc=='seed data'] #capture output to screen once
  1545. add_args = ([],['-S'])[bool(stdout)] + get_segwit_arg(cfg)
  1546. t = MMGenExpect(name,'mmgen-addrgen', add_args +
  1547. ['-i'+in_fmt,'-d',cfg['tmpdir'],wf,cfg['addr_idx_list']])
  1548. t.license()
  1549. t.expect_getend('Valid %s for Seed ID ' % desc)
  1550. vmsg('Comparing generated checksum with checksum from previous address file')
  1551. chk = t.expect_getend(r'Checksum for address data .*?: ',regex=True)
  1552. if stdout: t.read()
  1553. verify_checksum_or_exit(get_addrfile_checksum(),chk)
  1554. if in_fmt == 'seed':
  1555. t.ok()
  1556. else:
  1557. t.no_overwrite()
  1558. t.ok(exit_val=1)
  1559. def addrgen_hex(self,name,wf,foo,desc='hexadecimal seed data',in_fmt='hex'):
  1560. self.addrgen_seed(name,wf,foo,desc=desc,in_fmt=in_fmt)
  1561. def addrgen_mnemonic(self,name,wf,foo):
  1562. self.addrgen_seed(name,wf,foo,desc='mnemonic data',in_fmt='words')
  1563. def addrgen_incog(self,name,wf=[],foo='',in_fmt='i',desc='incognito data',args=[]):
  1564. t = MMGenExpect(name,'mmgen-addrgen', args + get_segwit_arg(cfg) + ['-i'+in_fmt,'-d',cfg['tmpdir']]+
  1565. ([],[wf])[bool(wf)] + [cfg['addr_idx_list']])
  1566. t.license()
  1567. t.expect_getend('Incog Wallet ID: ')
  1568. t.hash_preset(desc,'1')
  1569. t.passphrase('%s \w{8}' % desc, cfg['wpasswd'])
  1570. vmsg('Comparing generated checksum with checksum from address file')
  1571. chk = t.expect_getend(r'Checksum for address data .*?: ',regex=True)
  1572. verify_checksum_or_exit(get_addrfile_checksum(),chk)
  1573. t.no_overwrite()
  1574. t.ok(exit_val=1)
  1575. def addrgen_incog_hex(self,name,wf,foo):
  1576. self.addrgen_incog(name,wf,'',in_fmt='xi',desc='hex incognito data')
  1577. def addrgen_incog_hidden(self,name,wf,foo):
  1578. rf = os.path.join(cfg['tmpdir'],hincog_fn)
  1579. self.addrgen_incog(name,[],'',in_fmt='hi',desc='hidden incognito data',
  1580. args=['-H','%s,%s'%(rf,hincog_offset),'-l',str(hincog_seedlen)])
  1581. def keyaddrgen(self,name,wf,pf=None,check_ref=False,mmtype=None):
  1582. if cfg['segwit'] and not mmtype:
  1583. mmtype = ('segwit','bech32')[bool(opt.bech32)]
  1584. args = ['-d',cfg['tmpdir'],usr_rand_arg,wf,cfg['addr_idx_list']]
  1585. t = MMGenExpect(name,'mmgen-keygen',
  1586. ([],['--type='+str(mmtype)])[bool(mmtype)] + args,
  1587. extra_desc='({})'.format(mmtype) if mmtype in ('segwit','bech32') else '')
  1588. t.license()
  1589. t.passphrase('MMGen wallet',cfg['wpasswd'])
  1590. chk = t.expect_getend(r'Checksum for key-address data .*?: ',regex=True)
  1591. if check_ref:
  1592. k = 'keyaddrfile{}_chk'.format('_'+mmtype if mmtype else '')
  1593. refcheck('key-address data checksum',chk,cfg[k][fork][g.testnet])
  1594. return
  1595. t.expect('Encrypt key list? (y/N): ','y')
  1596. t.usr_rand(usr_rand_chars)
  1597. t.hash_preset('new key list','1')
  1598. # t.passphrase_new('new key list','kafile password')
  1599. t.passphrase_new('new key list',cfg['kapasswd'])
  1600. t.written_to_file('Encrypted secret keys',oo=True)
  1601. t.ok()
  1602. def refkeyaddrgen(self,name,wf,pf):
  1603. self.keyaddrgen(name,wf,pf,check_ref=True)
  1604. def refkeyaddrgen_compressed(self,name,wf,pf):
  1605. if opt.segwit or opt.bech32:
  1606. msg('Skipping non-Segwit key-address generation'); return True
  1607. self.keyaddrgen(name,wf,pf,check_ref=True,mmtype='compressed')
  1608. def refpasswdgen(self,name,wf,pf):
  1609. self.addrgen(name,wf,pf,check_ref=True,ftype='pass',id_str='alice@crypto.org')
  1610. def ref_b32passwdgen(self,name,wf,pf):
  1611. ea = ['--base32','--passwd-len','17']
  1612. self.addrgen(name,wf,pf,check_ref=True,ftype='pass32',id_str='фубар@crypto.org',extra_args=ea)
  1613. def ref_hexpasswdgen(self,name,wf,pf):
  1614. ea = ['--hex']
  1615. self.addrgen(name,wf,pf,check_ref=True,ftype='passhex',id_str='фубар@crypto.org',extra_args=ea)
  1616. def txsign_keyaddr(self,name,keyaddr_file,txfile):
  1617. t = MMGenExpect(name,'mmgen-txsign', ['-d',cfg['tmpdir'],'-M',keyaddr_file,txfile])
  1618. t.license()
  1619. t.hash_preset('key-address data','1')
  1620. t.passphrase('key-address data',cfg['kapasswd'])
  1621. t.expect('Check key-to-address validity? (y/N): ','y')
  1622. t.tx_view()
  1623. self.txsign_end(t)
  1624. t.ok()
  1625. def walletgen2(self,name,del_dw_run='dummy'):
  1626. self.walletgen(name,seed_len=128)
  1627. def addrgen2(self,name,wf):
  1628. self.addrgen(name,wf,pf='')
  1629. def txcreate2(self,name,addrfile):
  1630. self.txcreate_common(name,sources=['2'])
  1631. def txsign2(self,name,txf1,wf1,txf2,wf2):
  1632. t = MMGenExpect(name,'mmgen-txsign', ['-d',cfg['tmpdir'],txf1,wf1,txf2,wf2])
  1633. t.license()
  1634. for cnum in ('1','2'):
  1635. t.tx_view()
  1636. t.passphrase('MMGen wallet',cfgs[cnum]['wpasswd'])
  1637. self.txsign_end(t,cnum)
  1638. t.ok()
  1639. def export_mnemonic2(self,name,wf):
  1640. self.export_mnemonic(name,wf)
  1641. def walletgen3(self,name,del_dw_run='dummy'):
  1642. self.walletgen(name)
  1643. def addrgen3(self,name,wf):
  1644. self.addrgen(name,wf,pf='')
  1645. def txcreate3(self,name,addrfile1,addrfile2):
  1646. self.txcreate_common(name,sources=['1','3'])
  1647. def txsign3(self,name,wf1,wf2,txf2):
  1648. t = MMGenExpect(name,'mmgen-txsign', ['-d',cfg['tmpdir'],wf1,wf2,txf2])
  1649. t.license()
  1650. t.tx_view()
  1651. for cnum in ('1','3'):
  1652. # t.expect_getend('Getting MMGen wallet data from file ')
  1653. t.passphrase('MMGen wallet',cfgs[cnum]['wpasswd'])
  1654. self.txsign_end(t)
  1655. t.ok()
  1656. def walletgen4(self,name,del_dw_run='dummy'):
  1657. bwf = os.path.join(cfg['tmpdir'],cfg['bw_filename'])
  1658. make_brainwallet_file(bwf)
  1659. seed_len = str(cfg['seed_len'])
  1660. args = ['-d',cfg['tmpdir'],'-p1',usr_rand_arg,'-l'+seed_len,'-ib']
  1661. t = MMGenExpect(name,'mmgen-walletconv', args + [bwf])
  1662. t.license()
  1663. t.passphrase_new('new MMGen wallet',cfg['wpasswd'])
  1664. t.usr_rand(usr_rand_chars)
  1665. t.label()
  1666. t.written_to_file('MMGen wallet')
  1667. t.ok()
  1668. def addrgen4(self,name,wf):
  1669. self.addrgen(name,wf,pf='')
  1670. def txcreate4(self,name,f1,f2,f3,f4,f5,f6):
  1671. self.txcreate_common(name,sources=['1','2','3','4','14'],non_mmgen_input='4',do_label=1)
  1672. def txdo4(self,name,f1,f2,f3,f4,f5,f6,f7,f8,f9,f10,f11,f12):
  1673. non_mm_fn = os.path.join(cfg['tmpdir'],non_mmgen_fn)
  1674. add_args = ['-d',cfg['tmpdir'],'-i','brain','-b'+cfg['bw_params'],'-p1','-k',non_mm_fn,'-M',f12]
  1675. t = self.txcreate_common(name,sources=['1','2','3','4','14'],non_mmgen_input='4',do_label=1,txdo_args=[f7,f8,f9,f10],add_args=add_args)
  1676. os.system('rm -f %s/*.sigtx' % cfg['tmpdir'])
  1677. self.txsign4(name,f7,f8,f9,f10,f11,f12,txdo_handle=t)
  1678. self.txsend(name,'',txdo_handle=t)
  1679. os.system('touch ' + os.path.join(cfg['tmpdir'],'txdo'))
  1680. def txbump4(self,name,f1,f2,f3,f4,f5,f6,f7,f8,f9): # f7:txfile,f9:'txdo'
  1681. non_mm_fn = os.path.join(cfg['tmpdir'],non_mmgen_fn)
  1682. self.txbump(name,f7,prepend_args=['-p1','-k',non_mm_fn,'-M',f1],seed_args=[f2,f3,f4,f5,f6,f8])
  1683. def txsign4(self,name,f1,f2,f3,f4,f5,f6,txdo_handle=None):
  1684. if txdo_handle:
  1685. t = txdo_handle
  1686. else:
  1687. non_mm_fn = os.path.join(cfg['tmpdir'],non_mmgen_fn)
  1688. a = ['-d',cfg['tmpdir'],'-i','brain','-b'+cfg['bw_params'],'-p1','-k',non_mm_fn,'-M',f6,f1,f2,f3,f4,f5]
  1689. t = MMGenExpect(name,'mmgen-txsign',a)
  1690. t.license()
  1691. t.hash_preset('key-address data','1')
  1692. t.passphrase('key-address data',cfgs['14']['kapasswd'])
  1693. t.expect('Check key-to-address validity? (y/N): ','y')
  1694. t.tx_view(view='terse')
  1695. for cnum,desc in (('1','incognito data'),('3','MMGen wallet')):
  1696. t.passphrase(('%s' % desc),cfgs[cnum]['wpasswd'])
  1697. if txdo_handle: return
  1698. self.txsign_end(t,has_label=True)
  1699. t.ok()
  1700. def tool_encrypt(self,name,infile=''):
  1701. if infile:
  1702. infn = infile
  1703. else:
  1704. d = os.urandom(1033)
  1705. tmp_fn = cfg['tool_enc_infn']
  1706. write_to_tmpfile(cfg,tmp_fn,d,binary=True)
  1707. infn = get_tmpfile_fn(cfg,tmp_fn)
  1708. t = MMGenExpect(name,'mmgen-tool',['-d',cfg['tmpdir'],usr_rand_arg,'encrypt',infn])
  1709. t.usr_rand(usr_rand_chars)
  1710. t.hash_preset('user data','1')
  1711. t.passphrase_new('user data',tool_enc_passwd)
  1712. t.written_to_file('Encrypted data')
  1713. t.ok()
  1714. # Generate the reference mmenc file
  1715. # def tool_encrypt_ref(self,name):
  1716. # infn = get_tmpfile_fn(cfg,cfg['tool_enc_ref_infn'])
  1717. # write_data_to_file(infn,cfg['tool_enc_reftext'],silent=True)
  1718. # self.tool_encrypt(name,infn)
  1719. def tool_decrypt(self,name,f1,f2):
  1720. of = name + '.out'
  1721. pre = []
  1722. t = MMGenExpect(name,'mmgen-tool',
  1723. pre+['-d',cfg['tmpdir'],'decrypt',f2,'outfile='+of,'hash_preset=1'])
  1724. t.passphrase('user data',tool_enc_passwd)
  1725. t.written_to_file('Decrypted data')
  1726. d1 = read_from_file(f1,binary=True)
  1727. d2 = read_from_file(get_tmpfile_fn(cfg,of),binary=True)
  1728. cmp_or_die(d1,d2,skip_ok=False)
  1729. def tool_find_incog_data(self,name,f1,f2):
  1730. i_id = read_from_file(f2).rstrip()
  1731. vmsg('Incog ID: %s' % cyan(i_id))
  1732. t = MMGenExpect(name,'mmgen-tool',
  1733. ['-d',cfg['tmpdir'],'find_incog_data',f1,i_id])
  1734. o = t.expect_getend('Incog data for ID %s found at offset ' % i_id)
  1735. os.unlink(f1)
  1736. cmp_or_die(hincog_offset,int(o))
  1737. # Miscellaneous tests
  1738. def autosign(self,name): # tests everything except device detection, mount/unmount
  1739. if skip_for_win(): return
  1740. fdata = (('btc',''),('bch',''),('ltc','litecoin'))
  1741. tfns = [cfgs['8']['ref_tx_file'][c].format('') for c,d in fdata]
  1742. tfs = [os.path.join(ref_dir,d[1],fn) for d,fn in zip(fdata,tfns)]
  1743. try: os.mkdir(os.path.join(cfg['tmpdir'],'tx'))
  1744. except: pass
  1745. for f,fn in zip(tfs,tfns):
  1746. shutil.copyfile(f,os.path.join(cfg['tmpdir'],'tx',fn))
  1747. # make a bad tx file
  1748. with open(os.path.join(cfg['tmpdir'],'tx','bad.rawtx'),'w') as f:
  1749. f.write('bad tx data')
  1750. ls = os.listdir(cfg['tmpdir'])
  1751. opts = ['--mountpoint='+cfg['tmpdir'],'--coins=btc,bch,ltc']
  1752. # opts += ['--quiet']
  1753. mn_fn = os.path.join(ref_dir,cfgs['8']['seed_id']+'.mmwords')
  1754. mn = read_from_file(mn_fn).strip().split()
  1755. t = MMGenExpect(name,'mmgen-autosign',opts+['gen_key'],extra_desc='(gen_key)')
  1756. t.expect_getend('Wrote key file ')
  1757. t.ok()
  1758. t = MMGenExpect(name,'mmgen-autosign',opts+['setup'],extra_desc='(setup)')
  1759. t.expect('words: ','3')
  1760. t.expect('OK? (Y/n): ','\n')
  1761. for i in range(24):
  1762. t.expect('word #{}: '.format(i+1),mn[i]+'\n')
  1763. wf = t.written_to_file('Autosign wallet')
  1764. t.ok()
  1765. t = MMGenExpect(name,'mmgen-autosign',opts+['wait'],extra_desc='(sign)')
  1766. t.expect('3 transactions signed')
  1767. t.expect('1 transaction failed to sign')
  1768. t.expect('Waiting.')
  1769. t.kill(2)
  1770. t.ok(exit_val=1)
  1771. # Saved reference file tests
  1772. def ref_wallet_conv(self,name):
  1773. wf = os.path.join(ref_dir,cfg['ref_wallet'])
  1774. self.walletconv_in(name,wf,'MMGen wallet',pw=True,oo=True)
  1775. def ref_mn_conv(self,name,ext='mmwords',desc='Mnemonic data'):
  1776. wf = os.path.join(ref_dir,cfg['seed_id']+'.'+ext)
  1777. self.walletconv_in(name,wf,desc,oo=True)
  1778. def ref_seed_conv(self,name):
  1779. self.ref_mn_conv(name,ext='mmseed',desc='Seed data')
  1780. def ref_hex_conv(self,name):
  1781. self.ref_mn_conv(name,ext='mmhex',desc='Hexadecimal seed data')
  1782. def ref_brain_conv(self,name):
  1783. uopts = ['-i','b','-p','1','-l',str(cfg['seed_len'])]
  1784. self.walletconv_in(name,None,'brainwallet',uopts,oo=True)
  1785. def ref_incog_conv(self,name,wfk='ic_wallet',in_fmt='i',desc='incognito data'):
  1786. uopts = ['-i',in_fmt,'-p','1','-l',str(cfg['seed_len'])]
  1787. wf = os.path.join(ref_dir,cfg[wfk])
  1788. self.walletconv_in(name,wf,desc,uopts,oo=True,pw=True)
  1789. def ref_incox_conv(self,name):
  1790. self.ref_incog_conv(name,in_fmt='xi',wfk='ic_wallet_hex',desc='hex incognito data')
  1791. def ref_hincog_conv(self,name,wfk='hic_wallet',add_uopts=[]):
  1792. ic_f = os.path.join(ref_dir,cfg[wfk])
  1793. uopts = ['-i','hi','-p','1','-l',str(cfg['seed_len'])] + add_uopts
  1794. hi_opt = ['-H','%s,%s' % (ic_f,ref_wallet_incog_offset)]
  1795. self.walletconv_in(name,None,'hidden incognito data',uopts+hi_opt,oo=True,pw=True)
  1796. def ref_hincog_conv_old(self,name):
  1797. self.ref_hincog_conv(name,wfk='hic_wallet_old',add_uopts=['-O'])
  1798. def ref_wallet_conv_out(self,name):
  1799. self.walletconv_out(name,'MMGen wallet','w',pw=True)
  1800. def ref_mn_conv_out(self,name):
  1801. self.walletconv_out(name,'mnemonic data','mn')
  1802. def ref_seed_conv_out(self,name):
  1803. self.walletconv_out(name,'seed data','seed')
  1804. def ref_hex_conv_out(self,name):
  1805. self.walletconv_out(name,'hexadecimal seed data','hexseed')
  1806. def ref_incog_conv_out(self,name):
  1807. self.walletconv_out(name,'incognito data',out_fmt='i',pw=True)
  1808. def ref_incox_conv_out(self,name):
  1809. self.walletconv_out(name,'hex incognito data',out_fmt='xi',pw=True)
  1810. def ref_hincog_conv_out(self,name,extra_uopts=[]):
  1811. ic_f = os.path.join(cfg['tmpdir'],hincog_fn)
  1812. hi_parms = '%s,%s' % (ic_f,ref_wallet_incog_offset)
  1813. sl_parm = '-l' + str(cfg['seed_len'])
  1814. self.walletconv_out(name,
  1815. 'hidden incognito data', 'hi',
  1816. uopts=['-J',hi_parms,sl_parm] + extra_uopts,
  1817. uopts_chk=['-H',hi_parms,sl_parm],
  1818. pw=True
  1819. )
  1820. def ref_wallet_chk(self,name):
  1821. wf = os.path.join(ref_dir,cfg['ref_wallet'])
  1822. self.walletchk(name,wf,pf=None,pw=True,sid=cfg['seed_id'])
  1823. def ref_ss_chk(self,name,ss=None):
  1824. wf = os.path.join(ref_dir,'%s.%s' % (cfg['seed_id'],ss.ext))
  1825. self.walletchk(name,wf,pf=None,desc=ss.desc,sid=cfg['seed_id'])
  1826. def ref_seed_chk(self,name):
  1827. from mmgen.seed import SeedFile
  1828. self.ref_ss_chk(name,ss=SeedFile)
  1829. def ref_hex_chk(self,name):
  1830. from mmgen.seed import HexSeedFile
  1831. self.ref_ss_chk(name,ss=HexSeedFile)
  1832. def ref_mn_chk(self,name):
  1833. from mmgen.seed import Mnemonic
  1834. self.ref_ss_chk(name,ss=Mnemonic)
  1835. def ref_brain_chk(self,name,bw_file=ref_bw_file):
  1836. wf = os.path.join(ref_dir,bw_file)
  1837. add_args = ['-l%s' % cfg['seed_len'], '-p'+ref_bw_hash_preset]
  1838. self.walletchk(name,wf,pf=None,add_args=add_args,
  1839. desc='brainwallet',sid=cfg['ref_bw_seed_id'])
  1840. def ref_brain_chk_spc3(self,name):
  1841. self.ref_brain_chk(name,bw_file=ref_bw_file_spc)
  1842. def ref_hincog_chk(self,name,desc='hidden incognito data'):
  1843. for wtype,edesc,of_arg in ('hic_wallet','',[]), \
  1844. ('hic_wallet_old','(old format)',['-O']):
  1845. ic_arg = ['-H%s,%s' % (
  1846. os.path.join(ref_dir,cfg[wtype]),
  1847. ref_wallet_incog_offset
  1848. )]
  1849. slarg = ['-l%s ' % cfg['seed_len']]
  1850. hparg = ['-p1']
  1851. if wtype == 'hic_wallet_old' and opt.profile: msg('')
  1852. t = MMGenExpect(name,'mmgen-walletchk',
  1853. slarg + hparg + of_arg + ic_arg,
  1854. extra_desc=edesc)
  1855. t.passphrase(desc,cfg['wpasswd'])
  1856. if wtype == 'hic_wallet_old':
  1857. t.expect('Is the Seed ID correct? (Y/n): ','\n')
  1858. chk = t.expect_getend('Seed ID: ')
  1859. t.close()
  1860. cmp_or_die(cfg['seed_id'],chk)
  1861. def ref_addrfile_chk(self,name,ftype='addr',coin=None,subdir=None,pfx=None,mmtype=None,add_args=[]):
  1862. af_key = 'ref_{}file'.format(ftype)
  1863. af_fn = cfg[af_key].format(pfx or altcoin_pfx,'' if coin else tn_ext)
  1864. af = os.path.join(ref_dir,(subdir or ref_subdir,'')[ftype=='passwd'],af_fn)
  1865. coin_arg = [] if coin == None else ['--coin='+coin]
  1866. tool_cmd = ftype.replace('segwit','').replace('bech32','')+'file_chksum'
  1867. t = MMGenExpect(name,'mmgen-tool',coin_arg+[tool_cmd,af]+add_args)
  1868. if ftype == 'keyaddr':
  1869. w = 'key-address data'
  1870. t.hash_preset(w,ref_kafile_hash_preset)
  1871. t.passphrase(w,ref_kafile_pass)
  1872. t.expect('Check key-to-address validity? (y/N): ','y')
  1873. o = t.read().strip().split('\n')[-1]
  1874. rc = cfg[ 'ref_' + ftype + 'file_chksum' +
  1875. ('_'+coin.lower() if coin else '') +
  1876. ('_'+mmtype if mmtype else '')]
  1877. ref_chksum = rc if (ftype == 'passwd' or coin) else rc[g.proto.base_coin.lower()][g.testnet]
  1878. cmp_or_die(ref_chksum,o)
  1879. def ref_addrfile_chk_zec(self,name):
  1880. self.ref_addrfile_chk(name,ftype='addr',coin='ZEC',subdir='zcash',pfx='-ZEC-C')
  1881. def ref_addrfile_chk_zec_z(self,name):
  1882. if skip_for_win(): return
  1883. self.ref_addrfile_chk(name,ftype='addr',coin='ZEC',subdir='zcash',pfx='-ZEC-Z',
  1884. mmtype='z',add_args=['mmtype=zcash_z'])
  1885. def ref_addrfile_chk_xmr(self,name):
  1886. self.ref_addrfile_chk(name,ftype='addr',coin='XMR',subdir='monero',pfx='-XMR-M')
  1887. def ref_addrfile_chk_dash(self,name):
  1888. self.ref_addrfile_chk(name,ftype='addr',coin='DASH',subdir='dash',pfx='-DASH-C')
  1889. def ref_addrfile_chk_eth(self,name):
  1890. self.ref_addrfile_chk(name,ftype='addr',coin='ETH',subdir='ethereum',pfx='-ETH')
  1891. def ref_addrfile_chk_etc(self,name):
  1892. self.ref_addrfile_chk(name,ftype='addr',coin='ETC',subdir='ethereum_classic',pfx='-ETC')
  1893. def ref_keyaddrfile_chk_zec(self,name):
  1894. self.ref_addrfile_chk(name,ftype='keyaddr',coin='ZEC',subdir='zcash',pfx='-ZEC-C')
  1895. def ref_keyaddrfile_chk_zec_z(self,name):
  1896. if skip_for_win(): return
  1897. self.ref_addrfile_chk(name,ftype='keyaddr',coin='ZEC',subdir='zcash',pfx='-ZEC-Z',
  1898. mmtype='z',add_args=['mmtype=zcash_z'])
  1899. def ref_keyaddrfile_chk_xmr(self,name):
  1900. self.ref_addrfile_chk(name,ftype='keyaddr',coin='XMR',subdir='monero',pfx='-XMR-M')
  1901. def ref_keyaddrfile_chk_dash(self,name):
  1902. self.ref_addrfile_chk(name,ftype='keyaddr',coin='DASH',subdir='dash',pfx='-DASH-C')
  1903. def ref_keyaddrfile_chk_eth(self,name):
  1904. self.ref_addrfile_chk(name,ftype='keyaddr',coin='ETH',subdir='ethereum',pfx='-ETH')
  1905. def ref_keyaddrfile_chk_etc(self,name):
  1906. self.ref_addrfile_chk(name,ftype='keyaddr',coin='ETC',subdir='ethereum_classic',pfx='-ETC')
  1907. def ref_keyaddrfile_chk(self,name):
  1908. self.ref_addrfile_chk(name,ftype='keyaddr')
  1909. def ref_passwdfile_chk(self,name):
  1910. self.ref_addrfile_chk(name,ftype='passwd')
  1911. def ref_segwitaddrfile_chk(self,name):
  1912. if not 'S' in g.proto.mmtypes:
  1913. msg_r('Skipping {} (not supported)'.format(name)); ok()
  1914. else:
  1915. self.ref_addrfile_chk(name,ftype='segwitaddr')
  1916. def ref_bech32addrfile_chk(self,name):
  1917. if not 'B' in g.proto.mmtypes:
  1918. msg_r('Skipping {} (not supported)'.format(name)); ok()
  1919. else:
  1920. self.ref_addrfile_chk(name,ftype='bech32addr')
  1921. # def txcreate8(self,name,addrfile):
  1922. # self.txcreate_common(name,sources=['8'])
  1923. def ref_tx_chk(self,name):
  1924. tf = os.path.join(ref_dir,ref_subdir,cfg['ref_tx_file'][g.coin.lower()].format(tn_ext))
  1925. wf = os.path.join(ref_dir,cfg['ref_wallet'])
  1926. write_to_tmpfile(cfg,pwfile,cfg['wpasswd'])
  1927. pf = get_tmpfile_fn(cfg,pwfile)
  1928. self.txsign(name,tf,wf,pf,save=False,has_label=True)
  1929. def ref_tool_decrypt(self,name):
  1930. f = os.path.join(ref_dir,ref_enc_fn)
  1931. t = MMGenExpect(name,'mmgen-tool', ['-q','decrypt',f,'outfile=-','hash_preset=1'])
  1932. t.passphrase('user data',tool_enc_passwd)
  1933. # t.expect("Type uppercase 'YES' to confirm: ",'YES\n') # comment out with popen_spawn
  1934. t.expect(NL,nonl=True)
  1935. import re
  1936. o = re.sub('\r\n','\n',t.read())
  1937. cmp_or_die(sample_text,o)
  1938. # wallet conversion tests
  1939. def walletconv_in(self,name,infile,desc,uopts=[],pw=False,oo=False):
  1940. opts = ['-d',cfg['tmpdir'],'-o','words',usr_rand_arg]
  1941. if_arg = [infile] if infile else []
  1942. d = '(convert)'
  1943. t = MMGenExpect(name,'mmgen-walletconv',opts+uopts+if_arg,extra_desc=d)
  1944. t.license()
  1945. if desc == 'brainwallet':
  1946. t.expect('Enter brainwallet: ',ref_wallet_brainpass+'\n')
  1947. if pw:
  1948. t.passphrase(desc,cfg['wpasswd'])
  1949. if name[:19] == 'ref_hincog_conv_old':
  1950. t.expect('Is the Seed ID correct? (Y/n): ','\n')
  1951. else:
  1952. t.expect(['Passphrase is OK',' are correct'])
  1953. # Output
  1954. wf = t.written_to_file('Mnemonic data',oo=oo)
  1955. t.close()
  1956. t.ok()
  1957. # back check of result
  1958. if opt.profile: msg('')
  1959. self.walletchk(name,wf,pf=None,
  1960. desc='mnemonic data',
  1961. sid=cfg['seed_id'],
  1962. extra_desc='(check)'
  1963. )
  1964. def walletconv_out(self,name,desc,out_fmt='w',uopts=[],uopts_chk=[],pw=False):
  1965. opts = ['-d',cfg['tmpdir'],'-p1','-o',out_fmt] + uopts
  1966. infile = os.path.join(ref_dir,cfg['seed_id']+'.mmwords')
  1967. t = MMGenExpect(name,'mmgen-walletconv',[usr_rand_arg]+opts+[infile],extra_desc='(convert)')
  1968. add_args = ['-l%s' % cfg['seed_len']]
  1969. t.license()
  1970. if pw:
  1971. t.passphrase_new('new '+desc,cfg['wpasswd'])
  1972. t.usr_rand(usr_rand_chars)
  1973. if ' '.join(desc.split()[-2:]) == 'incognito data':
  1974. for i in (1,2,3):
  1975. t.expect('Generating encryption key from OS random data ')
  1976. if desc == 'hidden incognito data':
  1977. ret = t.expect(['Create? (Y/n): ',"'YES' to confirm: "])
  1978. if ret == 0:
  1979. t.send('\n')
  1980. t.expect('Enter file size: ',str(hincog_bytes)+'\n')
  1981. else:
  1982. t.send('YES\n')
  1983. if out_fmt == 'w': t.label()
  1984. wf = t.written_to_file(capfirst(desc),oo=True)
  1985. pf = None
  1986. t.ok()
  1987. if desc == 'hidden incognito data':
  1988. add_args += uopts_chk
  1989. wf = None
  1990. if opt.profile: msg('')
  1991. self.walletchk(name,wf,pf=pf,
  1992. desc=desc,sid=cfg['seed_id'],pw=pw,
  1993. add_args=add_args,
  1994. extra_desc='(check)')
  1995. def regtest_setup(self,name):
  1996. if g.testnet:
  1997. die(2,'--testnet option incompatible with regtest test suite')
  1998. try: shutil.rmtree(os.path.join(data_dir,'regtest'))
  1999. except: pass
  2000. os.environ['MMGEN_TEST_SUITE'] = '' # mnemonic is piped to stdin, so stop being a terminal
  2001. t = MMGenExpect(name,'mmgen-regtest',['-n','setup'])
  2002. os.environ['MMGEN_TEST_SUITE'] = '1'
  2003. for s in 'Starting setup','Creating','Mined','Creating','Creating','Setup complete':
  2004. t.expect(s)
  2005. t.ok()
  2006. def regtest_walletgen(self,name,user):
  2007. t = MMGenExpect(name,'mmgen-walletgen',['-q','-r0','-p1','--'+user])
  2008. t.passphrase_new('new MMGen wallet','abc')
  2009. t.label()
  2010. t.expect('move it to the data directory? (Y/n): ','y')
  2011. t.written_to_file('MMGen wallet')
  2012. t.ok()
  2013. def regtest_walletgen_bob(self,name): return self.regtest_walletgen(name,'bob')
  2014. def regtest_walletgen_alice(self,name): return self.regtest_walletgen(name,'alice')
  2015. @staticmethod
  2016. def regtest_user_dir(user,coin=None):
  2017. return os.path.join(data_dir,'regtest',coin or g.coin.lower(),user)
  2018. def regtest_user_sid(self,user):
  2019. return os.path.basename(get_file_with_ext('mmdat',self.regtest_user_dir(user)))[:8]
  2020. def regtest_addrgen(self,name,user,wf=None,passwd='abc',addr_range='1-5'):
  2021. from mmgen.addr import MMGenAddrType
  2022. for mmtype in g.proto.mmtypes:
  2023. t = MMGenExpect(name,'mmgen-addrgen',
  2024. ['--quiet','--'+user,'--type='+mmtype,'--outdir={}'.format(self.regtest_user_dir(user))] +
  2025. ([],[wf])[bool(wf)] + [addr_range],
  2026. extra_desc='({})'.format(MMGenAddrType.mmtypes[mmtype]['name']))
  2027. t.passphrase('MMGen wallet',passwd)
  2028. t.written_to_file('Addresses')
  2029. t.ok()
  2030. def regtest_addrgen_bob(self,name): self.regtest_addrgen(name,'bob')
  2031. def regtest_addrgen_alice(self,name): self.regtest_addrgen(name,'alice')
  2032. def regtest_addrimport(self,name,user,sid=None,addr_range='1-5',num_addrs=5):
  2033. id_strs = { 'legacy':'', 'compressed':'-C', 'segwit':'-S', 'bech32':'-B' }
  2034. if not sid: sid = self.regtest_user_sid(user)
  2035. from mmgen.addr import MMGenAddrType
  2036. for mmtype in g.proto.mmtypes:
  2037. desc = MMGenAddrType.mmtypes[mmtype]['name']
  2038. fn = os.path.join(self.regtest_user_dir(user),
  2039. '{}{}{}[{}].addrs'.format(sid,altcoin_pfx,id_strs[desc],addr_range))
  2040. t = MMGenExpect(name,'mmgen-addrimport', ['--quiet','--'+user,'--batch',fn],extra_desc='('+desc+')')
  2041. t.expect('Importing')
  2042. t.expect('{} addresses imported'.format(num_addrs))
  2043. t.ok()
  2044. def regtest_addrimport_bob(self,name): self.regtest_addrimport(name,'bob')
  2045. def regtest_addrimport_alice(self,name): self.regtest_addrimport(name,'alice')
  2046. def regtest_fund_wallet(self,name,user,mmtype,amt,sid=None,addr_range='1-5'):
  2047. if not sid: sid = self.regtest_user_sid(user)
  2048. addr = self.get_addr_from_regtest_addrlist(user,sid,mmtype,0,addr_range=addr_range)
  2049. t = MMGenExpect(name,'mmgen-regtest', ['send',str(addr),str(amt)])
  2050. t.expect('Sending {} {}'.format(amt,g.coin))
  2051. t.expect('Mined 1 block')
  2052. t.ok()
  2053. def regtest_fund_bob(self,name): self.regtest_fund_wallet(name,'bob','C',rtFundAmt)
  2054. def regtest_fund_alice(self,name): self.regtest_fund_wallet(name,'alice',('L','S')[g.proto.cap('segwit')],rtFundAmt)
  2055. def regtest_user_bal(self,name,user,bal):
  2056. t = MMGenExpect(name,'mmgen-tool',['--'+user,'listaddresses','showempty=1'])
  2057. total = t.expect_getend('TOTAL: ')
  2058. cmp_or_die('{} {}'.format(bal,g.coin),total)
  2059. def regtest_alice_bal1(self,name):
  2060. return self.regtest_user_bal(name,'alice',rtFundAmt)
  2061. def regtest_bob_bal1(self,name):
  2062. return self.regtest_user_bal(name,'bob',rtFundAmt)
  2063. def regtest_bob_bal2(self,name):
  2064. return self.regtest_user_bal(name,'bob',rtBals[0])
  2065. def regtest_bob_bal3(self,name):
  2066. return self.regtest_user_bal(name,'bob',rtBals[1])
  2067. def regtest_bob_bal4(self,name):
  2068. return self.regtest_user_bal(name,'bob',rtBals[2])
  2069. def regtest_bob_bal5(self,name):
  2070. return self.regtest_user_bal(name,'bob',rtBals[3])
  2071. def regtest_bob_alice_bal(self,name):
  2072. t = MMGenExpect(name,'mmgen-regtest',['get_balances'])
  2073. t.expect('Switching')
  2074. ret = t.expect_getend("Bob's balance:").strip()
  2075. cmp_or_die(rtBals[4],ret,skip_ok=True)
  2076. ret = t.expect_getend("Alice's balance:").strip()
  2077. cmp_or_die(rtBals[5],ret,skip_ok=True)
  2078. ret = t.expect_getend("Total balance:").strip()
  2079. cmp_or_die(rtBals[6],ret,skip_ok=True)
  2080. t.ok()
  2081. def regtest_user_txdo( self,name,user,fee,
  2082. outputs_cl,
  2083. outputs_prompt,
  2084. extra_args=[],
  2085. wf=None,
  2086. pw='abc',
  2087. no_send=False,
  2088. do_label=False,
  2089. bad_locktime=False):
  2090. os.environ['MMGEN_BOGUS_SEND'] = ''
  2091. t = MMGenExpect(name,'mmgen-txdo',
  2092. ['-d',cfg['tmpdir'],'-B','--'+user] +
  2093. (['--tx-fee='+fee] if fee else []) +
  2094. extra_args + ([],[wf])[bool(wf)] + outputs_cl)
  2095. os.environ['MMGEN_BOGUS_SEND'] = '1'
  2096. t.expect(r"'q'=quit view, .*?:.",'M',regex=True) # sort by mmid
  2097. t.expect(r"'q'=quit view, .*?:.",'q',regex=True)
  2098. t.expect('outputs to spend: ',outputs_prompt+'\n')
  2099. if not fee:
  2100. t.expect('Enter transaction fee: ','{}\n'.format(tx_fee))
  2101. t.expect('OK? (Y/n): ','y') # fee OK?
  2102. t.expect('OK? (Y/n): ','y') # change OK?
  2103. if do_label:
  2104. t.expect('Add a comment to transaction? (y/N): ','y')
  2105. t.expect('Comment: ',ref_tx_label.encode('utf8')+'\n')
  2106. t.expect('View decoded transaction\? .*?: ','n',regex=True)
  2107. else:
  2108. t.expect('Add a comment to transaction? (y/N): ','\n')
  2109. t.expect('View decoded transaction\? .*?: ','t',regex=True)
  2110. t.expect('to continue: ','\n')
  2111. t.passphrase('MMGen wallet',pw)
  2112. t.written_to_file('Signed transaction')
  2113. if no_send:
  2114. t.read()
  2115. exit_val = 0
  2116. else:
  2117. t.expect('to confirm: ','YES, I REALLY WANT TO DO THIS\n')
  2118. s,exit_val = (('Transaction sent',0),("can't be included",1))[bad_locktime]
  2119. t.expect(s)
  2120. t.ok(exit_val)
  2121. def regtest_bob_split1(self,name):
  2122. sid = self.regtest_user_sid('bob')
  2123. outputs_cl = [sid+':C:1,100', sid+':L:2,200',sid+':'+rtBobOp3]
  2124. return self.regtest_user_txdo(name,'bob',rtFee[0],outputs_cl,'1',do_label=True)
  2125. def get_addr_from_regtest_addrlist(self,user,sid,mmtype,idx,addr_range='1-5'):
  2126. id_str = { 'L':'', 'S':'-S', 'C':'-C' }[mmtype]
  2127. ext = '{}{}{}[{}].addrs'.format(sid,altcoin_pfx,id_str,addr_range)
  2128. fn = get_file_with_ext(ext,self.regtest_user_dir(user),no_dot=True)
  2129. silence()
  2130. psave = g.proto
  2131. g.proto = CoinProtocol(g.coin,True)
  2132. addr = AddrList(fn).data[idx].addr
  2133. g.proto = psave
  2134. end_silence()
  2135. return addr
  2136. def create_tx_outputs(self,user,data):
  2137. sid = self.regtest_user_sid(user)
  2138. return [self.get_addr_from_regtest_addrlist(user,sid,mmtype,idx-1)+amt_str for mmtype,idx,amt_str in data]
  2139. def regtest_bob_rbf_send(self,name):
  2140. outputs_cl = self.create_tx_outputs('alice',(('L',1,',60'),('C',1,',40'))) # alice_sid:L:1, alice_sid:C:1
  2141. outputs_cl += [self.regtest_user_sid('bob')+':'+rtBobOp3]
  2142. return self.regtest_user_txdo(name,'bob',rtFee[1],outputs_cl,'3',
  2143. extra_args=([],['--rbf'])[g.proto.cap('rbf')])
  2144. def regtest_bob_send_non_mmgen(self,name):
  2145. outputs_cl = self.create_tx_outputs('alice',(
  2146. (('L','S')[g.proto.cap('segwit')],2,',10'),
  2147. (('L','S')[g.proto.cap('segwit')],3,'')
  2148. )) # alice_sid:S:2, alice_sid:S:3
  2149. fn = os.path.join(cfg['tmpdir'],'non-mmgen.keys')
  2150. return self.regtest_user_txdo(name,'bob',rtFee[3],outputs_cl,'1,4-10',
  2151. extra_args=['--keys-from-file='+fn,'--vsize-adj=1.02'])
  2152. def regtest_alice_send_estimatefee(self,name):
  2153. outputs_cl = self.create_tx_outputs('bob',(('L',1,''),)) # bob_sid:L:1
  2154. return self.regtest_user_txdo(name,'alice',None,outputs_cl,'1') # fee=None
  2155. def regtest_user_txbump(self,name,user,txfile,fee,red_op,no_send=False):
  2156. if not g.proto.cap('rbf'):
  2157. msg('Skipping RBF'); return True
  2158. os.environ['MMGEN_BOGUS_SEND'] = ''
  2159. t = MMGenExpect(name,'mmgen-txbump',
  2160. ['-d',cfg['tmpdir'],'--send','--'+user,'--tx-fee='+fee,'--output-to-reduce='+red_op] + [txfile])
  2161. os.environ['MMGEN_BOGUS_SEND'] = '1'
  2162. t.expect('OK? (Y/n): ','y') # output OK?
  2163. t.expect('OK? (Y/n): ','y') # fee OK?
  2164. t.expect('Add a comment to transaction? (y/N): ','n')
  2165. t.passphrase('MMGen wallet','abc')
  2166. t.written_to_file('Signed transaction')
  2167. if not no_send:
  2168. t.expect('to confirm: ','YES, I REALLY WANT TO DO THIS\n')
  2169. t.expect('Transaction sent')
  2170. t.written_to_file('Signed transaction')
  2171. t.read()
  2172. t.ok()
  2173. def regtest_bob_rbf_bump(self,name):
  2174. txfile = get_file_with_ext(',{}].sigtx'.format(rtFee[1][:-1]),cfg['tmpdir'],delete=False,no_dot=True)
  2175. return self.regtest_user_txbump(name,'bob',txfile,rtFee[2],'c')
  2176. def regtest_generate(self,name,coin=None,num_blocks=1):
  2177. int(num_blocks)
  2178. if coin: opt.coin = coin
  2179. t = MMGenExpect(name,'mmgen-regtest',['generate',str(num_blocks)])
  2180. t.expect('Mined {} block'.format(num_blocks))
  2181. t.ok()
  2182. def regtest_get_mempool(self,name):
  2183. ds = disable_debug()
  2184. ret = MMGenExpect(name,'mmgen-regtest',['show_mempool']).read()
  2185. restore_debug(ds)
  2186. from ast import literal_eval
  2187. return literal_eval(ret)
  2188. def regtest_get_mempool1(self,name):
  2189. mp = self.regtest_get_mempool(name)
  2190. if len(mp) != 1:
  2191. rdie(2,'Mempool has more or less than one TX!')
  2192. write_to_tmpfile(cfg,'rbf_txid',mp[0]+'\n')
  2193. ok()
  2194. def regtest_get_mempool2(self,name):
  2195. if not g.proto.cap('rbf'):
  2196. msg('Skipping post-RBF mempool check'); return True
  2197. mp = self.regtest_get_mempool(name)
  2198. if len(mp) != 1:
  2199. rdie(2,'Mempool has more or less than one TX!')
  2200. chk = read_from_tmpfile(cfg,'rbf_txid')
  2201. if chk.strip() == mp[0]:
  2202. rdie(2,'TX in mempool has not changed! RBF bump failed')
  2203. ok()
  2204. @staticmethod
  2205. def gen_pairs(n):
  2206. ds = disable_debug()
  2207. ret = [subprocess.check_output(
  2208. ['python',os.path.join('cmds','mmgen-tool'),'--testnet=1'] +
  2209. (['--type=compressed'],[])[i==0] +
  2210. ['-r0','randpair']
  2211. ).split() for i in range(n)]
  2212. restore_debug(ds)
  2213. return ret
  2214. def regtest_bob_pre_import(self,name):
  2215. pairs = self.gen_pairs(5)
  2216. write_to_tmpfile(cfg,'non-mmgen.keys','\n'.join([a[0] for a in pairs])+'\n')
  2217. write_to_tmpfile(cfg,'non-mmgen.addrs','\n'.join([a[1] for a in pairs])+'\n')
  2218. return self.regtest_user_txdo(name,'bob',rtFee[4],[pairs[0][1]],'3')
  2219. def regtest_user_import(self,name,user,args):
  2220. t = MMGenExpect(name,'mmgen-addrimport',['--quiet','--'+user]+args)
  2221. t.expect('Importing')
  2222. t.expect('OK')
  2223. t.ok()
  2224. def regtest_bob_import_addr(self,name):
  2225. addr = read_from_tmpfile(cfg,'non-mmgen.addrs').split()[0]
  2226. return self.regtest_user_import(name,'bob',['--rescan','--address='+addr])
  2227. def regtest_bob_import_list(self,name):
  2228. fn = os.path.join(cfg['tmpdir'],'non-mmgen.addrs')
  2229. return self.regtest_user_import(name,'bob',['--addrlist',fn])
  2230. def regtest_bob_split2(self,name):
  2231. addrs = read_from_tmpfile(cfg,'non-mmgen.addrs').split()
  2232. amts = (a for a in (1.12345678,2.87654321,3.33443344,4.00990099,5.43214321))
  2233. outputs1 = ['{},{}'.format(a,amts.next()) for a in addrs]
  2234. sid = self.regtest_user_sid('bob')
  2235. l1,l2 = (':S',':B') if g.proto.cap('segwit') else (':L',':L')
  2236. outputs2 = [sid+':C:2,6.333', sid+':L:3,6.667',sid+l1+':4,0.123',sid+l2+':5']
  2237. return self.regtest_user_txdo(name,'bob',rtFee[5],outputs1+outputs2,'1-2')
  2238. def regtest_user_add_label(self,name,user,addr,label):
  2239. t = MMGenExpect(name,'mmgen-tool',['--'+user,'add_label',addr,label])
  2240. t.expect('Added label.*in tracking wallet',regex=True)
  2241. t.ok()
  2242. def regtest_user_remove_label(self,name,user,addr):
  2243. t = MMGenExpect(name,'mmgen-tool',['--'+user,'remove_label',addr])
  2244. t.expect('Removed label.*in tracking wallet',regex=True)
  2245. t.ok()
  2246. def regtest_alice_add_label1(self,name):
  2247. sid = self.regtest_user_sid('alice')
  2248. return self.regtest_user_add_label(name,'alice',sid+':C:1','Original Label')
  2249. def regtest_alice_add_label2(self,name):
  2250. sid = self.regtest_user_sid('alice')
  2251. return self.regtest_user_add_label(name,'alice',sid+':C:1','Replacement Label')
  2252. def regtest_alice_add_label_coinaddr(self,name):
  2253. mmaddr = self.regtest_user_sid('alice') + ':C:2'
  2254. t = MMGenExpect(name,'mmgen-tool',['--alice','listaddress',mmaddr],no_msg=True)
  2255. btcaddr = [i for i in t.read().splitlines() if i.lstrip()[0:len(mmaddr)] == mmaddr][0].split()[1]
  2256. return self.regtest_user_add_label(name,'alice',btcaddr,'Label added using coin address')
  2257. def regtest_alice_chk_label_coinaddr(self,name):
  2258. sid = self.regtest_user_sid('alice')
  2259. return self.regtest_user_chk_label(name,'alice',sid+':C:2','Label added using coin address')
  2260. def regtest_alice_add_label_badaddr(self,name,addr,reply):
  2261. t = MMGenExpect(name,'mmgen-tool',['--alice','add_label',addr,'(none)'])
  2262. t.expect(reply,regex=True)
  2263. t.ok()
  2264. def regtest_alice_add_label_badaddr1(self,name):
  2265. return self.regtest_alice_add_label_badaddr(name,'abc','Invalid coin address for this chain: abc')
  2266. def regtest_alice_add_label_badaddr2(self,name):
  2267. addr = g.proto.pubhash2addr('00'*20,False) # mainnet zero address
  2268. return self.regtest_alice_add_label_badaddr(name,addr,'Invalid coin address for this chain: '+addr)
  2269. def regtest_alice_add_label_badaddr3(self,name):
  2270. addr = self.regtest_user_sid('alice') + ':C:123'
  2271. return self.regtest_alice_add_label_badaddr(name,addr,
  2272. "MMGen address '{}' not found in tracking wallet".format(addr))
  2273. def regtest_alice_add_label_badaddr4(self,name):
  2274. addr = CoinProtocol(g.coin,True).pubhash2addr('00'*20,False) # testnet zero address
  2275. return self.regtest_alice_add_label_badaddr(name,addr,
  2276. "Address '{}' not found in tracking wallet".format(addr))
  2277. def regtest_alice_add_label_rpcfail(self,name):
  2278. addr = self.regtest_user_sid('alice') + ':C:2'
  2279. os.environ['MMGEN_RPC_FAIL_ON_COMMAND'] = 'importaddress'
  2280. self.regtest_alice_add_label_badaddr(name,addr,'Label could not be added')
  2281. os.environ['MMGEN_RPC_FAIL_ON_COMMAND'] = ''
  2282. def regtest_alice_remove_label1(self,name):
  2283. sid = self.regtest_user_sid('alice')
  2284. return self.regtest_user_remove_label(name,'alice',sid+':C:1')
  2285. def regtest_user_chk_label(self,name,user,addr,label):
  2286. t = MMGenExpect(name,'mmgen-tool',['--'+user,'listaddresses','all_labels=1'])
  2287. t.expect('{}\s+\S{{30}}\S+\s+{}\s+'.format(addr,label),regex=True)
  2288. t.ok()
  2289. def regtest_alice_chk_label1(self,name):
  2290. sid = self.regtest_user_sid('alice')
  2291. return self.regtest_user_chk_label(name,'alice',sid+':C:1','Original Label')
  2292. def regtest_alice_chk_label2(self,name):
  2293. sid = self.regtest_user_sid('alice')
  2294. return self.regtest_user_chk_label(name,'alice',sid+':C:1','Replacement Label')
  2295. def regtest_alice_chk_label3(self,name):
  2296. sid = self.regtest_user_sid('alice')
  2297. return self.regtest_user_chk_label(name,'alice',sid+':C:1','Edited Label')
  2298. def regtest_alice_chk_label4(self,name):
  2299. sid = self.regtest_user_sid('alice')
  2300. return self.regtest_user_chk_label(name,'alice',sid+':C:1','-')
  2301. def regtest_user_edit_label(self,name,user,output,label):
  2302. t = MMGenExpect(name,'mmgen-txcreate',['-B','--'+user,'-i'])
  2303. t.expect(r"'q'=quit view, .*?:.",'M',regex=True)
  2304. t.expect(r"'q'=quit view, .*?:.",'l',regex=True)
  2305. t.expect(r"Enter unspent.*return to main menu\):.",output+'\n',regex=True)
  2306. t.expect(r"Enter label text.*return to main menu\):.",label+'\n',regex=True)
  2307. t.expect(r"'q'=quit view, .*?:.",'q',regex=True)
  2308. t.ok()
  2309. def regtest_alice_edit_label1(self,name):
  2310. return self.regtest_user_edit_label(name,'alice','1','Edited Label')
  2311. def regtest_stop(self,name):
  2312. t = MMGenExpect(name,'mmgen-regtest',['stop'])
  2313. t.ok()
  2314. def regtest_split_setup(self,name):
  2315. if g.coin != 'BTC': die(1,'Test valid only for coin BTC')
  2316. opt.coin = 'BTC'
  2317. return self.regtest_setup(name)
  2318. def regtest_split_fork(self,name):
  2319. opt.coin = 'B2X'
  2320. t = MMGenExpect(name,'mmgen-regtest',['fork','btc'])
  2321. t.expect('Creating fork from coin')
  2322. t.expect('successfully created')
  2323. t.ok()
  2324. def regtest_split_start(self,name,coin):
  2325. opt.coin = coin
  2326. t = MMGenExpect(name,'mmgen-regtest',['bob'])
  2327. t.expect('Starting')
  2328. t.expect('done')
  2329. t.ok()
  2330. def regtest_split_start_btc(self,name): self.regtest_split_start(name,coin='BTC')
  2331. def regtest_split_start_b2x(self,name): self.regtest_split_start(name,coin='B2X')
  2332. def regtest_split_gen_btc(self,name): self.regtest_generate(name,coin='BTC')
  2333. def regtest_split_gen_b2x(self,name): self.regtest_generate(name,coin='B2X',num_blocks=100)
  2334. def regtest_split_gen_b2x2(self,name): self.regtest_generate(name,coin='B2X')
  2335. def regtest_split_do_split(self,name):
  2336. opt.coin = 'B2X'
  2337. sid = self.regtest_user_sid('bob')
  2338. t = MMGenExpect(name,'mmgen-split',[
  2339. '--bob',
  2340. '--outdir='+cfg['tmpdir'],
  2341. '--tx-fees=0.0001,0.0003',
  2342. sid+':S:1',sid+':S:2'])
  2343. t.expect(r"'q'=quit view, .*?:.",'q', regex=True)
  2344. t.expect('outputs to spend: ','1\n')
  2345. for tx in ('timelocked','split'):
  2346. for q in ('fee','change'): t.expect('OK? (Y/n): ','y')
  2347. t.expect('Add a comment to transaction? (y/N): ','n')
  2348. t.expect('View decoded transaction\? .*?: ','t',regex=True)
  2349. t.expect('to continue: ','\n')
  2350. t.written_to_file('Long chain (timelocked) transaction')
  2351. t.written_to_file('Short chain transaction')
  2352. t.ok()
  2353. def regtest_split_sign(self,name,coin,ext):
  2354. wf = get_file_with_ext('mmdat',self.regtest_user_dir('bob',coin=coin.lower()))
  2355. txfile = get_file_with_ext(ext,cfg['tmpdir'],no_dot=True)
  2356. opt.coin = coin
  2357. self.txsign(name,txfile,wf,extra_opts=['--bob'])
  2358. def regtest_split_sign_b2x(self,name):
  2359. return self.regtest_split_sign(name,coin='B2X',ext='533].rawtx')
  2360. def regtest_split_sign_btc(self,name):
  2361. return self.regtest_split_sign(name,coin='BTC',ext='9997].rawtx')
  2362. def regtest_split_send(self,name,coin,ext):
  2363. opt.coin = coin
  2364. txfile = get_file_with_ext(ext,cfg['tmpdir'],no_dot=True)
  2365. self.txsend(name,txfile,really_send=True,extra_opts=['--bob'])
  2366. def regtest_split_send_b2x(self,name):
  2367. return self.regtest_split_send(name,coin='B2X',ext='533].sigtx')
  2368. def regtest_split_send_btc(self,name):
  2369. return self.regtest_split_send(name,coin='BTC',ext='9997].sigtx')
  2370. def regtest_split_txdo_timelock(self,name,coin,locktime,bad_locktime):
  2371. opt.coin = coin
  2372. sid = self.regtest_user_sid('bob')
  2373. self.regtest_user_txdo(
  2374. name,'bob','0.0001',[sid+':S:5'],'1',pw='abc',
  2375. extra_args=['--locktime='+str(locktime)],
  2376. bad_locktime=bad_locktime)
  2377. def regtest_split_txdo_timelock_bad_btc(self,name):
  2378. self.regtest_split_txdo_timelock(name,'BTC',locktime=8888,bad_locktime=True)
  2379. def regtest_split_txdo_timelock_good_btc(self,name):
  2380. self.regtest_split_txdo_timelock(name,'BTC',locktime=1321009871,bad_locktime=False)
  2381. def regtest_split_txdo_timelock_bad_b2x(self,name):
  2382. self.regtest_split_txdo_timelock(name,'B2X',locktime=8888,bad_locktime=True)
  2383. def regtest_split_txdo_timelock_good_b2x(self,name):
  2384. self.regtest_split_txdo_timelock(name,'B2X',locktime=1321009871,bad_locktime=False)
  2385. # def regtest_user_txdo(self,name,user,fee,outputs_cl,outputs_prompt,extra_args=[],wf=None,pw='abc',no_send=False,do_label=False):
  2386. # undocumented admin commands
  2387. ref_tx_setup = regtest_setup
  2388. ref_tx_generate = regtest_generate
  2389. def ref_tx_addrgen_bob_ref_wallet(self,name):
  2390. wf = os.path.join(ref_dir,cfg['ref_wallet'])
  2391. self.regtest_addrgen(name,'bob',wf=wf,passwd=cfg['wpasswd'],addr_range='1-100')
  2392. def ref_tx_addrimport_bob_ref_wallet(self,name):
  2393. self.regtest_addrimport(name,'bob',sid=cfg['seed_id'],addr_range='1-100',num_addrs=100)
  2394. def ref_tx_fund_bob(self,name):
  2395. mmtype = g.proto.mmtypes[-1]
  2396. self.regtest_fund_wallet(name,'bob',mmtype,rtFundAmt,sid=cfg['seed_id'],addr_range='1-100')
  2397. def ref_tx_bob_split(self,name):
  2398. sid = cfg['seed_id']
  2399. outputs_cl = [sid+':C:1,100', sid+':L:2,200',sid+':'+rtBobOp3]
  2400. wf = os.path.join(ref_dir,cfg['ref_wallet'])
  2401. return self.regtest_user_txdo(name,'bob',rtFee[0],outputs_cl,'1',do_label=True,wf=wf,pw=cfg['wpasswd'])
  2402. def ref_tx_bob_create_tx(self,name):
  2403. sid = cfg['seed_id']
  2404. psave = g.proto
  2405. g.proto = CoinProtocol(g.coin,True)
  2406. privhex = PrivKey(os.urandom(32),compressed=True,pubkey_type='std')
  2407. addr = AddrGenerator('p2pkh').to_addr(KeyGenerator('std').to_pubhex(privhex))
  2408. g.proto = psave
  2409. outputs_cl = [sid+':{}:3,1.1234'.format(g.proto.mmtypes[-1]), sid+':C:5,5.5555',sid+':L:4',addr+',100']
  2410. pw = cfg['wpasswd']
  2411. # create tx in cwd
  2412. t = MMGenExpect(name,'mmgen-txcreate',[
  2413. '-B',
  2414. '--bob',
  2415. '--tx-fee='+rtFee[0],
  2416. '--locktime=1320969600'
  2417. ] + outputs_cl)
  2418. # [os.path.join(ref_dir,cfg['ref_wallet'])])
  2419. t.expect(r"'q'=quit view, .*?:.",'M',regex=True) # sort by mmid
  2420. t.expect(r"'q'=quit view, .*?:.",'q',regex=True)
  2421. t.expect('outputs to spend: ','1 2 3\n')
  2422. t.expect('OK? (Y/n): ','y') # fee OK?
  2423. t.expect('OK? (Y/n): ','y') # change OK?
  2424. t.expect('Add a comment to transaction? (y/N): ','y')
  2425. t.expect('Comment: ',ref_tx_label.encode('utf8')+'\n')
  2426. t.expect('View decoded transaction\? .*?: ','n',regex=True)
  2427. t.expect('Save transaction? (y/N): ','y')
  2428. fn = t.written_to_file('Transaction')
  2429. write_to_tmpfile(cfg,'ref_tx_fn',fn)
  2430. t.ok()
  2431. def ref_tx_bob_modify_tx(self,name):
  2432. MMGenExpect(name,'',msg_only=True)
  2433. fn = read_from_tmpfile(cfg,'ref_tx_fn')
  2434. with open(fn) as f:
  2435. lines = f.read().splitlines()
  2436. tx = {}
  2437. from ast import literal_eval
  2438. for k,i in (('in',3),('out',4)):
  2439. tx[k] = literal_eval(lines[i])
  2440. tx[k+'_addrs'] = [i['addr'] for i in tx[k]]
  2441. psave = g.proto
  2442. g.proto = CoinProtocol(g.coin,True)
  2443. from mmgen.obj import CoinAddr
  2444. for k in ('in_addrs','out_addrs'):
  2445. tx[k+'_hex'] = [(CoinAddr(a).hex,CoinAddr(a).addr_fmt) for a in tx[k]]
  2446. g.proto = psave
  2447. for k in ('in_addrs','out_addrs'):
  2448. tx[k+'_conv'] = [g.proto.pubhash2addr(h,(False,True)[f=='p2sh']) for h,f in tx[k+'_hex']]
  2449. for k in ('in','out'):
  2450. for i in range(len(tx[k])):
  2451. tx[k][i]['addr'] = tx[k+'_addrs_conv'][i]
  2452. lines_tn = [lines[1].replace('REGTEST','TESTNET')] + lines[2:]
  2453. o_tn = '\n'.join([make_chksum_6(' '.join(lines_tn))] + lines_tn)+'\n'
  2454. fn_tn = fn.replace('.rawtx','.testnet.rawtx')
  2455. lines_mn = [lines[1].replace('REGTEST','MAINNET'),
  2456. lines[2],
  2457. repr(tx['in']),
  2458. repr(tx['out'])] + lines[5:]
  2459. o_mn = '\n'.join([make_chksum_6(' '.join(lines_mn))] + lines_mn)+'\n'
  2460. fn_mn = fn.replace('.rawtx','.mainnet.rawtx')
  2461. ok()
  2462. write_data_to_file(fn_tn,o_tn,'testnet TX data',ask_overwrite=False)
  2463. write_data_to_file(fn_mn,o_mn,'mainnet TX data',ask_overwrite=False)
  2464. # END methods
  2465. for k in (
  2466. 'ref_wallet_conv',
  2467. 'ref_mn_conv',
  2468. 'ref_seed_conv',
  2469. 'ref_hex_conv',
  2470. 'ref_brain_conv',
  2471. 'ref_incog_conv',
  2472. 'ref_incox_conv',
  2473. 'ref_hincog_conv',
  2474. 'ref_hincog_conv_old',
  2475. 'ref_wallet_conv_out',
  2476. 'ref_mn_conv_out',
  2477. 'ref_seed_conv_out',
  2478. 'ref_hex_conv_out',
  2479. 'ref_incog_conv_out',
  2480. 'ref_incox_conv_out',
  2481. 'ref_hincog_conv_out',
  2482. 'ref_wallet_chk',
  2483. 'refwalletgen',
  2484. 'ref_seed_chk',
  2485. 'ref_hex_chk',
  2486. 'ref_mn_chk',
  2487. 'ref_brain_chk',
  2488. 'ref_hincog_chk',
  2489. 'refaddrgen',
  2490. 'refkeyaddrgen',
  2491. 'refaddrgen_compressed',
  2492. 'refkeyaddrgen_compressed',
  2493. 'refpasswdgen',
  2494. 'ref_b32passwdgen',
  2495. 'ref_hexpasswdgen'
  2496. ):
  2497. for i in ('1','2','3'):
  2498. locals()[k+i] = locals()[k]
  2499. for k in ('walletgen','addrgen','keyaddrgen'): locals()[k+'14'] = locals()[k]
  2500. # create temporary dirs
  2501. if not opt.resume and not opt.skip_deps:
  2502. if g.platform == 'win':
  2503. for cfg in sorted(cfgs):
  2504. mk_tmpdir(cfgs[cfg]['tmpdir'])
  2505. else:
  2506. for cfg in sorted(cfgs):
  2507. src = os.path.join(shm_dir,cfgs[cfg]['tmpdir'].split('/')[-1])
  2508. mk_tmpdir(src)
  2509. try:
  2510. os.unlink(cfgs[cfg]['tmpdir'])
  2511. except OSError as e:
  2512. if e.errno != 2: raise
  2513. finally:
  2514. os.symlink(src,cfgs[cfg]['tmpdir'])
  2515. have_dfl_wallet = False
  2516. # main()
  2517. if opt.pause:
  2518. import termios,atexit
  2519. fd = sys.stdin.fileno()
  2520. old = termios.tcgetattr(fd)
  2521. def at_exit():
  2522. termios.tcsetattr(fd, termios.TCSADRAIN, old)
  2523. atexit.register(at_exit)
  2524. start_time = int(time.time())
  2525. def end_msg():
  2526. t = int(time.time()) - start_time
  2527. m = '{} test{} performed. Elapsed time: {:02d}:{:02d}\n'
  2528. sys.stderr.write(green(m.format(cmd_total,suf(cmd_total),t/60,t%60)))
  2529. ts = MMGenTestSuite()
  2530. if cmd_args and cmd_args[0] == 'admin':
  2531. cmd_args.pop(0)
  2532. cmd_data = cmd_data_admin
  2533. cmd_list = cmd_list_admin
  2534. try:
  2535. if cmd_args:
  2536. for arg in cmd_args:
  2537. if arg in utils:
  2538. globals()[arg](cmd_args[cmd_args.index(arg)+1:])
  2539. sys.exit(0)
  2540. elif 'info_'+arg in cmd_data:
  2541. dirs = cmd_data['info_'+arg][1]
  2542. if dirs: clean(dirs)
  2543. for cmd in cmd_list[arg]:
  2544. check_needs_rerun(ts,cmd,build=True)
  2545. elif arg in meta_cmds:
  2546. for cmd in meta_cmds[arg]:
  2547. check_needs_rerun(ts,cmd,build=True)
  2548. elif arg in cmd_data:
  2549. check_needs_rerun(ts,arg,build=True)
  2550. else:
  2551. die(1,'%s: unrecognized command' % arg)
  2552. else:
  2553. clean()
  2554. for cmd in cmd_data:
  2555. if cmd == 'info_regtest': break # don't run everything after this by default
  2556. if cmd[:5] == 'info_':
  2557. msg(green('%sTesting %s' % (('\n','')[bool(opt.resume)],cmd_data[cmd][0])))
  2558. continue
  2559. ts.do_cmd(cmd)
  2560. if cmd is not cmd_data.keys()[-1]: do_between()
  2561. except KeyboardInterrupt:
  2562. die(1,'\nExiting at user request')
  2563. except opt.traceback and Exception:
  2564. with open('my.err') as f:
  2565. t = f.readlines()
  2566. if t: msg_r('\n'+yellow(''.join(t[:-1]))+red(t[-1]))
  2567. die(1,blue('Test script exited with error'))
  2568. except:
  2569. sys.stderr = stderr_save
  2570. raise
  2571. end_msg()