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

Bugfix, also change PBA interface

This commit is contained in:
Daniel Goldberg 2019-11-12 19:39:06 +02:00
parent 7f58bd9693
commit e9538e9a42
1 changed files with 4 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
monkey/infection_monkey/post_breach/post_breach_handler.py
View File

@ -1,8 +1,8 @@
import logging import logging
from typing import Sequence
from infection_monkey.utils.environment import is_windows_os from infection_monkey.utils.environment import is_windows_os
from infection_monkey.utils.plugins.load_plugins import get_instances
from infection_monkey.post_breach.pba import PBA from infection_monkey.post_breach.pba import PBA
import infection_monkey.post_breach.actions
LOG = logging.getLogger(__name__) LOG = logging.getLogger(__name__)
@ -33,10 +33,8 @@ class PostBreach(object):
LOG.info("All PBAs executed. Total {} executed.".format(len(self.pba_list))) LOG.info("All PBAs executed. Total {} executed.".format(len(self.pba_list)))
@staticmethod @staticmethod
def config_to_pba_list(): def config_to_pba_list() -> Sequence[PBA]:
""" """
Passes config to each post breach action class and aggregates results into a list.
:return: A list of PBA objects. :return: A list of PBA objects.
""" """
return get_instances(infection_monkey.post_breach.actions.__package__, return PBA.get_instances()
infection_monkey.post_breach.actions.__file__, PBA)