def setExpose(self, expose=80): if expose == "None": Logger.logError("[ERROR] Not supported expose port!") self.expose = 80 Logger.logInfo("[Info] set expose to {0}!".format(self.expose)) else: self.expose = expose return self.expose
def setExpose(self, expose=80): if expose == "None": Logger.logError("[ERROR] Not supported expose port!") self.expose = 80 Logger.logInfo("[Info] set expose to {0}!".format(self.expose)) else: self.expose = expose return self.expose
def setModules(self, modules="+unfilter"): if modules == "None": Logger.logError("[ERROR] Not supported modules!") self.modules = "+unfilter" Logger.logInfo("[Info] set modules to {0}!".format(self.modules)) else: self.modules = modules return modules
def setModules(self, modules="+unfilter"): if modules == "None": Logger.logError("[ERROR] Not supported modules!") self.modules = "+unfilter" Logger.logInfo("[Info] set modules to {0}!".format(self.modules)) else: self.modules = modules return modules
def setTheme(self, theme="startbootstrap-agency-1.0.6"): if theme == "None": Logger.logError("[ERROR] Not supported theme!") self.theme = "startbootstrap-agency-1.0.6" self.theme_path = os.path.join(THEME_DIR, "themes", self.theme) Logger.logInfo("[Info] set theme to {0}!".format(self.theme)) else: self.theme = theme self.theme_path = os.path.join(THEME_DIR, "themes", self.theme) return self.theme
def setTheme(self, theme="startbootstrap-agency-1.0.6"): if theme == "None": Logger.logError("[ERROR] Not supported theme!") self.theme = "startbootstrap-agency-1.0.6" self.theme_path = os.path.join(THEME_DIR, "themes", self.theme) Logger.logInfo("[Info] set theme to {0}!".format(self.theme)) else: self.theme = theme self.theme_path = os.path.join(THEME_DIR, "themes", self.theme) return self.theme
def start(self): path = self.generate() web.path = path if web.payloads is not None: if self.dbms: if self.dbms == 'Mongo': cmd = "run -id -p {0}:80 -v {1}:{2}:rw -v {3}:/etc/php5/fpm/php.ini:ro -v {4}:/usr/lib/php5/modules/mongodb.so:ro --link {5}:{6} --name VW --workdir {2} {7} ".format( self.expose, web.path, self.mount_point, os.path.join(web.path, 'php.ini'), os.path.join(web.path, 'mongodb.so'), web.container_name, self.dbms.lower(), self.image) else: cmd = "run -id -p {0}:80 -v {1}:{2} -v {3}:/etc/php5/fpm/php.ini --link {4}:{5} --name VW --workdir {2} {6} ".format( self.expose, web.path, self.mount_point, os.path.join(web.path, 'php.ini'), web.container_name, self.dbms.lower(), self.image) if self.command: cmd = cmd + self.command web.dAgent.send(cmd) else: cmd = "run -id -p {0}:80 -v {1}:{2}:rw -v {3}:/etc/php5/fpm/php.ini:ro --name VW --workdir {2} {4} ".format( self.expose, web.path, self.mount_point, os.path.join(web.path, 'php.ini'), self.image) if self.command: cmd = cmd + self.command web.dAgent.send(cmd) web.ctr = web.dAgent.recv() if "cmd" in web.payloads: Logger.logInfo( "[INFO] " + "CMD: {0}".format(web.payloads['cmd'])) web.dAgent.send( "exec {0} -- {1}".format(web.ctr, web.payloads['cmd'])) if "warning" in web.payloads: for warning in web.payloads['warning']: Logger.logWarning("[WARNING] " + warning) if "error" in web.payloads: for error in web.payloads['error']: Logger.logError("[ERROR] " + error) url = ['http', '127.0.0.1:{0}'.format( self.expose), '/', '', '', ''] params = {} if web.payloads['key'] is not None: for index, _ in enumerate(web.payloads['key']): if re.search("page", web.payloads['key'][index], flags=re.IGNORECASE): web.payloads['value'][index] = "index" params.update({'{0}'.format(web.payloads['key'][index]): '{0}'.format( web.payloads['value'][index])}) query = params url[4] = urlencode(query) t = Terminal() with t.location(0, t.height - 1): Logger.logSuccess( t.center(t.blink("Browse: {0}".format(urlparse.urlunparse(url))))) web.dAgent.send("logs {0} -f".format(web.ctr))
def start(self): path = self.generate() web.path = path if web.payloads is not None: web.ctr = web.dAgent.startContainer(image='{0}'.format(self.image), ports=[80], volumes=['{0}'.format(self.mount_point), '/etc/php5/fpm/php.ini'], host_config=web.dAgent.createHostConfig( port_bindings={ 80: self.expose }, binds=self.bindsOperation(), links={'{0}'.format(web.container_name): '{0}'.format( self.dbms.lower())} if self.dbms is not None else None ), name='VW') if "cmd" in web.payloads: Logger.logInfo( "[INFO] " + "CMD: cd {0} && {1}".format(self.mount_point, web.payloads['cmd'])) web.dAgent.execute(web.ctr, web.payloads[ 'cmd'], self.mount_point) if "warning" in web.payloads: for warning in web.payloads['warning']: Logger.logWarning("[WARNING] " + warning) if "error" in web.payloads: for error in web.payloads['error']: Logger.logError("[ERROR] " + error) url = ['http', '{0}:{1}'.format( web.dAgent.host, self.expose), '/', '', '', ''] params = {} if web.payloads['key'] is not None: for index, _ in enumerate(web.payloads['key']): if re.search("page", web.payloads['key'][index], flags=re.IGNORECASE): web.payloads['value'][index] = "index" params.update({'{0}'.format(web.payloads['key'][index]): '{0}'.format( web.payloads['value'][index])}) query = params url[4] = urlencode(query) t = Terminal() with t.location(0, t.height - 1): Logger.logSuccess( t.center(t.blink("Browse: {0}".format(urlparse.urlunparse(url))))) web.dAgent.logs(web.ctr)
def enter_shell(gen_instance): sA = shellAgent() gen_instance.parse("set backend = php") gen_instance.parse("set dbms = None") gen_instance.parse("set theme = startbootstrap-agency-1.0.6") gen_instance.parse("set expose = 80") gen_instance.parse("set modules = +unfilter") Logger.logInfo("VWGen ready (press Ctrl+D to end input)") while True: result = gen_instance.parse(sA.prompt()) if result == "CTRL+D": Logger.logInfo("[INFO] CTRL+D captured. Exit.") raise RuntimeError elif result is not None: Logger.logSuccess(result) else: Logger.logError("Unreconized keyword!")
def enter_shell(gen_instance): sA = shellAgent() gen_instance.parse("set backend = php") gen_instance.parse("set dbms = None") gen_instance.parse("set theme = startbootstrap-agency-1.0.6") gen_instance.parse("set expose = 80") gen_instance.parse("set modules = +unfilter") Logger.logInfo("VWGen ready (press Ctrl+D to end input)") while True: result = gen_instance.parse(sA.prompt()) if result == "CTRL+D": Logger.logInfo("[INFO] CTRL+D captured. Exit.") raise RuntimeError elif result is not None: Logger.logSuccess(result) else: Logger.logError("Unreconized keyword!")
def setBackend(self, backend="php"): if backend == "None": backend = None self.backend = backend for case in switch(self.backend): if case('php'): self.image = 'richarvey/nginx-php-fpm:php5' self.mount_point = '/var/www/html' return self.backend if case('php7'): self.image = 'richarvey/nginx-php-fpm:php7' self.mount_point = '/var/www/html' return self.backend if case(): Logger.logError("[ERROR] Not supported backend!") self.backend = "php" Logger.logInfo( "[Info] set backend to {0}!".format(self.backend)) return self.backend
def setBackend(self, backend="php"): if backend == "None": backend = None self.backend = backend for case in switch(self.backend): if case('php'): self.image = 'richarvey/nginx-php-fpm:php5' self.mount_point = '/var/www/html' return self.backend if case('php7'): self.image = 'richarvey/nginx-php-fpm:php7' self.mount_point = '/var/www/html' return self.backend if case('node'): self.tty = True self.image = 'node:latest' self.mount_point = '/usr/src/app' return self.backend if case(): Logger.logError("[ERROR] Not supported backend!") self.backend = "php" Logger.logInfo("[Info] set backend to {0}!".format( self.backend)) return self.backend
else: if options.inputFile is not None: web.fp.processInputFile(options.inputFile) from core.customization.instanceSample import instanceSample instance = instanceSample(gen) instance.main() web.fp.customizationClass = instanceSample instance.gen_instance.setThemeEnv() web.fp.observer.start() try: instance.gen_instance.start() except (KeyboardInterrupt, SystemExit, RuntimeError): Logger.logInfo("[INFO] See you next time.") except APIError as e: Logger.logError("\n" + "[ERROR] " + str(e.explanation)) Logger.logInfo( "\n[INFO] Taking you to safely leave the program.") finally: web.fp.observer.stop() web.fp.observer.join() web.fp.rmtree(web.fp.path) web.dAgent.removeContainer(web.db_ctr) web.dAgent.removeContainer(web.ctr) web.fp.finishProcessInputFile() raise SystemExit if options.color: gen.setColor()
def cli(options=None): try: if options is None: usage = "usage: %prog [options]" p = optparse.OptionParser(usage=usage, version="VWGen v0.2.0") p.add_option('--console', '-c', action="store_true", metavar='CONSOLE', help="enter console mode") p.add_option('--backend', action="store", dest="backend", type="string", default="php", metavar='BACKEND', help="configure the backend (Default: php)") p.add_option( '--theme', action="store", dest="theme", type="string", default="startbootstrap-agency-1.0.6", metavar='THEME', help= "configure the theme (Default: startbootstrap-agency-1.0.6)") p.add_option( '--expose', action="store", dest="expose", type="int", default=80, metavar='EXPOSE_PORT', help= "configure the port of the host for container binding (Default: 80)" ) p.add_option('--database', '--db', action="store", dest="dbms", type="string", default=None, metavar='DBMS', help="configure the dbms for container linking") p.add_option('--modules', action="store", dest="modules", default="+unfilter", metavar='LIST', help="list of modules to load (Default: +unfilter)") p.add_option('--color', action="store_true", dest="color", help="set terminal color") p.add_option('--verbose', '-v', action="store_true", dest="verbosity", metavar='LEVEL', help="set verbosity level") group = optparse.OptionGroup( p, 'Under development', 'Following options are still in development!') group.add_option( '--file', action="store", dest="inputFile", type="string", default=None, metavar='FILENAME', help="specify the file that VWGen will gonna operate on") p.add_option_group(group) options, _ = p.parse_args() options = vars(options) gen = VWGen() if options["console"]: enter_shell(gen) else: if options["inputFile"] is not None: web.fp.processInputFile(options["inputFile"]) from core.customization.instanceSample import instanceSample instance = instanceSample(gen) instance.main() web.fp.customizationClass = instanceSample instance.gen_instance.setThemeEnv() try: instance.gen_instance.start() except (KeyboardInterrupt, SystemExit, RuntimeError): Logger.logInfo("[INFO] See you next time.") except APIError as e: Logger.logError("\n" + "[ERROR] " + str(e.explanation)) Logger.logInfo( "\n[INFO] Taking you to safely leave the program.") finally: web.fp.rmtree(web.fp.path) if web.db_ctr: web.dAgent.send("rm -f {0}".format(web.db_ctr)) web.dAgent.send("rm -f {0}".format(web.ctr)) web.fp.finishProcessInputFile() raise SystemExit if options["color"]: gen.setColor() if options["verbosity"]: gen.setVerbose() gen.setBackend(options["backend"]) gen.setDbms(options["dbms"]) gen.setTheme(options["theme"]) gen.setExpose(options["expose"]) gen.setModules(options["modules"]) gen.setThemeEnv() try: gen.start() except (KeyboardInterrupt, SystemExit, RuntimeError): Logger.logInfo("[INFO] See you next time.") except APIError as e: Logger.logError("\n" + "[ERROR] " + str(e.explanation)) Logger.logInfo( "\n[INFO] Taking you to safely leave the program.") finally: web.fp.rmtree(web.fp.path) if web.db_ctr: web.dAgent.send("rm -f {0}".format(web.db_ctr)) web.dAgent.send("rm -f {0}".format(web.ctr)) except (KeyboardInterrupt, SystemExit): pass except RuntimeError: if web.db_ctr: web.dAgent.send("rm -f {0}".format(web.db_ctr)) web.dAgent.send("rm -f {0}".format(web.ctr))
def start(self): path = self.generate() web.path = path if web.payloads is not None: if self.dbms: if self.dbms == 'Mongo': cmd = "run -id -p {0}:80 -v {1}:{2}:rw -v {3}:/etc/php5/fpm/php.ini:ro -v {4}:/usr/lib/php5/modules/mongodb.so:ro --link {5}:{6} --name VW --workdir {2} {7} ".format( self.expose, web.path, self.mount_point, os.path.join(web.path, 'php.ini'), os.path.join(web.path, 'mongodb.so'), web.container_name, self.dbms.lower(), self.image) else: cmd = "run -id -p {0}:80 -v {1}:{2} -v {3}:/etc/php5/fpm/php.ini --link {4}:{5} --name VW --workdir {2} {6} ".format( self.expose, web.path, self.mount_point, os.path.join(web.path, 'php.ini'), web.container_name, self.dbms.lower(), self.image) if self.command: cmd = cmd + self.command web.dAgent.send(cmd) else: cmd = "run -id -p {0}:80 -v {1}:{2}:rw -v {3}:/etc/php5/fpm/php.ini:ro --name VW --workdir {2} {4} ".format( self.expose, web.path, self.mount_point, os.path.join(web.path, 'php.ini'), self.image) if self.command: cmd = cmd + self.command web.dAgent.send(cmd) web.ctr = web.dAgent.recv() if "cmd" in web.payloads: Logger.logInfo("[INFO] " + "CMD: {0}".format(web.payloads['cmd'])) web.dAgent.send("exec {0} -- {1}".format( web.ctr, web.payloads['cmd'])) if "warning" in web.payloads: for warning in web.payloads['warning']: Logger.logWarning("[WARNING] " + warning) if "error" in web.payloads: for error in web.payloads['error']: Logger.logError("[ERROR] " + error) url = [ 'http', '127.0.0.1:{0}'.format(self.expose), '/', '', '', '' ] params = {} if web.payloads['key'] is not None: for index, _ in enumerate(web.payloads['key']): if re.search("page", web.payloads['key'][index], flags=re.IGNORECASE): web.payloads['value'][index] = "index" params.update({ '{0}'.format(web.payloads['key'][index]): '{0}'.format(web.payloads['value'][index]) }) query = params url[4] = urlencode(query) t = Terminal() with t.location(0, t.height - 1): Logger.logSuccess( t.center( t.blink("Browse: {0}".format( urlparse.urlunparse(url))))) web.dAgent.send("logs {0} -f".format(web.ctr))
def parse(self, arg): from core.attack import attack arg = arg.strip() try: if arg.startswith("help"): arg = arg[4:].strip() for case in switch(arg): if case('set'): Logger.logSuccess("[*] set A = B") break if case('unset'): Logger.logSuccess("[*] unset A") break if case('show'): Logger.logSuccess("[*] show [modules, themes, infos]") break if case(): Logger.logSuccess("[*] help [set, unset, show]") return True elif arg.startswith("set"): arg = arg[3:].strip() list = re.split("[\s=]+", arg) return getattr(self, ''.join(['set', list[0].capitalize()]))(list[1]) elif arg.startswith("unset"): arg = arg[5:].strip() print arg setattr(self, arg, None) return True elif arg.startswith("show"): arg = arg[4:].strip() for case in switch(arg): if case('modules'): Logger.logSuccess(u"{0}".format(u", ".join( attack.modules))) break if case('themes'): Logger.logSuccess(u"{0}".format(u", ".join( attack.themes))) break if case('infos'): Logger.logSuccess("Backend: {0}".format(self.backend)) Logger.logSuccess("Dbms: {0}".format(self.dbms)) Logger.logSuccess("Theme: {0}".format(self.theme)) Logger.logSuccess("Expose Port: {0}".format( self.expose)) Logger.logSuccess("Color: {0}".format( str(bool(self.color)))) Logger.logSuccess("Verbose: {0}".format( str(bool(self.verbose)))) Logger.logSuccess("Modules: {0}".format(self.modules)) break if case(): Logger.logSuccess("[*] show [modules, themes, infos]") return True elif arg.startswith("start"): gen.setThemeEnv() try: self.start() except (KeyboardInterrupt, SystemExit, RuntimeError): Logger.logInfo("[INFO] See you next time.") except APIError as e: Logger.logError("\n" + "[ERROR] " + str(e.explanation)) Logger.logInfo( "\n[INFO] Taking you to safely leave the program.") finally: self.fp.rmtree(self.fp.path) web.dAgent.send("rm -f {0}".format(web.db_ctr)) web.dAgent.send("rm -f {0}".format(web.ctr)) gen.reset() return True elif arg.startswith("CTRL+D"): return "CTRL+D" except AttributeError as e: Logger.logError(e) return True
def __initAttacks(self): from core.attack import attack Logger.logInfo("[INFO] Loading modules:") Logger.logInfo( u"[INFO] " + "\t {0}".format(u", ".join(attack.modules))) for mod_name in attack.modules: f, filename, description = imp.find_module( mod_name, [os.path.join(ROOT_DIR, 'core', 'attack')]) mod = imp.load_module( mod_name, f, filename, description) mod_instance = getattr(mod, mod_name)(self.fp) self.attacks.append(mod_instance) self.attacks.sort(lambda a, b: a.PRIORITY - b.PRIORITY) # Custom list of modules was specified if self.modules is not None: # First deactivate all modules for attack_module in self.attacks: attack_module.doReturn = False opts = self.modules.split(",") for opt in opts: if opt.strip() == "": continue module = opt # deactivate some modules if module.startswith("-"): module = module[1:] if module == "all": for attack_module in self.attacks: if attack_module.name in attack.modules: attack_module.doReturn = False else: found = False for attack_module in self.attacks: if attack_module.name == module: found = True attack_module.doReturn = False if not found: Logger.logError( "[ERROR] Unable to find a module named {0}".format(module)) # activate some modules else: if module.startswith("+"): module = module[1:] else: module = attack.default if module == "all": Logger.logError( "[ERROR] Keyword 'all' was not safe enough for activating all modules at once. Specify modules names instead.") else: found = False for attack_module in self.attacks: if attack_module.name == module: found = True attack_module.doReturn = True if not found: Logger.logError( "[ERROR] Unable to find a module named {0}".format(module))
def parse(self, arg): from core.attack import attack arg = arg.strip() try: if arg.startswith("help"): arg = arg[4:].strip() for case in switch(arg): if case('set'): Logger.logSuccess("[*] set A = B") break if case('unset'): Logger.logSuccess("[*] unset A") break if case('show'): Logger.logSuccess("[*] show [modules, themes, infos]") break if case(): Logger.logSuccess("[*] help [set, unset, show]") return True elif arg.startswith("set"): arg = arg[3:].strip() list = re.split("[\s=]+", arg) return getattr(self, ''.join(['set', list[0].capitalize()]))(list[1]) elif arg.startswith("unset"): arg = arg[5:].strip() print arg setattr(self, arg, None) return True elif arg.startswith("show"): arg = arg[4:].strip() for case in switch(arg): if case('modules'): Logger.logSuccess(u"{0}".format( u", ".join(attack.modules))) break if case('themes'): Logger.logSuccess(u"{0}".format( u", ".join(attack.themes))) break if case('infos'): Logger.logSuccess("Backend: {0}".format(self.backend)) Logger.logSuccess("Dbms: {0}".format(self.dbms)) Logger.logSuccess("Theme: {0}".format(self.theme)) Logger.logSuccess( "Expose Port: {0}".format(self.expose)) Logger.logSuccess( "Color: {0}".format(str(bool(self.color)))) Logger.logSuccess("Verbose: {0}".format( str(bool(self.verbose)))) Logger.logSuccess("Modules: {0}".format(self.modules)) break if case(): Logger.logSuccess("[*] show [modules, themes, infos]") return True elif arg.startswith("start"): gen.setThemeEnv() try: self.start() except (KeyboardInterrupt, SystemExit, RuntimeError): Logger.logInfo("[INFO] See you next time.") except APIError as e: Logger.logError("\n" + "[ERROR] " + str(e.explanation)) Logger.logInfo( "\n[INFO] Taking you to safely leave the program.") finally: self.fp.rmtree(self.fp.path) web.dAgent.send("rm -f {0}".format(web.db_ctr)) web.dAgent.send("rm -f {0}".format(web.ctr)) gen.reset() return True elif arg.startswith("CTRL+D"): return "CTRL+D" except AttributeError as e: Logger.logError(e) return True
def cli(options=None): try: if options is None: usage = "usage: %prog [options]" p = optparse.OptionParser(usage=usage, version="VWGen v0.2.0") p.add_option('--console', '-c', action="store_true", metavar='CONSOLE', help="enter console mode") p.add_option('--backend', action="store", dest="backend", type="string", default="php", metavar='BACKEND', help="configure the backend (Default: php)") p.add_option('--theme', action="store", dest="theme", type="string", default="startbootstrap-agency-1.0.6", metavar='THEME', help="configure the theme (Default: startbootstrap-agency-1.0.6)") p.add_option('--expose', action="store", dest="expose", type="int", default=80, metavar='EXPOSE_PORT', help="configure the port of the host for container binding (Default: 80)") p.add_option('--database', '--db', action="store", dest="dbms", type="string", default=None, metavar='DBMS', help="configure the dbms for container linking") p.add_option('--modules', action="store", dest="modules", default="+unfilter", metavar='LIST', help="list of modules to load (Default: +unfilter)") p.add_option('--color', action="store_true", dest="color", help="set terminal color") p.add_option('--verbose', '-v', action="store_true", dest="verbosity", metavar='LEVEL', help="set verbosity level") group = optparse.OptionGroup( p, 'Under development', 'Following options are still in development!') group.add_option('--file', action="store", dest="inputFile", type="string", default=None, metavar='FILENAME', help="specify the file that VWGen will gonna operate on") p.add_option_group(group) options, _ = p.parse_args() options = vars(options) gen = VWGen() if options["console"]: enter_shell(gen) else: if options["inputFile"] is not None: web.fp.processInputFile(options["inputFile"]) from core.customization.instanceSample import instanceSample instance = instanceSample(gen) instance.main() web.fp.customizationClass = instanceSample instance.gen_instance.setThemeEnv() try: instance.gen_instance.start() except (KeyboardInterrupt, SystemExit, RuntimeError): Logger.logInfo("[INFO] See you next time.") except APIError as e: Logger.logError("\n" + "[ERROR] " + str(e.explanation)) Logger.logInfo( "\n[INFO] Taking you to safely leave the program.") finally: web.fp.rmtree(web.fp.path) if web.db_ctr: web.dAgent.send("rm -f {0}".format(web.db_ctr)) web.dAgent.send("rm -f {0}".format(web.ctr)) web.fp.finishProcessInputFile() raise SystemExit if options["color"]: gen.setColor() if options["verbosity"]: gen.setVerbose() gen.setBackend(options["backend"]) gen.setDbms(options["dbms"]) gen.setTheme(options["theme"]) gen.setExpose(options["expose"]) gen.setModules(options["modules"]) gen.setThemeEnv() try: gen.start() except (KeyboardInterrupt, SystemExit, RuntimeError): Logger.logInfo("[INFO] See you next time.") except APIError as e: Logger.logError("\n" + "[ERROR] " + str(e.explanation)) Logger.logInfo( "\n[INFO] Taking you to safely leave the program.") finally: web.fp.rmtree(web.fp.path) if web.db_ctr: web.dAgent.send("rm -f {0}".format(web.db_ctr)) web.dAgent.send("rm -f {0}".format(web.ctr)) except (KeyboardInterrupt, SystemExit): pass except RuntimeError: if web.db_ctr: web.dAgent.send("rm -f {0}".format(web.db_ctr)) web.dAgent.send("rm -f {0}".format(web.ctr))
def __initAttacks(self): from core.attack import attack Logger.logInfo("[INFO] Loading modules:") Logger.logInfo(u"[INFO] " + "\t {0}".format(u", ".join(attack.modules))) for mod_name in attack.modules: f, filename, description = imp.find_module( mod_name, [os.path.join(ROOT_DIR, 'core', 'attack')]) mod = imp.load_module(mod_name, f, filename, description) mod_instance = getattr(mod, mod_name)(self.fp) self.attacks.append(mod_instance) self.attacks.sort(lambda a, b: a.PRIORITY - b.PRIORITY) # Custom list of modules was specified if self.modules is not None: # First deactivate all modules for attack_module in self.attacks: attack_module.doReturn = False opts = self.modules.split(",") for opt in opts: if opt.strip() == "": continue module = opt # deactivate some modules if module.startswith("-"): module = module[1:] if module == "all": for attack_module in self.attacks: if attack_module.name in attack.modules: attack_module.doReturn = False else: found = False for attack_module in self.attacks: if attack_module.name == module: found = True attack_module.doReturn = False if not found: Logger.logError( "[ERROR] Unable to find a module named {0}". format(module)) # activate some modules else: if module.startswith("+"): module = module[1:] else: module = attack.default if module == "all": Logger.logError( "[ERROR] Keyword 'all' was not safe enough for activating all modules at once. Specify modules names instead." ) else: found = False for attack_module in self.attacks: if attack_module.name == module: found = True attack_module.doReturn = True if not found: Logger.logError( "[ERROR] Unable to find a module named {0}". format(module))