From 50ca81f0fc3996480f98b7c851a5e5e72790dfd3 Mon Sep 17 00:00:00 2001
From: Mike Salvatore <mike.s.salvatore@gmail.com>
Date: Mon, 28 Feb 2022 15:27:59 -0500
Subject: [PATCH] Agent: Add IAgentRepository

---
 monkey/infection_monkey/exploit/__init__.py   |  1 +
 .../exploit/i_agent_repository.py             | 21 +++++++++++++++++++
 2 files changed, 22 insertions(+)
 create mode 100644 monkey/infection_monkey/exploit/i_agent_repository.py

diff --git a/monkey/infection_monkey/exploit/__init__.py b/monkey/infection_monkey/exploit/__init__.py
index 42d8d18bf..105d7947e 100644
--- a/monkey/infection_monkey/exploit/__init__.py
+++ b/monkey/infection_monkey/exploit/__init__.py
@@ -1 +1,2 @@
+from .i_agent_repository import IAgentRepository
 from .exploiter_wrapper import ExploiterWrapper
diff --git a/monkey/infection_monkey/exploit/i_agent_repository.py b/monkey/infection_monkey/exploit/i_agent_repository.py
new file mode 100644
index 000000000..f63ca4038
--- /dev/null
+++ b/monkey/infection_monkey/exploit/i_agent_repository.py
@@ -0,0 +1,21 @@
+import abc
+import io
+
+
+class IAgentRepository(metaclass=abc.ABCMeta):
+    """
+    IAgentRepository provides an interface for other components to access agent binaries. Notably,
+    this is used by exploiters during propagation to retrieve the appropriate agent binary so that
+    it can be uploaded to a victim and executed.
+    """
+
+    @abc.abstractmethod
+    def get_agent_binary(self, os: str, architecture: str = None) -> io.BytesIO:
+        """
+        Retrieve the appropriate agent binary from the repository.
+        :param str os: The name of the operating system on which the agent binary will run
+        :param str architecture: Reserved
+        :return: A file-like object for the requested agent binary
+        :rtype: io.BytesIO
+        """
+        pass