Agent: Modify PBA base class to accept options in its run method
This commit is contained in:
parent
0be6af2d5c
commit
9f8463f707
|
@ -1,6 +1,6 @@
|
|||
import logging
|
||||
import subprocess
|
||||
from typing import Iterable
|
||||
from typing import Dict, Iterable
|
||||
|
||||
from common.utils.attack_utils import ScanStatus
|
||||
from infection_monkey.i_puppet.i_puppet import PostBreachData
|
||||
|
@ -30,7 +30,7 @@ class PBA:
|
|||
self.pba_data = []
|
||||
self.telemetry_messenger = telemetry_messenger
|
||||
|
||||
def run(self) -> Iterable[PostBreachData]:
|
||||
def run(self, options: Dict) -> Iterable[PostBreachData]:
|
||||
"""
|
||||
Runs post breach action command
|
||||
"""
|
||||
|
|
Loading…
Reference in New Issue