diff --git a/rsa.py b/rsa.py index a46d885..320560f 100755 --- a/rsa.py +++ b/rsa.py @@ -30,7 +30,12 @@ def main(): print("hesham-rsa version ", VERSION) if sys.argv[1] == "gen" and len(sys.argv) == 4: ##rsa gen keyFileName = sys.argv[3] - key = generateKeys(keyFileName, int(sys.argv[2])) + if keyExist(keyFileName): + ##ask + print("key already exists") + sys.exit(0) + else: + key = generateKeys(keyFileName, int(sys.argv[2])) print("e: ", key[E]) print("n: ", key[N]) print("d: ", key[D]) @@ -351,5 +356,8 @@ def backTrack(x): print("\b",end="") length -= 1 +def keyExist(keyName): + exist = os.path.exists(keysFolder+keyName) + return exist if __name__ == "__main__": main()