p15670423
/
monkey
1
0
Fork 5
Code Issues Pull Requests 3 Projects Releases Wiki Activity

Merge pull request #462 from guardicore/460/hotfix/exception-processing-systeminfo-telem 

460/hotfix/exception processing systeminfo telem to develop as well
This commit is contained in:
Shay Nehmad 2019-10-11 21:22:08 +03:00 committed by GitHub
parent 68e2a83aac ef04c341f5
commit 68d9cae8e0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 27 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
monkey/monkey_island/cc/services/telemetry/processing/system_info.py
View File

@ -1,3 +1,5 @@
import logging
from monkey_island.cc.database import mongo from monkey_island.cc.database import mongo
from monkey_island.cc.models import Monkey from monkey_island.cc.models import Monkey
from monkey_island.cc.services import mimikatz_utils from monkey_island.cc.services import mimikatz_utils
@ -7,14 +9,32 @@ from monkey_island.cc.services.telemetry.zero_trust_tests.antivirus_existence im
from monkey_island.cc.services.wmi_handler import WMIHandler from monkey_island.cc.services.wmi_handler import WMIHandler
from monkey_island.cc.encryptor import encryptor from monkey_island.cc.encryptor import encryptor
logger = logging.getLogger(__name__)
def process_system_info_telemetry(telemetry_json): def process_system_info_telemetry(telemetry_json):
process_ssh_info(telemetry_json) telemetry_processing_stages = [
process_credential_info(telemetry_json) process_ssh_info,
process_mimikatz_and_wmi_info(telemetry_json) process_credential_info,
process_aws_data(telemetry_json) process_mimikatz_and_wmi_info,
update_db_with_new_hostname(telemetry_json) process_aws_data,
test_antivirus_existence(telemetry_json) update_db_with_new_hostname,
test_antivirus_existence,
]
# Calling safe_process_telemetry so if one of the stages fail, we log and move on instead of failing the rest of
# them, as they are independent.
for stage in telemetry_processing_stages:
safe_process_telemetry(stage, telemetry_json)
def safe_process_telemetry(processing_function, telemetry_json):
# noinspection PyBroadException
try:
processing_function(telemetry_json)
except Exception as err:
logger.error("Error while in {} stage of processing telemetry.".format(processing_function.func_name),
exc_info=True)
def process_ssh_info(telemetry_json): def process_ssh_info(telemetry_json):
@ -102,4 +122,4 @@ def process_aws_data(telemetry_json):
def update_db_with_new_hostname(telemetry_json): def update_db_with_new_hostname(telemetry_json):
Monkey.get_single_monkey_by_id(telemetry_json['_id']).set_hostname(telemetry_json['data']['hostname']) Monkey.get_single_monkey_by_guid(telemetry_json['monkey_guid']).set_hostname(telemetry_json['data']['hostname'])