p20319658
/
monkey
forked from p34709852/monkey
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

* add the right key in telemetry 

* added error handling in report.py
This commit is contained in:
maor.rayzin 2018-11-28 15:30:46 +02:00
parent 0fe7a9c6e1
commit 9d36cf3990
2 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
monkey/monkey_island/cc/resources/telemetry.py
View File

@ -192,7 +192,8 @@ class Telemetry(flask_restful.Resource):
wmi_handler = WMIHandler(monkey_id, telemetry_json['data']['wmi'], users_secrets) wmi_handler = WMIHandler(monkey_id, telemetry_json['data']['wmi'], users_secrets)
wmi_handler.process_and_handle_wmi_info() wmi_handler.process_and_handle_wmi_info()
if 'aws' in telemetry_json['data']: if 'aws' in telemetry_json['data']:
mongo.db.monkey.update_one({'_id': monkey_id}, {'aws_instance_id': telemetry_json['data']['instance-id']}) mongo.db.monkey.update_one({'_id': monkey_id},
{'aws_instance_id': telemetry_json['data']['aws']['instance-id']})
@staticmethod @staticmethod
def add_ip_to_ssh_keys(ip, ssh_info): def add_ip_to_ssh_keys(ip, ssh_info):

6
monkey/monkey_island/cc/services/report.py
View File

@ -553,7 +553,11 @@ class ReportService:
@staticmethod @staticmethod
def get_machine_aws_instance_id(hostname): def get_machine_aws_instance_id(hostname):
return str(list(mongo.db.monkey.find({'hostname': hostname}, {'aws_instance_id': 1}))[0].get('aws_instance_id', None)) aws_instance_id_list = list(mongo.db.monkey.find({'hostname': hostname}, {'aws_instance_id': 1}))
if aws_instance_id_list:
return str(aws_instance_id_list[0].get('aws_instance_id', None))
else:
return None
@staticmethod @staticmethod
def get_issues(): def get_issues():