p41275903
/
monkey
forked from p15670423/monkey
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Island: Modify all telemetry processing functions to accept an extra agent configuration argument

This commit is contained in:
Shreya Malviya 2022-07-14 14:18:38 +05:30
parent c70627c74b
commit 8daa6db81f
7 changed files with 10 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
monkey/monkey_island/cc/services/telemetry/processing/aws_info.py
View File

@ -5,7 +5,7 @@ from monkey_island.cc.models.monkey import Monkey
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
def process_aws_telemetry(telemetry_json): def process_aws_telemetry(telemetry_json, _):
relevant_monkey = Monkey.get_single_monkey_by_guid(telemetry_json["monkey_guid"]) relevant_monkey = Monkey.get_single_monkey_by_guid(telemetry_json["monkey_guid"])
if "instance_id" in telemetry_json["data"]: if "instance_id" in telemetry_json["data"]:

2
monkey/monkey_island/cc/services/telemetry/processing/exploit.py
View File

@ -14,7 +14,7 @@ from monkey_island.cc.services.telemetry.zero_trust_checks.machine_exploited imp
) )
def process_exploit_telemetry(telemetry_json): def process_exploit_telemetry(telemetry_json, _):
encrypt_exploit_creds(telemetry_json) encrypt_exploit_creds(telemetry_json)
edge = get_edge_by_scan_or_exploit_telemetry(telemetry_json) edge = get_edge_by_scan_or_exploit_telemetry(telemetry_json)
update_network_with_exploit(edge, telemetry_json) update_network_with_exploit(edge, telemetry_json)

2
monkey/monkey_island/cc/services/telemetry/processing/post_breach.py
View File

@ -32,7 +32,7 @@ POST_BREACH_TELEMETRY_PROCESSING_FUNCS = {
} }
def process_post_breach_telemetry(telemetry_json): def process_post_breach_telemetry(telemetry_json, _):
def convert_telem_data_to_list(data): def convert_telem_data_to_list(data):
modified_data = [data] modified_data = [data]
if type(data["result"][0]) is list: # multiple results in one pba if type(data["result"][0]) is list: # multiple results in one pba

4
monkey/monkey_island/cc/services/telemetry/processing/processing.py
View File

@ -34,7 +34,9 @@ def process_telemetry(telemetry_json, agent_configuration: AgentConfiguration):
try: try:
telem_category = telemetry_json.get("telem_category") telem_category = telemetry_json.get("telem_category")
if telem_category in TELEMETRY_CATEGORY_TO_PROCESSING_FUNC: if telem_category in TELEMETRY_CATEGORY_TO_PROCESSING_FUNC:
TELEMETRY_CATEGORY_TO_PROCESSING_FUNC[telem_category](telemetry_json) TELEMETRY_CATEGORY_TO_PROCESSING_FUNC[telem_category](
telemetry_json, agent_configuration
)
else: else:
logger.info("Got unknown type of telemetry: %s" % telem_category) logger.info("Got unknown type of telemetry: %s" % telem_category)

2
monkey/monkey_island/cc/services/telemetry/processing/scan.py
View File

@ -14,7 +14,7 @@ from monkey_island.cc.services.telemetry.zero_trust_checks.segmentation import (
) )
def process_scan_telemetry(telemetry_json): def process_scan_telemetry(telemetry_json, _):
if not _host_responded(telemetry_json["data"]["machine"]): if not _host_responded(telemetry_json["data"]["machine"]):
return return

3
monkey/monkey_island/cc/services/telemetry/processing/state.py
View File

@ -1,5 +1,6 @@
import logging import logging
from common.configuration import AgentConfiguration
from monkey_island.cc.models import Monkey from monkey_island.cc.models import Monkey
from monkey_island.cc.services.node import NodeService from monkey_island.cc.services.node import NodeService
from monkey_island.cc.services.telemetry.zero_trust_checks.segmentation import ( from monkey_island.cc.services.telemetry.zero_trust_checks.segmentation import (
@ -9,7 +10,7 @@ from monkey_island.cc.services.telemetry.zero_trust_checks.segmentation import (
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
def process_state_telemetry(telemetry_json): def process_state_telemetry(telemetry_json, agent_configuration: AgentConfiguration):
monkey = NodeService.get_monkey_by_guid(telemetry_json["monkey_guid"]) monkey = NodeService.get_monkey_by_guid(telemetry_json["monkey_guid"])
NodeService.add_communication_info(monkey, telemetry_json["command_control_channel"]) NodeService.add_communication_info(monkey, telemetry_json["command_control_channel"])
if telemetry_json["data"]["done"]: if telemetry_json["data"]["done"]:

2
monkey/monkey_island/cc/services/telemetry/processing/tunnel.py
View File

@ -5,7 +5,7 @@ from monkey_island.cc.services.telemetry.zero_trust_checks.tunneling import (
) )
def process_tunnel_telemetry(telemetry_json): def process_tunnel_telemetry(telemetry_json, _):
check_tunneling_violation(telemetry_json) check_tunneling_violation(telemetry_json)
monkey_id = NodeService.get_monkey_by_guid(telemetry_json["monkey_guid"])["_id"] monkey_id = NodeService.get_monkey_by_guid(telemetry_json["monkey_guid"])["_id"]
if telemetry_json["data"]["proxy"] is not None: if telemetry_json["data"]["proxy"] is not None: