Hesham/hesham-rsa
1
0
Fork 0
Code Issues Pull Requests Releases 5 Wiki Activity

Improved interface:

Browse Source 
- Catch wrong number of args.
	- Logic was not correct.
	- Print command template.

Signed-off-by: HeshamTB <hishaminv@gmail.com>
This commit is contained in:
HeshamTB 2020-04-18 14:12:01 +03:00
parent 998089f94c
commit bfc572386e
Signed by: Hesham
GPG Key ID: 74876157D199B09E
1 changed files with 13 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
rsa.py
View File

@ -38,8 +38,8 @@ def main():
except Exception as ex: except Exception as ex:
print(ex) print(ex)
exit(1) exit(1)
if len(sys.argv) == 5 or len(sys.argv) == 4: if sys.argv[1] == "encrypt": ##rsa encrypt <message> <key> <signer>
if sys.argv[1] == "encrypt": ##rsa encrypt <message> <key> <signer> if len(sys.argv) == 5:
msg = sys.argv[2] msg = sys.argv[2]
msg_list = msg.split() msg_list = msg.split()
keyName = sys.argv[3] keyName = sys.argv[3]
@ -52,8 +52,13 @@ def main():
msg_encrypted = msg_encrypted + " " + str(encrypt(word, key_public)) msg_encrypted = msg_encrypted + " " + str(encrypt(word, key_public))
#msg_encrypted = encrypt(msg, key_public) #msg_encrypted = encrypt(msg, key_public)
print("Encrypted msg: \n", msg_encrypted) print("Encrypted msg: \n", msg_encrypted)
print("Signed: \n", sign(msg_encrypted, signing_key)) print("Signed: \n", sign(msg_encrypted, signing_key)) ## Adds an encrypted sig at the end of message.
if sys.argv[1] == "decrypt": ##rsa decrypt <cipher> <key> else:
print("Not enough arguments")
print("rsa encrypt <message> <key> <signer>")
#Make help function
if sys.argv[1] == "decrypt": ##rsa decrypt "<cipher>" <key>
if len(sys.argv) == 4:
cipher = sys.argv[2] cipher = sys.argv[2]
cipher_list = cipher.split() cipher_list = cipher.split()
sig = verify(cipher_list) sig = verify(cipher_list)
@ -62,10 +67,12 @@ def main():
key = readKeyFile(sys.argv[3]) key = readKeyFile(sys.argv[3])
for cipher_word in cipher_list: for cipher_word in cipher_list:
msg_decrypted = msg_decrypted + " " + str(decrypt(int(cipher_word),key[D],key[N])) msg_decrypted = msg_decrypted + " " + str(decrypt(int(cipher_word),key[D],key[N]))
#with open(fileName, "r") as cipherFile:
# cipher = int(cipherFile.readline()) ##one line may make problems later with padding
print("Signed by: ", sig) print("Signed by: ", sig)
print("Decrypted message: \n", msg_decrypted) print("Decrypted message: \n", msg_decrypted)
else:
print("Not enough arguments")
print("rsa decrypt \"<cipher>\" <key>")
if sys.argv[1] == "list": if sys.argv[1] == "list":
listKeys() listKeys()