p17860254
/
monkey
forked from p15670423/monkey
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

island: Move code for generating reverse schema into functions for better readibility

This commit is contained in:
Shreya Malviya 2021-10-13 18:06:42 +05:30
parent e42a9d8b8f
commit 55fcfa9813
1 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
monkey/monkey_island/cc/services/config_schema/config_schema_per_attack_technique.py
View File

@ -15,6 +15,13 @@ def get_config_schema_per_attack_technique(schema: Dict) -> Dict[str, Dict[str,
""" """
reverse_schema = {} reverse_schema = {}
_crawl_config_schema_definitions_for_reverse_schema(schema, reverse_schema)
_crawl_config_schema_properties_for_reverse_schema(schema, reverse_schema)
return reverse_schema
def _crawl_config_schema_definitions_for_reverse_schema(schema: Dict, reverse_schema: Dict):
definitions = schema["definitions"] definitions = schema["definitions"]
for definition in definitions: for definition in definitions:
definition_type = definitions[definition]["title"] definition_type = definitions[definition]["title"]
@ -25,6 +32,8 @@ def get_config_schema_per_attack_technique(schema: Dict) -> Dict[str, Dict[str,
definition_type, config_field, attack_technique, reverse_schema definition_type, config_field, attack_technique, reverse_schema
) )
def _crawl_config_schema_properties_for_reverse_schema(schema: Dict, reverse_schema: Dict):
properties = schema["properties"] properties = schema["properties"]
for prop in properties: for prop in properties:
property_type = properties[prop]["title"] property_type = properties[prop]["title"]
@ -40,8 +49,6 @@ def get_config_schema_per_attack_technique(schema: Dict) -> Dict[str, Dict[str,
property_type, config_field, attack_technique, reverse_schema property_type, config_field, attack_technique, reverse_schema
) )
return reverse_schema
def _add_config_field_to_reverse_schema( def _add_config_field_to_reverse_schema(
definition_type: str, config_field: str, attack_technique: str, reverse_schema: Dict definition_type: str, config_field: str, attack_technique: str, reverse_schema: Dict