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

Agent: Pass agent_id to ControlChannel constructor

This commit is contained in:
Mike Salvatore 2021-12-01 12:18:32 -05:00
parent 1944040328
commit f074b3e388
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
monkey/infection_monkey/master/control_channel.py
View File

@ -4,7 +4,7 @@ import logging
import requests import requests
from common.common_consts.timeouts import SHORT_REQUEST_TIMEOUT from common.common_consts.timeouts import SHORT_REQUEST_TIMEOUT
from infection_monkey.config import GUID, WormConfiguration from infection_monkey.config import WormConfiguration
from infection_monkey.control import ControlClient from infection_monkey.control import ControlClient
from monkey.infection_monkey.i_control_channel import IControlChannel from monkey.infection_monkey.i_control_channel import IControlChannel
@ -14,7 +14,8 @@ logger = logging.getLogger(__name__)
class ControlChannel(IControlChannel): class ControlChannel(IControlChannel):
def __init__(self, server: str): def __init__(self, server: str, agent_id: str):
self._agent_id = agent_id
self._control_channel_server = server self._control_channel_server = server
def should_agent_stop(self) -> bool: def should_agent_stop(self) -> bool:
@ -23,7 +24,7 @@ class ControlChannel(IControlChannel):
try: try:
response = requests.get( # noqa: DUO123 response = requests.get( # noqa: DUO123
f"{self._control_channel_server}/api/monkey_control/{GUID}", f"{self._control_channel_server}/api/monkey_control/{self._agent_id}",
verify=False, verify=False,
timeout=SHORT_REQUEST_TIMEOUT, timeout=SHORT_REQUEST_TIMEOUT,
) )