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

Agent: Add TODOs in automated master and process collection list PBA

This commit is contained in:
Shreya Malviya 2022-02-14 17:13:59 +05:30
parent 7cee2e49a2
commit 417f40d62d
2 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
monkey/infection_monkey/master/automated_master.py
View File

@ -176,6 +176,9 @@ class AutomatedMaster(IMaster):
) )
def _run_pba(self, pba: Tuple[str, Dict]): def _run_pba(self, pba: Tuple[str, Dict]):
# TODO: This is the class's name right now. We need `display_name` (see the
# ProcessListCollection PBA). This is shown in the Security report as the PBA
# name and is checked against in the T1082's mongo query in the ATT&CK report.
name = pba[0] name = pba[0]
options = pba[1] options = pba[1]

5
monkey/infection_monkey/post_breach/actions/collect_processes_list.py
View File

@ -16,6 +16,9 @@ except NameError:
class ProcessListCollection(PBA): class ProcessListCollection(PBA):
# TODO: (?) Move all PBA consts into their classes
display_name = POST_BREACH_PROCESS_LIST_COLLECTION
def __init__(self): def __init__(self):
super().__init__(POST_BREACH_PROCESS_LIST_COLLECTION) super().__init__(POST_BREACH_PROCESS_LIST_COLLECTION)
@ -51,4 +54,4 @@ class ProcessListCollection(PBA):
} }
continue continue
return self.command, [str(processes), success_state] return self.command, (str(processes), success_state)