LBRYCommunity/lbry-sdk
1
0
Fork 1
Code Issues Pull requests Projects Releases 282 Packages Wiki Activity

fix double-double sha

Browse source
This commit is contained in:
Jack Robison 2018-09-21 13:10:16 -04:00 committed by Lex Berezhny
parent 4832526477
commit 9268b6ca13
1 changed files with 7 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
torba/baseaccount.py
View file

@ -291,16 +291,17 @@ class BaseAccount:
def decrypt(self, password):
assert self.encrypted, "Key is not encrypted."
secret = double_sha256(password)
self.seed = aes_decrypt(secret, self.seed)
self.private_key = from_extended_key_string(self.ledger, aes_decrypt(secret, self.private_key))
self.seed = aes_decrypt(password, self.seed.encode()).decode()
self.private_key = from_extended_key_string(
self.ledger, aes_decrypt(password, self.private_key.encode()).decode()
)
self.encrypted = False
def encrypt(self, password):
assert not self.encrypted, "Key is already encrypted."
secret = double_sha256(password)
self.seed = aes_encrypt(secret, self.seed)
self.private_key = aes_encrypt(secret, self.private_key.extended_key_string())
self.seed = aes_encrypt(password, self.seed.encode()).decode()
private_key: PrivateKey = self.private_key
self.private_key = aes_encrypt(password, private_key.extended_key_string().encode()).decode()
self.encrypted = True
@defer.inlineCallbacks