Check number of args in 'gen' command interface

2020-09-06 06:09:12 +03:00 · 2020-09-06 06:09:12 +03:00 · ea5a9195be
commit ea5a9195be
parent d51a42589c
1 changed files with 1 additions and 1 deletions
--- a/rsa.py
+++ b/rsa.py
@ -29,7 +29,7 @@ ID=6
 def main():
    print("hesham-rsa version ", VERSION)
    if len(sys.argv) > 1:
-        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])