def island_client(island, quick_performance_tests): client_established = False try: island_client_object = MonkeyIslandClient(island) client_established = island_client_object.get_api_status() except Exception: logging.exception("Got an exception while trying to establish connection to the Island.") finally: if not client_established: pytest.exit("BB tests couldn't establish communication to the island.") if not quick_performance_tests: island_client_object.reset_env() island_client_object.set_scenario(IslandModeEnum.ADVANCED.value) yield island_client_object
def island_client(island): island_client_object = MonkeyIslandClient(island) yield island_client_object
def island_client(island): island_client_object = MonkeyIslandClient(island) island_client_object.reset_env() yield island_client_object
def island_client(island, quick_performance_tests): island_client_object = MonkeyIslandClient(island) if not quick_performance_tests: island_client_object.reset_env() yield island_client_object
from envs.monkey_zoo.blackbox.island_client.monkey_island_client import ( MonkeyIslandClient, ) DST_DIR_NAME = "generated_configs" DST_DIR_PATH = pathlib.Path( pathlib.Path(__file__).parent.absolute(), DST_DIR_NAME) parser = argparse.ArgumentParser(description="Generate config files.") parser.add_argument( "island_ip", metavar="IP:PORT", help="Island endpoint. Example: 123.123.123.123:5000", ) args = parser.parse_args() island_client = MonkeyIslandClient(args.island_ip) CONFIG_TEMPLATES = [ Elastic, Hadoop, Mssql, Performance, ShellShock, SmbMimikatz, SmbPth, Ssh, Struts2, Tunneling, Weblogic, WmiMimikatz, WmiPth,
def get_raw_config(config_template: Type[ConfigTemplate], island_client: MonkeyIslandClient) -> str: response = island_client.get_config() config = IslandConfigParser.apply_template_to_config( config_template, response["configuration"]) return json.dumps(config)