def test_read_dictionary_from_file(self):
some_dict = {"a": "b", "c": "d"}
contents = json.dumps(some_dict)
with open(self.wallet_path, "w") as f:
contents = f.write(contents)
storage = WalletStorage(self.wallet_path)
db = WalletDB(storage.read(), manual_upgrades=True)
self.assertEqual("b", db.get("a"))
self.assertEqual("d", db.get("c"))
def run_upgrades(self, storage: WalletStorage, db: 'WalletDB') -> None:
path = storage.path
if db.requires_split():
self.hide()
msg = _(
"The wallet '{}' contains multiple accounts, which are no longer supported since Dash Electrum 2.7.\n\n"
"Do you want to split your wallet into multiple files?"
).format(path)
if not self.question(msg):
return
file_list = db.split_accounts(path)
msg = _('Your accounts have been moved to') + ':\n' + '\n'.join(
file_list) + '\n\n' + _(
'Do you want to delete the old file') + ':\n' + path
if self.question(msg):
os.remove(path)
self.show_warning(_('The file was removed'))
# raise now, to avoid having the old storage opened
raise UserCancelled()
action = db.get_action()
if action and db.requires_upgrade():
raise WalletFileException(
'Incomplete wallet files cannot be upgraded.')
if action:
self.hide()
msg = _("The file '{}' contains an incompletely created wallet.\n"
"Do you want to complete its creation now?").format(path)
if not self.question(msg):
if self.question(
_("Do you want to delete '{}'?").format(path)):
os.remove(path)
self.show_warning(_('The file was removed'))
return
self.show()
self.data = json.loads(storage.read())
self.run(action)
for k, v in self.data.items():
db.put(k, v)
db.write(storage)
return
if db.requires_upgrade():
self.upgrade_db(storage, db)
def test_update_password_with_app_restarts(self):
wallet_str = '{"addr_history":{"Xcmu97gPDoJn6NELShGPRnaRDvbLWyN4ph":[],"Xetp3vzZd25tqMTHUs826okUcpnxAtG73J":[],"XpeViGqbFaUYUQYHeYx62tSBNfrctAu6er":[]},"addresses":{"change":[],"receiving":["Xcmu97gPDoJn6NELShGPRnaRDvbLWyN4ph","XpeViGqbFaUYUQYHeYx62tSBNfrctAu6er","Xetp3vzZd25tqMTHUs826okUcpnxAtG73J"]},"keystore":{"keypairs":{"0344b1588589958b0bcab03435061539e9bcf54677c104904044e4f8901f4ebdf5":"XGw9fNSxGB9e7c9G1wHwMdVsfgvLZGxrUVDEiuozYtEJbQQ8Djc9","0389508c13999d08ffae0f434a085f4185922d64765c0bff2f66e36ad7f745cc5f":"XHLdYVniEEZajZEreWmi6AwfqgzjJwKDAyZATAjVavZZ58gdjf38","04575f52b82f159fa649d2a4c353eb7435f30206f0a6cb9674fbd659f45082c37d559ffd19bea9c0d3b7dcc07a7b79f4cffb76026d5d4dff35341efe99056e22d2":"7ri5P9aS73pjQp62MU6MdhNz2cp6XamakdXtp91aRLEJUtZLd1M"},"type":"imported"},"pruned_txo":{},"seed_version":13,"stored_height":-1,"transactions":{},"tx_fees":{},"txi":{},"txo":{},"use_encryption":false,"verified_tx3":{},"wallet_type":"standard","winpos-qt":[100,100,840,405]}'
db = WalletDB(wallet_str, manual_upgrades=False)
storage = WalletStorage(self.wallet_path)
wallet = Wallet(db, storage, config=self.config)
wallet.stop()
storage = WalletStorage(self.wallet_path)
# if storage.is_encrypted():
# storage.decrypt(password)
db = WalletDB(storage.read(), manual_upgrades=False)
wallet = Wallet(db, storage, config=self.config)
wallet.check_password(None)
wallet.update_password(None, "1234")
with self.assertRaises(InvalidPassword):
wallet.check_password(None)
with self.assertRaises(InvalidPassword):
wallet.check_password("wrong password")
wallet.check_password("1234")
def __init__(self, config: 'SimpleConfig', daemon: 'Daemon',
plugins: 'Plugins'):
self.config = config
self.network = network = daemon.network
if config.get('tor_auto_on', True):
if network:
proxy_modifiable = config.is_modifiable('proxy')
if not proxy_modifiable or not network.detect_tor_proxy():
print(network.TOR_WARN_MSG_TXT)
c = ''
while c != 'y':
c = input("Continue without Tor (y/n)?")
if c == 'n':
exit()
storage = WalletStorage(config.get_wallet_path())
if not storage.file_exists():
print("Wallet not found. try 'electrum-dash create'")
exit()
if storage.is_encrypted():
password = getpass.getpass('Password:'******'')
self.encoding = locale.getpreferredencoding()
self.stdscr = curses.initscr()
curses.noecho()
curses.cbreak()
curses.start_color()
curses.use_default_colors()
curses.init_pair(1, curses.COLOR_WHITE, curses.COLOR_BLUE)
curses.init_pair(2, curses.COLOR_WHITE, curses.COLOR_CYAN)
curses.init_pair(3, curses.COLOR_BLACK, curses.COLOR_WHITE)
self.stdscr.keypad(1)
if getattr(storage, 'backup_message', None):
msg_key = 'Press any key to continue...'
self.stdscr.addstr(f'{storage.backup_message}\n\n{msg_key}')
self.stdscr.getch()
self.stdscr.border(0)
self.maxy, self.maxx = self.stdscr.getmaxyx()
self.set_cursor(0)
self.w = curses.newwin(10, 50, 5, 5)
console_stderr_handler.setLevel(logging.CRITICAL)
self.tab = 0
self.pos = 0
self.popup_pos = 0
self.str_recipient = ""
self.str_description = ""
self.str_amount = ""
self.str_fee = ""
self.history = None
def_dip2 = not self.wallet.psman.unsupported
self.show_dip2 = self.config.get('show_dip2_tx_type', def_dip2)
util.register_callback(self.update,
['wallet_updated', 'network_updated'])
self.tab_names = [
_("History"),
_("Send"),
_("Receive"),
_("Addresses"),
_("Contacts"),
_("Banner")
]
self.num_tabs = len(self.tab_names)
class SignApp(object):
def __init__(self, **kwargs):
'''Get app settings from options'''
self.make_commit = kwargs.pop('make_commit')
self.password = kwargs.pop('password')
self.signing_key = kwargs.pop('signing_key')
self.testnet = kwargs.pop('testnet')
self.wallet_path = kwargs.pop('wallet')
script_config = read_config()
if script_config:
self.make_commit = (self.make_commit
or script_config.get('make_commit', False))
self.signing_key = (self.signing_key
or script_config.get('signing_key', False))
self.testnet = (self.testnet
or script_config.get('testnet', False))
self.wallet_path = (self.wallet_path
or script_config.get('wallet', False))
if self.wallet_path:
self.wallet_path = os.path.expanduser(self.wallet_path)
self.config_options = {'cwd': os.getcwd()}
self.config_options['password'] = self.password
self.config_options['testnet'] = self.testnet
self.config_options['wallet_path'] = self.wallet_path
self.config = SimpleConfig(self.config_options)
if self.config.get('testnet'):
constants.set_testnet()
self.storage = WalletStorage(self.config.get_wallet_path())
def load_wallet(self, storage):
print('Lodaing wallet: %s' % self.config.get_wallet_path())
password = None
if storage.is_encrypted():
if storage.is_encrypted_with_hw_device():
plugins = Plugins(self.config, 'cmdline')
password = get_passwd_for_hw_device_encrypted_storage(plugins)
storage.decrypt(password)
else:
password = get_password(storage.decrypt)
db = WalletDB(self.storage.read(), manual_upgrades=True)
if db.requires_upgrade():
print('Error: Wallet db need to be upgraded.'
' Do it with the wallet app')
sys.exit(1)
self.wallet = Wallet(db, self.storage, config=self.config)
if self.wallet.has_password() and not password:
password = get_password(self.wallet.check_password)
self.config_options['password'] = password
def commit_latest_version(self):
commit_msg = COMMIT_MSG_TEMPLATE.format(fname=LATEST_VER_FNAME,
version=ELECTRUM_VERSION)
print('commiting: %s' % commit_msg)
os.system('git add ./%s' % LATEST_VER_FNAME)
os.system('git commit -m "%s"' % commit_msg)
def run(self):
self.load_wallet(self.storage)
if self.signing_key:
address = self.signing_key
else:
address = SIGNING_KEYS[0]
message = ELECTRUM_VERSION
password = self.config_options.get('password')
print('Signing version: %s' % message)
print('with address: %s' % address)
sig = self.wallet.sign_message(address, message, password)
sig = base64.b64encode(sig).decode('ascii')
content = {
'version': ELECTRUM_VERSION,
'signatures': {
address: sig
}
}
content_json = json.dumps(content, indent=4)
print(content_json)
with open('./%s' % LATEST_VER_FNAME, 'w') as fd:
fd.write('%s\n' % content_json)
if self.make_commit:
self.commit_latest_version()
print(
f"> tested {num_tested} passwords so far... most recently tried: {password!r}"
)
if __name__ == '__main__':
if len(sys.argv) < 2:
print("ERROR. usage: bruteforce_pw.py <path_to_wallet_file>")
sys.exit(1)
path = sys.argv[1]
config = SimpleConfig()
storage = WalletStorage(path)
if not storage.file_exists():
print(f"ERROR. wallet file not found at path: {path}")
sys.exit(1)
if storage.is_encrypted():
test_password = partial(test_password_for_storage_encryption, storage)
print(f"wallet found: with storage encryption.")
else:
db = WalletDB(storage.read(), manual_upgrades=True)
wallet = Wallet(db, storage, config=config)
if not wallet.has_password():
print("wallet found but it is not encrypted.")
sys.exit(0)
test_password = partial(test_password_for_keystore_encryption, wallet)
print(f"wallet found: with keystore encryption.")
password = bruteforce_loop(test_password)
print(f"====================")
print(f"password found: {password}")