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

Island: Inject local_ip_addresses into LocalMonkeyRunService's constructor

This commit is contained in:
Shreya Malviya 2022-08-04 12:59:18 +05:30
parent e4d7f36985
commit 80c6ee8805
1 changed files with 9 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
monkey/monkey_island/cc/services/run_local_monkey.py
View File

@ -4,10 +4,10 @@ import stat
import subprocess import subprocess
from pathlib import Path from pathlib import Path
from shutil import copyfileobj from shutil import copyfileobj
from typing import Sequence
from monkey_island.cc.repository import IAgentBinaryRepository, RetrievalError from monkey_island.cc.repository import IAgentBinaryRepository, RetrievalError
from monkey_island.cc.server_utils.consts import ISLAND_PORT from monkey_island.cc.server_utils.consts import ISLAND_PORT
from monkey_island.cc.services.utils.network_utils import get_local_ip_addresses
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
@ -15,9 +15,15 @@ AGENT_NAMES = {"linux": "monkey-linux-64", "windows": "monkey-windows-64.exe"}
class LocalMonkeyRunService: class LocalMonkeyRunService:
def __init__(self, data_dir: Path, agent_binary_repository: IAgentBinaryRepository): def __init__(
self,
data_dir: Path,
agent_binary_repository: IAgentBinaryRepository,
local_ip_addresses: Sequence[str],
):
self._data_dir = data_dir self._data_dir = data_dir
self._agent_binary_repository = agent_binary_repository self._agent_binary_repository = agent_binary_repository
self._local_ips = local_ip_addresses
def run_local_monkey(self): def run_local_monkey(self):
# get the monkey executable suitable to run on the server # get the monkey executable suitable to run on the server
@ -59,7 +65,7 @@ class LocalMonkeyRunService:
# run the monkey # run the monkey
try: try:
ip = get_local_ip_addresses()[0] ip = self._local_ips[0]
port = ISLAND_PORT port = ISLAND_PORT
args = [str(dest_path), "m0nk3y", "-s", f"{ip}:{port}"] args = [str(dest_path), "m0nk3y", "-s", f"{ip}:{port}"]