diff --git a/rsa.py b/rsa.py index 320560f..d534067 100755 --- a/rsa.py +++ b/rsa.py @@ -31,9 +31,14 @@ def main(): if sys.argv[1] == "gen" and len(sys.argv) == 4: ##rsa gen keyFileName = sys.argv[3] if keyExist(keyFileName): - ##ask - print("key already exists") - sys.exit(0) + choice = input("overwrite key %s (y/n)" % keyFileName) + if choice == "y": + key = generateKeys(keyFileName, int(sys.argv[2])) + elif choice == "n": + sys.exit(0) + else: + print("unrecognized choice!") + sys.exit(1) else: key = generateKeys(keyFileName, int(sys.argv[2])) print("e: ", key[E])