ts_xmrwallet.py 20 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663
  1. #!/usr/bin/env python3
  2. #
  3. # mmgen = Multi-Mode GENerator, command-line Bitcoin cold storage solution
  4. # Copyright (C)2013-2021 The MMGen Project <mmgen@tuta.io>
  5. #
  6. # This program is free software: you can redistribute it and/or modify
  7. # it under the terms of the GNU General Public License as published by
  8. # the Free Software Foundation, either version 3 of the License, or
  9. # (at your option) any later version.
  10. #
  11. # This program is distributed in the hope that it will be useful,
  12. # but WITHOUT ANY WARRANTY; without even the implied warranty of
  13. # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  14. # GNU General Public License for more details.
  15. #
  16. # You should have received a copy of the GNU General Public License
  17. # along with this program. If not, see <http://www.gnu.org/licenses/>.
  18. """
  19. ts_xmrwallet.py: xmrwallet tests for the test.py test suite
  20. """
  21. import sys,os,atexit,asyncio
  22. from subprocess import run,PIPE
  23. from mmgen.globalvars import g
  24. from mmgen.opts import opt
  25. from mmgen.obj import MMGenRange,XMRAmt
  26. from mmgen.addr import KeyAddrList,AddrIdxList
  27. from ..include.common import *
  28. from .common import *
  29. from .ts_base import *
  30. class TestSuiteXMRWallet(TestSuiteBase):
  31. """
  32. Monero wallet operations
  33. """
  34. networks = ('xmr',)
  35. passthru_opts = ()
  36. tmpdir_nums = [29]
  37. dfl_random_txs = 3
  38. color = True
  39. cmd_group = (
  40. ('gen_kafiles', 'generating key-address files'),
  41. ('create_wallets_miner', 'creating Monero wallets (Miner)'),
  42. ('mine_initial_coins', 'mining initial coins'),
  43. ('create_wallets_alice', 'creating Monero wallets (Alice)'),
  44. ('fund_alice', 'sending funds'),
  45. ('sync_wallets_all', 'syncing all wallets'),
  46. ('sync_wallets_selected', 'syncing selected wallets'),
  47. ('sweep_to_address_proxy', 'sweeping to new address (via TX relay + proxy)'),
  48. ('sweep_to_account', 'sweeping to new account'),
  49. ('sweep_to_address_noproxy', 'sweeping to new address (via TX relay, no proxy)'),
  50. ('transfer_to_miner_proxy', 'transferring funds to Miner (via TX relay + proxy)'),
  51. ('transfer_to_miner_noproxy', 'transferring funds to Miner (via TX relay, no proxy)'),
  52. ('transfer_to_miner_create1', 'transferring funds to Miner (create TX)'),
  53. ('transfer_to_miner_send1', 'transferring funds to Miner (send TX via proxy)'),
  54. ('transfer_to_miner_create2', 'transferring funds to Miner (create TX)'),
  55. ('transfer_to_miner_send2', 'transferring funds to Miner (send TX, no proxy)'),
  56. ('sweep_create_and_send', 'sweeping to new account (create TX + send TX, in stages)'),
  57. )
  58. def __init__(self,trunner,cfgs,spawn):
  59. TestSuiteBase.__init__(self,trunner,cfgs,spawn)
  60. if trunner == None:
  61. return
  62. from mmgen.protocol import init_proto
  63. self.proto = init_proto('XMR',network='testnet')
  64. self.datadir_base = os.path.join('test','daemons','xmrtest')
  65. self.long_opts = ['--testnet=1', '--monero-wallet-rpc-password=passw0rd']
  66. self.init_users()
  67. self.init_daemon_args()
  68. for v in self.users.values():
  69. run(['mkdir','-p',v.udir])
  70. self.init_proxy()
  71. self.tx_relay_daemon_parm = 'localhost:{}'.format( self.users['bob'].md.rpc_port )
  72. self.tx_relay_daemon_proxy_parm = (
  73. self.tx_relay_daemon_parm + f':127.0.0.1:{self.socks_port}' # proxy must be IP, not 'localhost'
  74. if self.use_proxy else None )
  75. if not opt.no_daemon_autostart:
  76. self.start_daemons()
  77. self.start_wallet_daemons()
  78. if not opt.no_daemon_stop:
  79. atexit.register(self.stop_daemons)
  80. atexit.register(self.stop_wallet_daemons)
  81. self.balance = None
  82. # init methods
  83. def init_proxy(self):
  84. def port_in_use(port):
  85. import socket
  86. try: socket.create_connection(('localhost',port)).close()
  87. except: return False
  88. else: return True
  89. def start_proxy():
  90. if not opt.no_daemon_autostart:
  91. run(a+b2)
  92. omsg(f'SSH SOCKS server started, listening at localhost:{self.socks_port}')
  93. def kill_proxy():
  94. omsg(f'Killing SSH SOCKS server at localhost:{self.socks_port}')
  95. cmd = [ 'pkill', '-f', ' '.join(a + b2) ]
  96. run(cmd)
  97. self.use_proxy = False
  98. self.socks_port = 9060
  99. a = ['ssh','-x','-o','ExitOnForwardFailure=True','-D',f'localhost:{self.socks_port}']
  100. b0 = ['-o','PasswordAuthentication=False']
  101. b1 = ['localhost','true']
  102. b2 = ['-fN','-E','txrelay-proxy.debug','localhost']
  103. if port_in_use(self.socks_port):
  104. omsg(f'Port {self.socks_port} already in use. Assuming SSH SOCKS server is running')
  105. self.use_proxy = True
  106. else:
  107. cp = run(a+b0+b1,stdout=PIPE,stderr=PIPE)
  108. err = cp.stderr.decode()
  109. if err:
  110. omsg(err)
  111. if cp.returncode == 0:
  112. start_proxy()
  113. self.use_proxy = True
  114. elif 'onnection refused' in err:
  115. die(2,fmt("""
  116. The SSH daemon must be running and listening on localhost in order to test
  117. XMR TX relaying via SOCKS proxy. If sshd is not running, please start it.
  118. Otherwise, add the line 'ListenAddress 127.0.0.1' to your sshd_config, and
  119. then restart the daemon.
  120. """,indent=' '))
  121. elif 'ermission denied' in err:
  122. msg(fmt(f"""
  123. In order to test XMR TX relaying via SOCKS proxy, it’s desirable to enable
  124. SSH to localhost without a password, which is not currently supported by
  125. your configuration. Your possible courses of action:
  126. 1. Continue by answering 'y' at this prompt, and enter your system password
  127. at the following prompt;
  128. 2. Exit the test here, add your user SSH public key to your user
  129. 'authorized_keys' file, and restart the test; or
  130. 3. Exit the test here, start the SSH SOCKS proxy manually by entering the
  131. following command, and restart the test:
  132. {' '.join(a+b2)}
  133. """,indent=' ',strip_char='\t'))
  134. if keypress_confirm('Continue?'):
  135. start_proxy()
  136. self.use_proxy = True
  137. else:
  138. die(1,'Exiting at user request')
  139. else:
  140. die(2,fmt(f"""
  141. Please start the SSH SOCKS proxy by entering the following command:
  142. {' '.join(a+b2)}
  143. Then restart the test.
  144. """,indent=' '))
  145. if not opt.no_daemon_stop:
  146. atexit.register(kill_proxy)
  147. def init_users(self):
  148. from mmgen.daemon import CoinDaemon,MoneroWalletDaemon
  149. from mmgen.rpc import MoneroRPCClient,MoneroRPCClientRaw,MoneroWalletRPCClient
  150. self.users = {}
  151. n = self.tmpdir_nums[0]
  152. ud = namedtuple('user_data',[
  153. 'sid',
  154. 'mmwords',
  155. 'udir',
  156. 'datadir',
  157. 'kal_range',
  158. 'kafile',
  159. 'walletfile_fs',
  160. 'addrfile_fs',
  161. 'md',
  162. 'md_rpc',
  163. 'md_json_rpc',
  164. 'wd',
  165. 'wd_rpc',
  166. ])
  167. for user,sid,shift,kal_range in ( # kal_range must be None, a single digit, or a single hyphenated range
  168. ('miner', '98831F3A', 130, '1-2'),
  169. ('bob', '1378FC64', 140, None),
  170. ('alice', 'FE3C6545', 150, '1-4'),
  171. ):
  172. udir = os.path.join('test',f'tmp{n}',user)
  173. datadir = os.path.join(self.datadir_base,user)
  174. md = CoinDaemon(
  175. proto = self.proto,
  176. test_suite = True,
  177. port_shift = shift,
  178. opts = ['online'],
  179. datadir = datadir
  180. )
  181. md_rpc = MoneroRPCClientRaw(
  182. host = md.host,
  183. port = md.rpc_port,
  184. user = None,
  185. passwd = None,
  186. test_connection = False,
  187. )
  188. md_json_rpc = MoneroRPCClient(
  189. host = md.host,
  190. port = md.rpc_port,
  191. user = None,
  192. passwd = None,
  193. test_connection = False,
  194. )
  195. wd = MoneroWalletDaemon(
  196. proto = self.proto,
  197. test_suite = True,
  198. wallet_dir = udir,
  199. user = 'foo',
  200. passwd = 'bar',
  201. port_shift = shift,
  202. datadir = os.path.join('test','daemons'),
  203. daemon_addr = f'127.0.0.1:{md.rpc_port}',
  204. )
  205. wd_rpc = MoneroWalletRPCClient(
  206. host = wd.host,
  207. port = wd.rpc_port,
  208. user = wd.user,
  209. passwd = wd.passwd,
  210. test_connection = False,
  211. )
  212. self.users[user] = ud(
  213. sid = sid,
  214. mmwords = f'test/ref/{sid}.mmwords',
  215. udir = udir,
  216. datadir = datadir,
  217. kal_range = kal_range,
  218. kafile = f'{udir}/{sid}-XMR-M[{kal_range}].testnet.akeys',
  219. walletfile_fs = f'{udir}/{sid}-{{}}-MoneroWallet.testnet',
  220. addrfile_fs = f'{udir}/{sid}-{{}}-MoneroWallet.testnet.address.txt',
  221. md = md,
  222. md_rpc = md_rpc,
  223. md_json_rpc = md_json_rpc,
  224. wd = wd,
  225. wd_rpc = wd_rpc,
  226. )
  227. def init_daemon_args(self):
  228. common_args = ['--p2p-bind-ip=127.0.0.1','--fixed-difficulty=1'] # ,'--rpc-ssl-allow-any-cert']
  229. for u in self.users:
  230. other_ports = [self.users[u2].md.p2p_port for u2 in self.users if u2 != u]
  231. node_args = [f'--add-exclusive-node=127.0.0.1:{p}' for p in other_ports]
  232. self.users[u].md.usr_coind_args = common_args + node_args
  233. # cmd_group methods
  234. def gen_kafiles(self):
  235. for user,data in self.users.items():
  236. if not data.kal_range:
  237. continue
  238. run(['mkdir','-p',data.udir])
  239. run(f'rm -f {data.kafile}',shell=True)
  240. t = self.spawn(
  241. 'mmgen-keygen', [
  242. '--testnet=1','-q', '--accept-defaults', '--coin=xmr',
  243. f'--outdir={data.udir}', data.mmwords, data.kal_range
  244. ],
  245. extra_desc = f'({capfirst(user)})' )
  246. t.read()
  247. t.ok()
  248. t.skip_ok = True
  249. return t
  250. def create_wallets_miner(self): return self.create_wallets('miner')
  251. def create_wallets_alice(self): return self.create_wallets('alice')
  252. def create_wallets(self,user,wallet=None):
  253. assert wallet is None or is_int(wallet), 'wallet arg'
  254. data = self.users[user]
  255. run(
  256. 'rm -f {}*'.format( data.walletfile_fs.format(wallet or '*') ),
  257. shell = True
  258. )
  259. dir_opt = [f'--wallet-dir={data.udir}']
  260. t = self.spawn(
  261. 'mmgen-xmrwallet',
  262. self.long_opts + dir_opt + [ 'create', data.kafile, (wallet or data.kal_range) ] )
  263. t.expect('Check key-to-address validity? (y/N): ','n')
  264. for i in MMGenRange(wallet or data.kal_range).items:
  265. t.expect('Address: ')
  266. t.read()
  267. return t
  268. async def mine_initial_coins(self):
  269. return await self.mine_chk('miner',1,0,lambda x: x > 20,'unlocked balance > 20')
  270. async def fund_alice(self):
  271. await self.open_wallet_user('miner',1)
  272. await self.transfer(
  273. 'miner',
  274. 1234567891234,
  275. read_from_file(self.users['alice'].addrfile_fs.format(1)),
  276. )
  277. bal = '1.234567891234'
  278. return await self.mine_chk(
  279. 'alice',1,0,
  280. lambda x: str(x) == bal,f'unlocked balance == {bal}',
  281. random_txs = self.dfl_random_txs
  282. )
  283. def sync_wallets_all(self):
  284. return self.sync_wallets('alice',add_opts=['--rescan-blockchain'])
  285. def sync_wallets_selected(self):
  286. return self.sync_wallets('alice',wallets='1-2,4')
  287. def sync_wallets(self,user,wallets=None,add_opts=None):
  288. data = self.users[user]
  289. cmd_opts = list_gen(
  290. [f'--wallet-dir={data.udir}'],
  291. [f'--daemon=localhost:{data.md.rpc_port}'],
  292. )
  293. t = self.spawn(
  294. 'mmgen-xmrwallet',
  295. self.long_opts + cmd_opts + (add_opts or []) + [ 'sync', data.kafile ] + ([wallets] if wallets else []) )
  296. t.expect('Check key-to-address validity? (y/N): ','n')
  297. wlist = AddrIdxList(wallets) if wallets else MMGenRange(data.kal_range).items
  298. for n,wnum in enumerate(wlist):
  299. t.expect('Syncing wallet {}/{} ({})'.format(
  300. n+1,
  301. len(wlist),
  302. os.path.basename(data.walletfile_fs.format(wnum)),
  303. ))
  304. t.expect('Chain height: ')
  305. t.expect('Wallet height: ')
  306. t.expect('Balance: ')
  307. t.read()
  308. return t
  309. def do_op(self, op, user, arg2,
  310. tx_relay_parm = None,
  311. do_not_relay = False,
  312. return_amt = False,
  313. reuse_acct = False,
  314. add_desc = None,
  315. do_ret = False ):
  316. data = self.users[user]
  317. cmd_opts = list_gen(
  318. [f'--wallet-dir={data.udir}'],
  319. [f'--outdir={data.udir}'],
  320. [f'--daemon=localhost:{data.md.rpc_port}'],
  321. [f'--tx-relay-daemon={tx_relay_parm}', tx_relay_parm],
  322. ['--do-not-relay', do_not_relay]
  323. )
  324. add_desc = (', ' + add_desc) if add_desc else ''
  325. t = self.spawn(
  326. 'mmgen-xmrwallet',
  327. self.long_opts + cmd_opts + [ op, data.kafile, arg2 ],
  328. extra_desc = f'({capfirst(user)}{add_desc})' )
  329. t.expect('Check key-to-address validity? (y/N): ','n')
  330. if op == 'sweep':
  331. t.expect(
  332. 'Create new {} .* \(y/N\): '.format(('address','account')[',' in arg2]),
  333. ('y','n')[reuse_acct],
  334. regex=True )
  335. if reuse_acct:
  336. t.expect( 'to last existing account .* \(y/N\): ','y', regex=True )
  337. if return_amt:
  338. amt = XMRAmt(strip_ansi_escapes(t.expect_getend('Amt: ')).replace('XMR','').strip())
  339. if do_not_relay:
  340. t.expect('Save MoneroMMGenTX data? (y/N): ','y')
  341. t.written_to_file('MoneroMMGenTX data')
  342. else:
  343. t.expect(f'Relay {op} transaction? (y/N): ','y')
  344. t.read()
  345. return t if do_ret else amt if return_amt else t.ok()
  346. def sweep_to_address_proxy(self):
  347. self.do_op('sweep','alice','1:0',self.tx_relay_daemon_proxy_parm)
  348. return self.mine_chk('alice',1,0,lambda x: x > 1,'unlocked balance > 1')
  349. def sweep_to_account(self):
  350. self.do_op('sweep','alice','1:0,2')
  351. return self.mine_chk('alice',2,1,lambda x: x > 1,'unlocked balance > 1')
  352. def sweep_to_address_noproxy(self):
  353. self.do_op('sweep','alice','2:1',self.tx_relay_daemon_parm)
  354. return self.mine_chk('alice',2,1,lambda x: x > 0.9,'unlocked balance > 0.9')
  355. def transfer_to_miner_proxy(self):
  356. addr = read_from_file(self.users['miner'].addrfile_fs.format(2))
  357. amt = '0.135'
  358. self.do_op('transfer','alice',f'2:1:{addr},{amt}',self.tx_relay_daemon_proxy_parm)
  359. return self.mine_chk('miner',2,0,lambda x: str(x) == amt,f'unlocked balance == {amt}')
  360. def transfer_to_miner_noproxy(self):
  361. addr = read_from_file(self.users['miner'].addrfile_fs.format(2))
  362. self.do_op('transfer','alice',f'2:1:{addr},0.0995',self.tx_relay_daemon_parm)
  363. return self.mine_chk('miner',2,0,lambda x: str(x) == '0.2345','unlocked balance == 0.2345')
  364. def transfer_to_miner_create(self,amt):
  365. get_file_with_ext(self.users['alice'].udir,'sigtx',delete_all=True)
  366. addr = read_from_file(self.users['miner'].addrfile_fs.format(2))
  367. return self.do_op('transfer','alice',f'2:1:{addr},{amt}',do_not_relay=True,do_ret=True)
  368. def transfer_to_miner_create1(self):
  369. return self.transfer_to_miner_create('0.0111')
  370. def transfer_to_miner_create2(self):
  371. return self.transfer_to_miner_create('0.0012')
  372. def relay_tx(self,relay_opt=None,add_desc=None):
  373. user = 'alice'
  374. data = self.users[user]
  375. fn = get_file_with_ext(data.udir,'sigtx')
  376. add_desc = (', ' + add_desc) if add_desc else ''
  377. t = self.spawn(
  378. 'mmgen-xmrwallet',
  379. self.long_opts
  380. + ([relay_opt] if relay_opt else [])
  381. + [ 'relay', fn ],
  382. extra_desc = f'(relaying TX, {capfirst(user)}{add_desc})' )
  383. t.expect('Relay transaction? ','y')
  384. t.read()
  385. t.ok()
  386. def transfer_to_miner_send1(self):
  387. self.relay_tx(f'--tx-relay-daemon={self.tx_relay_daemon_proxy_parm}',add_desc='via proxy')
  388. return self.mine_chk('miner',2,0,lambda x: str(x) == '0.2456','unlocked balance == 0.2456')
  389. def transfer_to_miner_send2(self):
  390. self.relay_tx(f'--tx-relay-daemon={self.tx_relay_daemon_parm}',add_desc='no proxy')
  391. return self.mine_chk('miner',2,0,lambda x: str(x) == '0.2468','unlocked balance == 0.2468')
  392. async def sweep_create_and_send(self):
  393. bal = XMRAmt('0')
  394. min_bal = XMRAmt('0.9')
  395. for i in range(4):
  396. if i: ok()
  397. get_file_with_ext(self.users['alice'].udir,'sigtx',delete_all=True)
  398. send_amt = self.do_op(
  399. 'sweep','alice','2:1,3', # '2:1,3'
  400. do_not_relay = True,
  401. reuse_acct = True,
  402. add_desc = f'TX #{i+1}',
  403. return_amt = True )
  404. ok()
  405. self.relay_tx(f'--tx-relay-daemon={self.tx_relay_daemon_parm}',add_desc=f'send amt: {send_amt} XMR')
  406. bal += await self.mine_chk('alice',3,0,lambda x: 'chk_bal_chg','balance has changed') # 3,0
  407. if bal >= min_bal:
  408. return 'ok'
  409. return False
  410. # wallet methods
  411. async def open_wallet_user(self,user,wnum):
  412. data = self.users[user]
  413. silence()
  414. kal = KeyAddrList(self.proto,data.kafile,skip_key_address_validity_check=True)
  415. end_silence()
  416. if user != 'miner':
  417. self.users[user].wd.start(silent=not (opt.exact_output or opt.verbose))
  418. return await data.wd_rpc.call(
  419. 'open_wallet',
  420. filename = os.path.basename(data.walletfile_fs.format(wnum)),
  421. password = kal.entry(wnum).wallet_passwd )
  422. async def close_wallet_user(self,user):
  423. ret = await self.users[user].wd_rpc.call('close_wallet')
  424. if user != 'miner':
  425. self.users[user].wd.stop(silent=not (opt.exact_output or opt.verbose))
  426. return 'ok'
  427. # mining methods
  428. async def start_mining(self):
  429. data = self.users['miner']
  430. addr = read_from_file(data.addrfile_fs.format(1)) # mine to wallet #1, account 0
  431. for i in range(20):
  432. ret = await data.md_rpc.call(
  433. 'start_mining',
  434. do_background_mining = False, # run mining in background or foreground
  435. ignore_battery = True, # ignore battery state (on laptop)
  436. miner_address = addr, # account address to mine to
  437. threads_count = 3 ) # number of mining threads to run
  438. status = self.get_status(ret)
  439. if status == 'OK':
  440. return True
  441. elif status == 'BUSY':
  442. await asyncio.sleep(5)
  443. omsg('Daemon busy. Attempting to start mining...')
  444. else:
  445. die(2,f'Monerod returned status {status}')
  446. else:
  447. die(2,'Max retries exceeded')
  448. async def stop_mining(self):
  449. ret = await self.users['miner'].md_rpc.call('stop_mining')
  450. return self.get_status(ret)
  451. async def mine_chk(self,user,wnum,account,test,test_desc,random_txs=None):
  452. """
  453. - open destination wallet
  454. - optionally create and broadcast random TXs
  455. - start mining
  456. - mine until funds appear in wallet
  457. - stop mining
  458. - close wallet
  459. """
  460. async def get_height():
  461. u = self.users['miner']
  462. for i in range(20):
  463. try:
  464. return (await u.md_json_rpc.call('get_last_block_header'))['block_header']['height']
  465. except Exception as e:
  466. if 'onnection refused' in str(e):
  467. omsg(f'{e}\nMonerod appears to have crashed. Attempting to restart...')
  468. await asyncio.sleep(5)
  469. u.md.restart()
  470. await asyncio.sleep(5)
  471. await self.start_mining()
  472. else:
  473. raise
  474. else:
  475. die(2,'Restart attempt limit exceeded')
  476. async def send_random_txs():
  477. from mmgen.tool import tool_api
  478. t = tool_api()
  479. t.init_coin('XMR','testnet')
  480. t.usr_randchars = 0
  481. imsg_r(f'Sending random transactions: ')
  482. for i in range(random_txs):
  483. await self.transfer(
  484. 'miner',
  485. 123456789,
  486. t.randpair()[1],
  487. )
  488. imsg_r(f'{i+1} ')
  489. oqmsg_r('+')
  490. await asyncio.sleep(0.5)
  491. imsg('')
  492. def print_balance(dest,ub):
  493. imsg('Total balance in {}’s wallet {}, account #{}: {}'.format(
  494. capfirst(dest.user),
  495. dest.wnum,
  496. dest.account,
  497. ub.hl()
  498. ))
  499. async def get_balance(dest,count):
  500. data = self.users[dest.user]
  501. await data.wd_rpc.call('refresh')
  502. if count and not count % 20:
  503. await data.wd_rpc.call('rescan_blockchain')
  504. ret = await data.wd_rpc.call('get_accounts')
  505. return XMRAmt(ret['subaddress_accounts'][dest.account]['unlocked_balance'],from_unit='atomic')
  506. # start execution:
  507. self.do_msg(extra_desc =
  508. (f'sending {random_txs} random TXs, ' if random_txs else '') +
  509. f'mining, checking wallet {user}:{wnum}:{account}' )
  510. dest = namedtuple(
  511. 'dest_info',['user','wnum','account','test','test_desc'])(user,wnum,account,test,test_desc)
  512. await self.open_wallet_user(dest.user,dest.wnum)
  513. ub_start = await get_balance(dest,0)
  514. chk_bal_chg = dest.test(ub_start) == 'chk_bal_chg'
  515. if random_txs:
  516. await send_random_txs()
  517. await self.start_mining()
  518. h = await get_height()
  519. imsg_r(f'Chain height: {h} ')
  520. for count in range(500):
  521. ub = await get_balance(dest,count)
  522. if dest.test(ub) is True or ( chk_bal_chg and ub != ub_start ):
  523. imsg('')
  524. oqmsg_r('+')
  525. print_balance(dest,ub)
  526. break
  527. await asyncio.sleep(2)
  528. h = await get_height()
  529. imsg_r(f'{h} ')
  530. oqmsg_r('+')
  531. else:
  532. die(2,'Timeout exceeded, balance {ub!r}')
  533. await self.stop_mining()
  534. await self.close_wallet_user(dest.user)
  535. return ub if chk_bal_chg else 'ok'
  536. # util methods
  537. def get_status(self,ret):
  538. if ret['status'] != 'OK':
  539. imsg( 'RPC status: {}'.format( ret['status'] ))
  540. return ret['status']
  541. def do_msg(self,extra_desc=None):
  542. self.spawn(
  543. '',
  544. msg_only = True,
  545. extra_desc = f'({extra_desc})' if extra_desc else None
  546. )
  547. async def transfer(self,user,amt,addr):
  548. return await self.users[user].wd_rpc.call('transfer',destinations=[{'amount':amt,'address':addr}])
  549. # daemon start/stop methods
  550. def start_daemons(self):
  551. self.stop_daemons()
  552. for v in self.users.values():
  553. run(['mkdir','-p',v.datadir])
  554. v.md.start()
  555. def stop_daemons(self):
  556. for v in self.users.values():
  557. if v.md.state != 'stopped':
  558. v.md.stop()
  559. run(['rm','-rf',self.datadir_base])
  560. def start_wallet_daemons(self):
  561. self.users['miner'].wd.start()
  562. def stop_wallet_daemons(self):
  563. self.users['miner'].wd.stop()