feat: Analyze frequency of a msg with a wordlist

Author: disalad

project/__main__.py

@@ -1,4 +1,4 @@
-from .handlers import bruteforce_handler, decrypt_handler, encrypt_handler
+from .handlers import bruteforce_handler, decrypt_handler, encrypt_handler, freq_analysis_handler
 import argparse
 
 parser = argparse.ArgumentParser(
@@ -25,6 +25,12 @@
 parser.add_argument('-br', '--bruteforce', nargs='+',
                     metavar='PLAIN', help='Cipher text to crack', type=str)
 
+# -fr & --freq-analysis
+parser.add_argument('-fr', '--freq-analysis', nargs='+', metavar='CIPHER', help='Cipher text to crack by analysing', type=str)
+
+# -w & --wordlist
+parser.add_argument('-w', '--wordlist', nargs=1, metavar='WORDLIST', help='Path to the wordlist containing language words', type=str)
+
 args = parser.parse_args()
 
 
@@ -33,6 +39,8 @@ 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")
+        elif args.freq_analysis:
+            freq_analysis_handler(args)
         # If the bruteforce flag used
         elif args.bruteforce:
             bruteforce_handler(args)

project/handlers.py

@@ -1,6 +1,7 @@
 from .crack import bruteforce
-from .utils import write_file
+from .utils import write_file, analyse_text
 from .crypt import encrypt, decrypt
+import re
 
 
 def bruteforce_handler(args):
@@ -62,3 +63,15 @@ def decrypt_handler(args):
         write_file(plain, args.output)
     else:
         print("Decrypted: " + plain)
+
+
+def freq_analysis_handler(args):
+    text = ''
+    a_value = ' '.join(args.freq_analysis)
+    text = re.sub(r'[^A-Za-z0-9 ]+', '', a_value)
+    matches = analyse_text(text, ''.join(args.wordlist))
+    l = len(text.split(' '))
+    avg = float(matches) / len(text.split(' ')) * 100
+    if avg > 80:
+        print('True')
+    print('False')

project/utils.py

@@ -11,3 +11,21 @@ def write_file(text, path):
                 f.write("%s\n" % item)
             return
         f.write(text)
+
+
+def get_wordlist(path):
+    WORDLIST = []
+    words = open(path, 'r').read().split('\n')
+    for word in words:
+        WORDLIST.append(word.upper())
+    return WORDLIST
+
+
+def analyse_text(text, wordlist_path):
+    wordlist = get_wordlist(wordlist_path)
+    text = text.upper()
+    matches = 0
+    for word in text.split(' '):
+        if word in wordlist:
+            matches += 1
+    return matches