From 7766e27f16083d345cae5b30bbe5a3e84a965b39 Mon Sep 17 00:00:00 2001
From: Ilija Lazoroski <ilija.la@live.com>
Date: Mon, 22 Nov 2021 13:46:09 +0100
Subject: [PATCH] Island: Add mock endpoint to check if the agent should stop

---
 monkey/monkey_island/cc/app.py                           | 2 ++
 .../cc/resources/monkey_control/stop_agent_check.py      | 9 +++++++++
 2 files changed, 11 insertions(+)
 create mode 100644 monkey/monkey_island/cc/resources/monkey_control/stop_agent_check.py

diff --git a/monkey/monkey_island/cc/app.py b/monkey/monkey_island/cc/app.py
index 7ea91c0db..113c20d06 100644
--- a/monkey/monkey_island/cc/app.py
+++ b/monkey/monkey_island/cc/app.py
@@ -33,6 +33,7 @@ from monkey_island.cc.resources.monkey import Monkey
 from monkey_island.cc.resources.monkey_configuration import MonkeyConfiguration
 from monkey_island.cc.resources.monkey_control.remote_port_check import RemotePortCheck
 from monkey_island.cc.resources.monkey_control.started_on_island import StartedOnIsland
+from monkey_island.cc.resources.monkey_control.stop_agent_check import StopAgentCheck
 from monkey_island.cc.resources.monkey_download import MonkeyDownload
 from monkey_island.cc.resources.netmap import NetMap
 from monkey_island.cc.resources.node import Node
@@ -168,6 +169,7 @@ def init_api_resources(api):
     api.add_resource(VersionUpdate, "/api/version-update", "/api/version-update/")
     api.add_resource(RemotePortCheck, "/api/monkey_control/check_remote_port/<string:port>")
     api.add_resource(StartedOnIsland, "/api/monkey_control/started_on_island")
+    api.add_resource(StopAgentCheck, "/api/monkey_control/<int:monkey_guid>")
     api.add_resource(ScoutSuiteAuth, "/api/scoutsuite_auth/<string:provider>")
     api.add_resource(AWSKeys, "/api/aws_keys")
 
diff --git a/monkey/monkey_island/cc/resources/monkey_control/stop_agent_check.py b/monkey/monkey_island/cc/resources/monkey_control/stop_agent_check.py
new file mode 100644
index 000000000..817d6db94
--- /dev/null
+++ b/monkey/monkey_island/cc/resources/monkey_control/stop_agent_check.py
@@ -0,0 +1,9 @@
+import flask_restful
+
+
+class StopAgentCheck(flask_restful.Resource):
+    def get(self, monkey_guid: int):
+        if monkey_guid % 2:
+            return {"stop_agent": True}
+        else:
+            return {"stop_agent": False}