Remove old Agent registration logic

CHANGELOG.md

@@ -142,6 +142,7 @@ Changelog](https://keepachangelog.com/en/1.0.0/).
 - "POST /api/log" endpoint. #2485
 - "GET /api/local-monkey" endpoint. #2506
 - "/api/telemetry" endpoint. #2503
+- "/api/agent" endpoint. #2542
 
 ### Fixed
 - A bug in network map page that caused delay of telemetry log loading. #1545

monkey/infection_monkey/control.py

@@ -1,15 +1,8 @@
-import json
 import logging
-from socket import gethostname
 
-import requests
 from urllib3 import disable_warnings
 
-from common.common_consts.timeouts import MEDIUM_REQUEST_TIMEOUT
-from common.network.network_utils import get_my_ip_addresses_legacy
 from common.types import SocketAddress
-from infection_monkey.config import GUID
-from infection_monkey.utils import agent_process
 
 disable_warnings()  # noqa DUO131
 
@@ -19,27 +12,3 @@
 class ControlClient:
     def __init__(self, server_address: SocketAddress):
         self.server_address = server_address
-
-    def wakeup(self, parent=None):
-        if parent:
-            logger.debug("parent: %s" % (parent,))
-
-        hostname = gethostname()
-        if not parent:
-            parent = GUID
-
-        monkey = {
-            "guid": GUID,
-            "hostname": hostname,
-            "ip_addresses": get_my_ip_addresses_legacy(),
-            "parent": parent,
-            "launch_time": agent_process.get_start_time(),
-        }
-
-        requests.post(  # noqa: DUO123
-            f"https://{self.server_address}/api/agent",
-            data=json.dumps(monkey),
-            headers={"content-type": "application/json"},
-            verify=False,
-            timeout=MEDIUM_REQUEST_TIMEOUT,
-        )

monkey/infection_monkey/monkey.py

@@ -182,8 +182,6 @@ def start(self):
         logger.info("Agent is starting...")
         logger.info(f"Agent GUID: {GUID}")
 
-        self._control_client.wakeup(parent=self._opts.parent)
-
         should_stop = self._control_channel.should_agent_stop()
         if should_stop:
             logger.info("The Monkey Island has instructed this agent to stop")

monkey/monkey_island/cc/app.py

@@ -34,7 +34,6 @@
 from monkey_island.cc.resources.exploitations.monkey_exploitation import MonkeyExploitation
 from monkey_island.cc.resources.island_mode import IslandMode
 from monkey_island.cc.resources.local_run import LocalRun
-from monkey_island.cc.resources.monkey import Monkey
 from monkey_island.cc.resources.ransomware_report import RansomwareReport
 from monkey_island.cc.resources.root import Root
 from monkey_island.cc.resources.security_report import SecurityReport
@@ -150,7 +149,6 @@ def init_restful_endpoints(api: FlaskDIWrapper):
     api.add_resource(RegistrationStatus)
     api.add_resource(Authenticate)
     api.add_resource(Agents)
-    api.add_resource(Monkey)
     api.add_resource(LocalRun)
 
     api.add_resource(IslandMode)

monkey/monkey_island/cc/models/__init__.py

@@ -1,7 +1,6 @@
 # Order of importing matters here, for registering the embedded and referenced documents before
 # using them.
 from .monkey import Monkey
-from .monkey_ttl import MonkeyTtl
 from monkey_island.cc.models.report.report import Report
 from .simulation import Simulation, IslandMode
 from .user_credentials import UserCredentials

monkey/monkey_island/cc/models/monkey.py

@@ -2,23 +2,16 @@
 Define a Document Schema for the Monkey document.
 
 """
-import ring
 from mongoengine import (
-    BooleanField,
     DateTimeField,
     Document,
-    DoesNotExist,
     DynamicField,
     FloatField,
     ListField,
     ReferenceField,
     StringField,
 )
 
-from common.network.network_utils import get_my_ip_addresses_legacy
-from monkey_island.cc.models.monkey_ttl import MonkeyTtl, create_monkey_ttl_document
-from monkey_island.cc.server_utils.consts import DEFAULT_MONKEY_TTL_EXPIRY_DURATION_IN_SECONDS
-
 
 class Monkey(Document):
     """
@@ -31,8 +24,6 @@ class Monkey(Document):
 
     # SCHEMA
     guid = StringField(required=True)
-    should_stop = BooleanField()
-    dead = BooleanField()
     hostname = StringField()
     ip_addresses = ListField(StringField())
     launch_time = FloatField()
@@ -43,70 +34,15 @@ class Monkey(Document):
     # (even with required=False of null=True).
     # See relevant issue: https://github.com/MongoEngine/mongoengine/issues/1904
     parent = ListField(ListField(DynamicField()))
-    ttl_ref = ReferenceField(MonkeyTtl)
     tunnel = ReferenceField("self")
 
     # This field only exists when the monkey is running on an AWS
     aws_instance_id = StringField(required=False)
 
     # instance. See https://github.com/guardicore/monkey/issues/426.
 
-    # LOGIC
-    @staticmethod
-    def get_single_monkey_by_id(db_id):
-        try:
-            return Monkey.objects.get(id=db_id)
-        except DoesNotExist as ex:
-            raise MonkeyNotFoundError("info: {0} | id: {1}".format(ex, str(db_id)))
-
-    @staticmethod
-    # See https://www.python.org/dev/peps/pep-0484/#forward-references
-    def get_single_monkey_by_guid(monkey_guid) -> "Monkey":
-        try:
-            return Monkey.objects.get(guid=monkey_guid)
-        except DoesNotExist as ex:
-            raise MonkeyNotFoundError("info: {0} | guid: {1}".format(ex, str(monkey_guid)))
-
     @staticmethod
     def get_latest_modifytime():
         if Monkey.objects.count() > 0:
             return Monkey.objects.order_by("-modifytime").first().modifytime
         return None
-
-    @ring.lru()
-    @staticmethod
-    def get_label_by_id(object_id):
-        current_monkey = Monkey.get_single_monkey_by_id(object_id)
-        label = Monkey.get_hostname_by_id(object_id) + " : " + current_monkey.ip_addresses[0]
-        local_ips = map(str, get_my_ip_addresses_legacy())
-        if len(set(current_monkey.ip_addresses).intersection(local_ips)) > 0:
-            label = "MonkeyIsland - " + label
-        return label
-
-    @ring.lru()
-    @staticmethod
-    def get_hostname_by_id(object_id):
-        """
-        :param object_id: the object ID of a Monkey in the database.
-        :return: The hostname of that machine.
-        :note: Use this and not monkey.hostname for performance - this is lru-cached.
-        """
-        return Monkey.get_single_monkey_by_id(object_id).hostname
-
-    # data has TTL of 1 second. This is useful for rapid calls for report generation.
-    @ring.lru(expire=1)
-    @staticmethod
-    def is_monkey(object_id):
-        try:
-            _ = Monkey.get_single_monkey_by_id(object_id)
-            return True
-        except:  # noqa: E722
-            return False
-
-    def renew_ttl(self, duration=DEFAULT_MONKEY_TTL_EXPIRY_DURATION_IN_SECONDS):
-        self.ttl_ref = create_monkey_ttl_document(duration)
-        self.save()
-
-
-class MonkeyNotFoundError(Exception):
-    pass

monkey/monkey_island/cc/server_utils/consts.py

@@ -26,8 +26,6 @@ def _get_monkey_island_abs_path() -> str:
 
 DEFAULT_DATA_DIR = expand_path(get_default_data_dir())
 
-DEFAULT_MONKEY_TTL_EXPIRY_DURATION_IN_SECONDS = 60 * 5
-
 _MONGO_BINARY_DIR = os.path.join(MONKEY_ISLAND_ABS_PATH, "bin", "mongodb")
 _MONGO_EXECUTABLE_PATH_WIN = os.path.join(_MONGO_BINARY_DIR, "mongod.exe")
 _MONGO_EXECUTABLE_PATH_LINUX = os.path.join(_MONGO_BINARY_DIR, "bin", "mongod")