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

Agent: Add docstrings to notify_relay_on_propagation

This commit is contained in:
Mike Salvatore 2022-10-03 13:24:09 -04:00
parent d3ff56138f
commit a8383f4a79
1 changed files with 12 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
monkey/infection_monkey/agent_event_handlers/notify_relay_on_propagation.py
View File

@ -8,10 +8,22 @@ logger = logging.getLogger(__name__)
class notify_relay_on_propagation: class notify_relay_on_propagation:
"""
Notifies a TCPRelay of potential relay users if propagation is successful
"""
def __init__(self, tcp_relay: Optional[TCPRelay]): def __init__(self, tcp_relay: Optional[TCPRelay]):
"""
:param tcp_relay: A TCPRelay to notify on successful propagation
"""
self._tcp_relay = tcp_relay self._tcp_relay = tcp_relay
def __call__(self, event: PropagationEvent): def __call__(self, event: PropagationEvent):
"""
Notify a TCPRelay of potential relay users if propagation is successful
:param event: A `PropagationEvent`
"""
if self._tcp_relay is None: if self._tcp_relay is None:
return return