def __init__(self, server): self.commands = { "exploit": self.start_module, "message": self.register_module_message, "on_modules_log": self.get_modules_log, "kill_process": self.kill_process, "options": self.get_module_options, "get_args_for_module": self.get_module_args, "get_all_server_data": self.get_all_server_data, "listener_message": self.on_listener_message, "listener_get_options": self.get_listener_options, "gui_command_to_listener": self.gui_command_to_listener, "get_source": self.get_source, "save_source": self.save_source, "generate_report": self.generate_report, "is_listener_connected": self.is_listener_connected } self.server = server self.using_module = "" self.available_modules = self.get_all_modules_paths() self.modules_handler = ModulesHandler(server) self.listener_handler = ListenerHandler(server) self.logger = logging.getLogger() self.options_parser = OptionsParser() self.port_scanner = PortScannerMT.Scanner(4000, 5000) self.report_generator = ReportGenerator()
def __init__(self, server): API.__init__(self) self.commands = self.get_api_functions() self.server = server self.available_modules = self.get_all_modules_paths() self.modules_handler = ModulesHandler(self) self.logger = logging.getLogger() self.options_parser = OptionsParser() self.port_scanner = PortScannerMT.Scanner(4000, 5000) self.report_generator = ReportGenerator() self.service_messages_handler = ServiceMessagesHandler()