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

Agent: Add logging in CachingAgentBinaryRepository._download_binary_from_island()

This commit is contained in:
Shreya Malviya 2022-09-20 15:18:03 +05:30
parent 3b3f39065d
commit 4605722874
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
monkey/infection_monkey/exploit/caching_agent_binary_repository.py
View File

@ -1,4 +1,5 @@
import io import io
import logging
import threading import threading
from functools import lru_cache from functools import lru_cache
@ -7,6 +8,8 @@ from infection_monkey.island_api_client import IIslandAPIClient
from . import IAgentBinaryRepository from . import IAgentBinaryRepository
logger = logging.getLogger(__name__)
class CachingAgentBinaryRepository(IAgentBinaryRepository): class CachingAgentBinaryRepository(IAgentBinaryRepository):
""" """
@ -32,4 +35,7 @@ class CachingAgentBinaryRepository(IAgentBinaryRepository):
@lru_cache(maxsize=None) @lru_cache(maxsize=None)
def _download_binary_from_island(self, operating_system: OperatingSystem) -> bytes: def _download_binary_from_island(self, operating_system: OperatingSystem) -> bytes:
os_name = operating_system.value os_name = operating_system.value
return self._island_api_client.get_agent_binary(os_name) try:
return self._island_api_client.get_agent_binary(os_name)
except Exception as exc:
logger.warning(f"Error connecting to control server: {exc}")