Add fn to check if disabled in config

This commit is contained in:
Shreya 2020-07-14 17:42:49 +05:30
parent 2685c50841
commit ce5921eaad
1 changed files with 6 additions and 4 deletions

View File

@ -62,7 +62,7 @@ class AttackTechnique(object, metaclass=abc.ABCMeta):
Gets the status of a certain attack technique.
:return: ScanStatus numeric value
"""
if not cls._is_enabled_in_config():
if cls._is_disabled_in_config():
return ScanStatus.DISABLED.value
elif mongo.db.telemetry.find_one({'telem_category': 'attack',
'data.status': ScanStatus.USED.value,
@ -72,8 +72,6 @@ class AttackTechnique(object, metaclass=abc.ABCMeta):
'data.status': ScanStatus.SCANNED.value,
'data.technique': cls.tech_id}):
return ScanStatus.SCANNED.value
elif not AttackConfig.get_technique_values()[cls.tech_id]:
return ScanStatus.DISABLED.value
else:
return ScanStatus.UNSCANNED.value
@ -145,6 +143,10 @@ class AttackTechnique(object, metaclass=abc.ABCMeta):
@classmethod
def _check_status(cls, status):
if status == ScanStatus.UNSCANNED.value and not AttackConfig.get_technique_values()[cls.tech_id]:
if status == ScanStatus.UNSCANNED.value and cls._is_disabled_in_config():
return ScanStatus.DISABLED.value
return status
@classmethod
def _is_disabled_in_config(cls):
return not AttackConfig.get_technique_values()[cls.tech_id]