p15670423
/
monkey
1
0
Fork 5
Code Issues Pull Requests 3 Projects Releases Wiki Activity

Agent: Add RetrievalError to i_agent_binary_repository.py

This commit is contained in:
Shreya Malviya 2022-09-20 18:30:21 +05:30
parent e8ecaa2169
commit 0e9397b283
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
monkey/infection_monkey/exploit/i_agent_binary_repository.py
View File

@ -7,6 +7,12 @@ from common import OperatingSystem
# moment, the Island and Agent have different needs, but at some point we should unify these.
class RetrievalError(RuntimeError):
"""
Raised when a repository encounters an error while attempting to retrieve data
"""
class IAgentBinaryRepository(metaclass=abc.ABCMeta):
"""
IAgentBinaryRepository provides an interface for other components to access agent binaries.
@ -23,5 +29,6 @@ class IAgentBinaryRepository(metaclass=abc.ABCMeta):
:param operating_system: The name of the operating system on which the agent binary will run
:param architecture: Reserved
:return: A file-like object for the requested agent binary
:raises RetrievalError: If an error occurs when retrieving the agent binary
"""
pass