Add files via upload

app/encryption.py

@@ -0,0 +1,59 @@
+from __future__ import annotations
+
+from base64 import b64decode
+from base64 import b64encode
+
+from py3rijndael import Pkcs7Padding
+from py3rijndael import RijndaelCbc
+
+
+def encrypt_score_aes_data(
+    # to encode
+    score_data: list[str],
+    client_hash: str,
+    # used for encoding
+    iv_b64: bytes,
+    osu_version: str,
+) -> tuple[bytes, bytes]:
+    """Encrypt the score data to base64."""
+    # TODO: perhaps this should return TypedDict?
+
+    # attempt to encrypt score data
+    aes = RijndaelCbc(
+        key=f"osu!-scoreburgr---------{osu_version}".encode(),
+        iv=b64decode(iv_b64),
+        padding=Pkcs7Padding(32),
+        block_size=32,
+    )
+
+    score_data_joined = ":".join(score_data)
+    score_data_b64 = b64encode(aes.encrypt(score_data_joined.encode()))
+    client_hash_b64 = b64encode(aes.encrypt(client_hash.encode()))
+
+    return score_data_b64, client_hash_b64
+
+
+def decrypt_score_aes_data(
+    # to decode
+    score_data_b64: bytes,
+    client_hash_b64: bytes,
+    # used for decoding
+    iv_b64: bytes,
+    osu_version: str,
+) -> tuple[list[str], str]:
+    """Decrypt the base64'ed score data."""
+    # TODO: perhaps this should return TypedDict?
+
+    # attempt to decrypt score data
+    aes = RijndaelCbc(
+        key=f"osu!-scoreburgr---------{osu_version}".encode(),
+        iv=b64decode(iv_b64),
+        padding=Pkcs7Padding(32),
+        block_size=32,
+    )
+
+    score_data = aes.decrypt(b64decode(score_data_b64)).decode().split(":")
+    client_hash_decoded = aes.decrypt(b64decode(client_hash_b64)).decode()
+
+    # score data is delimited by colons (:).
+    return score_data, client_hash_decoded