def __init__(self): super(PocsuiteInterpreter, self).__init__() self.current_module = None self.raw_prompt_template = None self.module_prompt_template = None self.prompt_hostname = "Pocsuite3" self.show_sub_commands = ("info", "options", "ip", "all") self.global_commands = sorted( ["use ", "help", "exit", "show ", "search "]) self.module_commands = ["run", "back", "set ", "setg ", "check"] self.module_commands.extend(self.global_commands) self.module_commands.sort() self.modules = index_modules() self.modules_count = len(self.modules) # init conf.console_mode = True banner() logger.info("Load Pocs :{}".format(self.modules_count)) self.last_search = [] self.main_modules_dirs = [] for module in self.modules: temp_module = module if IS_WIN: temp_module = temp_module.replace("/", "\\") temp_module = temp_module.replace(paths.POCSUITE_ROOT_PATH, "").lstrip("\\") temp_module = temp_module.replace(paths.POCSUITE_ROOT_PATH, "").lstrip("/") self.main_modules_dirs.append(temp_module) self.__parse_prompt()
def __init__(self, module_directory=paths.POCSUITE_POCS_PATH): super(PocsuiteInterpreter, self).__init__() self.current_module = None self.raw_prompt_template = None self.module_prompt_template = None self.prompt_hostname = "Pocsuite3" self.show_sub_commands = ("info", "options", "ip", "all") self.global_commands = sorted( ["use ", "help", "exit", "show ", "search ", "clear"]) self.module_commands = ["run", "back", "set ", "setg ", "check"] self.module_commands.extend(self.global_commands) self.module_commands.sort() self.modules = index_modules(module_directory) self.module_parent_directory = os.sep.join( module_directory.rstrip(os.sep).split(os.sep)[0:-1]) + os.sep self.modules_count = len(self.modules) # init conf.console_mode = True banner() logger.info("Load Pocs :{}".format(self.modules_count)) self.last_search = [] self.last_ip = [] self.main_modules_dirs = [] for module in self.modules: temp_module = module temp_module = ltrim(temp_module, self.module_parent_directory).lstrip(os.sep) self.main_modules_dirs.append(temp_module) self.__parse_prompt()