aljaz/python-nostr
1
0
Fork 0
mirror of https://github.com/aljazceru/python-nostr.git synced 2025-12-19 07:14:23 +01:00
Code Issues Packages Projects Releases Wiki Activity

handle unpadding decrypted messages

Browse Source
This commit is contained in:
jeffthibault
2022-07-28 19:45:20 -04:00
parent df9eb450ce
commit 71f6d79599
1 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
nostr/key.py
View File

@@ -30,12 +30,11 @@ def encrypt_message(content: str, shared_secret: str) -> str:
iv = os.urandom(16)
cipher = Cipher(algorithms.AES(bytes.fromhex(shared_secret)), modes.CBC(iv))
encryptor = cipher.encryptor()
padder = padding.PKCS7(128).padder()
padded_data = padder.update(content.encode()) + padder.finalize()
encryptor = cipher.encryptor()
encrypted_message = encryptor.update(padded_data) + encryptor.finalize()
return f"{base64.b64encode(encrypted_message).decode()}?iv={base64.b64encode(iv).decode()}"
def decrypt_message(encoded_message: str, shared_secret: str) -> str:
@@ -49,5 +48,8 @@ def decrypt_message(encoded_message: str, shared_secret: str) -> str:
decryptor = cipher.decryptor()
decrypted_message = decryptor.update(encrypted_content) + decryptor.finalize()
return decrypted_message.decode()
unpadder = padding.PKCS7(128).unpadder()
unpadded_data = unpadder.update(decrypted_message) + unpadder.finalize()
return unpadded_data.decode()