mrtoth/baseimage-tutorial-framework
1
0
Fork 0
mirror of https://github.com/avatao-content/baseimage-tutorial-framework synced 2024-11-22 19:01:33 +00:00
Code Issues Releases Wiki Activity

Refactor message signing and verifying logic

Browse Source
This commit is contained in:
Kristóf Tóth 2018-07-16 10:29:06 +02:00
parent 5770e29733
commit 7c13d31de0
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
lib/tfw/crypto.py
View File

@ -19,16 +19,20 @@ def message_checksum(message):
def sign_message(key, message): def sign_message(key, message):
signature = HMAC(key, message_bytes(message)).signature signature = message_signature(key, message)
message['signature'] = b64encode(signature).decode() message['signature'] = b64encode(signature).decode()
def message_signature(key, message):
return HMAC(key, message_bytes(message)).signature
def verify_message(key, message): def verify_message(key, message):
message = deepcopy(message) message = deepcopy(message)
try: try:
signature_b64 = message.pop('signature') signature_b64 = message.pop('signature')
signature = b64decode(signature_b64) signature = b64decode(signature_b64)
actual_signature = HMAC(key, message_bytes(message)).signature actual_signature = message_signature(key, message)
return signature == actual_signature return signature == actual_signature
except KeyError: except KeyError:
return False return False