LBRYFoundation/LBRY-Vault
1
0
Fork 0
mirror of https://github.com/LBRYFoundation/LBRY-Vault.git synced 2025-09-01 01:35:20 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

pw_decode, pw_encode: separate bytes functions

Browse source
This commit is contained in:
ThomasV 2020-04-03 12:29:55 +02:00
parent 764c18b3c8
commit c8f602c9d7
1 changed files with 14 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

23
electrum/crypto.py
View file

@ -189,22 +189,17 @@ def _hash_password(password: Union[bytes, str], *, version: int) -> bytes:
raise UnexpectedPasswordHashVersion(version) raise UnexpectedPasswordHashVersion(version)
def pw_encode(data: str, password: Union[bytes, str, None], *, version: int) -> str: def pw_encode_bytes(data: bytes, password: Union[bytes, str], *, version: int) -> str:
if not password:
return data
if version not in KNOWN_PW_HASH_VERSIONS: if version not in KNOWN_PW_HASH_VERSIONS:
raise UnexpectedPasswordHashVersion(version) raise UnexpectedPasswordHashVersion(version)
# derive key from password # derive key from password
secret = _hash_password(password, version=version) secret = _hash_password(password, version=version)
# encrypt given data # encrypt given data
ciphertext = EncodeAES_bytes(secret, to_bytes(data, "utf8")) ciphertext = EncodeAES_bytes(secret, data)
ciphertext_b64 = base64.b64encode(ciphertext) ciphertext_b64 = base64.b64encode(ciphertext)
return ciphertext_b64.decode('utf8') return ciphertext_b64.decode('utf8')
def pw_decode_bytes(data: str, password: Union[bytes, str], *, version: int) -> bytes:
def pw_decode(data: str, password: Union[bytes, str, None], *, version: int) -> str:
if password is None:
return data
if version not in KNOWN_PW_HASH_VERSIONS: if version not in KNOWN_PW_HASH_VERSIONS:
raise UnexpectedPasswordHashVersion(version) raise UnexpectedPasswordHashVersion(version)
data_bytes = bytes(base64.b64decode(data)) data_bytes = bytes(base64.b64decode(data))
@ -212,11 +207,21 @@ def pw_decode(data: str, password: Union[bytes, str, None], *, version: int) ->
secret = _hash_password(password, version=version) secret = _hash_password(password, version=version)
# decrypt given data # decrypt given data
try: try:
d = to_string(DecodeAES_bytes(secret, data_bytes), "utf8") d = DecodeAES_bytes(secret, data_bytes)
except Exception as e: except Exception as e:
raise InvalidPassword() from e raise InvalidPassword() from e
return d return d
def pw_encode(data: str, password: Union[bytes, str, None], *, version: int) -> str:
if not password:
return data
return pw_encode_bytes(to_bytes(data, "utf8"), password, version=version)
def pw_decode(data: str, password: Union[bytes, str, None], *, version: int) -> str:
if password is None:
return data
return to_string(pw_decode_bytes(data, password, version=version), "utf8")
def sha256(x: Union[bytes, str]) -> bytes: def sha256(x: Union[bytes, str]) -> bytes:
x = to_bytes(x, 'utf8') x = to_bytes(x, 'utf8')