1 files changed, 4 insertions, 11 deletions
diff --git a/final/rsa-decrypt.py b/final/rsa-decrypt.py
index 39a2720..3bc0a06 100644
--- a/final/rsa-decrypt.py
+++ b/final/rsa-decrypt.py
@@ -1,7 +1,8 @@
-encrypted = [340, 316, 145, 278, 250, 435, 321, 109, 115, 490, 156, 212, 122, 288, 287, 164, 225] # Directly from demo output of encrypted
+import rsautils as ru
 p = int(input('prime 1: '))
 q = int(input('prime 2: '))
 n = p*q
+encrypted = ru.int2list(ru.base642int(input('encrypted message:\n')),n)
 tot = (p-1)*(q-1)
 e = int(input('e: '))
 
@@ -15,15 +16,7 @@ def decrypt_block(blk):
   return blk**d % n
 
 decrypted = [decrypt_block(block) for block in encrypted]
-
-messagenum = 0
-for blockind in range(len(decrypted)):
-  block = decrypted[blockind]
-  messagenum += block * n ** blockind
-
-message = ''
-while messagenum > 0:
-  message = chr(messagenum%128) + message
-  messagenum = (messagenum-messagenum%128)//128
+messagenum = ru.list2int(decrypted,n,endian="big")
+message = ru.int2string(messagenum)
 
 print(message)