From a9138f37e6053f1a2831c16b85bb1b5f1a50e1a4 Mon Sep 17 00:00:00 2001
From: Kaushal Kumar Agarwal <kaushalag29@gmail.com>
Date: Thu, 21 Jun 2018 16:38:12 +0530
Subject: [PATCH] Update Aes256.py

---
 Aes256.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/Aes256.py b/Aes256.py
index 7c72d44..c722ba8 100644
--- a/Aes256.py
+++ b/Aes256.py
@@ -4,10 +4,10 @@ from Crypto.Cipher import AES
 
 def pad(data):
     padding = 16 - len(data) % 16
-    return data + padding * chr(padding)
+    return data + padding * ' '
 
 def unpad(data):
-    return data[0:-ord(data[-1])]
+    return data[:length]
 
 def keyGen():
     #Generating random key of 32 bytes
@@ -33,6 +33,7 @@ def decryptMsg(ciphertext, key):
     
 
 msg=input('Enter The Message To Be Encrypted : ')
+length=len(msg)
 key = keyGen()
 print("Key generated : "+str(key))
 ciphertext = encryptMsg(msg,key)