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

Agent: Modify clear command history PBA to return PostBreachData

This commit is contained in:
Shreya Malviya 2022-03-25 14:21:15 +05:30 committed by vakarisz
parent 2e48d9ead9
commit ee24538407
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1,11 +1,11 @@
import subprocess import subprocess
from common.common_consts.post_breach_consts import POST_BREACH_CLEAR_CMD_HISTORY from common.common_consts.post_breach_consts import POST_BREACH_CLEAR_CMD_HISTORY
from infection_monkey.i_puppet.i_puppet import PostBreachData
from infection_monkey.post_breach.clear_command_history.clear_command_history import ( from infection_monkey.post_breach.clear_command_history.clear_command_history import (
get_commands_to_clear_command_history, get_commands_to_clear_command_history,
) )
from infection_monkey.post_breach.pba import PBA from infection_monkey.post_breach.pba import PBA
from infection_monkey.telemetry.post_breach_telem import PostBreachTelem
class ClearCommandHistory(PBA): class ClearCommandHistory(PBA):
@ -15,7 +15,8 @@ class ClearCommandHistory(PBA):
def run(self): def run(self):
results = [pba.run() for pba in self.clear_command_history_PBA_list()] results = [pba.run() for pba in self.clear_command_history_PBA_list()]
if results: if results:
PostBreachTelem(self, results).send() # Note: `self.command` is empty here
return PostBreachData(self.name, self.command, results)
def clear_command_history_PBA_list(self): def clear_command_history_PBA_list(self):
return self.CommandHistoryPBAGenerator().get_clear_command_history_pbas() return self.CommandHistoryPBAGenerator().get_clear_command_history_pbas()