示例#1
0
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
示例#2
0
def island_client(island):
    island_client_object = MonkeyIslandClient(island)
    yield island_client_object
示例#3
0
def island_client(island):
    island_client_object = MonkeyIslandClient(island)
    island_client_object.reset_env()
    yield island_client_object
示例#4
0
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
示例#5
0
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,
示例#6
0
 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)