minor updates

Author: Ekultek

.gitignore

@@ -7,4 +7,3 @@ uid.p
 etc/tokens/*
 autosploit_out/*
 venv/*
-etc/json/*

lib/banner.py

@@ -1,7 +1,7 @@
 import os
 import random
 
-VERSION = "2.2.4"
+VERSION = "2.2.5"
 
 
 def banner_1(line_sep="#--", space=" " * 30):

lib/cmdline/cmd.py

@@ -141,7 +141,25 @@ def single_run_args(opt, keys, loaded_modules):
                     "You should take this ethical lesson into consideration "
                     "before you continue with the use of this tool:\n\n{}\n".format(ethic))
         if opt.downloadModules is not None:
-            print "downloading MODULES!"
+            import re
+
+            modules_to_download = opt.downloadModules
+            links_list = "{}/etc/text_files/links.txt".format(lib.settings.CUR_DIR)
+            possibles = open(links_list).readlines()
+            for module in modules_to_download:
+                searcher = re.compile("{}".format(module))
+                for link in possibles:
+                    if searcher.search(link) is not None:
+                        filename = lib.settings.download_modules(link.strip())
+                        download_filename = "{}.json".format(link.split("/")[-1].split(".")[0])
+                        download_path = "{}/etc/json".format(os.getcwd())
+                        current_files = os.listdir(download_path)
+                        if download_filename not in current_files:
+                            full_path = "{}/{}".format(download_path, download_filename)
+                            lib.jsonize.text_file_to_dict(filename, filename=full_path)
+                            lib.output.info("downloaded into: {}".format(download_path))
+                        else:
+                            lib.output.warning("file already downloaded, skipping")
         if opt.exploitList:
             try:
                 lib.output.info("converting {} to JSON format".format(opt.exploitList))

lib/settings.py

@@ -328,3 +328,23 @@ def save_error_to_file(error_info, error_message, error_class):
             "Traceback (most recent call):\n " + error_info.strip() + "\n{}: {}".format(error_class, error_message)
         )
     return file_path
+
+
+def download_modules(link):
+    import re
+    import requests
+    import tempfile
+
+    lib.output.info('downloading: {}'.format(link))
+    retval = ""
+    req = requests.get(link)
+    content = req.content
+    split_data = content.split(" ")
+    searcher = re.compile("exploit/\w+/\w+")
+    storage_file = tempfile.NamedTemporaryFile(delete=False)
+    for item in split_data:
+        if searcher.search(item) is not None:
+            retval += item + "\n"
+    with open(storage_file.name, 'a+') as tmp:
+        tmp.write(retval)
+    return storage_file.name