fixes the bug in sorting relevant modules (issue #141)

Author: ekultek

autosploit/main.py

@@ -32,6 +32,7 @@ def main():
     try:
         is_admin = os.getuid() == 0
     except AttributeError:
+        # we'll make it cross platform because it seems like a cool idea
         is_admin = ctypes.windll.shell32.IsUserAnAdmin() != 0
 
     if not is_admin:
@@ -87,9 +88,6 @@ def main():
         loaded_tokens = load_api_keys()
         AutoSploitParser().parse_provided(opts)
 
-        # TODO[5] figure out why this isn't used anywhere
-        # maybe we can just remove it, idk
-        loaded_exploits = []
         if not opts.exploitFile:
             misc_info("checking if there are multiple exploit files")
             loaded_exploits = load_exploits(EXPLOIT_FILES_PATH)

lib/exploitation/exploiter.py

@@ -44,10 +44,7 @@ def __init__(self, configuration, all_modules, hosts=None, **kwargs):
         # is run multiple times, it will not recreate the file and instead
         # opens an empty
         self.query = kwargs.get("query", lib.settings.QUERY_FILE_PATH)
-        try:
-            self.query_file = open(self.query).read()
-        except:
-            self.query_file = ""
+        self.query_file = open(self.query).read()
         self.single = kwargs.get("single", None)
         self.ruby_exec = kwargs.get("ruby_exec", False)
         self.msf_path = kwargs.get("msf_path", None)

lib/settings.py

@@ -34,7 +34,7 @@
 RC_SCRIPTS_PATH = "{}/autosploit_out/".format(CUR_DIR)
 
 # path to the file that will contain our query
-# QUERY_FILE_PATH = tempfile.NamedTemporaryFile(delete=False).name
+QUERY_FILE_PATH = tempfile.NamedTemporaryFile(delete=False).name
 
 # default HTTP User-Agent
 DEFAULT_USER_AGENT = "AutoSploit/{} (Language=Python/{}; Platform={})".format(

lib/term/terminal.py

@@ -307,8 +307,15 @@ def __config_headers():
                     elif choice == 2:
                         print(self.sep)
                         query = lib.output.prompt("enter your search query", lowercase=False)
-                        with open(lib.settings.QUERY_FILE_PATH, "a+") as _query:
-                            _query.write(query)
+                        try:
+                            with open(lib.settings.QUERY_FILE_PATH, "w") as _query:
+                                _query.write(query)
+                        except AttributeError:
+                            filename = tempfile.NamedTemporaryFile(delete=False).name
+                            with open(filename, "w") as _query:
+                                _query.write(query)
+                                lib.settings.QUERY_FILE_PATH = filename
+                        print lib.settings.QUERY_FILE_PATH
                         proxy, agent = __config_headers()
                         # possibly needs to change here (see TODO[2])
                         self.gather_hosts(query, proxy=proxy, agent=agent)