def wallet_encrypt(ctx):
wallet = Wallet(wallet_path=ctx.obj.wallet_path)
click.echo('Encrypting wallet at {}'.format(wallet.wallet_path))
secret = click.prompt('Enter password', hide_input=True, confirmation_prompt=True)
wallet.encrypt(secret)
wallet.save()
def test_encrypt_save_reload(self):
with set_xrd_dir("no_data"):
wallet = Wallet()
wallet.add_new_address(height=4)
wallet.save()
wallet_b = Wallet()
self.assertEqual(1, len(wallet_b.address_items))
self.assertEqual(wallet.address_items[0],
wallet_b.address_items[0])
TEST_KEY = 'mytestkey'
wallet_b.encrypt(TEST_KEY)
wallet_b.save()
self.assertEqual(1, len(wallet_b.address_items))
self.assertNotEqual(wallet.address_items[0],
wallet_b.address_items[0])
wallet_c = Wallet()
self.assertEqual(1, len(wallet_c.address_items))
self.assertTrue(wallet_c.address_items[0].encrypted)
wallet_c.decrypt(TEST_KEY)
self.assertFalse(wallet_c.address_items[0].encrypted)
self.assertEqual(wallet.address_items[0],
wallet_c.address_items[0])
def test_encrypt_wallet(self):
# 2 unencrypted addresses. This should work.
with set_xrd_dir("no_data"):
TEST_KEY = 'mytestkey'
wallet = Wallet()
wallet.add_new_address(height=4)
wallet.add_new_address(height=4)
wallet.encrypt(TEST_KEY)
self.assertTrue(wallet.encrypted)
self.assertFalse(wallet.encrypted_partially)
def test_decrypt_wallet(self):
with set_xrd_dir("no_data"):
wallet = Wallet()
wallet.add_new_address(height=4)
wallet.add_new_address(height=4)
addresses = wallet.addresses
TEST_KEY = 'mytestkey'
wallet.encrypt(TEST_KEY)
self.assertTrue(wallet.encrypted)
wallet.decrypt(TEST_KEY)
self.assertEqual(addresses, wallet.addresses)
self.assertFalse(wallet.encrypted_partially)
def wallet_gen(ctx, height, hash_function, encrypt):
"""
Generates a new wallet with one address
"""
wallet = Wallet(wallet_path=ctx.obj.wallet_path)
if len(wallet.address_items) > 0:
click.echo("Wallet already exists")
return
wallet.add_new_address(height, hash_function)
_print_addresses(ctx, wallet.address_items, ctx.obj.wallet_path)
if encrypt:
secret = click.prompt('Enter password to encrypt wallet with', hide_input=True, confirmation_prompt=True)
wallet.encrypt(secret)
wallet.save()
def wallet_add(ctx, height, hash_function):
"""
Adds an address or generates a new wallet (working directory)
"""
secret = None
wallet = Wallet(wallet_path=ctx.obj.wallet_path)
wallet_was_encrypted = wallet.encrypted
if wallet.encrypted:
secret = click.prompt('The wallet is encrypted. Enter password', hide_input=True)
wallet.decrypt(secret)
wallet.add_new_address(height, hash_function)
_print_addresses(ctx, wallet.address_items, config.user.wallet_dir)
if wallet_was_encrypted:
wallet.encrypt(secret)
wallet.save()
def test_read_wallet_secure_ver0_saves_wallet_ver1_encrypted(self):
with set_xrd_dir("wallet_secure_ver0"):
wallet = Wallet()
self.assertEqual(wallet.version, 0)
# Wallet will not let you save an encrypted ver0 wallet as ver1. You have to decrypt it first.
# This is because Qaddress is unencrypted in the ver1 wallet.
with self.assertRaises(WalletVersionError):
wallet.save()
wallet.decrypt('test1234')
wallet.encrypt('test1234')
wallet.version = 1
wallet.save()
wallet_reloaded = Wallet()
self.assertEqual(wallet_reloaded.version, 1)
self.assertEqual(
wallet_reloaded.address_items[0].qaddress,
'Q010400d9f1efe5b272e042dcc8ef690f0e90ca8b0b6edba0d26f81e7aff12a6754b21788169f7f'
)
wallet_reloaded.decrypt('test1234')
addr_item = wallet_reloaded.address_items[0]
self.assertEqual(
'Q010400d9f1efe5b272e042dcc8ef690f0e90ca8b0b6edba0d26f81e7aff12a6754b21788169f7f',
addr_item.qaddress)
xmss0 = wallet_reloaded.get_xmss_by_index(0)
self.assertEqual(
'010400d9f1efe5b272e042dcc8ef690f0e90ca8b0b6edba0d26f81e7aff12a6754b21788169f7f',
bin2hstr(xmss0.address))
xmss0b = wallet_reloaded.get_xmss_by_address(xmss0.address)
self.assertEqual(
'010400d9f1efe5b272e042dcc8ef690f0e90ca8b0b6edba0d26f81e7aff12a6754b21788169f7f',
bin2hstr(xmss0b.address))
def test_integrity_behaviour(self):
with set_xrd_dir("no_data"):
TEST_KEY = 'mytestkey'
wallet = Wallet()
wallet.add_new_address(4)
wallet.add_new_address(4)
wallet.encrypt(TEST_KEY)
# An encrypted wallet is not editable.
with self.assertRaises(WalletEncryptionError):
wallet.add_new_address(4)
# You may not re-encrypt it.
with self.assertRaises(WalletEncryptionError):
wallet.encrypt(TEST_KEY)
# You can save it.
wallet.save()
wallet.decrypt_item(1, TEST_KEY)
# A partially encrypted wallet is not editable.
with self.assertRaises(WalletEncryptionError):
wallet.add_new_address(4)
# You may not re-encrypt it.
with self.assertRaises(WalletEncryptionError):
wallet.encrypt(TEST_KEY)
# You may not re-decrypt it.
with self.assertRaises(WalletEncryptionError):
wallet.decrypt(TEST_KEY)
# You can't even save it.
with self.assertRaises(WalletEncryptionError):
wallet.save()
wallet.decrypt_item(0, TEST_KEY)
# A fully decrypted wallet is editable.
wallet.add_new_address(4)
# You may not re-decrypt it.
with self.assertRaises(WalletEncryptionError):
wallet.decrypt(TEST_KEY)
# You can save it.
wallet.save()