Island: Add /api/clear-simulation-data

monkey/monkey_island/cc/app.py

@@ -10,7 +10,12 @@ from werkzeug.exceptions import NotFound
 
 from common import DIContainer
 from monkey_island.cc.database import database, mongo
-from monkey_island.cc.resources import AgentBinaries, RemoteRun, ResetAgentConfiguration
+from monkey_island.cc.resources import (
+    AgentBinaries,
+    ClearSimulationData,
+    RemoteRun,
+    ResetAgentConfiguration,
+)
 from monkey_island.cc.resources.AbstractResource import AbstractResource
 from monkey_island.cc.resources.agent_configuration import AgentConfiguration
 from monkey_island.cc.resources.agent_controls import StopAgentCheck, StopAllAgents
@@ -199,6 +204,7 @@ def init_restful_endpoints(api: FlaskDIWrapper):
 
 def init_rpc_endpoints(api: FlaskDIWrapper):
     api.add_resource(ResetAgentConfiguration)
+    api.add_resource(ClearSimulationData)
 
 
 def init_app(mongo_url: str, container: DIContainer):

monkey/monkey_island/cc/resources/__init__.py

@@ -1,3 +1,4 @@
 from .remote_run import RemoteRun
 from .agent_binaries import AgentBinaries
+from .clear_simulation_data import ClearSimulationData
 from .reset_agent_configuration import ResetAgentConfiguration

monkey/monkey_island/cc/resources/clear_simulation_data.py

@@ -0,0 +1,18 @@
+from flask import make_response
+
+from monkey_island.cc.resources.AbstractResource import AbstractResource
+from monkey_island.cc.resources.request_authentication import jwt_required
+from monkey_island.cc.services.database import Database
+
+
+class ClearSimulationData(AbstractResource):
+    urls = ["/api/clear-simulation-data"]
+
+    @jwt_required
+    def post(self):
+        """
+        Clear all data collected during the simulation
+        """
+        Database.reset_db(reset_config=False)
+
+        return make_response({}, 200)