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

Agent: Modify ExploitTelem to accept param of type ExploiterResultData

This commit is contained in:
Shreya Malviya 2022-02-22 12:47:42 +05:30
parent 1cce742692
commit afb7210179
3 changed files with 15 additions and 59 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

40
monkey/infection_monkey/master/mock_master.py
View File

@ -101,44 +101,16 @@ class MockMaster(IMaster):
def _exploit(self):
logger.info("Exploiting victims")
(
exploitation_result,
propagation_result,
os,
info,
attempts,
error_message,
) = self._puppet.exploit_host("PowerShellExploiter", "10.0.0.1", {}, None)
logger.info(f"Attempts for exploiting {attempts}")
result = self._puppet.exploit_host("PowerShellExploiter", "10.0.0.1", {}, None)
logger.info(f"Attempts for exploiting {result.attempts}")
self._telemetry_messenger.send_telemetry(
ExploitTelem(
"PowerShellExploiter",
self._hosts["10.0.0.1"],
exploitation_result,
propagation_result,
info,
attempts,
)
ExploitTelem("PowerShellExploiter", self._hosts["10.0.0.1"], result)
)
(
exploitation_result,
propagation_result,
os,
info,
attempts,
error_message,
) = self._puppet.exploit_host("SSHExploiter", "10.0.0.3", {}, None)
logger.info(f"Attempts for exploiting {attempts}")
result = self._puppet.exploit_host("SSHExploiter", "10.0.0.3", {}, None)
logger.info(f"Attempts for exploiting {result.attempts}")
self._telemetry_messenger.send_telemetry(
ExploitTelem(
"SSHExploiter",
self._hosts["10.0.0.3"],
exploitation_result,
propagation_result,
info,
attempts,
)
ExploitTelem("SSHExploiter", self._hosts["10.0.0.3"], result)
)
logger.info("Finished exploiting victims")

11
monkey/infection_monkey/master/propagator.py
View File

@ -165,13 +165,4 @@ class Propagator:
f"{result.error_message}"
)
self._telemetry_messenger.send_telemetry(
ExploitTelem(
exploiter_name,
host,
result.exploitation_success,
result.propagation_success,
result.info,
result.attempts,
)
)
self._telemetry_messenger.send_telemetry(ExploitTelem(exploiter_name, host, result))

23
monkey/infection_monkey/telemetry/exploit_telem.py
View File

@ -1,8 +1,9 @@
from typing import Dict, List
from typing import Dict
from common.common_consts.telem_categories import TelemCategoryEnum
from infection_monkey.model.host import VictimHost
from infection_monkey.telemetry.base_telem import BaseTelem
from monkey.infection_monkey.i_puppet.i_puppet import ExploiterResultData
class ExploitTelem(BaseTelem):
@ -10,30 +11,22 @@ class ExploitTelem(BaseTelem):
self,
name: str,
host: VictimHost,
exploitation_result: bool,
propagation_result: bool,
info: Dict,
attempts: List,
result: ExploiterResultData,
):
"""
Default exploit telemetry constructor
:param name: The name of exploiter used
:param host: The host machine
:param exploitation_result: The result of the exploitation attempt from the 'exploit_host'
method
:param propagation_result: The result of the propagation attempt from the 'exploit_host'
method
:param info: Information about the exploiter
:param attempts: Information about the exploiter's attempts
:param result: Data about the exploitation attempt (success status, info, attempts, etc)
"""
super(ExploitTelem, self).__init__()
self.name = name
self.host = host.__dict__
self.exploitation_result = exploitation_result
self.propagation_result = propagation_result
self.info = info
self.attempts = attempts
self.exploitation_result = result.exploitation_success
self.propagation_result = result.propagation_success
self.info = result.info
self.attempts = result.attempts
telem_category = TelemCategoryEnum.EXPLOIT