Island: Add /api/agents endpoint

CHANGELOG.md

@@ -23,6 +23,7 @@ Changelog](https://keepachangelog.com/en/1.0.0/).
 - `/api/events` endpoint. #2155
 - The ability to customize the file extension used by ransomware when
   encrypting files. #1242
+- `/api/agents` endpoint.
 
 ### Changed
 - Reset workflow. Now it's possible to delete data gathered by agents without

monkey/monkey_island/cc/app.py

@@ -13,6 +13,7 @@ from monkey_island.cc.database import database, mongo
 from monkey_island.cc.resources import (
     AgentBinaries,
     AgentConfiguration,
+    Agents,
     ClearSimulationData,
     Events,
     IPAddresses,
@@ -162,6 +163,7 @@ def init_restful_endpoints(api: FlaskDIWrapper):
     api.add_resource(Register)
     api.add_resource(RegistrationStatus)
     api.add_resource(Authenticate)
+    api.add_resource(Agents)
     api.add_resource(Monkey)
     api.add_resource(LocalRun)
     api.add_resource(Telemetry)

monkey/monkey_island/cc/resources/__init__.py

@@ -9,3 +9,4 @@ from .agent_configuration import AgentConfiguration
 from .pba_file_upload import PBAFileUpload, LINUX_PBA_TYPE, WINDOWS_PBA_TYPE
 from .pba_file_download import PBAFileDownload
 from .events import Events
+from .agents import Agents

monkey/monkey_island/cc/resources/agents.py

@@ -0,0 +1,22 @@
+import json
+from http import HTTPStatus
+
+from flask import make_response, request
+
+from common import AgentRegistrationData
+from monkey_island.cc.resources.AbstractResource import AbstractResource
+
+
+class Agents(AbstractResource):
+    urls = ["/api/agents"]
+
+    def post(self):
+        try:
+            # Just parse for now
+            AgentRegistrationData(**request.json)
+            return make_response({}, HTTPStatus.NO_CONTENT)
+        except (TypeError, ValueError, json.JSONDecodeError) as err:
+            return make_response(
+                {"error": f"Invalid configuration supplied: {err}"},
+                HTTPStatus.BAD_REQUEST,
+            )

monkey/tests/unit_tests/monkey_island/cc/resources/test_agents.py

@@ -0,0 +1,42 @@
+from http import HTTPStatus
+from uuid import UUID
+
+from tests.unit_tests.monkey_island.conftest import get_url_for_resource
+
+from monkey_island.cc.resources import Agents
+
+AGENTS_URL = get_url_for_resource(Agents)
+
+AGENT_REGISTRATION_DICT = {
+    "id": UUID("6bfd8b64-43d8-4449-8c70-d898aca74ad8"),
+    "machine_hardware_id": 1,
+    "start_time": 0,
+    "parent_id": UUID("9d55ba33-95c2-417d-bd86-d3d11e47daeb"),
+    "cc_server": "10.0.0.1:5000",
+    "network_interfaces": ["10.1.1.2/24"],
+}
+
+
+def test_agent_registration(flask_client):
+    print(AGENTS_URL)
+    resp = flask_client.post(
+        AGENTS_URL,
+        json=AGENT_REGISTRATION_DICT,
+        follow_redirects=True,
+    )
+
+    assert resp.status_code == HTTPStatus.NO_CONTENT
+
+
+def test_agent_registration_invalid_data(flask_client):
+    agent_registration_dict = AGENT_REGISTRATION_DICT.copy()
+
+    agent_registration_dict["id"] = 1
+
+    resp = flask_client.post(
+        AGENTS_URL,
+        json=agent_registration_dict,
+        follow_redirects=True,
+    )
+
+    assert resp.status_code == HTTPStatus.BAD_REQUEST