test.py 72 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105210621072108210921102111211221132114211521162117211821192120212121222123212421252126212721282129213021312132213321342135213621372138213921402141214221432144214521462147214821492150215121522153215421552156215721582159
  1. #!/usr/bin/env python
  2. # -*- coding: UTF-8 -*-
  3. # mmgen = Multi-Mode GENerator, command-line Bitcoin cold storage solution
  4. # Copyright (C)2013-2016 Philemon <mmgen-py@yandex.com>
  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
  22. def run_in_tb():
  23. fn = sys.argv[0]
  24. source = open(fn)
  25. try:
  26. exec source in {'inside_tb':1}
  27. except SystemExit:
  28. pass
  29. except:
  30. def color(s): return '\033[36;1m' + s + '\033[0m'
  31. e = sys.exc_info()
  32. sys.stdout.write(color('\nTest script returned: %s\n' % (e[0].__name__)))
  33. if not 'inside_tb' in globals() and 'MMGEN_TEST_TRACEBACK' in os.environ:
  34. run_in_tb()
  35. sys.exit()
  36. pn = os.path.dirname(sys.argv[0])
  37. os.chdir(os.path.join(pn,os.pardir))
  38. sys.path.__setitem__(0,os.path.abspath(os.curdir))
  39. # Import these _after_ local path's been added to sys.path
  40. from mmgen.common import *
  41. from mmgen.test import *
  42. g.quiet = False # if 'quiet' was set in config file, disable here
  43. os.environ['MMGEN_QUIET'] = '0' # and for the spawned scripts
  44. tb_cmd = 'scripts/traceback.py'
  45. log_file = 'test.py_log'
  46. scripts = (
  47. 'addrgen', 'addrimport', 'keygen',
  48. 'passchg', 'tool',
  49. 'txcreate', 'txsend', 'txsign',
  50. 'walletchk', 'walletconv', 'walletgen'
  51. )
  52. hincog_fn = 'rand_data'
  53. hincog_bytes = 1024*1024
  54. hincog_offset = 98765
  55. hincog_seedlen = 256
  56. incog_id_fn = 'incog_id'
  57. non_mmgen_fn = 'btckey'
  58. pwfile = 'passwd_file'
  59. ref_dir = os.path.join('test','ref')
  60. ref_wallet_brainpass = 'abc'
  61. ref_wallet_hash_preset = '1'
  62. ref_wallet_incog_offset = 123
  63. from mmgen.obj import MMGenTXLabel
  64. ref_tx_label = ''.join([unichr(i) for i in range(65,91) +
  65. range(1040,1072) + # cyrillic
  66. range(913,939) + # greek
  67. range(97,123)])[:MMGenTXLabel.max_len]
  68. ref_bw_hash_preset = '1'
  69. ref_bw_file = 'wallet.mmbrain'
  70. ref_bw_file_spc = 'wallet-spaced.mmbrain'
  71. ref_kafile_pass = 'kafile password'
  72. ref_kafile_hash_preset = '1'
  73. ref_enc_fn = 'sample-text.mmenc'
  74. tool_enc_passwd = "Scrypt it, don't hash it!"
  75. sample_text = \
  76. 'The Times 03/Jan/2009 Chancellor on brink of second bailout for banks\n'
  77. # Laggy flash media cause pexpect to crash, so create a temporary directory
  78. # under '/dev/shm' and put datadir and temp files here.
  79. shortopts = ''.join([e[1:] for e in sys.argv if len(e) > 1 and e[0] == '-' and e[1] != '-'])
  80. shortopts = ['-'+e for e in list(shortopts)]
  81. data_dir = os.path.join('test','data_dir')
  82. if not any(e in ('--skip-deps','--resume','-S','-r') for e in sys.argv+shortopts):
  83. if g.platform == 'win':
  84. try: os.listdir(data_dir)
  85. except: pass
  86. else:
  87. import shutil
  88. shutil.rmtree(data_dir)
  89. os.mkdir(data_dir,0755)
  90. else:
  91. d,pfx = '/dev/shm','mmgen-test-'
  92. try:
  93. import subprocess
  94. subprocess.call('rm -rf %s/%s*'%(d,pfx),shell=True)
  95. except Exception as e:
  96. die(2,'Unable to delete directory tree %s/%s* (%s)'%(d,pfx,e))
  97. try:
  98. import tempfile
  99. shm_dir = tempfile.mkdtemp('',pfx,d)
  100. except Exception as e:
  101. die(2,'Unable to create temporary directory in %s (%s)'%(d,e))
  102. dd = os.path.join(shm_dir,'data_dir')
  103. os.mkdir(dd,0755)
  104. try: os.unlink(data_dir)
  105. except: pass
  106. os.symlink(dd,data_dir)
  107. opts_data = {
  108. # 'sets': [('interactive',bool,'verbose',None)],
  109. 'desc': 'Test suite for the MMGen suite',
  110. 'usage':'[options] [command(s) or metacommand(s)]',
  111. 'options': """
  112. -h, --help Print this help message
  113. --, --longhelp Print help message for long options (common options)
  114. -b, --buf-keypress Use buffered keypresses as with real human input
  115. -c, --print-cmdline Print the command line of each spawned command
  116. -d, --debug-scripts Turn on debugging output in executed scripts
  117. -x, --debug-pexpect Produce debugging output for pexpect calls
  118. -D, --direct-exec Bypass pexpect and execute a command directly (for
  119. debugging only)
  120. -e, --exact-output Show the exact output of the MMGen script(s) being run
  121. -l, --list-cmds List and describe the commands in the test suite
  122. -L, --log Log commands to file {lf}
  123. -n, --names Display command names instead of descriptions
  124. -I, --interactive Interactive mode (without pexpect)
  125. -O, --popen-spawn Use pexpect's popen_spawn instead of popen
  126. -p, --pause Pause between tests, resuming on keypress
  127. -P, --profile Record the execution time of each script
  128. -q, --quiet Produce minimal output. Suppress dependency info
  129. -r, --resume=c Resume at command 'c' after interrupted run
  130. -s, --system Test scripts and modules installed on system rather
  131. than those in the repo root
  132. -S, --skip-deps Skip dependency checking for command
  133. -u, --usr-random Get random data interactively from user
  134. -t, --traceback Run the command inside the '{tb_cmd}' script
  135. -v, --verbose Produce more verbose output
  136. -W, --no-dw-delete Don't remove default wallet from data dir after dw tests are done
  137. """.format(tb_cmd=tb_cmd,lf=log_file),
  138. 'notes': """
  139. If no command is given, the whole suite of tests is run.
  140. """
  141. }
  142. sys.argv = [sys.argv[0]] + ['--data-dir',data_dir] + sys.argv[1:]
  143. cmd_args = opts.init(opts_data)
  144. tn_desc = ('','.testnet')[g.testnet]
  145. cfgs = {
  146. '15': {
  147. 'tmpdir': os.path.join('test','tmp15'),
  148. 'wpasswd': 'Dorian',
  149. 'kapasswd': 'Grok the blockchain',
  150. 'addr_idx_list': '12,99,5-10,5,12', # 8 addresses
  151. 'dep_generators': {
  152. pwfile: 'walletgen_dfl_wallet',
  153. 'addrs': 'addrgen_dfl_wallet',
  154. 'rawtx': 'txcreate_dfl_wallet',
  155. 'sigtx': 'txsign_dfl_wallet',
  156. 'mmseed': 'export_seed_dfl_wallet',
  157. 'del_dw_run': 'delete_dfl_wallet',
  158. },
  159. },
  160. '16': {
  161. 'tmpdir': os.path.join('test','tmp16'),
  162. 'wpasswd': 'My changed password',
  163. 'hash_preset': '2',
  164. 'dep_generators': {
  165. pwfile: 'passchg_dfl_wallet',
  166. },
  167. },
  168. '1': {
  169. 'tmpdir': os.path.join('test','tmp1'),
  170. 'wpasswd': 'Dorian',
  171. 'kapasswd': 'Grok the blockchain',
  172. 'addr_idx_list': '12,99,5-10,5,12', # 8 addresses
  173. 'dep_generators': {
  174. pwfile: 'walletgen',
  175. 'mmdat': 'walletgen',
  176. 'addrs': 'addrgen',
  177. 'rawtx': 'txcreate',
  178. 'sigtx': 'txsign',
  179. 'mmwords': 'export_mnemonic',
  180. 'mmseed': 'export_seed',
  181. 'mmincog': 'export_incog',
  182. 'mmincox': 'export_incog_hex',
  183. hincog_fn: 'export_incog_hidden',
  184. incog_id_fn: 'export_incog_hidden',
  185. 'akeys.mmenc': 'keyaddrgen'
  186. },
  187. },
  188. '2': {
  189. 'tmpdir': os.path.join('test','tmp2'),
  190. 'wpasswd': 'Hodling away',
  191. 'addr_idx_list': '37,45,3-6,22-23', # 8 addresses
  192. 'seed_len': 128,
  193. 'dep_generators': {
  194. 'mmdat': 'walletgen2',
  195. 'addrs': 'addrgen2',
  196. 'rawtx': 'txcreate2',
  197. 'sigtx': 'txsign2',
  198. 'mmwords': 'export_mnemonic2',
  199. },
  200. },
  201. '3': {
  202. 'tmpdir': os.path.join('test','tmp3'),
  203. 'wpasswd': 'Major miner',
  204. 'addr_idx_list': '73,54,1022-1023,2-5', # 8 addresses
  205. 'dep_generators': {
  206. 'mmdat': 'walletgen3',
  207. 'addrs': 'addrgen3',
  208. 'rawtx': 'txcreate3',
  209. 'sigtx': 'txsign3'
  210. },
  211. },
  212. '4': {
  213. 'tmpdir': os.path.join('test','tmp4'),
  214. 'wpasswd': 'Hashrate good',
  215. 'addr_idx_list': '63,1004,542-544,7-9', # 8 addresses
  216. 'seed_len': 192,
  217. 'dep_generators': {
  218. 'mmdat': 'walletgen4',
  219. 'mmbrain': 'walletgen4',
  220. 'addrs': 'addrgen4',
  221. 'rawtx': 'txcreate4',
  222. 'sigtx': 'txsign4',
  223. },
  224. 'bw_filename': 'brainwallet.mmbrain',
  225. 'bw_params': '192,1',
  226. },
  227. '14': {
  228. 'kapasswd': 'Maxwell',
  229. 'tmpdir': os.path.join('test','tmp14'),
  230. 'wpasswd': 'The Halving',
  231. 'addr_idx_list': '61,998,502-504,7-9', # 8 addresses
  232. 'seed_len': 256,
  233. 'dep_generators': {
  234. 'mmdat': 'walletgen14',
  235. 'addrs': 'addrgen14',
  236. 'akeys.mmenc': 'keyaddrgen14',
  237. },
  238. },
  239. '5': {
  240. 'tmpdir': os.path.join('test','tmp5'),
  241. 'wpasswd': 'My changed password',
  242. 'hash_preset': '2',
  243. 'dep_generators': {
  244. 'mmdat': 'passchg',
  245. pwfile: 'passchg',
  246. },
  247. },
  248. '6': {
  249. 'name': 'reference wallet check (128-bit)',
  250. 'seed_len': 128,
  251. 'seed_id': 'FE3C6545',
  252. 'ref_bw_seed_id': '33F10310',
  253. 'addrfile_chk': ('B230 7526 638F 38CB','B64D 7327 EF2A 60FE')[g.testnet],
  254. 'keyaddrfile_chk': ('CF83 32FB 8A8B 08E2','FEBF 7878 97BB CC35')[g.testnet],
  255. 'wpasswd': 'reference password',
  256. 'ref_wallet': 'FE3C6545-D782B529[128,1].mmdat',
  257. 'ic_wallet': 'FE3C6545-E29303EA-5E229E30[128,1].mmincog',
  258. 'ic_wallet_hex': 'FE3C6545-BC4BE3F2-32586837[128,1].mmincox',
  259. 'hic_wallet': 'FE3C6545-161E495F-BEB7548E[128,1].incog-offset123',
  260. 'hic_wallet_old': 'FE3C6545-161E495F-9860A85B[128,1].incog-old.offset123',
  261. 'tmpdir': os.path.join('test','tmp6'),
  262. 'kapasswd': '',
  263. 'addr_idx_list': '1010,500-501,31-33,1,33,500,1011', # 8 addresses
  264. 'dep_generators': {
  265. 'mmdat': 'refwalletgen1',
  266. pwfile: 'refwalletgen1',
  267. 'addrs': 'refaddrgen1',
  268. 'akeys.mmenc': 'refkeyaddrgen1'
  269. },
  270. },
  271. '7': {
  272. 'name': 'reference wallet check (192-bit)',
  273. 'seed_len': 192,
  274. 'seed_id': '1378FC64',
  275. 'ref_bw_seed_id': 'CE918388',
  276. 'addrfile_chk': ('8C17 A5FA 0470 6E89','0A59 C8CD 9439 8B81')[g.testnet],
  277. 'keyaddrfile_chk': ('9648 5132 B98E 3AD9','2F72 C83F 44C5 0FAC')[g.testnet],
  278. 'wpasswd': 'reference password',
  279. 'ref_wallet': '1378FC64-6F0F9BB4[192,1].mmdat',
  280. 'ic_wallet': '1378FC64-2907DE97-F980D21F[192,1].mmincog',
  281. 'ic_wallet_hex': '1378FC64-4DCB5174-872806A7[192,1].mmincox',
  282. 'hic_wallet': '1378FC64-B55E9958-77256FC1[192,1].incog.offset123',
  283. 'hic_wallet_old': '1378FC64-B55E9958-D85FF20C[192,1].incog-old.offset123',
  284. 'tmpdir': os.path.join('test','tmp7'),
  285. 'kapasswd': '',
  286. 'addr_idx_list': '1010,500-501,31-33,1,33,500,1011', # 8 addresses
  287. 'dep_generators': {
  288. 'mmdat': 'refwalletgen2',
  289. pwfile: 'refwalletgen2',
  290. 'addrs': 'refaddrgen2',
  291. 'akeys.mmenc': 'refkeyaddrgen2'
  292. },
  293. },
  294. '8': {
  295. 'name': 'reference wallet check (256-bit)',
  296. 'seed_len': 256,
  297. 'seed_id': '98831F3A',
  298. 'ref_bw_seed_id': 'B48CD7FC',
  299. 'addrfile_chk': ('6FEF 6FB9 7B13 5D91','3C2C 8558 BB54 079E')[g.testnet],
  300. 'keyaddrfile_chk': ('9F2D D781 1812 8BAD','7410 8F95 4B33 B4B2')[g.testnet],
  301. 'wpasswd': 'reference password',
  302. 'ref_wallet': '98831F3A-{}[256,1].mmdat'.format(('27F2BF93','E2687906')[g.testnet]),
  303. 'ref_addrfile': '98831F3A[1,31-33,500-501,1010-1011]{}.addrs'.format(tn_desc),
  304. 'ref_keyaddrfile': '98831F3A[1,31-33,500-501,1010-1011]{}.akeys.mmenc'.format(tn_desc),
  305. 'ref_addrfile_chksum': ('6FEF 6FB9 7B13 5D91','3C2C 8558 BB54 079E')[g.testnet],
  306. 'ref_keyaddrfile_chksum': ('9F2D D781 1812 8BAD','7410 8F95 4B33 B4B2')[g.testnet],
  307. # 'ref_fake_unspent_data':'98831F3A_unspent.json',
  308. 'ref_tx_file': 'FFB367[1.234]{}.rawtx'.format(tn_desc),
  309. 'ic_wallet': '98831F3A-5482381C-18460FB1[256,1].mmincog',
  310. 'ic_wallet_hex': '98831F3A-1630A9F2-870376A9[256,1].mmincox',
  311. 'hic_wallet': '98831F3A-F59B07A0-559CEF19[256,1].incog.offset123',
  312. 'hic_wallet_old': '98831F3A-F59B07A0-848535F3[256,1].incog-old.offset123',
  313. 'tmpdir': os.path.join('test','tmp8'),
  314. 'kapasswd': '',
  315. 'addr_idx_list': '1010,500-501,31-33,1,33,500,1011', # 8 addresses
  316. 'dep_generators': {
  317. 'mmdat': 'refwalletgen3',
  318. pwfile: 'refwalletgen3',
  319. 'addrs': 'refaddrgen3',
  320. 'akeys.mmenc': 'refkeyaddrgen3'
  321. },
  322. },
  323. '9': {
  324. 'tmpdir': os.path.join('test','tmp9'),
  325. 'tool_enc_infn': 'tool_encrypt.in',
  326. # 'tool_enc_ref_infn': 'tool_encrypt_ref.in',
  327. 'wpasswd': 'reference password',
  328. 'dep_generators': {
  329. 'tool_encrypt.in': 'tool_encrypt',
  330. 'tool_encrypt.in.mmenc': 'tool_encrypt',
  331. # 'tool_encrypt_ref.in': 'tool_encrypt_ref',
  332. # 'tool_encrypt_ref.in.mmenc': 'tool_encrypt_ref',
  333. },
  334. },
  335. }
  336. from copy import deepcopy
  337. for a,b in ('6','11'),('7','12'),('8','13'):
  338. cfgs[b] = deepcopy(cfgs[a])
  339. cfgs[b]['tmpdir'] = os.path.join('test','tmp'+b)
  340. from collections import OrderedDict
  341. cmd_group = OrderedDict()
  342. cmd_group['help'] = OrderedDict([
  343. # test description depends
  344. ['helpscreens', (1,'help screens', [],1)],
  345. ['longhelpscreens', (1,'help screens (--longhelp)',[],1)],
  346. ])
  347. cmd_group['dfl_wallet'] = OrderedDict([
  348. ['walletgen_dfl_wallet', (15,'wallet generation (default wallet)',[[[],15]],1)],
  349. ['export_seed_dfl_wallet',(15,'seed export to mmseed format (default wallet)',[[[pwfile],15]],1)],
  350. ['addrgen_dfl_wallet',(15,'address generation (default wallet)',[[[pwfile],15]],1)],
  351. ['txcreate_dfl_wallet',(15,'transaction creation (default wallet)',[[['addrs'],15]],1)],
  352. ['txsign_dfl_wallet',(15,'transaction signing (default wallet)',[[['rawtx',pwfile],15]],1)],
  353. ['passchg_dfl_wallet',(16,'password, label and hash preset change (default wallet)',[[[pwfile],15]],1)],
  354. ['walletchk_newpass_dfl_wallet',(16,'wallet check with new pw, label and hash preset',[[[pwfile],16]],1)],
  355. ['delete_dfl_wallet',(15,'delete default wallet',[[[pwfile],15]],1)],
  356. ])
  357. cmd_group['main'] = OrderedDict([
  358. ['walletgen', (1,'wallet generation', [[['del_dw_run'],15]],1)],
  359. # ['walletchk', (1,'wallet check', [[['mmdat'],1]])],
  360. ['passchg', (5,'password, label and hash preset change',[[['mmdat',pwfile],1]],1)],
  361. ['walletchk_newpass',(5,'wallet check with new pw, label and hash preset',[[['mmdat',pwfile],5]],1)],
  362. ['addrgen', (1,'address generation', [[['mmdat',pwfile],1]],1)],
  363. ['addrimport', (1,'address import', [[['addrs'],1]],1)],
  364. ['txcreate', (1,'transaction creation', [[['addrs'],1]],1)],
  365. ['txsign', (1,'transaction signing', [[['mmdat','rawtx',pwfile],1]],1)],
  366. ['txsend', (1,'transaction sending', [[['sigtx'],1]])],
  367. # txdo must go after txsign
  368. ['txdo', (1,'online transaction', [[['sigtx','mmdat'],1]])],
  369. ['export_seed', (1,'seed export to mmseed format', [[['mmdat'],1]])],
  370. ['export_mnemonic', (1,'seed export to mmwords format', [[['mmdat'],1]])],
  371. ['export_incog', (1,'seed export to mmincog format', [[['mmdat'],1]])],
  372. ['export_incog_hex',(1,'seed export to mmincog hex format', [[['mmdat'],1]])],
  373. ['export_incog_hidden',(1,'seed export to hidden mmincog format', [[['mmdat'],1]])],
  374. ['addrgen_seed', (1,'address generation from mmseed file', [[['mmseed','addrs'],1]])],
  375. ['addrgen_mnemonic',(1,'address generation from mmwords file',[[['mmwords','addrs'],1]])],
  376. ['addrgen_incog', (1,'address generation from mmincog file',[[['mmincog','addrs'],1]])],
  377. ['addrgen_incog_hex',(1,'address generation from mmincog hex file',[[['mmincox','addrs'],1]])],
  378. ['addrgen_incog_hidden',(1,'address generation from hidden mmincog file', [[[hincog_fn,'addrs'],1]])],
  379. ['keyaddrgen', (1,'key-address file generation', [[['mmdat',pwfile],1]])],
  380. ['txsign_keyaddr',(1,'transaction signing with key-address file', [[['akeys.mmenc','rawtx'],1]])],
  381. ['walletgen2',(2,'wallet generation (2), 128-bit seed', [[['del_dw_run'],15]])],
  382. ['addrgen2', (2,'address generation (2)', [[['mmdat'],2]])],
  383. ['txcreate2', (2,'transaction creation (2)', [[['addrs'],2]])],
  384. ['txsign2', (2,'transaction signing, two transactions',[[['mmdat','rawtx'],1],[['mmdat','rawtx'],2]])],
  385. ['export_mnemonic2', (2,'seed export to mmwords format (2)',[[['mmdat'],2]])],
  386. ['walletgen3',(3,'wallet generation (3)', [[['del_dw_run'],15]])],
  387. ['addrgen3', (3,'address generation (3)', [[['mmdat'],3]])],
  388. ['txcreate3', (3,'tx creation with inputs and outputs from two wallets', [[['addrs'],1],[['addrs'],3]])],
  389. ['txsign3', (3,'tx signing with inputs and outputs from two wallets',[[['mmdat'],1],[['mmdat','rawtx'],3]])],
  390. ['walletgen14', (14,'wallet generation (14)', [[['del_dw_run'],15]],14)],
  391. ['addrgen14', (14,'address generation (14)', [[['mmdat'],14]])],
  392. ['keyaddrgen14',(14,'key-address file generation (14)', [[['mmdat'],14]],14)],
  393. ['walletgen4',(4,'wallet generation (4) (brainwallet)', [[['del_dw_run'],15]])],
  394. ['addrgen4', (4,'address generation (4)', [[['mmdat'],4]])],
  395. ['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]])],
  396. ['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]])],
  397. ['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
  398. ])
  399. cmd_group['tool'] = OrderedDict([
  400. ['tool_encrypt', (9,"'mmgen-tool encrypt' (random data)", [],1)],
  401. ['tool_decrypt', (9,"'mmgen-tool decrypt' (random data)", [[[cfgs['9']['tool_enc_infn'],cfgs['9']['tool_enc_infn']+'.mmenc'],9]],1)],
  402. # ['tool_encrypt_ref', (9,"'mmgen-tool encrypt' (reference text)", [])],
  403. ['tool_find_incog_data', (9,"'mmgen-tool find_incog_data'", [[[hincog_fn],1],[[incog_id_fn],1]])],
  404. # ['pywallet', (9,"'mmgen-pywallet'", [],1)],
  405. ])
  406. # saved reference data
  407. cmd_group['ref'] = (
  408. # reading
  409. ('ref_wallet_chk', ([],'saved reference wallet')),
  410. ('ref_seed_chk', ([],'saved seed file')),
  411. ('ref_mn_chk', ([],'saved mnemonic file')),
  412. ('ref_hincog_chk', ([],'saved hidden incog reference wallet')),
  413. ('ref_brain_chk', ([],'saved brainwallet')),
  414. # generating new reference ('abc' brainwallet) files:
  415. ('refwalletgen', ([],'gen new refwallet')),
  416. ('refaddrgen', (['mmdat',pwfile],'new refwallet addr chksum')),
  417. ('refkeyaddrgen', (['mmdat',pwfile],'new refwallet key-addr chksum'))
  418. )
  419. # misc. saved reference data
  420. cmd_group['ref_other'] = (
  421. ('ref_addrfile_chk', 'saved reference address file'),
  422. ('ref_keyaddrfile_chk','saved reference key-address file'),
  423. # Create the fake inputs:
  424. # ('txcreate8', 'transaction creation (8)'),
  425. ('ref_tx_chk', 'saved reference tx file'),
  426. ('ref_brain_chk_spc3', 'saved brainwallet (non-standard spacing)'),
  427. ('ref_tool_decrypt', 'decryption of saved MMGen-encrypted file'),
  428. )
  429. # mmgen-walletconv:
  430. cmd_group['conv_in'] = ( # reading
  431. ('ref_wallet_conv', 'conversion of saved reference wallet'),
  432. ('ref_mn_conv', 'conversion of saved mnemonic'),
  433. ('ref_seed_conv', 'conversion of saved seed file'),
  434. ('ref_brain_conv', 'conversion of ref brainwallet'),
  435. ('ref_incog_conv', 'conversion of saved incog wallet'),
  436. ('ref_incox_conv', 'conversion of saved hex incog wallet'),
  437. ('ref_hincog_conv', 'conversion of saved hidden incog wallet'),
  438. ('ref_hincog_conv_old','conversion of saved hidden incog wallet (old format)')
  439. )
  440. cmd_group['conv_out'] = ( # writing
  441. ('ref_wallet_conv_out', 'ref seed conversion to wallet'),
  442. ('ref_mn_conv_out', 'ref seed conversion to mnemonic'),
  443. ('ref_seed_conv_out', 'ref seed conversion to seed'),
  444. ('ref_incog_conv_out', 'ref seed conversion to incog data'),
  445. ('ref_incox_conv_out', 'ref seed conversion to hex incog data'),
  446. ('ref_hincog_conv_out', 'ref seed conversion to hidden incog data')
  447. )
  448. cmd_list = OrderedDict()
  449. for k in cmd_group: cmd_list[k] = []
  450. cmd_data = OrderedDict()
  451. for k,v in (
  452. ('help', ('help screens',[])),
  453. ('dfl_wallet', ('basic operations with default wallet',[15,16])),
  454. ('main', ('basic operations',[1,2,3,4,5,15,16])),
  455. ('tool', ('tools',[9]))
  456. ):
  457. cmd_data['info_'+k] = v
  458. for i in cmd_group[k]:
  459. cmd_list[k].append(i)
  460. cmd_data[i] = cmd_group[k][i]
  461. cmd_data['info_ref'] = 'reference data',[6,7,8]
  462. for a,b in cmd_group['ref']:
  463. for i,j in (1,128),(2,192),(3,256):
  464. k = a+str(i)
  465. cmd_list['ref'].append(k)
  466. cmd_data[k] = (5+i,'%s (%s-bit)' % (b[1],j),[[b[0],5+i]],1)
  467. cmd_data['info_ref_other'] = 'other reference data',[8]
  468. for a,b in cmd_group['ref_other']:
  469. cmd_list['ref_other'].append(a)
  470. cmd_data[a] = (8,b,[[[],8]],1)
  471. cmd_data['info_conv_in'] = 'wallet conversion from reference data',[11,12,13]
  472. for a,b in cmd_group['conv_in']:
  473. for i,j in (1,128),(2,192),(3,256):
  474. k = a+str(i)
  475. cmd_list['conv_in'].append(k)
  476. cmd_data[k] = (10+i,'%s (%s-bit)' % (b,j),[[[],10+i]],1)
  477. cmd_data['info_conv_out'] = 'wallet conversion to reference data',[11,12,13]
  478. for a,b in cmd_group['conv_out']:
  479. for i,j in (1,128),(2,192),(3,256):
  480. k = a+str(i)
  481. cmd_list['conv_out'].append(k)
  482. cmd_data[k] = (10+i,'%s (%s-bit)' % (b,j),[[[],10+i]],1)
  483. utils = {
  484. 'check_deps': 'check dependencies for specified command',
  485. 'clean': 'clean specified tmp dir(s) 1,2,3,4,5 or 6 (no arg = all dirs)',
  486. }
  487. addrs_per_wallet = 8
  488. # total of two outputs must be < 10 BTC
  489. for k in cfgs:
  490. cfgs[k]['amts'] = [0,0]
  491. for idx,mod in (0,6),(1,4):
  492. cfgs[k]['amts'][idx] = '%s.%s' % ((getrandnum(2) % mod), str(getrandnum(4))[:5])
  493. meta_cmds = OrderedDict([
  494. ['ref1', ('refwalletgen1','refaddrgen1','refkeyaddrgen1')],
  495. ['ref2', ('refwalletgen2','refaddrgen2','refkeyaddrgen2')],
  496. ['ref3', ('refwalletgen3','refaddrgen3','refkeyaddrgen3')],
  497. ['gen', ('walletgen','addrgen')],
  498. ['pass', ('passchg','walletchk_newpass')],
  499. ['tx', ('addrimport','txcreate','txsign','txsend')],
  500. ['export', [k for k in cmd_data if k[:7] == 'export_' and cmd_data[k][0] == 1]],
  501. ['gen_sp', [k for k in cmd_data if k[:8] == 'addrgen_' and cmd_data[k][0] == 1]],
  502. ['online', ('keyaddrgen','txsign_keyaddr')],
  503. ['2', [k for k in cmd_data if cmd_data[k][0] == 2]],
  504. ['3', [k for k in cmd_data if cmd_data[k][0] == 3]],
  505. ['4', [k for k in cmd_data if cmd_data[k][0] == 4]],
  506. ['saved_ref1', [c[0]+'1' for c in cmd_group['ref']]],
  507. ['saved_ref2', [c[0]+'2' for c in cmd_group['ref']]],
  508. ['saved_ref3', [c[0]+'3' for c in cmd_group['ref']]],
  509. ['saved_ref_other', [c[0] for c in cmd_group['ref_other']]],
  510. ['saved_ref_conv_in1', [c[0]+'1' for c in cmd_group['conv_in']]],
  511. ['saved_ref_conv_in2', [c[0]+'2' for c in cmd_group['conv_in']]],
  512. ['saved_ref_conv_in3', [c[0]+'3' for c in cmd_group['conv_in']]],
  513. ['saved_ref_conv_out1', [c[0]+'1' for c in cmd_group['conv_out']]],
  514. ['saved_ref_conv_out2', [c[0]+'2' for c in cmd_group['conv_out']]],
  515. ['saved_ref_conv_out3', [c[0]+'3' for c in cmd_group['conv_out']]],
  516. ])
  517. del cmd_group
  518. add_spawn_args = ' '.join(['{} {}'.format(
  519. '--'+k.replace('_','-'),
  520. getattr(opt,k) if getattr(opt,k) != True else ''
  521. ) for k in 'testnet','rpc_host' if getattr(opt,k)]).split()
  522. add_spawn_args += ['--data-dir',data_dir]
  523. if opt.profile: opt.names = True
  524. if opt.resume: opt.skip_deps = True
  525. if opt.log:
  526. log_fd = open(log_file,'a')
  527. log_fd.write('\nLog started: %s\n' % make_timestr())
  528. usr_rand_chars = (5,30)[bool(opt.usr_random)]
  529. usr_rand_arg = '-r%s' % usr_rand_chars
  530. if opt.system: sys.path.pop(0)
  531. ia = bool(opt.interactive)
  532. # Disable color in spawned scripts so we can parse their output
  533. os.environ['MMGEN_DISABLE_COLOR'] = '1'
  534. os.environ['MMGEN_NO_LICENSE'] = '1'
  535. os.environ['MMGEN_MIN_URANDCHARS'] = '3'
  536. os.environ['MMGEN_BOGUS_SEND'] = '1'
  537. if opt.debug_scripts: os.environ['MMGEN_DEBUG'] = '1'
  538. if opt.buf_keypress:
  539. send_delay = 0.3
  540. else:
  541. send_delay = 0
  542. os.environ['MMGEN_DISABLE_HOLD_PROTECT'] = '1'
  543. if opt.exact_output:
  544. def msg(s): pass
  545. vmsg = vmsg_r = msg_r = msg
  546. else:
  547. def msg(s): sys.stderr.write(s+'\n')
  548. def vmsg(s):
  549. if opt.verbose: sys.stderr.write(s+'\n')
  550. def msg_r(s): sys.stderr.write(s)
  551. def vmsg_r(s):
  552. if opt.verbose: sys.stderr.write(s)
  553. stderr_save = sys.stderr
  554. def silence():
  555. if not (opt.verbose or opt.exact_output):
  556. f = ('/dev/null','stderr.out')[g.platform=='win']
  557. sys.stderr = open(f,'a')
  558. def end_silence():
  559. if not (opt.verbose or opt.exact_output):
  560. sys.stderr = stderr_save
  561. def errmsg(s): stderr_save.write(s+'\n')
  562. def errmsg_r(s): stderr_save.write(s)
  563. if opt.list_cmds:
  564. fs = ' {:<{w}} - {}'
  565. Msg(green('AVAILABLE COMMANDS:'))
  566. w = max([len(i) for i in cmd_data])
  567. for cmd in cmd_data:
  568. if cmd[:5] == 'info_':
  569. m = capfirst(cmd_data[cmd][0])
  570. msg(green(' %s:' % m))
  571. continue
  572. Msg(' '+fs.format(cmd,cmd_data[cmd][1],w=w))
  573. w = max([len(i) for i in meta_cmds])
  574. Msg(green('\nAVAILABLE METACOMMANDS:'))
  575. for cmd in meta_cmds:
  576. Msg(fs.format(cmd,' '.join(meta_cmds[cmd]),w=w))
  577. w = max([len(i) for i in cmd_list])
  578. Msg(green('\nAVAILABLE COMMAND GROUPS:'))
  579. for g in cmd_list:
  580. Msg(fs.format(g,' '.join(cmd_list[g]),w=w))
  581. Msg(green('\nAVAILABLE UTILITIES:'))
  582. w = max([len(i) for i in utils])
  583. for cmd in sorted(utils):
  584. Msg(fs.format(cmd,utils[cmd],w=w))
  585. sys.exit()
  586. import time,re
  587. if g.platform == 'linux':
  588. import pexpect
  589. if opt.popen_spawn:
  590. import termios,atexit
  591. def at_exit(): os.system('stty sane')
  592. atexit.register(at_exit)
  593. from pexpect.popen_spawn import PopenSpawn
  594. use_popen_spawn,NL = True,'\n'
  595. else:
  596. use_popen_spawn,NL = False,'\r\n'
  597. else: # Windows
  598. use_popen_spawn,NL = True,'\r\n'
  599. try:
  600. import pexpect
  601. from pexpect.popen_spawn import PopenSpawn
  602. except:
  603. ia = True
  604. m1 = ('Missing pexpect module detected. Skipping some tests and running in'
  605. '\ninteractive mode. User prompts and control value checks will be ')
  606. m2 = 'HIGHLIGHTED IN GREEN'
  607. m3 = '.\nControl values should be checked against the program output.\nContinue?'
  608. if not keypress_confirm(green(m1)+grnbg(m2)+green(m3),default_yes=True):
  609. errmsg('Exiting at user request')
  610. sys.exit()
  611. def my_send(p,t,delay=send_delay,s=False):
  612. if delay: time.sleep(delay)
  613. ret = p.send(t) # returns num bytes written
  614. if delay: time.sleep(delay)
  615. if opt.verbose:
  616. ls = (' ','')[bool(opt.debug or not s)]
  617. es = (' ','')[bool(s)]
  618. msg('%sSEND %s%s' % (ls,es,yellow("'%s'"%t.replace('\n',r'\n'))))
  619. return ret
  620. def my_expect(p,s,t='',delay=send_delay,regex=False,nonl=False):
  621. quo = ('',"'")[type(s) == str]
  622. if opt.verbose: msg_r('EXPECT %s' % yellow(quo+str(s)+quo))
  623. else: msg_r('+')
  624. try:
  625. if s == '': ret = 0
  626. else:
  627. f = (p.expect_exact,p.expect)[bool(regex)]
  628. ret = f(s,timeout=(60,5)[bool(opt.debug_pexpect)])
  629. except pexpect.TIMEOUT:
  630. if opt.debug_pexpect: raise
  631. errmsg(red('\nERROR. Expect %s%s%s timed out. Exiting' % (quo,s,quo)))
  632. sys.exit(1)
  633. debug_pexpect_msg(p)
  634. if opt.debug or (opt.verbose and type(s) != str): msg_r(' ==> %s ' % ret)
  635. if ret == -1:
  636. errmsg('Error. Expect returned %s' % ret)
  637. sys.exit(1)
  638. else:
  639. if t == '':
  640. if not nonl: vmsg('')
  641. else:
  642. my_send(p,t,delay,s)
  643. return ret
  644. def get_file_with_ext(ext,mydir,delete=True,no_dot=False):
  645. dot = ('.','')[bool(no_dot)]
  646. flist = [os.path.join(mydir,f) for f in os.listdir(mydir)
  647. if f == ext or f[-len(dot+ext):] == dot+ext]
  648. if not flist: return False
  649. if len(flist) > 1:
  650. if delete:
  651. if not opt.quiet:
  652. msg("Multiple *.%s files in '%s' - deleting" % (ext,mydir))
  653. for f in flist: os.unlink(f)
  654. return False
  655. else:
  656. return flist[0]
  657. def find_generated_exts(cmd):
  658. out = []
  659. for k in cfgs:
  660. for ext,prog in cfgs[k]['dep_generators'].items():
  661. if prog == cmd:
  662. out.append((ext,cfgs[k]['tmpdir']))
  663. return out
  664. def get_addrfile_checksum(display=False):
  665. addrfile = get_file_with_ext('addrs',cfg['tmpdir'])
  666. silence()
  667. from mmgen.addr import AddrList
  668. chk = AddrList(addrfile).chksum
  669. if opt.verbose and display: msg('Checksum: %s' % cyan(chk))
  670. end_silence()
  671. return chk
  672. def verify_checksum_or_exit(checksum,chk):
  673. if checksum != chk:
  674. errmsg(red('Checksum error: %s' % chk))
  675. sys.exit(1)
  676. vmsg(green('Checksums match: %s') % (cyan(chk)))
  677. def debug_pexpect_msg(p):
  678. if opt.debug_pexpect:
  679. errmsg('\n{}{}{}'.format(red('BEFORE ['),p.before,red(']')))
  680. errmsg('{}{}{}'.format(red('MATCH ['),p.after,red(']')))
  681. class MMGenExpect(object):
  682. def __init__(self,name,mmgen_cmd,cmd_args=[],extra_desc='',no_output=False):
  683. cmd = (('./','')[bool(opt.system)]+mmgen_cmd,'python')[g.platform=='win']
  684. cmd_args = add_spawn_args + cmd_args
  685. args = (cmd_args,[mmgen_cmd]+cmd_args)[g.platform=='win']
  686. desc = (cmd_data[name][1],name)[bool(opt.names)] + (' ' + extra_desc).strip()
  687. for i in args:
  688. if type(i) not in (str,unicode):
  689. m1 = 'Error: missing input files in cmd line?:'
  690. m2 = '\nName: {}\nCmd: {}\nCmd args: {}'
  691. die(2,(m1+m2).format(name,cmd,args))
  692. if use_popen_spawn:
  693. args = [("'"+a+"'" if ' ' in a else a) for a in args]
  694. cmd_str = '{} {}'.format(cmd,' '.join(args))
  695. if use_popen_spawn:
  696. cmd_str = cmd_str.replace('\\','/')
  697. if opt.log:
  698. log_fd.write(cmd_str+'\n')
  699. if opt.verbose or opt.print_cmdline or opt.exact_output:
  700. clr1,clr2,eol = ((green,cyan,'\n'),(nocolor,nocolor,' '))[bool(opt.print_cmdline)]
  701. sys.stderr.write(green('Testing: {}\n'.format(desc)))
  702. sys.stderr.write(clr1('Executing {}{}'.format(clr2(cmd_str),eol)))
  703. else:
  704. m = 'Testing %s: ' % desc
  705. msg_r((m,yellow(m))[ia])
  706. if mmgen_cmd == '': return
  707. if opt.direct_exec or ia:
  708. msg('')
  709. from subprocess import call,check_output
  710. f = (call,check_output)[bool(no_output)]
  711. ret = f([cmd] + args)
  712. if f == call and ret != 0:
  713. m = 'ERROR: process returned a non-zero exit status (%s)'
  714. die(1,red(m % ret))
  715. else:
  716. if opt.traceback:
  717. cmd,args = tb_cmd,[cmd]+args
  718. if use_popen_spawn:
  719. self.p = PopenSpawn(cmd_str)
  720. else:
  721. self.p = pexpect.spawn(cmd,args)
  722. if opt.exact_output: self.p.logfile = sys.stdout
  723. def license(self):
  724. if 'MMGEN_NO_LICENSE' in os.environ: return
  725. p = "'w' for conditions and warranty info, or 'c' to continue: "
  726. my_expect(self.p,p,'c')
  727. def label(self,label='Test Label'):
  728. p = 'Enter a wallet label, or hit ENTER for no label: '
  729. my_expect(self.p,p,label+'\n')
  730. def usr_rand_out(self,saved=False):
  731. m = '%suser-supplied entropy' % (('','saved ')[saved])
  732. my_expect(self.p,'Generating encryption key from OS random data plus ' + m)
  733. def usr_rand(self,num_chars):
  734. if opt.usr_random:
  735. self.interactive()
  736. my_send(self.p,'\n')
  737. else:
  738. rand_chars = list(getrandstr(num_chars,no_space=True))
  739. my_expect(self.p,'symbols left: ','x')
  740. try:
  741. vmsg_r('SEND ')
  742. while self.p.expect('left: ',0.1) == 0:
  743. ch = rand_chars.pop(0)
  744. msg_r(yellow(ch)+' ' if opt.verbose else '+')
  745. self.p.send(ch)
  746. except:
  747. vmsg('EOT')
  748. my_expect(self.p,'ENTER to continue: ','\n')
  749. def passphrase_new(self,desc,passphrase):
  750. my_expect(self.p,('Enter passphrase for %s: ' % desc), passphrase+'\n')
  751. my_expect(self.p,'Repeat passphrase: ', passphrase+'\n')
  752. def passphrase(self,desc,passphrase,pwtype=''):
  753. if pwtype: pwtype += ' '
  754. my_expect(self.p,('Enter %spassphrase for %s.*?: ' % (pwtype,desc)),
  755. passphrase+'\n',regex=True)
  756. def hash_preset(self,desc,preset=''):
  757. my_expect(self.p,('Enter hash preset for %s' % desc))
  758. my_expect(self.p,('or hit ENTER .*?:'), str(preset)+'\n',regex=True)
  759. def written_to_file(self,desc,overwrite_unlikely=False,query='Overwrite? ',oo=False):
  760. s1 = '%s written to file ' % desc
  761. s2 = query + "Type uppercase 'YES' to confirm: "
  762. ret = my_expect(self.p,([s1,s2],s1)[overwrite_unlikely])
  763. if ret == 1:
  764. my_send(self.p,'YES\n')
  765. # if oo:
  766. outfile = self.expect_getend("Overwriting file '").rstrip("'")
  767. return outfile
  768. # else:
  769. # ret = my_expect(self.p,s1)
  770. self.expect(NL,nonl=True)
  771. outfile = self.p.before.strip().strip("'")
  772. if opt.debug_pexpect: msgred('Outfile [%s]' % outfile)
  773. vmsg('%s file: %s' % (desc,cyan(outfile.replace("'",''))))
  774. return outfile
  775. def no_overwrite(self):
  776. self.expect("Overwrite? Type uppercase 'YES' to confirm: ",'\n')
  777. self.expect('Exiting at user request')
  778. def tx_view(self):
  779. my_expect(self.p,r'View .*?transaction.*? \(y\)es, \(N\)o, pager \(v\)iew.*?: ','\n',regex=True)
  780. def expect_getend(self,s,regex=False):
  781. ret = self.expect(s,regex=regex,nonl=True)
  782. debug_pexpect_msg(self.p)
  783. # end = self.readline().strip()
  784. # readline() of partial lines doesn't work with PopenSpawn, so do this instead:
  785. self.expect(NL,nonl=True)
  786. debug_pexpect_msg(self.p)
  787. end = self.p.before
  788. vmsg(' ==> %s' % cyan(end))
  789. return end
  790. def interactive(self):
  791. return self.p.interact()
  792. def logfile(self,arg):
  793. self.p.logfile = arg
  794. def expect(self,*args,**kwargs):
  795. return my_expect(self.p,*args,**kwargs)
  796. def send(self,*args,**kwargs):
  797. return my_send(self.p,*args,**kwargs)
  798. # def readline(self):
  799. # return self.p.readline()
  800. # def readlines(self):
  801. # return [l.rstrip()+'\n' for l in self.p.readlines()]
  802. def read(self,n=None):
  803. return self.p.read(n)
  804. def close(self):
  805. if not use_popen_spawn:
  806. self.p.close()
  807. from mmgen.obj import BTCAmt
  808. from mmgen.bitcoin import verify_addr
  809. def create_fake_unspent_entry(address,sid=None,idx=None,lbl=None,non_mmgen=None):
  810. if lbl: lbl = ' ' + lbl
  811. return {
  812. 'account': (non_mmgen or ('%s:%s%s' % (sid,idx,lbl))).decode('utf8'),
  813. 'vout': int(getrandnum(4) % 8),
  814. 'txid': hexlify(os.urandom(32)).decode('utf8'),
  815. 'amount': BTCAmt('%s.%s' % (10+(getrandnum(4) % 40), getrandnum(4) % 100000000)),
  816. 'address': address,
  817. 'spendable': False,
  818. 'scriptPubKey': ('76a914'+verify_addr(address,return_hex=True)+'88ac'),
  819. 'confirmations': getrandnum(4) % 50000
  820. }
  821. labels = [
  822. "Automotive",
  823. "Travel expenses",
  824. "Healthcare",
  825. "Freelancing 1",
  826. "Freelancing 2",
  827. "Alice's allowance",
  828. "Bob's bequest",
  829. "House purchase",
  830. "Real estate fund",
  831. "Job 1",
  832. "XYZ Corp.",
  833. "Eddie's endowment",
  834. "Emergency fund",
  835. "Real estate fund",
  836. "Ian's inheritance",
  837. "",
  838. "Rainy day",
  839. "Fred's funds",
  840. "Job 2",
  841. "Carl's capital",
  842. ]
  843. label_iter = None
  844. def create_fake_unspent_data(adata,unspent_data_file,tx_data,non_mmgen_input=''):
  845. out = []
  846. for s in tx_data:
  847. sid = tx_data[s]['sid']
  848. a = adata.addrlist(sid)
  849. for n,(idx,btcaddr) in enumerate(a.addrpairs(),1):
  850. while True:
  851. try: lbl = next(label_iter)
  852. except: label_iter = iter(labels)
  853. else: break
  854. out.append(create_fake_unspent_entry(btcaddr,sid,idx,lbl))
  855. if n == 1: # create a duplicate address. This means addrs_per_wallet += 1
  856. out.append(create_fake_unspent_entry(btcaddr,sid,idx,lbl))
  857. if non_mmgen_input:
  858. from mmgen.bitcoin import privnum2addr,hex2wif
  859. privnum = getrandnum(32)
  860. btcaddr = privnum2addr(privnum,compressed=True)
  861. of = os.path.join(cfgs[non_mmgen_input]['tmpdir'],non_mmgen_fn)
  862. write_data_to_file(of, hex2wif('{:064x}'.format(privnum),
  863. compressed=True)+'\n','compressed bitcoin key',silent=True)
  864. out.append(create_fake_unspent_entry(btcaddr,non_mmgen='Non-MMGen address'))
  865. # msg('\n'.join([repr(o) for o in out])); sys.exit()
  866. write_data_to_file(unspent_data_file,repr(out),'Unspent outputs',silent=True)
  867. def add_comments_to_addr_file(addrfile,outfile):
  868. silence()
  869. msg(green("Adding comments to address file '%s'" % addrfile))
  870. from mmgen.addr import AddrList
  871. a = AddrList(addrfile)
  872. for n,idx in enumerate(a.idxs(),1):
  873. if n % 2: a.set_comment(idx,'Test address %s' % n)
  874. a.format(enable_comments=True)
  875. write_data_to_file(outfile,a.fmt_data,silent=True)
  876. end_silence()
  877. def make_brainwallet_file(fn):
  878. # Print random words with random whitespace in between
  879. from mmgen.mn_tirosh import words
  880. wl = words.split()
  881. nwords,ws_list,max_spaces = 10,' \n',5
  882. def rand_ws_seq():
  883. nchars = getrandnum(1) % max_spaces + 1
  884. return ''.join([ws_list[getrandnum(1)%len(ws_list)] for i in range(nchars)])
  885. rand_pairs = [wl[getrandnum(4) % len(wl)] + rand_ws_seq() for i in range(nwords)]
  886. d = ''.join(rand_pairs).rstrip() + '\n'
  887. if opt.verbose: msg_r('Brainwallet password:\n%s' % cyan(d))
  888. write_data_to_file(fn,d,'brainwallet password',silent=True)
  889. def do_between():
  890. if opt.pause:
  891. if keypress_confirm(green('Continue?'),default_yes=True):
  892. if opt.verbose or opt.exact_output: sys.stderr.write('\n')
  893. else:
  894. errmsg('Exiting at user request')
  895. sys.exit()
  896. elif opt.verbose or opt.exact_output:
  897. sys.stderr.write('\n')
  898. rebuild_list = OrderedDict()
  899. def check_needs_rerun(
  900. ts,
  901. cmd,
  902. build=False,
  903. root=True,
  904. force_delete=False,
  905. dpy=False
  906. ):
  907. rerun = (False,True)[root] # force_delete is not passed to recursive call
  908. fns = []
  909. if force_delete or not root:
  910. # does cmd produce a needed dependency(ies)?
  911. ret = ts.get_num_exts_for_cmd(cmd,dpy)
  912. if ret:
  913. for ext in ret[1]:
  914. fn = get_file_with_ext(ext,cfgs[ret[0]]['tmpdir'],delete=build)
  915. if fn:
  916. if force_delete: os.unlink(fn)
  917. else: fns.append(fn)
  918. else: rerun = True
  919. fdeps = ts.generate_file_deps(cmd)
  920. cdeps = ts.generate_cmd_deps(fdeps)
  921. # print 'cmd,fdeps,cdeps,fns: ',cmd,fdeps,cdeps,fns # DEBUG
  922. for fn in fns:
  923. my_age = os.stat(fn).st_mtime
  924. for num,ext in fdeps:
  925. f = get_file_with_ext(ext,cfgs[num]['tmpdir'],delete=build)
  926. if f and os.stat(f).st_mtime > my_age:
  927. rerun = True
  928. for cdep in cdeps:
  929. if check_needs_rerun(ts,cdep,build=build,root=False,dpy=cmd):
  930. rerun = True
  931. if build:
  932. if rerun:
  933. for fn in fns:
  934. if not root: os.unlink(fn)
  935. if not (dpy and opt.skip_deps):
  936. ts.do_cmd(cmd)
  937. if not root: do_between()
  938. else:
  939. # If prog produces multiple files:
  940. if cmd not in rebuild_list or rerun == True:
  941. rebuild_list[cmd] = (rerun,fns[0] if fns else '') # FIX
  942. return rerun
  943. def refcheck(desc,chk,refchk):
  944. vmsg("Comparing %s '%s' to stored reference" % (desc,chk))
  945. if chk == refchk:
  946. ok()
  947. else:
  948. if not opt.verbose: errmsg('')
  949. errmsg(red("""
  950. Fatal error - %s '%s' does not match reference value '%s'. Aborting test
  951. """.strip() % (desc,chk,refchk)))
  952. sys.exit(3)
  953. def check_deps(cmds):
  954. if len(cmds) != 1:
  955. die(1,'Usage: %s check_deps <command>' % g.prog_name)
  956. cmd = cmds[0]
  957. if cmd not in cmd_data:
  958. die(1,"'%s': unrecognized command" % cmd)
  959. if not opt.quiet:
  960. msg("Checking dependencies for '%s'" % (cmd))
  961. check_needs_rerun(ts,cmd,build=False)
  962. w = max(len(i) for i in rebuild_list) + 1
  963. for cmd in rebuild_list:
  964. c = rebuild_list[cmd]
  965. m = 'Rebuild' if (c[0] and c[1]) else 'Build' if c[0] else 'OK'
  966. msg('cmd {:<{w}} {}'.format(cmd+':', m, w=w))
  967. # mmsg(cmd,c)
  968. def clean(usr_dirs=[]):
  969. if opt.skip_deps and not ia: return
  970. all_dirs = MMGenTestSuite().list_tmp_dirs()
  971. dirs = (usr_dirs or all_dirs)
  972. for d in sorted(dirs):
  973. if str(d) in all_dirs:
  974. cleandir(all_dirs[str(d)])
  975. else:
  976. die(1,'%s: invalid directory number' % d)
  977. class MMGenTestSuite(object):
  978. def __init__(self):
  979. pass
  980. def list_tmp_dirs(self):
  981. d = {}
  982. for k in cfgs: d[k] = cfgs[k]['tmpdir']
  983. return d
  984. def get_num_exts_for_cmd(self,cmd,dpy=False): # dpy ignored here
  985. num = str(cmd_data[cmd][0])
  986. dgl = cfgs[num]['dep_generators']
  987. # mmsg(num,cmd,dgl)
  988. if cmd in dgl.values():
  989. exts = [k for k in dgl if dgl[k] == cmd]
  990. return (num,exts)
  991. else:
  992. return None
  993. def do_cmd(self,cmd):
  994. if ia and (len(cmd_data[cmd]) < 4 or cmd_data[cmd][3] != 1): return
  995. # delete files produced by this cmd
  996. # for ext,tmpdir in find_generated_exts(cmd):
  997. # print cmd, get_file_with_ext(ext,tmpdir)
  998. d = [(str(num),ext) for exts,num in cmd_data[cmd][2] for ext in exts]
  999. # delete files depended on by this cmd
  1000. al = [get_file_with_ext(ext,cfgs[num]['tmpdir']) for num,ext in d]
  1001. global cfg
  1002. cfg = cfgs[str(cmd_data[cmd][0])]
  1003. if opt.resume:
  1004. if cmd == opt.resume:
  1005. msg(yellow("Resuming at '%s'" % cmd))
  1006. opt.resume = False
  1007. opt.skip_deps = False
  1008. else:
  1009. return
  1010. if opt.profile: start = time.time()
  1011. self.__class__.__dict__[cmd](*([self,cmd] + al))
  1012. if opt.profile:
  1013. msg('\r\033[50C{:.4f}'.format(time.time() - start))
  1014. def generate_file_deps(self,cmd):
  1015. return [(str(n),e) for exts,n in cmd_data[cmd][2] for e in exts]
  1016. def generate_cmd_deps(self,fdeps):
  1017. return [cfgs[str(n)]['dep_generators'][ext] for n,ext in fdeps]
  1018. def helpscreens(self,name,arg='--help'):
  1019. for s in scripts:
  1020. t = MMGenExpect(name,('mmgen-'+s),[arg],
  1021. extra_desc='(mmgen-%s)'%s,no_output=True)
  1022. if not ia:
  1023. t.read(); ok()
  1024. def longhelpscreens(self,name): self.helpscreens(name,arg='--longhelp')
  1025. def walletgen(self,name,del_dw_run='dummy',seed_len=None,gen_dfl_wallet=False):
  1026. if ia:
  1027. m = "\nAnswer '{}' at the the interactive prompt".format(('n','y')[gen_dfl_wallet])
  1028. msg(grnbg(m))
  1029. write_to_tmpfile(cfg,pwfile,cfg['wpasswd']+'\n')
  1030. add_args = ([usr_rand_arg],
  1031. ['-q','-r0','-L','Interactive Mode Wallet','-P',get_tmpfile_fn(cfg,pwfile)])[bool(ia)]
  1032. args = ['-d',cfg['tmpdir'],'-p1']
  1033. if seed_len: args += ['-l',str(seed_len)]
  1034. t = MMGenExpect(name,'mmgen-walletgen', args + add_args)
  1035. if ia: return
  1036. t.license()
  1037. t.usr_rand(usr_rand_chars)
  1038. t.passphrase_new('new MMGen wallet',cfg['wpasswd'])
  1039. t.label()
  1040. global have_dfl_wallet
  1041. if not have_dfl_wallet:
  1042. t.expect('move it to the data directory? (Y/n): ',('n','y')[gen_dfl_wallet])
  1043. if gen_dfl_wallet: have_dfl_wallet = True
  1044. t.written_to_file('MMGen wallet')
  1045. ok()
  1046. def walletgen_dfl_wallet(self,name,seed_len=None):
  1047. self.walletgen(name,seed_len=seed_len,gen_dfl_wallet=True)
  1048. def brainwalletgen_ref(self,name):
  1049. sl_arg = '-l%s' % cfg['seed_len']
  1050. hp_arg = '-p%s' % ref_wallet_hash_preset
  1051. label = "test.py ref. wallet (pw '%s', seed len %s)" \
  1052. % (ref_wallet_brainpass,cfg['seed_len'])
  1053. bf = 'ref.mmbrain'
  1054. args = ['-d',cfg['tmpdir'],hp_arg,sl_arg,'-ib','-L',label]
  1055. write_to_tmpfile(cfg,bf,ref_wallet_brainpass)
  1056. write_to_tmpfile(cfg,pwfile,cfg['wpasswd'])
  1057. if ia:
  1058. add_args = ['-r0', '-q', '-P%s' % get_tmpfile_fn(cfg,pwfile),
  1059. get_tmpfile_fn(cfg,bf)]
  1060. else:
  1061. add_args = [usr_rand_arg]
  1062. t = MMGenExpect(name,'mmgen-walletconv', args + add_args)
  1063. if ia: return
  1064. t.license()
  1065. t.expect('Enter brainwallet: ', ref_wallet_brainpass+'\n')
  1066. t.passphrase_new('new MMGen wallet',cfg['wpasswd'])
  1067. t.usr_rand(usr_rand_chars)
  1068. sid = os.path.basename(t.written_to_file('MMGen wallet').split('-')[0])
  1069. refcheck('Seed ID',sid,cfg['seed_id'])
  1070. def refwalletgen(self,name): self.brainwalletgen_ref(name)
  1071. def passchg(self,name,wf,pf):
  1072. # ia: reuse password, since there's no way to change it non-interactively
  1073. silence()
  1074. write_to_tmpfile(cfg,pwfile,get_data_from_file(pf))
  1075. end_silence()
  1076. add_args = ([usr_rand_arg],['-q','-r0','-P',pf])[bool(ia)]
  1077. t = MMGenExpect(name,'mmgen-passchg', add_args +
  1078. ['-d',cfg['tmpdir'],'-p','2','-L','Changed label'] + ([],[wf])[bool(wf)])
  1079. if ia: return
  1080. t.license()
  1081. t.passphrase('MMGen wallet',cfgs['1']['wpasswd'],pwtype='old')
  1082. t.expect_getend('Hash preset changed to ')
  1083. t.passphrase('MMGen wallet',cfg['wpasswd'],pwtype='new') # reuse passphrase?
  1084. t.expect('Repeat passphrase: ',cfg['wpasswd']+'\n')
  1085. t.usr_rand(usr_rand_chars)
  1086. # t.expect('Enter a wallet label.*: ','Changed Label\n',regex=True)
  1087. t.expect_getend('Label changed to ')
  1088. # t.expect_getend('Key ID changed: ')
  1089. if not wf:
  1090. t.expect("Type uppercase 'YES' to confirm: ",'YES\n')
  1091. t.written_to_file('New wallet')
  1092. t.expect('Securely deleting old wallet')
  1093. # t.expect('Okay to WIPE 1 regular file ? (Yes/No)','Yes\n')
  1094. t.expect('Wallet passphrase has changed')
  1095. t.expect_getend('has been changed to ')
  1096. else:
  1097. t.written_to_file('MMGen wallet')
  1098. ok()
  1099. def passchg_dfl_wallet(self,name,pf):
  1100. if ia:
  1101. m = "\nAnswer 'YES'<ENTER> at the the interactive prompt"
  1102. msg(grnbg(m))
  1103. return self.passchg(name=name,wf=None,pf=pf)
  1104. def walletchk(self,name,wf,pf,desc='MMGen wallet',
  1105. add_args=[],sid=None,pw=False,extra_desc=''):
  1106. args = ([],['-P',pf,'-q'])[bool(ia and pf)]
  1107. hp = cfg['hash_preset'] if 'hash_preset' in cfg else '1'
  1108. wf_arg = ([],[wf])[bool(wf)]
  1109. t = MMGenExpect(name,'mmgen-walletchk',
  1110. add_args+args+['-p',hp]+wf_arg,
  1111. extra_desc=extra_desc)
  1112. if ia:
  1113. if sid:
  1114. n = (' should be','')[desc=='MMGen wallet']
  1115. m = grnbg('Seed ID%s:' % n)
  1116. msg(grnbg('%s %s' % (m,cyan(sid))))
  1117. return
  1118. if desc != 'hidden incognito data':
  1119. t.expect("Getting %s from file '" % (desc))
  1120. if pw:
  1121. t.passphrase(desc,cfg['wpasswd'])
  1122. t.expect(
  1123. ['Passphrase is OK', 'Passphrase.* are correct'],
  1124. regex=True
  1125. )
  1126. chk = t.expect_getend('Valid %s for Seed ID ' % desc)[:8]
  1127. if sid: cmp_or_die(chk,sid)
  1128. else: ok()
  1129. def walletchk_newpass(self,name,wf,pf):
  1130. return self.walletchk(name,wf,pf,pw=True)
  1131. def walletchk_newpass_dfl_wallet(self,name,pf):
  1132. return self.walletchk_newpass(name,wf=None,pf=pf)
  1133. def delete_dfl_wallet(self,name,pf):
  1134. with open(os.path.join(cfg['tmpdir'],'del_dw_run'),'w') as f: pass
  1135. if opt.no_dw_delete: return True
  1136. for wf in [f for f in os.listdir(g.data_dir) if f[-6:]=='.mmdat']:
  1137. os.unlink(os.path.join(g.data_dir,wf))
  1138. MMGenExpect(name,'')
  1139. global have_dfl_wallet
  1140. have_dfl_wallet = False
  1141. if not ia: ok()
  1142. def addrgen(self,name,wf,pf=None,check_ref=False):
  1143. add_args = ([],['-q'] + ([],['-P',pf])[bool(pf)])[ia]
  1144. t = MMGenExpect(name,'mmgen-addrgen', add_args +
  1145. ['-d',cfg['tmpdir']] + ([],[wf])[bool(wf)] + [cfg['addr_idx_list']])
  1146. if ia: return
  1147. t.license()
  1148. t.passphrase('MMGen wallet',cfg['wpasswd'])
  1149. t.expect('Passphrase is OK')
  1150. chk = t.expect_getend(r'Checksum for address data .*?: ',regex=True)
  1151. if check_ref:
  1152. refcheck('address data checksum',chk,cfg['addrfile_chk'])
  1153. return
  1154. t.written_to_file('Addresses',oo=True)
  1155. ok()
  1156. def addrgen_dfl_wallet(self,name,pf=None,check_ref=False):
  1157. return self.addrgen(name,wf=None,pf=pf,check_ref=check_ref)
  1158. def refaddrgen(self,name,wf,pf):
  1159. d = ' (%s-bit seed)' % cfg['seed_len']
  1160. self.addrgen(name,wf,pf=pf,check_ref=True)
  1161. def addrimport(self,name,addrfile):
  1162. add_args = ([],['-q','-t'])[ia]
  1163. outfile = os.path.join(cfg['tmpdir'],'addrfile_w_comments')
  1164. add_comments_to_addr_file(addrfile,outfile)
  1165. t = MMGenExpect(name,'mmgen-addrimport', add_args + [outfile])
  1166. if ia: return
  1167. t.expect_getend(r'Checksum for address data .*\[.*\]: ',regex=True)
  1168. t.expect_getend('Validating addresses...OK. ')
  1169. t.expect("Type uppercase 'YES' to confirm: ",'\n')
  1170. vmsg('This is a simulation, so no addresses were actually imported into the tracking\nwallet')
  1171. ok()
  1172. def txcreate_common(self,name,sources=['1'],non_mmgen_input='',do_label=False,txdo_args=[],add_args=[]):
  1173. if opt.verbose or opt.exact_output:
  1174. sys.stderr.write(green('Generating fake tracking wallet info\n'))
  1175. silence()
  1176. from mmgen.addr import AddrList,AddrData,AddrIdxList
  1177. tx_data,ad = {},AddrData()
  1178. for s in sources:
  1179. afile = get_file_with_ext('addrs',cfgs[s]['tmpdir'])
  1180. ai = AddrList(afile)
  1181. ad.add(ai)
  1182. aix = AddrIdxList(fmt_str=cfgs[s]['addr_idx_list'])
  1183. if len(aix) != addrs_per_wallet:
  1184. errmsg(red('Address index list length != %s: %s' %
  1185. (addrs_per_wallet,repr(aix))))
  1186. sys.exit()
  1187. tx_data[s] = {
  1188. 'addrfile': afile,
  1189. 'chk': ai.chksum,
  1190. 'sid': ai.seed_id,
  1191. 'addr_idxs': aix[-2:],
  1192. }
  1193. unspent_data_file = os.path.join(cfg['tmpdir'],'unspent.json')
  1194. create_fake_unspent_data(ad,unspent_data_file,tx_data,non_mmgen_input)
  1195. if opt.verbose or opt.exact_output:
  1196. sys.stderr.write("Fake transaction wallet data written to file '%s'\n" % unspent_data_file)
  1197. # make the command line
  1198. from mmgen.bitcoin import privnum2addr
  1199. btcaddr = privnum2addr(getrandnum(32),compressed=True)
  1200. cmd_args = ['-d',cfg['tmpdir']]
  1201. for num in tx_data:
  1202. s = tx_data[num]
  1203. cmd_args += [
  1204. '%s:%s,%s' % (s['sid'],s['addr_idxs'][0],cfgs[num]['amts'][0]),
  1205. ]
  1206. # + one BTC address
  1207. # + one change address and one BTC address
  1208. if num is tx_data.keys()[-1]:
  1209. cmd_args += ['%s:%s' % (s['sid'],s['addr_idxs'][1])]
  1210. cmd_args += ['%s,%s' % (btcaddr,cfgs[num]['amts'][1])]
  1211. for num in tx_data: cmd_args += [tx_data[num]['addrfile']]
  1212. os.environ['MMGEN_BOGUS_WALLET_DATA'] = unspent_data_file
  1213. end_silence()
  1214. if opt.verbose or opt.exact_output: sys.stderr.write('\n')
  1215. if ia:
  1216. add_args += ['-q']
  1217. m = '\nAnswer the interactive prompts as follows:\n' + \
  1218. " 'y', 'y', 'q', '1-9'<ENTER>, ENTER, ENTER, ENTER, ENTER, 'y'"
  1219. msg(grnbg(m))
  1220. bwd_msg = 'MMGEN_BOGUS_WALLET_DATA=%s' % unspent_data_file
  1221. if opt.print_cmdline: msg(bwd_msg)
  1222. if opt.log: log_fd.write(bwd_msg + ' ')
  1223. t = MMGenExpect(name,'mmgen-'+('txcreate','txdo')[bool(txdo_args)],['-f','0.0001'] + add_args + cmd_args + txdo_args)
  1224. if ia: return
  1225. t.license()
  1226. if txdo_args and add_args: # txdo4
  1227. t.hash_preset('key-address data','1')
  1228. t.passphrase('key-address data',cfgs['14']['kapasswd'])
  1229. t.expect('Check key-to-address validity? (y/N): ','y')
  1230. for num in tx_data:
  1231. t.expect_getend('Getting address data from file ')
  1232. chk=t.expect_getend(r'Checksum for address data .*?: ',regex=True)
  1233. verify_checksum_or_exit(tx_data[num]['chk'],chk)
  1234. # not in tracking wallet warning, (1 + num sources) times
  1235. if t.expect(['Continue anyway? (y/N): ',
  1236. 'Unable to connect to bitcoind']) == 0:
  1237. t.send('y')
  1238. else:
  1239. errmsg(red('Error: unable to connect to bitcoind. Exiting'))
  1240. sys.exit(1)
  1241. for num in tx_data:
  1242. t.expect('Continue anyway? (y/N): ','y')
  1243. t.expect(r"'q'=quit view, .*?:.",'M', regex=True)
  1244. t.expect(r"'q'=quit view, .*?:.",'q', regex=True)
  1245. outputs_list = [(addrs_per_wallet+1)*i + 1 for i in range(len(tx_data))]
  1246. if non_mmgen_input: outputs_list.append(len(tx_data)*(addrs_per_wallet+1) + 1)
  1247. t.expect('Enter a range or space-separated list of outputs to spend: ',
  1248. ' '.join([str(i) for i in outputs_list])+'\n')
  1249. if non_mmgen_input and not txdo_args: t.expect('Accept? (y/N): ','y')
  1250. t.expect('OK? (Y/n): ','y') # fee OK?
  1251. t.expect('OK? (Y/n): ','y') # change OK?
  1252. if do_label:
  1253. t.expect('Add a comment to transaction? (y/N): ','y')
  1254. t.expect('Comment: ',ref_tx_label.encode('utf8')+'\n')
  1255. else:
  1256. t.expect('Add a comment to transaction? (y/N): ','\n')
  1257. t.tx_view()
  1258. if txdo_args: return t
  1259. t.expect('Save transaction? (y/N): ','y')
  1260. t.written_to_file('Transaction')
  1261. ok()
  1262. def txcreate(self,name,addrfile):
  1263. self.txcreate_common(name,sources=['1'])
  1264. def txdo(self,name,addrfile,wallet):
  1265. t = self.txcreate_common(name,sources=['1'],txdo_args=[wallet])
  1266. self.txsign(name,'','',pf='',save=True,has_label=False,txdo_handle=t)
  1267. self.txsend(name,'',txdo_handle=t)
  1268. def txcreate_dfl_wallet(self,name,addrfile):
  1269. self.txcreate_common(name,sources=['15'])
  1270. def txsign_end(self,t,tnum=None,has_label=False):
  1271. t.expect('Signing transaction')
  1272. cprompt = ('Add a comment to transaction','Edit transaction comment')[has_label]
  1273. t.expect('%s? (y/N): ' % cprompt,'\n')
  1274. t.expect('Save signed transaction.*?\? \(Y/n\): ','y',regex=True)
  1275. add = ' #' + tnum if tnum else ''
  1276. t.written_to_file('Signed transaction' + add, oo=True)
  1277. def txsign(self,name,txfile,wf,pf='',save=True,has_label=False,txdo_handle=None):
  1278. add_args = ([],['-q','-P',pf])[ia]
  1279. if ia:
  1280. m = '\nAnswer the interactive prompts as follows:\n ENTER, ENTER, ENTER'
  1281. msg(grnbg(m))
  1282. if txdo_handle:
  1283. t = txdo_handle
  1284. if ia: return
  1285. else:
  1286. t = MMGenExpect(name,'mmgen-txsign', add_args+['-d',cfg['tmpdir'],txfile]+([],[wf])[bool(wf)])
  1287. if ia: return
  1288. t.license()
  1289. t.tx_view()
  1290. t.passphrase('MMGen wallet',cfg['wpasswd'])
  1291. if txdo_handle: return
  1292. if save:
  1293. self.txsign_end(t,has_label=has_label)
  1294. else:
  1295. cprompt = ('Add a comment to transaction','Edit transaction comment')[has_label]
  1296. t.expect('%s? (y/N): ' % cprompt,'\n')
  1297. t.close()
  1298. ok()
  1299. def txsign_dfl_wallet(self,name,txfile,pf='',save=True,has_label=False):
  1300. return self.txsign(name,txfile,wf=None,pf=pf,save=save,has_label=has_label)
  1301. def txsend(self,name,sigfile,txdo_handle=None):
  1302. if txdo_handle:
  1303. t = txdo_handle
  1304. else:
  1305. t = MMGenExpect(name,'mmgen-txsend', ['-d',cfg['tmpdir'],sigfile])
  1306. t.license()
  1307. t.tx_view()
  1308. t.expect('Add a comment to transaction? (y/N): ','\n')
  1309. t.expect('broadcast this transaction to the network?')
  1310. m = 'YES, I REALLY WANT TO DO THIS'
  1311. t.expect("'%s' to confirm: " % m,m+'\n')
  1312. t.expect('BOGUS transaction NOT sent')
  1313. t.written_to_file('Transaction ID')
  1314. ok()
  1315. def walletconv_export(self,name,wf,desc,uargs=[],out_fmt='w',pf=None,out_pw=False):
  1316. opts = ['-d',cfg['tmpdir'],'-o',out_fmt] + uargs + \
  1317. ([],[wf])[bool(wf)] + ([],['-P',pf])[bool(pf)]
  1318. t = MMGenExpect(name,'mmgen-walletconv',opts)
  1319. if ia: return
  1320. t.license()
  1321. if not pf:
  1322. t.passphrase('MMGen wallet',cfg['wpasswd'])
  1323. if out_pw:
  1324. t.passphrase_new('new '+desc,cfg['wpasswd'])
  1325. t.usr_rand(usr_rand_chars)
  1326. if ' '.join(desc.split()[-2:]) == 'incognito data':
  1327. t.expect('Generating encryption key from OS random data ')
  1328. t.expect('Generating encryption key from OS random data ')
  1329. ic_id = t.expect_getend('New Incog Wallet ID: ')
  1330. t.expect('Generating encryption key from OS random data ')
  1331. if desc == 'hidden incognito data':
  1332. write_to_tmpfile(cfg,incog_id_fn,ic_id)
  1333. ret = t.expect(['Create? (Y/n): ',"'YES' to confirm: "])
  1334. if ret == 0:
  1335. t.send('\n')
  1336. t.expect('Enter file size: ',str(hincog_bytes)+'\n')
  1337. else:
  1338. t.send('YES\n')
  1339. if out_fmt == 'w': t.label()
  1340. return t.written_to_file(capfirst(desc),oo=True)
  1341. def export_seed(self,name,wf,desc='seed data',out_fmt='seed',pf=None):
  1342. f = self.walletconv_export(name,wf,desc=desc,out_fmt=out_fmt,pf=pf)
  1343. if ia: return
  1344. silence()
  1345. msg('%s: %s' % (capfirst(desc),cyan(get_data_from_file(f,desc))))
  1346. end_silence()
  1347. ok()
  1348. def export_seed_dfl_wallet(self,name,pf,desc='seed data',out_fmt='seed'):
  1349. self.export_seed(name,wf=None,desc=desc,out_fmt=out_fmt,pf=pf)
  1350. def export_mnemonic(self,name,wf):
  1351. self.export_seed(name,wf,desc='mnemonic data',out_fmt='words')
  1352. def export_incog(self,name,wf,desc='incognito data',out_fmt='i',add_args=[]):
  1353. uargs = ['-p1',usr_rand_arg] + add_args
  1354. self.walletconv_export(name,wf,desc=desc,out_fmt=out_fmt,uargs=uargs,out_pw=True)
  1355. ok()
  1356. def export_incog_hex(self,name,wf):
  1357. self.export_incog(name,wf,desc='hex incognito data',out_fmt='xi')
  1358. # TODO: make outdir and hidden incog compatible (ignore --outdir and warn user?)
  1359. def export_incog_hidden(self,name,wf):
  1360. rf = os.path.join(cfg['tmpdir'],hincog_fn)
  1361. add_args = ['-J','%s,%s'%(rf,hincog_offset)]
  1362. self.export_incog(
  1363. name,wf,desc='hidden incognito data',out_fmt='hi',add_args=add_args)
  1364. def addrgen_seed(self,name,wf,foo,desc='seed data',in_fmt='seed'):
  1365. stdout = (False,True)[desc=='seed data'] #capture output to screen once
  1366. add_arg = ([],['-S'])[bool(stdout)]
  1367. t = MMGenExpect(name,'mmgen-addrgen', add_arg +
  1368. ['-i'+in_fmt,'-d',cfg['tmpdir'],wf,cfg['addr_idx_list']])
  1369. t.license()
  1370. t.expect_getend('Valid %s for Seed ID ' % desc)
  1371. vmsg('Comparing generated checksum with checksum from previous address file')
  1372. chk = t.expect_getend(r'Checksum for address data .*?: ',regex=True)
  1373. if stdout: t.read()
  1374. verify_checksum_or_exit(get_addrfile_checksum(),chk)
  1375. # t.no_overwrite()
  1376. ok()
  1377. def addrgen_mnemonic(self,name,wf,foo):
  1378. self.addrgen_seed(name,wf,foo,desc='mnemonic data',in_fmt='words')
  1379. def addrgen_incog(self,name,wf=[],foo='',in_fmt='i',desc='incognito data',args=[]):
  1380. t = MMGenExpect(name,'mmgen-addrgen', args+['-i'+in_fmt,'-d',cfg['tmpdir']]+
  1381. ([],[wf])[bool(wf)] + [cfg['addr_idx_list']])
  1382. t.license()
  1383. t.expect_getend('Incog Wallet ID: ')
  1384. t.hash_preset(desc,'1')
  1385. t.passphrase('%s \w{8}' % desc, cfg['wpasswd'])
  1386. vmsg('Comparing generated checksum with checksum from address file')
  1387. chk = t.expect_getend(r'Checksum for address data .*?: ',regex=True)
  1388. t.close()
  1389. verify_checksum_or_exit(get_addrfile_checksum(),chk)
  1390. # t.no_overwrite()
  1391. ok()
  1392. def addrgen_incog_hex(self,name,wf,foo):
  1393. self.addrgen_incog(name,wf,'',in_fmt='xi',desc='hex incognito data')
  1394. def addrgen_incog_hidden(self,name,wf,foo):
  1395. rf = os.path.join(cfg['tmpdir'],hincog_fn)
  1396. self.addrgen_incog(name,[],'',in_fmt='hi',desc='hidden incognito data',
  1397. args=['-H','%s,%s'%(rf,hincog_offset),'-l',str(hincog_seedlen)])
  1398. def keyaddrgen(self,name,wf,pf=None,check_ref=False):
  1399. args = ['-d',cfg['tmpdir'],usr_rand_arg,wf,cfg['addr_idx_list']]
  1400. if ia:
  1401. m = "\nAnswer 'n' at the interactive prompt"
  1402. msg(grnbg(m))
  1403. args = ['-q'] + ([],['-P',pf])[bool(pf)] + args
  1404. t = MMGenExpect(name,'mmgen-keygen', args)
  1405. if ia: return
  1406. t.license()
  1407. t.passphrase('MMGen wallet',cfg['wpasswd'])
  1408. chk = t.expect_getend(r'Checksum for key-address data .*?: ',regex=True)
  1409. if check_ref:
  1410. refcheck('key-address data checksum',chk,cfg['keyaddrfile_chk'])
  1411. return
  1412. t.expect('Encrypt key list? (y/N): ','y')
  1413. t.usr_rand(usr_rand_chars)
  1414. t.hash_preset('new key list','1')
  1415. # t.passphrase_new('new key list','kafile password')
  1416. t.passphrase_new('new key list',cfg['kapasswd'])
  1417. t.written_to_file('Secret keys',oo=True)
  1418. ok()
  1419. def refkeyaddrgen(self,name,wf,pf):
  1420. self.keyaddrgen(name,wf,pf,check_ref=True)
  1421. def txsign_keyaddr(self,name,keyaddr_file,txfile):
  1422. t = MMGenExpect(name,'mmgen-txsign', ['-d',cfg['tmpdir'],'-M',keyaddr_file,txfile])
  1423. t.license()
  1424. t.hash_preset('key-address data','1')
  1425. t.passphrase('key-address data',cfg['kapasswd'])
  1426. t.expect('Check key-to-address validity? (y/N): ','y')
  1427. t.tx_view()
  1428. self.txsign_end(t)
  1429. ok()
  1430. def walletgen2(self,name,del_dw_run='dummy'):
  1431. self.walletgen(name,seed_len=128)
  1432. def addrgen2(self,name,wf):
  1433. self.addrgen(name,wf,pf='')
  1434. def txcreate2(self,name,addrfile):
  1435. self.txcreate_common(name,sources=['2'])
  1436. def txsign2(self,name,txf1,wf1,txf2,wf2):
  1437. t = MMGenExpect(name,'mmgen-txsign', ['-d',cfg['tmpdir'],txf1,wf1,txf2,wf2])
  1438. t.license()
  1439. for cnum in ('1','2'):
  1440. t.tx_view()
  1441. t.passphrase('MMGen wallet',cfgs[cnum]['wpasswd'])
  1442. self.txsign_end(t,cnum)
  1443. ok()
  1444. def export_mnemonic2(self,name,wf):
  1445. self.export_mnemonic(name,wf)
  1446. def walletgen3(self,name,del_dw_run='dummy'):
  1447. self.walletgen(name)
  1448. def addrgen3(self,name,wf):
  1449. self.addrgen(name,wf,pf='')
  1450. def txcreate3(self,name,addrfile1,addrfile2):
  1451. self.txcreate_common(name,sources=['1','3'])
  1452. def txsign3(self,name,wf1,wf2,txf2):
  1453. t = MMGenExpect(name,'mmgen-txsign', ['-d',cfg['tmpdir'],wf1,wf2,txf2])
  1454. t.license()
  1455. t.tx_view()
  1456. for cnum in ('1','3'):
  1457. # t.expect_getend('Getting MMGen wallet data from file ')
  1458. t.passphrase('MMGen wallet',cfgs[cnum]['wpasswd'])
  1459. self.txsign_end(t)
  1460. ok()
  1461. def walletgen4(self,name,del_dw_run='dummy'):
  1462. bwf = os.path.join(cfg['tmpdir'],cfg['bw_filename'])
  1463. make_brainwallet_file(bwf)
  1464. seed_len = str(cfg['seed_len'])
  1465. args = ['-d',cfg['tmpdir'],'-p1',usr_rand_arg,'-l'+seed_len,'-ib']
  1466. t = MMGenExpect(name,'mmgen-walletconv', args + [bwf])
  1467. t.license()
  1468. t.passphrase_new('new MMGen wallet',cfg['wpasswd'])
  1469. t.usr_rand(usr_rand_chars)
  1470. t.label()
  1471. t.written_to_file('MMGen wallet')
  1472. ok()
  1473. def addrgen4(self,name,wf):
  1474. self.addrgen(name,wf,pf='')
  1475. def txcreate4(self,name,f1,f2,f3,f4,f5,f6):
  1476. self.txcreate_common(name,sources=['1','2','3','4','14'],non_mmgen_input='4',do_label=1)
  1477. def txdo4(self,name,f1,f2,f3,f4,f5,f6,f7,f8,f9,f10,f11,f12):
  1478. non_mm_fn = os.path.join(cfg['tmpdir'],non_mmgen_fn)
  1479. add_args = ['-d',cfg['tmpdir'],'-i','brain','-b'+cfg['bw_params'],'-p1','-k',non_mm_fn,'-M',f12]
  1480. 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)
  1481. self.txsign4(name,f7,f8,f9,f10,f11,f12,txdo_handle=t)
  1482. self.txsend(name,'',txdo_handle=t)
  1483. def txsign4(self,name,f1,f2,f3,f4,f5,f6,txdo_handle=None):
  1484. if txdo_handle:
  1485. t = txdo_handle
  1486. else:
  1487. non_mm_fn = os.path.join(cfg['tmpdir'],non_mmgen_fn)
  1488. a = ['-d',cfg['tmpdir'],'-i','brain','-b'+cfg['bw_params'],'-p1','-k',non_mm_fn,'-M',f6,f1,f2,f3,f4,f5]
  1489. t = MMGenExpect(name,'mmgen-txsign',a)
  1490. t.license()
  1491. t.hash_preset('key-address data','1')
  1492. t.passphrase('key-address data',cfgs['14']['kapasswd'])
  1493. t.expect('Check key-to-address validity? (y/N): ','y')
  1494. t.tx_view()
  1495. for cnum,desc in ('1','incognito data'),('3','MMGen wallet'):
  1496. t.passphrase(('%s' % desc),cfgs[cnum]['wpasswd'])
  1497. if txdo_handle: return
  1498. self.txsign_end(t,has_label=True)
  1499. ok()
  1500. def tool_encrypt(self,name,infile=''):
  1501. if infile:
  1502. infn = infile
  1503. else:
  1504. d = os.urandom(1033)
  1505. tmp_fn = cfg['tool_enc_infn']
  1506. write_to_tmpfile(cfg,tmp_fn,d,binary=True)
  1507. infn = get_tmpfile_fn(cfg,tmp_fn)
  1508. if ia:
  1509. pwfn = 'ni_pw'
  1510. write_to_tmpfile(cfg,pwfn,tool_enc_passwd+'\n')
  1511. pre = ['-P', get_tmpfile_fn(cfg,pwfn)]
  1512. app = ['hash_preset=1']
  1513. else:
  1514. pre,app = [],[]
  1515. t = MMGenExpect(name,'mmgen-tool',pre+['-d',cfg['tmpdir'],usr_rand_arg,'encrypt',infn]+app)
  1516. if ia: return
  1517. t.usr_rand(usr_rand_chars)
  1518. t.hash_preset('user data','1')
  1519. t.passphrase_new('user data',tool_enc_passwd)
  1520. t.written_to_file('Encrypted data')
  1521. ok()
  1522. # Generate the reference mmenc file
  1523. # def tool_encrypt_ref(self,name):
  1524. # infn = get_tmpfile_fn(cfg,cfg['tool_enc_ref_infn'])
  1525. # write_data_to_file(infn,cfg['tool_enc_reftext'],silent=True)
  1526. # self.tool_encrypt(name,infn)
  1527. def tool_decrypt(self,name,f1,f2):
  1528. of = name + '.out'
  1529. if ia:
  1530. pwfn = 'ni_pw'
  1531. pre = ['-P', get_tmpfile_fn(cfg,pwfn)]
  1532. else:
  1533. pre = []
  1534. t = MMGenExpect(name,'mmgen-tool',
  1535. pre+['-d',cfg['tmpdir'],'decrypt',f2,'outfile='+of,'hash_preset=1'])
  1536. if not ia:
  1537. t.passphrase('user data',tool_enc_passwd)
  1538. t.written_to_file('Decrypted data')
  1539. d1 = read_from_file(f1,binary=True)
  1540. d2 = read_from_file(get_tmpfile_fn(cfg,of),binary=True)
  1541. cmp_or_die(d1,d2,skip_ok=ia)
  1542. def tool_find_incog_data(self,name,f1,f2):
  1543. i_id = read_from_file(f2).rstrip()
  1544. vmsg('Incog ID: %s' % cyan(i_id))
  1545. t = MMGenExpect(name,'mmgen-tool',
  1546. ['-d',cfg['tmpdir'],'find_incog_data',f1,i_id])
  1547. if ia: return
  1548. o = t.expect_getend('Incog data for ID %s found at offset ' % i_id)
  1549. os.unlink(f1)
  1550. cmp_or_die(hincog_offset,int(o))
  1551. # Saved reference file tests
  1552. def ref_wallet_conv(self,name):
  1553. wf = os.path.join(ref_dir,cfg['ref_wallet'])
  1554. self.walletconv_in(name,wf,'MMGen wallet',pw=True,oo=True)
  1555. def ref_mn_conv(self,name,ext='mmwords',desc='Mnemonic data'):
  1556. wf = os.path.join(ref_dir,cfg['seed_id']+'.'+ext)
  1557. self.walletconv_in(name,wf,desc,oo=True)
  1558. def ref_seed_conv(self,name):
  1559. self.ref_mn_conv(name,ext='mmseed',desc='Seed data')
  1560. def ref_brain_conv(self,name):
  1561. uopts = ['-i','b','-p','1','-l',str(cfg['seed_len'])]
  1562. self.walletconv_in(name,None,'brainwallet',uopts,oo=True)
  1563. def ref_incog_conv(self,name,wfk='ic_wallet',in_fmt='i',desc='incognito data'):
  1564. uopts = ['-i',in_fmt,'-p','1','-l',str(cfg['seed_len'])]
  1565. wf = os.path.join(ref_dir,cfg[wfk])
  1566. self.walletconv_in(name,wf,desc,uopts,oo=True,pw=True)
  1567. def ref_incox_conv(self,name):
  1568. self.ref_incog_conv(name,in_fmt='xi',wfk='ic_wallet_hex',desc='hex incognito data')
  1569. def ref_hincog_conv(self,name,wfk='hic_wallet',add_uopts=[]):
  1570. ic_f = os.path.join(ref_dir,cfg[wfk])
  1571. uopts = ['-i','hi','-p','1','-l',str(cfg['seed_len'])] + add_uopts
  1572. hi_opt = ['-H','%s,%s' % (ic_f,ref_wallet_incog_offset)]
  1573. self.walletconv_in(name,None,'hidden incognito data',uopts+hi_opt,oo=True,pw=True)
  1574. def ref_hincog_conv_old(self,name):
  1575. self.ref_hincog_conv(name,wfk='hic_wallet_old',add_uopts=['-O'])
  1576. def ref_wallet_conv_out(self,name):
  1577. self.walletconv_out(name,'MMGen wallet','w',pw=True)
  1578. def ref_mn_conv_out(self,name):
  1579. self.walletconv_out(name,'mnemonic data','mn')
  1580. def ref_seed_conv_out(self,name):
  1581. self.walletconv_out(name,'seed data','seed')
  1582. def ref_incog_conv_out(self,name):
  1583. self.walletconv_out(name,'incognito data',out_fmt='i',pw=True)
  1584. def ref_incox_conv_out(self,name):
  1585. self.walletconv_out(name,'hex incognito data',out_fmt='xi',pw=True)
  1586. def ref_hincog_conv_out(self,name,extra_uopts=[]):
  1587. ic_f = os.path.join(cfg['tmpdir'],hincog_fn)
  1588. hi_parms = '%s,%s' % (ic_f,ref_wallet_incog_offset)
  1589. sl_parm = '-l' + str(cfg['seed_len'])
  1590. self.walletconv_out(name,
  1591. 'hidden incognito data', 'hi',
  1592. uopts=['-J',hi_parms,sl_parm] + extra_uopts,
  1593. uopts_chk=['-H',hi_parms,sl_parm],
  1594. pw=True
  1595. )
  1596. def ref_wallet_chk(self,name):
  1597. wf = os.path.join(ref_dir,cfg['ref_wallet'])
  1598. if ia:
  1599. write_to_tmpfile(cfg,pwfile,cfg['wpasswd'])
  1600. pf = get_tmpfile_fn(cfg,pwfile)
  1601. else:
  1602. pf = None
  1603. self.walletchk(name,wf,pf=pf,pw=True,sid=cfg['seed_id'])
  1604. from mmgen.seed import SeedFile
  1605. def ref_seed_chk(self,name,ext=SeedFile.ext):
  1606. wf = os.path.join(ref_dir,'%s.%s' % (cfg['seed_id'],ext))
  1607. from mmgen.seed import SeedFile
  1608. desc = ('mnemonic data','seed data')[ext==SeedFile.ext]
  1609. self.walletchk(name,wf,pf=None,desc=desc,sid=cfg['seed_id'])
  1610. def ref_mn_chk(self,name):
  1611. from mmgen.seed import Mnemonic
  1612. self.ref_seed_chk(name,ext=Mnemonic.ext)
  1613. def ref_brain_chk(self,name,bw_file=ref_bw_file):
  1614. wf = os.path.join(ref_dir,bw_file)
  1615. add_args = ['-l%s' % cfg['seed_len'], '-p'+ref_bw_hash_preset]
  1616. self.walletchk(name,wf,pf=None,add_args=add_args,
  1617. desc='brainwallet',sid=cfg['ref_bw_seed_id'])
  1618. def ref_brain_chk_spc3(self,name):
  1619. self.ref_brain_chk(name,bw_file=ref_bw_file_spc)
  1620. def ref_hincog_chk(self,name,desc='hidden incognito data'):
  1621. for wtype,edesc,of_arg in ('hic_wallet','',[]), \
  1622. ('hic_wallet_old','(old format)',['-O']):
  1623. ic_arg = ['-H%s,%s' % (
  1624. os.path.join(ref_dir,cfg[wtype]),
  1625. ref_wallet_incog_offset
  1626. )]
  1627. slarg = ['-l%s ' % cfg['seed_len']]
  1628. hparg = ['-p1']
  1629. if ia:
  1630. write_to_tmpfile(cfg,pwfile,cfg['wpasswd'])
  1631. add_args = ['-q','-P%s' % get_tmpfile_fn(cfg,pwfile)]
  1632. else:
  1633. add_args = []
  1634. if ia and wtype == 'hic_wallet_old':
  1635. m = grnbg("Answer 'y' at the interactive prompt if Seed ID is")
  1636. n = cyan(cfg['seed_id'])
  1637. msg('\n%s %s' % (m,n))
  1638. if wtype == 'hic_wallet_old' and opt.profile: msg('')
  1639. t = MMGenExpect(name,'mmgen-walletchk',
  1640. add_args + slarg + hparg + of_arg + ic_arg,
  1641. extra_desc=edesc)
  1642. if ia: continue
  1643. t.passphrase(desc,cfg['wpasswd'])
  1644. if wtype == 'hic_wallet_old':
  1645. t.expect('Is the Seed ID correct? (Y/n): ','\n')
  1646. chk = t.expect_getend('Seed ID: ')
  1647. t.close()
  1648. cmp_or_die(cfg['seed_id'],chk)
  1649. def ref_addrfile_chk(self,name,ftype='addr'):
  1650. wf = os.path.join(ref_dir,cfg['ref_'+ftype+'file'])
  1651. if ia:
  1652. m = "\nAnswer the interactive prompts as follows: '1'<ENTER>, ENTER"
  1653. msg(grnbg(m))
  1654. pfn = 'ref_kafile_passwd'
  1655. write_to_tmpfile(cfg,pfn,ref_kafile_pass)
  1656. aa = ['-P',get_tmpfile_fn(cfg,pfn)]
  1657. else:
  1658. aa = []
  1659. t = MMGenExpect(name,'mmgen-tool',aa+[ftype+'file_chksum',wf])
  1660. if ia:
  1661. k = 'ref_%saddrfile_chksum' % ('','key')[ftype == 'keyaddr']
  1662. m = grnbg('Checksum should be:')
  1663. n = cyan(cfg[k])
  1664. msg(grnbg('%s %s' % (m,n)))
  1665. return
  1666. if ftype == 'keyaddr':
  1667. w = 'key-address data'
  1668. t.hash_preset(w,ref_kafile_hash_preset)
  1669. t.passphrase(w,ref_kafile_pass)
  1670. t.expect('Check key-to-address validity? (y/N): ','y')
  1671. o = t.read().strip().split('\n')[-1]
  1672. cmp_or_die(cfg['ref_'+ftype+'file_chksum'],o)
  1673. def ref_keyaddrfile_chk(self,name):
  1674. self.ref_addrfile_chk(name,ftype='keyaddr')
  1675. # def txcreate8(self,name,addrfile):
  1676. # self.txcreate_common(name,sources=['8'])
  1677. def ref_tx_chk(self,name):
  1678. tf = os.path.join(ref_dir,cfg['ref_tx_file'])
  1679. wf = os.path.join(ref_dir,cfg['ref_wallet'])
  1680. write_to_tmpfile(cfg,pwfile,cfg['wpasswd'])
  1681. pf = get_tmpfile_fn(cfg,pwfile)
  1682. self.txsign(name,tf,wf,pf,save=False,has_label=True)
  1683. def ref_tool_decrypt(self,name):
  1684. f = os.path.join(ref_dir,ref_enc_fn)
  1685. aa = []
  1686. if ia:
  1687. pfn = 'tool_enc_passwd'
  1688. write_to_tmpfile(cfg,pfn,tool_enc_passwd)
  1689. aa = ['-P',get_tmpfile_fn(cfg,pfn)]
  1690. t = MMGenExpect(name,'mmgen-tool',
  1691. aa + ['-q','decrypt',f,'outfile=-','hash_preset=1'])
  1692. if ia: return
  1693. t.passphrase('user data',tool_enc_passwd)
  1694. t.expect(NL,nonl=True)
  1695. import re
  1696. o = re.sub('\r\n','\n',t.read())
  1697. cmp_or_die(sample_text,o)
  1698. # wallet conversion tests
  1699. def walletconv_in(self,name,infile,desc,uopts=[],pw=False,oo=False):
  1700. opts = ['-d',cfg['tmpdir'],'-o','words',usr_rand_arg]
  1701. if_arg = [infile] if infile else []
  1702. d = '(convert)'
  1703. if ia:
  1704. opts += ['-q']
  1705. msg('')
  1706. if pw:
  1707. pfn = 'ni_passwd'
  1708. write_to_tmpfile(cfg,pfn,cfg['wpasswd'])
  1709. opts += ['-P',get_tmpfile_fn(cfg,pfn)]
  1710. if desc == 'brainwallet':
  1711. m = "\nAnswer the interactive prompt as follows: '%s'<ENTER>"
  1712. msg(grnbg(m % ref_wallet_brainpass))
  1713. if '-O' in uopts:
  1714. m = grnbg("Answer 'y' at the interactive prompt if Seed ID is")
  1715. n = cyan(cfg['seed_id'])
  1716. msg('\n%s %s' % (m,n))
  1717. t = MMGenExpect(name,'mmgen-walletconv',opts+uopts+if_arg,extra_desc=d)
  1718. if ia:
  1719. m = grnbg('Seed ID should be:')
  1720. n = cyan(cfg['seed_id'])
  1721. msg(grnbg('%s %s' % (m,n)))
  1722. return
  1723. t.license()
  1724. if desc == 'brainwallet':
  1725. t.expect('Enter brainwallet: ',ref_wallet_brainpass+'\n')
  1726. if pw:
  1727. t.passphrase(desc,cfg['wpasswd'])
  1728. if name[:19] == 'ref_hincog_conv_old':
  1729. t.expect('Is the Seed ID correct? (Y/n): ','\n')
  1730. else:
  1731. t.expect(['Passphrase is OK',' are correct'])
  1732. # Output
  1733. wf = t.written_to_file('Mnemonic data',oo=oo)
  1734. t.close()
  1735. ok()
  1736. # back check of result
  1737. if opt.profile: msg('')
  1738. self.walletchk(name,wf,pf=None,
  1739. desc='mnemonic data',
  1740. sid=cfg['seed_id'],
  1741. extra_desc='(check)'
  1742. )
  1743. def walletconv_out(self,name,desc,out_fmt='w',uopts=[],uopts_chk=[],pw=False):
  1744. opts = ['-d',cfg['tmpdir'],'-p1','-o',out_fmt] + uopts
  1745. if ia:
  1746. pfn = 'ni_passwd'
  1747. write_to_tmpfile(cfg,pfn,cfg['wpasswd'])
  1748. l = 'Non-Interactive Test Wallet'
  1749. aa = ['-q','-L',l,'-r0','-P',get_tmpfile_fn(cfg,pfn)]
  1750. if desc == 'hidden incognito data':
  1751. rd = os.urandom(ref_wallet_incog_offset+128)
  1752. write_to_tmpfile(cfg,hincog_fn,rd)
  1753. else:
  1754. aa = [usr_rand_arg]
  1755. infile = os.path.join(ref_dir,cfg['seed_id']+'.mmwords')
  1756. t = MMGenExpect(name,'mmgen-walletconv',aa+opts+[infile],extra_desc='(convert)')
  1757. add_args = ['-l%s' % cfg['seed_len']]
  1758. if ia:
  1759. pfn = 'ni_passwd'
  1760. write_to_tmpfile(cfg,pfn,cfg['wpasswd'])
  1761. pf = get_tmpfile_fn(cfg,pfn)
  1762. if desc != 'hidden incognito data':
  1763. from mmgen.seed import SeedSource
  1764. ext = SeedSource.fmt_code_to_type(out_fmt).ext
  1765. hps = ('',',1')[bool(pw)] # TODO real hp
  1766. pre_ext = '[%s%s].' % (cfg['seed_len'],hps)
  1767. wf = get_file_with_ext(pre_ext+ext,cfg['tmpdir'],no_dot=True)
  1768. else:
  1769. t.license()
  1770. if pw:
  1771. t.passphrase_new('new '+desc,cfg['wpasswd'])
  1772. t.usr_rand(usr_rand_chars)
  1773. if ' '.join(desc.split()[-2:]) == 'incognito data':
  1774. for i in (1,2,3):
  1775. t.expect('Generating encryption key from OS random data ')
  1776. if desc == 'hidden incognito data':
  1777. ret = t.expect(['Create? (Y/n): ',"'YES' to confirm: "])
  1778. if ret == 0:
  1779. t.send('\n')
  1780. t.expect('Enter file size: ',str(hincog_bytes)+'\n')
  1781. else:
  1782. t.send('YES\n')
  1783. if out_fmt == 'w': t.label()
  1784. wf = t.written_to_file(capfirst(desc),oo=True)
  1785. pf = None
  1786. ok()
  1787. if desc == 'hidden incognito data':
  1788. add_args += uopts_chk
  1789. wf = None
  1790. if opt.profile: msg('')
  1791. self.walletchk(name,wf,pf=pf,
  1792. desc=desc,sid=cfg['seed_id'],pw=pw,
  1793. add_args=add_args,
  1794. extra_desc='(check)')
  1795. # END methods
  1796. for k in (
  1797. 'ref_wallet_conv',
  1798. 'ref_mn_conv',
  1799. 'ref_seed_conv',
  1800. 'ref_brain_conv',
  1801. 'ref_incog_conv',
  1802. 'ref_incox_conv',
  1803. 'ref_hincog_conv',
  1804. 'ref_hincog_conv_old',
  1805. 'ref_wallet_conv_out',
  1806. 'ref_mn_conv_out',
  1807. 'ref_seed_conv_out',
  1808. 'ref_incog_conv_out',
  1809. 'ref_incox_conv_out',
  1810. 'ref_hincog_conv_out',
  1811. 'ref_wallet_chk',
  1812. 'refwalletgen',
  1813. 'refaddrgen',
  1814. 'ref_seed_chk',
  1815. 'ref_mn_chk',
  1816. 'ref_brain_chk',
  1817. 'ref_hincog_chk',
  1818. 'refkeyaddrgen',
  1819. ):
  1820. for i in ('1','2','3'):
  1821. locals()[k+i] = locals()[k]
  1822. for k in ('walletgen','addrgen','keyaddrgen'): locals()[k+'14'] = locals()[k]
  1823. # create temporary dirs
  1824. if not opt.resume and not opt.skip_deps:
  1825. if g.platform == 'win':
  1826. for cfg in sorted(cfgs):
  1827. mk_tmpdir(cfgs[cfg]['tmpdir'])
  1828. else:
  1829. for cfg in sorted(cfgs):
  1830. src = os.path.join(shm_dir,cfgs[cfg]['tmpdir'].split('/')[-1])
  1831. mk_tmpdir(src)
  1832. try:
  1833. os.unlink(cfgs[cfg]['tmpdir'])
  1834. except OSError as e:
  1835. if e.errno != 2: raise
  1836. finally:
  1837. os.symlink(src,cfgs[cfg]['tmpdir'])
  1838. have_dfl_wallet = False
  1839. # main()
  1840. if opt.pause:
  1841. import termios,atexit
  1842. fd = sys.stdin.fileno()
  1843. old = termios.tcgetattr(fd)
  1844. def at_exit():
  1845. termios.tcsetattr(fd, termios.TCSADRAIN, old)
  1846. atexit.register(at_exit)
  1847. start_time = int(time.time())
  1848. def end_msg():
  1849. t = int(time.time()) - start_time
  1850. m1 = 'All requested tests finished OK, elapsed time: {:02d}:{:02d}\n'
  1851. m2 = ('','Please re-check all {} control values against the program output.\n'.format(grnbg('HIGHLIGHTED')))[ia]
  1852. sys.stderr.write(green(m1.format(t/60,t%60)))
  1853. sys.stderr.write(m2)
  1854. ts = MMGenTestSuite()
  1855. try:
  1856. if cmd_args:
  1857. for arg in cmd_args:
  1858. if arg in utils:
  1859. globals()[arg](cmd_args[cmd_args.index(arg)+1:])
  1860. sys.exit()
  1861. elif 'info_'+arg in cmd_data:
  1862. dirs = cmd_data['info_'+arg][1]
  1863. if dirs: clean(dirs)
  1864. for cmd in cmd_list[arg]:
  1865. check_needs_rerun(ts,cmd,build=True)
  1866. elif arg in meta_cmds:
  1867. for cmd in meta_cmds[arg]:
  1868. check_needs_rerun(ts,cmd,build=True)
  1869. elif arg in cmd_data:
  1870. check_needs_rerun(ts,arg,build=True)
  1871. else:
  1872. die(1,'%s: unrecognized command' % arg)
  1873. else:
  1874. clean()
  1875. for cmd in cmd_data:
  1876. if cmd[:5] == 'info_':
  1877. msg(green('%sTesting %s' % (('\n','')[bool(opt.resume)],cmd_data[cmd][0])))
  1878. continue
  1879. ts.do_cmd(cmd)
  1880. if cmd is not cmd_data.keys()[-1]: do_between()
  1881. except KeyboardInterrupt:
  1882. die(1,'\nExiting at user request')
  1883. raise
  1884. except:
  1885. sys.stderr = stderr_save
  1886. raise
  1887. end_msg()