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

Island: Change agent log file name in FileAgentLogRepository

This commit is contained in:
Shreya Malviya 2022-09-28 18:29:53 +05:30
parent 30d3124cb4
commit c5d26749b7
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
monkey/monkey_island/cc/repository/file_agent_log_repository.py
View File

@ -9,7 +9,7 @@ from monkey_island.cc.repository import (
UnknownRecordError, UnknownRecordError,
) )
AGENT_LOG_FILE_NAME_PREFIX = "agent_log_" AGENT_LOG_FILE_NAME_PATTERN = "agent-*.log"
class FileAgentLogRepository(IAgentLogRepository): class FileAgentLogRepository(IAgentLogRepository):
@ -18,12 +18,14 @@ class FileAgentLogRepository(IAgentLogRepository):
def upsert_agent_log(self, agent_id: AgentID, log_contents: str): def upsert_agent_log(self, agent_id: AgentID, log_contents: str):
self._file_repository.save_file( self._file_repository.save_file(
f"{AGENT_LOG_FILE_NAME_PREFIX}{agent_id}", io.BytesIO(log_contents.encode()) AGENT_LOG_FILE_NAME_PATTERN.replace("*", agent_id), io.BytesIO(log_contents.encode())
) )
def get_agent_log(self, agent_id: AgentID) -> str: def get_agent_log(self, agent_id: AgentID) -> str:
try: try:
with self._file_repository.open_file(f"{AGENT_LOG_FILE_NAME_PREFIX}{agent_id}") as f: with self._file_repository.open_file(
AGENT_LOG_FILE_NAME_PATTERN.replace("*", agent_id)
) as f:
log_contents = f.read().decode() log_contents = f.read().decode()
return log_contents return log_contents
except repository.FileNotFoundError as err: except repository.FileNotFoundError as err:
@ -32,4 +34,4 @@ class FileAgentLogRepository(IAgentLogRepository):
raise RetrievalError(f"Error retrieving the agent logs: {err}") raise RetrievalError(f"Error retrieving the agent logs: {err}")
def reset(self): def reset(self):
self._file_repository.delete_files_by_pattern(f"{AGENT_LOG_FILE_NAME_PREFIX}*") self._file_repository.delete_files_by_pattern(f"{AGENT_LOG_FILE_NAME_PATTERN}")