rsa.py

@@ -28,7 +28,7 @@ ID=6
 
 def main():
     print("hesham-rsa version ", VERSION)
-    if sys.argv[1] == "gen": ##rsa gen <keysize> <keyname>
+    if sys.argv[1] == "gen" and len(sys.argv) == 4: ##rsa gen <keysize> <keyname>
         keyFileName = sys.argv[3]
         key = generateKeys(keyFileName, int(sys.argv[2]))
         print("e: ", key[E])
@@ -170,11 +170,11 @@ def getPrime(bits):
     while True:
         #Byte order "little" or "big" does not matter here since we want a random number from os.urandom()
         x = int.from_bytes(os.urandom(int(bits/8)), byteOrder)
-        print("trying: ", x, end="")
+        print(x, end="")
         if mr.is_prime(x):
             print("\nprime: ", x)
             return x
-        print("\r",end="")
+        backTrack(x)
 
 
 def isPrime(number):
@@ -344,5 +344,12 @@ def printHelp():
     print("rsa print <key>")
     print("rsa list")
 
+def backTrack(x):
+    #Back track and clear terminal with length of x
+    length = len(str(x))
+    while length > 0:
+        print("\b",end="")
+        length -= 1
+
 if __name__ == "__main__":
     main()