def handle_exception(error):
ErrorHandler.handle_error('The server encounter a ValueError', error)
return make_response(
jsonify({
'error':
"The server could not process request, please check your input data"
}), 400)
def handle_exception(error):
ErrorHandler.handle_error('The server encountered an error', error)
return make_response(
jsonify({
'error':
"The server encountered an error and failed to process your request"
}), 400)
def __config_vm_vulnerable_to_vms(self):
"""Configure the vm_vulnerable_to_vms
Using all configurations, map each VM to all VMs that can potentially attack it
"""
app.logger.info(f'initializing the vm_vulnerable_to_vms config')
vm_vulnerable_to_vms = {}
for vm in self.__vms_cfg:
try:
vm_id = vm["vm_id"]
vm_vulnerable_to_vms[vm_id] = set()
source_tags = set()
for tag in vm["tags"]:
if tag in self.dtag_to_stag:
source_tags = source_tags.union(self.dtag_to_stag[tag])
for source_tag in source_tags:
if source_tag in self.tag_to_vmid:
vm_vulnerable_to_vms[vm_id] = vm_vulnerable_to_vms[
vm_id].union(self.tag_to_vmid[source_tag])
except KeyError as error:
ErrorHandler.handle_error(
'KeyError: Error parsing data, skipping value', error)
continue
self.__vm_to_vms = vm_vulnerable_to_vms
app.logger.info(
f'Configured vm_vulnerable_to_vms: {vm_vulnerable_to_vms}')
def __config_tag_to_vmid(self):
"""Configure the tag to vm_ids
Using the VM config, map each tag to all VMs that have it
"""
app.logger.info(f'initializing the tag_to_vmid config')
tag_to_vmid = {}
for vm in self.__vms_cfg:
try:
for tag in vm["tags"]:
if tag in tag_to_vmid:
tag_to_vmid[tag].add(vm["vm_id"])
else:
tag_to_vmid[tag] = {vm["vm_id"]}
except KeyError as error:
ErrorHandler.handle_error(
'KeyError: Error parsing VM config, skipping value', error)
continue
self.tag_to_vmid = tag_to_vmid
app.logger.info(f'Configured tag_to_vmid: {tag_to_vmid}')
def __config_dtag_to_stag(self):
"""Configure the dest_tag to src_tag
Using the FW rules config, map each dest tag to the src tag that can
potentially attack it
"""
app.logger.info(f'initializing the dtag_to_stag config')
dtag_to_stag = {}
for rule in self.__fw_rules:
try:
dest_tag = rule["dest_tag"]
if dest_tag in dtag_to_stag:
dtag_to_stag[dest_tag].add(rule["source_tag"])
else:
dtag_to_stag[dest_tag] = {rule["source_tag"]}
except KeyError as error:
ErrorHandler.handle_error(
'KeyError: Error parsing FW config, skipping value', error)
continue
self.dtag_to_stag = dtag_to_stag
app.logger.info(f'Configured dtag_to_stag: {dtag_to_stag}')
def __init__(self, cloud_env_cfg=None):
app.logger.info(f'initializing the cloud environment')
if cloud_env_cfg:
try:
with open(cloud_env_cfg) as cloud_json_cfg:
cloud_env = json.load(cloud_json_cfg)
self.__vms_cfg = cloud_env["vms"]
self.vm_count = len(self.__vms_cfg)
self.__fw_rules = cloud_env["fw_rules"]
self.__init_cloud_env()
except FileNotFoundError as error:
ErrorHandler.handle_error(
'FileNotFound: Could not load service',
error,
sys_exit=True)
except KeyError as error:
ErrorHandler.handle_error(
'KeyError: Error parsing the config, data may be corrupt',
error,
sys_exit=True)
except Exception as error:
ErrorHandler.handle_error('Could not load service',
error,
sys_exit=True)
