forked from p15670423/monkey
Added error suppression where required and documented where it isn't.
This commit is contained in:
parent
177e1ea990
commit
177f902838
|
@ -9,6 +9,7 @@ __author__ = 'itay.mizeretz'
|
|||
class AwsEnvironment(Environment):
|
||||
def __init__(self):
|
||||
super(AwsEnvironment, self).__init__()
|
||||
# Not suppressing error here on purpose. This is critical if we're on AWS env.
|
||||
self.aws_info = AwsInstance()
|
||||
self._instance_id = self._get_instance_id()
|
||||
self.region = self._get_region()
|
||||
|
|
|
@ -1,3 +1,5 @@
|
|||
import logging
|
||||
|
||||
from monkey_island.cc.services.config import ConfigService
|
||||
from common.cloud.aws_instance import AwsInstance
|
||||
from common.cloud.aws_service import AwsService
|
||||
|
@ -7,6 +9,8 @@ from common.cmd.cmd_runner import CmdRunner
|
|||
|
||||
__author__ = "itay.mizeretz"
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
|
||||
class RemoteRunAwsService:
|
||||
aws_instance = None
|
||||
|
@ -23,7 +27,15 @@ class RemoteRunAwsService:
|
|||
:return: None
|
||||
"""
|
||||
if RemoteRunAwsService.aws_instance is None:
|
||||
RemoteRunAwsService.try_init_aws_instance()
|
||||
|
||||
@staticmethod
|
||||
def try_init_aws_instance():
|
||||
# noinspection PyBroadException
|
||||
try:
|
||||
RemoteRunAwsService.aws_instance = AwsInstance()
|
||||
except Exception:
|
||||
logger.error("Failed init aws instance. Exception info: ", exc_info=True)
|
||||
|
||||
@staticmethod
|
||||
def run_aws_monkeys(instances, island_ip):
|
||||
|
|
|
@ -24,6 +24,7 @@ class AWSExporter(Exporter):
|
|||
logger.info('No issues were found by the monkey, no need to send anything')
|
||||
return True
|
||||
|
||||
# Not suppressing error here on purpose.
|
||||
current_aws_region = AwsInstance().get_region()
|
||||
|
||||
for machine in issues_list:
|
||||
|
@ -70,6 +71,7 @@ class AWSExporter(Exporter):
|
|||
configured_product_arn = load_server_configuration_from_file()['aws'].get('sec_hub_product_arn', '')
|
||||
product_arn = 'arn:aws:securityhub:{region}:{arn}'.format(region=region, arn=configured_product_arn)
|
||||
instance_arn = 'arn:aws:ec2:' + str(region) + ':instance:{instance_id}'
|
||||
# Not suppressing error here on purpose.
|
||||
account_id = AwsInstance().get_account_id()
|
||||
logger.debug("aws account id acquired: {}".format(account_id))
|
||||
|
||||
|
|
|
@ -22,5 +22,5 @@ def try_add_aws_exporter_to_manager(manager):
|
|||
RemoteRunAwsService.init()
|
||||
if RemoteRunAwsService.is_running_on_aws() and ('aws' == env.get_deployment()):
|
||||
manager.add_exporter_to_list(AWSExporter)
|
||||
except Exception as err:
|
||||
logger.error("Failed adding aws exporter to manager.", exc_info=True)
|
||||
except Exception:
|
||||
logger.error("Failed adding aws exporter to manager. Exception info:", exc_info=True)
|
||||
|
|
Loading…
Reference in New Issue