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

Agent: Modify collect running processes PBA to return PostBreachData

This commit is contained in:
Shreya Malviya 2022-03-25 20:48:54 +05:30 committed by vakarisz
parent ee24538407
commit 24ba5e37da
1 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -3,6 +3,7 @@ import logging
import psutil import psutil
from common.common_consts.post_breach_consts import POST_BREACH_PROCESS_LIST_COLLECTION from common.common_consts.post_breach_consts import POST_BREACH_PROCESS_LIST_COLLECTION
from infection_monkey.i_puppet.i_puppet import PostBreachData
from infection_monkey.post_breach.pba import PBA from infection_monkey.post_breach.pba import PBA
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
@ -16,9 +17,6 @@ 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)
@ -54,4 +52,5 @@ class ProcessListCollection(PBA):
} }
continue continue
return self.command, (processes, success_state) # No command here; used psutil
return PostBreachData(self.name, "", (processes, success_state))