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

Island: Reduce code duplication in agent_binary_repository.py

This commit is contained in:
vakarisz 2022-06-08 11:31:12 +03:00
parent e7459412e4
commit df6b1cd4d3
1 changed files with 8 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
monkey/monkey_island/cc/repository/agent_binary_repository.py
View File

@ -10,20 +10,18 @@ class AgentBinaryRepository(IAgentBinaryRepository):
def __init__(self, file_repository: IFileRepository): def __init__(self, file_repository: IFileRepository):
self._file_repository = file_repository self._file_repository = file_repository
def get_linux_binary(self) -> BinaryIO: def __get_binary(self, filename) -> BinaryIO:
try: try:
agent_binary = self._file_repository.open_file(LINUX_AGENT_FILE_NAME) agent_binary = self._file_repository.open_file(filename)
return agent_binary return agent_binary
except FileRetrivalError as err: except FileRetrivalError as err:
raise AgentRetrivalError( raise AgentRetrivalError(
f"An error occurred while retrieving the Linux agent binary: {err}" f"An error occurred while retrieving the {filename}"
f" agent binary from {self._file_repository}: {err}"
) )
def get_linux_binary(self) -> BinaryIO:
self.__get_binary(LINUX_AGENT_FILE_NAME)
def get_windows_binary(self) -> BinaryIO: def get_windows_binary(self) -> BinaryIO:
try: self.__get_binary(WINDOWS_AGENT_FILE_NAME)
agent_binary = self._file_repository.open_file(WINDOWS_AGENT_FILE_NAME)
return agent_binary
except FileRetrivalError as err:
raise AgentRetrivalError(
f"An error occurred while retrieving the Windows agent binary: {err}"
)