if __name__ == '__main__':
while True:
print("Please input command:")
print("0) Generate public and private key")
print("1) Encryption")
print("2) Decryption")
print("3) Exit")
command = input("> ")
if command == "0":
print("Generate 1024 bits key...")
n, p, q, pub_key, pri_key = RSA.KeyGenerator()
print("N: %X" % n)
print("p: %X" % p)
print("q: %X" % q)
print("Public key: %X" % pub_key)
print("Private key: %X" % pri_key)
print("")
elif command == "1":
n = int(input("Input n(base 16): "), 16)
key = int(input("Input key(base 16): "), 16)
message = int(input("Input message(base 16): "), 16)
print("\nResult: %X\n" % RSA.Encryption(n, key, message))
elif command == "2":
n = int(input("Input n(base 16): "), 16)
p = int(input("Input p(base 16): "), 16)
q = int(input("Input q(base 16): "), 16)
key = int(input("Input key(base 16): "), 16)
message = int(input("Input message(base 16): "), 16)
print("\nResult: %X\n" % RSA.DecryptionCRT(n, p, q, key, message))
else:
break
