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

Agent: Fix get_target_monkey() bug when running from source

This commit is contained in:
Mike Salvatore 2022-02-28 08:29:04 -05:00
parent 01a21f744f
commit d970271016
1 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
monkey/infection_monkey/exploit/tools/helpers.py
View File

@ -1,4 +1,5 @@
import logging import logging
from pathlib import Path
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
@ -20,7 +21,15 @@ def get_target_monkey(host):
return None return None
if host.os.get("type") == platform.system().lower(): if host.os.get("type") == platform.system().lower():
try:
# When running from source, sys.executable will be "python", not an agent.
if "monkey" in Path(sys.executable).name:
return sys.executable return sys.executable
except Exception as ex:
logger.warning(
"Unable to retrieve this executable's path, downloading executable from the island "
f"instead: {ex}"
)
return ControlClient.download_monkey_exe(host) return ControlClient.download_monkey_exe(host)