AaronFiore/electrum
1
0
Fork 0
Code Activity

crypto: (trivial) add some type-hints

Browse Source
This commit is contained in:
SomberNight
2024-10-12 03:51:59 +00:00
parent 106a1aca60
commit ffdd45ee6f
1 changed files with 12 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
electrum/crypto.py
View File

@@ -446,7 +446,12 @@ def chacha20_decrypt(*, key: bytes, nonce: bytes, data: bytes) -> bytes:
raise Exception("no chacha20 backend found")
def ecies_encrypt_message(ec_pubkey, message: bytes, *, magic: bytes = b'BIE1') -> bytes:
def ecies_encrypt_message(
ec_pubkey: 'ecc.ECPubkey',
message: bytes,
*,
magic: bytes = b'BIE1',
) -> bytes:
"""
ECIES encryption/decryption methods; AES-128-CBC with PKCS7 is used as the cipher; hmac-sha256 is used as the mac
"""
@@ -462,7 +467,12 @@ def ecies_encrypt_message(ec_pubkey, message: bytes, *, magic: bytes = b'BIE1')
return base64.b64encode(encrypted + mac)
def ecies_decrypt_message(ec_privkey, encrypted: Union[str, bytes], *, magic: bytes=b'BIE1') -> bytes:
def ecies_decrypt_message(
ec_privkey: 'ecc.ECPrivkey',
encrypted: Union[str, bytes],
*,
magic: bytes = b'BIE1',
) -> bytes:
encrypted = base64.b64decode(encrypted) # type: bytes
if len(encrypted) < 85:
raise Exception('invalid ciphertext: length')