feat: bruteforce cipher key

Author: disalad

project/__main__.py

@@ -1,5 +1,6 @@
 from .crypt import encrypt, decrypt
 from .utils import write_file
+from .crack import bruteforce
 import argparse
 
 parser = argparse.ArgumentParser(
@@ -22,13 +23,33 @@
 # -s & --source
 parser.add_argument('-sf', '--sourceFile', action='store_true')
 
+# -b & --brute
+parser.add_argument('-br', '--bruteforce', nargs='+',
+                    metavar='PLAIN', help='Cipher text to crack', type=str)
+
 args = parser.parse_args()
 
 
 def handle_args():
     # If key isn't in the args,
     if not args.key and (args.encrypt or args.decrypt):
         raise NameError("Key is not defined")
+    # If the bruteforce flag used
+    elif args.brute:
+        c_value = ' '.join(args.brute)
+        cracked = []
+        if args.sourceFile:
+            f = open(c_value, 'r').read()
+            cracked = bruteforce(f)
+        else:
+            cracked = bruteforce(c_value)
+        # Write to the file,
+        # If an output path specified
+        if args.output:
+            write_file(cracked, args.output)
+        else:
+            for j in cracked:
+                print(j)
     # If the encrypt flag used
     elif args.encrypt:
         # Extract key as an int from args

project/crack.py

@@ -0,0 +1,7 @@
+from .crypt import ALPHEBET, decrypt
+
+def bruteforce(text):
+    cracked = []
+    for i in range(1, len(ALPHEBET)):
+        cracked.append('Result with key ' + str(i) + ' ' + decrypt(text, i))
+    return cracked

project/utils.py

@@ -1,8 +1,13 @@
 from pathlib import Path
 
+
 def write_file(text, path):
     f = Path(''.join(map(str, path)))
     if f.exists():
         raise OSError("File exists")
     with f.open("w", encoding="utf-8") as f:
+        if type(text) == list:
+            for item in text:
+                f.write("%s\n" % item)
+            return
         f.write(text)