update uint encryption

Author: spencer-xyz

coti/crypto_utils.py

@@ -5,7 +5,6 @@
 from cryptography.hazmat.primitives.asymmetric import padding
 from cryptography.hazmat.primitives.asymmetric import rsa
 from eth_keys import keys
-from math import ceil
 
 block_size = AES.block_size
 address_size = 20
@@ -21,7 +20,6 @@ def encrypt(key, plaintext):
 
     # Ensure key size is 128 bits (16 bytes)
     if len(key) != block_size:
-        print(len(key), block_size)
         raise ValueError("Key size must be 128 bits.")
 
     # Create a new AES cipher block using the provided key
@@ -120,7 +118,10 @@ def build_input_text(plaintext, user_aes_key, sender, contract, function_selecto
     # Convert the ct to an integer
     int_cipher_text = int.from_bytes(ct, byteorder='big')
 
-    return int_cipher_text, signature
+    return {
+        'ciphertext': int_cipher_text,
+        'signature': signature
+    }
 
 
 def build_string_input_text(plaintext, user_aes_key, sender, contract, function_selector, signing_key):
@@ -133,13 +134,12 @@ def build_string_input_text(plaintext, user_aes_key, sender, contract, function_
 
     encoded_plaintext = bytearray(list(plaintext.encode('utf-8')))
 
-    for i in range(ceil(len(encoded_plaintext) / 8)):
-        start_idx = i * 8
+    for start_idx in range(0, len(encoded_plaintext), 8):
         end_idx = min(start_idx + 8, len(encoded_plaintext))
 
         byte_arr = encoded_plaintext[start_idx:end_idx] + bytearray(8 - (end_idx - start_idx))
 
-        ct_int, sig = build_input_text(
+        it_int = build_input_text(
             int.from_bytes(byte_arr, 'big'),
             user_aes_key,
             sender,
@@ -148,8 +148,8 @@ def build_string_input_text(plaintext, user_aes_key, sender, contract, function_
             signing_key
         )
 
-        input_text['ciphertext']['value'].append(ct_int)
-        input_text['signature'].append(sig)
+        input_text['ciphertext']['value'].append(it_int['ciphertext'])
+        input_text['signature'].append(it_int['signature'])
 
     return input_text