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

Agent: Modify remaining PBAs to yield PostBreachData

This commit is contained in:
Shreya Malviya 2022-03-28 20:26:03 +05:30 committed by vakarisz
parent ec2b2beca5
commit 778f230589
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
monkey/infection_monkey/post_breach/actions/schedule_jobs.py
View File

@ -23,4 +23,4 @@ class ScheduleJobs(PBA):
def run(self): def run(self):
post_breach_data = super(ScheduleJobs, self).run() post_breach_data = super(ScheduleJobs, self).run()
remove_scheduled_jobs() remove_scheduled_jobs()
return post_breach_data yield post_breach_data

2
monkey/infection_monkey/post_breach/actions/use_signed_scripts.py
View File

@ -24,7 +24,7 @@ class SignedScriptProxyExecution(PBA):
original_comspec = subprocess.check_output( # noqa: DUO116 original_comspec = subprocess.check_output( # noqa: DUO116
"if defined COMSPEC echo %COMSPEC%", shell=True "if defined COMSPEC echo %COMSPEC%", shell=True
).decode() ).decode()
return super().run() yield super().run()
except Exception as e: except Exception as e:
logger.warning( logger.warning(
f"An exception occurred on running PBA " f"An exception occurred on running PBA "