p15670423
/
monkey
1
0
Fork 5
Code Issues Pull Requests 3 Projects Releases Wiki Activity

Comments fixed

This commit is contained in:
VakarisZ 2019-05-08 17:08:48 +03:00
parent 67f8ef4a0a
commit bff95fa470
2 changed files with 19 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
monkey/infection_monkey/post_breach/actions/__init__.py
View File

@ -3,5 +3,9 @@ import glob
def get_pba_files(): def get_pba_files():
"""
Gets all files under current directory(/actions)
:return: list of all files without .py ending
"""
files = glob.glob(join(dirname(__file__), "*.py")) files = glob.glob(join(dirname(__file__), "*.py"))
return [basename(f)[:-3] for f in files if isfile(f) and not f.endswith('__init__.py')] return [basename(f)[:-3] for f in files if isfile(f) and not f.endswith('__init__.py')]

16
monkey/infection_monkey/post_breach/pba.py
View File

@ -34,6 +34,15 @@ class PBA(object):
@staticmethod @staticmethod
def default_get_pba(name, pba_class, linux_cmd="", windows_cmd=""): def default_get_pba(name, pba_class, linux_cmd="", windows_cmd=""):
"""
Default get_pba() method implementation
:param name: PBA name
:param pba_class: class instance. Class's name is matched to config to determine
if corresponding field was enabled in post breach array or not.
:param linux_cmd: commands for linux
:param windows_cmd: commands for windows
:return: post breach action
"""
if pba_class.__name__ in WormConfiguration.post_breach_actions: if pba_class.__name__ in WormConfiguration.post_breach_actions:
command = PBA.choose_command(linux_cmd, windows_cmd) command = PBA.choose_command(linux_cmd, windows_cmd)
if command: if command:
@ -55,7 +64,6 @@ class PBA(object):
def _execute_default(self): def _execute_default(self):
""" """
Default post breach command execution routine Default post breach command execution routine
:param command: What command to execute
:return: Tuple of command's output string and boolean, indicating if it succeeded :return: Tuple of command's output string and boolean, indicating if it succeeded
""" """
try: try:
@ -66,4 +74,10 @@ class PBA(object):
@staticmethod @staticmethod
def choose_command(linux_cmd, windows_cmd): def choose_command(linux_cmd, windows_cmd):
"""
Helper method that chooses between linux and windows commands.
:param linux_cmd:
:param windows_cmd:
:return: Command for current os
"""
return windows_cmd if is_windows_os() else linux_cmd return windows_cmd if is_windows_os() else linux_cmd