The MMGen Project
00a8ff30ad
DAEMON VERSION - Bitcoin Core: 25.1.0 - Bitcoin Cash Node: 26.1.0 - Geth: 1.13.4
1499 lines
52 KiB
Python
Executable file
1499 lines
52 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
#
|
|
# mmgen = Multi-Mode GENerator, command-line Bitcoin cold storage solution
|
|
# Copyright (C)2013-2023 The MMGen Project <mmgen@tuta.io>
|
|
#
|
|
# This program is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
"""
|
|
test.cmdtest_py_d.ct_ethdev: Ethdev tests for the cmdtest.py test suite
|
|
"""
|
|
|
|
import sys,os,re,shutil,asyncio,json
|
|
from decimal import Decimal
|
|
from collections import namedtuple
|
|
from subprocess import run,PIPE,DEVNULL
|
|
|
|
from mmgen.color import yellow,blue,cyan,set_vt100
|
|
from mmgen.util import msg,rmsg,die
|
|
|
|
from ..include.common import (
|
|
cfg,
|
|
check_solc_ver,
|
|
omsg,
|
|
imsg,
|
|
imsg_r,
|
|
joinpath,
|
|
read_from_file,
|
|
write_to_file,
|
|
cmp_or_die,
|
|
strip_ansi_escapes,
|
|
silence,
|
|
end_silence,
|
|
gr_uc,
|
|
stop_test_daemons
|
|
)
|
|
from .common import (
|
|
ref_dir,
|
|
dfl_words_file,
|
|
tx_comment_jp,
|
|
tx_comment_lat_cyr_gr,
|
|
tw_comment_zh,
|
|
tw_comment_lat_cyr_gr,
|
|
get_file_with_ext,
|
|
ok_msg,
|
|
Ctrl_U
|
|
)
|
|
from .ct_base import CmdTestBase
|
|
from .ct_shared import CmdTestShared
|
|
|
|
del_addrs = ('4','1')
|
|
dfl_sid = '98831F3A'
|
|
|
|
# The OpenEthereum dev address with lots of coins. Create with "ethkey -b info ''":
|
|
dfl_devaddr = '00a329c0648769a73afac7f9381e08fb43dbea72'
|
|
dfl_devkey = '4d5db4107d237df6a3d58ee5f70ae63d73d7658d4026f2eefd2f204c81682cb7'
|
|
|
|
burn_addr = 'deadbeef'*5
|
|
burn_addr2 = 'beadcafe'*5
|
|
|
|
amt1 = '999999.12345689012345678'
|
|
amt2 = '888.111122223333444455'
|
|
|
|
parity_devkey_fn = 'parity.devkey'
|
|
|
|
def set_vbals(daemon_id):
|
|
global vbal1, vbal2, vbal3, vbal4, vbal5, vbal6, vbal7, vbal9
|
|
if daemon_id == 'geth':
|
|
vbal1 = '1.2288347'
|
|
vbal2 = '99.996561415'
|
|
vbal3 = '1.23141825'
|
|
vbal4 = '127.0287847'
|
|
vbal5 = '1000126.14775300512345678'
|
|
vbal6 = '1000126.14880300512345678'
|
|
vbal7 = '1000124.91891830512345678'
|
|
vbal9 = '1.22625235'
|
|
else:
|
|
vbal1 = '1.2288409'
|
|
vbal2 = '99.997088755'
|
|
vbal3 = '1.23142525'
|
|
vbal4 = '127.0287909'
|
|
vbal5 = '1000126.14828654512345678'
|
|
vbal6 = '1000126.14933654512345678'
|
|
vbal7 = '1000124.91944564512345678'
|
|
vbal9 = '1.22626295'
|
|
|
|
bals = lambda k: {
|
|
'1': [ ('98831F3A:E:1','123.456')],
|
|
'2': [ ('98831F3A:E:1','123.456'),('98831F3A:E:11','1.234')],
|
|
'3': [ ('98831F3A:E:1','123.456'),('98831F3A:E:11','1.234'),('98831F3A:E:21','2.345')],
|
|
'4': [ ('98831F3A:E:1','100'),
|
|
('98831F3A:E:2','23.45495'),
|
|
('98831F3A:E:11','1.234'),
|
|
('98831F3A:E:21','2.345')],
|
|
'5': [ ('98831F3A:E:1','100'),
|
|
('98831F3A:E:2','23.45495'),
|
|
('98831F3A:E:11','1.234'),
|
|
('98831F3A:E:21','2.345'),
|
|
(burn_addr + r'\s+non-MMGen',amt1)],
|
|
'8': [ ('98831F3A:E:1','0'),
|
|
('98831F3A:E:2','23.45495'),
|
|
('98831F3A:E:11',vbal1,'a1'),
|
|
('98831F3A:E:12','99.99895'),
|
|
('98831F3A:E:21','2.345'),
|
|
(burn_addr + r'\s+non-MMGen',amt1)],
|
|
'9': [ ('98831F3A:E:1','0'),
|
|
('98831F3A:E:2','23.45495'),
|
|
('98831F3A:E:11',vbal1,'a1'),
|
|
('98831F3A:E:12',vbal2),
|
|
('98831F3A:E:21','2.345'),
|
|
(burn_addr + r'\s+non-MMGen',amt1)],
|
|
'10': [ ('98831F3A:E:1','0'),
|
|
('98831F3A:E:2','23.0218'),
|
|
('98831F3A:E:3','0.4321'),
|
|
('98831F3A:E:11',vbal1,'a1'),
|
|
('98831F3A:E:12',vbal2),
|
|
('98831F3A:E:21','2.345'),
|
|
(burn_addr + r'\s+non-MMGen',amt1)]
|
|
}[k]
|
|
|
|
token_bals = lambda k: {
|
|
'1': [ ('98831F3A:E:11','1000','1.234')],
|
|
'2': [ ('98831F3A:E:11','998.76544',vbal3,'a1'),
|
|
('98831F3A:E:12','1.23456','0')],
|
|
'3': [ ('98831F3A:E:11','110.654317776666555545',vbal1,'a1'),
|
|
('98831F3A:E:12','1.23456','0')],
|
|
'4': [ ('98831F3A:E:11','110.654317776666555545',vbal1,'a1'),
|
|
('98831F3A:E:12','1.23456','0'),
|
|
(burn_addr + r'\s+non-MMGen',amt2,amt1)],
|
|
'5': [ ('98831F3A:E:11','110.654317776666555545',vbal1,'a1'),
|
|
('98831F3A:E:12','1.23456','99.99895'),
|
|
(burn_addr + r'\s+non-MMGen',amt2,amt1)],
|
|
'6': [ ('98831F3A:E:11','110.654317776666555545',vbal1,'a1'),
|
|
('98831F3A:E:12','0',vbal2),
|
|
('98831F3A:E:13','1.23456','0'),
|
|
(burn_addr + r'\s+non-MMGen',amt2,amt1)],
|
|
'7': [ ('98831F3A:E:11','67.444317776666555545',vbal9,'a2'),
|
|
('98831F3A:E:12','43.21',vbal2),
|
|
('98831F3A:E:13','1.23456','0'),
|
|
(burn_addr + r'\s+non-MMGen',amt2,amt1)]
|
|
}[k]
|
|
|
|
token_bals_getbalance = lambda k: {
|
|
'1': (vbal4,'999999.12345689012345678'),
|
|
'2': ('111.888877776666555545','888.111122223333444455')
|
|
}[k]
|
|
|
|
coin = cfg.coin
|
|
|
|
class CmdTestEthdev(CmdTestBase,CmdTestShared):
|
|
'Ethereum transacting, token deployment and tracking wallet operations'
|
|
networks = ('eth','etc')
|
|
passthru_opts = ('coin','daemon_id','http_timeout','rpc_backend')
|
|
extra_spawn_args = ['--regtest=1']
|
|
tmpdir_nums = [22]
|
|
color = True
|
|
cmd_group_in = (
|
|
('setup', f'dev mode tests for coin {coin} (start daemon)'),
|
|
('subgroup.misc', []),
|
|
('subgroup.init', []),
|
|
('subgroup.msg', ['init']),
|
|
('subgroup.main', ['init']),
|
|
('subgroup.contract', ['main']),
|
|
('subgroup.token', ['contract']),
|
|
('subgroup.twexport', ['token']),
|
|
('subgroup.cached', ['token']),
|
|
('subgroup.view', ['cached']),
|
|
('subgroup.label', ['cached']),
|
|
('subgroup.remove', ['cached']),
|
|
('stop', 'stopping daemon'),
|
|
)
|
|
cmd_subgroups = {
|
|
'misc': (
|
|
'miscellaneous commands',
|
|
('daemon_version', 'mmgen-tool daemon_version'),
|
|
),
|
|
'init': (
|
|
'initializing wallets',
|
|
('wallet_upgrade1', 'upgrading the tracking wallet (v1 -> v2)'),
|
|
('wallet_upgrade2', 'upgrading the tracking wallet (v2 -> v3)'),
|
|
('addrgen', 'generating addresses'),
|
|
('addrimport', 'importing addresses'),
|
|
('addrimport_dev_addr', "importing dev faucet address 'Ox00a329c..'"),
|
|
('fund_dev_address', 'funding the default (Parity dev) address'),
|
|
),
|
|
'msg': (
|
|
'message signing',
|
|
('msgsign_chk', "signing a message (low-level, check against 'eth_sign' RPC call)"),
|
|
('msgcreate', 'creating a message file'),
|
|
('msgsign', 'signing the message file'),
|
|
('msgverify', 'verifying the message file'),
|
|
('msgexport', 'exporting the message file data to JSON for third-party verifier'),
|
|
('msgverify_export', 'verifying the exported JSON data'),
|
|
|
|
('msgcreate_raw', 'creating a message file (--msghash-type=raw)'),
|
|
('msgsign_raw', 'signing the message file (msghash_type=raw)'),
|
|
('msgverify_raw', 'verifying the message file (msghash_type=raw)'),
|
|
('msgexport_raw', 'exporting the message file data to JSON (msghash_type=raw)'),
|
|
('msgverify_export_raw', 'verifying the exported JSON data (msghash_type=raw)'),
|
|
),
|
|
'main': (
|
|
'creating, signing, sending and bumping Ethereum transactions',
|
|
('txcreate1', 'creating a transaction (spend from dev address to address :1)'),
|
|
('txview1_raw', 'viewing the raw transaction'),
|
|
('txsign1', 'signing the transaction'),
|
|
('txview1_sig', 'viewing the signed transaction'),
|
|
('tx_status0_bad', 'getting the transaction status'),
|
|
('txsign1_ni', 'signing the transaction (non-interactive)'),
|
|
('txsend1', 'sending the transaction'),
|
|
('bal1', f'the {coin} balance'),
|
|
|
|
('txcreate2', 'creating a transaction (spend from dev address to address :11)'),
|
|
('txsign2', 'signing the transaction'),
|
|
('txsend2', 'sending the transaction'),
|
|
('bal2', f'the {coin} balance'),
|
|
|
|
('txcreate3', 'creating a transaction (spend from dev address to address :21)'),
|
|
('txsign3', 'signing the transaction'),
|
|
('txsend3', 'sending the transaction'),
|
|
('bal3', f'the {coin} balance'),
|
|
|
|
('tx_status1', 'getting the transaction status'),
|
|
|
|
('txcreate4', 'creating a transaction (spend from MMGen address, low TX fee)'),
|
|
('txbump', 'bumping the transaction fee'),
|
|
|
|
('txsign4', 'signing the transaction'),
|
|
('txsend4', 'sending the transaction'),
|
|
('tx_status1a', 'getting the transaction status'),
|
|
('bal4', f'the {coin} balance'),
|
|
|
|
('txcreate5', 'creating a transaction (fund burn address)'),
|
|
('txsign5', 'signing the transaction'),
|
|
('txsend5', 'sending the transaction'),
|
|
|
|
('addrimport_burn_addr', 'importing burn address'),
|
|
('bal5', f'the {coin} balance'),
|
|
|
|
('add_comment1', 'adding a UTF-8 label (zh)'),
|
|
('chk_comment1', 'checking the label'),
|
|
('add_comment2', 'adding a UTF-8 label (lat+cyr+gr)'),
|
|
('chk_comment2', 'checking the label'),
|
|
('remove_comment', 'removing the label'),
|
|
),
|
|
'contract': (
|
|
'creating and deploying ERC20 tokens',
|
|
('token_compile1', 'compiling ERC20 token #1'),
|
|
('token_deploy1a', 'deploying ERC20 token #1 (SafeMath)'),
|
|
('token_deploy1b', 'deploying ERC20 token #1 (Owned)'),
|
|
('token_deploy1c', 'deploying ERC20 token #1 (Token)'),
|
|
|
|
('tx_status2', 'getting the transaction status'),
|
|
('bal6', f'the {coin} balance'),
|
|
|
|
('token_compile2', 'compiling ERC20 token #2'),
|
|
('token_deploy2a', 'deploying ERC20 token #2 (SafeMath)'),
|
|
('token_deploy2b', 'deploying ERC20 token #2 (Owned)'),
|
|
('token_deploy2c', 'deploying ERC20 token #2 (Token)'),
|
|
|
|
('contract_deploy', 'deploying contract (create,sign,send)'),
|
|
),
|
|
'token': (
|
|
'creating, signing, sending and bumping ERC20 token transactions',
|
|
|
|
('token_fund_users', 'transferring token funds from dev to user'),
|
|
('token_user_bals', 'show balances after transfer'),
|
|
('token_addrgen', 'generating token addresses'),
|
|
('token_addrimport_badaddr1', 'importing token addresses (no token address)'),
|
|
('token_addrimport_badaddr2', 'importing token addresses (bad token address)'),
|
|
('token_addrimport_addr1', 'importing token addresses using token address (MM1)'),
|
|
('token_addrimport_addr2', 'importing token addresses using token address (MM2)'),
|
|
('token_addrimport_batch', 'importing token addresses (dummy batch mode) (MM1)'),
|
|
('token_addrimport_sym', 'importing token addresses using token symbol (MM2)'),
|
|
|
|
('bal7', f'the {coin} balance'),
|
|
('token_bal1', f'the {coin} balance and token balance'),
|
|
|
|
('token_txcreate1', 'creating a token transaction'),
|
|
('token_txview1_raw', 'viewing the raw transaction'),
|
|
('token_txsign1', 'signing the transaction'),
|
|
('token_txsend1', 'sending the transaction'),
|
|
('token_txview1_sig', 'viewing the signed transaction'),
|
|
('tx_status3', 'getting the transaction status'),
|
|
('token_bal2', f'the {coin} balance and token balance'),
|
|
|
|
('token_txcreate2', 'creating a token transaction (to burn address)'),
|
|
('token_txbump', 'bumping the transaction fee'),
|
|
|
|
('token_txsign2', 'signing the transaction'),
|
|
('token_txsend2', 'sending the transaction'),
|
|
('token_bal3', f'the {coin} balance and token balance'),
|
|
|
|
('del_dev_addr', 'deleting the dev address'),
|
|
|
|
('bal1_getbalance', f'the {coin} balance (getbalance)'),
|
|
|
|
('addrimport_token_burn_addr', 'importing the token burn address'),
|
|
|
|
('token_bal4', f'the {coin} balance and token balance'),
|
|
('token_bal_getbalance', 'the token balance (getbalance)'),
|
|
|
|
('txcreate_noamt', 'creating a transaction (full amount send)'),
|
|
('txsign_noamt', 'signing the transaction'),
|
|
('txsend_noamt', 'sending the transaction'),
|
|
|
|
('bal8', f'the {coin} balance'),
|
|
('token_bal5', 'the token balance'),
|
|
|
|
('token_txcreate_noamt', 'creating a token transaction (full amount send)'),
|
|
('token_txsign_noamt', 'signing the transaction'),
|
|
('token_txsend_noamt', 'sending the transaction'),
|
|
|
|
('bal9', f'the {coin} balance'),
|
|
('token_bal6', 'the token balance'),
|
|
|
|
('listaddresses1', 'listaddresses'),
|
|
('listaddresses2', 'listaddresses minconf=999999999 (ignored)'),
|
|
('listaddresses3', 'listaddresses sort=age (ignored)'),
|
|
('listaddresses4', 'listaddresses showempty=1 sort=age (ignored)'),
|
|
|
|
('token_listaddresses1', 'listaddresses --token=mm1'),
|
|
('token_listaddresses2', 'listaddresses --token=mm1 showempty=1'),
|
|
),
|
|
'twexport': (
|
|
'exporting and importing tracking wallet to JSON',
|
|
('twexport_noamt', 'exporting the tracking wallet (include_amts=0)'),
|
|
('twmove', 'moving the tracking wallet'),
|
|
('twimport', 'importing the tracking wallet'),
|
|
('twview7', 'twview (cached_balances=1)'),
|
|
('twview8', 'twview'),
|
|
('twexport', 'exporting the tracking wallet'),
|
|
('tw_chktotal', 'checking total value in tracking wallet dump'),
|
|
('twmove', 'moving the tracking wallet'),
|
|
('twimport', 'importing the tracking wallet'),
|
|
('twcompare', 'comparing imported tracking wallet with original'),
|
|
('edit_json_twdump', 'editing the tracking wallet JSON dump'),
|
|
('twmove', 'moving the tracking wallet'),
|
|
('twimport_nochksum', 'importing the edited tracking wallet JSON dump (ignore_checksum=1)'),
|
|
|
|
('token_listaddresses3', 'listaddresses --token=mm1 showempty=1'),
|
|
('token_listaddresses4', 'listaddresses --token=mm2 showempty=1'),
|
|
('twview9', 'twview (check balance)'),
|
|
),
|
|
'cached': (
|
|
'creating and sending transactions using cached balances',
|
|
('twview_cached_balances', 'twview (cached balances)'),
|
|
('token_twview_cached_balances', 'token twview (cached balances)'),
|
|
('txcreate_cached_balances', 'txcreate (cached balances)'),
|
|
('token_txcreate_cached_balances', 'token txcreate (cached balances)'),
|
|
|
|
('txdo_cached_balances', 'txdo (cached balances)'),
|
|
('txcreate_refresh_balances', 'refreshing balances'),
|
|
('bal10', f'the {coin} balance'),
|
|
|
|
('token_txdo_cached_balances', 'token txdo (cached balances)'),
|
|
('token_txcreate_refresh_balances', 'refreshing token balances'),
|
|
('token_bal7', 'the token balance'),
|
|
),
|
|
'view': (
|
|
'viewing addresses and unspent outputs',
|
|
('twview1', 'twview'),
|
|
('twview2', 'twview wide=1'),
|
|
('twview3', 'twview wide=1 sort=age (ignored)'),
|
|
('twview4', 'twview wide=1 minconf=999999999 (ignored)'),
|
|
('twview5', 'twview wide=1 minconf=0 (ignored)'),
|
|
('token_twview1', 'twview --token=mm1'),
|
|
('token_twview2', 'twview --token=mm1 wide=1'),
|
|
('token_twview3', 'twview --token=mm1 wide=1 sort=age (ignored)'),
|
|
),
|
|
'label': (
|
|
'creating, editing and removing labels',
|
|
('edit_comment1', f'adding label to addr #{del_addrs[0]} in {coin} tracking wallet (zh)'),
|
|
('edit_comment2', f'editing label for addr #{del_addrs[0]} in {coin} tracking wallet (zh)'),
|
|
('edit_comment3', f'adding label to addr #{del_addrs[1]} in {coin} tracking wallet (lat+cyr+gr)'),
|
|
('edit_comment4', f'removing label from addr #{del_addrs[0]} in {coin} tracking wallet'),
|
|
('token_edit_comment1', f'adding label to addr #{del_addrs[0]} in {coin} token tracking wallet'),
|
|
),
|
|
'remove': (
|
|
'removing addresses from tracking wallet',
|
|
('remove_addr1', f'removing addr #{del_addrs[0]} from {coin} tracking wallet'),
|
|
('twview6', 'twview (balance reduced after address removal)'),
|
|
('remove_addr2', f'removing addr #{del_addrs[1]} from {coin} tracking wallet'),
|
|
('token_remove_addr1', f'removing addr #{del_addrs[0]} from {coin} token tracking wallet'),
|
|
('token_remove_addr2', f'removing addr #{del_addrs[1]} from {coin} token tracking wallet'),
|
|
),
|
|
}
|
|
|
|
def __init__(self,trunner,cfgs,spawn):
|
|
CmdTestBase.__init__(self,trunner,cfgs,spawn)
|
|
if trunner is None:
|
|
return
|
|
|
|
from mmgen.protocol import init_proto
|
|
self.proto = init_proto( cfg, cfg.coin, network='regtest', need_amt=True )
|
|
|
|
from mmgen.daemon import CoinDaemon
|
|
self.daemon = CoinDaemon( cfg, self.proto.coin+'_rt', test_suite=True )
|
|
|
|
set_vbals(self.daemon.id)
|
|
|
|
self.using_solc = check_solc_ver()
|
|
if not self.using_solc:
|
|
omsg(yellow('Using precompiled contract data'))
|
|
|
|
omsg(blue(f'Coin daemon {self.daemon.id!r} selected'))
|
|
|
|
self.genesis_fn = joinpath(self.tmpdir,'genesis.json')
|
|
self.keystore_dir = os.path.relpath(joinpath(self.daemon.datadir,'keystore'))
|
|
|
|
write_to_file(
|
|
joinpath(self.tmpdir,parity_devkey_fn),
|
|
dfl_devkey+'\n' )
|
|
|
|
self.message = 'attack at dawn'
|
|
self.spawn_env['MMGEN_BOGUS_SEND'] = ''
|
|
|
|
@property
|
|
def eth_args(self):
|
|
return [
|
|
f'--outdir={self.tmpdir}',
|
|
f'--coin={self.proto.coin}',
|
|
f'--rpc-port={self.daemon.rpc_port}',
|
|
'--quiet'
|
|
]
|
|
|
|
@property
|
|
def eth_args_noquiet(self):
|
|
return self.eth_args[:-1]
|
|
|
|
@property
|
|
async def rpc(self):
|
|
from mmgen.rpc import rpc_init
|
|
return await rpc_init(cfg,self.proto)
|
|
|
|
async def setup(self):
|
|
self.spawn('',msg_only=True)
|
|
|
|
if not self.using_solc:
|
|
srcdir = os.path.join(self.tr.repo_root,'test','ref','ethereum','bin')
|
|
from shutil import copytree
|
|
for d in ('mm1','mm2'):
|
|
copytree(os.path.join(srcdir,d),os.path.join(self.tmpdir,d))
|
|
|
|
d = self.daemon
|
|
|
|
if d.id in ('geth','erigon'):
|
|
self.genesis_setup(d)
|
|
set_vt100()
|
|
|
|
if d.id == 'erigon':
|
|
self.write_to_tmpfile('signer_key',self.keystore_data['key']+'\n')
|
|
d.usr_coind_args = [
|
|
'--miner.sigfile={}'.format(os.path.join(self.tmpdir,'signer_key')),
|
|
'--miner.etherbase={}'.format(self.keystore_data['address']) ]
|
|
|
|
if d.id in ('geth','erigon'):
|
|
imsg(' {:19} {}'.format('Cmdline:',' '.join(e for e in d.start_cmd if not 'verbosity' in e)))
|
|
|
|
if not cfg.no_daemon_autostart:
|
|
if not d.id in ('geth','erigon'):
|
|
d.stop(silent=True)
|
|
d.remove_datadir()
|
|
d.start( silent = not (cfg.verbose or cfg.exact_output) )
|
|
rpc = await self.rpc
|
|
imsg(f'Daemon: {rpc.daemon.coind_name} v{rpc.daemon_version_str}')
|
|
|
|
return 'ok'
|
|
|
|
@property
|
|
def keystore_data(self):
|
|
if not hasattr(self,'_keystore_data'):
|
|
|
|
wallet_fn = os.path.join( self.keystore_dir, os.listdir(self.keystore_dir)[0] )
|
|
|
|
from mmgen.proto.eth.misc import extract_key_from_geth_keystore_wallet
|
|
key = extract_key_from_geth_keystore_wallet(
|
|
cfg = cfg,
|
|
wallet_fn = wallet_fn,
|
|
passwd = b'' )
|
|
|
|
with open(wallet_fn) as fh:
|
|
res = json.loads(fh.read())
|
|
|
|
res.update( { 'key': key.hex() } )
|
|
self._keystore_data = res
|
|
|
|
return self._keystore_data
|
|
|
|
def genesis_setup(self,d):
|
|
|
|
def make_key():
|
|
pwfile = joinpath(self.tmpdir,'account_passwd')
|
|
write_to_file(pwfile,'')
|
|
run(['rm','-rf',self.keystore_dir])
|
|
cmd = f'geth account new --password={pwfile} --lightkdf --keystore {self.keystore_dir}'
|
|
cp = run(cmd.split(),stdout=PIPE,stderr=PIPE)
|
|
if cp.returncode:
|
|
die(1,cp.stderr.decode())
|
|
|
|
def make_genesis(signer_addr,prealloc_addr):
|
|
return {
|
|
'config': {
|
|
'chainId': 1337, # TODO: replace constant with var
|
|
'homesteadBlock': 0,
|
|
'eip150Block': 0,
|
|
'eip155Block': 0,
|
|
'eip158Block': 0,
|
|
'byzantiumBlock': 0,
|
|
'constantinopleBlock': 0,
|
|
'petersburgBlock': 0,
|
|
'istanbulBlock': 0,
|
|
'muirGlacierBlock': 0,
|
|
'berlinBlock': 0,
|
|
'londonBlock': 0,
|
|
'arrowGlacierBlock': 0,
|
|
'grayGlacierBlock': 0,
|
|
'shanghaiTime': 0,
|
|
'terminalTotalDifficulty': 0,
|
|
'terminalTotalDifficultyPassed': True,
|
|
'isDev': True
|
|
},
|
|
'nonce': '0x0',
|
|
'timestamp': '0x0',
|
|
'extraData': '0x',
|
|
'gasLimit': '0xaf79e0',
|
|
'difficulty': '0x0',
|
|
'mixHash': '0x0000000000000000000000000000000000000000000000000000000000000000',
|
|
'coinbase': '0x0000000000000000000000000000000000000000',
|
|
'number': '0x0',
|
|
'gasUsed': '0x0',
|
|
'parentHash': '0x0000000000000000000000000000000000000000000000000000000000000000',
|
|
'baseFeePerGas': '0x3b9aca00',
|
|
'excessBlobGas': None,
|
|
'blobGasUsed': None,
|
|
'alloc': {
|
|
prealloc_addr: { 'balance': hex(prealloc_amt.toWei()) }
|
|
}
|
|
}
|
|
|
|
def init_genesis(fn):
|
|
cmd = f'{d.exec_fn} init --datadir {d.datadir} {fn}'
|
|
cp = run( cmd.split(), stdout=PIPE, stderr=PIPE )
|
|
if cp.returncode:
|
|
die(1,cp.stderr.decode())
|
|
|
|
d.stop(quiet=True)
|
|
d.remove_datadir()
|
|
|
|
imsg(cyan('Initializing Genesis Block:'))
|
|
|
|
prealloc_amt = self.proto.coin_amt('1_000_000_000')
|
|
|
|
make_key()
|
|
signer_addr = self.keystore_data['address']
|
|
self.write_to_tmpfile( 'signer_addr', signer_addr + '\n' )
|
|
|
|
imsg(f' Keystore: {self.keystore_dir}')
|
|
imsg(f' Signer key: {self.keystore_data["key"]}')
|
|
imsg(f' Signer address: {signer_addr}')
|
|
imsg(f' Faucet: {dfl_devaddr} ({prealloc_amt} ETH)')
|
|
imsg(f' Genesis block data: {self.genesis_fn}')
|
|
|
|
genesis_data = make_genesis(signer_addr,dfl_devaddr)
|
|
write_to_file( self.genesis_fn, json.dumps(genesis_data,indent=' ')+'\n' )
|
|
init_genesis(self.genesis_fn)
|
|
|
|
def wallet_upgrade(self,src_file):
|
|
if self.proto.coin == 'ETC':
|
|
msg(f'skipping test {self.test_name!r} for ETC')
|
|
return 'skip'
|
|
src_dir = joinpath(ref_dir,'ethereum')
|
|
dest_dir = joinpath(self.tr.data_dir,'altcoins',self.proto.coin.lower())
|
|
w_from = joinpath(src_dir,src_file)
|
|
w_to = joinpath(dest_dir,'tracking-wallet.json')
|
|
os.makedirs(dest_dir,mode=0o750,exist_ok=True)
|
|
dest = shutil.copy2(w_from,w_to)
|
|
assert dest == w_to, dest
|
|
t = self.spawn('mmgen-tool', self.eth_args + ['twview'])
|
|
t.read()
|
|
os.unlink(w_to)
|
|
return t
|
|
|
|
def daemon_version(self):
|
|
t = self.spawn('mmgen-tool', self.eth_args + ['daemon_version'])
|
|
t.expect('version')
|
|
return t
|
|
|
|
def wallet_upgrade1(self):
|
|
return self.wallet_upgrade('tracking-wallet-v1.json')
|
|
def wallet_upgrade2(self):
|
|
return self.wallet_upgrade('tracking-wallet-v2.json')
|
|
|
|
def addrgen(self,addrs='1-3,11-13,21-23'):
|
|
t = self.spawn('mmgen-addrgen', self.eth_args + [dfl_words_file,addrs])
|
|
t.written_to_file('Addresses')
|
|
return t
|
|
|
|
def addrimport(self,ext='21-23]{}.regtest.addrs',expect='9/9',add_args=[],bad_input=False):
|
|
ext = ext.format('-α' if cfg.debug_utf8 else '')
|
|
fn = self.get_file_with_ext(ext,no_dot=True,delete=False)
|
|
t = self.spawn('mmgen-addrimport', self.eth_args[1:-1] + add_args + [fn])
|
|
if bad_input:
|
|
return t
|
|
t.expect('Importing')
|
|
t.expect(expect)
|
|
return t
|
|
|
|
def addrimport_one_addr(self,addr=None,extra_args=[]):
|
|
t = self.spawn('mmgen-addrimport', self.eth_args[1:] + extra_args + ['--address='+addr])
|
|
t.expect('OK')
|
|
return t
|
|
|
|
def addrimport_dev_addr(self):
|
|
return self.addrimport_one_addr(addr=dfl_devaddr)
|
|
|
|
def addrimport_burn_addr(self):
|
|
return self.addrimport_one_addr(addr=burn_addr)
|
|
|
|
def txcreate(
|
|
self,
|
|
args = [],
|
|
menu = [],
|
|
acct = '1',
|
|
caller = 'txcreate',
|
|
interactive_fee = '50G',
|
|
fee_info_data = ('0.00105','50'),
|
|
no_read = False,
|
|
tweaks = []):
|
|
fee_info_pat = r'\D{}\D.*{c} .*\D{}\D.*gas price in Gwei'.format( *fee_info_data, c=self.proto.coin )
|
|
t = self.spawn('mmgen-'+caller, self.eth_args + ['-B'] + args)
|
|
t.expect(r'add \[l\]abel, .*?:.','p', regex=True)
|
|
t.written_to_file('Account balances listing')
|
|
t = self.txcreate_ui_common(
|
|
t,
|
|
menu = menu,
|
|
caller = caller,
|
|
input_sels_prompt = 'to spend from',
|
|
inputs = acct,
|
|
file_desc = 'transaction',
|
|
bad_input_sels = True,
|
|
interactive_fee = interactive_fee,
|
|
fee_info_pat = fee_info_pat,
|
|
fee_desc = 'transaction fee or gas price',
|
|
add_comment = tx_comment_jp,
|
|
tweaks = tweaks)
|
|
if not no_read:
|
|
t.read()
|
|
return t
|
|
|
|
def txsign(self,ni=False,ext='{}.regtest.rawtx',add_args=[],dev_send=False):
|
|
ext = ext.format('-α' if cfg.debug_utf8 else '')
|
|
keyfile = joinpath(self.tmpdir,parity_devkey_fn)
|
|
txfile = self.get_file_with_ext(ext,no_dot=True)
|
|
t = self.spawn(
|
|
'mmgen-txsign',
|
|
[f'--outdir={self.tmpdir}']
|
|
+ [f'--coin={self.proto.coin}']
|
|
+ ['--quiet']
|
|
+ ['--rpc-host=bad_host'] # ETH signing must work without RPC
|
|
+ add_args
|
|
+ ([],['--yes'])[ni]
|
|
+ ([f'--keys-from-file={keyfile}'] if dev_send else [])
|
|
+ [txfile, dfl_words_file])
|
|
return self.txsign_ui_common(t,ni=ni,has_label=True)
|
|
|
|
def txsend(self,ext='{}.regtest.sigtx',add_args=[]):
|
|
ext = ext.format('-α' if cfg.debug_utf8 else '')
|
|
txfile = self.get_file_with_ext(ext,no_dot=True)
|
|
t = self.spawn('mmgen-txsend', self.eth_args + add_args + [txfile])
|
|
self.txsend_ui_common(
|
|
t,
|
|
quiet = not cfg.debug,
|
|
bogus_send = False,
|
|
has_label = True)
|
|
return t
|
|
|
|
def txview(self,ext_fs):
|
|
ext = ext_fs.format('-α' if cfg.debug_utf8 else '')
|
|
txfile = self.get_file_with_ext(ext,no_dot=True)
|
|
return self.spawn( 'mmgen-tool',['--verbose','txview',txfile] )
|
|
|
|
def fund_dev_address(self):
|
|
"""
|
|
For Erigon, fund the default (Parity) dev address from the Erigon dev address
|
|
For the others, send a junk TX to keep block counts equal for all daemons
|
|
"""
|
|
dt = namedtuple('data',['devkey_fn','dest','amt'])
|
|
d = dt( parity_devkey_fn, burn_addr2, '1' )
|
|
t = self.txcreate(
|
|
args = [
|
|
f'--keys-from-file={joinpath(self.tmpdir,d.devkey_fn)}',
|
|
f'{d.dest},{d.amt}',
|
|
],
|
|
menu = ['a','r'],
|
|
caller = 'txdo',
|
|
acct = '1',
|
|
no_read = True )
|
|
self._do_confirm_send(t,quiet=not cfg.debug,sure=False)
|
|
t.read()
|
|
self.get_file_with_ext('sigtx',delete_all=True)
|
|
return t
|
|
|
|
def txcreate1(self):
|
|
# include one invalid keypress 'X' -- see EthereumTwUnspentOutputs.key_mappings
|
|
menu = ['a','d','r','M','X','e','m','m']
|
|
args = ['98831F3A:E:1,123.456']
|
|
return self.txcreate(args=args,menu=menu,acct='1',tweaks=['confirm_non_mmgen'])
|
|
def txview1_raw(self):
|
|
return self.txview(ext_fs='{}.regtest.rawtx')
|
|
def txsign1(self):
|
|
return self.txsign(add_args=['--use-internal-keccak-module'],dev_send=True)
|
|
def tx_status0_bad(self):
|
|
return self.tx_status(ext='{}.regtest.sigtx',expect_str='neither in mempool nor blockchain',exit_val=1)
|
|
def txsign1_ni(self):
|
|
return self.txsign(ni=True,dev_send=True)
|
|
def txsend1(self):
|
|
return self.txsend()
|
|
def txview1_sig(self): # do after send so that TxID is displayed
|
|
return self.txview(ext_fs='{}.regtest.sigtx')
|
|
def bal1(self):
|
|
return self.bal(n='1')
|
|
|
|
def txcreate2(self):
|
|
args = ['98831F3A:E:11,1.234']
|
|
return self.txcreate(args=args,acct='10',tweaks=['confirm_non_mmgen'])
|
|
def txsign2(self):
|
|
return self.txsign(ni=True,ext='1.234,50000]{}.regtest.rawtx',dev_send=True)
|
|
def txsend2(self):
|
|
return self.txsend(ext='1.234,50000]{}.regtest.sigtx')
|
|
def bal2(self):
|
|
return self.bal(n='2')
|
|
|
|
def txcreate3(self):
|
|
args = ['98831F3A:E:21,2.345']
|
|
return self.txcreate(args=args,acct='10',tweaks=['confirm_non_mmgen'])
|
|
def txsign3(self):
|
|
return self.txsign(ni=True,ext='2.345,50000]{}.regtest.rawtx',dev_send=True)
|
|
def txsend3(self):
|
|
return self.txsend(ext='2.345,50000]{}.regtest.sigtx')
|
|
def bal3(self):
|
|
return self.bal(n='3')
|
|
|
|
def tx_status(self,ext,expect_str,expect_str2='',add_args=[],exit_val=0):
|
|
ext = ext.format('-α' if cfg.debug_utf8 else '')
|
|
txfile = self.get_file_with_ext(ext,no_dot=True)
|
|
t = self.spawn('mmgen-txsend', self.eth_args + add_args + ['--status',txfile])
|
|
t.expect(expect_str)
|
|
if expect_str2:
|
|
t.expect(expect_str2)
|
|
t.req_exit_val = exit_val
|
|
return t
|
|
|
|
def tx_status1(self):
|
|
return self.tx_status(ext='2.345,50000]{}.regtest.sigtx',expect_str='has 1 confirmation')
|
|
|
|
def tx_status1a(self):
|
|
return self.tx_status(ext='2.345,50000]{}.regtest.sigtx',expect_str='has 2 confirmations')
|
|
|
|
async def msgsign_chk(self): # NB: Geth only!
|
|
|
|
def create_signature_mmgen():
|
|
key = self.keystore_data['key']
|
|
imsg(f'Key: {key}')
|
|
from mmgen.proto.eth.misc import ec_sign_message_with_privkey
|
|
return ec_sign_message_with_privkey(cfg,self.message,bytes.fromhex(key),'eth_sign')
|
|
|
|
async def create_signature_rpc():
|
|
addr = self.read_from_tmpfile('signer_addr').strip()
|
|
imsg(f'Address: {addr}')
|
|
rpc = await self.rpc
|
|
return await rpc.call(
|
|
'eth_sign',
|
|
'0x' + addr,
|
|
'0x' + self.message.encode().hex() )
|
|
|
|
if not self.daemon.id == 'geth':
|
|
return 'skip'
|
|
|
|
self.spawn('',msg_only=True)
|
|
|
|
sig = '0x' + create_signature_mmgen()
|
|
sig_chk = await create_signature_rpc()
|
|
|
|
# Compare signatures
|
|
imsg(f'Message: {self.message}')
|
|
imsg(f'Signature: {sig}')
|
|
cmp_or_die(sig,sig_chk,'message signatures')
|
|
imsg('Geth and MMGen signatures match')
|
|
|
|
return 'ok'
|
|
|
|
def msgcreate(self,add_args=[]):
|
|
t = self.spawn('mmgen-msg', self.eth_args_noquiet + add_args + [ 'create', self.message, '98831F3A:E:1' ])
|
|
t.written_to_file('Unsigned message data')
|
|
return t
|
|
|
|
def msgsign(self):
|
|
fn = get_file_with_ext(self.tmpdir,'rawmsg.json')
|
|
t = self.spawn('mmgen-msg', self.eth_args_noquiet + [ 'sign', fn, dfl_words_file ])
|
|
t.written_to_file('Signed message data')
|
|
return t
|
|
|
|
def msgverify(self,fn=None,msghash_type='eth_sign'):
|
|
fn = fn or get_file_with_ext(self.tmpdir,'sigmsg.json')
|
|
t = self.spawn('mmgen-msg', self.eth_args_noquiet + [ 'verify', fn ])
|
|
t.expect(msghash_type)
|
|
t.expect('1 signature verified')
|
|
return t
|
|
|
|
def msgexport(self):
|
|
fn = get_file_with_ext(self.tmpdir,'sigmsg.json')
|
|
t = self.spawn('mmgen-msg', self.eth_args_noquiet + [ 'export', fn ])
|
|
t.written_to_file('Signature data')
|
|
return t
|
|
|
|
def msgverify_export(self):
|
|
return self.msgverify(
|
|
fn = os.path.join(self.tmpdir,'signatures.json') )
|
|
|
|
def msgcreate_raw(self):
|
|
get_file_with_ext(self.tmpdir,'rawmsg.json',delete_all=True)
|
|
return self.msgcreate(add_args=['--msghash-type=raw'])
|
|
|
|
def msgsign_raw(self):
|
|
get_file_with_ext(self.tmpdir,'sigmsg.json',delete_all=True)
|
|
return self.msgsign()
|
|
|
|
def msgverify_raw(self):
|
|
return self.msgverify(msghash_type='raw')
|
|
|
|
def msgexport_raw(self):
|
|
get_file_with_ext(self.tmpdir,'signatures.json',no_dot=True,delete_all=True)
|
|
return self.msgexport()
|
|
|
|
def msgverify_export_raw(self):
|
|
return self.msgverify(
|
|
fn = os.path.join(self.tmpdir,'signatures.json'),
|
|
msghash_type = 'raw' )
|
|
|
|
def txcreate4(self):
|
|
return self.txcreate(
|
|
args = ['98831F3A:E:2,23.45495'],
|
|
acct = '1',
|
|
interactive_fee = '40G',
|
|
fee_info_data = ('0.00084','40') )
|
|
|
|
def txbump(self,ext=',40000]{}.regtest.rawtx',fee='50G',add_args=[]):
|
|
ext = ext.format('-α' if cfg.debug_utf8 else '')
|
|
txfile = self.get_file_with_ext(ext,no_dot=True)
|
|
t = self.spawn('mmgen-txbump', self.eth_args + add_args + ['--yes',txfile])
|
|
t.expect('or gas price: ',fee+'\n')
|
|
return t
|
|
|
|
def txsign4(self):
|
|
return self.txsign(ni=True,ext='.45495,50000]{}.regtest.rawtx',dev_send=True)
|
|
def txsend4(self):
|
|
return self.txsend(ext='.45495,50000]{}.regtest.sigtx')
|
|
def bal4(self):
|
|
return self.bal(n='4')
|
|
|
|
def txcreate5(self):
|
|
args = [burn_addr + ','+amt1]
|
|
return self.txcreate(args=args,acct='10',tweaks=['confirm_non_mmgen'])
|
|
def txsign5(self):
|
|
return self.txsign(ni=True,ext=amt1+',50000]{}.regtest.rawtx',dev_send=True)
|
|
def txsend5(self):
|
|
return self.txsend(ext=amt1+',50000]{}.regtest.sigtx')
|
|
def bal5(self):
|
|
return self.bal(n='5')
|
|
|
|
#bal_corr = Decimal('0.0000032') # gas use for token sends varies between ETH and ETC!
|
|
bal_corr = Decimal('0.0000000') # update: OpenEthereum team seems to have corrected this
|
|
|
|
def bal(self,n):
|
|
t = self.spawn('mmgen-tool', self.eth_args + ['twview','wide=1'])
|
|
text = t.read(strip_color=True)
|
|
for b in bals(n):
|
|
addr,amt,adj = b if len(b) == 3 else b + (False,)
|
|
if adj and self.proto.coin == 'ETC':
|
|
amt = str(Decimal(amt) + Decimal(adj[1]) * self.bal_corr)
|
|
pat = r'\D{}\D.*\D{}\D'.format( addr, amt.replace('.',r'\.') )
|
|
assert re.search(pat,text), pat
|
|
ss = f'Total {self.proto.coin}:'
|
|
assert re.search(ss,text),ss
|
|
return t
|
|
|
|
def token_bal(self,n=None):
|
|
t = self.spawn('mmgen-tool', self.eth_args + ['--token=mm1','twview','wide=1'])
|
|
text = t.read(strip_color=True)
|
|
for b in token_bals(n):
|
|
addr,_amt1,_amt2,adj = b if len(b) == 4 else b + (False,)
|
|
if adj and self.proto.coin == 'ETC':
|
|
_amt2 = str(Decimal(_amt2) + Decimal(adj[1]) * self.bal_corr)
|
|
pat = fr'{addr}\b.*\D{_amt1}\D.*\b{_amt2}\D'
|
|
assert re.search(pat,text), pat
|
|
ss = 'Total MM1:'
|
|
assert re.search(ss,text),ss
|
|
return t
|
|
|
|
def bal_getbalance(self,sid,idx,etc_adj=False,extra_args=[]):
|
|
bal1 = token_bals_getbalance(idx)[0]
|
|
bal2 = token_bals_getbalance(idx)[1]
|
|
bal1 = Decimal(bal1)
|
|
if etc_adj and self.proto.coin == 'ETC':
|
|
bal1 += self.bal_corr
|
|
t = self.spawn('mmgen-tool', self.eth_args + extra_args + ['getbalance'])
|
|
t.expect(rf'{sid}:.*'+str(bal1),regex=True)
|
|
t.expect(r'Non-MMGen:.*'+bal2,regex=True)
|
|
total = strip_ansi_escapes(t.expect_getend(rf'TOTAL {self.proto.coin}')).split()[0]
|
|
assert Decimal(bal1) + Decimal(bal2) == Decimal(total)
|
|
return t
|
|
|
|
def add_comment(self,comment,addr='98831F3A:E:3'):
|
|
t = self.spawn('mmgen-tool', self.eth_args + ['add_label',addr,comment])
|
|
t.expect('Added label.*in tracking wallet',regex=True)
|
|
return t
|
|
|
|
def chk_comment(self,comment_pat,addr='98831F3A:E:3'):
|
|
t = self.spawn('mmgen-tool', self.eth_args + ['listaddresses','all_labels=1'])
|
|
t.expect(fr'{addr}\b.*{comment_pat}',regex=True)
|
|
return t
|
|
|
|
def add_comment1(self):
|
|
return self.add_comment(comment=tw_comment_zh)
|
|
def chk_comment1(self):
|
|
return self.chk_comment(comment_pat=tw_comment_zh[:3])
|
|
def add_comment2(self):
|
|
return self.add_comment(comment=tw_comment_lat_cyr_gr)
|
|
def chk_comment2(self):
|
|
return self.chk_comment(comment_pat=tw_comment_lat_cyr_gr[:3])
|
|
|
|
def remove_comment(self,addr='98831F3A:E:3'):
|
|
t = self.spawn('mmgen-tool', self.eth_args + ['remove_label',addr])
|
|
t.expect('Removed label.*in tracking wallet',regex=True)
|
|
return t
|
|
|
|
def token_compile(self,token_data={}):
|
|
odir = joinpath(self.tmpdir,token_data['symbol'].lower())
|
|
if not self.using_solc:
|
|
imsg(f'Using precompiled contract data in {odir}')
|
|
return 'skip' if os.path.exists(odir) else False
|
|
self.spawn('',msg_only=True)
|
|
cmd_args = [f'--{k}={v}' for k,v in list(token_data.items())]
|
|
imsg("Compiling solidity token contract '{}' with 'solc'".format( token_data['symbol'] ))
|
|
try:
|
|
os.mkdir(odir)
|
|
except:
|
|
pass
|
|
cmd = [
|
|
'python3',
|
|
'scripts/exec_wrapper.py',
|
|
'scripts/create-token.py',
|
|
'--coin=' + self.proto.coin,
|
|
'--outdir=' + odir
|
|
] + cmd_args + [self.proto.checksummed_addr(dfl_devaddr)]
|
|
imsg('Executing: {}'.format( ' '.join(cmd) ))
|
|
cp = run(cmd,stdout=DEVNULL,stderr=PIPE)
|
|
if cp.returncode != 0:
|
|
rmsg('solc failed with the following output:')
|
|
die(2,cp.stderr.decode())
|
|
imsg('ERC20 token {!r} compiled'.format( token_data['symbol'] ))
|
|
return 'ok'
|
|
|
|
def token_compile1(self):
|
|
token_data = { 'name':'MMGen Token 1', 'symbol':'MM1', 'supply':10**26, 'decimals':18 }
|
|
return self.token_compile(token_data)
|
|
|
|
def token_compile2(self):
|
|
token_data = { 'name':'MMGen Token 2', 'symbol':'MM2', 'supply':10**18, 'decimals':10 }
|
|
return self.token_compile(token_data)
|
|
|
|
async def get_tx_receipt(self,txid):
|
|
from mmgen.tx import NewTX
|
|
tx = await NewTX(cfg=cfg,proto=self.proto)
|
|
tx.rpc = await self.rpc
|
|
res = await tx.get_receipt(txid)
|
|
imsg(f'Gas sent: {res.gas_sent.hl():<9} {(res.gas_sent*res.gas_price).hl2(encl="()")}')
|
|
imsg(f'Gas used: {res.gas_used.hl():<9} {(res.gas_used*res.gas_price).hl2(encl="()")}')
|
|
imsg(f'Gas price: {res.gas_price.hl()}')
|
|
if res.gas_used == res.gas_sent:
|
|
omsg(yellow('Warning: all gas was used!'))
|
|
return res
|
|
|
|
async def token_deploy(self,num,key,gas,mmgen_cmd='txdo',tx_fee='8G'):
|
|
keyfile = joinpath(self.tmpdir,parity_devkey_fn)
|
|
fn = joinpath(self.tmpdir,'mm'+str(num),key+'.bin')
|
|
args = [
|
|
'-B',
|
|
f'--fee={tx_fee}',
|
|
f'--gas={gas}',
|
|
f'--contract-data={fn}',
|
|
f'--inputs={dfl_devaddr}',
|
|
'--yes',
|
|
]
|
|
if mmgen_cmd == 'txdo':
|
|
args += ['-k',keyfile]
|
|
t = self.spawn( 'mmgen-'+mmgen_cmd, self.eth_args + args)
|
|
if mmgen_cmd == 'txcreate':
|
|
t.written_to_file('transaction')
|
|
ext = '[0,8000]{}.regtest.rawtx'.format('-α' if cfg.debug_utf8 else '')
|
|
txfile = self.get_file_with_ext(ext,no_dot=True)
|
|
t = self.spawn('mmgen-txsign', self.eth_args + ['--yes','-k',keyfile,txfile],no_msg=True)
|
|
self.txsign_ui_common(t,ni=True)
|
|
txfile = txfile.replace('.rawtx','.sigtx')
|
|
t = self.spawn('mmgen-txsend', self.eth_args + [txfile],no_msg=True)
|
|
|
|
txid = self.txsend_ui_common(t,
|
|
caller = mmgen_cmd,
|
|
quiet = mmgen_cmd == 'txdo' or not cfg.debug,
|
|
bogus_send = False )
|
|
addr = strip_ansi_escapes(t.expect_getend('Contract address: '))
|
|
if (await self.get_tx_receipt(txid)).status == 0:
|
|
die(2,f'Contract {num}:{key} failed to execute. Aborting')
|
|
if key == 'Token':
|
|
self.write_to_tmpfile( f'token_addr{num}', addr+'\n' )
|
|
imsg(f'\nToken MM{num} deployed!')
|
|
return t
|
|
|
|
async def token_deploy1a(self):
|
|
return await self.token_deploy(num=1,key='SafeMath',gas=500_000)
|
|
async def token_deploy1b(self):
|
|
return await self.token_deploy(num=1,key='Owned', gas=1_000_000)
|
|
async def token_deploy1c(self):
|
|
return await self.token_deploy(num=1,key='Token', gas=4_000_000,tx_fee='7G')
|
|
|
|
def tx_status2(self):
|
|
return self.tx_status(
|
|
ext = self.proto.coin+'[0,7000]{}.regtest.sigtx',
|
|
expect_str = 'successfully executed')
|
|
|
|
def bal6(self):
|
|
return self.bal5()
|
|
|
|
async def token_deploy2a(self):
|
|
return await self.token_deploy(num=2,key='SafeMath',gas=500_000)
|
|
async def token_deploy2b(self):
|
|
return await self.token_deploy(num=2,key='Owned', gas=1_000_000)
|
|
async def token_deploy2c(self):
|
|
return await self.token_deploy(num=2,key='Token', gas=4_000_000)
|
|
|
|
async def contract_deploy(self): # test create,sign,send
|
|
return await self.token_deploy(num=2,key='SafeMath',gas=500_000,mmgen_cmd='txcreate')
|
|
|
|
async def token_transfer_ops(self,op,amt=1000):
|
|
self.spawn('',msg_only=True)
|
|
sid = dfl_sid
|
|
from mmgen.tool.wallet import tool_cmd
|
|
usr_mmaddrs = [f'{sid}:E:{i}' for i in (11,21)]
|
|
|
|
from mmgen.proto.eth.contract import ResolvedToken
|
|
async def do_transfer(rpc):
|
|
for i in range(2):
|
|
tk = await ResolvedToken(
|
|
cfg,
|
|
self.proto,
|
|
rpc,
|
|
self.read_from_tmpfile(f'token_addr{i+1}').strip() )
|
|
imsg_r( '\n' + await tk.info() )
|
|
imsg('dev token balance (pre-send): {}'.format( await tk.get_balance(dfl_devaddr) ))
|
|
imsg(f'Sending {amt} {self.proto.dcoin} to address {usr_addrs[i]} ({usr_mmaddrs[i]})')
|
|
txid = await tk.transfer(
|
|
dfl_devaddr,
|
|
usr_addrs[i],
|
|
amt,
|
|
dfl_devkey,
|
|
start_gas = self.proto.coin_amt(60000,'wei'),
|
|
gasPrice = self.proto.coin_amt(8,'Gwei') )
|
|
if self.daemon.id == 'geth': # yet another Geth bug
|
|
await asyncio.sleep(0.5)
|
|
if (await self.get_tx_receipt(txid)).status == 0:
|
|
die(2,'Transfer of token funds failed. Aborting')
|
|
|
|
async def show_bals(rpc):
|
|
for i in range(2):
|
|
tk = await ResolvedToken(
|
|
cfg,
|
|
self.proto,
|
|
rpc,
|
|
self.read_from_tmpfile(f'token_addr{i+1}').strip() )
|
|
imsg('Token: {}'.format( await tk.get_symbol() ))
|
|
imsg(f'dev token balance: {await tk.get_balance(dfl_devaddr)}')
|
|
imsg('usr token balance: {} ({} {})'.format(
|
|
await tk.get_balance(usr_addrs[i]),
|
|
usr_mmaddrs[i],
|
|
usr_addrs[i] ))
|
|
|
|
def gen_addr(addr):
|
|
return tool_cmd(cfg,cmdname='gen_addr',proto=self.proto).gen_addr(addr,dfl_words_file)
|
|
|
|
silence()
|
|
usr_addrs = list(map(gen_addr,usr_mmaddrs))
|
|
if op == 'show_bals':
|
|
await show_bals(await self.rpc)
|
|
elif op == 'do_transfer':
|
|
await do_transfer(await self.rpc)
|
|
end_silence()
|
|
return 'ok'
|
|
|
|
def token_fund_users(self):
|
|
return self.token_transfer_ops(op='do_transfer')
|
|
|
|
def token_user_bals(self):
|
|
return self.token_transfer_ops(op='show_bals')
|
|
|
|
def token_addrgen(self):
|
|
self.addrgen(addrs='11-13')
|
|
ok_msg()
|
|
return self.addrgen(addrs='21-23')
|
|
|
|
def token_addrimport_badaddr1(self):
|
|
t = self.addrimport(ext='[11-13]{}.regtest.addrs',add_args=['--token=abc'],bad_input=True)
|
|
t.expect('could not be resolved')
|
|
t.req_exit_val = 2
|
|
return t
|
|
|
|
def token_addrimport_badaddr2(self):
|
|
t = self.addrimport(ext='[11-13]{}.regtest.addrs',add_args=['--token='+'00deadbeef'*4],bad_input=True)
|
|
t.expect('could not be resolved')
|
|
t.req_exit_val = 2
|
|
return t
|
|
|
|
def token_addrimport(self,addr_file,addr_range,expect,extra_args=[]):
|
|
token_addr = self.read_from_tmpfile(addr_file).strip()
|
|
return self.addrimport(
|
|
ext = f'[{addr_range}]{{}}.regtest.addrs',
|
|
expect = expect,
|
|
add_args = ['--token-addr='+token_addr]+extra_args )
|
|
|
|
def token_addrimport_addr1(self):
|
|
return self.token_addrimport('token_addr1','11-13',expect='3/3')
|
|
|
|
def token_addrimport_addr2(self):
|
|
return self.token_addrimport('token_addr2','21-23',expect='3/3')
|
|
|
|
def token_addrimport_batch(self):
|
|
return self.token_addrimport('token_addr1','11-13',expect='3 addresses',extra_args=['--batch'])
|
|
|
|
def token_addrimport_sym(self):
|
|
return self.addrimport(
|
|
ext = '[21-23]{}.regtest.addrs',
|
|
expect = '3/3',
|
|
add_args = ['--token=MM2'] )
|
|
|
|
def bal7(self):
|
|
return self.bal5()
|
|
def token_bal1(self):
|
|
return self.token_bal(n='1')
|
|
|
|
def token_txcreate(self,args=[],token='',inputs='1',fee='50G'):
|
|
return self.txcreate_ui_common(
|
|
self.spawn('mmgen-txcreate', self.eth_args + ['--token='+token,'-B','--fee='+fee] + args),
|
|
menu = [],
|
|
inputs = inputs,
|
|
input_sels_prompt = 'to spend from',
|
|
add_comment = tx_comment_lat_cyr_gr )
|
|
def token_txsign(self,ext='',token=''):
|
|
return self.txsign(ni=True,ext=ext,add_args=['--token='+token])
|
|
def token_txsend(self,ext='',token=''):
|
|
return self.txsend(ext=ext,add_args=['--token='+token])
|
|
|
|
def token_txcreate1(self):
|
|
return self.token_txcreate(args=['98831F3A:E:12,1.23456'],token='mm1')
|
|
def token_txview1_raw(self):
|
|
return self.txview(ext_fs='1.23456,50000]{}.regtest.rawtx')
|
|
def token_txsign1(self):
|
|
return self.token_txsign(ext='1.23456,50000]{}.regtest.rawtx',token='mm1')
|
|
def token_txsend1(self):
|
|
return self.token_txsend(ext='1.23456,50000]{}.regtest.sigtx',token='mm1')
|
|
def token_txview1_sig(self):
|
|
return self.txview(ext_fs='1.23456,50000]{}.regtest.sigtx')
|
|
|
|
def tx_status3(self):
|
|
return self.tx_status(
|
|
ext = '1.23456,50000]{}.regtest.sigtx',
|
|
add_args = ['--token=mm1'],
|
|
expect_str = 'successfully executed',
|
|
expect_str2 = 'has 1 confirmation' )
|
|
|
|
def token_bal2(self):
|
|
return self.token_bal(n='2')
|
|
|
|
def twview(self,args=[],expect_str='',tool_args=[]):
|
|
t = self.spawn('mmgen-tool', self.eth_args + args + ['twview'] + tool_args)
|
|
if expect_str:
|
|
t.expect(expect_str,regex=True)
|
|
return t
|
|
|
|
def token_txcreate2(self):
|
|
return self.token_txcreate(args=[burn_addr+','+amt2],token='mm1')
|
|
def token_txbump(self):
|
|
return self.txbump(ext=amt2+',50000]{}.regtest.rawtx',fee='56G',add_args=['--token=mm1'])
|
|
def token_txsign2(self):
|
|
return self.token_txsign(ext=amt2+',50000]{}.regtest.rawtx',token='mm1')
|
|
def token_txsend2(self):
|
|
return self.token_txsend(ext=amt2+',50000]{}.regtest.sigtx',token='mm1')
|
|
|
|
def token_bal3(self):
|
|
return self.token_bal(n='3')
|
|
|
|
def del_dev_addr(self):
|
|
t = self.spawn('mmgen-tool', self.eth_args + ['remove_address', dfl_devaddr])
|
|
t.expect(f"'{dfl_devaddr}' deleted")
|
|
return t
|
|
|
|
def bal1_getbalance(self):
|
|
return self.bal_getbalance(dfl_sid,'1',etc_adj=True)
|
|
|
|
def addrimport_token_burn_addr(self):
|
|
return self.addrimport_one_addr(addr=burn_addr,extra_args=['--token=mm1'])
|
|
|
|
def token_bal4(self):
|
|
return self.token_bal(n='4')
|
|
|
|
def token_bal_getbalance(self):
|
|
return self.bal_getbalance(dfl_sid,'2',extra_args=['--token=mm1'])
|
|
|
|
def txcreate_noamt(self):
|
|
return self.txcreate(args=['98831F3A:E:12'])
|
|
def txsign_noamt(self):
|
|
return self.txsign(ext='99.99895,50000]{}.regtest.rawtx')
|
|
def txsend_noamt(self):
|
|
return self.txsend(ext='99.99895,50000]{}.regtest.sigtx')
|
|
|
|
def bal8(self):
|
|
return self.bal(n='8')
|
|
def token_bal5(self):
|
|
return self.token_bal(n='5')
|
|
|
|
def token_txcreate_noamt(self):
|
|
return self.token_txcreate(args=['98831F3A:E:13'],token='mm1',inputs='2',fee='51G')
|
|
def token_txsign_noamt(self):
|
|
return self.token_txsign(ext='1.23456,51000]{}.regtest.rawtx',token='mm1')
|
|
def token_txsend_noamt(self):
|
|
return self.token_txsend(ext='1.23456,51000]{}.regtest.sigtx',token='mm1')
|
|
|
|
def bal9(self):
|
|
return self.bal(n='9')
|
|
def token_bal6(self):
|
|
return self.token_bal(n='6')
|
|
|
|
def listaddresses(self,args=[],tool_args=['all_labels=1']):
|
|
return self.spawn('mmgen-tool', self.eth_args + args + ['listaddresses'] + tool_args)
|
|
|
|
def listaddresses1(self):
|
|
return self.listaddresses()
|
|
def listaddresses2(self):
|
|
return self.listaddresses(tool_args=['minconf=999999999'])
|
|
def listaddresses3(self):
|
|
return self.listaddresses(tool_args=['sort=amt','reverse=1'])
|
|
def listaddresses4(self):
|
|
return self.listaddresses(tool_args=['sort=age','showempty=0'])
|
|
|
|
def token_listaddresses1(self):
|
|
return self.listaddresses(args=['--token=mm1'])
|
|
def token_listaddresses2(self):
|
|
return self.listaddresses(args=['--token=mm1'],tool_args=['showempty=1'])
|
|
def token_listaddresses3(self):
|
|
return self.listaddresses(args=['--token=mm1'],tool_args=['showempty=0'])
|
|
def token_listaddresses4(self):
|
|
return self.listaddresses(args=['--token=mm2'],tool_args=['sort=age','reverse=1'])
|
|
|
|
def twview_cached_balances(self):
|
|
return self.twview(args=['--cached-balances'])
|
|
def token_twview_cached_balances(self):
|
|
return self.twview(args=['--token=mm1','--cached-balances'])
|
|
|
|
def txcreate_cached_balances(self):
|
|
args = ['--fee=20G','--cached-balances','98831F3A:E:3,0.1276']
|
|
return self.txcreate(args=args,acct='2')
|
|
def token_txcreate_cached_balances(self):
|
|
args=['--cached-balances','--fee=12G','98831F3A:E:12,1.2789']
|
|
return self.token_txcreate(args=args,token='mm1')
|
|
|
|
def txdo_cached_balances(
|
|
self,
|
|
acct = '2',
|
|
fee_info_data = ('0.00105','50'),
|
|
add_args = ['98831F3A:E:3,0.4321']):
|
|
t = self.txcreate(
|
|
args = ['--fee=20G','--cached-balances'] + add_args + [dfl_words_file],
|
|
acct = acct,
|
|
caller = 'txdo',
|
|
fee_info_data = fee_info_data,
|
|
no_read = True)
|
|
self._do_confirm_send(t,quiet=not cfg.debug,sure=False)
|
|
return t
|
|
|
|
def txcreate_refresh_balances(self):
|
|
return self._txcreate_refresh_balances(
|
|
bals = ['2','3'],
|
|
args = ['-B','--cached-balances','-i'],
|
|
total = vbal5,
|
|
adj_total = True,
|
|
total_coin = None)
|
|
|
|
def _txcreate_refresh_balances(self,bals,args,total,adj_total,total_coin):
|
|
|
|
if total_coin is None:
|
|
total_coin = self.proto.coin
|
|
|
|
if self.proto.coin == 'ETC' and adj_total:
|
|
total = str(Decimal(total) + self.bal_corr)
|
|
t = self.spawn('mmgen-txcreate', self.eth_args + args)
|
|
for n in bals:
|
|
t.expect('[R]efresh balance:\b','R')
|
|
t.expect(' main menu): ',n+'\n')
|
|
t.expect('Is this what you want? (y/N): ','y')
|
|
t.expect('[R]efresh balance:\b','q')
|
|
t.expect(rf'Total unspent:.*\D{total}\D.*{total_coin}',regex=True)
|
|
return t
|
|
|
|
def bal10(self):
|
|
return self.bal(n='10')
|
|
|
|
def token_txdo_cached_balances(self):
|
|
return self.txdo_cached_balances(
|
|
acct = '1',
|
|
fee_info_data = ('0.0026','50'),
|
|
add_args = ['--token=mm1','98831F3A:E:12,43.21'] )
|
|
|
|
def token_txcreate_refresh_balances(self):
|
|
return self._txcreate_refresh_balances(
|
|
bals = ['1','2'],
|
|
args = ['--token=mm1','-B','--cached-balances','-i'],
|
|
total = '1000',
|
|
adj_total = False,
|
|
total_coin = 'MM1' )
|
|
|
|
def token_bal7(self):
|
|
return self.token_bal(n='7')
|
|
|
|
def twview1(self):
|
|
return self.twview()
|
|
def twview2(self):
|
|
return self.twview(tool_args=['wide=1'])
|
|
def twview3(self):
|
|
return self.twview(tool_args=['wide=1','sort=age'])
|
|
def twview4(self):
|
|
return self.twview(tool_args=['wide=1','minconf=999999999'])
|
|
def twview5(self):
|
|
return self.twview(tool_args=['wide=1','minconf=0'])
|
|
def twview6(self):
|
|
return self.twview(expect_str=vbal7)
|
|
def twview7(self):
|
|
return self.twview(args=['--cached-balances'])
|
|
def twview8(self):
|
|
return self.twview()
|
|
def twview9(self):
|
|
return self.twview(args=['--cached-balances'],expect_str=vbal6)
|
|
|
|
def token_twview1(self):
|
|
return self.twview(args=['--token=mm1'])
|
|
def token_twview2(self):
|
|
return self.twview(args=['--token=mm1'],tool_args=['wide=1'])
|
|
def token_twview3(self):
|
|
return self.twview(args=['--token=mm1'],tool_args=['wide=1','sort=age'])
|
|
|
|
def edit_comment(
|
|
self,
|
|
out_num,
|
|
args = [],
|
|
action = 'l',
|
|
comment_text = None,
|
|
changed = False,
|
|
pexpect_spawn = None):
|
|
t = self.spawn('mmgen-txcreate', self.eth_args + args + ['-B','-i'],pexpect_spawn=pexpect_spawn)
|
|
|
|
menu_prompt = 'efresh balance:\b'
|
|
|
|
t.expect(menu_prompt,'M')
|
|
t.expect(menu_prompt,action)
|
|
t.expect(r'return to main menu): ',out_num+'\n')
|
|
|
|
for p,r in (
|
|
('Enter label text.*: ',comment_text+'\n') if comment_text is not None else (r'\(y/N\): ','y'),
|
|
(r'\(y/N\): ','y') if comment_text == Ctrl_U else (None,None),
|
|
):
|
|
if p:
|
|
t.expect(p,r,regex=True)
|
|
|
|
m = (
|
|
'Label for account #{} edited' if changed else
|
|
'Account #{} removed' if action == 'D' else
|
|
'Label added to account #{}' if comment_text and comment_text != Ctrl_U else
|
|
'Label removed from account #{}' )
|
|
|
|
t.expect(m.format(out_num))
|
|
t.expect(menu_prompt,'M')
|
|
t.expect(menu_prompt,'q')
|
|
|
|
t.expect('Total unspent:')
|
|
|
|
return t
|
|
|
|
def edit_comment1(self):
|
|
return self.edit_comment(out_num=del_addrs[0],comment_text=tw_comment_zh[:3])
|
|
def edit_comment2(self):
|
|
spawn = not sys.platform == 'win32'
|
|
return self.edit_comment(
|
|
out_num = del_addrs[0],
|
|
comment_text = tw_comment_zh[3:],
|
|
changed = True,
|
|
pexpect_spawn = spawn)
|
|
def edit_comment3(self):
|
|
return self.edit_comment(out_num=del_addrs[1],comment_text=tw_comment_lat_cyr_gr)
|
|
def edit_comment4(self):
|
|
if self.skip_for_win():
|
|
return 'skip'
|
|
return self.edit_comment(out_num=del_addrs[0],comment_text=Ctrl_U,pexpect_spawn=True)
|
|
|
|
def token_edit_comment1(self):
|
|
return self.edit_comment(out_num='1',comment_text='Token label #1',args=['--token=mm1'])
|
|
|
|
def remove_addr1(self):
|
|
return self.edit_comment(out_num=del_addrs[0],action='D')
|
|
def remove_addr2(self):
|
|
return self.edit_comment(out_num=del_addrs[1],action='D')
|
|
def token_remove_addr1(self):
|
|
return self.edit_comment(out_num=del_addrs[0],args=['--token=mm1'],action='D')
|
|
def token_remove_addr2(self):
|
|
return self.edit_comment(out_num=del_addrs[1],args=['--token=mm1'],action='D')
|
|
|
|
def twexport_noamt(self):
|
|
return self.twexport(add_args=['include_amts=0'])
|
|
|
|
def twexport(self,add_args=[]):
|
|
t = self.spawn('mmgen-tool', self.eth_args + ['twexport'] + add_args)
|
|
t.written_to_file('JSON data')
|
|
return t
|
|
|
|
async def twmove(self):
|
|
self.spawn('',msg_only=True)
|
|
from mmgen.tw.ctl import TwCtl
|
|
twctl = await TwCtl(cfg,self.proto)
|
|
imsg('Moving tracking wallet')
|
|
bakfile = twctl.tw_fn + '.bak.json'
|
|
if os.path.exists(bakfile):
|
|
os.unlink(bakfile)
|
|
os.rename( twctl.tw_fn, bakfile )
|
|
return 'ok'
|
|
|
|
def twimport(self,add_args=[],expect_str=None):
|
|
from mmgen.tw.json import TwJSON
|
|
fn = joinpath( self.tmpdir, TwJSON.Base(cfg,self.proto).dump_fn )
|
|
t = self.spawn('mmgen-tool',self.eth_args_noquiet + ['twimport',fn] + add_args)
|
|
t.expect('(y/N): ','y')
|
|
if expect_str:
|
|
t.expect(expect_str)
|
|
t.written_to_file('tracking wallet data')
|
|
return t
|
|
|
|
def twimport_nochksum(self):
|
|
return self.twimport(add_args=['ignore_checksum=true'],expect_str='ignoring incorrect checksum')
|
|
|
|
def tw_chktotal(self):
|
|
self.spawn('',msg_only=True)
|
|
from mmgen.tw.json import TwJSON
|
|
fn = joinpath( self.tmpdir, TwJSON.Base(cfg,self.proto).dump_fn )
|
|
res = json.loads(read_from_file(fn))
|
|
cmp_or_die( res['data']['value'], vbal6, 'value in tracking wallet JSON dump' )
|
|
return 'ok'
|
|
|
|
async def twcompare(self):
|
|
self.spawn('',msg_only=True)
|
|
from mmgen.tw.ctl import TwCtl
|
|
twctl = await TwCtl(cfg,self.proto)
|
|
fn = twctl.tw_fn
|
|
imsg('Comparing imported tracking wallet with original')
|
|
data = [json.dumps(json.loads(read_from_file(f)),sort_keys=True) for f in (fn,fn+'.bak.json')]
|
|
cmp_or_die(*data,'tracking wallets')
|
|
return 'ok'
|
|
|
|
def edit_json_twdump(self):
|
|
self.spawn('',msg_only=True)
|
|
from mmgen.tw.json import TwJSON
|
|
fn = TwJSON.Base(cfg,self.proto).dump_fn
|
|
text = json.loads(self.read_from_tmpfile(fn))
|
|
token_addr = self.read_from_tmpfile('token_addr2').strip()
|
|
text['data']['entries']['tokens'][token_addr][2][3] = f'edited comment [фубар] [{gr_uc}]'
|
|
self.write_to_tmpfile( fn, json.dumps(text,indent=4) )
|
|
return 'ok'
|
|
|
|
def stop(self):
|
|
self.spawn('',msg_only=True)
|
|
if not cfg.no_daemon_stop:
|
|
if not stop_test_daemons(self.proto.coin+'_rt'):
|
|
return False
|
|
set_vt100()
|
|
return 'ok'
|